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

import java.awt.Graphics;

import java.awt.event.InputEvent;

import java.awt.geom.PathIterator;

import java.awt.geom.Point2D;

import java.util.ArrayList;

import java.util.Iterator;

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.primitive.GeneralPathX;

import org.gvsig.fmap.geom.util.Converter;

import org.gvsig.fmap.geom.util.UtilFunctions;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

import com.iver.cit.gvsig.CADExtension;

import com.iver.cit.gvsig.gui.cad.DefaultCADTool;

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

import com.iver.cit.gvsig.gui.cad.tools.smc.ExtendCADToolContext;

import com.iver.cit.gvsig.gui.cad.tools.smc.ExtendCADToolContext.ExtendCADToolState;

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

/**

 * DOCUMENT ME!

 *

 * @author Vicente Caballero Navarro

 */

public class ExtendCADTool extends DefaultCADTool {

        private ExtendCADToolContext _fsm;

        private GeometryFactory geomFactory = GeometryManager.getInstance()

                        .getGeometryFactory();

        /**

         * Crea un nuevo ExtendCADTool.

         */

        public ExtendCADTool() {

        }

        /**

         * M?todo de incio, para poner el c?digo de todo lo que se requiera de una

         * carga previa a la utilizaci?n de la herramienta.

         */

        public void init() {

                _fsm = new ExtendCADToolContext(this);

        }

        /*

         * (non-Javadoc)

         *

         * @see com.iver.cit.gvsig.gui.cad.CADTool#transition(com.iver.cit.gvsig.fmap.layers.FBitSet,

         *      double, double)

         */

        public void transition(double x, double y, InputEvent event) {

                _fsm.addPoint(x, y, event);

        }

        /*

         * (non-Javadoc)

         *

         * @see com.iver.cit.gvsig.gui.cad.CADTool#transition(com.iver.cit.gvsig.fmap.layers.FBitSet,

         *      double)

         */

        public void transition(double d) {

                _fsm.addValue(d);

        }

        /*

         * (non-Javadoc)

         *

         * @see com.iver.cit.gvsig.gui.cad.CADTool#transition(com.iver.cit.gvsig.fmap.layers.FBitSet,

         *      java.lang.String)

         */

        public void transition(String s) throws CommandException {

                if (!super.changeCommand(s)) {

                        _fsm.addOption(s);

                }

        }

        /**

         * DOCUMENT ME!

         */

        public void selection() {

                FeatureSet selection = null;

                try {

                        selection = (FeatureSet) getVLE().getFeatureStore().getSelection();

                        if (selection.getSize() == 0

                                        && !CADExtension

                                                        .getCADTool()

                                                        .getClass()

                                                        .getName()

                                                        .equals(

                                                                        "com.iver.cit.gvsig.gui.cad.tools.SelectionCADTool")) {

                                CADExtension.setCADTool("_selection", false);

                                ((SelectionCADTool) CADExtension.getCADTool())

                                                .setNextTool("_extend");

                        }

                } catch (ReadException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                } catch (DataException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

        }

        /**

         * Equivale al transition del prototipo pero sin pasarle como par?metro el

         * editableFeatureSource que ya estar? creado.

         *

         * @param x

         *            par?metro x del punto que se pase en esta transici?n.

         * @param y

         *            par?metro y del punto que se pase en esta transici?n.

         */

        public void addPoint(double x, double y, InputEvent event) {

                ExtendCADToolState actualState = (ExtendCADToolState) _fsm

                                .getPreviousState();

                String status = actualState.getName();

                if (status.equals("Extend.SelectGeometryToExtend")) {

                        VectorialLayerEdited vle = getVLE();

                        FeatureStore featureStore = null;

                        try {

                                featureStore = vle.getFeatureStore();

                                featureStore.beginEditingGroup(getName());

                                Iterator iterator = null;

                                iterator = ((FeatureSelection) featureStore.getSelection())

                                                .iterator();

                                // ArrayList selectedRow=getSelectedRows();

                                ArrayList selectedRowAux = new ArrayList();

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

                                // selectedRowAux.addAll(selectedRow);

                                // }

                                // selection();

                                vle.selectWithPoint(x, y, false);

                                // ArrayList newSelectedRow=new ArrayList();

                                while (iterator.hasNext()) {

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

                                        // }

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

                                        // IRowEdited edRow1 = (IRowEdited) selectedRowAux.get(i);

                                        // DefaultFeature fea1 = (DefaultFeature)

                                        // edRow1.getLinkedRow().cloneRow();

                                        Geometry geometry1 = ((Geometry) feature

                                                        .getDefaultGeometry()).cloneGeometry();

                                        // IRowEdited edRow2 = (IRowEdited) newSelectedRow.get(i);

                                        // DefaultFeature fea2 = (DefaultFeature)

                                        // edRow2.getLinkedRow().cloneRow();

                                        Geometry geometry2 = ((Geometry) feature

                                                        .getDefaultGeometry()).cloneGeometry();

                                        // for (int j=0;j<newSelectedRow.size();j++) {

                                        // if (geometry1 instanceof FPolygon2D) {

                                        EditableFeature eFeature = feature.getEditable();

                                        eFeature.setGeometry(featureStore.getDefaultFeatureType()

                                                        .getDefaultGeometryAttributeName(),

                                                        intersectsGeometry(geometry2, geometry1));

                                        // fea2.setGeometry(intersectsGeometry(geometry2,geometry1));

                                        // }

                                        // }

                                        featureStore.update(eFeature);

                                        // vea.modifyRow(edRow2.getIndex(),fea2,getName(),EditionEvent.GRAPHIC);

                                        clearSelection();

                                        // newSelectedRow.add(new

                                        // DefaultRowEdited(fea2,IRowEdited.STATUS_MODIFIED,edRow2.getIndex()));

                                }

                                featureStore.endEditingGroup();

                                // vle.setSelectionCache(VectorialLayerEdited.NOTSAVEPREVIOUS,

                                // newSelectedRow);

                        } catch (ReadException e) {

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

                        } catch (DataException e) {

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

                        }

                }

        }

        private Geometry intersectsGeometry(Geometry geometry1, Geometry geometry2) {

                Point2D p3 = null;

                Point2D p4 = null;

                Point2D p1 = null;

                Point2D p2 = null;

                GeneralPathX gpx = new GeneralPathX();

                PathIterator theIterator = geometry1.getInternalShape()

                                .getPathIterator(null, Converter.FLATNESS);

                boolean first = true;

                double[] theData = new double[6];

                int theType;

                while (!theIterator.isDone()) {

                        theType = theIterator.currentSegment(theData);

                        switch (theType) {

                        case PathIterator.SEG_MOVETO:

                                p1 = new Point2D.Double(theData[0], theData[1]);

                                gpx.moveTo(p1.getX(), p1.getY());

                                break;

                        case PathIterator.SEG_LINETO:

                                p2 = new Point2D.Double(theData[0], theData[1]);

                                ArrayList lines = getLines(geometry2);

                                boolean isLineTo = true;

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

                                        Point2D[] ps1 = (Point2D[]) lines.get(i);

                                        Point2D p = UtilFunctions.getIntersection(ps1[0], ps1[1],

                                                        p1, p2);

                                        // GeneralPathX gpxAux=new GeneralPathX();

                                        // gpxAux.moveTo(p.getX(),p.getY());

                                        // gpxAux.lineTo(ps1[0].getX(),ps1[0].getY());

                                        // Geometry

                                        // gjts1=FConverter.java2d_to_jts((FShape)ShapeFactory.createPolyline2D(gpxAux).getInternalShape());

                                        // Geometry

                                        // gjts2=FConverter.java2d_to_jts((FShape)geometry2.getInternalShape());

                                        // GeometryCollection

                                        // result=(GeometryCollection)gjts1.intersection(gjts2);

                                        // Point point=(Point)result.getGeometryN(1);

                                        // p=new Point2D.Double(point.getX(),point.getY());

                                        if (p != null && first) {

                                                gpx.lineTo(p.getX(), p.getY());

                                                first = false;

                                                isLineTo = false;

                                                break;

                                        } else {

                                                // gpx.lineTo(p2.getX(),p2.getY());

                                        }

                                }

                                if (!first && isLineTo)

                                        gpx.lineTo(p2.getX(), p2.getY());

                                break;

                        } // end switch

                        theIterator.next();

                } // end while loop

                return geomFactory.createPolyline2D(gpx);

        }

        private ArrayList getLines(Geometry geometry1) {

                Point2D p1 = null;

                Point2D p2 = null;

                ArrayList lines = new ArrayList();

                PathIterator theIterator = geometry1.getInternalShape()

                                .getPathIterator(null, Converter.FLATNESS);

                double[] theData = new double[6];

                int theType;

                while (!theIterator.isDone()) {

                        theType = theIterator.currentSegment(theData);

                        switch (theType) {

                        case PathIterator.SEG_MOVETO:

                                p1 = new Point2D.Double(theData[0], theData[1]);

                                break;

                        case PathIterator.SEG_LINETO:

                                p2 = new Point2D.Double(theData[0], theData[1]);

                                lines.add(new Point2D[] { p1, p2 });

                                break;

                        } // end switch

                        theIterator.next();

                } // end while loop

                return lines;

        }

        /**

         * M?todo para dibujar la lo necesario para el estado en el que nos

         * encontremos.

         *

         * @param g

         *            Graphics sobre el que dibujar.

         * @param x

         *            par?metro x del punto que se pase para dibujar.

         * @param y

         *            par?metro x del punto que se pase para dibujar.

         */

        public void drawOperation(Graphics g, double x, double y) {

        }

        /**

         * Add a diferent option.

         *

         * @param s

         *            Diferent option.

         */

        public void addOption(String s) {

        }

        /*

         * (non-Javadoc)

         *

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

         */

        public void addValue(double d) {

        }

        public String getName() {

                return PluginServices.getText(this, "extend_");

        }

        public String toString() {

                return "_extend";

        }

        public boolean isApplicable(int shapeType) {

                switch (shapeType) {

                case Geometry.TYPES.CURVE:

                case Geometry.TYPES.GEOMETRY:

                        return true;

                }

                return false;

        }

}