Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.gdal.provider;

import java.io.File;

import java.io.IOException;

import org.cresques.cts.ICRSFactory;

import org.cresques.cts.IProjection;

import org.gdal.gdal.Dataset;

import org.gdal.gdal.Driver;

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

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

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

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

import org.gvsig.tools.dynobject.DynObject;

/**

 * Raster GDAL store provider that stores new raster through create method

 * @author dmartinezizquierdo

 *

 */

public class RasterGdalStoreProviderCreate extends DefaultRasterGdalStoreProvider{

    protected RasterGdalStoreProviderCreate(RasterGdalFileStoreParameters params,

        DataStoreProviderServices storeServices, DynObject metadata)

        throws InitializeException {

        super(params, storeServices, metadata);

    }

    /**

     * Constructor

     * @param params

     * @param storeServices

     * @throws InitializeException

     */

    public RasterGdalStoreProviderCreate(RasterGdalFileStoreParameters params,

        DataStoreProviderServices storeServices) throws InitializeException {

        super(params, storeServices);

    }

    protected RasterGdalStoreProviderCreate(NewRasterGdalStoreParameters params,

        Driver gdalDriver) {

        super(params, gdalDriver);

    }

    @Override

    public void store(boolean overwrite)

        throws CreateException {

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

        File file=newRasterParams.getFile();

        Buffer buffer=newRasterParams.getBuffer();

        String[] options = getStoringOptions(newRasterParams);

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

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

        }

        Dataset gdalDataset = null;

        try {

            if (options.length == 0) {

                gdalDataset =

                    gdalDriver.Create(file.getAbsolutePath(), buffer.getColumns(), buffer.getRows(),

                        buffer.getBandCount());

            } else {

                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));

            fillDataset(buffer, gdalDataset);

            gdalDataset.FlushCache();

        } finally {

            if (gdalDataset != null) {

                gdalDataset.delete();

            }

        }

        //Create WLD

        try {

            createWorldFile(file,  buffer);

        } catch (IOException e) {

            throw new CreateException("WLD File", e);

        }

    }

}