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.gui.cad;

import java.awt.Color;

import java.awt.Graphics;

import java.awt.Graphics2D;

import java.awt.event.MouseEvent;

import java.awt.geom.AffineTransform;

import java.awt.geom.Point2D;

import java.awt.image.BufferedImage;

import java.util.ArrayList;

import java.util.Iterator;

import org.apache.log4j.Logger;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.exception.ReadException;

import org.gvsig.fmap.dal.feature.EditableFeature;

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.dal.feature.FeatureSelection;

import org.gvsig.fmap.dal.feature.FeatureSet;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryFactory;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.handler.Handler;

import org.gvsig.fmap.geom.operation.Draw;

import org.gvsig.fmap.geom.operation.DrawOperationContext;

import org.gvsig.fmap.geom.operation.GeometryOperationException;

import org.gvsig.fmap.geom.operation.GeometryOperationNotSupportedException;

import org.gvsig.fmap.geom.primitive.DefaultEnvelope;

import org.gvsig.fmap.geom.primitive.Envelope;

import org.gvsig.fmap.geom.primitive.GeneralPathX;

import org.gvsig.fmap.mapcontext.ViewPort;

import org.gvsig.fmap.mapcontext.layers.SpatialCache;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.fmap.mapcontext.rendering.symbols.FGraphicUtilities;

import org.gvsig.fmap.mapcontext.rendering.symbols.ISymbol;

import org.gvsig.fmap.mapcontext.rendering.symbols.SymbologyFactory;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

import com.iver.cit.gvsig.CADExtension;

import com.iver.cit.gvsig.gui.cad.exception.CommandException;

import com.iver.cit.gvsig.layers.VectorialLayerEdited;

import com.iver.cit.gvsig.project.documents.view.gui.View;

import com.iver.utiles.console.JConsole;

/**

 * DOCUMENT ME!

 *

 * @author Vicente Caballero Navarro

 */

public abstract class DefaultCADTool implements CADTool {

        public static ISymbol selectionSymbol = SymbologyFactory

                        .createDefaultSymbolByShapeType(Geometry.TYPES.GEOMETRY, new Color(

                                        255, 0, 0, 100)); // Le ponemos una transparencia

        public static ISymbol axisReferencesSymbol = SymbologyFactory

                        .createDefaultSymbolByShapeType(Geometry.TYPES.GEOMETRY, new Color(

                                        100, 100, 100, 100));

        public static ISymbol geometrySelectSymbol = SymbologyFactory

                        .createDefaultSymbolByShapeType(Geometry.TYPES.GEOMETRY, Color.RED);

        public static ISymbol handlerSymbol = SymbologyFactory

                        .createDefaultSymbolByShapeType(Geometry.TYPES.GEOMETRY,

                                        Color.ORANGE);

        private static Logger logger = Logger.getLogger(DefaultCADTool.class

                        .getName());

        private CADToolAdapter cadToolAdapter;

        private String question;

        private String[] currentdescriptions;

        private String tool = "selection";

        private DefaultCADTool previousTool;

        private ArrayList temporalCache = new ArrayList();

        protected GeometryFactory geomFactory = GeometryManager.getInstance()

                        .getGeometryFactory();

        public void addTemporalCache(Geometry geom) {

                temporalCache.add(geom);

                insertSpatialCache(geom);

        }

        public void clearTemporalCache() {

                Geometry[] geoms = (Geometry[]) temporalCache.toArray(new Geometry[0]);

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

                        removeSpatialCache(geoms[i]);

                }

                temporalCache.clear();

        }

        private void insertSpatialCache(Geometry geom) {

                VectorialLayerEdited vle = getVLE();

                SpatialCache spatialCache = ((FLyrVect) vle.getLayer())

                                .getSpatialCache();

                Envelope r = geom.getEnvelope();

                if (geom.getType() == Geometry.TYPES.POINT) {

                        r = new DefaultEnvelope(r.getMinimum(0), r.getMinimum(1), r

                                        .getMinimum(0) + 1, r.getMinimum(1) + 1);// Rectangle2D.Double(r.getX(),r.getY(),1,1);

                }

                spatialCache.insert(r, geom);

        }

        private void removeSpatialCache(Geometry geom) {

                VectorialLayerEdited vle = getVLE();

                SpatialCache spatialCache = ((FLyrVect) vle.getLayer())

                                .getSpatialCache();

                Envelope r = geom.getEnvelope();

                if (geom.getType() == Geometry.TYPES.POINT) {

                        r = new DefaultEnvelope(r.getMinimum(0), r.getMinimum(1), r

                                        .getMinimum(0) + 1, r.getMinimum(1) + 1);// Rectangle2D.Double(r.getX(),r.getY(),1,1);

                }

                spatialCache.remove(r, geom);

        }

        /**

         * DOCUMENT ME!

         */

        public void draw(Geometry geometry) {

                if (geometry != null) {

                        BufferedImage img = getCadToolAdapter().getMapControl().getImage();

                        Graphics2D gImag = (Graphics2D) img.getGraphics();

                        ViewPort vp = getCadToolAdapter().getMapControl().getViewPort();

                        DrawOperationContext doc = new DrawOperationContext();

                        doc.setGraphics((Graphics2D) gImag);

                        doc.setViewPort(vp);

                        doc.setSymbol(DefaultCADTool.geometrySelectSymbol);

                        try {

                                geometry.cloneGeometry().invokeOperation(Draw.CODE, doc);

                        } catch (GeometryOperationNotSupportedException e) {

                                e.printStackTrace();

                        } catch (GeometryOperationException e) {

                                e.printStackTrace();

                        }

                        // geometry.draw(gImag, vp, DefaultCADTool.selectionSymbol);

                }

        }

        /**

         * DOCUMENT ME!

         *

         * @param cta

         *            DOCUMENT ME!

         */

        public void setCadToolAdapter(CADToolAdapter cta) {

                cadToolAdapter = cta;

        }

        /**

         * DOCUMENT ME!

         *

         * @return DOCUMENT ME!

         */

        public CADToolAdapter getCadToolAdapter() {

                return cadToolAdapter;

        }

        public VectorialLayerEdited getVLE() {

                return (VectorialLayerEdited) CADExtension.getEditionManager()

                                .getActiveLayerEdited();

        }

        /**

         * DOCUMENT ME!

         *

         * @param g

         *            DOCUMENT ME!

         * @param firstPoint

         *            DOCUMENT ME!

         * @param endPoint

         *            DOCUMENT ME!

         */

        public void drawLine(Graphics2D g, Point2D firstPoint, Point2D endPoint,

                        ISymbol symbol) {

                GeneralPathX elShape = new GeneralPathX(GeneralPathX.WIND_EVEN_ODD, 2);

                elShape.moveTo(firstPoint.getX(), firstPoint.getY());

                elShape.lineTo(endPoint.getX(), endPoint.getY());

                GeometryFactory geomFactory = GeometryManager.getInstance()

                                .getGeometryFactory();

                DrawOperationContext doc = new DrawOperationContext();

                doc.setGraphics(g);

                doc.setViewPort(getCadToolAdapter().getMapControl().getViewPort());

                doc.setSymbol(symbol);

                try {

                        geomFactory.createPolyline2D(elShape).invokeOperation(Draw.CODE,

                                        doc);

                } catch (GeometryOperationNotSupportedException e) {

                        e.printStackTrace();

                } catch (GeometryOperationException e) {

                        e.printStackTrace();

                }

                // draw(g,

                // getCadToolAdapter().getMapControl().getViewPort(),

                // symbol);

        }

        /**

         * DOCUMENT ME!

         *

         * @param geometry

         *            DOCUMENT ME!

         */

        public Feature insertGeometry(Geometry geometry, Feature feature) {

                VectorialLayerEdited vle = getVLE();

                try {

                        FeatureStore featureStore = ((FLyrVect) vle.getLayer())

                                        .getFeatureStore();

                        EditableFeature eFeature = featureStore.createNewFeature(

                                        featureStore.getDefaultFeatureType(), feature);

                        eFeature.setGeometry(featureStore.getDefaultFeatureType()

                                        .getDefaultGeometryAttributeName(), geometry);

                        featureStore.insert(eFeature);

                        drawToImage(featureStore, vle, eFeature);

                        draw(geometry.cloneGeometry());

                        return eFeature;

                } catch (ReadException e) {

                        NotificationManager.addError(e.getMessage(), e);

                        return null;

                } catch (GeometryOperationNotSupportedException e) {

                        NotificationManager.addError(e.getMessage(), e);

                        return null;

                } catch (GeometryOperationException e) {

                        NotificationManager.addError(e.getMessage(), e);

                        return null;

                } catch (DataException e) {

                        NotificationManager.addError(e.getMessage(), e);

                        return null;

                }

        }

        public Feature insertGeometry(Geometry geometry) {

                VectorialLayerEdited vle = getVLE();

                try {

                        FeatureStore featureStore = ((FLyrVect) vle.getLayer())

                                        .getFeatureStore();

                        EditableFeature eFeature = featureStore.createNewFeature(true);

                        eFeature.setGeometry(featureStore.getDefaultFeatureType()

                                        .getDefaultGeometryAttributeName(), geometry);

                        featureStore.insert(eFeature);

                        drawToImage(featureStore, vle, eFeature);

                        draw(geometry.cloneGeometry());

                        return eFeature;

                } catch (ReadException e) {

                        NotificationManager.addError(e.getMessage(), e);

                        return null;

                } catch (GeometryOperationNotSupportedException e) {

                        NotificationManager.addError(e.getMessage(), e);

                        return null;

                } catch (GeometryOperationException e) {

                        NotificationManager.addError(e.getMessage(), e);

                        return null;

                } catch (DataException e) {

                        NotificationManager.addError(e.getMessage(), e);

                        return null;

                }

        }

        private void drawToImage(FeatureStore featureStore,VectorialLayerEdited vle, Feature feature) throws DataException, GeometryOperationNotSupportedException, GeometryOperationException {

                clearSelection();

                ViewPort vp = vle.getLayer().getMapContext().getViewPort();

                BufferedImage selectionImage = new BufferedImage(

                                vp.getImageWidth(), vp.getImageHeight(),

                                BufferedImage.TYPE_INT_ARGB);

                Graphics2D gs = selectionImage.createGraphics();

                // int inversedIndex=vea.getInversedIndex(index);

                FeatureSelection selection = (FeatureSelection) featureStore.getSelection();

//                selection.select(feature);

                // vle.addSelectionCache(new DefaultRowEdited(df,

                // IRowEdited.STATUS_ADDED, inversedIndex ));

                // vea.getSelection().set(inversedIndex);

                // Geometry geom = (Geometry)feature.getDefaultGeometry();

                DrawOperationContext doc = new DrawOperationContext();

                doc.setGraphics(gs);

                doc.setViewPort(vp);

                doc.setSymbol(DefaultCADTool.selectionSymbol);

                Geometry geometry=feature.getDefaultGeometry();

                geometry.cloneGeometry().invokeOperation(Draw.CODE, doc);

                // draw(gs, vp, DefaultCADTool.selectionSymbol);

                vle.drawHandlers(geometry.cloneGeometry(), gs, vp);

                vle.setSelectionImage(selectionImage);

                insertSpatialCache(geometry);

        }

        /**

         * DOCUMENT ME!

         *

         * @param geometry

         *            DOCUMENT ME!

         */

        // public void modifyFeature(Feature feature, Feature oldFeature) {

        // try {

        // CommandsRecord

        // cr=((FLyrVect)getVLE().getLayer()).getFeatureStore().getCommandsRecord();

        // cr.update(feature, oldFeature);

        // } catch (ReadException e) {

        // NotificationManager.addError(e.getMessage(),e);

        // }

        // draw(((Geometry)feature.getDefaultGeometry()).cloneGeometry());

        // }

        /**

         * DOCUMENT ME!

         *

         * @param geometry

         *            DOCUMENT ME!

         * @param values

         *            DOCUMENT ME!

         */

//        public Feature addGeometry(Geometry geometry, Object[] values) {

//                // int index = 0;

//                try {

//                        FeatureStore featureStore = ((FLyrVect) getVLE().getLayer())

//                                        .getFeatureStore();

//                        EditableFeature eFeature = featureStore.createNewFeature(true);

//                        eFeature.setGeometry(featureStore.getDefaultFeatureType()

//                                        .getDefaultGeometryAttributeName(), geometry);

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

//                                eFeature.set(i, values[i]);

//                        }

//                        featureStore.insert(eFeature);

//                        return eFeature;

//                } catch (DataException e) {

//                        // TODO Auto-generated catch block

//                        e.printStackTrace();

//                }

//                // try {

//                // String newFID = vea.getNewFID();

//                // DefaultFeature df = new DefaultFeature(geometry, values, newFID);

//                // index = vea.addRow(df, getName(), EditionEvent.GRAPHIC);

//                // insertSpatialCache(geometry);

//                // } catch (ValidateRowException e) {

//                // NotificationManager.addError(e);

//                // } catch (ReadException e) {

//                // NotificationManager.addError(e);

//                // }

//                // return vea.getInversedIndex(index);

//                return null;

//        }

        /**

         * Devuelve la cadena que corresponde al estado en el que nos encontramos.

         *

         * @return Cadena para mostrar por consola.

         */

        public String getQuestion() {

                return question;

        }

        /**

         * Actualiza la cadena que corresponde al estado actual.

         *

         * @param s

         *            Cadena que aparecer? en consola.

         */

        public void setQuestion(String s) {

                question = s;

                // ConsoleToken.addQuestion(s);

        }

        /**

         * Provoca un repintado "soft" de la capa activa en edici?n. Las capas por

         * debajo de ella no se dibujan de verdad, solo se dibuja la que est? en

         * edici?n y las que est?n por encima de ella en el TOC.

         */

        public void refresh() {

                // getCadToolAdapter().getMapControl().drawMap(false);

                getVLE().getLayer().setDirty(true);

                getCadToolAdapter().getMapControl().rePaintDirtyLayers();

        }

        /*

         * public void drawHandlers(Graphics g, FBitSet sel, AffineTransform at)

         * throws DriverIOException { for (int i = sel.nextSetBit(0); i >= 0; i =

         * sel.nextSetBit(i + 1)) { IGeometry ig =

         * getCadToolAdapter().getVectorialAdapter() .getShape(i).cloneGeometry();

         * if (ig == null) continue; Handler[] handlers =

         * ig.getHandlers(IGeometry.SELECTHANDLER);

         * FGraphicUtilities.DrawHandlers((Graphics2D) g, at, handlers); } }

         */

        public void drawHandlers(Graphics g, ArrayList selectedRows,

                        AffineTransform at) {

                FeatureSet selection = null;

                try {

                        selection = (FeatureSet) ((FLyrVect) getVLE().getLayer())

                                        .getFeatureStore().getSelection();

                        Iterator iterator = selection.iterator();

                        while (iterator.hasNext()) {

                                Feature feature = (Feature) iterator.next();

                                // }

                                // for (int i = 0; i < selectedRows.size(); i++) {

                                // IRowEdited edRow = (IRowEdited) selectedRows.get(i);

                                // IFeature feat = (IFeature) edRow.getLinkedRow();

                                // IFeature feat = (IFeature) selectedRows.get(i);

                                Geometry ig = ((Geometry) feature.getDefaultGeometry())

                                                .cloneGeometry();

                                if (ig == null)

                                        continue;

                                Handler[] handlers = ig.getHandlers(Geometry.SELECTHANDLER);

                                FGraphicUtilities.DrawHandlers((Graphics2D) g, at, handlers,

                                                DefaultCADTool.handlerSymbol);

                        }

                } catch (ReadException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                } catch (DataException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

        }

        public void setDescription(String[] currentdescriptions) {

                this.currentdescriptions = currentdescriptions;

        }

        public String[] getDescriptions() {

                return currentdescriptions;

        }

        /*

         * (non-Javadoc)

         *

         * @see com.iver.cit.gvsig.gui.cad.CADTool#end()

         */

        public void end() {

                CADExtension.setCADTool("_selection", true);

                PluginServices.getMainFrame().setSelectedTool("_selection");

                CADTool cadtool = CADExtension.getCADTool();

                cadtool.setPreviosTool(this);

        }

        public void init() {

                // jaume, should not be necessary

                // CADTool.drawingSymbol.setOutlined(true);

                // CADTool.drawingSymbol.setOutlineColor(Color.GREEN);

        }

        // protected ArrayList getSelectedRows() {

        // VectorialLayerEdited vle = getVLE();

        // ArrayList selectedRow = vle.getSelectedRow();

        // return selectedRow;

        // }

        protected ArrayList getSelectedHandlers() {

                VectorialLayerEdited vle = getVLE();

                ArrayList selectedHandlers = vle.getSelectedHandler();

                return selectedHandlers;

        }

        public void clearSelection() throws DataException {

                VectorialLayerEdited vle = getVLE();

                FeatureSet selection = null;

                selection = (FeatureSet) ((FLyrVect) vle.getLayer())

                                        .getFeatureStore().getSelection();

                // ArrayList selectedRow = vle.getSelectedRow();

                ArrayList selectedHandlers = vle.getSelectedHandler();

                selection.dispose();

                selectedHandlers.clear();

                // VectorialEditableAdapter vea = vle.getVEA();

                // FBitSet selection = vea.getSelection();

                // selection.clear();

                vle.setSelectionImage(null);

                vle.setHandlersImage(null);

        }

        public String getNextTool() {

                return tool;

        }

        public void setNextTool(String tool) {

                this.tool = tool;

        }

        public boolean changeCommand(String name) throws CommandException {

                CADTool[] cadtools = CADExtension.getCADTools();

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

                        CADTool ct = cadtools[i];

                        if (name.equalsIgnoreCase(ct.getName())

                                        || name.equalsIgnoreCase(ct.toString())) {

                                int type = Geometry.TYPES.POINT;

                                try {

                                        type = ((FLyrVect) getVLE().getLayer()).getShapeType();

                                } catch (ReadException e) {

                                        throw new CommandException(e);

                                }

                                if (ct.isApplicable(type)) {

                                        getCadToolAdapter().setCadTool(ct);

                                        ct.init();

                                        View vista = (View) PluginServices.getMDIManager()

                                                        .getActiveWindow();

                                        vista.getConsolePanel().addText("\n" + ct.getName(),

                                                        JConsole.COMMAND);

                                        String question = ct.getQuestion();

                                        vista.getConsolePanel().addText(

                                                        "\n" + "#" + question + " > ", JConsole.MESSAGE);

                                        return true;

                                }

                                throw new CommandException(name);

                        }

                }

                return false;

        }

        public boolean isApplicable(int shapeType) {

                return true;

        }

        public abstract String toString();

        public void throwValueException(String s, double d) {

                View vista = (View) PluginServices.getMDIManager().getActiveWindow();

                vista.getConsolePanel().addText(s + " : " + d, JConsole.ERROR);

        }

        public void throwOptionException(String s, String o) {

                View vista = (View) PluginServices.getMDIManager().getActiveWindow();

                vista.getConsolePanel().addText(s + " : " + o, JConsole.ERROR);

        }

        public void throwPointException(String s, double x, double y) {

                View vista = (View) PluginServices.getMDIManager().getActiveWindow();

                vista.getConsolePanel().addText(s + " : " + " X = " + x + ", Y = " + y,

                                JConsole.ERROR);

        }

        public void setPreviosTool(DefaultCADTool tool) {

                previousTool = tool;

        }

        public void restorePreviousTool() {

                CADExtension.setCADTool(previousTool.toString(), true);

                PluginServices.getMainFrame().setSelectedTool(previousTool.toString());

        }

        public void endTransition(double x, double y, MouseEvent e) {

                // TODO Auto-generated method stub

        }

}