Application: gvSIG desktop » gvSIG data providers based on gdal

package org.gvsig.raster.gdal.provider;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.List;

import org.apache.commons.lang3.StringUtils;

import org.gdal.gdal.Driver;

import org.gdal.gdal.gdal;

import org.gdal.ogr.ogr;

import org.gvsig.fmap.dal.DALFileLibrary;

import org.gvsig.fmap.dal.DALFileLocator;

import org.gvsig.fmap.dal.DALLibrary;

import org.gvsig.fmap.dal.FileHelper;

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

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

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

import org.gvsig.metadata.exceptions.MetadataException;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynStruct_v2;

import org.gvsig.tools.dynobject.Tags;

import org.gvsig.tools.library.AbstractLibrary;

import org.gvsig.tools.library.LibraryException;

/**

 * Raster GDAL Provider Library to register provider commponents

 *

 * @author dmartinezizquierdo

 *

 */

public class RasterGdalLibrary extends AbstractLibrary {

    /**

     * Driver name that must not be used to open files

     */

    public static final String[] EXCLUDED_OPEN_DRIVERS={"netCDF"};

    /**

     * Driver name that must not be used to create new files

     */

    public static final String[] EXCLUDED_STORE_DRIVERS={};

    /**

     * This drivers will be not used by GDAL

     */

    public static final String[] NOT_REGISTERED_DRIVERS={"PostGISRaster", "WMS", "WCS"};

    @Override

    public void doRegistration() {

        registerAsServiceOf(DALLibrary.class);

        require(DALFileLibrary.class);

    }

    @Override

    protected void doInitialize() throws LibraryException {

    }

    @Override

    protected void doPostInitialize() throws LibraryException {

        List<Throwable> exs = new ArrayList<Throwable>();

        // initialize gdal drivers

        try {

            gdal.AllRegister();

            ogr.RegisterAll();

        } catch (RuntimeException e) {

            exs.add(e);

        }

        DataManagerProviderServices dataman =

            DALSPILocator.getDataManagerProviderServices();

        //Register supported parameters

        try {

            FileHelper.registerParametersDefinition(

                RasterGdalFileStoreParameters.PARAMETERS_DEFINITION_NAME,

                RasterGdalFileStoreParameters.class, "RasterGdalParameters.xml");

        } catch (Exception e) {

            exs.add(e);

        }

        try {

            FileHelper.registerMetadataDefinition(

                AbstractFileRasterGdalStoreProvider.METADATA_DEFINITION_NAME,

                AbstractFileRasterGdalStoreProvider.class, "RasterGdalMetadata.xml");

        } catch (MetadataException e) {

            exs.add(e);

        }

        try {

            FileHelper.registerParametersDefinition(

                NewRasterGdalStoreParameters.PARAMETERS_DEFINITION_NAME,

                NewRasterGdalStoreParameters.class, "NewRasterGdalStoreParameters.xml");

        } catch (Exception e) {

            exs.add(e);

        }

        try {

            FileHelper.registerParametersDefinition(

                RasterGdalStoreParameters.PARAMETERS_DEFINITION_NAME,

                RasterGdalStoreParameters.class, "RasterGdalParameters.xml");

        } catch (Exception e) {

            exs.add(e);

        }

        List<String> notRegisteredDrivers=Arrays.asList(NOT_REGISTERED_DRIVERS);

        try {

            for (int i = 0; i < gdal.GetDriverCount(); i++) {

                //For each driver must be a StoreProviderFactory and a FilesystemProviderFactory

                Driver gdalDriver = gdal.GetDriver(i);

                String driverName = gdalDriver.getShortName();

                if (notRegisteredDrivers.contains(driverName)){

                    continue;

                }

                String provName =driverName;

                    //RasterGdalStoreProvider.NAME + "_" + driverName;

                String provDescription = AbstractFileRasterGdalStoreProvider.DESCRIPTION

                    + " for driver " + driverName;

                String openParams;

                //Try to load specific params with file

                try {

                    openParams=RasterGdalFileStoreParameters.PARAMETERS_DEFINITION_NAME+"_"+driverName;

                    FileHelper.registerParametersDefinition(

                        openParams,

                        RasterGdalFileStoreParameters.class, "RasterGdalParameters.xml");

//                    FileHelper.registerParametersDefinition(

//                        openParams,

//                        RasterGdalFileStoreParameters.class, "RasterGdalParameters.xml");

                } catch (Exception e) {

                    try {

                        //Try to load specific params without file

                        openParams=RasterGdalStoreParameters.PARAMETERS_DEFINITION_NAME+"_"+driverName;

                        FileHelper.registerParametersDefinition(

                            openParams,

                            RasterGdalStoreParameters.class, "RasterGdalParameters.xml");

                    } catch (Exception e1) {

                        //If specific params are not found, generic ones with file are used

                        openParams=RasterGdalFileStoreParameters.PARAMETERS_DEFINITION_NAME;

                    }

                }

                //The store provider factory is created

                RasterGdalStoreProviderFactory providerFactory=null;

                try {

                    if (!dataman.getStoreProviders().contains(provName)) {

                        providerFactory= new RasterGdalStoreProviderFactory(provName,

                            provDescription,

                            gdalDriver, openParams);

                        dataman.registerStoreProviderFactory(providerFactory);

                    }

                } catch (RuntimeException e) {

                    exs.add(e);

                }

                String newParams;

                //Try to load specific params

                try {

                    newParams=NewRasterGdalStoreParameters.PARAMETERS_DEFINITION_NAME+"_"+driverName;

                    FileHelper.registerParametersDefinition(

                        newParams,

                        NewRasterGdalStoreParameters.class, "NewRasterGdalStoreParameters.xml");

                } catch (Exception e) {

                    //If specific params are not found, generic ones are used

                    newParams=NewRasterGdalStoreParameters.PARAMETERS_DEFINITION_NAME;

                }

                // The FilesystemServerExplorerProvider Factory is created

                FilesystemServerExplorerProviderFactory filesystemProviderFactory =null;

                try {

                    if (providerFactory!=null){

                        filesystemProviderFactory = (FilesystemServerExplorerProviderFactory)

                            new RasterGdalFileSystemServerProviderFactory(

                            providerFactory, newParams);

                        DALFileLocator.getFilesystemServerExplorerManager()

                        .registerProviderFactory(filesystemProviderFactory);

                    }

                } catch (RuntimeException e) {

                    exs.add(e);

                }

                //The valid extensions from the parameters are added to the filesystem server factory

                String validExtensions=null;

                DynStruct_v2 params = (DynStruct_v2)ToolsLocator.getPersistenceManager().getDefinition(openParams);

                Tags paramsTags=params.getTags();

                if (paramsTags.has(RasterGdalFileStoreParameters.VALID_EXTENSIONS_TAG_NAME)){

                    validExtensions=(String)paramsTags.get(RasterGdalFileStoreParameters.VALID_EXTENSIONS_TAG_NAME);

                }

                if (filesystemProviderFactory!=null &&

                    filesystemProviderFactory instanceof RasterGdalFileSystemServerProviderFactory &&

                    StringUtils.isNotEmpty(validExtensions)){

                    String[] extensions=validExtensions.split(",");

                    ((RasterGdalFileSystemServerProviderFactory)filesystemProviderFactory).

                        addAcceptedExtensions(extensions);

                }

            }

        } catch (RuntimeException e) {

            exs.add(e);

        }

        if (exs.size() > 0) {

            throw new LibraryException(this.getClass(), exs);

        }

    }

}