Application: gvSIG desktop

/*

 * Created on 25-oct-2005

 *

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

 * 

 * Copyright (C) 2004 IVER T.I. 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

 * 

 *   IVER T.I. S.A

 *   Salamanca 50

 *   46005 Valencia

 *   Spain

 * 

 *   +34 963163400

 *   dac@iver.es

 */

package com.iver.cit.gvsig.jdbc_spatial;

import java.sql.Connection;

import java.sql.DriverManager;

import java.sql.SQLException;

import java.sql.Statement;

import java.util.ArrayList;

import javax.swing.JOptionPane;

import com.hardcode.driverManager.DriverLoadException;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

import com.iver.andami.plugins.Extension;

import com.iver.cit.gvsig.fmap.DriverException;

import com.iver.cit.gvsig.fmap.MapControl;

import com.iver.cit.gvsig.fmap.core.FShape;

import com.iver.cit.gvsig.fmap.core.IFeature;

import com.iver.cit.gvsig.fmap.core.IGeometry;

import com.iver.cit.gvsig.fmap.drivers.DBLayerDefinition;

import com.iver.cit.gvsig.fmap.drivers.DriverIOException;

import com.iver.cit.gvsig.fmap.drivers.FieldDescription;

import com.iver.cit.gvsig.fmap.drivers.VectorialJDBCDriver;

import com.iver.cit.gvsig.fmap.layers.FLayer;

import com.iver.cit.gvsig.fmap.layers.FLyrVect;

import com.iver.cit.gvsig.fmap.layers.LayerFactory;

import com.iver.cit.gvsig.fmap.layers.SelectableDataSource;

import com.iver.cit.gvsig.fmap.layers.VectorialAdapter;

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

import com.iver.cit.gvsig.jdbc_spatial.gui.jdbcwizard.ConnectionSettings;

import com.iver.cit.gvsig.jdbc_spatial.util.PostGIS;

public class ImportLayer implements Extension {

    public void inicializar() {

        // TODO Auto-generated method stub

    }

    public void execute(String actionCommand) {

        View vista = (View) PluginServices.getMDIManager().getActiveView();

        MapControl mapCtrl = vista.getMapControl();

        FLayer[]  activeLayers = 

        mapCtrl.getMapContext().getLayers().getActives();

        FLyrVect lyrVect = (FLyrVect) activeLayers[0];

        VectorialAdapter adapter =  lyrVect.getSource();

        try {

            adapter.start();

            SelectableDataSource rsSel = lyrVect.getRecordset();

            int sc = adapter.getShapeCount();

            ArrayList newList = new ArrayList();

            DlgConnection dlg = new DlgConnection();

            dlg.setModal(true);

            dlg.setVisible(true);

            ConnectionSettings cs = dlg.getConnSettings();

            if (cs == null)

                return;

            Connection conex = DriverManager.getConnection(cs.getConnectionString(), cs.getUser(),

                    cs.getPassw());

            Statement st = conex.createStatement();

            DBLayerDefinition dbLayerDef = new DBLayerDefinition();

            dbLayerDef.setCatalogName(cs.getDb());

            dbLayerDef.setTableName(lyrVect.getName());

            String strGeometryFieldType = "GEOMETRY";

           /*  switch (lyrVect.getShapeType())

            {

                case FShape.POINT:

                    strGeometryFieldType = XTypes.fieldTypeToString(XTypes.POINT2D);

                    break;

                case FShape.LINE:

                    strGeometryFieldType = XTypes.fieldTypeToString(XTypes.LINE2D);

                    break;

                case FShape.POLYGON:

                    strGeometryFieldType = XTypes.fieldTypeToString(XTypes.POLYGON2D);

                    break;

                case FShape.MULTI:

                    strGeometryFieldType = XTypes.fieldTypeToString(XTypes.MULTI2D);

                    break;

            } */

            dbLayerDef.setFieldGeometry("the_geom");

            FieldDescription[] fieldsDescrip = new FieldDescription[rsSel.getFieldNames().length];

            dbLayerDef.setFieldNames(rsSel.getFieldNames());

            for (int i=0; i < rsSel.getFieldNames().length; i++)

            {

                fieldsDescrip[i] = new FieldDescription();

                fieldsDescrip[i].setFieldType(rsSel.getFieldType(i));

                fieldsDescrip[i].setFieldName(rsSel.getFieldName(i));

                // TODO: Por ahora le ponemos 200, a falta

                // de recompilar GDBMS con la posibilidad

                // de obtener el ancho de un campo.

                fieldsDescrip[i].setFieldLength(200);

            }

            String strSRID = lyrVect.getProjection().getAbrev().substring(5);

            dbLayerDef.setSRID_EPSG(strSRID);

            st.execute("DROP TABLE " + dbLayerDef.getTableName() + ";");

            String sqlCreate = PostGIS.getSqlCreateSpatialTable(dbLayerDef, fieldsDescrip);

            System.out.println("sqlCreate =" + sqlCreate);

            st.execute(sqlCreate);

            int dimension = 2;

            String sqlAlter = PostGIS.getSqlAlterTable(dbLayerDef, strSRID, strGeometryFieldType, dimension);

            System.out.println("sqlAlter =" + sqlAlter);

            st.execute(sqlAlter);

            // CREATE TABLE PARKS ( PARK_ID int4, PARK_NAME varchar(128), PARK_DATE date, PARK_TYPE varchar(2) );

            // SELECT AddGeometryColumn('parks_db', 'parks', 'park_geom', 128, 'MULTIPOLYGON', 2 );

            /* BEGIN;

            INSERT INTO ROADS_GEOM (ID,GEOM,NAME ) VALUES (1,GeometryFromText('LINESTRING(191232 243118,191108 243242)',-1),'Jeff Rd'); 

            INSERT INTO ROADS_GEOM (ID,GEOM,NAME ) VALUES (2,GeometryFromText('LINESTRING(189141 244158,189265 244817)',-1),'Geordie Rd'); 

            COMMIT; */       

            conex.setAutoCommit(false);

            for (int i = 0; i < sc; i++)

            {

                // IGeometry geom = adapter.getShape(i);

                IFeature feat = adapter.getFeature(i);

                String sqlInsert = PostGIS.getSqlInsertFeature(dbLayerDef, feat);

                // System.out.println("sqlInsert =" + sqlInsert);

                st.execute(sqlInsert);

            }

            conex.setAutoCommit(true);

            conex.close();

            adapter.stop();

            JOptionPane.showMessageDialog(null, "Capa importada.");

        } catch (Exception e) {

            NotificationManager.addError("Error en la carga.", e);

        }

    }

    private String[] getDriverNames(){

        Class[] classes = new Class[] { VectorialJDBCDriver.class };

        ArrayList ret = new ArrayList();

        String[] driverNames = LayerFactory.getDM().getDriverNames();

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

            boolean is = false;

            for (int j = 0; j < classes.length; j++) {

                if (LayerFactory.getDM().isA(driverNames[i], classes[j])) {

                    ret.add(driverNames[i]);

                }

            }

        }

        return (String[]) ret.toArray(new String[0]);

    }

    public boolean isEnabled() {

        // TODO Auto-generated method stub

        return true;

    }

    public boolean isVisible() {

        // TODO Auto-generated method stub

        return true;

    }

}