Application: gvSIG desktop » gvSIG data providers based on gdal

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;

        IProjection projection = buffer.getProjection();

        try {

            if (options.length == 0) {

                gdalDataset =

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

                        buffer.getBandCount(), getGdalTypeFromRasterBufType(buffer.getBandTypes()[0]));

            } else {

                gdalDataset =

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

                        buffer.getBandCount(), getGdalTypeFromRasterBufType(buffer.getBandTypes()[0]), options);

            }

            if (projection != null) {

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

                if (wkt != null) {

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

        }

        // Create PRJ file

        try {

            savePrjFile(file, projection);

        } catch (IOException e) {

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

        }

    }

}