Application: gvSIG desktop

/*

 * Created on 08-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.operations.strategies;

import java.awt.Graphics2D;

import java.awt.geom.Rectangle2D;

import java.awt.image.BufferedImage;

import org.cresques.cts.ICoordTrans;

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

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

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

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

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

import com.iver.cit.gvsig.fmap.core.v02.FGraphicUtilities;

import com.iver.cit.gvsig.fmap.core.v02.FSymbol;

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

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

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

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

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

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.layerOperations.ClassifiableVectorial;

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

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

import com.iver.cit.gvsig.fmap.operations.Cancellable;

import com.iver.cit.gvsig.fmap.rendering.VectorialLegend;

/**

 * @author FJP

 *

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

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

 */

public class DBStrategy extends DefaultStrategy {

        public DBStrategy(FLayer capa) {

                super(capa);

        }

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.fmap.operations.strategies.Strategy#draw(java.awt.image.BufferedImage, java.awt.Graphics2D, com.iver.cit.gvsig.fmap.ViewPort, com.iver.cit.gvsig.fmap.operations.Cancellable)

     */

    public void draw(BufferedImage image, Graphics2D g, ViewPort viewPort, Cancellable cancel) throws DriverException {

        // Nos aprovechamos del SQL para lanzar la consulta

        // teniendo en cuenta el boundingbox que toca.

        FLyrVect lyr = (FLyrVect) getCapa();

        ISpatialDB dbAdapter = (ISpatialDB) ((SingleLayer) capa).getSource();

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

        try {

                        dbAdapter.start();

                Selectable selectable=lyr.getRecordset();

                        ICoordTrans ct = lyr.getCoordTrans();

                        FBitSet bitSet = selectable.getSelection();

                String strEPSG = viewPort.getProjection().getAbrev().substring(5);

                // TODO: EXPLORAR LAS LEYENDAS PARA SABER LOS CAMPOS QUE VAMOS

                // A NECESITAR RECUPERAR, PARA INCLUIR EN LA CONSULTA SOLO

                // AQUELLOS QUE VAMOS A NECESITAR. CON ESO GANAREMOS VELOCIDAD.

                // Ejemplo:

                // En ArcSDE:

                // con Vias cogiendo un campo solo: 5 segundos

                // con todos los campos de Vias: 11 segundos.

                // => MODIFICAR EL getFeatureIterator para que admita los nombres

                // de los campos adem?s del rect?ngulo que pides.

                String[] usedFields = null;

                VectorialLegend l = null;

                if (lyr.getLegend() instanceof VectorialLegend)

                {

                    l = (VectorialLegend) lyr.getLegend();

                }

                Rectangle2D rectAux = viewPort.getAdjustedExtent();

                if (ct != null) {

                    ICoordTrans invertedCT = ct.getInverted();

                    rectAux = invertedCT.convert(rectAux);            

                }

                usedFields = l.getUsedFields();

                IFeatureIterator geomIt = dbAdapter.getFeatureIterator(rectAux, strEPSG, usedFields);

                if (geomIt == null)

                {

                    // dbAdapter.stop();

                    return;

                }

                DriverAttributes attr = dbAdapter.getDriverAttributes();

                boolean bMustClone = false;

                if (attr != null)

                {

                    if (attr.isLoadedInMemory())

                    {

                        bMustClone = attr.isLoadedInMemory();               

                    }

                }

                int i;

                FSymbol symbol;

                while (geomIt.hasNext())

                {

                        if (cancel.isCanceled()) {

                            geomIt.closeIterator();

                                break;

                        }

                        IFeature feat = geomIt.next();

                        if (feat == null)

                        {

                                continue;

                        }

                    IGeometry geom = feat.getGeometry();

                    System.out.println("PINTANDO FEATURE " + feat.getID());

                    if (ct != null) {

                        if (bMustClone)

                            geom = geom.cloneGeometry();

                        geom.reProject(ct);

                    }

                    i = dbAdapter.getRowIndexByFID(feat);

                    // System.out.println("Antes de pintar " + i);

                    // Value val = feat.getAttribute(0);                

                    symbol = l.getSymbolByFeature(feat);

                    // symbol = l.getSymbol(i);

                    if (symbol == null) continue;

                    if (bitSet.get(i)) {

                        symbol = FSymbol.getSymbolForSelection(symbol);

                    } 

                    // symbol = l.getDefaultSymbol();

                    geom.draw(g,viewPort, symbol);

                    if (lyr.isEditing())

                    {

                            if (bitSet.get(i))

                            {

                                    Handler[] handlers = geom.getHandlers(IGeometry.SELECTHANDLER);

                                    FGraphicUtilities.DrawHandlers(g, viewPort.getAffineTransform(), handlers);

                            }                    

                    }

                }

                dbAdapter.stop();

                } catch (DriverIOException e) {

                        throw new DriverException(e);

                }

    }

        /* (non-Javadoc)

         * @see com.iver.cit.gvsig.fmap.operations.strategies.DefaultStrategy#queryByRect(java.awt.geom.Rectangle2D)

         */

        public FBitSet queryByRect(Rectangle2D rect) throws DriverException {

        // Nos aprovechamos del SQL para lanzar la consulta

        // teniendo en cuenta el boundingbox que toca.

        FLyrVect lyr = (FLyrVect) getCapa();

        ISpatialDB dbAdapter = (ISpatialDB) ((SingleLayer) capa).getSource();

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

        try {

                        dbAdapter.start();

                        ICoordTrans ct = lyr.getCoordTrans();

                        String strEPSG = lyr.getProjection().getAbrev().substring(5);                

                Rectangle2D rectAux = rect;

                if (ct != null) {

                    ICoordTrans invertedCT = ct.getInverted();

                    rectAux = invertedCT.convert(rectAux);            

                }

                FBitSet selection = new FBitSet();

                IFeatureIterator geomIt = dbAdapter.getFeatureIterator(rectAux, strEPSG, null);

                if (geomIt == null)

                {

                    return selection;

                }

                DriverAttributes attr = dbAdapter.getDriverAttributes();

                boolean bMustClone = false;

                if (attr != null)

                {

                    if (attr.isLoadedInMemory())

                    {

                        bMustClone = attr.isLoadedInMemory();               

                    }

                }

                int i;

                while (geomIt.hasNext())

                {

                        IFeature feat = geomIt.next();

                    IGeometry geom = feat.getGeometry();

                    if (ct != null) {

                        if (bMustClone)

                            geom = geom.cloneGeometry();

                        geom.reProject(ct);

                    }

                    i = dbDriver.getRowIndexByFID(feat);

                    if (geom.intersects(rect))

                            selection.set(i, true);

                }

                dbAdapter.stop();

                return selection;

                } catch (DriverIOException e) {

                        throw new DriverException(e);

                }

        }

}