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 org.gvsig.editing.gui.cad.tools;

import java.awt.Color;

import java.awt.event.InputEvent;

import java.awt.geom.Point2D;

import java.util.ArrayList;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.messages.NotificationManager;

import org.gvsig.app.project.documents.table.gui.FeatureTableDocumentPanel;

import org.gvsig.editing.CADExtension;

import org.gvsig.editing.gui.cad.DefaultCADTool;

import org.gvsig.editing.gui.cad.exception.CommandException;

import org.gvsig.editing.gui.cad.tools.smc.SelectionCADToolContext;

import org.gvsig.editing.gui.cad.tools.smc.SelectionCADToolContext.SelectionCADToolState;

import org.gvsig.editing.layers.VectorialLayerEdited;

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.dal.feature.exception.NeedEditingModeException;

import org.gvsig.fmap.geom.Geometry;

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

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

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

import org.gvsig.tools.dispose.DisposableIterator;

/**

 * DOCUMENT ME!

 *

 * @author Vicente Caballero Navarro

 */

public class SelectionCADTool extends DefaultCADTool {

    //        public static int tolerance = 4;

    protected SelectionCADToolContext _fsm;

    protected Point2D firstPoint;

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

    }

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

            PluginServices.getMDIManager().restoreCursor();

        }

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

        org.gvsig.andami.ui.mdiManager.IWindow[] views = PluginServices.getMDIManager().getAllWindows();

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

            if (views[i] instanceof FeatureTableDocumentPanel){

                FeatureTableDocumentPanel table=(FeatureTableDocumentPanel)views[i];

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

                    table.updateSelection();

                }

            }

        }

    }

    public void transition(double d) {

        _fsm.addValue(d);

    }

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

    }

    /**

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

        SelectionCADToolState actualState = (SelectionCADToolState) _fsm.getPreviousState();

        String status = actualState.getName();

        System.out.println("PREVIOUSSTATE =" + status);         

        VectorialLayerEdited vle = getVLE();      

        try {            

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

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

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

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

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

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

                addPointWithHandlers(x, y, vle.getFeatureStore(), vle.getSelectedHandler());        

            }

        } catch (DataException e) {

            LOG.error("Error reding the store", e);

        }

    }

    /**

     * AddPoint method for the WithHandlers option

     * @param x

     *          selected x coordinate.

     * @param y

     *          selected y coordinate.

     * @param featureStore

     *          the selected feature store.

     * @param selectedHandlers

     *          the selected handlers

     */

    protected void addPointWithHandlers(double x, double y, FeatureStore featureStore, ArrayList selectedHandlers){

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

        DisposableIterator iterator = null;

        try {

            featureStore.beginEditingGroup(description); 

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

            while (iterator.hasNext()) {

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

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

                // Movemos los handlers que hemos seleccionado

                // previamente dentro del m?todo select()

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

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

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

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

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

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

                        }

                    }

                }

                EditableFeature eFeature = feature.getEditable();

                eFeature.setDefaultGeometry(geometry);

                featureStore.update(eFeature);

            }

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

            featureStore.endEditingGroup();

        } catch (DataException e1) {           

            LOG.error("Error reading the store", e1);

        } finally {

            if (iterator != null) {

                iterator.dispose();

            }

        }

    }

    /**

     * Receives second point

     * @param x

     * @param y

     * @return numFeatures selected

     */

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

        VectorialLayerEdited vle = getVLE();

        PluginServices.getMDIManager().setWaitCursor();

        vle.selectWithSecondPoint(x,y);

        FeatureSet selection=null;

        try {

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

            PluginServices.getMDIManager().restoreCursor();

            long countSel=selection.getSize();

            if (countSel > 0) {

                nextState = "Selection.WithSelectedFeatures";

            } else {

                nextState = "Selection.FirstPoint";

            }

            return countSel;

        } catch (ReadException e) {

            e.printStackTrace();

            return 0;

        } catch (DataException e) {

            e.printStackTrace();

            return 0;

        }

    }

    /**

     * 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(MapControlDrawer renderer, 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

            Curve curve = createCurve();

            curve.addMoveToVertex(createPoint(firstPoint.getX(), firstPoint.getY()));

            curve.addVertex(createPoint(x, firstPoint.getY()));

            curve.addVertex(createPoint(x, y));

            curve.addVertex(createPoint(firstPoint.getX(), y));

            curve.addVertex(createPoint(firstPoint.getX(), firstPoint.getY()));

            renderer.draw(curve, mapControlManager.getGeometrySelectionSymbol());

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

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

                renderer.setColor(Color.gray);

                renderer.draw(geom, mapControlManager.getAxisReferenceSymbol());

            }

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

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

                h.set(xPrev, yPrev);

            }

            return;

        }

    }

    /**

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

        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 = _fsm.getState();

        String status = actualState.getName();

        VectorialLayerEdited vle = getVLE();

        multipleSelection = event.isControlDown();

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

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

            PluginServices.getMDIManager().setWaitCursor();

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

            try {

                vle.getFeatureStore().beginEditingGroup(getName());

                vle.selectWithPoint(x, y, multipleSelection);

                vle.getFeatureStore().endEditingGroup();

            } catch (NeedEditingModeException e) {

                NotificationManager.showMessageError(getName(), e);

            } catch (ReadException e) {

                NotificationManager.showMessageError(getName(), e);

            }

            PluginServices.getMDIManager().restoreCursor();

        }

        FeatureSet selection = null;

        try {

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

            long countSel=selection.getSize();

            if (countSel > 0) {

                nextState = "Selection.WithSelectedFeatures";

                return true;

            } else {

                {

                    nextState = "Selection.SecondPoint";

                    return true;

                }

            }

        } catch (ReadException e) {

            LOG.error("Error selecting the features", e);

            return false;

        } catch (DataException e) {

            LOG.error("Error selecting the features", e);

            return false;

        }

    }

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

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

        VectorialLayerEdited vle = getVLE();

        ArrayList selectedHandler = vle.getSelectedHandler();

        FeatureSet selection=null;

        DisposableIterator iterator = null;

        try {

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

            long countSel = selection.getSize();

            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(mapControlManager.getTolerance());

            Handler[] handlers = null;

            rowselectedHandlers.clear();

            iterator = selection.fastIterator();

            while (iterator.hasNext()) {

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

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

                // y miramos los handlers de cada entidad seleccionada

                double min = tam;

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

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

                    double distance = auxPoint.distance(handlerPoint);

                    if (distance <= min) {

                        min = distance;                                                

                        selectedHandler.add(handlers[j]);

                        rowselectedHandlers.add(feature);

                    }

                }

            }

            PluginServices.getMDIManager().restoreCursor();

        } catch (DataException e) {

            LOG.error("Error reading the store", e);

        } finally {

            if (iterator != null) {

                iterator.dispose();

            }

        }

        return selectedHandler.size();

    }

    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;

        }

    }

    public String toString() {

        return "_selection";

    }

    public void multipleSelection(boolean b) {

        multipleSelection=b;

    }

}