ShortStructSequence2DWriter.java

package org.esa.beam.smos.ee2netcdf.variable;

import com.bc.ceres.binio.CompoundData;
import com.bc.ceres.binio.SequenceData;
import org.esa.beam.dataio.netcdf.nc.NVariable;
import ucar.ma2.Array;
import ucar.ma2.Index;

import java.io.IOException;

class ShortStructSequence2DWriter extends AbstractVariableWriter {

    private final int memberIndex;

    ShortStructSequence2DWriter(NVariable variable, int width, int height, int memberIndex, short fillValue) {
        this.variable = variable;
        this.memberIndex = memberIndex;
        final short[][] shortArray = VariableHelper.getShortArray(width, height, fillValue);
        array = Array.factory(shortArray);
    }

    @Override
    public void write(CompoundData gridPointData, SequenceData btDataList, int index) throws IOException {
        final Index arrayIndex = array.getIndex();
        final int size = btDataList.getElementCount();
        for (int i = 0; i < size; i++) {
            final short data = btDataList.getCompound(i).getShort(memberIndex);
            arrayIndex.set(index, i);
            array.setShort(arrayIndex, data);
        }
    }
}