Application: gvSIG desktop » gvSIG data providers based on gdal

/* 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.gdal.app.gpx.gui;

import java.awt.FlowLayout;

import java.awt.GridBagConstraints;

import java.awt.GridBagLayout;

import java.awt.event.MouseEvent;

import java.awt.event.MouseListener;

import java.io.File;

import java.net.URL;

import java.util.List;

import javax.swing.JButton;

import javax.swing.JLabel;

import javax.swing.JPanel;

import javax.swing.JScrollPane;

import javax.swing.JTextArea;

import org.gdal.gdal.gdal;

import org.gdal.ogr.DataSource;

import org.gdal.ogr.Feature;

import org.gdal.ogr.FeatureDefn;

import org.gdal.ogr.FieldDefn;

import org.gdal.ogr.Layer;

import org.gdal.ogr.ogr;

import org.gvsig.andami.PluginServices;

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

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.ApplicationManager;

import org.gvsig.app.gui.WizardPanel;

import org.gvsig.app.gui.wizards.WizardListener;

import org.gvsig.app.gui.wizards.WizardListenerSupport;

import org.gvsig.app.project.ProjectManager;

import org.gvsig.app.project.documents.view.ViewDocument;

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

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

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

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.store.gpx.OGRStoreParameters;

import org.gvsig.fmap.dal.store.gpx.OGRStoreProvider;

import org.gvsig.fmap.dal.store.gpx.OGRStoreProviderFactory;

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

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.gdal.app.gpx.gui.utils.JFileField;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * Wizard panel to add a new OSM raster layer

 * @author Jos? Bad?a (jbadia@scolab.es)

 */

public class GPXWizard extends WizardPanel implements MouseListener {

        private static final long           serialVersionUID      = 1L;

        private static final String         TAB_NAME              = "GDAL/OGR";

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

        private AbstractViewPanel           view                  = null;

        private WizardListenerSupport       listenerSupport       = null;

        private JPanel chooserPanel;

        private JPanel jPanel1;

        private JFileField cmbHost;

        private JButton btnConnect;

        private JPanel jPanel;

        private JScrollPane jScrollPane;

        private JTextArea txtAbstract;

        private JPanel pnlName;

        private JLabel jLabel1;

        private JLabel lblTitle;

        private DataSource poDS = null;

        private File file = null;

        private OGRStoreParameters ogrStoreParameters;

        public GPXWizard() {

                super();

                initialize();

        }

        /**

         * This method initializes this

         *

         * @return void

         */

        private void initialize() {

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

                setLayout(new GridBagLayout());

                gdal.AllRegister();

                ogr.RegisterAll();

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

                if (iw == null) {

                        return;

                }

                if (!(iw instanceof AbstractViewPanel)) {

                        return;

                }

                view = (AbstractViewPanel) iw;

                setMapCtrl(view.getMapControl());

                GridBagConstraints gridBagConstraints = new GridBagConstraints();

                gridBagConstraints.fill = GridBagConstraints.BOTH;

                gridBagConstraints.weightx = 1.0;

                gridBagConstraints.weighty = 1.0;

                gridBagConstraints.gridx = 0;

                gridBagConstraints.gridy = 0;

                add(getFileChooserPanel(), gridBagConstraints);

                gridBagConstraints.weightx = 1.0;

                gridBagConstraints.gridy = 1;

                gridBagConstraints.weighty = 4;

                add(getDescriptionPanel(), gridBagConstraints);

        }

        private WizardListenerSupport getWizardListenerSupport() {

                if(listenerSupport == null)

                        listenerSupport = new WizardListenerSupport();

                return listenerSupport;

        }

        /**

         * This method checks for the options selected within the configuration

         * dialog are correct.

         *

         * @return true if you're done, false if not.

         */

        public boolean areSettingsValid() {

                return true;

        }

        public void addWizardListener(WizardListener listener) {

                getWizardListenerSupport().addWizardListener(listener);

                getWizardListenerSupport().callStateChanged(false);

        }

        public void removeWizardListener(WizardListener listener) {

                getWizardListenerSupport().removeWizardListener(listener);

        }

        @Override

        public void close() {

        }

        @Override

        public void execute() {

                if(this.file != null){

                        isFileSupported(this.file);

                        // Si el fichero es soportado por OGR

                        if(poDS != null){

                                DataManager dataManager = DALLocator.getDataManager();

                                OGRStoreProviderFactory provFact = (OGRStoreProviderFactory) dataManager.getStoreProviderFactory(OGRStoreProvider.NAME);

                                OGRStoreParameters params = (OGRStoreParameters) provFact.createParameters();

                                ApplicationManager application = ApplicationLocator.getManager();

                                ProjectManager projectManager = application.getProjectManager();

                                ViewDocument view = (ViewDocument) projectManager.getCurrentProject().getActiveDocument();

                                try {

                                        FLyrVect layer = (FLyrVect) application.getMapContextManager().createLayer("prueba-ogr",  openFile(this.file));

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

                                } catch (LoadLayerException e) {

                                        e.printStackTrace();

                                }

                        }

                }else{

                        // TODO mensaje de error que no hay fichero escogido

                }

        }

        private void isFileSupported(File path) {

                if(path != null){

                        lblTitle.setText(path.getAbsolutePath());

                }else{

                        poDS = null;

                        lblTitle.setText("-");

                        return ;

                }

                String aux = "Running under GDAL/OGR "+ gdal.VersionInfo()  +" version.\n\n";

                txtAbstract.setText(aux);

                if(ogr.GetDriverCount() != 0){

                        poDS = ogr.Open(path.getAbsolutePath(), false);

                        if(poDS != null){

                                aux = aux + "Opened file '"+ path.getName()  +"' with "+ poDS.GetDriver().getName() +" driver.\n";

                        }else{

                                aux = aux + "ERROR: This file cannot be opened. Available drivers are the following:\n"; 

                                for(int i = 0; i<ogr.GetDriverCount(); i++){

                                        aux = aux + ogr.GetDriver(i).getName();

                                        if(i+1<ogr.GetDriverCount()){

                                                aux = aux + ",   ";

                                        }

                                }

                                txtAbstract.setText(aux);

                                return;

                        }

                }else{

                        aux = aux + "ERROR: GDAL/OGR driver are not initialized"; 

                }

                txtAbstract.setText(aux);

        }

        private JPanel getFileChooserPanel() {

                if (chooserPanel == null) {

                        chooserPanel = new javax.swing.JPanel();

                        chooserPanel.setLayout(null);

                        chooserPanel.setPreferredSize(new java.awt.Dimension(480, 220));

                        chooserPanel.setVisible(true);

                        chooserPanel.setBounds(15, 5, 480, 407);

                        chooserPanel.add(getFilePanel(), null);

                }

                return chooserPanel;

        }

        private JPanel getFilePanel() {

                if (jPanel1 == null) {

                        jPanel1 = new javax.swing.JPanel();

                        jPanel1.setLayout(null);

                        jPanel1.setBorder(javax.swing.BorderFactory.createTitledBorder(

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

                                        javax.swing.border.TitledBorder.DEFAULT_JUSTIFICATION,

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

                        jPanel1.setBounds(2, 5, 477, 85);

                        jPanel1.add(getTxtHost(), null);

                        jPanel1.add(getBtnConnect(), null);

                }

                return jPanel1;

        }

        private JFileField getTxtHost() {

                if (cmbHost == null) {

                        cmbHost = new JFileField(null, "...");

                        cmbHost.setMode(JFileField.MODE_OPEN);

                        cmbHost.getFileChooser().setCurrentDirectory(new File(System.getProperty("user.home")));

                        cmbHost.setPreferredSize(new java.awt.Dimension(350, 40));

                        cmbHost.setBounds(11, 26, 470, 30);

                        //cmbHost.setEditable(true);

                }

                return cmbHost;

        }

        private JButton getBtnConnect() {

                if (btnConnect == null) {

                        btnConnect = new JButton();

                        btnConnect.setPreferredSize(new java.awt.Dimension(100, 20));

                        btnConnect.setBounds(366, 60, 100, 20);

                        btnConnect.setText(PluginServices.getText(this, "Cargar"));

                        btnConnect.addActionListener(new java.awt.event.ActionListener() {

                                public void actionPerformed(java.awt.event.ActionEvent e) {

                                        file = cmbHost.getFileChooser().getSelectedFile();

                                        isFileSupported(file);

                                        String aux = txtAbstract.getText();

                                        DataManager dataManager = DALLocator.getDataManager();

                                        OGRStoreProviderFactory provFact = (OGRStoreProviderFactory) dataManager.getStoreProviderFactory(OGRStoreProvider.NAME);

                                        OGRStoreParameters params = (OGRStoreParameters) provFact.createParameters();

                                        params.setFile(file);

                                        OGRStoreProvider provider = null;

                                        try {

                                                provider = (OGRStoreProvider) provFact.createProvider(params, null);

                                        } catch (InitializeException e1) {

                                                // TODO Auto-generated catch block

                                                e1.printStackTrace();

                                        }

                                        if(provider != null){

                                                aux.concat("Running under GDAL/OGR "+ provider.getOGRVersion()  +" version.\n");

                                                try {

                                                        provider.open();

                                                        if(((OGRStoreParameters)provider.getParameters()).getDriver() != null){

                                                                aux.concat("Opened file '"+ file.getName()  +"' with "+ ((OGRStoreParameters)provider.getParameters()).getDriver() +" driver.\n");

                                                        }else{

                                                                List<String> availableDrivers = provider.getAvailableDrivers();

                                                                aux.concat("ERROR: This file cannot be opened. Available drivers are the following:\n"); 

                                                                for(int i = 0; i<provider.getDriverCount(); i++){

                                                                        aux.concat(provider.getDriver(i)+" ");

                                                                }

                                                        }

                                                } catch (OpenException e1) {

                                                        // TODO Auto-generated catch block

                                                        e1.printStackTrace();

                                                }

                                        }

                                        for(int index = 0; index < poDS.GetLayerCount();index++){

                                                Layer layer = poDS.GetLayer(index);

                                                for(int feats = 0; feats < layer.GetFeatureCount();feats++){

                                                        Feature feat = layer.GetFeature(feats);

                                                        FeatureDefn definition = feat.GetDefnRef();

                                                        for(int fields = 0; fields < definition.GetFieldCount();fields++){

                                                                FieldDefn field = definition.GetFieldDefn(fields);

                                                                String fieldName = field.GetName();

                                                                int fieldType = field.GetFieldType();

                                                                switch(fieldType){

                                                                // Ejemplo -> Type 1 : Integer

                                                                case 1:

                                                                        // Crear una feature en la vista

                                                                        int feature = feat.GetFieldAsInteger(field.GetFieldTypeName(fieldType));

                                                                        // A?adirla

                                                                        break;

                                                                default:

                                                                        break;

                                                                }

                                                        }

                                                }

                                        }

                                }

                        });

                }

                return btnConnect;

        }

        private JPanel getDescriptionPanel() {

                if (jPanel == null) {

                        jPanel = new JPanel();

                        jPanel.setLayout(null);

                        jPanel.setPreferredSize(new java.awt.Dimension(480, 330));

                        jPanel.setBorder(javax.swing.BorderFactory.createTitledBorder(null,

                                        PluginServices.getText(this, "descripcion"),

                                        javax.swing.border.TitledBorder.DEFAULT_JUSTIFICATION,

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

                        jPanel.setBounds(15, 5, 480, 407);

                        jPanel.add(getJScrollPane(), null);

                        jPanel.add(getPnlName(), null);

                }

                return jPanel;

        }

        private JScrollPane getJScrollPane() {

                if (jScrollPane == null) {

                        jScrollPane = new JScrollPane();

                        jScrollPane.setViewportView(getTxtAbstract());

                        jScrollPane.setPreferredSize(new java.awt.Dimension(450, 60));

                        jScrollPane.setBounds(10, 47, 457, 247);

                }

                return jScrollPane;

        }

        private JTextArea getTxtAbstract() {

                if (txtAbstract == null) {

                        txtAbstract = new JTextArea();

                        txtAbstract.setWrapStyleWord(true);

                        txtAbstract.setColumns(30);

                        txtAbstract.setLineWrap(true);

                        txtAbstract.setEditable(false);

                }

                return txtAbstract;

        }

        private JPanel getPnlName() {

                if (pnlName == null) {

                        pnlName = new JPanel();

                        pnlName.setLayout(new FlowLayout(FlowLayout.LEFT, 10, 0));

                        pnlName.setBounds(new java.awt.Rectangle(9, 19, 452, 24));

                        pnlName.add(getJLabel1(), null);

                        pnlName.add(getLblTitle(), null);

                }

                return pnlName;

        }

        private JLabel getJLabel1() {

                if (jLabel1 == null) {

                        jLabel1 = new javax.swing.JLabel();

                        jLabel1.setText(PluginServices.getText(this, "nombre") + ":");

                        jLabel1.setBounds(15, 26, 65, 15);

                }

                return jLabel1;

        }

        private JLabel getLblTitle() {

                if (lblTitle == null) {

                        lblTitle = new javax.swing.JLabel();

                        lblTitle.setText("-");

                        lblTitle.setPreferredSize(new java.awt.Dimension(350, 16));

                        lblTitle.setBounds(82, 26, 350, 16);

                }

                return lblTitle;

        }

        @Override

        public DataStoreParameters[] getParameters() {

                return null;

        }

        @Override

        public void initWizard() {

        }

        public void mouseClicked(MouseEvent e) {

        }

        public void mouseEntered(MouseEvent e) {

        }

        public void mouseExited(MouseEvent e) {

        }

        public void mousePressed(MouseEvent e) {

                getWizardListenerSupport().callStateChanged(true);

        }

        public void mouseReleased(MouseEvent e) {

        }

        /**

         * Get a resource as a File from a path name in the

         * class path.

         * 

         * @param pathname

         * 

         * @return resource as a File

         */

        private File getResource(String pathname) {

                URL res = this.getClass().getClassLoader().getResource(pathname);

                return new File(res.getPath());

        }

        private FeatureStore openFile(File shape) {

                //            try {

                //

                //                    DataStoreParameters parameters;

                //                    DataManager manager = DALLocator.getDataManager();

                //                    

                //                        parameters = manager.createStoreParameters("Shape");

                //                    parameters.setDynValue("shpfile", shape);

                //                    parameters.setDynValue("crs", "EPSG:23030");

                //                    return (FeatureStore) manager.openStore("Shape", parameters);

                //

                //            } catch (InitializeException e) {

                //                        logger.error(e.getMessageStack());

                //                        throw new RuntimeException(e);

                //                } catch (ProviderNotRegisteredException e) {

                //                        logger.error(e.getMessageStack());

                //                        throw new RuntimeException(e);

                //                } catch (ValidateDataParametersException e) {

                //                        logger.error(e.getMessageStack());

                //                        throw new RuntimeException(e);

                //                }

                DataManager dataManager = DALLocator.getDataManager();

                try {        

                        if(this.ogrStoreParameters == null){

                                this.ogrStoreParameters = 

                                                (OGRStoreParameters) dataManager.createStoreParameters(OGRStoreProvider.NAME);

                        }

                        this.ogrStoreParameters.setFile(file);

                        return (FeatureStore) dataManager.openStore(

                                        OGRStoreProvider.NAME, this.ogrStoreParameters);

                } catch (ValidateDataParametersException e) {

                        e.printStackTrace();

                } catch (InitializeException e) {

                        e.printStackTrace();

                } catch (ProviderNotRegisteredException e) {

                        e.printStackTrace();

                }

                return null;

        }

}