Application: gvSIG desktop » gvSIG data providers based on gdal

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2016 gvSIG Association

 *

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

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.gdal.app.ogr.mainplugin.gui;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.io.File;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Collections;

import java.util.List;

import java.util.Locale;

import java.util.prefs.Preferences;

import javax.swing.JComponent;

import javax.swing.JFileChooser;

import javax.swing.JOptionPane;

import javax.swing.ListSelectionModel;

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.ListSelectionListener;

import javax.swing.table.TableModel;

import org.apache.commons.lang.StringUtils;

import org.apache.commons.lang3.ArrayUtils;

import org.gvsig.filedialogchooser.FileDialogChooser;

import org.gvsig.filedialogchooser.FileDialogChooserManager;

import org.gvsig.fmap.IconThemeHelper;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.DataStoreProviderFactory;

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 static org.gvsig.fmap.dal.serverexplorer.filesystem.swing.FilesystemExplorerWizardPanelController.OPEN_LAYER_FILE_CHOOSER_ID;

import org.gvsig.fmap.mapcontrol.swing.dynobject.DynObjectEditor;

import org.gvsig.gdal.prov.ogr.OGRDataExplorer;

import org.gvsig.gdal.prov.ogr.OGRDataExplorerParameters;

import org.gvsig.gdal.prov.ogr.OGRDataStoreParameters;

import org.gvsig.gdal.prov.ogr.OGRDataStoreProvider;

import org.gvsig.gdal.prov.ogr.OGRDataStoreProviderFactory;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.folders.FoldersManager;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.service.ServiceException;

import org.gvsig.tools.util.ToolsUtilLocator;

import org.gvsig.tools.util.ToolsUtilManager;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author <a href="mailto:lmarques@disid.com">Lluis Marques</a>

 *

 */

public class JOGRDataExplorerController extends JOGRDataExplorerView implements JOGRDataExplorer {

    private static final Logger LOG = LoggerFactory.getLogger(JOGRDataExplorerController.class);

    private static final long serialVersionUID = 2347040531301288139L;

    private String lastConnectionString;

    private OGRDataExplorerParameters dataExplorerParameters;

    /**

     *

     */

    public JOGRDataExplorerController() {

        init();

    }

    private void init() {

        initFileChooserButton();

        initDataExplorerPropertiesButton();

        initOpenButton();

        initPropertiesButton();

        initLayerTable();

        translate();

    }

    private void initDataExplorerPropertiesButton() {

        this.dataExplorerPropertiesButton.setIcon(IconThemeHelper

            .getImageIcon("ogr-dataexplorer-settings"));

        this.dataExplorerPropertiesButton.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                String connectionString = connectionField.getText();

                if (StringUtils.isBlank(connectionString)) {

                    lastConnectionString = null;

                    I18nManager i18nManager = ToolsLocator.getI18nManager();

                    JOptionPane.showMessageDialog(JOGRDataExplorerController.this,

                        i18nManager.getTranslation("connection_string_can_not_be_empty"),

                        i18nManager.getTranslation("error"), JOptionPane.ERROR_MESSAGE);

                    return;

                }

                if (!getConnectionString().equals(lastConnectionString)) {

                    setExplorerParameters();

                }

                try {

                    DynObjectEditor editor = new DynObjectEditor(dataExplorerParameters);

                    editor.editObject(true);

                } catch (ServiceException e1) {

                    LOG.error(String.format("Can not create DynObjectEditor with %1s",

                        dataExplorerParameters), e);

                    I18nManager i18nManager = ToolsLocator.getI18nManager();

                    JOptionPane.showMessageDialog(JOGRDataExplorerController.this,

                        i18nManager.getTranslation("error_creating_properties_panel"),

                        i18nManager.getTranslation("error"), JOptionPane.ERROR_MESSAGE);

                    return;

                }

            }

        });

    }

    @SuppressWarnings("unchecked")

    private void setExplorerParameters() {

        DataManager dataManager = DALLocator.getDataManager();

        List<String> storeProviders = dataManager.getStoreProviders();

        for (String storeProviderName : storeProviders) {

            // First we have to search specific data explorer

            // parameters

            // to open connection string so we have to avoid create

            // default OGR parameters.

            if (storeProviderName.equals(OGRDataStoreProvider.NAME)) {

                continue;

            }

            DataStoreProviderFactory factory =

                dataManager.getStoreProviderFactory(storeProviderName);

            if (factory instanceof OGRDataStoreProviderFactory) {

                OGRDataStoreProviderFactory ogrFactory = (OGRDataStoreProviderFactory) factory;

                DynObject explorerParameters =

                    ogrFactory.createDataExplorerParameters(getConnectionString());

                dataExplorerParameters = (OGRDataExplorerParameters) explorerParameters;

            }

        }

        // If there are not specific data explorer parameters,

        // create

        // default OGR data explorer parameters.

        if (dataExplorerParameters == null) {

            OGRDataStoreProviderFactory factory =

                (OGRDataStoreProviderFactory) dataManager

                    .getStoreProviderFactory(OGRDataStoreProvider.NAME);

            dataExplorerParameters =

                (OGRDataExplorerParameters) factory

                    .createDataExplorerParameters(getConnectionString());

        }

        File file = new File(getConnectionString());

        if (file.exists() && file.isFile()) {

            dataExplorerParameters.setFile(file);

        } else {

            dataExplorerParameters.setConnectionString(getConnectionString());

        }

        lastConnectionString = getConnectionString();

    }

    private void initLayerTable() {

        layerTable.setSelectionMode(ListSelectionModel.MULTIPLE_INTERVAL_SELECTION);

        layerTable.getSelectionModel().addListSelectionListener(new ListSelectionListener() {

            @Override

            public void valueChanged(ListSelectionEvent e) {

                if (layerTable.getSelectedRowCount() == 1) {

                    propertiesButton.setEnabled(true);

                } else {

                    propertiesButton.setEnabled(false);

                }

            }

        });

    }

    private void initOpenButton() {

        this.openButton.addActionListener(new ActionListener() {

            @SuppressWarnings("unchecked")

            @Override

            public void actionPerformed(ActionEvent e) {

                String connectionString = connectionField.getText();

                if (StringUtils.isBlank(connectionString)) {

                    lastConnectionString = null;

                    I18nManager i18nManager = ToolsLocator.getI18nManager();

                    JOptionPane.showMessageDialog(JOGRDataExplorerController.this,

                        i18nManager.getTranslation("connection_string_can_not_be_empty"),

                        i18nManager.getTranslation("error"), JOptionPane.ERROR_MESSAGE);

                    return;

                }

                DataManager dataManager = DALLocator.getDataManager();

                OGRDataExplorer serverExplorer = null;

                if (!getConnectionString().equals(lastConnectionString)) {

                    setExplorerParameters();

                }

                // If there are not specific data explorer parameters,

                // create

                // default OGR data explorer parameters.

                if (dataExplorerParameters == null) {

                    OGRDataStoreProviderFactory factory =

                        (OGRDataStoreProviderFactory) dataManager

                            .getStoreProviderFactory(OGRDataStoreProvider.NAME);

                    dataExplorerParameters =

                        (OGRDataExplorerParameters) factory

                            .createDataExplorerParameters(connectionString);

                }

                File file = new File(connectionString);

                if (file.exists() && file.isFile()) {

                    dataExplorerParameters.setFile(file);

                } else {

                    dataExplorerParameters.setConnectionString(connectionString);

                }

                lastConnectionString = connectionString;

                try {

                    serverExplorer =

                        (OGRDataExplorer) dataManager.openServerExplorer(

                            dataExplorerParameters.getExplorerName(), dataExplorerParameters);

                } catch (InitializeException ex) {

                    LOG.error("Can not create OGR server explorer parameters", ex);

                    I18nManager i18nManager = ToolsLocator.getI18nManager();

                    JOptionPane.showMessageDialog(JOGRDataExplorerController.this,

                        i18nManager.getTranslation("error_opening_connection_string"),

                        i18nManager.getTranslation("error"), JOptionPane.ERROR_MESSAGE);

                    return;

                } catch (ProviderNotRegisteredException ex) {

                    LOG.error("OGR Provider is not registered", ex);

                    I18nManager i18nManager = ToolsLocator.getI18nManager();

                    JOptionPane.showMessageDialog(JOGRDataExplorerController.this,

                        i18nManager.getTranslation("error_getting_OGR_provider"),

                        i18nManager.getTranslation("error"), JOptionPane.ERROR_MESSAGE);

                    return;

                } catch (ValidateDataParametersException ex) {

                    LOG.error(String.format("Can not open OGR server explorer. Parameters: %1s",

                        connectionString), ex);

                    I18nManager i18nManager = ToolsLocator.getI18nManager();

                    JOptionPane.showMessageDialog(JOGRDataExplorerController.this,

                        i18nManager.getTranslation("error_opening_connection_string"),

                        i18nManager.getTranslation("error"), JOptionPane.ERROR_MESSAGE);

                    return;

                }

                List storeParametersList;

                try {

                    storeParametersList = serverExplorer.list();

                } catch (DataException ex) {

                    LOG.error("Can not get layer list of OGR Data Explorer", ex);

                    I18nManager i18nManager = ToolsLocator.getI18nManager();

                    JOptionPane.showMessageDialog(JOGRDataExplorerController.this,

                        i18nManager.getTranslation("error_getting_layers_from_connection_string"),

                        i18nManager.getTranslation("error"), JOptionPane.ERROR_MESSAGE);

                    return;

                }

                layerTable.setModel(new OGRDataStoreParameterTableModel(storeParametersList));

            }

        });

    }

    private void initPropertiesButton() {

        this.propertiesButton.setEnabled(false);

        this.propertiesButton.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent arg0) {

                OGRDataStoreParameterTableModel model =

                    (OGRDataStoreParameterTableModel) layerTable.getModel();

                OGRDataStoreParameters parameters =

                    model.getDataStoreParameter(layerTable.getSelectedRow());

                try {

                    DynObjectEditor editor = new DynObjectEditor(parameters);

                    editor.editObject(true);

                } catch (ServiceException e) {

                    LOG.error(String.format("Can not create DynObjectEditor with %1s", parameters),

                        e);

                    I18nManager i18nManager = ToolsLocator.getI18nManager();

                    JOptionPane.showMessageDialog(JOGRDataExplorerController.this,

                        i18nManager.getTranslation("error_creating_properties_panel"),

                        i18nManager.getTranslation("error"), JOptionPane.ERROR_MESSAGE);

                    return;

                }

            }

        });

    }

    private void initFileChooserButton() {

        fileChooserButton.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                FoldersManager foldersManager = ToolsLocator.getFoldersManager();

                FileDialogChooserManager fileDialogChooserManager = ToolsUtilLocator.getFileDialogChooserManager();

                FileDialogChooser chooser = fileDialogChooserManager.create(OPEN_LAYER_FILE_CHOOSER_ID);

                chooser.setCurrentDirectory(foldersManager.get("DataFolder"));

                if (StringUtils.isNotBlank(getConnectionString())) {

                    File file = new File(getConnectionString());

                    if (file.exists()) {

                        chooser.setCurrentDirectory(file.getParentFile());

                    }

                }

                int reply = chooser.showOpenDialog(asJComponent());

                if (reply == JFileChooser.APPROVE_OPTION) {

                    File file = chooser.getSelectedFile();

                    if (file.isFile() && file.exists() && file.canRead()) {

                        connectionField.setText(file.getAbsolutePath());

                    }

                }

            }

        });

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    @Override

    public void setLocale(Locale l) {

        if (!this.getLocale().equals(l)) {

            this.translate();

        }

        super.setLocale(l);

    }

    private void translate() {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        this.connectionLabel.setText(i18nManager.getTranslation(this.connectionLabel.getText()));

        this.openButton.setText(i18nManager.getTranslation(this.openButton.getText()));

        this.propertiesButton.setText(i18nManager.getTranslation(this.propertiesButton.getText()));

    }

    @Override

    public List<String> getSelectedLayerNames() {

        List<String> selectedLayerNames = new ArrayList<String>();

        OGRDataStoreParameterTableModel model = getOGRTableModel();

        int[] selectedRows = layerTable.getSelectedRows();

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

            OGRDataStoreParameters dataStoreParameter =

                model.getDataStoreParameter(selectedRows[i]);

            selectedLayerNames.add(dataStoreParameter.getLayerName());

        }

        return selectedLayerNames;

    }

    public List<DataStoreParameters> getSelectedLayerProperties() throws InitializeException,

        ProviderNotRegisteredException {

        OGRDataStoreParameterTableModel model = getOGRTableModel();

        if( model == null ) {

            return Collections.EMPTY_LIST;

        }

        int[] selectedRows = layerTable.getSelectedRows();

        List<DataStoreParameters> parameters = new ArrayList<DataStoreParameters>();

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

            parameters.add(model.getDataStoreParameter(selectedRows[i]));

        }

        return parameters;

    }

    @Override

    public List<Integer> getSelectedLayerIndexes() {

        int[] selectedRows = layerTable.getSelectedRows();

        return Arrays.asList(ArrayUtils.toObject(selectedRows));

    }

    @Override

    public String getConnectionString() {

        return connectionField.getText();

    }

    private OGRDataStoreParameterTableModel getOGRTableModel() {

        TableModel model = layerTable.getModel();

        if( model instanceof OGRDataStoreParameterTableModel) {

            return (OGRDataStoreParameterTableModel) model;

        }

        return null;

    }

}