| gvSIG desktop 1

/*

 * Created on 04-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.mysql;

import java.awt.geom.Rectangle2D;

import java.sql.Connection;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Statement;

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

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

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

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

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

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

import com.vividsolutions.jts.io.ParseException;

/**

 * @author FJP

 *

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

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

 */

public class MySQLdriver extends DefaultDBDriver {

    private WKBParser parser = new WKBParser();

    /* private int fetch_min=-1;

    private int fetch_max=-1; */

    private Statement st;

    private Rectangle2D fullExtent = null;

    private String strAux;

    /**

     * 

     */

    public MySQLdriver() {

    }

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.fmap.drivers.VectorialDriver#getDriverAttributes()

     */

    public DriverAttributes getDriverAttributes() {

        return null;

    }

    /* (non-Javadoc)

     * @see com.hardcode.driverManager.Driver#getName()

     */

    public String getName() {

        return "mySQL JDBC Driver";

    }

	/**

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

	 */

	public IGeometry getShape(int index) {

	    IGeometry geom = null;

	    boolean resul;

	        try {

	            // EL ABSOLUTE NO HACE QUE SE VUELVAN A LEER LAS

	            // FILAS, ASI QUE MONTAMOS ESTA HISTORIA PARA QUE

	            // LO HAGA

	            // System.out.println("getShape " + index);

	            /* if (index < fetch_min)

	            {

	                rs.close();

	    	        rs = st.executeQuery(sqlOrig);

	                fetch_min = 0;

	                fetch_max = rs.getFetchSize();

	            }

	            while (index >= fetch_max)

	            {

	                rs.last();

	                // forzamos una carga

	                rs.next();

	                fetch_min = fetch_max;

	                fetch_max = fetch_max + rs.getFetchSize();

	                // System.out.println("fetchSize = " + rs.getFetchSize() + " " + fetch_min + "-" + fetch_max);

	            } 

	            rs.absolute(index+1 - fetch_min); */

	            rs.absolute(index+1);

    	        // strAux = rs.getString(1);	        

    	        // geom = parser.read(strAux);

	            byte[] data = rs.getBytes(1);	        

	            geom = parser.parse(data);

            } catch (SQLException e) {

                e.printStackTrace();

            }

	    return geom;

	}

	/**

	 * @param conn

	 * @param tableName

	 * @param fields OJO: EL PRIMER CAMPO HA DE SER EL DE GEOMETRIA

	 * @param whereClause

	 */

	public void setData(Connection conn, String tableName, String fields, String whereClause, int id_FID_field)

	{

	    this.conn = conn;	    

	    this.tableName = tableName;

	    this.fields = fields;

	    this.whereClause = whereClause;

	    this.sqlOrig = "SELECT " + fields + " FROM " + tableName + " " + whereClause;

        this.idFID_FieldName = id_FID_field;

	    try {

	        st = conn.createStatement(ResultSet.TYPE_FORWARD_ONLY, ResultSet.CONCUR_READ_ONLY);

	        rs = st.executeQuery(sqlOrig);

            metaData = rs.getMetaData();

            // Le pegamos un primera pasada para poder relacionar

            // un campo de identificador ?nico (parecido al OID en

            // postgresql) con el ?ndice dentro del recordset.

            // Esto cuando haya ediciones, no es v?lido, y hay

            // que refrescarlo.

            doRelateID_FID();

        } catch (SQLException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

	}

	/**

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

	 */

	public Rectangle2D getFullExtent(){

	    if (fullExtent == null)

	    {

    	    try

            {

    	        IFeatureIterator itGeom = getFeatureIterator("SELECT ASBINARY(ogc_geom) AS the_geom FROM " + tableName);

    	        IGeometry geom;

    	        int cont = 0;

    	        while (itGeom.hasNext())

    	        {

    	            geom = itGeom.next().getGeometry();

    	            if (cont==0)

    	                fullExtent = geom.getBounds2D();

    	            else

    	                fullExtent.add(geom.getBounds2D());

    	            cont++;

    	        }

            }

    	    catch (SQLException e)

    	    {

    	        System.err.println(e.getMessage());

    	    } catch (DriverException e) {

                // TODO Auto-generated catch block

                e.printStackTrace();

            }

	    }

	    return fullExtent;

	}

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.fmap.drivers.VectorialDatabaseDriver#getGeometryIterator(java.lang.String)

     */

    public IFeatureIterator getFeatureIterator(String sql) throws com.iver.cit.gvsig.fmap.DriverException {

        Statement st;

        MySqlFeatureIterator geomIterator = null;

        try {

            st = conn.createStatement();

            // st.setFetchSize(2000);

            ResultSet rs = st.executeQuery(sql);

            geomIterator = new MySqlFeatureIterator(rs);

        } catch (SQLException e) {

            e.printStackTrace();

            throw new com.iver.cit.gvsig.fmap.DriverException(e);

        }

        return geomIterator;

    }

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.fmap.drivers.VectorialDatabaseDriver#getGeometryIterator(java.awt.geom.Rectangle2D)

     */

    public IFeatureIterator getFeatureIterator(Rectangle2D r, String strEPSG) throws DriverException {

        double xMin = r.getMinX();

        double yMin = r.getMinY();

        double xMax = r.getMaxX();

        double yMax = r.getMaxY();

        String wktBox = "GeomFromText('LINESTRING(" + xMin + " " + yMin + ", "

		+ xMax + " " + yMin + ", "

		+ xMax + " " + yMax + ", "

		+ xMin + " " + yMax + ")', "

		+ strEPSG + ")";

        String sqlAux;

        if (getWhereClause().startsWith("WHERE")) 

            sqlAux = sqlOrig + " MBRIntersects(" + wktBox + ",ogc_geom);" ;

        else

            sqlAux = sqlOrig + "WHERE MBRIntersects(" + wktBox + ",ogc_geom);" ;

        return getFeatureIterator(sqlAux);

    }

    public void open() {

        // TODO Auto-generated method stub

    }

}