Application: gvSIG desktop

/*

 * Created on 03-mar-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.fmap.drivers.jdbc.postgis;

import java.awt.geom.Rectangle2D;

import java.sql.Connection;

import java.sql.DriverManager;

import java.sql.PreparedStatement;

import java.sql.ResultSet;

import java.sql.ResultSetMetaData;

import java.sql.SQLException;

import java.sql.Statement;

import java.util.Enumeration;

import org.cresques.cts.IProjection;

import org.cresques.cts.ProjectionPool;

import org.postgresql.fastpath.Fastpath;

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

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

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

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

import com.iver.cit.gvsig.fmap.drivers.jdbc.WKBParser;

import com.iver.cit.gvsig.fmap.drivers.jdbc.WKBParser2;

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

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

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

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

import com.iver.cit.gvsig.fmap.layers.layerOperations.SingleLayer;

/**

 * @author FJP

 *

 * TODO To change the template for this generated type comment go to

 * Window - Preferences - Java - Code Generation - Code and Comments

 */

public class testPostGis {

      public static void main(String[] args) 

      { 

      /*    System.err.println("dburl has the following format:");

          System.err.println("jdbc:postgresql://HOST:PORT/DATABASENAME");

          System.err.println("tablename is 'jdbc_test' by default.");

          System.exit(1); */

          String dburl = "jdbc:postgresql://localhost/latin1";

          String dbuser = "postgres";

          String dbpass = "aquilina";

          String dbtable = "vias";

          Connection conn = null; 

          System.out.println("Creating JDBC connection...");

          try {

            Class.forName("org.postgresql.Driver");

              Enumeration enumDrivers = DriverManager.getDrivers();

              while (enumDrivers.hasMoreElements())

              {

                  System.out.println("Driver " + enumDrivers.nextElement().toString());

              }

              conn = DriverManager.getConnection(dburl, dbuser, dbpass);

              conn.setAutoCommit(false);

              long t1 = System.currentTimeMillis();

              // test1(conn, dburl, dbuser, dbpass, dbtable);

              long t2 = System.currentTimeMillis();          

              System.out.println("Tiempo de consulta1:" + (t2 - t1) + " milisegundos");

              FLyrVect lyr = initLayerPostGIS();

              t1 = System.currentTimeMillis();

              test4(lyr);

              t2 = System.currentTimeMillis();

              System.out.println("Tiempo de consulta2:" + (t2 - t1) + " milisegundos");

              conn.close();

          } catch (ClassNotFoundException e) {

              // TODO Auto-generated catch block

              e.printStackTrace();

          } catch (SQLException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

          }

      }

      private static void test1(Connection conn, String dburl, String dbuser, String dbpass, String dbtable)

      {          

          try 

          { 

              // magic trickery to be pgjdbc 7.2 compatible

              // This works due to the late binding of data types in most java VMs. As

              // this is more a demo source than a real-world app, we can risk this

              // problem.

              /* if (conn.getClass().getName().equals("org.postgresql.jdbc2.Connection")) {

                  ((org.postgresql.Connection) conn).addDataType("geometry", "org.postgis.PGgeometry");

                  ((org.postgresql.Connection) conn).addDataType("box3d", "org.postgis.PGbox3d");

              } else {

                  ((org.postgresql.PGConnection) conn).addDataType("geometry", "org.postgis.PGgeometry");

                  ((org.postgresql.PGConnection) conn).addDataType("box3d", "org.postgis.PGbox3d");

              } */

            /* 

            * Create a statement and execute a select query. 

            */

              // String strSQL = "select AsBinary(the_geom) as geom, nom_provin from " + dbtable;

              String strSQL = "select ASBINARY(the_geom) as geom, gid from " + dbtable;

              // strSQL = "select ASTEXT(the_geom), nom_provin as geom from " + dbtable;

              /* String strSQL = "SELECT gid, rd_3, rd_5, rd_6, rd_10, rd_11, rd_12, rd_13, rd_14,"; 

              strSQL = strSQL + " rd_15, rd_16, kilometers, cost, metros, AsText(force_2d(the_geom)) FROM vias"; 

              strSQL = strSQL + " WHERE TRUE";

              */

              // PreparedStatement s = conn.prepareStatement(strSQL);

              Statement s = conn.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_READ_ONLY);

              // s.execute("begin");

              // Statement s = conn.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_READ_ONLY);

              // s.setFetchSize(5);

              int fetchSize = 150000;

            s.execute("declare wkb_cursor binary cursor for " + strSQL);

            ResultSet r =   s.executeQuery("fetch forward " + fetchSize + " in wkb_cursor"); 

              /// ResultSet r =   s.executeQuery("fetch forward all in wkb_cursor");

            // String strSQL2 = "select AsBinary(the_geom) as geom, nom_provin from " + dbtable;

            // PreparedStatement ps = conn.prepareStatement(strSQL2);

            // ResultSet r = ps.executeQuery();

              // ResultSet r = s.executeQuery(strSQL);

            WKBParser2 parser2 = new WKBParser2();

            WKBParser parser = new WKBParser();

            long id=0;

            /* for (int i=1; i< 100; i++)

            {

                r.absolute(i);

                System.out.println("Row " + i + ":" + r.getString(2));

            }

            r.beforeFirst();

              for (int i=1; i< 100; i++)

              {

                  r.absolute(i);

                  System.out.println("Row " + i + ":" + r.getString(2));

              } */

            while( r.next() ) 

            { 

              /* 

              * Retrieve the geometry as an object then cast it to the geometry type. 

              * Print things out. 

              */ 

                // Object obj = r.getObject(2);

                byte[] arrayByte = r.getBytes(1);

                // IGeometry gp = parser.parse(arrayByte);

                IGeometry gp2 = parser2.parse(arrayByte);

                // String strAux = r.getString(2);

                // int id = r.getInt(2);

                // System.out.println("Fila " + id + ":" + obj.toString());

                id++;

                // Geometry regeom = PGgeometry.geomFromString(obj.toString());

              // PGgeometry geom = (PGgeometry)obj; 

               // int id = r.getInt(2);

              // System.out.println("Row " + id + ":" + strAux); 

              // System.out.println(geom.toString()); 

                // System.out.println("provin=" + r.getString(2));

                /* if ((id % fetchSize) == 0)

                {

                    r =   s.executeQuery("fetch forward " + fetchSize + " in wkb_cursor");

                } */

            }

            // s.execute("end");

            s.close();

          } 

          catch( Exception e ) 

          { 

            e.printStackTrace(); 

          }  

        }

      private static void test2(Connection conn, String dburl, String dbuser, String dbpass, String dbtable)

      {          

          try 

          { 

            /* 

            * Create a statement and execute a select query. 

            */

              String strSQL = "select gid from " + dbtable;

              PreparedStatement s = conn.prepareStatement(strSQL);

              // Statement s = conn.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_READ_ONLY);

              int fetchSize = 5000;

              ResultSet r = s.executeQuery(strSQL);

              int id=0;

              while( r.next() ) 

              { 

                String strAux = r.getString(1);

                id++;

                // System.out.println("Row " + id + ":" + strAux); 

              } 

              s.close();

          } 

          catch( Exception e ) 

          { 

            e.printStackTrace(); 

          }  

        }

      private static void test3(Connection conn, String dburl, String dbuser, String dbpass, String dbtable)

      {          

          try 

          { 

                  Fastpath  fp;

                  if (conn.getClass().getName().equals("org.postgresql.jdbc2.Connection")) {

                          // ((org.postgresql.Connection) conn).addDataType("geometry", "org.postgis.PGgeometry");

                          // ((org.postgresql.Connection) conn).addDataType("box3d", "org.postgis.PGbox3d");

                  } else {

                          ((org.postgresql.PGConnection) conn).addDataType("geometry", "org.postgis.PGgeometry");

                          ((org.postgresql.PGConnection) conn).addDataType("box3d", "org.postgis.PGbox3d");

                          fp =  ((org.postgresql.PGConnection) conn).getFastpathAPI();

                  } 

            /* 

            * Create a statement and execute a select query. 

            */

              String strSQL = "select * from " + dbtable;

              // String strSQL = "select ASBINARY(the_geom) as geom from " + dbtable;

              // strSQL = "select ASTEXT(the_geom), nom_provin as geom from " + dbtable;

              /* String strSQL = "SELECT gid, rd_3, rd_5, rd_6, rd_10, rd_11, rd_12, rd_13, rd_14,"; 

              strSQL = strSQL + " rd_15, rd_16, kilometers, cost, metros, AsText(force_2d(the_geom)) FROM vias"; 

              strSQL = strSQL + " WHERE TRUE";

              */

              // PreparedStatement s = conn.prepareStatement(strSQL);

              Statement s = conn.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_READ_ONLY);

              // s.execute("begin");

              // Statement s = conn.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_READ_ONLY);

              // s.setFetchSize(5);

            /*  int fetchSize = 150000;

            s.execute("declare wkb_cursor2 binary cursor for " + strSQL);

            ResultSet r =   s.executeQuery("fetch forward " + fetchSize + " in wkb_cursor2"); */

            // String strSQL2 = "select AsBinary(the_geom) as geom, nom_provin from " + dbtable;

            // PreparedStatement ps = conn.prepareStatement(strSQL2);

            // ResultSet r = ps.executeQuery();

            ResultSet r = s.executeQuery(strSQL);

            WKBParser parser = new WKBParser();

            long id=0;

            /* FastpathArg args[] = new FastpathArg[2];

            args[0] = new FastpathArg(fd);

            args[1] = new FastpathArg(len);

            return fp.getData("loread", args); */  

            while( r.next() ) 

            { 

              /* 

              * Retrieve the geometry as an object then cast it to the geometry type. 

              * Print things out. 

              */ 

                Object obj = r.getObject(9);

                // fp.

                    // final Shape current = (Shape) r.getObject(1);

                // byte[] arrayByte = r.getBytes(1);

                // String strAux = r.getString(2);

                id++;

                // Geometry regeom = PGgeometry.geomFromString(obj.toString());

              // PGgeometry geom = (PGgeometry)obj; 

               // int id = r.getInt(2);

              // System.out.println("Row " + id + ":" + strAux); 

              // System.out.println(geom.toString()); 

                // System.out.println("provin=" + r.getString(2));

                /* if ((id % fetchSize) == 0)

                {

                    r =   s.executeQuery("fetch forward " + fetchSize + " in wkb_cursor");

                } */

            }

            // s.execute("end");

            s.close();

          } 

          catch( Exception e ) 

          { 

            e.printStackTrace(); 

          }  

      }

      private static void test4(FLyrVect lyr)

      {

            try

            {

                    ISpatialDB dbAdapter = (ISpatialDB) lyr.getSource();

            VectorialDatabaseDriver dbDriver = (VectorialDatabaseDriver) dbAdapter.getDriver();

                IFeatureIterator geomIt = dbDriver.getFeatureIterator(lyr.getFullExtent(), "23030");

                while (geomIt.hasNext())

                {

                        IFeature feat = geomIt.next();

                        IGeometry geom = feat.getGeometry();

                }

                  }

                  catch(Exception e)

                  {

                          e.printStackTrace();

                  }

      }

        private static FLyrVect initLayerPostGIS()

        {

        String dbURL = "jdbc:postgresql://localhost:5432/latin1"; // latin1 is the catalog name 

        String user = "postgres";

        String pwd = "aquilina";

        String layerName = "vias";

        String tableName = "vias";

        Connection conn;

        LayerFactory.setDriversPath("D:/eclipse/workspace/_fwAndami/gvSIG/extensiones/com.iver.cit.gvsig/drivers");

                try {

                        conn = DriverManager.getConnection(dbURL, user, pwd);

                conn.setAutoCommit(false);

                String fidField = "gid"; // BE CAREFUL => MAY BE NOT!!!

                String geomField = "the_geom"; // BE CAREFUL => MAY BE NOT!!! => You should read table GEOMETRY_COLUMNS.

                                                                                // See PostGIS help.

                // To obtain the fields, make a connection and get them. 

                        /* Statement st = conn.createStatement();

                        ResultSet rs = st.executeQuery("select * from " + tableName + " LIMIT 1");

                        ResultSetMetaData rsmd = rs.getMetaData();

                        String[] fields = new String[rsmd.getColumnCount()-1]; // We don't want to include the_geom field

                        int j = 0;

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

                                if (!rsmd.getColumnName(i+1).equalsIgnoreCase(geomField))

                                {

                                        fields[j++] = rsmd.getColumnName(i+1);                                        

                                }

                        }

                        rs.close(); */        

                String[] fields = new String[1];                

                fields[0] = "gid"; 

                String whereClause = "";

                VectorialJDBCDriver driver = (VectorialJDBCDriver) LayerFactory.getDM()

                                        .getDriver("PostGIS JDBC Driver");

                // Here you can set the workingArea

                // driver.setWorkingArea(dbLayerDefinition.getWorkingArea());

                String strEPSG = "23030";

                DBLayerDefinition lyrDef = new DBLayerDefinition();                

                lyrDef.setName(layerName);

                lyrDef.setTableName(tableName);

                lyrDef.setWhereClause(whereClause);

                lyrDef.setFieldNames(fields);

                lyrDef.setFieldGeometry(geomField);

                lyrDef.setFieldID(fidField);

                // if (dbLayerDefinition.getWorkingArea() != null)

                //     lyrDef.setWorkingArea(dbLayerDefinition.getWorkingArea());

                lyrDef.setSRID_EPSG(strEPSG);

                if (driver instanceof ICanReproject)

                {                    

                    ((ICanReproject)driver).setDestProjection(strEPSG);

                }

                driver.setData(conn, lyrDef);

                IProjection proj = null; 

                if (driver instanceof ICanReproject)

                {                                        

                    proj = ProjectionPool.get("EPSG:" + ((ICanReproject)driver).getSourceProjection()); 

                }

                FLayer lyr = LayerFactory.createDBLayer(driver, layerName, proj);

                Rectangle2D rectAux = lyr.getFullExtent();

                return (FLyrVect) lyr;

                }

                catch(Exception e)

                {

                        e.printStackTrace();

                }

                return null;

        }

}