Application: gvSIG desktop » gvSIG tools: Vector editing

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2014 gvSIG Association

 *

 * 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., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.vectorediting.lib.prov.rotate;

import java.awt.geom.AffineTransform;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import org.gvsig.fmap.dal.exception.DataException;

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

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.aggregate.Aggregate;

import org.gvsig.fmap.geom.aggregate.MultiCurve;

import org.gvsig.fmap.geom.aggregate.MultiPoint;

import org.gvsig.fmap.geom.aggregate.MultiSurface;

import org.gvsig.fmap.geom.operation.GeometryOperationException;

import org.gvsig.fmap.geom.operation.GeometryOperationNotSupportedException;

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

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

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

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

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

import org.gvsig.fmap.mapcontext.rendering.symbols.ISymbol;

import org.gvsig.symbology.SymbologyLocator;

import org.gvsig.symbology.SymbologyManager;

import org.gvsig.symbology.fmap.mapcontext.rendering.symbol.text.ISimpleTextSymbol;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposableIterator;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.exception.BaseException;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.service.spi.ProviderServices;

import org.gvsig.tools.visitor.VisitCanceledException;

import org.gvsig.tools.visitor.Visitor;

import org.gvsig.vectorediting.lib.api.DrawingStatus;

import org.gvsig.vectorediting.lib.api.EditingServiceParameter;

import org.gvsig.vectorediting.lib.api.EditingServiceParameter.TYPE;

import org.gvsig.vectorediting.lib.api.exceptions.DrawServiceException;

import org.gvsig.vectorediting.lib.api.exceptions.FinishServiceException;

import org.gvsig.vectorediting.lib.api.exceptions.InvalidEntryException;

import org.gvsig.vectorediting.lib.api.exceptions.StartServiceException;

import org.gvsig.vectorediting.lib.spi.AbstractEditingProvider;

import org.gvsig.vectorediting.lib.spi.DefaultDrawingStatus;

import org.gvsig.vectorediting.lib.spi.DefaultEditingServiceParameter;

import org.gvsig.vectorediting.lib.spi.EditingProvider;

import org.gvsig.vectorediting.lib.spi.EditingProviderFactory;

import org.gvsig.vectorediting.lib.spi.EditingProviderLocator;

import org.gvsig.vectorediting.lib.spi.EditingProviderManager;

import org.gvsig.vectorediting.lib.spi.EditingProviderServices;

public class RotateEditingProvider extends AbstractEditingProvider implements

    EditingProvider {

    private I18nManager i18nManager = ToolsLocator.getI18nManager();

    private EditingServiceParameter selectionParameter;

    private EditingServiceParameter centerPointParameter;

    private EditingServiceParameter angleParameter;

    private Map<EditingServiceParameter, Object> values;

    private FeatureStore featureStore;

    public RotateEditingProvider(ProviderServices providerServices,

        DynObject parameters) {

        super(providerServices);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.selectionParameter =

            new DefaultEditingServiceParameter("selection",

                i18nManager.getTranslation("selection"), TYPE.SELECTION);

        this.centerPointParameter =

            new DefaultEditingServiceParameter("center_of_rotation",

                i18nManager.getTranslation("center_of_rotation"), TYPE.POSITION);

        this.angleParameter =

            new DefaultEditingServiceParameter("angle_of_rotation",

                i18nManager.getTranslation("angle_of_rotation"), TYPE.POSITION,

                TYPE.VALUE);

    }

    private ISimpleTextSymbol getTextSymbol(){

        SymbologyManager symbologyManager = SymbologyLocator.getSymbologyManager();

        ISimpleTextSymbol textSymbol = symbologyManager.createSimpleTextSymbol();

        textSymbol.setFontSize(10);

        return textSymbol;

    }

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        DefaultDrawingStatus drawingStatus = new DefaultDrawingStatus();

        EditingProviderManager editingProviderManager =

            EditingProviderLocator.getProviderManager();

        ISymbol auxiliaryPointSymbolEditing = editingProviderManager.getSymbol("auxiliary-point-symbol-editing");

        ISymbol auxiliaryLineSymbolEditing = editingProviderManager.getSymbol("auxiliary-line-symbol-editing");

        ISymbol ruleAxisSymbol = editingProviderManager.getSymbol("rule-axis-symbol");

        ISymbol lineSymbolEditing = editingProviderManager.getSymbol("line-symbol-editing");

        ISymbol polygonSymbolEditing = editingProviderManager.getSymbol("polygon-symbol-editing");

        FeatureSelection selected =

            (FeatureSelection) values.get(selectionParameter);

        try {

            if ((selected != null) && !selected.isEmpty()) {

                Point p1 = (Point) values.get(centerPointParameter);

                if (p1 != null) {

                    drawingStatus.addStatus(p1, auxiliaryPointSymbolEditing, "");

                    Object angleValue = values.get(angleParameter);

                    Double angle = null;

                    EditingProviderServices editingProviderServices =

                        (EditingProviderServices) getProviderServices();

                    if ((angleValue != null) && (angleValue instanceof Double)) {

                        angle = (Double) angleValue;

                    } else {

                        angle =

                            editingProviderServices.getAngle(

                                (Point) values.get(centerPointParameter),

                                mousePosition);

                        Line line;

                        GeometryManager geometryManager =

                            GeometryLocator.getGeometryManager();

                        int subType =

                            editingProviderServices.getSubType(featureStore);

                        line = geometryManager.createLine(subType);

                        line.addVertex(p1);

                        Point p2 =

                            geometryManager.createPoint(

                                p1.getX() + p1.distance(mousePosition),

                                p1.getY(), subType);

                        line.addVertex(p2);

                        drawingStatus.addStatus(line, ruleAxisSymbol, "");

                        Line line2;

                        line2 = geometryManager.createLine(subType);

                        line2.addVertex(p1);

                        line2.addVertex(mousePosition);

                        drawingStatus.addStatus(line2, ruleAxisSymbol, "");

                        Double ext = (2 * Math.PI) - angle;

                        Arc arc =

                            editingProviderServices

                                .createArc(p1, p1.distance(mousePosition) / 2,

                                    0, ext, subType);

                        drawingStatus.addStatus(arc, auxiliaryLineSymbolEditing, "");

                        double textDistance = 3*p1.distance(mousePosition) / 4;

                        Point pointText = geometryManager.createPoint(p1.getX()+textDistance*Math.cos(angle/2), p1.getY()+textDistance*Math.sin(angle/2), subType);

                        ISimpleTextSymbol textSymbol = getTextSymbol();

                        drawingStatus.addStatus(pointText, textSymbol, degToDms(Math.toDegrees(angle)));

                    }

                    DisposableIterator it;

                    it = selected.fastIterator();

                    AffineTransform at;

                    try {

                        at = getRotateAffineTransform(p1, angle);

                    } catch (Exception e) {

                        throw new DrawServiceException(e);

                    }

                    while (it.hasNext()) {

                        Feature feat = (Feature) it.next();

                        Geometry transformedGeometry = feat.getDefaultGeometry().cloneGeometry();

                        transformedGeometry.transform(at);

                        ISymbol symbol=null;

                        if(transformedGeometry instanceof Curve || transformedGeometry instanceof MultiCurve){

                            symbol = lineSymbolEditing;

                        } else if(transformedGeometry instanceof Surface || transformedGeometry instanceof MultiSurface){

                            symbol = polygonSymbolEditing;

                        } else if(transformedGeometry instanceof Point || transformedGeometry instanceof MultiPoint){

                            symbol = auxiliaryPointSymbolEditing;

                        }

                        if(transformedGeometry instanceof Aggregate){

                            int primitivesNumber = ((Aggregate)transformedGeometry).getPrimitivesNumber();

                            for (int i = 0; i < primitivesNumber; i++) {

                                drawingStatus.addStatus(((Aggregate)transformedGeometry).getPrimitiveAt(i), symbol, "");

                            }

                        } else {

                            drawingStatus.addStatus(transformedGeometry, symbol, "");

                        }

                    }

                    it.dispose();

                }

                return drawingStatus;

            }

        } catch (Exception e) {

            throw new DrawServiceException(e);

        }

        return null;

    }

    private String degToDms (double deg) {

        int d = (int)Math.floor(deg);

        double minfloat = (deg-d)*60;

        int m = (int)Math.floor(minfloat);

        double secfloat = (minfloat-m)*60;

        int s = (int)Math.round(secfloat);

        // After rounding, the seconds might become 60. These two

        // if-tests are not necessary if no rounding is done.

        if (s==60) {

          m++;

          s=0;

        }

        if (m==60) {

          d++;

          m=0;

        }

        return ("" + d + "\u00B0" + m + "\u2032" + s+"\u2033");

     }

    private AffineTransform getRotateAffineTransform(Point axisP1, Double angle)

        throws GeometryOperationNotSupportedException,

        GeometryOperationException {

        AffineTransform translate =

            AffineTransform

                .getTranslateInstance(-axisP1.getX(), -axisP1.getY());

        AffineTransform rotate = AffineTransform.getRotateInstance(angle);

        AffineTransform inverseTranslate =

            AffineTransform.getTranslateInstance(axisP1.getX(), axisP1.getY());

        AffineTransform at = new AffineTransform(translate);

        at.preConcatenate(rotate);

        at.preConcatenate(inverseTranslate);

        return at;

    }

    public EditingServiceParameter next() {

        if (values.get(selectionParameter) == null) {

            return this.selectionParameter;

        } else if (values.get(centerPointParameter) == null) {

            return this.centerPointParameter;

        } else if (values.get(angleParameter) == null) {

            return this.angleParameter;

        }

        return null;

    }

    public void stop() {

        values.clear();

    }

    private void validateAndInsertValue(EditingServiceParameter param,

        Object value) throws InvalidEntryException {

        if (param == selectionParameter) {

            if (value instanceof FeatureSelection) {

                values.put(param, value);

                return;

            }

        } else if (param == centerPointParameter) {

            if (value instanceof Point) {

                values.put(param, value);

                return;

            }

        } else if (param == angleParameter) {

            if (value instanceof Point) {

                Double angleValue = null;

                try {

                    EditingProviderServices editingProviderServices =

                        (EditingProviderServices) getProviderServices();

                    angleValue =

                        editingProviderServices.getAngle(

                            (Point) values.get(centerPointParameter),

                            ((Point) value));

                } catch (Exception e) {

                    throw new InvalidEntryException(e);

                }

                if (angleValue != null) {

                    values.put(param, angleValue);

                    return;

                }

            } else if (value instanceof Double) {

                values.put(param, (Double)value*Math.PI/180);

                return;

            }

        }

    }

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> list =

            new ArrayList<EditingServiceParameter>();

        list.add(selectionParameter);

        list.add(centerPointParameter);

        list.add(angleParameter);

        return list;

    }

    public void setValue(Object value) throws InvalidEntryException {

        EditingServiceParameter param = next();

        validateAndInsertValue(param, value);

    }

    public void finishAndStore() throws FinishServiceException {

        FeatureSelection selected =

            (FeatureSelection) values.get(selectionParameter);

        try {

            if (!selected.isEmpty()) {

                Point p1 = (Point) values.get(centerPointParameter);

                Double angle = (Double) values.get(angleParameter);

                if ((p1 != null) && (angle != null)) {

                    final AffineTransform at;

                    try {

                        at = getRotateAffineTransform(p1, angle);

                    } catch (GeometryOperationNotSupportedException e) {

                        throw new FinishServiceException(e);

                    } catch (GeometryOperationException e) {

                        throw new FinishServiceException(e);

                    }

                    try {

                        selected.accept(new Visitor() {

                            public void visit(Object obj)

                                throws VisitCanceledException, BaseException {

                                Feature feature = (Feature) obj;

                                Geometry geom =

                                    feature.getDefaultGeometry()

                                        .cloneGeometry();

                                geom.transform(at);

                                // Se sustituye la geometr?a original por la

                                // calculada

                                EditableFeature editableFeature =

                                    feature.getEditable();

                                editableFeature.setDefaultGeometry(geom);

                                ((EditingProviderServices) getProviderServices())

                                    .updateFeatureInFeatureStore(

                                        editableFeature, featureStore);

                            }

                        });

                    } catch (BaseException e) {

                        throw new FinishServiceException(e);

                    }

                    featureStore.getFeatureSelection().deselectAll();

                }

            }

        } catch (DataException e) {

            throw new FinishServiceException(e);

        }

    }

    public Geometry finish() throws FinishServiceException {

        return null;

    }

    public void start() throws StartServiceException {

        this.values = new HashMap<EditingServiceParameter, Object>();

        FeatureSelection selected = null;

        if (featureStore != null) {

            try {

                selected =

                    (FeatureSelection) featureStore.getFeatureSelection()

                        .clone();

            } catch (DataException e) {

                throw new StartServiceException(e);

            } catch (CloneNotSupportedException e) {

                // Do nothing

            }

            if ((selected != null) && (selected.getSelectedCount() > 0)) {

                values.put(selectionParameter, selected);

            }

        }

    }

    public String getName() {

        return RotateEditingProviderFactory.PROVIDER_NAME;

    }

}