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

import java.awt.geom.Rectangle2D;

import java.awt.image.BufferedImage;

import java.io.IOException;

import java.util.BitSet;

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

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

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.layers.FLayer;

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

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

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

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

/**

 * Esta clase definir? las operaciones de la interfaz FLyrVect de la manera m?s

 * ?ptima para los ficheros shp.

 */

public class ShpStrategy extends DefaultStrategy {

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

        /**

         * Crea una ShpStrategy.

         *

         * @param capa

         */

        public ShpStrategy(FLayer capa) {

                super(capa);

        }

        /**

         * @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 {

                try {

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

                        if (l instanceof ClassifiedLegendInfo) {

                                ClassifiedLegendInfo clsfLegend = (ClassifiedLegendInfo) l;

                                FSymbol[] symbs = clsfLegend.getSymbols();

                                //double rSym = 0;

                                //double maxRSym = -1;

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

                                        // TODO: REVISAR LOS SIMBOLOS Y SUS TAMA?OS

                                        /* Style2D pointSymbol = symbs[i].getPointStyle2D();

                                           if (pointSymbol instanceof MarkStyle2D)

                                           {

                                                   MarkStyle2D mrk2D = (MarkStyle2D) pointSymbol;

                                                   rSym = viewPort.toMapDistance(mrk2D.getSize());

                                                   if (maxRSym < rSym)

                                                           maxRSym = rSym;

                                           }*/

                                }

                        }

                        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++) {

                                // Salimos si alguien cancela

                                if (cancel.isCanceled()) {

                                        break;

                                }

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

                                        }

                                        boolean bPoint = (shapeBounds.getShapeType(i) == FShape.POINT);

                                        if (bPoint ||

                                                        ((bounds.getHeight() > viewPort.getDist1pixel()) ||

                                                        (bounds.getWidth() > viewPort.getDist1pixel()))) {

                                                geom = driver.getShape(i);

                                                if (ct != null) {

                                                    if (bMustClone)

                                                        geom = geom.cloneGeometry();

                                                        geom.reProject(ct);

                                                }

                                                geom.draw(g, viewPort, symbol);

                                        } else {

                                                Point2D.Double pOrig = new Point2D.Double(bounds.getMinX(),

                                                                bounds.getMinY());

                                                Point2D pDest;

                                                // pDest = viewPort.getAffineTransform().transform(pOrig, null);

                        pDest = g.getTransform().transform(pOrig, null);

                                                int pixX = (int) pDest.getX();

                                                int pixY = (int) pDest.getY();

                                                if (symbol ==null)

                                                        continue;

                                                if ((pixX > 0) && (pixX < image.getWidth())) {

                                                        if ((pixY > 0) && (pixY < image.getHeight())) {

                                                                image.setRGB(pixX, pixY, symbol.getRgb());

                                                        }

                                                }

                                        }

                                }

                        }

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

                }

        }

        /**

         * 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!!

        }

}