Application: gvSIG desktop » gvSIG data providers based on gdal

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2016 gvSIG Association

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.raster.gdal.provider;

import java.io.File;

import java.io.IOException;

import org.gdal.gdal.Driver;

import org.gdal.gdalconst.gdalconstConstants;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.NewDataStoreParameters;

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

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

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

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

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

import org.gvsig.fmap.dal.resource.spi.ResourceConsumer;

import org.gvsig.fmap.dal.resource.spi.ResourceProvider;

import org.gvsig.fmap.dal.serverexplorer.filesystem.impl.AbstractFilesystemServerExplorerProvider;

import org.gvsig.fmap.dal.serverexplorer.filesystem.spi.FilesystemProviderServices;

import org.gvsig.fmap.dal.serverexplorer.filesystem.spi.FilesystemServerExplorerProvider;

import org.gvsig.fmap.dal.serverexplorer.filesystem.spi.FilesystemServerExplorerProviderServices;

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

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

import org.gvsig.tools.dispose.DisposeUtils;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * Filesystem Provider for Raster GDAL.

 * @author dmartinezizquierdo

 *

 */

public class RasterGdalFileSystemServerProvider extends AbstractFilesystemServerExplorerProvider

implements FilesystemServerExplorerProvider, ResourceConsumer{

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

    protected FilesystemServerExplorerProviderServices serverExplorer;

    private RasterGdalFileSystemServerProviderFactory factory;

    //private ArrayList<String> acceptedExtensions;

    RasterGdalFileSystemServerProvider (RasterGdalFileSystemServerProviderFactory factory){

        this.factory=factory;

    }

    public int getMode() {

        return factory.getMode();

    }

    @Override

    public boolean canCreate() {

        return factory.allowCreate()==FilesystemProviderServices.YES;

    }

    @Override

    public boolean canCreate(NewDataStoreParameters newDataStoreParameters) {

        if (canCreate()){

            if (newDataStoreParameters instanceof NewRasterGdalStoreParameters){

                if ( newDataStoreParameters.isValid() ){

                    Driver gdalDriver=factory.getProviderFactory().getDriver();

                    Buffer buffer=((NewRasterGdalStoreParameters)newDataStoreParameters).getBuffer();

                    String creationTypes=(String)gdalDriver.GetMetadata_Dict().get(gdalconstConstants.GDAL_DMD_CREATIONDATATYPES);

                    if  (creationTypes == null || (creationTypes!=null && creationTypes.contains(getBufferGDALDataType(buffer)))){

                        return true;

                    }

                }

            }

        }

        return false;

    }

    private String getBufferGDALDataType(Buffer buffer){

        int[] bufferTypes=buffer.getBandTypes();

        int type=BufferManager.TYPE_UNDEFINED;

        for (int i = 0; i < bufferTypes.length; i++) {

            if(type==BufferManager.TYPE_UNDEFINED){

                type = bufferTypes[i];

            } else if(type != bufferTypes[i]){

                type = BufferManager.TYPE_UNDEFINED;

                break;

            }

        }

        //Byte UInt16 Int16 UInt32 Int32 Float32 Float64 CInt16 CInt32 CFloat32 CFloat64

        switch (type) {

        case BufferManager.TYPE_BYTE:

            return "Byte";

        case BufferManager.TYPE_SHORT:

            return "Int16";

        case BufferManager.TYPE_USHORT:

            return "UInt16"; // OJO

        case BufferManager.TYPE_INT:

            return "Int32";

        case BufferManager.TYPE_FLOAT:

            return "Float32";

        case BufferManager.TYPE_DOUBLE:

            return "Float64";

        default:

            return "Unsupported";

        }

    }

    @Override

    public void create(NewDataStoreParameters params, boolean overwrite)

        throws CreateException {

        if ( !canCreate(params) ){

            throw new CreateException(params.getDataStoreName(), new IllegalArgumentException());

        }

        if (params instanceof NewRasterGdalStoreParameters){

            RasterGdalStoreProviderFactory providerFactory = factory.getProviderFactory();

            AbstractFileRasterGdalStoreProvider provider;

            try {

                provider = (AbstractFileRasterGdalStoreProvider) providerFactory.createProvider(params);

            } catch (InitializeException e) {

                throw new CreateException(params.getDataStoreName(), e);

            }

            provider.store(overwrite);

            DisposeUtils.dispose(provider);

        }

    }

    @Override

    public NewDataStoreParameters getCreateParameters() throws DataException {

       return new NewRasterGdalStoreParameters(factory.getNewParamsDefinitionName(), factory.getName());

    }

    @Override

    public void initialize(FilesystemServerExplorerProviderServices serverExplorer) {

        this.serverExplorer = serverExplorer;

    }

    @Override

    public void remove(DataStoreParameters parameters) throws RemoveException {

        RasterGdalFileStoreParameters params = (RasterGdalFileStoreParameters) parameters;

        File file = params.getFile();

        if (!file.exists()) {

            throw new RemoveException(this.getDataStoreProviderName(),

                    new FileNotFoundException(params.getFile()));

        }

        if (!file.delete()) {

            throw new RemoveException(this.getDataStoreProviderName(),

                    new IOException("Error deleting file: "+file.getName()));

        }

    }

    @Override

    public String getDataStoreProviderName() {

        return factory.getName();

    }

    @Override

    public String getDescription() {

        return factory.getDescription();

    }

    @Override

    public boolean accept(File pathname) {

        return factory.accept(pathname);

    }

    @Override

    public boolean closeResourceRequested(ResourceProvider resource) {

        return !(this.equals(resource));

    }

    @Override

    public void resourceChanged(ResourceProvider resource) {

        //Do nothing

    }

}