Application: gvSIG desktop

/*

 * Created on 13-may-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.sde;

import java.awt.geom.Rectangle2D;

import java.io.IOException;

import java.sql.Types;

import java.util.Hashtable;

import java.util.Vector;

import com.esri.sde.sdk.client.SeColumnDefinition;

import com.esri.sde.sdk.client.SeConnection;

import com.esri.sde.sdk.client.SeException;

import com.esri.sde.sdk.client.SeExtent;

import com.esri.sde.sdk.client.SeFilter;

import com.esri.sde.sdk.client.SeLayer;

import com.esri.sde.sdk.client.SeObjectId;

import com.esri.sde.sdk.client.SeQuery;

import com.esri.sde.sdk.client.SeQueryInfo;

import com.esri.sde.sdk.client.SeRow;

import com.esri.sde.sdk.client.SeShape;

import com.esri.sde.sdk.client.SeShapeFilter;

import com.esri.sde.sdk.client.SeSqlConstruct;

import com.esri.sde.sdk.client.SeTable;

import com.hardcode.gdbms.engine.data.driver.ObjectDriver;

import com.hardcode.gdbms.engine.data.edition.DataWare;

import com.hardcode.gdbms.engine.values.Value;

import com.hardcode.gdbms.engine.values.ValueFactory;

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

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

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

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

public class ArcSdeDriver implements VectorialDatabaseDriver, ObjectDriver {

    private SeLayer layer;

    private SeQuery query;

    private SeQuery queryAux;

    private SeSqlConstruct sqlConstruct;

    private SeColumnDefinition[] colDefs;

    private SeRow row = null;

    private int numReg = -1;

    private Rectangle2D fullExtent = null;

    private Hashtable hashRelate = null;

    private int idSpatialColumn = -1;

    private String[] fields;

    private int[] fieldTypes;

    private long lastRowServed = -1;

    private SeConnection conn = null;

    /**

     * Recorre el recordset creando una tabla Hash que usaremos para 

     * relacionar el n?mero de un registro con su identificador ?nico.

     * Debe ser llamado en el setData justo despu?s de crear el recorset

     * principal

     */

    private void doRelateID_FID()

    {

        hashRelate = new Hashtable();

        try {

            // query.execute();

            // row = query.fetch();

            if( row == null ) {

                System.out.println(" No rows fetched");

                return;

            }

            int index = 0;

            while( row != null ) 

            {                 

                SeShape shpVal = row.getShape(idSpatialColumn);

                SeObjectId objID = shpVal.getFeatureId();

                String theKey = "" + objID.longValue();

                // System.out.println("Antes de meter en el hash " + theKey);

                hashRelate.put(theKey, new Integer(index));

                row = query.fetch();

                index++;

            }                

            numReg = index;

            // Close the query.    

            // query.close();

        } catch (SeException e)

        {

            e.printStackTrace();

        }

    }

    public void setData(String host, int port, String db, String userDB, String pwdDB,

            String tableName, String[] cols,

            String whereClause)

    {

        // Conexi?n:

        String server     = host;

        int instance      = port;

        String database   = db;

        String user       = userDB;

        String password   = pwdDB;

        try {

            conn = new SeConnection(server, instance, database, user, password); 

        }catch (SeException e) {

            e.printStackTrace();

            return;

        }        

        // Fetching data

        String layerName = tableName;

        SeObjectId layerID = null;

        String strSpatialColumn = "";

        try {

            Vector theLayers = conn.getLayers();

            for (int i=0; i < theLayers.size(); i++)

            {

                SeLayer layer = (SeLayer)theLayers.elementAt(i);

                if (layer.getName().compareToIgnoreCase(layerName) == 0 );

                {

                    layerID = layer.getID();

                    strSpatialColumn = layer.getSpatialColumn();

                }

            }            

            if (layerID == null)

            {

                System.err.println("Capa no encontrada");

                return;

            }

            // layerName = layer.getName();

            layer = new SeLayer( conn, layerName, strSpatialColumn );

            if (cols != null)

                this.fields = cols;

            else

            {                

                SeSqlConstruct sqlConstruct = new SeSqlConstruct( tableName );

                SeTable table1 = new SeTable(conn, tableName );

                /*

                 *   Get the table's column definition.

                 */

                SeColumnDefinition[] tableDef = table1.describe();

                this.fields = new String[tableDef.length];

                /*

                 *   Store the names of all the table's columns in the

                 *   String array cols. This array specifies the columns

                 *   to be retrieved from the database.

                 */

                int idField=1;

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

                    if (tableDef[i].getType() == SeColumnDefinition.TYPE_SHAPE)

                    {

                        this.fields[0] = tableDef[i].getName();

                    }

                    else

                    {

                        this.fields[idField] = tableDef[i].getName();

                        idField++;

                    }

                }            

            }

            sqlConstruct = new SeSqlConstruct( layerName);

    //      Create a query stream between the client and server

            query = new SeQuery( conn, fields, sqlConstruct );

            query.prepareQuery();

            query.execute();

            row = query.fetch();

            if (row == null)

            {

                System.out.println("La tabla " + getName() + " no tiene registros");

                return;

            }

            colDefs = row.getColumns();

            fieldTypes = new int[colDefs.length-1];

            for (int colNum = 0; colNum < colDefs.length; colNum++)

            {                

                SeColumnDefinition colDef = colDefs[colNum];

                int dataType = colDef.getType();

                if (dataType == SeColumnDefinition.TYPE_SHAPE)

                {

                    if (colNum != 0)

                    {

                        System.err.println("Por favor, el campo de shapes deber ser el primero");

                        return;

                    }                    

                    idSpatialColumn = colNum;

                }

                if (dataType == SeColumnDefinition.TYPE_STRING)

                    fieldTypes[colNum-1] = Types.VARCHAR;

                if (dataType == SeColumnDefinition.TYPE_FLOAT)

                    fieldTypes[colNum-1] = Types.FLOAT;

                if (dataType == SeColumnDefinition.TYPE_DOUBLE)

                    fieldTypes[colNum-1] = Types.DOUBLE;

                if (dataType == SeColumnDefinition.TYPE_INTEGER)

                    fieldTypes[colNum-1] = Types.INTEGER;

                if (dataType == SeColumnDefinition.TYPE_SMALLINT)

                    fieldTypes[colNum-1] = Types.INTEGER;

                if (dataType == SeColumnDefinition.TYPE_DATE)

                    fieldTypes[colNum-1] = Types.DATE;

            }

            // query.close();

            SeQuery extentQuery = new SeQuery( conn, fields, sqlConstruct );                       

            SeQueryInfo queryInfo = new SeQueryInfo();            

            queryInfo.setConstruct(sqlConstruct);

            // queryInfo.setQueryType(SeQueryInfo.SE_QUERYTYPE_JFA);

            // extentQuery.prepareQueryInfo(queryInfo);

            SeExtent seExtent = extentQuery.calculateLayerExtent(queryInfo);

            fullExtent = new Rectangle2D.Double(

                    seExtent.getMinX(), seExtent.getMinY(),

                    seExtent.getMaxX()- seExtent.getMinX(),

                    seExtent.getMaxY()- seExtent.getMinY());

            extentQuery.close();

            doRelateID_FID();

        }

        catch (SeException e)

        {

            e.printStackTrace();

        }

    }

    public Rectangle2D getFullExtent() {

        return fullExtent;

    }

    public int getShapeType() {

        // TODO Auto-generated method stub

        int shapeType = -1;

        switch (layer.getShapeTypes())

        {

        case SeLayer.TYPE_POINT:

            shapeType = FShape.POINT;

            break;

        case SeLayer.TYPE_LINE:

        case SeLayer.TYPE_MULTI_LINE:

        case SeLayer.TYPE_MULTI_SIMPLE_LINE:            

            shapeType = FShape.LINE;

            break;

        case SeLayer.TYPE_POLYGON:

        case SeLayer.TYPE_MULTI_POLYGON:            

            shapeType = FShape.POLYGON;

            break;

        }

        return FShape.MULTI;

    }

    public IGeometry getShape(int index) {

        SeRow row;

        try {

            row = obtainRow(index);

            SeShape spVal = row.getShape(idSpatialColumn);

            IGeometry geom = ArcSdeFeatureIterator.getGeometry(spVal);

            // queryAux.close();

            return geom;

        } catch (SeException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

        return null;        

    }

    public IFeatureIterator getFeatureIterator(String sql)

            throws DriverException {

        // TODO Auto-generated method stub

        return null;

    }

    public IFeatureIterator getFeatureIterator(Rectangle2D r, String strEPSG)

            throws DriverException {

        /*

         *   Generate a rectangular shape that will be used as a filter

         */

        try {

            SeShape shape = new SeShape(layer.getCoordRef());

            SeExtent extent = new SeExtent( r.getMinX(), 

                    r.getMinY(),r.getMaxX(),r.getMaxY());

            shape.generateRectangle(extent);

            SeShape[] shapes = new SeShape[1];

            shapes[0] = shape;

            /*

             *   Retrieve all the shapes that are contained within the

             *   rectangles envelope.

             */

            SeShapeFilter filters[] = new SeShapeFilter[1];

            SeShapeFilter filter = null;

            filter = new SeShapeFilter( layer.getName(), layer.getSpatialColumn(), shape, SeFilter.METHOD_ENVP);

            filters[0] = filter;

            SeQuery spatialQuery = null;

            SeSqlConstruct sqlCons = new SeSqlConstruct( layer.getName() );

            spatialQuery = new SeQuery(conn, fields, sqlCons);

            /*

             *   Set spatial constraints

             */

            spatialQuery.setSpatialConstraints(SeQuery.SE_OPTIMIZE, false, filters);

            spatialQuery.prepareQuery();

            spatialQuery.execute();

            return new ArcSdeFeatureIterator(spatialQuery);

        } catch (SeException e) {

            // TODO Auto-generated catch block           

            e.printStackTrace();

            return null;

        }

    }

    public void close() {

        // TODO Auto-generated method stub

    }

    public void open() throws DriverException {

        // TODO Auto-generated method stub

    }

    public int getRowIndexByFID(IFeature FID) {

        int resul;

        String strID = FID.getID();

        // System.out.println("Recuperando " + strID + " del hash");

        Integer rowIndex = (Integer) hashRelate.get(strID);

        if (rowIndex == null)

        {

            System.err.println("Error: No se ha encontrado " + strID + " dentro del Hash");

        }

        resul = rowIndex.intValue();

        return resul;

    }

    public String getGeometryField(String fieldName) {

        return layer.getSpatialColumn();

    }

    public int getShapeCount() throws IOException {

        return numReg;

    }

    public DriverAttributes getDriverAttributes() {

        // TODO Auto-generated method stub

        return null;

    }

    public String getName() {

        return "ArcSDE driver";

    }

    public String[] getFields() {

        String[] attributes = new String[fields.length-1];

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

            attributes[i-1] = fields[i];

        return attributes;

    }

    public String getWhereClause() {

        // TODO Auto-generated method stub

        return "";

    }

    public String getTableName() {

        return layer.getName();

    }

    private SeRow obtainRow(long rowIndex) throws SeException

    {

        if (rowIndex == 0)

        {  

            if (query != null)

            {

                if (query.inProgress())

                    query.close();

            }

            query = new SeQuery( conn, fields, sqlConstruct );

            query.prepareQuery();

            query.execute();

            lastRowServed = -1;

        }

        if (lastRowServed == rowIndex-1)

        {

            row = query.fetch();

            lastRowServed++;

        }

        else

        {

            if (queryAux != null) queryAux.close();

            queryAux = new SeQuery( conn, fields, sqlConstruct );

            SeObjectId rowID = new SeObjectId(rowIndex+1);

            row = queryAux.fetchRow(layer.getName(), rowID, fields);

        }

        // queryAux.close();

        return row;

    }

    public Value getFieldValue(long rowIndex, int fieldId) throws com.hardcode.gdbms.engine.data.driver.DriverException {

        SeRow row;

        Value val = null;

        try {

            int idFieldArcSDE = fieldId + 1; // SIEMPRE CONTANDO CON

                // QUE NOS HAN PASADO EL PRIMER CAMPO EL DE SHAPE

            row = obtainRow(rowIndex);

            int dataType = colDefs[idFieldArcSDE].getType();

            if ( dataType == SeColumnDefinition.TYPE_STRING)

            {

                String strAux = row.getString(idFieldArcSDE);

                if (strAux == null) strAux = "";

                    val = ValueFactory.createValue(strAux);

            }

            if (dataType == SeColumnDefinition.TYPE_FLOAT)

            {

                val = ValueFactory.createValue(row.getFloat(idFieldArcSDE).floatValue());

            }

            if (dataType == SeColumnDefinition.TYPE_DOUBLE)

            {

                val = ValueFactory.createValue(row.getDouble(idFieldArcSDE).doubleValue());

            }

            if (dataType == SeColumnDefinition.TYPE_INTEGER)

            {

                val = ValueFactory.createValue(row.getInteger(idFieldArcSDE).intValue());

            }

            if (dataType == SeColumnDefinition.TYPE_SMALLINT)

            {

                val = ValueFactory.createValue(row.getShort(idFieldArcSDE).intValue());

            }

            if (dataType == SeColumnDefinition.TYPE_DATE)

            {

                val = ValueFactory.createValue(row.getDate(idFieldArcSDE));

            }

            // queryAux.close();

            return val;

        } catch (SeException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

        return ValueFactory.createNullValue();

    }

    public int getFieldCount() throws com.hardcode.gdbms.engine.data.driver.DriverException {

        return fields.length-1;

    }

    public String getFieldName(int fieldId) throws com.hardcode.gdbms.engine.data.driver.DriverException {

        // System.out.println("En ArcSdeDriver: getFieldName(" + fieldId + ")=" + fields[fieldId+1]);

        return fields[fieldId+1];

    }

    public long getRowCount() throws com.hardcode.gdbms.engine.data.driver.DriverException {

        return numReg;

    }

    public int getFieldType(int i) throws com.hardcode.gdbms.engine.data.driver.DriverException {

        return fieldTypes[i];

    }

    /**

     * @see com.hardcode.gdbms.engine.data.driver.ObjectDriver#getPrimaryKeys()

     */

    public int[] getPrimaryKeys() throws com.hardcode.gdbms.engine.data.driver.DriverException {

        throw new UnsupportedOperationException();

    }

    /**

     * @see com.hardcode.gdbms.engine.data.driver.ObjectDriver#write(com.hardcode.gdbms.engine.data.edition.DataWare)

     */

    public void write(DataWare arg0) throws com.hardcode.gdbms.engine.data.driver.DriverException {

        // TODO Auto-generated method stub

    }

}