Application: gvSIG desktop » gvSIG raster

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * 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.

 *

 */

package org.gvsig.raster.util;

import java.io.File;

import java.io.IOException;

import java.lang.reflect.Constructor;

import java.lang.reflect.InvocationTargetException;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.Map;

import java.util.Set;

import java.util.TreeMap;

import org.gvsig.fmap.dal.DALFileLocator;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.coverage.RasterLocator;

import org.gvsig.fmap.dal.coverage.exception.FileNotSupportedException;

import org.gvsig.fmap.dal.coverage.exception.NotSupportedExtensionException;

import org.gvsig.fmap.dal.coverage.exception.RasterDriverException;

import org.gvsig.fmap.dal.coverage.exception.RmfSerializerException;

import org.gvsig.fmap.dal.coverage.store.parameter.RasterDataParameters;

import org.gvsig.fmap.dal.coverage.store.parameter.RasterFileStoreParameters;

import org.gvsig.fmap.dal.coverage.util.ProviderServices;

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

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

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

import org.gvsig.fmap.dal.serverexplorer.filesystem.FilesystemServerExplorer;

import org.gvsig.fmap.dal.serverexplorer.filesystem.FilesystemServerExplorerParameters;

import org.gvsig.fmap.dal.serverexplorer.filesystem.FilesystemStoreParameters;

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

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

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

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

import org.gvsig.jgdal.GdalWarp;

import org.gvsig.raster.impl.provider.DefaultRasterProvider;

import org.gvsig.raster.impl.provider.RasterProvider;

import org.gvsig.raster.impl.store.WriteFileFormatFeatures;

import org.gvsig.raster.impl.store.rmf.ClassSerializer;

import org.gvsig.raster.impl.store.rmf.RmfBlocksManager;

import org.gvsig.raster.impl.store.writer.DefaultRasterWriter;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.extensionpoint.ExtensionPoint;

import org.gvsig.tools.extensionpoint.ExtensionPointManager;

import org.gvsig.tools.extensionpoint.ExtensionPoint.Extension;

/**

 * 

 * 31/10/2007

 * @author Nacho Brodin nachobrodin@gmail.com

 */

public class DefaultProviderServices implements ProviderServices {

        public TreeMap<String, WriteFileFormatFeatures>

                                  fileFeature    = new TreeMap<String, WriteFileFormatFeatures>();

        public TreeMap<String, WriteFileFormatFeatures> getFileFeature() {

                return fileFeature;

        }

        /**

         * Loads the specific provider to download data

         * @param file

         * @return

         * @throws NotSupportedExtensionException

         * @throws FileNotSupportedException 

         */

        public static DefaultRasterProvider loadProvider(File file) throws ProviderNotRegisteredException, InitializeException {

                DataManagerProviderServices dataManager = (DataManagerProviderServices)DALLocator.getDataManager();

                DataStoreProviderServices storeServices = null;

                DataStoreProvider prov = null;

                if(file != null) {

                        //We have to locate a provider's name which manages the selected file

                        //A FilesystemServerExplorer will give a getProviderNames service

                        FilesystemServerExplorerParameters paramsExplorer = (FilesystemServerExplorerParameters)dataManager.createServerExplorerParameters(FilesystemServerExplorer.NAME);

                        FilesystemServerExplorer serverExplorer = null;

                        try {

                                paramsExplorer.setRoot(File.separator);

                                serverExplorer = (FilesystemServerExplorer)dataManager.openServerExplorer(FilesystemServerExplorer.NAME, paramsExplorer);

                        } catch (ValidateDataParametersException e) {

                                throw new InitializeException(e);

                        }

                        //Gets the list of provider's name to manage the file

                        ArrayList<String> provNameList = serverExplorer.getProviderNameList(file);

                        if(provNameList.size() > 0) {

                                for (int i = 0; i < provNameList.size(); i++) {

                                        //Gets the first provider what is not a TileProvider

                                        if(provNameList.get(i).compareTo("Tile Store") != 0) {

                                                DataStoreParameters newparams = dataManager.createStoreParameters(provNameList.get(i));

                                                ((FilesystemStoreParameters)newparams).setFile(file); 

                                                prov = dataManager.createProvider(storeServices, newparams);

                                        }

                                }

                        } else { //Default provider. This should be Gdal

                                ExtensionPointManager extensionPoints = ToolsLocator.getExtensionPointManager();

                                ExtensionPoint point = extensionPoints.get("DefaultRasterProvider");

                                Extension ext = point.get("reader");

                                String provName = ext.getDescription();

                                DataStoreParameters newparams = dataManager.createStoreParameters(provName);

                                ((FilesystemStoreParameters)newparams).setFile(file); 

                                prov = dataManager.createProvider(storeServices, newparams);

                        }

                }

                if(prov != null && prov instanceof RasterProvider) {

                        if(((RasterProvider)prov).isRotated())

                                throw new InitializeException("Rotation not supported tiling files", new Throwable());

                        return (DefaultRasterProvider)prov;

                }

                return null;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.fmap.dal.coverage.util.ProviderServices#createNotTiledParameters(java.lang.String)

         */

        public RasterDataParameters createNotTiledParameters(String id) throws InitializeException, ProviderNotRegisteredException {

                DataManagerProviderServices dataManager = (DataManagerProviderServices)DALLocator.getDataManager();

                ArrayList<RasterDataParameters> parametersList = createParametersList(id);

                for (int i = 0; i < parametersList.size(); i++) {

                        DataStoreProvider p = dataManager.createProvider(null, parametersList.get(i));

                        if(p instanceof RasterProvider) {

                                if(!((RasterProvider)p).isTiled()) 

                                        return parametersList.get(i);

                        }

                }

                return null;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.fmap.dal.coverage.util.ProviderServices#createParameters(java.lang.String)

         */

        @SuppressWarnings("unchecked")

        public RasterDataParameters createParameters(String id) {

                Iterator it = DALFileLocator.getFilesystemServerExplorerManager().getRegisteredProviders();

                while (it.hasNext()) {

                        ExtensionPoint.Extension o = (ExtensionPoint.Extension)it.next();

                        Class c = o.getExtension();

                        try {

                                Constructor hazNuevo = c.getConstructor();

                                Object [] args2 = {};

                                Object obj = hazNuevo.newInstance(args2);

                                if(obj instanceof FilesystemServerExplorerProvider) {

                                        FilesystemServerExplorerProvider prov = (FilesystemServerExplorerProvider)obj;

                                        File file = new File(id);

                                        if(prov.accept(file)) {

                                                DataManager manager = DALLocator.getDataManager();

                                                RasterFileStoreParameters params = null;

                                                try {

                                                        params = (RasterFileStoreParameters) manager

                                                                        .createStoreParameters(prov.getDataStoreProviderName());

                                                } catch (InitializeException e) {

                                                } catch (ProviderNotRegisteredException e) {

                                                }

                                                params.setFile(file);

                                                return params;

                                        }

                                } 

                        } catch (SecurityException e) {

                                System.out.println("Error SecurityException in open");

                        } catch (NoSuchMethodException e) {

                                System.out.println("Error NoSuchMethodException in open");

                        } catch (IllegalArgumentException e) {

                                System.out.println("Error IllegalArgumentException in open");

                        } catch (InstantiationException e) {

                                System.out.println("Error InstantiationException in open");

                        } catch (IllegalAccessException e) {

                                System.out.println("Error IllegalAccessException in open");

                        } catch (InvocationTargetException e) {

                                System.out.println("Error opening this image with ");

                        }

                }

                return null;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.fmap.dal.coverage.util.ProviderServices#createParametersList(java.lang.String)

         */

        @SuppressWarnings("unchecked")

        public ArrayList<RasterDataParameters> createParametersList(String id) {

                ArrayList<RasterDataParameters> list = new ArrayList<RasterDataParameters>();

                Iterator it = DALFileLocator.getFilesystemServerExplorerManager().getRegisteredProviders();

                while (it.hasNext()) {

                        ExtensionPoint.Extension o = (ExtensionPoint.Extension)it.next();

                        Class c = o.getExtension();

                        try {

                                Constructor hazNuevo = c.getConstructor();

                                Object [] args2 = {};

                                Object obj = hazNuevo.newInstance(args2);

                                if(obj instanceof FilesystemServerExplorerProvider) {

                                        FilesystemServerExplorerProvider prov = (FilesystemServerExplorerProvider)obj;

                                        File file = new File(id);

                                        if(prov.accept(file)) {

                                                DataManager manager = DALLocator.getDataManager();

                                                RasterFileStoreParameters params = null;

                                                try {

                                                        params = (RasterFileStoreParameters) manager

                                                                        .createStoreParameters(prov.getDataStoreProviderName());

                                                } catch (InitializeException e) {

                                                } catch (ProviderNotRegisteredException e) {

                                                }

                                                params.setFile(file);

                                                list.add(params);

                                        }

                                } 

                        } catch (SecurityException e) {

                                System.out.println("Error SecurityException in open");

                        } catch (NoSuchMethodException e) {

                                System.out.println("Error NoSuchMethodException in open");

                        } catch (IllegalArgumentException e) {

                                System.out.println("Error IllegalArgumentException in open");

                        } catch (InstantiationException e) {

                                System.out.println("Error InstantiationException in open");

                        } catch (IllegalAccessException e) {

                                System.out.println("Error IllegalAccessException in open");

                        } catch (InvocationTargetException e) {

                                System.out.println("Error opening this image with ");

                        }

                }

                return list;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.fmap.dal.coverage.util.ProviderServices#getDriversExtensions()

         */

        @SuppressWarnings("unchecked")

        public String[] getDriversExtensions() {

                ExtensionPointManager extensionPoints = ToolsLocator.getExtensionPointManager();

                ExtensionPoint point = extensionPoints.get("RasterWriter");

                return (String[])point.getNames().toArray(new String[0]);

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.fmap.dal.coverage.util.ProviderServices#getExtensionsSupported(int, int, boolean)

         */

        @SuppressWarnings("unchecked")

        public ArrayList<String> getExtensionsSupported(int dataType, int bands, boolean reprojectable) throws RasterDriverException {

                ExtensionPointManager extensionPoints = ToolsLocator.getExtensionPointManager();

                ExtensionPoint point = extensionPoints.get("RasterWriter");

                Iterator iterator = point.iterator();

                ArrayList<String> result = new ArrayList<String>();

                while (iterator.hasNext()) {

                        ExtensionPoint.Extension extension = (ExtensionPoint.Extension) iterator

                                        .next();

                        String ext = extension.getName();

                        Class writerClass = extension.getExtension();

                        Class[] args = { String.class };

                        try {

                                Constructor hazNuevo = writerClass.getConstructor(args);

                                Object[] args2 = { ext };

                                DefaultRasterWriter grw = (DefaultRasterWriter) hazNuevo.newInstance(args2);

                                if (isSupportedThisFileToWrite(ext, dataType, bands))

                                        if (reprojectable) {

                                                if (GdalWarp.getDrivers().contains(grw.getDriverName()))

                                                        result.add(ext);

                                        } else

                                                result.add(ext);

                        } catch (SecurityException e) {

                                throw new RasterDriverException("Error SecurityException in open");

                        } catch (NoSuchMethodException e) {

                                throw new RasterDriverException("Error NoSuchMethodException in open");

                        } catch (IllegalArgumentException e) {

                                throw new RasterDriverException("Error IllegalArgumentException in open");

                        } catch (InstantiationException e) {

                                throw new RasterDriverException("Error InstantiationException in open");

                        } catch (IllegalAccessException e) {

                                throw new RasterDriverException("Error IllegalAccessException in open");

                        } catch (InvocationTargetException e) {

                                throw new RasterDriverException("Error in open");

                        }

                }

                return result;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.fmap.dal.coverage.util.ProviderServices#getWriteDriversType()

         */

        @SuppressWarnings("unchecked")

        public String[] getWriteDriversType() {

                if (fileFeature.size() == 0)

                        return null;

                String[] list = new String[fileFeature.size()];

                Set values = fileFeature.entrySet();

                int i = 0;

                for (Iterator it = values.iterator(); it.hasNext();) {

                        list[i] = ((WriteFileFormatFeatures) ((Map.Entry) it.next()).getValue()).getDriverName();

                        i++;

                }

                return list;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.fmap.dal.coverage.util.ProviderServices#getWriteDriverType(java.lang.String)

         */

        public String getWriteDriverType(String ext) {

                return ((WriteFileFormatFeatures) fileFeature.get(ext)).getDriverName();

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.fmap.dal.coverage.util.ProviderServices#getWriteNDrivers()

         */

        public int getWriteNDrivers() {

                ExtensionPointManager extensionPoints = ToolsLocator.getExtensionPointManager();

                ExtensionPoint point = extensionPoints.get("RasterWriter");

                return point.getCount();

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.fmap.dal.coverage.util.ProviderServices#getWriteNTypes()

         */

        public int getWriteNTypes() {

                return fileFeature.size();

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.fmap.dal.coverage.util.ProviderServices#isSupportedThisFileToWrite(java.lang.String, int, int)

         */

        public boolean isSupportedThisFileToWrite(String ext, int dataType, int bands) {

                WriteFileFormatFeatures features = (WriteFileFormatFeatures) fileFeature.get(ext);

                if (features == null)

                        return false;

                int[] bandsSupported = features.getNBandsSupported();

                if (bandsSupported[0] != -1 && bandsSupported[0] < bands)

                        return false;

                int[] dt = features.getDataTypesSupported();

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

                        if (dataType == dt[i])

                                return true;

                return false;

        }

        /**

         * Guarda en el RMF que trata el BlocksManager el valor value usando el serializador que soporta

         * la clase class1.

         *

         * @param blocksManager

         * @param class1

         * @param value

         * @throws RmfSerializerException

         */

        @SuppressWarnings("unchecked")

        public void saveObjectToRmfFile(RmfBlocksManager blocksManager, Class class1, Object value) throws RmfSerializerException {

                ClassSerializer serializerObject = getSerializerObject(class1, value);

                if (serializerObject == null)

                        throw new RmfSerializerException("No se ha podido encontrar el serializador para el Rmf");

                if (!blocksManager.checkRmf())

                        throw new RmfSerializerException("Error al comprobar el fichero Rmf");

                blocksManager.addClient(serializerObject);

                try {

                        blocksManager.write(true);

                } catch (IOException e) {

                        throw new RmfSerializerException("Error al escribir el fichero Rmf", e);

                }

                blocksManager.removeAllClients();

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.fmap.dal.coverage.util.ProviderServices#saveObjectToRmfFile(java.lang.String, java.lang.Class, java.lang.Object)

         */

        @SuppressWarnings("unchecked")

        public void saveObjectToRmfFile(String file, Class class1, Object value) throws RmfSerializerException {

                String fileRMF = RasterLocator.getManager().getFileUtils().getNameWithoutExtension(file) + ".rmf";

                RmfBlocksManager blocksManager = new RmfBlocksManager(fileRMF);

                saveObjectToRmfFile(blocksManager, class1, value);

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.fmap.dal.coverage.util.ProviderServices#saveObjectToRmfFile(java.lang.String, java.lang.Object)

         */

        public void saveObjectToRmfFile(String file, Object value) throws RmfSerializerException {

                saveObjectToRmfFile(file, value.getClass(), value);

        }

        /**

         * Construye un serializador que soporte la clase class1. El constructor se invoca

         * con el parametro value.

         *

         * @param class1

         * @param value

         * @return

         * @throws RmfSerializerException

         */

        @SuppressWarnings("unchecked")

        public ClassSerializer getSerializerObject(Class class1, Object value) throws RmfSerializerException {

                try {

                        Class[] args = { class1 };

                        Constructor constructor = getSerializerClass(class1).getConstructor(args);

                        Object[] args2 = { value };

                        return (ClassSerializer) constructor.newInstance(args2);

                } catch (Exception e) {

                        throw new RmfSerializerException("No se ha podido crear el serializador para el Rmf", e);

                }

        }

        /**

         * Devuelve un serializador que tenga un constructor con la clase pasada por parametro

         * @throws RmfSerializerException

         */

        @SuppressWarnings("unchecked")

        private Class getSerializerClass(Class class1)

                        throws RmfSerializerException {

                ExtensionPointManager extensionPoints = ToolsLocator.getExtensionPointManager();

                ExtensionPoint point = extensionPoints.get("Serializer");

                Iterator iterator = point.iterator();

                while (iterator.hasNext()) {

                        ExtensionPoint.Extension extension = (ExtensionPoint.Extension) iterator.next();

                        if (extension != null) {

                                Class[] args = { class1 };

                                Class clase = extension.getExtension();

                                try {

                                        clase.getConstructor(args);

                                        return clase;

                                } catch (Exception e) {

                                        // Si no encuentra un constructor para esta clase, seguir? buscando en los dem?s elementos

                                }

                        }

                }

                throw new RmfSerializerException("No se ha encontrado la clase para el serializador");

        }

}