Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.gdal.provider;

import java.io.IOException;

import java.nio.Buffer;

import java.nio.ByteBuffer;

import java.nio.DoubleBuffer;

import java.nio.FloatBuffer;

import java.nio.IntBuffer;

import java.nio.ShortBuffer;

import java.util.ArrayList;

import java.util.Hashtable;

import java.util.LinkedHashMap;

import java.util.Map;

import java.util.Set;

import org.gdal.gdal.Band;

import org.gdal.gdal.Dataset;

import org.gvsig.raster.lib.buffer.api.BandInfo;

import org.gvsig.raster.lib.buffer.api.BandPageManager;

import org.gvsig.raster.lib.buffer.api.BufferLocator;

import org.gvsig.raster.lib.buffer.api.BufferManager;

/**

 * Band Page Manager for a Memory Image

 * 

 * @author dmartinezizquierdo

 *

 */

public class RasterGdalBandPageManager implements BandPageManager {

    private Band gdalBand;

    private Dataset gdalDataSet;

    /**

     * @param gdalDataSet

     * @param bandNumber

     */

    public RasterGdalBandPageManager(Dataset gdalDataSet, int bandNumber) {

        this.gdalDataSet = gdalDataSet;

        this.gdalBand = gdalDataSet.GetRasterBand(bandNumber);

    }

    @Override

    public boolean isSupportedSave() {

        return false;

    }

    @Override

    public void load(Buffer buffer, int firstRow, int numberRows, int dataType) throws IOException {

        int xOffset = 0;

        int xSize = gdalDataSet.getRasterXSize();

        int yOffset = firstRow;

        int ySize = numberRows;

        switch (dataType) {

        case BufferManager.TYPE_INT:

            gdalBand.ReadRaster(xOffset, yOffset, xSize, ySize, ((IntBuffer) buffer).array());

            break;

        case BufferManager.TYPE_BYTE:

            gdalBand.ReadRaster(xOffset, yOffset, xSize, ySize, ((ByteBuffer) buffer).array());

            break;

        case BufferManager.TYPE_DOUBLE:

            gdalBand.ReadRaster(xOffset, yOffset, xSize, ySize, ((DoubleBuffer) buffer).array());

            break;

        case BufferManager.TYPE_FLOAT:

            gdalBand.ReadRaster(xOffset, yOffset, xSize, ySize, ((FloatBuffer) buffer).array());

            break;

        case BufferManager.TYPE_SHORT:

        case BufferManager.TYPE_USHORT:

            gdalBand.ReadRaster(xOffset, yOffset, xSize, ySize, ((ShortBuffer) buffer).array());

            break;

        }

    }

    @Override

    public void save(Buffer buffer, int row, int rows, int dataType) throws IOException {

        throw new UnsupportedOperationException();

    }

    @Override

    public BandInfo getBandInfo() {

        //TODO

        BufferManager bufferManager = BufferLocator.getBufferManager();

        String description = gdalBand.GetDescription();

        Map<Object, Object> values = new LinkedHashMap<Object, Object>();

        values.put("X size", gdalBand.getXSize());

        values.put("Y size", gdalBand.getYSize());

        values.put("Data type",

            RasterGdalStoreProvider.getRasterBufTypeFromGdalType(gdalBand.getDataType()));

        values.put("NoData value", this.getNoData(gdalBand));

        values.put("Block X size", gdalBand.GetBlockXSize());

        values.put("Block Y size", gdalBand.GetBlockYSize());

        values.put("Unit type", gdalBand.GetUnitType());

        Hashtable metadataDict = gdalBand.GetMetadata_Dict();

        Set keySet = metadataDict.keySet();

        for (Object object : keySet) {

            values.put(object, metadataDict.get(keySet));

        }

        return bufferManager.createBandInfo(gdalBand.GetBand(), String.valueOf(gdalBand.GetBand()),

            description, new ArrayList<Map.Entry<Object, Object>>(values.entrySet()));

    }

    private Object getNoData(Band gdalBand2) {

        // TODO Auto-generated method stub

        return null;

    }

}