Application: gvSIG desktop

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2006 Prodevelop and Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 *

 * For more information, contact:

 *

 *   Generalitat Valenciana

 *   Conselleria d'Infraestructures i Transport

 *   Av. Blasco Ib??ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *   +34 963862235

 *   gvsig@gva.es

 *   www.gvsig.gva.es

 *

 *    or

 *

 *   Prodevelop Integraci?n de Tecnolog?as SL

 *   Conde Salvatierra de ?lava , 34-10

 *   46004 Valencia

 *   Spain

 *

 *   +34 963 510 612

 *   +34 963 510 968

 *   gis@prodevelop.es

 *   http://www.prodevelop.es

 */

package com.prodevelop.cit.gvsig.vectorialdb.wizard;

import java.awt.BorderLayout;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.ArrayList;

import javax.swing.DefaultListModel;

import javax.swing.ImageIcon;

import javax.swing.JComboBox;

import javax.swing.JOptionPane;

import javax.swing.JPanel;

import javax.swing.JScrollPane;

import javax.swing.ListSelectionModel;

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.ListSelectionListener;

import org.apache.log4j.Logger;

import org.gvsig.fmap.data.DataManager;

import org.gvsig.fmap.data.DataStoreParameters;

import org.gvsig.fmap.data.InitializeException;

import org.gvsig.fmap.data.ReadException;

import org.gvsig.fmap.data.feature.db.DBExplorer;

import org.gvsig.fmap.data.feature.db.DBExplorerParameters;

import org.gvsig.fmap.data.feature.db.DBStoreParameters;

import org.gvsig.fmap.geom.primitive.Envelope;

import org.gvsig.gui.beans.swing.JButton;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

import com.iver.andami.ui.mdiManager.IWindow;

import com.iver.cit.gvsig.SingleVectorialDBConnectionExtension;

import com.iver.cit.gvsig.gui.WizardPanel;

import com.iver.cit.gvsig.project.documents.view.gui.View;

/**

 * Driver-independent GeoDB wizard. Queries the drivers to fill GUI controls.

 * Multi-table selection available.

 *

 * @author jldominguez

 *

 */

public class WizardVectorialDB extends WizardPanel implements ActionListener,

    ListSelectionListener {

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

    private static final String GEODB_WIZARD_TAB_NAME = "GeoDB";

    private JPanel namePanel = null;

    private JPanel tablesPanel = null;

    private JScrollPane tablesScrollPane = null;

    private AvailableTablesCheckBoxList tablesList = null;

    private JComboBox datasourceComboBox = null;

    private UserTableSettingsPanel settingsPanel = null;

    private UserSelectedFieldsPanel fieldsPanel = null;

    private UserTableSettingsPanel emptySettingsPanel = null;

    private UserSelectedFieldsPanel emptyFieldsPanel = null;

    private JButton dbButton = null;

    private DBExplorerParameters dbExplorerParameters;

    private View view = null;

    public WizardVectorialDB() {

        super();

        initialize();

    }

    /**

     * This method initializes this

     *

     * @return void

     */

    private void initialize() {

        setTabName(GEODB_WIZARD_TAB_NAME);

        setLayout(null);

        setSize(512, 478);

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

        if (iw == null) {

            return;

        }

        if ((iw instanceof View)) {

                 view = (View) iw;

             setMapCtrl(view.getMapControl());

        }

        emptySettingsPanel = new UserTableSettingsPanel(null, null, "",

                         getMapCtrl(), true, this, null);

        emptyFieldsPanel = new UserSelectedFieldsPanel(null, true, this);

        add(getNamePanel(), null);

        loadVectorialDBDatasourcesCombo(null);

        add(getTablesPanel(), null);

    }

    private void loadVectorialDBDatasourcesCombo(MyExplorer sel) {

        getDatasourceComboBox().removeAllItems();

        DataManager dm=DataManager.getManager();

        String[] names=dm.getRegisters();

                if (sel!=null) {

                        getDatasourceComboBox().addItem(sel);

                }

        if (names == null) {

            return;

        }

    }

    public void initWizard() {

    }

    public void execute() {

            SingleVectorialDBConnectionExtension.saveAllToPersistence();

    }

    private TablesListItem[] getSelectedTables() {

        int count = tablesList.getModel().getSize();

        ArrayList resp = new ArrayList();

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

            TablesListItem item = (TablesListItem) tablesList.getModel()

                                                             .getElementAt(i);

            if (item.isSelected()) {

                resp.add(item);

            }

        }

        return (TablesListItem[]) resp.toArray(new TablesListItem[0]);

    }

    /**

     * This method initializes namePanel

     *

     * @return javax.swing.JPanel

     */

    private JPanel getNamePanel() {

        if (namePanel == null) {

            namePanel = new JPanel();

            namePanel.setLayout(null);

            namePanel.setBounds(new java.awt.Rectangle(5, 5, 501, 51));

            namePanel.setBorder(javax.swing.BorderFactory.createTitledBorder(

                    null, PluginServices.getText(this, "choose_connection"),

                    javax.swing.border.TitledBorder.DEFAULT_JUSTIFICATION,

                    javax.swing.border.TitledBorder.DEFAULT_POSITION, null, null));

            namePanel.add(getDatasourceComboBox(), null);

            namePanel.add(getJdbcButton(), null);

        }

        return namePanel;

    }

    /**

     * This method initializes tablesPanel

     *

     * @return javax.swing.JPanel

     */

    private JPanel getTablesPanel() {

        if (tablesPanel == null) {

            tablesPanel = new JPanel();

            tablesPanel.setLayout(new BorderLayout());

            tablesPanel.setBorder(javax.swing.BorderFactory.createTitledBorder(

                    null, PluginServices.getText(this, "choose_table"),

                    javax.swing.border.TitledBorder.DEFAULT_JUSTIFICATION,

                    javax.swing.border.TitledBorder.DEFAULT_POSITION, null, null));

            tablesPanel.setBounds(new java.awt.Rectangle(5, 55, 246, 191));

            tablesPanel.add(getTablesScrollPane(), java.awt.BorderLayout.CENTER);

        }

        return tablesPanel;

    }

    /**

     * This method initializes settingsPanel

     *

     * @return javax.swing.JPanel

     */

    /**

     * This method initializes tablesScrollPane

     *

     * @return javax.swing.JScrollPane

     */

    private JScrollPane getTablesScrollPane() {

        if (tablesScrollPane == null) {

            tablesScrollPane = new JScrollPane();

            tablesScrollPane.setViewportView(getTablesList());

        }

        return tablesScrollPane;

    }

    /**

     * This method initializes tablesList

     *

     * @return javax.swing.JList

     */

    private AvailableTablesCheckBoxList getTablesList() {

        if (tablesList == null) {

            tablesList = new AvailableTablesCheckBoxList(this);

            tablesList.addListSelectionListener(this);

            tablesList.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);

        }

        return tablesList;

    }

    /**

     * This method initializes layerNameTextField

     *

     * @return javax.swing.JTextField

     */

    /**

     * This method initializes jComboBox

     *

     * @return javax.swing.JComboBox

     */

    private JComboBox getDatasourceComboBox() {

        if (datasourceComboBox == null) {

            datasourceComboBox = new JComboBox();

            datasourceComboBox.setBounds(new java.awt.Rectangle(10, 20, 446, 21));

            datasourceComboBox.addActionListener(this);

        }

        return datasourceComboBox;

    }

    public void actionPerformed(ActionEvent arg0) {

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

                        setEmptyPanels();

                }

            Object src = arg0.getSource();

        if (src == datasourceComboBox) {

            Object sel_obj = datasourceComboBox.getSelectedItem();

            if (sel_obj == null) {

                return;

            }

            getDatasourceComboBox().repaint();

            return;

        }

        if (src == dbButton) {

            MyExplorer sel = addNewConnection();

            if (sel != null) {

                    dbExplorerParameters=sel.getDbExplorerParameters();

                    loadVectorialDBDatasourcesCombo(sel);

                getDatasourceComboBox().setSelectedItem(sel);

            }

            updateTableList(dbExplorerParameters);

        }

    }

    private MyExplorer addNewConnection() {

            MyExplorer myExplorer=new MyExplorer();

        DBExplorerParameters resp = null;

        VectorialDBConnectionParamsDialog newco = new VectorialDBConnectionParamsDialog();

        newco.showDialog();

        if (newco.isOkPressed()) {

                try {

                                resp=newco.getParameters();

                        } catch (InitializeException e) {

                                showConnectionErrorMessage(e.getMessage());

                        return null;

                        }

            SingleVectorialDBConnectionExtension.saveAllToPersistence();

            myExplorer.setDbExplorerParameters(resp);

            myExplorer.setName(newco.getConnectionName());

            return myExplorer;

        }

        else {

            return null;

        }

    }

    private void updateTableList(DBExplorerParameters src) {

          if (src==null) {

                  return;

          }

            DataManager dm=DataManager.getManager();

            DBExplorer dbExplorer;

            try {

                    dbExplorer = (DBExplorer)dm.createDataExplorer(src);

                    DBStoreParameters[] parameters = (DBStoreParameters[]) dbExplorer

                                        .list();

                    DefaultListModel lmodel = new DefaultListModel();

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

                            lmodel.addElement(new TablesListItem(dbExplorer,parameters[i],

                                            getMapCtrl(), this));

                    }

                    getTablesList().setModel(lmodel);

                    getTablesScrollPane().setViewportView(tablesList);

                    tablesScrollPane.updateUI();

            } catch (InitializeException e) {

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

                        NotificationManager.showMessageError("While getting table names: "

                                        + e.getMessage(), e);

                    return;

            } catch (ReadException e) {

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

                        NotificationManager.showMessageError("While getting table names: "

                                        + e.getMessage(), e);

                    return;

            }

    }

    public void valueChanged(ListSelectionEvent arg0) {

        Object src = arg0.getSource();

        if (src == tablesList) {

            TablesListItem selected = (TablesListItem) tablesList.getSelectedValue();

            setSettingsPanels(selected);

            checkFinishable();

        }

    }

    private boolean validFormSettings() {

        int count = tablesList.getModel().getSize();

        boolean at_least_one = false;

        boolean resp = true;

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

            TablesListItem item = (TablesListItem) tablesList.getModel()

                                                             .getElementAt(i);

            if (item.isSelected()) {

                at_least_one = true;

            }

            if (item.disturbsWizardValidity()) {

                resp = false;

            }

        }

        return (at_least_one && resp);

    }

    public void checkFinishable() {

        boolean finishable = validFormSettings();

        callStateChanged(finishable);

    }

    /**

     * This method initializes jdbcButton

     *

     * @return javax.swing.JButton

     */

    private JButton getJdbcButton() {

        if (dbButton == null) {

            dbButton = new JButton();

            dbButton.addActionListener(this);

            dbButton.setToolTipText(PluginServices.getText(this,

                    "add_connection"));

            dbButton.setBounds(new java.awt.Rectangle(465, 20, 26, 21));

            String _file = createResourceUrl("images/jdbc.png").getFile();

            dbButton.setIcon(new ImageIcon(_file));

        }

        return dbButton;

    }

    private void showConnectionErrorMessage(String _msg) {

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

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

        JOptionPane.showMessageDialog(this, title + msg, title,

            JOptionPane.ERROR_MESSAGE);

    }

    private java.net.URL createResourceUrl(String path) {

        return getClass().getClassLoader().getResource(path);

    }

    public void setSettingsPanels(TablesListItem actTable){

        if (actTable == null) {

            setEmptyPanels();

            return;

        }

        String abrev=null;

        if (getMapCtrl()!=null){

                abrev=getMapCtrl().getViewPort().getProjection()

                       .getAbrev();

        }

        settingsPanel = actTable.getUserTableSettingsPanel(abrev);

        fieldsPanel = actTable.getUserSelectedFieldsPanel();

        removeFieldPanels();

        add(fieldsPanel);

        fieldsPanel.repaint();

        removeSettingsPanels();

        add(settingsPanel);

        settingsPanel.repaint();

        repaint();

    }

    private void setEmptyPanels() {

        removeFieldPanels();

        add(emptyFieldsPanel);

        removeSettingsPanels();

        add(emptySettingsPanel);

        settingsPanel = emptySettingsPanel;

        fieldsPanel = emptyFieldsPanel;

        repaint();

    }

    private void removeFieldPanels() {

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

            if (getComponent(i) instanceof UserSelectedFieldsPanel) {

                remove(i);

            }

        }

    }

    private void removeSettingsPanels() {

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

            if (getComponent(i) instanceof UserTableSettingsPanel) {

                remove(i);

            }

        }

    }

        public DataStoreParameters[] getParameters() {

                try {

            TablesListItem[] selected = getSelectedTables();

            int count = selected.length;

            DBStoreParameters[] dbParameters = new DBStoreParameters[count];

            String strEPSG =null;

            if (getMapCtrl()!=null) {

                                strEPSG = getMapCtrl().getViewPort().getProjection()

                                 .getAbrev();

                        }

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

                TablesListItem item = selected[i];

                DBStoreParameters parameters = item.getParameters();

                UserTableSettingsPanel userTableSettingsPanel=item.getUserTableSettingsPanel(strEPSG);

                Envelope _wa = userTableSettingsPanel.getWorkingArea();

                                //                String[] tokens = item.getTableName().split("\\u002E", 2);

                                //                String tableName;

                                //                String schema = null;

                                //                if (tokens.length > 1)

                                //                {

                                //                        schema = tokens[0];

                                //                        tableName = tokens[1];

                                //                }

                                //                else

                                //                {

                                //                        tableName = tokens[0];

                                //                }

                String fidField = userTableSettingsPanel.getIdFieldName();

                String geomField = userTableSettingsPanel.getGeoFieldName();

                String[] fields = item.getUserSelectedFieldsPanel()

                        .getUserSelectedFields(fidField, geomField);

                                //                parameters.setSchema(schema);

                                //                parameters.setTableName(tableName);

                if (userTableSettingsPanel.isSqlActive()) {

                    String whereClause = userTableSettingsPanel.getWhereClause();

                    parameters.setBaseFilter(whereClause);

                }

                else {

                    parameters.setBaseFilter("");

                }

                parameters.setFields(fields);

                parameters.setDefaultGeometryField(geomField);

                parameters.setFieldsId(new String[] { fidField });

                if (_wa != null) {

                   parameters.setWorkingArea(_wa);

                }

                parameters.setSRISD(strEPSG);

                dbParameters[i]=parameters;

            }

            return dbParameters;//layerArrayToGroup(all_layers, groupName);

        }

        catch (Exception e) {

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

            NotificationManager.addError("Error al cargar la capa: " +

                e.getMessage(), e);

        }

        return null;

        }

} //  @jve:decl-index=0:visual-constraint="10,10"