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.operations.strategies;

import java.awt.Graphics2D;

import java.awt.geom.AffineTransform;

import java.awt.geom.Point2D;

import java.awt.geom.Rectangle2D;

import java.awt.image.BufferedImage;

import java.util.BitSet;

import javax.print.attribute.PrintRequestAttributeSet;

import org.apache.log4j.Logger;

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

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

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

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

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

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

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

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.rendering.VectorialLegend;

import com.iver.utiles.swing.threads.Cancellable;

import com.iver.utiles.swing.threads.CancellableMonitorable;

/**

 * Implementa la Strategy por defecto. Los m?todos que tendr?n en com?n la

 * mayor parte de las estrategias

 */

public class DefaultStrategy implements Strategy {

    public static final int EQUALS = 0;

    public static final int DISJOINT = 1;

    public static final int INTERSECTS = 2;

    public static final int TOUCHES = 3;

    public static final int CROSSES = 4;

    public static final int WITHIN = 5;

    public static final int CONTAINS = 6;

    public static final int OVERLAPS = 7;

        private static Logger logger = Logger.getLogger(DefaultStrategy.class.getName());

        FLayer capa = null;

        /**

         * Crea un nuevo DefaultStrategy.

         *

         * @param capa DOCUMENT ME!

         */

        public DefaultStrategy(FLayer capa) {

                this.capa = capa;

                SingleLayer foo = (SingleLayer) capa;

                ClassifiableVectorial vectorial = (ClassifiableVectorial) capa;

        }

        /**

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

         */

        public FBitSet queryByRect(Rectangle2D rect, CancellableMonitorable cancel) throws DriverException {

                QueryByRectVisitor visitor = new QueryByRectVisitor();

                visitor.setRect(rect);

                try {

                        process(visitor, cancel);

                } catch (VisitException e) {

                        throw new RuntimeException(

                                "QueryByRectVisitor lanza una VisitException?");

                }

                return visitor.getBitSet();

        }

        /**

         * @see com.iver.cit.gvsig.fmap.operations.strategies.Strategy#queryByShape(com.iver.cit.gvsig.fmap.fshape.IGeometry,

         *                 int)

         */

        public FBitSet queryByShape(IGeometry g, int relationship, CancellableMonitorable cancel)

                throws DriverException, VisitException {

                QueryByGeometryVisitor visitor = new QueryByGeometryVisitor(g, relationship);

                process(visitor);

                return visitor.getBitSet();

        }

        /**

         * @see com.iver.cit.gvsig.fmap.operations.strategies.Strategy#getSelectionBounds()

         */

        public Rectangle2D getSelectionBounds() {

                return null;

        }

        /**

         * @see com.iver.cit.gvsig.fmap.operations.strategies.Strategy#createIndex()

         */

        public void createIndex() {

        }

        /**

         * @see com.iver.cit.gvsig.fmap.operations.LayerOperations#draw(java.awt.image.BufferedImage,

         *                 java.awt.Graphics2D, ISymbol)

         */

        public void draw(BufferedImage image, Graphics2D g, ViewPort viewPort,

                Cancellable cancel) throws DriverException {

                try {

                        ReadableVectorial adapter = ((SingleLayer) capa).getSource();

                        ICoordTrans ct = getCapa().getCoordTrans();

//                        logger.info("adapter.start()");

                        adapter.start();

//                        logger.info("getCapa().getRecordset().start()");

                        SelectableDataSource rsSel = ((AlphanumericData) getCapa()).getRecordset();

                        if (rsSel != null)

                            rsSel.start();

                        int sc;

                        Rectangle2D extent = viewPort.getAdjustedExtent();

                        AffineTransform at = viewPort.getAffineTransform();

                        sc = adapter.getShapeCount();

                        // TODO: A revisar si es o no conveniente este sistema

                        // de comunicaci?n con los drivers.

                        DriverAttributes attr = adapter.getDriverAttributes();

                        boolean bMustClone = false;

                        if (attr != null)

                        {

                            if (attr.isLoadedInMemory())

                            {

                                bMustClone = attr.isLoadedInMemory();

                            }

                        }

                        VectorialLegend l = (VectorialLegend) ((ClassifiableVectorial) capa).getLegend();

            FBitSet bitSet = rsSel.getSelection();

                        for (int i = 0; i < sc; i++) {

                                if (cancel.isCanceled()) {

                                        break;

                                }

                                IGeometry geom = adapter.getShape(i);

                                if (geom == null) {

                                        continue;

                                }

                                if (ct != null) {

                                    if (bMustClone)

                                        geom = geom.cloneGeometry();

                                        geom.reProject(ct);

                                }

                                // if (geom.intersects(extent)) {

                                if (geom.fastIntersects(extent.getMinX(), extent.getMinY(),

                                         extent.getWidth(), extent.getHeight())) {

                                        ISymbol symbol = l.getSymbol(i);

                    if (symbol ==null)

                        continue;

                    if (bitSet != null)

                        if (bitSet.get(i)) {

                                                    symbol = symbol.getSymbolForSelection();

                                            }

                    if (symbol != null)

                            geom.draw(g, viewPort, symbol);

                                }

                                /* else

                                {

                                    System.out.println("no pinto id=" + i);

                                } */

                        }

//                        logger.info("getCapa().getRecordset().stop()");

                        if (rsSel != null)

                            rsSel.stop();

//                        logger.debug("adapter.stop()");

                        adapter.stop();

                        // TODO: A revisar si es o no conveniente este sistema

                        // de comunicaci?n con los drivers.

                        // DriverAttributes attr = adapter.getDriverAttributes();

                        /* if (attr != null)

                        {

                            if (attr.isLoadedInMemory())

                            {

                                // Quitamos lo de la reproyecci?n al vuelo para que

                                // solo se haga una vez.

                                getCapa().setCoordTrans(null);

                            }

                        } */

                } catch (DriverIOException e) {

                        throw new DriverException(e);

                } catch (com.hardcode.gdbms.engine.data.driver.DriverException e) {

                    throw new DriverException(e);

                } catch (DriverException e) {

                    throw new DriverException(e);

        }

        }

        /**

         * Devuelve la capa.

         *

         * @return Returns the capa.

         */

        public FLayer getCapa() {

                return capa;

        }

        /**

         * @see com.iver.cit.gvsig.fmap.operations.strategies.Strategy#process(com.iver.cit.gvsig.fmap.operations.strategies.FeatureVisitor,

         *                 java.util.BitSet)

         */

        public void process(FeatureVisitor visitor, BitSet subset)

                throws DriverException, VisitException {

                process(visitor, subset, null);

        }

        /**

         * DOCUMENT ME!

         *

         * @param visitor DOCUMENT ME!

         *

         * @throws DriverException DOCUMENT ME!

         * @throws VisitException

         *

         * @see com.iver.cit.gvsig.fmap.operations.strategies.Strategy#process(com.iver.cit.gvsig.fmap.operations.strategies.FeatureVisitor)

         */

        public void process(FeatureVisitor visitor)

                throws DriverException, VisitException {

                process(visitor, (CancellableMonitorable)null);

        }

        /**

         * @see com.iver.cit.gvsig.fmap.operations.strategies.Strategy#queryByPoint(Point2D,

         *                 double)

         */

        public FBitSet queryByPoint(Point2D p, double tolerance, CancellableMonitorable cancel)

                throws DriverException {

        // TODO: OJO!!!!. Est? implementado como un rectangulo.

        // Lo correcto deber?a ser calculando las distancias reales

        // es decir, con un c?rculo.

        Rectangle2D recPoint = new Rectangle2D.Double(p.getX() - (tolerance / 2),

                p.getY() - (tolerance / 2), tolerance, tolerance);

                return queryByRect(recPoint);

        }

        /* (non-Javadoc)

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

         */

        public void print(Graphics2D g, ViewPort viewPort, Cancellable cancel, PrintRequestAttributeSet properties)

                throws DriverException {

                if (capa instanceof FLyrVect) {

                        ((FLyrVect)capa).beforePrinting(properties);

                        draw(null, g, viewPort, cancel); // Quiero ejecutar el draw del padre, que es el que va sin acelaraci?n!!

                        ((FLyrVect)capa).afterPrinting();

                }else{

                        draw(null, g, viewPort, cancel); // Quiero ejecutar el draw del padre, que es el que va sin acelaraci?n!!

                }

        }

        public void process(FeatureVisitor visitor, Rectangle2D rectangle) throws DriverException, VisitException {

                process(visitor, rectangle, null);

        }

        /**

         * Verifies cancelation events, and return a boolean flag

         * if processes must be stopped for this cancelations events.

         *

         * @param cancel

         * @param va

         * @param visitor

         * @return

         * @throws DriverIOException

         */

        protected boolean verifyCancelation(Cancellable cancel, ReadableVectorial va, FeatureVisitor visitor) throws DriverIOException{

                if(cancel != null){

                        if(cancel.isCanceled()){

                                va.stop();

                                visitor.stop(capa);

//                                logger.info("visitor canceled");

                                return true;

                        }

                }

                return false;

        }

        public void process(FeatureVisitor visitor, BitSet subset, CancellableMonitorable cancel) throws DriverException, VisitException {

                try {

//                        logger.info("visitor.start()");

                        if (visitor.start(capa)) {

                                ReadableVectorial va = ((SingleLayer) capa).getSource();

                                va.start();

                                for (int i = 0; i < va.getShapeCount(); i++) {

                                        if(verifyCancelation(cancel, va, visitor))

                                                return;

                                        if (subset.get(i)) {

                                                if(cancel != null){

                                                        cancel.reportStep();

                                                }

                                                visitor.visit(va.getShape(i), i);

                                        }

                                }

                                va.stop();

//                                logger.info("visitor.stop()");

                                visitor.stop(capa);

                        }

                } catch (DriverIOException e) {

                        throw new DriverException(e);

                }

        }

        public void process(FeatureVisitor visitor, CancellableMonitorable cancel) throws DriverException, VisitException {

                try {

//                        logger.info("visitor.start()");

                        if (visitor.start(capa)) {

                                ReadableVectorial va = ((SingleLayer) capa).getSource();

                                ICoordTrans ct = getCapa().getCoordTrans();

                                va.start();

                                for (int i = 0; i < va.getShapeCount(); i++) {

                                        if(cancel != null){

                                                cancel.reportStep();

                                        }

                                        if(verifyCancelation(cancel, va, visitor))

                                                return;

                                    IGeometry geom = va.getShape(i);

                                    if (ct != null) {

                                                geom.reProject(ct);

                                        }

                                        visitor.visit(geom, i);

                                }

                                va.stop();

//                                logger.info("visitor.stop()");

                                visitor.stop(capa);

                        }

                } catch (DriverIOException e) {

                        throw new DriverException(e);

                }

        }

        public void process(FeatureVisitor visitor, Rectangle2D rectangle, CancellableMonitorable cancel) throws DriverException, VisitException {

                FilterRectVisitor filterVisitor = new FilterRectVisitor();

                filterVisitor.setRectangle(rectangle);

                filterVisitor.setWrappedVisitor(visitor);

                try {

                        process(filterVisitor, cancel);

                } catch (VisitException e) {

                        throw new RuntimeException(

                                "process(visitor,rectangle) lanza una VisitException?");

                }

        }

        public FBitSet queryByPoint(Point2D p, double tolerance) throws DriverException {

                return queryByPoint(p, tolerance, null);

        }

        public FBitSet queryByRect(Rectangle2D rect) throws DriverException {

                return queryByRect(rect, null);

        }

        public FBitSet queryByShape(IGeometry g, int relationship) throws DriverException, VisitException {

                return queryByShape(g, relationship, null);

        }

        /**

         * Tells when in a spatial query the use of an spatial index is an advanced,

         * or it would be better to use a secuential search.

         * <br>

         * The criteria to decide is the area of the query region. If it is less than

         * 1/4 of full extent layer region, the spatial index will be an improve.

         * Else, a secuential scan would be better

         * @param rectangle

         * @return

         * @throws DriverException

         */

        protected boolean isSpatialIndexNecessary(Rectangle2D extent) throws DriverException {

                FLyrVect lyr = (FLyrVect) getCapa();

                double areaExtent = extent.getWidth() * extent.getHeight();

                double areaFullExtent = lyr.getFullExtent().getWidth() *

                                                        lyr.getFullExtent().getHeight();

                return areaExtent < (areaFullExtent / 4.0);

        }

        /**

         * Similar to process, but calls visitWithSymbol in FeatureVisitor.

         * @param visitor

         * @param rectangle

         * @param cancel

         * @throws DriverException

         * @throws VisitException

         */

        public void processWithSymbols(ExtendsFeatureVisitor visitor, Rectangle2D extent, CancellableMonitorable cancel) throws DriverException, VisitException {

                if (visitor==null)

                        return;

                /*IGeometry geom = null;

                int index = -1;

                ISymbol symbol = null;

                visitor.visitWithSymbol(geom, index, symbol); // TEST */

                try {

                        ReadableVectorial adapter = ((SingleLayer) capa).getSource();

                        ICoordTrans ct = getCapa().getCoordTrans();

                        //logger.debug("adapter.start()");

                        adapter.start();

                        //logger.debug("getCapa().getRecordset().start()");

                        SelectableDataSource rsSel = ((AlphanumericData) getCapa()).getRecordset();

                        if (rsSel != null)

                            rsSel.start();

                        int sc = adapter.getShapeCount();

                        // TODO: A revisar si es o no conveniente este sistema

                        // de comunicaci?n con los drivers.

                        DriverAttributes attr = adapter.getDriverAttributes();

                        boolean bMustClone = false;

                        if (attr != null)

                        {

                            if (attr.isLoadedInMemory())

                            {

                                bMustClone = attr.isLoadedInMemory();

                            }

                        }

                        VectorialLegend l = (VectorialLegend) ((ClassifiableVectorial) capa).getLegend();

            FBitSet bitSet = null;

            if (getCapa() instanceof Selectable){

                Selectable selection = (Selectable) getCapa();

                bitSet = selection.getSelection();

            }

                        for (int i = 0; i < sc; i++) {

                                if (cancel!=null && cancel.isCanceled()) {

                                        break;

                                }

                                IGeometry geom = adapter.getShape(i);

                                if (geom == null) {

                                        continue;

                                }

                                if (ct != null) {

                                    if (bMustClone)

                                        geom = geom.cloneGeometry();

                                        geom.reProject(ct);

                                }

                                if (extent == null ||

                                                geom.fastIntersects(extent.getMinX(), extent.getMinY(),

                                             extent.getWidth(), extent.getHeight())) {

                                        ISymbol symbol = l.getSymbol(i);

                    if (symbol ==null)

                        continue;

                    if (bitSet != null)

                        if (bitSet.get(i)) {

                                                    symbol = symbol.getSymbolForSelection();

                                            }

                                        // send feature for process in visitor

                                        visitor.visitWithSymbol(geom, i, symbol);

                                }

                                /* else

                                {

                                    System.out.println("no pinto id=" + i);

                                } */

                        }

                        //logger.debug("getCapa().getRecordset().stop()");

                        if (rsSel != null)

                            rsSel.stop();

                        //logger.debug("adapter.stop()");

                        adapter.stop();

                } catch (DriverIOException e) {

                        throw new DriverException(e);

                } catch (com.hardcode.gdbms.engine.data.driver.DriverException e) {

                    throw new DriverException(e);

                } catch (DriverException e) {

                    throw new DriverException(e);

        }

        }

}