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.gdal.gdal.gdal;

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 createCopy method

 *

 * @author dmartinezizquierdo

 *

 */

public class RasterGdalStoreProviderCreateCopy extends DefaultRasterGdalStoreProvider {

    private final static String AUX_DRIVER_NAME = "MEM";

    protected RasterGdalStoreProviderCreateCopy(RasterGdalFileStoreParameters params,

        DataStoreProviderServices storeServices, DynObject metadata) throws InitializeException {

        super(params, storeServices, metadata);

    }

    /**

     * Constructor

     *

     * @param params

     * @param storeServices

     * @throws InitializeException

     */

    public RasterGdalStoreProviderCreateCopy(RasterGdalFileStoreParameters params,

        DataStoreProviderServices storeServices) throws InitializeException {

        super(params, storeServices);

    }

    protected RasterGdalStoreProviderCreateCopy(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;

        // Provider driver doesn't support create

        // So we create another raster with a driver that supports it

        // and get its copy with our Provider driver

        IProjection projection = buffer.getProjection();

        try {

            Driver auxDriver = gdal.GetDriverByName(AUX_DRIVER_NAME);

            Dataset auxDataset = auxDriver.Create("", buffer.getColumns(), buffer.getRows(), buffer.getBandCount());

            if (projection != null) {

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

                auxDataset.SetProjection(wkt);

            }

            auxDataset.SetGeoTransform(createGeoTransform(buffer));

            fillDataset(buffer, auxDataset);

            if (options.length == 0) {

                gdalDataset = gdalDriver.CreateCopy(file.getAbsolutePath(), auxDataset, 1);

            } else {

                gdalDataset = gdalDriver.CreateCopy(file.getAbsolutePath(), auxDataset, 1, options);

            }

            if(gdalDataset==null){

                throw new CreateException("Can't create gdalDataset: '" + gdal.GetLastErrorMsg()+"'", null);

            }

            gdalDataset.FlushCache();

        } finally {

            if (gdalDataset != null) {

                gdalDataset.delete();

            }

        }

        // Create WLD

        try {

            createWorldFile(file, buffer);

        } catch (IOException e) {

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

        }

        // Create PRJ file

        try {

            savePrjFile(file, projection);

        } catch (IOException e) {

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

        }

    }

}