Application: gvSIG desktop

/* 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;

import java.awt.geom.Rectangle2D;

import java.io.IOException;

import java.sql.Connection;

import java.sql.ResultSet;

import java.sql.ResultSetMetaData;

import java.sql.SQLException;

import java.sql.Statement;

import java.sql.Types;

import java.util.Date;

import org.postgis.Geometry;

import org.postgis.PGbox3d;

import org.postgis.Point;

import com.hardcode.gdbms.engine.data.DataSource;

import com.hardcode.gdbms.engine.data.ReadDriver;

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

import com.iver.cit.gvsig.fmap.drivers.jdbc.postgis.WKTParser;

import com.vividsolutions.jts.io.ParseException;

/**

 * Clase abstracta qu

 */

public abstract class DefaultDBDriver implements VectorialDatabaseDriver, ReadDriver {    

    /**

     * 

     */

    protected Connection conn;

    protected String tableName;

    protected String whereClause;

    protected String fields;

    protected String sqlOrig;

    protected ResultSet rs;

    protected boolean bCursorActivo = false;

    protected Statement st;

    protected int numReg=-1;

    protected ResultSetMetaData metaData = null;

        /**

         * @return devuelve la Conexi?n a la base de datos, para que 

         * el usuario pueda hacer la consulta que quiera, si lo desea.

         * Por ejemplo, esto puede ser ?til para abrir un cuadro de dialogo

         * avanazado y lanzar peticiones del tipo "Devuelveme un buffer

         * a las autopistas", y con el resultset que te venga, escribir

         * un shape, o cosas as?.

         */

        public Connection getConnection()

        {

            return conn;

        }

        public String getFields()

        {

            return fields;            

        }

        public String getWhereClause()

        {

            return whereClause;

        }

        public String getTableName()

        {

            return tableName;

        }

        /**

         * @throws DriverIOException

         * @throws DriverException

         * @see com.iver.cit.gvsig.fmap.layers.ReadableVectorial#getShapeCount()

         */

        public int getShapeCount() throws IOException {

                    if (numReg == -1)

                    {

                        try

                    {

                            Statement s = conn.createStatement();                    

                            ResultSet r = s.executeQuery("SELECT COUNT(*) AS NUMREG FROM " + tableName + " " + whereClause);

                            r.next();

                            numReg = r.getInt(1);

                            System.err.println("numReg = " + numReg);

                    }

                        catch (SQLException e)

                        {

                            throw new IOException(e.getMessage());

                        }

                    }

            return numReg;

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.ReadableVectorial#getShapeType()

         */

        public int getShapeType() {

                return FShape.MULTI;

        }

        public Class getFieldType(int idField) throws com.hardcode.gdbms.engine.data.DriverException

        {

            String str = "";

            try {

                int i = idField + 2; // idField viene basado en 1, y

                                        // adem?s nos saltamos el campo de geometry

                str = metaData.getColumnClassName(i);

            if (metaData.getColumnType(i) == Types.VARCHAR)

                return Value.VARCHAR;

                    if (metaData.getColumnType(i) == Types.FLOAT)

                        return Value.FLOAT;

                    if (metaData.getColumnType(i) == Types.DOUBLE)

                        return Value.DOUBLE;

                    if (metaData.getColumnType(i) == Types.INTEGER)

                        return Value.INTEGER;

                    if (metaData.getColumnType(i) == Types.BIGINT)

                        return Value.BIGINT;

                    if (metaData.getColumnType(i) == Types.BIT)

                        return Value.BIT;

                    if (metaData.getColumnType(i) == Types.DATE)

                        return Value.DATE;

            } catch (SQLException e) {

            throw new com.hardcode.gdbms.engine.data.DriverException("Tipo no soportado: " + str);

            }

            return null;

        }

    /**

     * Obtiene el valor que se encuentra en la fila y columna indicada

     * Esta es la implementaci?n por defecto. Si lo del absolute

     * no va bien, como es el caso del PostGis, el driver lo

     * tiene que reimplementar

     *

     * @param rowIndex fila

     * @param fieldId columna

     *

     * @return subclase de Value con el valor del origen de datos

     *

     * @throws DriverException Si se produce un error accediendo al DataSource

     */

    public Value getFieldValue(long rowIndex, int idField)

        throws com.hardcode.gdbms.engine.data.DriverException

        {

                int i = (int) (rowIndex + 1);

                int fieldId = idField+2;

                try {

                    rs.absolute(i);

                if (metaData.getColumnType(fieldId) == Types.VARCHAR)

                    return ValueFactory.createValue(rs.getString(fieldId));

                        if (metaData.getColumnType(fieldId) == Types.FLOAT)

                            return ValueFactory.createValue(rs.getFloat(fieldId));

                        if (metaData.getColumnType(fieldId) == Types.DOUBLE)

                            return ValueFactory.createValue(rs.getDouble(fieldId));

                        if (metaData.getColumnType(fieldId) == Types.INTEGER)

                            return ValueFactory.createValue(rs.getInt(fieldId));

                        if (metaData.getColumnType(fieldId) == Types.BIGINT)

                            return ValueFactory.createValue(rs.getLong(fieldId));

                        if (metaData.getColumnType(fieldId) == Types.BIT)

                            return ValueFactory.createValue(rs.getBoolean(fieldId));

                        if (metaData.getColumnType(fieldId) == Types.DATE)

                            return ValueFactory.createValue(rs.getDate(fieldId));

                } catch (SQLException e) {

                throw new com.hardcode.gdbms.engine.data.DriverException("Tipo no soportado: columna " + fieldId );

                }

                return null;

        }

    /**

     * Obtiene el n?mero de campos del DataSource

     *

     * @return

     *

     * @throws DriverException Si se produce alg?n error accediendo al

     *         DataSource

     */

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

    {

        try {

            // Suponemos que el primer campo es el de las geometries, y no lo

            // contamos

            return rs.getMetaData().getColumnCount()-1;

        } catch (SQLException e) {

            throw new com.hardcode.gdbms.engine.data.DriverException(e);

        }

    }

    /**

     * Devuelve el nombre del campo fieldId-?simo

     *

     * @param fieldId ?ndice del campo cuyo nombre se quiere obtener

     *

     * @return

     * @throws com.hardcode.gdbms.engine.data.DriverException

     *

     * @throws DriverException Si se produce alg?n error accediendo al

     *         DataSource

     */

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

    {

        try {

            return rs.getMetaData().getColumnName(fieldId+2);

        } catch (SQLException e) {

            throw new com.hardcode.gdbms.engine.data.DriverException(e);

        }

    }

    /**

     * Obtiene el n?mero de registros del DataSource

     *

     * @return

     *

     * @throws DriverException Si se produce alg?n error accediendo al

     *         DataSource

     */

    public long getRowCount()

    {

        try {

            return getShapeCount();

        } catch (IOException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

        return -1;

    }

    public void close()

    {

        try {

            if (bCursorActivo)

            {

                st.execute("CLOSE wkb_cursor");

                bCursorActivo = false;

            }

        } catch (SQLException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

    }

}