| gvSIG desktop 1

/* 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.Color;

import java.awt.Graphics;

import java.awt.Graphics2D;

import java.awt.Image;

import java.awt.event.InputEvent;

import java.awt.geom.Point2D;

import java.util.ArrayList;

import java.util.Iterator;

import org.gvsig.fmap.data.ReadException;

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

import org.gvsig.fmap.data.feature.FeatureCollection;

import org.gvsig.fmap.data.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.GeneralPathX;

import org.gvsig.fmap.mapcontext.MapContext;

import org.gvsig.fmap.mapcontext.ViewPort;

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

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

import org.gvsig.fmap.mapcontrol.MapControl;

import com.iver.andami.PluginServices;

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

import com.iver.cit.gvsig.gui.cad.tools.smc.SelectionCADToolContext.SelectionCADToolState;

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

import com.iver.cit.gvsig.project.documents.table.gui.Table;

/**

 * DOCUMENT ME!

 *

 * @author Vicente Caballero Navarro

 */

public class SelectionCADTool extends DefaultCADTool {

//        public static int tolerance = 4;

        private SelectionCADToolContext _fsm;

        protected Point2D firstPoint;

        protected GeometryFactory geomFactory=GeometryManager.getInstance().getGeometryFactory();

        protected String nextState;

// Registros de los que se ha seleccionado alg?n handler.

        protected ArrayList rowselectedHandlers=new ArrayList();

        protected String type=PluginServices.getText(this,"simple");

//        protected ArrayList pointsPolygon=new ArrayList();

        protected boolean multipleSelection=false;

        /**

         * Crea un nuevo SelectionCADTool.

         */

        public SelectionCADTool() {

        }

        /**

         * 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 SelectionCADToolContext(this);

                setNextTool("selection");

                setType(PluginServices.getText(this,"simple"));

        }

        /*

         * (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) {

                System.out.println("TRANSICION DESDE ESTADO " + _fsm.getState()

                                + " x= " + x + " y=" + y);

                try{

                _fsm.addPoint(x, y, event);

                }catch (Exception e) {

                        init();

                }

                System.out.println("ESTADO ACTUAL: " + getStatus());

                // ESTO LO QUITO POR AHORA, PERO PUEDE QUE LO NECESITEMOS VOLVER A PONER.

                // Lo he quitado porque cuando seleccionas algo con CAD, molesta que

                // te hagan un redibujado.

                FLyrVect lv=(FLyrVect)((VectorialLayerEdited)CADExtension.getEditionManager().getActiveLayerEdited()).getLayer();

                //lv.getSource().getRecordset().getSelectionSupport().fireSelectionEvents();

                com.iver.andami.ui.mdiManager.IWindow[] views = (com.iver.andami.ui.mdiManager.IWindow[]) PluginServices.getMDIManager().getAllWindows();

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

                        if (views[i] instanceof Table){

                                Table table=(Table)views[i];

                                if (table.getModel().getAssociatedTable()!=null && table.getModel().getAssociatedTable().equals(lv))

                                        table.updateSelection();

                        }

                }

        }

        /*

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

            }

        }

        public String getNextState() {

                return nextState;

        }

        protected void pointDoubleClick(MapContext map) throws ReadException {

                FLayer[] actives = map

           .getLayers().getActives();

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

//            if (actives[i] instanceof FLyrAnnotation && actives[i].isEditing()) {

//                FLyrAnnotation lyrAnnotation = (FLyrAnnotation) actives[i];

//                       lyrAnnotation.setSelectedEditing();

//                       lyrAnnotation.setInEdition(lyrAnnotation.getRecordset().getSelection().nextSetBit(0));

//                       FLabel fl=lyrAnnotation.getLabel(lyrAnnotation.getInEdition());

//                       if (fl!=null){

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

//                                       TextFieldEdit tfe=new TextFieldEdit(lyrAnnotation);

//                                tfe.show(vista.getMapControl().getViewPort().fromMapPoint(fl.getOrig()),vista.getMapControl());

//                               }

//            }

//        }

}

        /**

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

         * editableFeatureSource que ya estar? creado.

         *

         * @param selection

         *            Bitset con las geometr?as que est?n seleccionadas.

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

//                if (event!=null && ((MouseEvent)event).getClickCount()==2){

//                        try {

//                                pointDoubleClick((MapControl)event.getComponent());

//                        } catch (ReadException e) {

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

//                        }

//                        return;

//                }

                SelectionCADToolState actualState = (SelectionCADToolState) _fsm

                                .getPreviousState();

                String status = actualState.getName();

                System.out.println("PREVIOUSSTATE =" + status); // + "ESTADO ACTUAL: " +

                                                                                                                // _fsm.getState());

                VectorialLayerEdited vle = getVLE();

                FeatureStore featureStore=null;

                try {

                        featureStore = vle.getFeatureStore();

                } catch (ReadException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                ArrayList selectedHandler = vle.getSelectedHandler();

                FeatureCollection selection = (FeatureCollection)featureStore.getSelection();//vle.getSelectedRow();

                System.out.println("STATUS ACTUAL = " + _fsm.getTransition());

                if (status.equals("Selection.FirstPoint")) {

                        firstPoint=new Point2D.Double(x,y);

//                        pointsPolygon.add(firstPoint);

                } else if (status.equals("Selection.SecondPoint")) {

                } else if (status.equals("Selection.WithFeatures")) {

                } else if (status.equals("Selection.WithHandlers")) {

                        featureStore.getCommandsRecord().startComplex();

                        ArrayList selectedRowsAux=new ArrayList();

                        Iterator iterator=selection.iterator();

                        while (iterator.hasNext()) {

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

//                        }

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

//                                IRowEdited row = (IRowEdited) selectedRow.get(i);

//                                IFeature feat = (IFeature) row.getLinkedRow().cloneRow();

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

//                                if (vea instanceof AnnotationEditableAdapter) {

//                                        // Movemos la geometr?a

//                    UtilFunctions.moveGeom(ig, x -

//                            firstPoint.getX(), y - firstPoint.getY());

//                                }else {

                                        // Movemos los handlers que hemos seleccionado

                                        // previamente dentro del m?todo select()

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

                                        for (int k = 0; k < selectedHandler.size(); k++) {

                                                Handler h = (Handler)selectedHandler.get(k);

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

                                                        if (h.getPoint().equals(handlers[j].getPoint()))

                                                                handlers[j].set(x,y);

                                                }

                                        }

//                                }

                                modifyFeature(feature, feature);

//                                selectedRowsAux.add(new DefaultRowEdited(feat,IRowEdited.STATUS_MODIFIED,row.getIndex()));

                        }

                        firstPoint=new Point2D.Double(x,y);

//                        vle.setSelectionCache(VectorialLayerEdited.SAVEPREVIOUS, selectedRowsAux);

                        //clearSelection();

                        //selectedRow.addAll(selectedRowsAux);

                        String description=PluginServices.getText(this,"move_handlers");

                        featureStore.getCommandsRecord().endComplex(description);

                }

        }

        /**

         * Receives second point

         * @param x

         * @param y

         * @return numFeatures selected

         */

        public int selectWithSecondPoint(double x, double y, InputEvent event) {

                VectorialLayerEdited vle = getVLE();

                PluginServices.getMDIManager().setWaitCursor();

                vle.selectWithSecondPoint(x,y);

                FeatureCollection selection=null;

                try {

                        selection = (FeatureCollection)vle.getFeatureStore().getSelection();

                } catch (ReadException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }//getSelectedRow();

                PluginServices.getMDIManager().restoreCursor();

                int countSel=selection.size();

                if (countSel > 0) {

                        nextState = "Selection.WithSelectedFeatures";

                } else

                        nextState = "Selection.FirstPoint";

                return countSel;

        }

        /**

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

         * encontremos.

         *

         * @param g

         *            Graphics sobre el que dibujar.

         * @param selectedGeometries

         *            BitSet con las geometr?as seleccionadas.

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

                SelectionCADToolState actualState = _fsm.getState();

                String status = actualState.getName();

                VectorialLayerEdited vle = getVLE();

                if (vle == null) return;

                ArrayList selectedHandler = vle.getSelectedHandler();

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

                if (status.equals("Selection.SecondPoint")) {

                        // Dibuja el rect?ngulo de selecci?n

                        GeneralPathX elShape = new GeneralPathX(GeneralPathX.WIND_EVEN_ODD,

                                        4);

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

                        elShape.lineTo(x, firstPoint.getY());

                        elShape.lineTo(x, y);

                        elShape.lineTo(firstPoint.getX(), y);

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

                        DrawOperationContext doc=new DrawOperationContext();

                        doc.setGraphics((Graphics2D)g);

                        doc.setViewPort(vp);

                        doc.setSymbol(DefaultCADTool.geometrySelectSymbol);

                try {

                        geomFactory.createPolyline2D(elShape).cloneGeometry().invokeOperation(Draw.CODE,doc);

                        } catch (GeometryOperationNotSupportedException e) {

                                e.printStackTrace();

                        } catch (GeometryOperationException e) {

                                e.printStackTrace();

                        }

//                        geomFactory.createPolyline2D(elShape).draw((Graphics2D) g,

//                                        vp,

//                                        DefaultCADTool.geometrySelectSymbol);

                        Image img = vle.getSelectionImage();

                g.drawImage(img, 0, 0, null);

                return;

                }else if (status.equals("Selection.WithHandlers")) {

                        // Movemos los handlers que hemos seleccionado

                        // previamente dentro del m?todo select()

                        double xPrev=0;

                        double yPrev=0;

                        for (int k = 0; k < selectedHandler.size(); k++) {

                                Handler h = (Handler)selectedHandler.get(k);

                                xPrev=h.getPoint().getX();

                                yPrev=h.getPoint().getY();

                                h.set(x, y);

                        }

                        // Y una vez movidos los v?rtices (handles)

                        // redibujamos la nueva geometr?a.

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

                                Feature feature=(Feature)rowselectedHandlers.get(i);

//                                IRowEdited rowEd = (IRowEdited) rowselectedHandlers.get(i);

//                                IGeometry geom = ((IFeature) rowEd.getLinkedRow())

//                                                .getGeometry().cloneGeometry();

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

                                g.setColor(Color.gray);

                                DrawOperationContext doc=new DrawOperationContext();

                            doc.setGraphics((Graphics2D)g);

                            doc.setViewPort(vp);

                            doc.setSymbol(DefaultCADTool.axisReferencesSymbol);

                    try {

                                        geom.invokeOperation(Draw.CODE,doc);

                                } catch (GeometryOperationNotSupportedException e) {

                                        e.printStackTrace();

                                } catch (GeometryOperationException e) {

                                        e.printStackTrace();

                                }

                        }

                        for (int k = 0; k < selectedHandler.size(); k++) {

                                Handler h = (Handler)selectedHandler.get(k);

                                h.set(xPrev, yPrev);

                        }

                        return;

                }else{

                        if (!vle.getLayer().isVisible())

                                return;

                        try{

                        Image imgSel = vle.getSelectionImage();

                if (imgSel!=null)

                        g.drawImage(imgSel, 0, 0, null);

                Image imgHand = vle.getHandlersImage();

                if (imgHand!=null)

                        g.drawImage(imgHand, 0, 0, null);

                        }catch (Exception e) {

                        }

                }

        }

        /**

         * Add a diferent option.

         *

         * @param sel

         *            DOCUMENT ME!

         * @param s

         *            Diferent option.

         */

        public void addOption(String s) {

                SelectionCADToolState actualState = (SelectionCADToolState) _fsm

                                .getPreviousState();

                String status = actualState.getName();

                System.out.println("PREVIOUSSTATE =" + status); // + "ESTADO ACTUAL: " +

                // _fsm.getState());

                System.out.println("STATUS ACTUAL = " + _fsm.getTransition());

                if (s.equals(PluginServices.getText(this,"cancel"))){

                        init();

                        return;

                }

                if (status.equals("Selection.FirstPoint")) {

                        setType(s);

                        return;

                }

                init();

        }

        /*

         * (non-Javadoc)

         *

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

         */

        public void addValue(double d) {

        }

        public String getStatus() {

                try {

                        SelectionCADToolState actualState = (SelectionCADToolState) _fsm

                                        .getPreviousState();

                        String status = actualState.getName();

                        return status;

                } catch (NullPointerException e) {

                        return "Selection.FirstPoint";

                }

        }

        public void end() {

                if (!getNextTool().equals("selection"))

                        CADExtension.setCADTool(getNextTool(),false);

        }

        public String getName() {

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

        }

        public boolean selectFeatures(double x, double y, InputEvent event) {

                SelectionCADToolState actualState = (SelectionCADToolState) _fsm

                                .getState();

                String status = actualState.getName();

                VectorialLayerEdited vle = getVLE();

                if ((status.equals("Selection.FirstPoint"))

                                || (status.equals("Selection.WithSelectedFeatures"))) {

                        PluginServices.getMDIManager().setWaitCursor();

                        firstPoint = new Point2D.Double(x, y);

                        vle.selectWithPoint(x,y,multipleSelection);

                        PluginServices.getMDIManager().restoreCursor();

                }

                FeatureCollection selection=null;

                try {

                        selection = (FeatureCollection)vle.getFeatureStore().getSelection();

                } catch (ReadException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

//                ArrayList selectedRow = vle.getSelectedRow();

                int countSel=selection.size();

                if (countSel > 0) {

                        nextState = "Selection.WithSelectedFeatures";

                        return true;

                } else {

                        {

                                nextState = "Selection.SecondPoint";

                                return true;

                        }

                }

        }

        public int selectHandlers(double x, double y, InputEvent event) {

                Point2D auxPoint = new Point2D.Double(x, y);

                VectorialLayerEdited vle = getVLE();

                ArrayList selectedHandler = vle.getSelectedHandler();

                FeatureCollection selection=null;

                try {

                        selection = (FeatureCollection)vle.getFeatureStore().getSelection();

                } catch (ReadException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

//                ArrayList selectedRow = vle.getSelectedRow();

                int countSel=selection.size();

                System.out.println("DENTRO DE selectHandlers. selectedRow.size= "

                                + countSel);

                selectedHandler.clear();

                // Se comprueba si se pincha en una gemometr?a

                PluginServices.getMDIManager().setWaitCursor();

                double tam = getCadToolAdapter().getMapControl().getViewPort()

                                .toMapDistance(MapControl.tolerance);

                Handler[] handlers = null;

                rowselectedHandlers.clear();

                Iterator iterator=selection.iterator();

                while (iterator.hasNext()) {

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

//                }

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

//                        IRowEdited rowEd = (IRowEdited) selectedRow.get(i);

//

//                        IFeature fea = (IFeature) rowEd.getLinkedRow();

                        handlers = ((Geometry)feature.getDefaultGeometry()).getHandlers(Geometry.SELECTHANDLER);

                        // y miramos los handlers de cada entidad seleccionada

                        double min = tam;

                        // int hSel = -1;

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

                                Point2D handlerPoint = handlers[j].getPoint();

                                double distance = auxPoint.distance(handlerPoint);

                                if (distance <= min) {

                                        min = distance;

                                        // hSel = j;

                                        selectedHandler.add(handlers[j]);

                                        rowselectedHandlers.add(feature);

                                }

                        }

                }

                PluginServices.getMDIManager().restoreCursor();

                int numHandlesSelected = selectedHandler.size();

                /*

                 * if (numHandlesSelected == 0) selectFeatures(x,y);

                 */

                return numHandlesSelected;

        }

        public String getType() {

                return type;

        }

        public void setType(String type) {

                if (type.equals("S") || type.equals("s")){

                        this.type=PluginServices.getText(this,"simple");

                }else{

                        this.type = type;

                }

//                pointsPolygon.clear();

        }

        public String toString() {

                return "_selection";

        }

        public void multipleSelection(boolean b) {

                multipleSelection=b;

        }

}