Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.gdal.provider;

import java.io.File;

import java.util.ArrayList;

import org.cresques.cts.ICRSFactory;

import org.cresques.cts.IProjection;

import org.gdal.gdal.Dataset;

import org.gdal.gdal.Driver;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.fmap.dal.exception.CreateException;

import org.gvsig.fmap.dal.exception.InitializeException;

import org.gvsig.fmap.dal.spi.DataStoreProviderServices;

import org.gvsig.fmap.geom.DataTypes;

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

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

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

import org.gvsig.tools.dynobject.DynClass;

import org.gvsig.tools.dynobject.DynField;

import org.gvsig.tools.dynobject.DynObject;

//FIXME: THIS PROVIDER IS NOT IN USE. IT WAS IMPLEMENTED AS AN EXAMPLE AS HOW IT SHOULD BE

/**

 * Class not in use. Generic provider is used. But in a future could be useful to use his own provider

 * @author dmartinezizquierdo

 *

 */

public class RasterGdalGTiffStoreProvider extends RasterGdalStoreProviderCreate {

    private static final Logger logger = LoggerFactory.getLogger(RasterGdalGTiffStoreProvider.class);

    protected RasterGdalGTiffStoreProvider(RasterGdalFileStoreParameters params,

        DataStoreProviderServices storeServices, DynObject metadata)

        throws InitializeException {

        super(params, storeServices, metadata);

    }

    /**

     * Creates the provider to be readed

     * @param params

     * @param storeServices

     * @throws InitializeException

     */

    public RasterGdalGTiffStoreProvider(RasterGdalFileStoreParameters params,

        DataStoreProviderServices storeServices) throws InitializeException {

        super(params, storeServices);

    }

    /**

     * Creates the provider to be written

     * @param params

     * @param gdalDriver

     * @throws InitializeException

     */

    public RasterGdalGTiffStoreProvider(NewRasterGdalStoreParameters params, Driver gdalDriver) throws InitializeException {

        super(params,gdalDriver);

    }

    /**

     * Stores a buffer into a new file

     * @param newRasterParams

     * @param overwrite

     * @throws CreateException

     */

    public void store(boolean overwrite)

        throws CreateException {

        NewRasterGdalStoreParameters newRasterParams = (NewRasterGdalStoreParameters)this.getParameters();

        File file=newRasterParams.getFile();

        Buffer buffer=newRasterParams.getBuffer();

        //We get the specific options of GTiff

        //IMPORTANT: They must use the same name in the xml properties file that in GDAL

        ArrayList<String> optionsList=new ArrayList<String>();

        DynClass dynClass = newRasterParams.getDynClass();

        DynField[] fields = dynClass.getDynFields();

        for (DynField field:fields){

            String fieldName = field.getName();

            String gdalOption = fieldName.toUpperCase();

            Object dynValue = newRasterParams.getDynValue(fieldName);

            if (dynValue!=null && field.getGroup().equalsIgnoreCase(NewRasterGdalStoreParameters.CREATE_OPTIONS_GROUP)){

                if(field.getType() == DataTypes.BOOLEAN) {

                    if((Boolean)dynValue){

                        optionsList.add(gdalOption+"=YES");

                    }else{

                        optionsList.add(gdalOption+"=NO");

                    }

                } else {

                    optionsList.add(gdalOption+"="+dynValue);

                }

            }

        }

        if (file.exists() && !overwrite){

            throw new CreateException("File already exists: "+file.toString(), null);

        }

        String[]options=new String[optionsList.size()];

        optionsList.toArray(options);

        Dataset gdalDataset = gdalDriver.Create(

            file.getAbsolutePath(),

            buffer.getColumns(), buffer.getRows(),

            buffer.getBandCount(),

            options);

        IProjection projection = buffer.getProjection();

        if (projection!=null){

            String wkt=projection.export(ICRSFactory.FORMAT_WKT_ESRI);

            gdalDataset.SetProjection(wkt);

        }

        gdalDataset.SetGeoTransform(createGeoTransform(buffer));

        for (int i=0;i<buffer.getBandCount();i++){

            Band band = buffer.getBand(i);

            org.gdal.gdal.Band gdalBand = gdalDataset.GetRasterBand(i+1);

            switch (band.getDataType()) {

            case BufferManager.TYPE_BYTE:

                byte[] byteRowBuffer = (byte[])band.createRowBuffer(); //new byte[band.getColumns()];

                for (int row=0;row<band.getRows();row++){

                    band.fetchRow(row, byteRowBuffer);

                    gdalBand.WriteRaster(0, row, band.getColumns(), 1,

                        byteRowBuffer);

                }

                break;

            case BufferManager.TYPE_DOUBLE:

                double[] doubleRowBuffer=(double[])band.createRowBuffer();

                for (int row=0;row<band.getRows();row++){

                    band.fetchRow(row, doubleRowBuffer);

                    gdalBand.WriteRaster(0, row, band.getColumns(), 1,

                        doubleRowBuffer);

                }

                break;

            case BufferManager.TYPE_FLOAT:

                float[] floatRowBuffer=(float[])band.createRowBuffer();

                for (int row=0;row<band.getRows();row++){

                    band.fetchRow(row, floatRowBuffer);

                    gdalBand.WriteRaster(0, row, band.getColumns(), 1,

                        floatRowBuffer);

                }

                break;

            case BufferManager.TYPE_INT:

                int[] intRowBuffer=(int[])band.createRowBuffer();

                for (int row=0;row<band.getRows();row++){

                    band.fetchRow(row, intRowBuffer);

                    gdalBand.WriteRaster(0, row, band.getColumns(), 1,

                        intRowBuffer);

                }

                break;

            case BufferManager.TYPE_SHORT:

            case BufferManager.TYPE_USHORT:

                short[] shortRowBuffer=(short[])band.createRowBuffer();

                for (int row=0;row<band.getRows();row++){

                    band.fetchRow(row, shortRowBuffer);

                    gdalBand.WriteRaster(0, row, band.getColumns(), 1,

                        shortRowBuffer);

                }

                break;

            default:

                break;

            }

            if (gdalDataset!=null){

                gdalDataset.FlushCache();

                // from gdal FlushCache doc:

                /*

                Flush all write cached data to disk.

                Any raster (or other GDAL) data written via GDAL calls, but buffered internally will be written to disk.

                Calling this method is generally not sufficient to ensure that the file is in a consistent state. You must call delete() for that

                */

//                try {

//                    gdalDataset.wait();

//                } catch (InterruptedException e) {

//                    logger.warn("An error has ocurred waiting for the cache is flushed.", e);

//                }

                gdalDataset.delete();

            }

        }

    }

}