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

import com.hardcode.driverManager.DriverLoadException;

import com.hardcode.gdbms.engine.data.DataSource;

import com.hardcode.gdbms.engine.instruction.FieldNotFoundException;

import com.hardcode.gdbms.engine.values.DoubleValue;

import com.hardcode.gdbms.engine.values.FloatValue;

import com.hardcode.gdbms.engine.values.NullValue;

import com.hardcode.gdbms.engine.values.Value;

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.v02.FLabel;

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

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

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

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

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

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

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

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

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

import com.iver.cit.gvsig.fmap.operations.strategies.FeatureVisitor;

import com.iver.cit.gvsig.fmap.operations.strategies.Strategy;

import com.iver.cit.gvsig.fmap.operations.strategies.StrategyManager;

import com.iver.cit.gvsig.fmap.operations.strategies.VisitException;

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

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

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

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

import com.iver.utiles.XMLEntity;

import org.apache.log4j.Logger;

import org.cresques.cts.ICoordTrans;

import java.awt.Graphics2D;

import java.awt.geom.Point2D;

import java.awt.geom.Rectangle2D;

import java.awt.image.BufferedImage;

import java.util.ArrayList;

import java.util.BitSet;

/**

 * Capa b?sica Vectorial.

 *

 * @author Fernando Gonz?lez Cort?s

 */

//TODO Cuando no sea para pruebas debe no ser public

public class FLyrVect extends FLyrDefault implements Labelable, Selectable,

        AlphanumericData, ClassifiableVectorial, SingleLayer, VectorialData,

        RandomVectorialData {

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

        /** Leyenda de la capa vectorial */

        private VectorialLegend legend;

        private int typeShape = -1;

        private SelectionSupport selectionSupport = new SelectionSupport();

        private LayerChangeSupport layerChangeSupport = new LayerChangeSupport();

        private VectorialAdapter source;

        private SelectableDataSource sds;

        /**

         * A?ade un SelectionListener a la lista de listeners.

         *

         * @param listener SelectionListener.

         */

        public void addSelectionListener(SelectionListener listener) {

                selectionSupport.addSelectionListener(listener);

        }

        /**

         * Borra un selectionListener de la lista de listeners.

         *

         * @param listener SelectionListener

         */

        public void removeSelectionListener(SelectionListener listener) {

                selectionSupport.removeSelectionListener(listener);

        }

        /**

         * Cuando ocurre un evento de cambio en la selecci?n, ?ste puede ser uno de

         * una gran cantidad de eventos. Con el fin de no propagar todos estos

         * eventos, se realiza la propagaci?n de manera manual al final de la

         * "r?faga" de eventos

         */

        public void fireSelectionEvents() {

                selectionSupport.fireSelectionEvents();

        }

        /**

         * Devuelve el VectorialAdapater de la capa.

         *

         * @return VectorialAdapter.

         */

        public VectorialAdapter getSource() {

                return source;

        }

        /**

         * Inserta el VectorialAdapter a la capa.

         *

         * @param va VectorialAdapter.

         */

        public void setSource(VectorialAdapter va) {

                source = va;

        }

        /**

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

         */

        public Rectangle2D getFullExtent() throws DriverException {

                try {

                        Rectangle2D rAux;

                        logger.debug("source.start()");

                        source.start();

                        rAux = source.getFullExtent();

                        logger.debug("source.stop()");

                        source.stop();

                        // Si existe reproyecci?n, reproyectar el extent

                        ICoordTrans ct = getCoordTrans();

                        if (ct != null) {

                                Point2D pt1 = new Point2D.Double(rAux.getMinX(), rAux.getMinY());

                                Point2D pt2 = new Point2D.Double(rAux.getMaxX(), rAux.getMaxY());

                                pt1 = ct.convert(pt1, null);

                                pt2 = ct.convert(pt2, null);

                                rAux = new Rectangle2D.Double();

                                rAux.setFrameFromDiagonal(pt1, pt2);

                        }

                        return rAux;

                } catch (DriverIOException e) {

                        throw new DriverException(e);

                }

        }

        /**

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

         *                 java.awt.Graphics2D, com.iver.cit.gvsig.fmap.ViewPort)

         */

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

                Cancellable cancel) throws DriverException {

                Strategy strategy = StrategyManager.getStrategy(this);

                strategy.draw(image, g, viewPort, cancel);

                if (getVirtualLayers() != null) {

                        getVirtualLayers().draw(image, g, viewPort, cancel);

                }

                if (getLayerText() != null) {

                        getLayerText().draw(image, g, viewPort, cancel);

                }

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.FLayer#print(java.awt.Graphics2D,

         *                 com.iver.cit.gvsig.fmap.ViewPort,

         *                 com.iver.cit.gvsig.fmap.operations.Cancellable)

         */

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

                throws DriverException {

                Strategy strategy = StrategyManager.getStrategy(this);

                strategy.print(g, viewPort, cancel);

                if (getLayerText() != null) {

                        getLayerText().draw(null, g, viewPort, cancel);

                }

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.VectorialOperations#createLabelLayer(int)

         */

        public FLayer createLabelLayer(int fieldId) {

                ArrayList labels = new ArrayList();

                try {

                        VectorialAdapter adapter = getSource();

                        DataSource ds = getRecordset();

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

                        adapter.start();

                        ds.start();

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

                        int sc;

                        sc = adapter.getShapeCount();

                        VectorialLegend l = (VectorialLegend) getLegend();

                        int idFieldHeightText = -1;

                        int idFieldRotationText = -1;

                        boolean bWithHeightText = false;

                        if (l.getLabelHeightField() != null) {

                                bWithHeightText = true;

                                idFieldHeightText = ds.getFieldIndexByName(l.getLabelHeightField());

                        }

                        boolean bWithRotationText = false;

                        if (l.getLabelRotationField() != null) {

                                bWithRotationText = true;

                                idFieldRotationText = ds.getFieldIndexByName(l.getLabelRotationField());

                        }

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

                                Value val = ds.getFieldValue(i, fieldId);

                                if ((val instanceof NullValue) || (val == null)) {

                                        continue;

                                }

                                IGeometry geom = adapter.getShape(i);

                                if (geom == null) {

                                        continue;

                                }

                                FSymbol symbol = l.getSymbol(i);

                                // TODO: El m?todo contenedor (createLabelLayer) debe recoger

                                // los par?metros de posicionamiento y de allowDuplicates

                                // if (i >= 328)

                                //         System.out.println("i= " + i + " " + val.toString());

                                FLabel[] lbls = geom.createLabels(0, true);

                                for (int j = 0; j < lbls.length; j++) {

                                        if (lbls[j] != null) {

                                                lbls[j].setString(val.toString());

                                                if (bWithHeightText) {

                                                        FloatValue height = (FloatValue) ds.getFieldValue(i,

                                                                        idFieldHeightText);

                                                        lbls[j].setHeight(height.getValue());

                                                } else {

                                                        lbls[j].setHeight(symbol.getFontSize());

                                                }

                                                if (bWithRotationText) {

                                                        DoubleValue rotation = (DoubleValue) ds.getFieldValue(i,

                                                                        idFieldRotationText);

                                                        lbls[j].setRotation(rotation.getValue());

                                                }

                                                labels.add(lbls[j]);

                                        }

                                }

                        }

                        long t2 = System.currentTimeMillis();

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

                        ds.stop();

                        adapter.stop();

                } catch (DriverIOException e) {

                        e.printStackTrace();

                } catch (DriverException e) {

                        e.printStackTrace();

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

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                } catch (FieldNotFoundException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                FLyrText layerText = new FLyrText(labels);

                try {

                        layerText.setLegend((VectorialLegend) getLegend());

                } catch (FieldNotFoundException e1) {

                        // TODO Auto-generated catch block

                        e1.printStackTrace();

                } catch (DriverException e1) {

                        // TODO Auto-generated catch block

                        e1.printStackTrace();

                }

                setLayerText(layerText);

                return layerText;

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.VectorialOperations#removeLabels()

         */

        public void removeLabels() {

                setLayerText(null);

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.VectorialOperations#createIndex()

         */

        public void createIndex() {

        }

        /**

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

         *                 com.iver.cit.gvsig.fmap.operations.selection.VectorialSubSet)

         */

        public void process(FeatureVisitor visitor, BitSet subset)

                throws DriverException, VisitException {

                Strategy s = StrategyManager.getStrategy(this);

                s.process(visitor, subset);

        }

        /**

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

         */

        public void process(FeatureVisitor visitor)

                throws DriverException, VisitException {

                Strategy s = StrategyManager.getStrategy(this);

                s.process(visitor);

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.CommonOperations#setSelection(com.iver.cit.gvsig.fmap.operations.selection.VectorialSubSet)

         */

        public void setSelection(FBitSet selection) {

                selectionSupport.setSelection(selection);

                fireSelectionEvents();

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.CommonOperations#isSelected(int)

         */

        public boolean isSelected(int index) {

                return selectionSupport.isSelected(index);

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.CommonOperations#getSelection()

         */

        public FBitSet getSelection() {

                return selectionSupport.getSelection();

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.CommonOperations#clearSelection()

         */

        public void clearSelection() {

                selectionSupport.clearSelection();

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.CommonOperations#queryByRect(java.awt.geom.Rectangle2D)

         */

        public BitSet queryByRect(Rectangle2D rect) throws DriverException {

                Strategy s = StrategyManager.getStrategy(this);

                return s.queryByRect(rect);

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.CommonOperations#getRecordset()

         */

        public SelectableDataSource getRecordset() throws DriverException {

                if (sds == null) {

                        //Nombre en el GDBMS de la tabla

                        String name = this.toString();

                        try {

                                DataSource ds = source.getRecordset(name);

                                if (ds == null) {

                                        return null;

                                }

                                sds = new SelectableDataSource(ds);

                                sds.setSelectionSupport(selectionSupport);

                        } catch (DriverLoadException e) {

                                throw new DriverException(e);

                        }

                }

                return sds;

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.CommonOperations#setLegend(int,

         *                 com.iver.cit.gvsig.fmap.rendering.Legend)

         */

        public void setLegend(VectorialLegend r)

                throws DriverException, FieldNotFoundException {

                VectorialLegend oldLegend = legend;

                legend = r;

                try {

                        legend.setDataSource(getRecordset());

                        if (legend.getLabelField() != null) {

                            sds.start();

                                int idLabelField = getRecordset().getFieldIndexByName(legend.getLabelField());

                                createLabelLayer(idLabelField);

                                sds.stop();

                        }

                        else

                            removeLabels();

                } catch (DriverException e) {

                        throw new DriverException(e);

                } catch (FieldNotFoundException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

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

                        throw new DriverException(e);

        }

                LegendChangedEvent e = LegendChangedEvent.createLegendChangedEvent(oldLegend, legend);

                callLegendChanged(e);

        }

        /**

         * Devuelve la Leyenda de la capa.

         *

         * @return Leyenda.

         */

        public Legend getLegend() {

                return legend;

        }

        /**

         * Devuelve el tipo de shape que contiene la capa.

         *

         * @return tipo de shape.

         *

         * @throws DriverException

         */

        public int getShapeType() throws DriverException {

                if (typeShape == -1) {

                        try {

                                logger.debug("source.start()");

                                source.start();

                                typeShape = source.getShapeType();

                                logger.debug("source.stop()");

                                source.stop();

                        } catch (DriverIOException e) {

                                throw new DriverException(e);

                        }

                }

                return typeShape;

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.FLayer#getProperties()

         */

        public XMLEntity getXMLEntity() {

                XMLEntity xml = super.getXMLEntity();

                xml.addChild(legend.getXMLEntity());

                xml.addChild(selectionSupport.getXMLEntity());

                if (source instanceof VectorialFileAdapter) {

                        xml.putProperty("file", ((VectorialFileAdapter) source).getFile());

                } else if (source instanceof VectorialDBAdapter) {

                } else if (source instanceof WFSAdapter) {

                }

                xml.putProperty("driverName", getSource().getDriver().getName());

                return xml;

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.FLyrDefault#setXMLEntity(com.iver.utiles.XMLEntity)

         */

        public void setXMLEntity(XMLEntity xml)

                throws XMLException {

                super.setXMLEntity(xml);

                legend = LegendFactory.createFromXML(xml.getChild(0));

                try {

                        // legend.setDataSource(getRecordset());

                        setLegend(legend);

                } catch (FieldNotFoundException e) {

                        throw new XMLException(e);

                } catch (DriverException e) {

                        throw new XMLException(e);

                }

                selectionSupport.setXMLEntity(xml.getChild(1));

        }

        /**

         * A?ade un LegendListener a la lista de Listeners.

         *

         * @param listener LegendListener.

         */

        public void addLegendListener(LegendListener listener) {

                layerChangeSupport.addLayerListener(listener);

        }

        /**

         * Llamada al m?todo callLegendChanged de los listener.

         *

         * @param e Evento.

         */

        private void callLegendChanged(LegendChangedEvent e) {

                layerChangeSupport.callLegendChanged(e);

        }

        /**

         * Borra un LegendListener de la lista de Listeners

         *

         * @param listener LegendListener.

         */

        public void removeLegendListener(LegendListener listener) {

                layerChangeSupport.removeLayerListener(listener);

        }

        /**

         * @see com.iver.cit.gvsig.fmap.layers.CommonOperations#changeRecordsetName()

         */

        public void changeRecordsetName(String newName) throws DriverException {

                source.changeRecordsetName(newName);

        }

        /**

         * Sobreimplementaci?n del m?todo toString para que las bases de datos

         * identifiquen la capa.

         *

         * @return DOCUMENT ME!

         */

        public String toString() {

                /*

                 * Se usa internamente para que la parte de datos

                 * identifique de forma un?voca las tablas

                 */

                String ret = super.toString();

                return "layer" + ret.substring(ret.indexOf('@') + 1);

        }

}