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

import java.awt.Graphics;

import java.awt.Graphics2D;

import java.awt.event.InputEvent;

import java.awt.geom.Point2D;

import java.util.ArrayList;

import java.util.Iterator;

import javax.swing.JOptionPane;

import org.gvsig.fmap.data.exceptions.DataException;

import org.gvsig.fmap.data.exceptions.ReadException;

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

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

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

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.geom.util.UtilFunctions;

import org.gvsig.fmap.mapcontext.ViewPort;

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

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

/**

 * DOCUMENT ME!

 *

 * @author Vicente Caballero Navarro

 */

public class JoinCADTool extends DefaultCADTool {

        private JoinCADToolContext _fsm;

        private SelectionCADTool selectionCADTool;

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

        /**

     * Crea un nuevo JoinCADTool.

     */

    public JoinCADTool() {

    }

    /**

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

        try {

                        clearSelection();

                } catch (ReadException e) {

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

                }

        selectionCADTool=new SelectionCADTool();//(SelectionCADTool) CADExtension.getCADTool();

        selectionCADTool.init();

        selectionCADTool.multipleSelection(true);

    }

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

            if (!CADExtension.getCADTool().getClass().getName().equals("com.iver.cit.gvsig.gui.cad.tools.SelectionCADTool")) {

            CADExtension.setCADTool("_selection",false);

            ((SelectionCADTool)CADExtension.getCADTool()).multipleSelection(true);

            ((SelectionCADTool) CADExtension.getCADTool()).setNextTool(

                "_join");

        }

    }

    /**

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

                selectionCADTool.selectFeatures(x,y,event);

    }

        /**

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

             VectorialLayerEdited vle=getVLE();

             Iterator iterator=null;

                try {

                        iterator = ((FeatureSelection)vle.getFeatureStore().getSelection()).iterator();

                } catch (ReadException e1) {

                        // TODO Auto-generated catch block

                        e1.printStackTrace();

                } catch (DataException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

//         ArrayList selectedRows=vle.getSelectedRow();

         ViewPort vp=CADExtension.getEditionManager().getMapControl().getViewPort();

//         DefaultRowEdited[] dres=(DefaultRowEdited[])selectedRows.toArray(new DefaultRowEdited[0]);

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

        while (iterator.hasNext()) {

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

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

                        DrawOperationContext doc=new DrawOperationContext();

                        doc.setGraphics((Graphics2D)g);

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

                        doc.setSymbol(DefaultCADTool.geometrySelectSymbol);

                try {

                                geom.invokeOperation(Draw.CODE,doc);

                        } catch (GeometryOperationNotSupportedException e) {

                                e.printStackTrace();

                        } catch (GeometryOperationException e) {

                                e.printStackTrace();

                        }

//                        geom.draw((Graphics2D)g,vp,DefaultCADTool.geometrySelectSymbol);

         }

    }

  /**

     * Add a diferent option.

     *

     * @param s Diferent option.

     */

    public void addOption(String s) {

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

                    init();

            }else {

                    try {

                                joinGeometries();

                        } catch (ReadException e) {

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

                        } catch (DataException e) {

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

                        }

            }

    }

    private Handler[] getHandlers() throws ReadException {

            Iterator iterator = null;

                try {

                        iterator = ((FeatureSelection)getVLE().getFeatureStore().getSelection()).iterator();

                } catch (DataException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

            ArrayList handlers=new ArrayList();

            while (iterator.hasNext()) {

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

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

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

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

                            handlers.add(hs[j]);

                    }

            }

            return (Handler[])handlers.toArray(new Handler[0]);

    }

    private boolean isIntoRect(Point2D p,Handler[] handlers) {

            double tol=0.02;

            double angle=Double.NEGATIVE_INFINITY;

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

            //if (handlers.length>0) {

            Point2D p0=p;

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

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

                            if (!handlers[0].equalsPoint(handlers[j])) {

                                    double angleAux=UtilFunctions.getAngle(p0,p1Aux);

                                    if (angle!=Double.NEGATIVE_INFINITY && (angle>angleAux+tol || angle<angleAux-tol)) {

                                            return false;

                                    }

                                    angle=angleAux;

                            }

                    }

            //}

            return true;

    }

private Point2D[] startAndEndPoints(Handler[] handlers) {

        Point2D first=null;

        Point2D end=null;

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

                Point2D aux=(Point2D)handlers[i].getPoint().clone();

                if (first == null || aux.getX()<first.getX()) {

                        first=aux;

                }

                if (end == null || aux.getX()>end.getX()) {

                        end=aux;

                }

        }

        if (first.getX()==end.getX()) {

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

                        Point2D aux=(Point2D)handlers[i].getPoint().clone();

                        if (first == null || aux.getY()<first.getY()) {

                                first=aux;

                        }

                        if (end == null || aux.getY()>end.getY()) {

                                end=aux;

                        }

                }

        }

        return new Point2D[] {first,end};

}

    private void joinGeometries() throws DataException{

//            DefaultRowEdited[] rows = (DefaultRowEdited[]) getSelectedRows()

//                                .toArray(new DefaultRowEdited[0]);

                Handler[] handlers = getHandlers();

                if (handlers.length < 2) {

                        return;

                }

                Point2D[] points = startAndEndPoints(handlers);

                if (!isIntoRect(points[0], handlers)){

                        JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),PluginServices.getText(this,"no_se_pueden_unir_las_geometrias_al_no_estar_en_la_misma_linea"));

                        return;

                }

                VectorialLayerEdited vle = getVLE();

//                VectorialEditableAdapter vea = vle.getVEA();

                FeatureStore featureStore=vle.getFeatureStore();

                featureStore.beginEditingGroup(getName());

//                String newFID = vea.getNewFID();

                GeneralPathX gpx = new GeneralPathX();

                gpx.moveTo(points[0].getX(), points[0].getY());

                gpx.lineTo(points[1].getX(), points[1].getY());

                Geometry geom = geomFactory.createPolyline2D(gpx);

                EditableFeature eFeature=featureStore.createNewFeature();

                eFeature.setGeometry(featureStore.getDefaultFeatureType().getDefaultGeometryAttributeName(), geom);

//                DefaultFeature df1 = new DefaultFeature(geom, null, newFID);

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

                while (iterator.hasNext()) {

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

//                }

//                for (int i = rows.length - 1; i >= 0; i--) {

                        featureStore.delete(feature);

                }

                featureStore.insert(eFeature);

//                int index1 = vea.addRow(df1, PluginServices.getText(this, "parte1"),

//                                EditionEvent.GRAPHIC);

                //clearSelection();

                ArrayList selectedRowAux = new ArrayList();

                selectedRowAux.add(eFeature);

//                vle.setSelectionCache(VectorialLayerEdited.NOTSAVEPREVIOUS, selectedRowAux);

                featureStore.endEditingGroup();

        }

        /*

         * (non-Javadoc)

         *

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

         */

    public void addValue(double d) {

    }

    public String getName() {

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

        }

        public String toString() {

                return "_join";

        }

        public boolean isApplicable(int shapeType) {

                switch (shapeType) {

                case Geometry.TYPES.GEOMETRY:

                case Geometry.TYPES.CURVE:

                        return true;

                }

                return false;

        }

}