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.postgis.app.postgisrasterclient.gui;

import java.util.List;

import javax.swing.JOptionPane;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.ui.mdiManager.IWindow;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataStoreParameters;

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

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

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.feature.FeatureStore;

import org.gvsig.fmap.dal.serverexplorer.db.DBServerExplorerParameters;

import org.gvsig.fmap.mapcontext.MapContext;

import org.gvsig.fmap.mapcontext.MapContextLocator;

import org.gvsig.fmap.mapcontext.MapContextManager;

import org.gvsig.fmap.mapcontext.exceptions.LoadLayerException;

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.raster.fmap.layers.DefaultFLyrRaster;

import org.gvsig.raster.fmap.layers.FLyrRaster;

import org.gvsig.raster.postgis.io.PostGISRasterDataParameters;

import org.gvsig.raster.postgis.io.PostGISRasterServerExplorer;

import org.gvsig.raster.postgis.io.PostGISRasterServerExplorerParameters;

import org.gvsig.raster.postgis.swing.AddLayerUI;

import org.gvsig.raster.util.RasterNotLoadException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * Manager to load layers in TOC. There are some strategies, load a entire layer

 * or load subdatasets depending on the user selection or the layer structure.

 * 

 * @author Nacho Brodin (nachobrodin@gmail.com)

 */

public class LayerLoaderManager {

        private static int                  ALL_SUBDATASETS       = -2;

        private static int                  ENTIRE                = -1;

        private static Logger               logger                = LoggerFactory.getLogger(AddPostGISRasterLayerWizard.class.getName());

        private AddLayerUI                  apiUI                 = null;

        private DBServerExplorerParameters  dbExplorerParameters  = null;

        private MapContext                  mapContext            = null;

        public LayerLoaderManager(AddLayerUI ui) {

                this.apiUI = ui;

        }

        /**

         * Assigns a {@link DBServerExplorerParameters} 

         * @param params

         */

        public void setDBServerExplorerParameters(DBServerExplorerParameters params) {

                this.dbExplorerParameters = params;

        }

        /**

         * Gets the {@link DBServerExplorerParameters}

         * @return {@link DBServerExplorerParameters}

         */

        public DBServerExplorerParameters getDBServerExplorerParameters() {

                return this.dbExplorerParameters;

        }

        /**

         * Return FLayers if user performs multi selection.

         */

        public void loadLayers() {

                if(dbExplorerParameters == null)

                        return;

                List<Object> tables = apiUI.getSelectedTables();

                //Para una tabla seleccionada

                if(tables.size() == 1) {

                        TableListItem item = (TableListItem)tables.get(0);

                        String schema = item.getSchema();

                        String table = item.getTableName();

                        //Sin subdatasets

                        if(getNumberOfColumns(schema, table) == 1) {

                                loadOneLayer(item, ENTIRE);

                        } 

                        //Con subdatasets

                        else { 

                                int[] subdatasets = apiUI.getSelectedSubdatasets();

                                if(subdatasets == null || subdatasets.length <= 0) {

                                        loadOneLayer(item, ALL_SUBDATASETS);

                                } else {

                                        long cols = getNumberOfColumns(schema, table);

                                        for (int i = 0; i < cols; i++) {

                                                loadOneLayer(item, i + 1);

                                        }

                                }

                        }

                }

                //Para m?s de una tabla seleccionada (se usan todos los subdatasets)

                if(tables.size() > 1) {

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

                                TableListItem item = (TableListItem)tables.get(i);

                                String schema = item.getSchema();

                                String table = item.getTableName();

                                long subdatasets = getNumberOfColumns(schema, table);

                                if(subdatasets == 1) {

                                        loadOneLayer(item, ENTIRE);

                                } 

                                if(subdatasets > 1) {

                                        loadOneLayer(item, ALL_SUBDATASETS);

                                }

                        }

                }

        }

        /**

         * Loads only one layer

         * @param item

         * @param subdataset

         */

        private void loadOneLayer(TableListItem item, int subdataset) {

                String host = dbExplorerParameters.getHost();

                if(host.compareTo("localhost") == 0)

                        host = "127.0.0.1";

                Integer port = dbExplorerParameters.getPort();

                String dbname = dbExplorerParameters.getDBName();

                String user = dbExplorerParameters.getUser();

                String password = dbExplorerParameters.getPassword();

                String schema = item.getSchema();

                String table = item.getTableName();

                String connString = null;

                if(schema == null)

                        connString = "PG:host='" + host + "' port='" + port + "' dbname='" + dbname + "' user='" + user + "' password='" + password + "' table='" + table + "'";

                else

                        connString = "PG:host='" + host + "' port='" + port + "' dbname='" + dbname + "' user='" + user + "' password='" + password + "' schema='" + schema + "' table='" + table + "'";

                if(subdataset == ALL_SUBDATASETS)

                        connString += " mode='2'";

                if(subdataset >= 0)

                        connString += " column='rast' where='rid = "+ subdataset + "'";

                try {

                        FLayer layer = createLayer(getFormatedRasterFileName(connString), connString, item, getNumberOfColumns(schema, table));

                        //Adds the layer to the view

                        getMapContext().beginAtomicEvent();

                        layer.setVisible(true);

                        getMapContext().getLayers().addLayer(layer);

                        getMapContext().callLegendChanged();

                        getMapContext().endAtomicEvent();

                        getMapContext().invalidate();

                } catch (LoadLayerException e) {

                        logger.info("Error creating layer: " + e.getMessage(), e);

                        //showConnectionErrorMessage(PluginServices.getText(this, "error_creating_layer") + "");

                }

        }

        /**

         * Gets the MapContext

         * @return

         */

        private MapContext getMapContext() {

                if(mapContext == null) {

                        IWindow[] w = PluginServices.getMDIManager().getAllWindows();

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

                                if (w[i] instanceof org.gvsig.app.project.documents.view.gui.AbstractViewPanel) {

                                        mapContext = ((org.gvsig.app.project.documents.view.gui.AbstractViewPanel)w[i]).getMapControl().getMapContext();

                                }

                        }

                }

                return mapContext;

        }

        /**

         * Builds a new raster layer

         * @param fileName

         * @return

         * @throws RasterNotLoadException 

         * @throws LoadLayerException 

         */

        @SuppressWarnings("deprecation")

        public FLyrRaster createLayer(String layerName, String conn, TableListItem item, long nColumns) throws LoadLayerException {

                DataManager dataManager = DALLocator.getDataManager();

                DefaultFLyrRaster lyr = null;

                try {

                        PostGISRasterServerExplorerParameters explorerParams = (PostGISRasterServerExplorerParameters) dataManager.createServerExplorerParameters(PostGISRasterServerExplorer.NAME);

                        explorerParams.setHost(conn);

                        PostGISRasterServerExplorer explorer = (PostGISRasterServerExplorer) dataManager.openServerExplorer(PostGISRasterServerExplorer.NAME,explorerParams);

                        PostGISRasterDataParameters storeParameters = (PostGISRasterDataParameters)explorer.getStoredParameters();

                        storeParameters.setURI(conn);

                        storeParameters.setDBExplorer(item.getExplorer());

                        storeParameters.setNumberOfBlocks(nColumns);

                        storeParameters.setDBStoreParameters(item.getDBStoreParameters());

                        storeParameters.setDBExplorerParameters(dbExplorerParameters);

                        MapContextManager mcm = MapContextLocator.getMapContextManager();

                        if(apiUI.isLayerTiled()) {

                                DataManager manager = DALLocator.getDataManager();

                                TileDataParameters tileParams = null;

                                try {

                                        tileParams = (TileDataParameters) manager.createStoreParameters("Tile Store");

                                } catch (InitializeException e) {

                                        throw new LoadLayerException("", e);

                                } catch (ProviderNotRegisteredException e) {

                                        throw new LoadLayerException("", e);

                                }

                                tileParams.setDataParameters(storeParameters);

                                lyr = (DefaultFLyrRaster) mcm.createLayer(layerName, tileParams);

                                //dataStore = dataManager.createStore(tileParams);

                        } else {

                                lyr = (DefaultFLyrRaster) mcm.createLayer(layerName, storeParameters);

                                //dataStore = dataManager.createStore(storeParameters);

                        }

                } catch (ValidateDataParametersException e) {

                        throw new LoadLayerException("Error validating parameters", e);

                } catch (InitializeException e) {

                        throw new LoadLayerException("Error initializating layer.", e);

                } catch (ProviderNotRegisteredException e) {

                        throw new LoadLayerException("Provider not registered.", e);

                } 

                return lyr;

        }

        /**

         * Gets the number of column of the selected table

         * @param schema

         * @param table

         * @return

         */

        @SuppressWarnings("deprecation")

        public long getNumberOfColumns(String schema, String table) {

                DataManager manager = DALLocator.getDataManager();

                try {

                        DataStoreParameters params = manager.createStoreParameters("PostgreSQL");

                        params.setDynValue("host", dbExplorerParameters.getHost());

                        params.setDynValue("port", dbExplorerParameters.getPort());

                        params.setDynValue("dbuser", dbExplorerParameters.getUser());

                        params.setDynValue("password", dbExplorerParameters.getPassword());

                        params.setDynValue("schema",  schema);

                        params.setDynValue("dbname", dbExplorerParameters.getDBName());

                        params.setDynValue("table", table);

                        FeatureStore store = (FeatureStore)manager.createStore(params);

                        long size = store.getFeatureCount();

                        store.dispose();

                        return size;

                } catch (DataException e) {

                        return 0;

                } catch (ValidateDataParametersException e) {

                        return 0;

                }

        }

        /**

         * Gets the name of a raster file

         */

        private String getFormatedRasterFileName(String name) {

                String newName = "";

                String schema = null;

                String table = null;

                int index = name.indexOf(" schema='") + 8;

                if(index != -1) {

                        try {

                                schema = name.substring(index + 1, name.indexOf("'", index + 1)); 

                                newName += schema + ".";

                        } catch (StringIndexOutOfBoundsException e) {

                        }

                }

                index = name.indexOf(" table='") + 7;

                if(index != -1) {

                        try {

                                table = name.substring(index + 1, name.indexOf("'", index + 1));

                                newName += table;

                        } catch (StringIndexOutOfBoundsException e) {

                        }

                }

                return newName;

        }

        /**

         * Shows an error

         * @param _msg

         */

        public void showConnectionErrorMessage(String _msg) {

                String msg = (_msg.length() > 300) ? "" : (": " + _msg);

                String title = PluginServices.getText(this, "connection_error");

                JOptionPane.showMessageDialog(apiUI.getComponent(), title + msg, title,

                                JOptionPane.ERROR_MESSAGE);

        }

        /**

         * Shows a information window

         * @param _msg

         */

        public void showConnectionInfoMessage(String _msg) {

                String msg = (_msg.length() > 300) ? "" : (": " + _msg);

                String title = PluginServices.getText(this, "warning");

                JOptionPane.showMessageDialog(apiUI.getComponent(), title + msg, title,

                                JOptionPane.WARNING_MESSAGE);

        }

}