Application: gvSIG desktop

package com.iver.cit.gvsig.fmap.operations.strategies;

import java.awt.Color;

import java.awt.FontMetrics;

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.io.IOException;

import java.util.BitSet;

import java.util.List;

import org.apache.log4j.Logger;

import org.cresques.cts.ICoordTrans;

import org.geotools.resources.geometry.XRectangle2D;

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

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

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

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

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

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

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

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

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

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

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

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

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

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

import com.iver.cit.gvsig.fmap.edition.EditableAdapter;

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

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

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

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

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

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

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

import com.iver.cit.gvsig.fmap.rendering.styling.FStyle2D;

import com.iver.cit.gvsig.fmap.spatialindex.ISpatialIndex;

import com.vividsolutions.jts.geom.Coordinate;

import com.vividsolutions.jts.geom.Envelope;

import com.vividsolutions.jts.geom.Geometry;

import com.vividsolutions.jts.geom.IntersectionMatrix;

/**

 * Esta clase se encargar? de dibujar de la forma m?s eficiente los temas de

 * anotaciones.

 *

 * @author Vicente Caballero Navarro

 */

public class AnnotationStrategy extends DefaultStrategy {

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

        private FSymbol symbolPoint=new FSymbol(FShape.POINT,Color.black);

        private Graphics2D graphics=null;

        private ViewPort viewPort=null;

        /**

     * Crea un nuevo AnotationStrategy.

     *

     * @param layer DOCUMENT ME!

     */

    public AnnotationStrategy(FLayer layer) {

        super(layer);

        capa = (FLyrAnnotation) layer;

        symbolPoint.setSize(5);

    }

    /**

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

         *                 java.awt.Graphics2D, FStyle2D)

         */

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

                Cancellable cancel) throws DriverException {

                Rectangle2D elExtent = viewPort.getAdjustedExtent();

                graphics=g;

                FLyrAnnotation lyrAnnotation=(FLyrAnnotation)capa;

                ISpatialIndex isi=lyrAnnotation.getISpatialIndex();

                List list=null;

                VectorialLegend l=(VectorialLegend)lyrAnnotation.getLegend();

                FBitSet bitSet=lyrAnnotation.getRecordset().getSelection();

                boolean inPixels=lyrAnnotation.isInPixels();

                FSymbol theSymbol = l.getDefaultSymbol();

                theSymbol.setFontSizeInPixels(inPixels);

                System.out.println("Dibujando Anotaciones...");

                this.viewPort=viewPort;//capa.getFMap().getViewPort();

                AffineTransform at=viewPort.getAffineTransform();

                try {

                        int sc;

                        if (isi==null){

                                sc=lyrAnnotation.getSource().getShapeCount();

                        }else{

                                list=isi.query(elExtent);

                                sc = list.size();

                        }

                        FontMetrics metrics = g.getFontMetrics();

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

                                if (cancel.isCanceled()) {

                                        break;

                                }

                                int numOriginal;

                                if (isi!=null){

                                        Integer idRec = (Integer) list.get(numReg);

                                        numOriginal= idRec.intValue();

                                }else{

                                        numOriginal=numReg;

                                }

                                if (lyrAnnotation.getSource() instanceof EditableAdapter)

                                        numOriginal=((EditableAdapter)lyrAnnotation.getSource()).getCalculatedIndex(numOriginal);

                                FLabel theLabel = (FLabel) ((FLyrAnnotation) capa).getLabel(numOriginal);

                                if ((theLabel == null) || (theLabel.getOrig() == null))

                                        continue;

                                Rectangle2D r=getBoundBox(theLabel.getOrig(),(float)theLabel.getHeight(), theLabel.getJustification(),theLabel.getString());

                                theLabel.setBoundBox(r);

                                FGraphicUtilities.DrawShape(g,at,new FPoint2D(viewPort.fromMapPoint(new Point2D.Double(r.getX(),r.getY()))),symbolPoint);

                                if (bitSet.get(numOriginal)) {

                                        FGraphicUtilities.DrawAnnotation(g, at, theSymbol, theLabel,metrics,true);

                                }else{

                                        FGraphicUtilities.DrawAnnotation(g, at, theSymbol, theLabel,metrics,false);

                                }

                        }

                } catch (DriverIOException e) {

                        e.printStackTrace();

                }

        }

         /**

     * Construcci?n del rect?ngulo

     *

     * @param p

     * @param g DOCUMENT ME!

     * @param justification DOCUMENT ME!

     * @param vp DOCUMENT ME!

     *

     * @return

     */

    public Rectangle2D getBoundBox(Point2D p, float hp,

        int justification,String s) {

        Rectangle2D bounding=null;

        if (((FLyrAnnotation)capa).isInPixels()){

                graphics.setFont(graphics.getFont().deriveFont(hp));

        }else{

                float alturaPixels = (float) ((hp * viewPort.getAffineTransform().getScaleX())*FLabel.SQUARE);

                graphics.setFont(graphics.getFont().deriveFont(alturaPixels));

        }

        FontMetrics metrics = graphics.getFontMetrics();

        int w = metrics.stringWidth(s);

        double width = viewPort.toMapDistance(w);

        int h = metrics.getMaxAscent();

        double height = viewPort.toMapDistance(h);

        //double dist = viewPort.toMapDistance(3);

        return new Rectangle2D.Double(p.getX(), p.getY(), width, height);

       /* switch (justification) {

            case FLabel.LEFT_BOTTOM:

                bounding=justification(p, width,height, 0, 0);

                break;

            case FLabel.LEFT_CENTER:

                     bounding=justification(p, width,height, 0, -(height / 2));

                break;

            case FLabel.LEFT_TOP:

                     bounding=justification(p,width,height, 0, -height);

                break;

            case FLabel.CENTER_BOTTOM:

                     bounding=justification(p, width,height, -(width / 2), -dist);

                break;

            case FLabel.CENTER_CENTER:

                     bounding=justification(p, width,height, -(width / 2), -(height / 2));

                break;

            case FLabel.CENTER_TOP:

                     bounding=justification(p, width,height, -(width / 2), -height);

                break;

            case FLabel.RIGHT_BOTTOM:

                     bounding=justification(p, width,height, -width, -dist);

                break;

            case FLabel.RIGHT_CENTER:

                     bounding=justification(p, width,height, -width, -(height / 2));

                break;

            case FLabel.RIGHT_TOP:

                     bounding=justification(p, width,height, -width, -height);

                break;

        }

        return bounding;

        */

    }

  /*  private Rectangle2D justification(Point2D p, double w,double h, double x, double y) {

        Rectangle2D r=new Rectangle2D.Double(p.getX() + x, p.getY() - y, w, h);

        return r;

    }

    */

        /**

         * M?todo utilizado para dibujar sobre el graphics que se pasa como

         * par?metro, pensado para utilizarse para imprimir.

         *

         * @param g

         *            Graphics2D

         * @param viewPort

         *            ViewPort.

         * @param cancel

         *

         * @throws DriverException

         */

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

                throws DriverException {

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

        try {

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

            if (adapter.getShapeCount() <= 0)

            {

                logger.debug("Layer:" + getCapa().getName() + " sin registros");

                return;

            }

            Selectable selection = (Selectable) getCapa();

            ICoordTrans ct = getCapa().getCoordTrans();

            BitSet bitSet = selection.getSelection();

            BoundedShapes shapeBounds = (BoundedShapes) adapter.getDriver();

            VectorialFileDriver driver = (VectorialFileDriver) adapter.getDriver();

            // logger.debug("adapter.start() -> Layer:" + getCapa().getName());

            adapter.start();

            IGeometry geom;

            if (adapter.getShapeCount()>0){

            geom = adapter.getShape(0);

            }

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

            Rectangle2D extent = viewPort.getAdjustedExtent();

            //AffineTransform at = viewPort.getAffineTransform();

            int sc;

            Rectangle2D bounds;

            sc = adapter.getShapeCount();

            long t1 = System.currentTimeMillis();

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

            ((AlphanumericData) getCapa()).getRecordset().start();

            // 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();

                }

            }

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

                bounds = shapeBounds.getShapeBounds(i);

                if (ct != null) {

                    bounds = ct.convert(bounds);

                }

                if (XRectangle2D.intersectInclusive(extent, bounds)) {

                    FSymbol symbol = l.getSymbol(i);

                    if (bitSet.get(i)) {

                        symbol = FSymbol.getSymbolForSelection(symbol);

                    }

                    geom = driver.getShape(i);

                    // PRUEBA DE VELOCIDAD

                    // geom = ShapeFactory.createPolygon2D(new GeneralPathX(bounds));

                    if (ct != null) {

                        if (bMustClone)

                            geom = geom.cloneGeometry();

                        geom.reProject(ct);

                    }

                    geom.draw(g, viewPort, symbol);

                }

            }

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

            ((AlphanumericData) getCapa()).getRecordset().stop();

            long t2 = System.currentTimeMillis();

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

            adapter.stop();

            // System.out.println(t2 - t1);

        } 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);

        } catch (IOException e) {

            throw new DriverException(e);

        }

        }

    /* (non-Javadoc)

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

     */

    public FBitSet queryByShape(IGeometry g, int relationship)

    throws DriverException, VisitException {

        // Si hay un ?ndice espacial, lo usamos para hacer el query.

        FLyrVect lyr = (FLyrVect) capa;

//        if (lyr.getSpatialIndex() == null)

        if(lyr.getISpatialIndex() == null)

            return super.queryByShape(g, relationship);

        long t1 = System.currentTimeMillis();

        ReadableVectorial va = lyr.getSource();

        ICoordTrans ct = lyr.getCoordTrans();

        Rectangle2D bounds = g.getBounds2D();

//        Coordinate c1 = new Coordinate(bounds.getMinX(), bounds.getMinY());

//        Coordinate c2 = new Coordinate(bounds.getMaxX(), bounds.getMaxY());

//        Envelope env = new Envelope(c1, c2);

//        List lstRecs = lyr.getSpatialIndex().query(env);

        List lstRecs = lyr.getISpatialIndex().query(bounds);

        Integer idRec;

        FBitSet bitset = new FBitSet();

        Geometry jtsShape = g.toJTSGeometry();

        IntersectionMatrix m;

        int index;

        try {

            va.start();

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

            {

                idRec = (Integer) lstRecs.get(i);

                index = idRec.intValue();

                IGeometry geom=getGeometry(((FLyrAnnotation)capa).getLabel(index).getBoundBox());

                if (ct != null) {

                    geom.reProject(ct);

                }

                Geometry jtsGeom = geom.toJTSGeometry();

                switch (relationship) {

                case CONTAINS:

                    m = jtsShape.relate(jtsGeom);

                    if (m.isContains()) {

                        bitset.set(index, true);

                    }

                    break;

                case CROSSES:

                    m = jtsShape.relate(jtsGeom);

                    if (m.isCrosses(jtsGeom.getDimension(), jtsShape.getDimension())) {

                        bitset.set(index, true);

                    }

                    break;

                case DISJOINT:

                    // TODO: CREO QUE EL DISJOINT NO SE PUEDE METER AQUI

                    m = jtsShape.relate(jtsGeom);

                    if (m.isDisjoint()) {

                        bitset.set(index, true);

                    }

                    break;

                case EQUALS:

                    m = jtsShape.relate(jtsGeom);

                    if (m.isEquals(jtsGeom.getDimension(), jtsShape.getDimension())) {

                        bitset.set(index, true);

                    }

                    break;

                case INTERSECTS:

                    m = jtsShape.relate(jtsGeom);

                    if (m.isIntersects()) {

                        bitset.set(index, true);

                    }

                    break;

                case OVERLAPS:

                    m = jtsShape.relate(jtsGeom);

                    if (m.isOverlaps(jtsGeom.getDimension(), jtsShape.getDimension()))

                    {

                        bitset.set(index, true);

                    }

                    break;

                case TOUCHES:

                    m = jtsShape.relate(jtsGeom);

                    if (m.isTouches(jtsGeom.getDimension(), jtsShape.getDimension()))

                    {

                        bitset.set(index, true);

                    }

                    break;

                case WITHIN:

                    m = jtsShape.relate(jtsGeom);

                    if (m.isWithin()) {

                        bitset.set(index, true);

                    }

                    break;

                }

            }

            va.stop();

        } catch (DriverIOException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

        long t2 = System.currentTimeMillis();

        logger.debug("queryByShape optimizado sobre la capa " + lyr.getName() + ". " + (t2-t1) + " mseg.");

        return bitset;

    }

    public FBitSet queryByRect(Rectangle2D rect) throws DriverException {

        // Si hay un ?ndice espacial, lo usamos para hacer el query.

        FLyrAnnotation lyr = (FLyrAnnotation) capa;

//        if (lyr.getSpatialIndex() == null)

          if(lyr.getISpatialIndex() == null)

            return super.queryByRect(rect);

        ReadableVectorial va = lyr.getSource();

        ICoordTrans ct = lyr.getCoordTrans();

        Rectangle2D bounds = rect;

//        Coordinate c1 = new Coordinate(bounds.getMinX(), bounds.getMinY());

//        Coordinate c2 = new Coordinate(bounds.getMaxX(), bounds.getMaxY());

//        Envelope env = new Envelope(c1, c2);

//

//        List lstRecs = lyr.getSpatialIndex().query(env);

        //azabala

        List lstRecs = lyr.getISpatialIndex().query(bounds);

        Integer idRec;

        FBitSet bitset = new FBitSet();

        int index;

        try {

            va.start();

            DriverAttributes attr = va.getDriverAttributes();

            boolean bMustClone = false;

            if (attr != null)

            {

                if (attr.isLoadedInMemory())

                {

                    bMustClone = attr.isLoadedInMemory();

                }

            }

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

            {

                idRec = (Integer) lstRecs.get(i);

                index = idRec.intValue();

                IGeometry geom=getGeometry(((FLyrAnnotation)capa).getLabel(index).getBoundBox());

                if (ct != null) {

                    if (bMustClone)

                        geom = geom.cloneGeometry();

                    geom.reProject(ct);

                }

                //System.out.println("Rect?ngulo de selecci?n = "+ rect);

                //System.out.println("Rect?ngulo de la geometr?a = "+ geom.getBounds2D());

                if (geom.intersects(rect)){

                    bitset.set(index, true);

                }

            }

            va.stop();

        } catch (DriverIOException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

        return bitset;

    }

    /* (non-Javadoc)

     * @see com.iver.cit.gvsig.fmap.operations.strategies.Strategy#queryByPoint(java.awt.geom.Point2D, double)

     */

    public FBitSet queryByPoint(Point2D p, double tolerance)

    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);

    }

    private IGeometry getGeometry(Rectangle2D r){

            GeneralPathX resul = new GeneralPathX();

                Point2D[] vs=new Point2D[4];

                vs[0]=new Point2D.Double(r.getX(),r.getY());

            vs[1]=new Point2D.Double(r.getMaxX(),r.getY());

            vs[2]=new Point2D.Double(r.getMaxX(),r.getMaxY());

            vs[3]=new Point2D.Double(r.getX(),r.getMaxY());

            //vs[4]=new Point2D.Double(r.getX(),r.getY());

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

                        if (i == 0) {

                                resul.moveTo(vs[i].getX(),vs[i].getY());

                        } else {

                                resul.lineTo(vs[i].getX(),vs[i].getY());

                        }

                }

                resul.closePath();

                return ShapeFactory.createPolygon2D(resul);

    }

}