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.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import java.util.Map.Entry;

import javax.swing.JComboBox;

import javax.swing.JOptionPane;

import javax.swing.JPanel;

import org.apache.commons.collections.map.HashedMap;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.PluginsLocator;

import org.gvsig.andami.PluginsManager;

import org.gvsig.andami.messages.NotificationManager;

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

import org.gvsig.app.gui.WizardPanel;

import org.gvsig.app.project.documents.view.gui.AbstractViewPanel;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.DataStoreParameters;

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

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.FeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.FeatureType;

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

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

import org.gvsig.fmap.dal.store.db.DBStoreParameters;

import org.gvsig.fmap.mapcontext.MapContext;

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

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

import org.gvsig.geodb.ExtDB_Spatial;

import org.gvsig.geodb.vectorialdb.wizard.MyExplorer;

import org.gvsig.geodb.vectorialdb.wizard.VectorialDBConnectionParamsDialog;

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

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

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

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.postgis.swing.PostGISRasterSwingLocator;

import org.gvsig.raster.util.RasterNotLoadException;

import org.gvsig.tools.dynobject.DynObject;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * Wizard panel to add a new PostGIS raster layer

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

 */

public class AddPostGISRasterLayerWizard extends WizardPanel implements ActionListener {

        private static final long           serialVersionUID      = 1L;

        private static final String         TAB_NAME              = "PostGIS Raster";

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

        private AbstractViewPanel           view                  = null;

        private AddLayerUI                  apiUI                 = null;

        private DBServerExplorerParameters  dbExplorerParameters  = null;

        public AddPostGISRasterLayerWizard() {

                super();

                initialize();

        }

        /**

         * This method initializes this

         *

         * @return void

         */

        private void initialize() {

                setTabName(PluginServices.getText(this, TAB_NAME));

                setLayout(null);

                IWindow iw = PluginServices.getMDIManager().getActiveWindow();

                if (iw == null) {

                        return;

                }

                if (!(iw instanceof AbstractViewPanel)) {

                        return;

                }

                apiUI = PostGISRasterSwingLocator.getSwingManager().createAddLayerUI();

                apiUI.addListenerToBDSelectionCombo(this);

                apiUI.addListenerToNewDBButton(this);

                view = (AbstractViewPanel) iw;

                setMapCtrl(view.getMapControl());

                JPanel mainPanel = (JPanel)apiUI.getComponent();

                mainPanel.setBounds(new java.awt.Rectangle(0, 0, 501, 350));

                add(mainPanel, null);

                loadVectorialDBDatasourcesCombo(null);

        }

        /**

         * Loads the ComboBox for listing connections

         * @param sel

         */

    @SuppressWarnings("unchecked")

        private void loadVectorialDBDatasourcesCombo(MyExplorer sel) {

            JComboBox combo = apiUI.getDBCombo();

        PluginsManager manager = PluginsLocator.getManager();

        DynObject values = manager.getPlugin(ExtDB_Spatial.class).getPluginProperties();

        Map connections = (Map) values.getDynValue("db_connections");

        if (connections != null){

            Iterator it = connections.entrySet().iterator();

            combo.removeAllItems();

            combo.addItem("");

            while (it.hasNext()){

                Map.Entry entry = (Entry) it.next();

                MyExplorer myExplorer = new MyExplorer();

                myExplorer.setDbExplorerParameters((DBServerExplorerParameters) entry.getValue());

                myExplorer.setName((String) entry.getKey());

                combo.addItem(myExplorer);

                if(sel!=null && sel.getName().equalsIgnoreCase(myExplorer.getName())){

                        combo.setSelectedItem(myExplorer);

                }

            }

        } else {

            connections = new HashedMap();

            values.setDynValue("db_connections", connections);

        }

        }

    /**

         * Return FLayers if user performs multi selection.

         */

        public FLayer getLayer() {

                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 table = null;

                String schema = null;

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

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

                schema = item.getSchema();

                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 + "'";

                //connString += " column='rast' where='rid = 1041'";

                try {

                        return createLayer(getFormatedRasterFileName(connString), connString, item);

                } catch (LoadLayerException e) {

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

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

                        return null;

                }

        }

        /**

         * Builds a new raster layer

         * @param fileName

         * @return

         * @throws RasterNotLoadException 

         * @throws LoadLayerException 

         */

        @SuppressWarnings("deprecation")

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

                DataManager dataManager = DALLocator.getDataManager();

                DataStore dataStore = null;

                try {

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

                        explorerParams.setHost(conn);

                        PostGISRasterServerExplorer explorer = (PostGISRasterServerExplorer) dataManager.createServerExplorer(explorerParams);

                        RasterDataParameters storeParameters = (RasterDataParameters)explorer.getStoreParameters();

                        storeParameters.setURI(conn);

                        storeParameters.setDynValue("DBPARAMS", item.getDBStoreParameters());

                        storeParameters.setDynValue("DBEXPLORER", item.getExplorer());

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

                } catch (PostGISRasterCoreException e) {

                        throw new LoadLayerException("", e);

                }

                DefaultFLyrRaster lyr = new DefaultFLyrRaster();

                lyr.setName(layerName);

                lyr.setDataStore(dataStore);

                return lyr;

        }

        /**

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

        }

        @Override

        public void close() {

        }

        @Override

        public void execute() {

                FLayer lyr = getLayer();

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

                        }

                }

                mapContext.beginAtomicEvent();

                lyr.setVisible(true);

                mapContext.getLayers().addLayer(lyr);

                mapContext.callLegendChanged();

                mapContext.endAtomicEvent();

                mapContext.invalidate();

        }

        @Override

        public DataStoreParameters[] getParameters() {

                return null;

        }

        @Override

        public void initWizard() {

        }

        /**

         * Updates the JList component with the list of tables of 

         * the selected database.

         * 

         * @param dbSeverExplorerParameters2

         */

        @SuppressWarnings("unchecked")

        private void updateTableList(

                        DBServerExplorerParameters dbSeverExplorerParameters2) {

                if (dbSeverExplorerParameters2 == null) {

                        return;

                }

                DataManager dm = DALLocator.getDataManager();

                DBServerExplorer dbExplorer = null;

                try {

                        dbExplorer = (DBServerExplorer) dm.openServerExplorer(dbSeverExplorerParameters2.getExplorerName(), dbSeverExplorerParameters2);

                        List parameters = dbExplorer.list();

                        Iterator iter = parameters.iterator();

                        DBStoreParameters param = null;

                        apiUI.clearList();

                        while (iter.hasNext()) {

                                param = (DBStoreParameters) iter.next();

                                //Criterios para a?adir la tabla a la lista

                                try {

                                        if(!param.getTable().startsWith("o_") && isRasterTable(param)) {

                                                apiUI.addElementToTable(new TableListItem(dbExplorer, param));

                                        }

                                } catch (PostGISRasterCoreException e) {

                                        logger.error("While getting table names: " + e.getMessage(), e);

                                }

                        }

                } catch (Exception e) {

                        logger.error("While getting table names: " + e.getMessage(), e);

                        showConnectionErrorMessage(PluginServices.getText(this, "error_getting_tables"));

                        return;

                }

        }

        /**

         * Checks is the table has the structure to contain a raster

         * @param param

         * @return

         * @throws PostGISRasterCoreException

         */

        @SuppressWarnings({ "deprecation" })

        private boolean isRasterTable(DBStoreParameters param) throws PostGISRasterCoreException {

                DataManager dm = DALLocator.getDataManager();

                FeatureStore store;

                try {

                        store = (FeatureStore)dm.createStore(param);

                        FeatureType featureType = store.getDefaultFeatureType();

                        if(featureType.size() == 2) {

                                if(existsAttr(featureType, "rid") && existsAttr(featureType, "rast"))

                                        return true;

                        }

                        store.dispose();

                } catch (ValidateDataParametersException e) {

                        throw new PostGISRasterCoreException(e);

                } catch (InitializeException e) {

                        throw new PostGISRasterCoreException(e);

                } catch (ProviderNotRegisteredException e) {

                        throw new PostGISRasterCoreException(e);

                } catch (DataException e) {

                        throw new PostGISRasterCoreException(e);

                }

                return false;

        }

        @SuppressWarnings("unchecked")

        private boolean existsAttr(FeatureType featureType, String attr) {

                Iterator it = featureType.iterator();

                while( it.hasNext() ) {

                        FeatureAttributeDescriptor attribute = (FeatureAttributeDescriptor)it.next();

                        if(attribute.getName().compareTo(attr) == 0) {

                                return true;

                        }

                }

                return false;

        }

        /*

         * (non-Javadoc)

         * @see java.awt.event.ActionListener#actionPerformed(java.awt.event.ActionEvent)

         */

        public void actionPerformed(ActionEvent e) {

                if (apiUI.isAddDBEventObject(e.getSource())) {

                        MyExplorer sel = addNewConnection();

                        if (sel != null) {

                                dbExplorerParameters = sel.getDbSeverExplorerParameters();

                                loadVectorialDBDatasourcesCombo(sel);

                                apiUI.getDBCombo().setSelectedItem(sel);

                        }

                }

                if(e.getSource() == apiUI.getDBCombo()) {

                        JComboBox dbCombo = apiUI.getDBCombo();

                        if(dbCombo.getItemCount() == 0) {

                                apiUI.clearList();

                        } else {

                                Object selected = dbCombo.getSelectedItem();

                                if(selected == null)

                                        return;

                                if (selected instanceof MyExplorer) {

                        MyExplorer sel_obj = (MyExplorer) selected;

                        dbCombo.repaint();

                        dbExplorerParameters = sel_obj.getDbSeverExplorerParameters();

                        updateTableList(dbExplorerParameters);

                    }

                        }

                }

        }

        /**

         * Adds a new connection using the generic dialog. This function

         * is used when the button add new connection is pressed by the user

         * @return

         */

        @SuppressWarnings("unchecked")

        private MyExplorer addNewConnection() {

                MyExplorer myExplorer = new MyExplorer();

                DBServerExplorerParameters resp = null;

                VectorialDBConnectionParamsDialog newco = new VectorialDBConnectionParamsDialog();

                newco.showDialog();

                if (newco.isOkPressed()) {

                        try {

                                resp = newco.getParameters();

                        } catch (Exception e) {

                                showConnectionErrorMessage(e.getMessage());

                                return null;

                        }

                        PluginsManager manager = PluginsLocator.getManager();

            DynObject values = manager.getPlugin(ExtDB_Spatial.class).getPluginProperties();

            Map connections = (Map) values.getDynValue("db_connections");

            if(connections == null){

                connections = new HashedMap();

                values.setDynValue("db_connections", connections);

            }

            connections.put(newco.getConnectionName(), resp);

                        myExplorer.setDbExplorerParameters(resp);

                        myExplorer.setName(newco.getConnectionName());

                        return myExplorer;

                } else {

                        return null;

                }

        }

        /**

         * Shows an error

         * @param _msg

         */

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

        }

}