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

import java.awt.Graphics2D;

import java.awt.Image;

import java.awt.event.InputEvent;

import java.awt.geom.AffineTransform;

import java.awt.geom.Point2D;

import java.awt.geom.Point2D.Double;

import java.awt.image.ImagingOpException;

import java.util.ArrayList;

import java.util.Iterator;

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

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

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

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

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

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

import org.gvsig.fmap.geom.Geometry;

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

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

import com.iver.cit.gvsig.gui.cad.tools.smc.ScaleCADToolContext.ScaleCADToolState;

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

/**

 * DOCUMENT ME!

 *

 * @author Vicente Caballero Navarro

 */

public class ScaleCADTool extends DefaultCADTool {

    private ScaleCADToolContext _fsm;

    private Point2D firstPoint;

    private Point2D lastPoint;

        private Point2D scalePoint;

        private Double orr;

        private Double frr;

        private Double ore;

        private Double fre;

    /**

     * Crea un nuevo PolylineCADTool.

     */

    public ScaleCADTool() {

    }

    /**

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

            } catch (ReadException e) {

                    // TODO Auto-generated catch block

                    e.printStackTrace();

            }

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

            CADExtension.setCADTool("_selection",false);

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

                "_scale");

        }

    }

    /**

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

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

        String status = actualState.getName();

        if (status.equals("Scale.PointMain")) {

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

                            scalePoint = firstPoint;

                } else if (status.equals("Scale.ScaleFactorOrReference")) {

                        PluginServices.getMDIManager().setWaitCursor();

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

                        //double w;

                        //double h;

                        //w = lastPoint.getX() - firstPoint.getX();

                        //h = lastPoint.getY() - firstPoint.getY();

                        try {

                                double size=getCadToolAdapter().getMapControl().getViewPort().toMapDistance(getCadToolAdapter().getMapControl().getWidth());

                                scale(firstPoint.distance(lastPoint)/(size/40));

                        } catch (ReadException e) {

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

                        }

                        PluginServices.getMDIManager().restoreCursor();

                } else if (status.equals("Scale.PointOriginOrScaleFactor")) {

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

                } else if (status.equals("Scale.EndPointReference")) {

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

                } else if (status.equals("Scale.OriginPointScale")) {

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

                        firstPoint = ore;

                } else if (status.equals("Scale.EndPointScale")) {

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

                        double distrr = orr.distance(frr);

                        double distre = ore.distance(fre);

                        double escalado = distre / distrr;

                        try {

                                scale(escalado);

                        } catch (ReadException e) {

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

                        }

                }

    }

    /**

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

                ScaleCADToolState actualState = _fsm.getState();

                String status = actualState.getName();

                FeatureSet selection=null;

                try {

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

//                ArrayList selectedRow = getSelectedRows();

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

                long selectionCount=selection.getSize();

                if (status.equals("Scale.ScaleFactorOrReference")) {

                        VectorialLayerEdited vle = getVLE();

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

                        Point2D point = vp.fromMapPoint(firstPoint.getX(), firstPoint

                                        .getY());

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

                                        .toMapDistance(

                                                        getCadToolAdapter().getMapControl().getWidth());

                        double scale = firstPoint.distance(currentPoint) / (size / 40);

                        drawLine((Graphics2D) g, firstPoint, currentPoint,DefaultCADTool.axisReferencesSymbol);

                        if (selectionCount < CADTool.TOPGEOMETRY) {

                                Iterator iterator=selection.iterator();

                                while (iterator.hasNext()) {

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

//                                }

//                                for (int i = 0; i < selectionCount; i++) {

//                                        DefaultFeature fea = (DefaultFeature) ((DefaultRowEdited) selectedRow

//                                                        .get(i)).getLinkedRow();

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

                                        UtilFunctions.scaleGeom(geometry, firstPoint, scale, scale);

                                        DrawOperationContext doc=new DrawOperationContext();

                                    doc.setGraphics((Graphics2D)g);

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

                                    doc.setSymbol(DefaultCADTool.axisReferencesSymbol);

                            try {

                                                geometry.invokeOperation(Draw.CODE,doc);

                                        } catch (GeometryOperationNotSupportedException e) {

                                                e.printStackTrace();

                                        } catch (GeometryOperationException e) {

                                                e.printStackTrace();

                                        }

//                                        geometry.draw((Graphics2D) g, getCadToolAdapter()

//                                                        .getMapControl().getViewPort(),

//                                                        DefaultCADTool.axisReferencesSymbol);

                                }

                        } else {

                                AffineTransform at = new AffineTransform();

                                at.setToTranslation(point.getX(), point.getY());

                                at.scale(scale, scale);

                                at.translate(-point.getX(), -point.getY());

                                Image imgSel = vle.getSelectionImage();

                                try {

                                        ((Graphics2D) g).drawImage(imgSel, at, null);

                                } catch (ImagingOpException e) {

                                }

                        }

                        PluginServices.getMainFrame().getStatusBar().setMessage("5",

                                        "Factor = " + scale);

                } else if (status.equals("Scale.EndPointScale")) {

                        VectorialLayerEdited vle = getVLE();

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

                        Point2D point = vp.fromMapPoint(scalePoint.getX(), scalePoint

                                        .getY());

                        double distrr = orr.distance(frr);

                        double distre = ore.distance(currentPoint);

                        double escalado = distre / distrr;

                        if (selectionCount < CADTool.TOPGEOMETRY) {

                                Iterator iterator=selection.iterator();

                                while (iterator.hasNext()) {

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

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

//                                                DefaultFeature fea = (DefaultFeature) ((DefaultRowEdited) selectedRow

//                                                                .get(i)).getLinkedRow();

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

                                                UtilFunctions.scaleGeom(geometry, scalePoint, escalado,

                                                                escalado);

                                                DrawOperationContext doc=new DrawOperationContext();

                                            doc.setGraphics((Graphics2D)g);

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

                                            doc.setSymbol(DefaultCADTool.axisReferencesSymbol);

                                    try {

                                                        geometry.invokeOperation(Draw.CODE,doc);

                                                } catch (GeometryOperationNotSupportedException e) {

                                                        e.printStackTrace();

                                                } catch (GeometryOperationException e) {

                                                        e.printStackTrace();

                                                }

//                                                geometry.draw((Graphics2D) g, getCadToolAdapter()

//                                                                .getMapControl().getViewPort(),

//                                                                DefaultCADTool.axisReferencesSymbol);

                                        }

                                } else {

                                        AffineTransform at = new AffineTransform();

                                        at.setToTranslation(point.getX(), point.getY());

                                        at.scale(escalado, escalado);

                                        at.translate(-point.getX(), -point.getY());

                                        Image imgSel = vle.getSelectionImage();

                                        try {

                                                ((Graphics2D) g).drawImage(imgSel, at, null);

                                        } catch (ImagingOpException e) {

                                        }

                                }

                                drawLine((Graphics2D) g, firstPoint, new Point2D.Double(x, y),DefaultCADTool.axisReferencesSymbol);

                        }

                } catch (ReadException e1) {

                        // TODO Auto-generated catch block

                        e1.printStackTrace();

                } catch (DataException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

        }

    /**

         * Add a diferent option.

         *

         * @param s

         *            Diferent option.

         */

    public void addOption(String s) {

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

            String status = actualState.getName();

            if (status.equals("Scale.ScaleFactorOrReference")) {

                    /*        try {

                                scale(2);

                        } catch (DriverIOException e) {

                                // TODO Auto-generated catch block

                                e.printStackTrace();

                        } catch (IOException e) {

                                // TODO Auto-generated catch block

                                e.printStackTrace();

                        }

                     */

            }

    }

    /* (non-Javadoc)

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

     */

    public void addValue(double d) {

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

            String status = actualState.getName();

            if (status.equals("Scale.ScaleFactorOrReference")) {

                    try {

                            scale(d);

                    } catch (ReadException e) {

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

                    }

            }

    }

    private void scale(double scaleFactor) throws ReadException{

            VectorialLayerEdited vle=getVLE();

            FeatureStore featureStore=vle.getFeatureStore();

            featureStore.getCommandsRecord().startComplex();

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

            ArrayList selectedRowAux=new ArrayList();

            while (iterator.hasNext()) {

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

//                }

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

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

//                    DefaultFeature fea = (DefaultFeature) edRow.getLinkedRow().cloneRow();

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

                        UtilFunctions.scaleGeom(geometry, scalePoint, scaleFactor, scaleFactor);

                    try {

                                feature.editing();

                                feature.setGeometry(featureStore.getDefaultFeatureType().getDefaultGeometryAttributeName(), geometry);

                                featureStore.update(feature);

                    } catch (DataException e) {

                                // TODO Auto-generated catch block

                                e.printStackTrace();

                        }

//                        vea.modifyRow(edRow.getIndex(), fea,getName(),EditionEvent.GRAPHIC);

                    selectedRowAux.add(feature);

            }

            featureStore.getCommandsRecord().endComplex(getName());

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

            //clearSelection();

            //selectedRow.addAll(selectedRowAux);

    }

    public String getName() {

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

        }

        public String toString() {

                return "_scale";

        }

}