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.offset.lib.prov.offset;

import java.awt.Color;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.LinkedHashMap;

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

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

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

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

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

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

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

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

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

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

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

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

import org.gvsig.fmap.mapcontext.MapContextLocator;

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

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposableIterator;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.i18n.I18nManager;

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

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 OffsetEditingProvider extends AbstractEditingProvider implements

    EditingProvider {

    private final Double PRECISION = new Double(1.0e-5);

    private I18nManager i18nManager = ToolsLocator.getI18nManager();

    private EditingServiceParameter selectionParameter;

    private EditingServiceParameter offsetParameter;

    private EditingServiceParameter deleteOriginalGeometriesParameter;

    private boolean deleteOriginalGeometries = false;

    private Map<EditingServiceParameter, Object> values;

    private Map<String, String> options;

    private FeatureStore featureStore;

    public OffsetEditingProvider(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.offsetParameter =

            new DefaultEditingServiceParameter("offset_distance",

                i18nManager.getTranslation("offset_distance"),

                TYPE.POSITION, TYPE.VALUE);

        this.options = new LinkedHashMap<String, String>();

        options.put(i18nManager.getTranslation("short_yes"),

            "delete_original_geometries");

        options.put(i18nManager.getTranslation("short_no"),

            "keep_original_geometries");

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        String consoleMsg =

            editingProviderServices.makeConsoleMessage(

                "delete_original_geometries_question", options);

        this.deleteOriginalGeometriesParameter =

            new DefaultEditingServiceParameter("Delete original geometries",

                consoleMsg, options, TYPE.OPTION);

    }

    public EditingServiceParameter next() {

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

            return selectionParameter;

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

            return offsetParameter;

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

            return this.deleteOriginalGeometriesParameter;

        }

        return null;

    }

    public DrawingStatus getDrawingStatus(Point mousePosition) throws DrawServiceException {

        DefaultDrawingStatus drawingStatus = new DefaultDrawingStatus();

        EditingProviderManager editingProviderManager =

            EditingProviderLocator.getProviderManager();

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

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

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

        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 point = null;

                double distance = 0.0;

                Object offsetValue = values.get(offsetParameter);

                if (offsetValue != null){

                    if (offsetValue instanceof Point) {

                        point = (Point) offsetValue;

                    } else {

                        distance = (Double) offsetValue;

                    }

                } else {

                    point = mousePosition;

                    distance = getMinDistance(selected, point);

                }

                DisposableIterator it;

                it = selected.fastIterator();

                while (it.hasNext()) {

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

                    Geometry transformedGeometry = feat.getDefaultGeometry().offset(distance);

                    ISymbol symbol = null;

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

                        symbol = lineSymbolEditing;

                        drawingStatus.addStatus(feat.getDefaultGeometry(), auxiliaryLineSymbolEditingDirection, "Direction");

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

        }

    }

    /**

     * @param selected

     * @param point

     * @return

     * @throws DataException

     * @throws GeometryOperationException

     * @throws GeometryOperationNotSupportedException

     * @throws GeometryException

     */

    private double getMinDistance(FeatureSelection selected, Point point) throws DataException, GeometryOperationNotSupportedException, GeometryOperationException, GeometryException {

        double minorDistance=Double.POSITIVE_INFINITY;

        DisposableIterator it;

        it = selected.fastIterator();

        while (it.hasNext()) {

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

            Geometry geometry = feat.getDefaultGeometry();

            double distance = getDistance(geometry, point);

            if(distance < minorDistance){

                minorDistance = distance;

            }

        }

        it.dispose();

        return minorDistance;

    }

    /**

     * @param geometry

     * @param point

     * @return

     * @throws GeometryOperationException

     * @throws GeometryOperationNotSupportedException

     * @throws GeometryException

     */

    private double getDistance(Geometry geometry, Point point) throws GeometryOperationNotSupportedException, GeometryOperationException, GeometryException {

        GeometryManager geomManager = GeometryLocator.getGeometryManager();

        if(geometry instanceof Arc){

            Arc arc = (Arc)geometry;

            Point center = arc.getCenterPoint();

            double radius = center.distance(arc.getInitPoint());

            double distance = center.distance(point)-radius;

            return distance;

        }

        if(geometry instanceof Circle){

            Circle circle = (Circle)geometry;

            return circle.getCenter().distance(point)-circle.getRadious();

        }

        if(geometry instanceof Circumference){

            Circumference circumference = (Circumference)geometry;

            return circumference.getCenter().distance(point)-circumference.getRadious();

        }

        if(geometry instanceof PeriEllipse){

            double minDistance = Double.POSITIVE_INFINITY;

            PeriEllipse ellipse = (PeriEllipse)geometry;

            Geometry[] closestPoints = point.closestPoints(ellipse);

            if (closestPoints!=null){

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

                    Geometry closestPoint = closestPoints[i];

                    if (!point.equals(closestPoint)) {

                        double distance = closestPoint.distance(point);

                        if (distance < minDistance) {

                            minDistance = distance;

                        }

                    }

                }

            }

            Ellipse auxEllipse = (Ellipse)geomManager.create(Geometry.TYPES.ELLIPSE, geometry.getGeometryType().getSubType());

            auxEllipse.setPoints(ellipse.getAxis1Start(), ellipse.getAxis1End(), ellipse.getAxis2Dist());

            if(auxEllipse.contains(point)){

                return -minDistance;

            }

            return minDistance;

        }

        if(geometry instanceof Ellipse){

            Ellipse ellipse = (Ellipse)geometry;

            PeriEllipse auxPeriEllipse = (PeriEllipse)geomManager.create(Geometry.TYPES.PERIELLIPSE, geometry.getGeometryType().getSubType());

            auxPeriEllipse.setPoints(ellipse.getAxis1Start(), ellipse.getAxis1End(), ellipse.getAxis2Dist());

            double distance = getDistance(auxPeriEllipse, point);

//            if(ellipse.contains(point)){

//                return -distance;

//            }

            return distance;

        }

        if(geometry instanceof Spline || geometry instanceof FilledSpline ){

            return getDistance(geometry.toLines().getPrimitiveAt(0), point);

        }

        if(geometry instanceof Line){

            Line line = (Line)geometry;

            double minDistance = Double.POSITIVE_INFINITY;

            Geometry[] closestPoints = point.closestPoints(line);

            Point closestPoint = null;

            if (closestPoints!=null){

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

                    Point p=(Point)closestPoints[i];

                    if (!point.equals(p)) {

                        double distance = p.distance(point);

                        if (distance < minDistance) {

                            minDistance = distance;

                            closestPoint = p;

                        }

                    }

                }

            }

            if(closestPoint != null){

                for(int i=0; i<line.getNumVertices()-1; i++){

                    Line segment = (Line) geomManager.create(Geometry.TYPES.LINE, geometry.getGeometryType().getSubType());

                    segment.addVertex(line.getVertex(i));

                    segment.addVertex(line.getVertex(i+1));

                    if(segment.isWithinDistance(closestPoint, PRECISION)){

                        if(line.getVertex(0).equals(line.getVertex(line.getNumVertices()-1))){ //isClosed

                            if(line.toPolygons().contains(point)){

                                return -minDistance;

                            };

                            return minDistance;

                        } else {

                            return getDirectedDistance(closestPoint, point, segment);

                        }

                    }

                }

            }

        }

        if(geometry instanceof Polygon){

            Polygon polygon = (Polygon)geometry;

            if(!polygon.contains(point)){

                double minDistance = Double.POSITIVE_INFINITY;

                Geometry[] closestPoints = point.closestPoints(polygon);

                Point closestPoint = null;

                if (closestPoints!=null){

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

                        Point p=(Point)closestPoints[i];

                        if (!point.equals(p)) {

                            double distance = p.distance(point);

                            if (distance < minDistance) {

                                minDistance = distance;

                                closestPoint = p;

                            }

                        }

                    }

                }

                if(closestPoint != null){

                    return closestPoint.distance(point);

                }

            } else {

                Line auxLine = (Line) polygon.toLines().getPrimitiveAt(0);

                if(auxLine!=null){

                    return getDistance(auxLine, point);

                }

            }

        }

        if(geometry instanceof Aggregate){

            double minDistance = Double.POSITIVE_INFINITY;

            Aggregate aggregate2 = (Aggregate)geometry;

            for(int i=0; i<aggregate2.getPrimitivesNumber(); i++){

                double distance = getDistance(aggregate2.getPrimitiveAt(i),point);

                if(distance<minDistance){

                    minDistance = distance;

                }

            }

            return minDistance;

        }

        return 0.0;

    }

    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 == offsetParameter) {

            if (value instanceof Point) {

                try {

                    values.put(param, getMinDistance((FeatureSelection) values.get(selectionParameter), (Point) value));

                } catch (Exception e) {

                    throw new InvalidEntryException(e);

                }

                return;

            }

            if (value instanceof Double) {

                values.put(param, value);

            }

        } else if (param == deleteOriginalGeometriesParameter) {

            if (value instanceof String) {

                if (((String) value).trim().equalsIgnoreCase(

                    i18nManager.getTranslation("short_yes"))) {

                    deleteOriginalGeometries = true;

                } else if (((String) value).trim().equalsIgnoreCase(

                    i18nManager.getTranslation("short_no"))) {

                    deleteOriginalGeometries = false;

                } else {

                    throw new InvalidEntryException(null);

                }

                values.put(param, value);

            }

        }

    }

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> list =

            new ArrayList<EditingServiceParameter>();

        list.add(selectionParameter);

        list.add(offsetParameter);

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

                double distance = (Double) values.get(offsetParameter);

                    DisposableIterator it;

                    it = selected.fastIterator();

                    while (it.hasNext()) {

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

                        Geometry geom;

                        try {

                            geom = feature.getDefaultGeometry().offset(distance);

                        } catch (GeometryOperationNotSupportedException e) {

                            throw new FinishServiceException(e);

                        } catch (GeometryOperationException e) {

                            throw new FinishServiceException(e);

                        }

                        if (this.deleteOriginalGeometries) {

                            // Se sustituye la geometr?a original por la

                            // calculada

                            EditableFeature editableFeature =

                                feature.getEditable();

                            editableFeature.setDefaultGeometry(geom);

                            ((EditingProviderServices) getProviderServices())

                                .updateFeatureInFeatureStore(editableFeature,

                                    featureStore);

                        } else {

                            // Se crea una feature nueva copiando los valores de

                            // la feature original excepto aquellos que sean PK

                            EditingProviderServices editingProviderServices =

                                (EditingProviderServices) getProviderServices();

                            EditableFeature editableFeature =

                                editingProviderServices

                                    .getFeatureCopyWithoutPK(featureStore,

                                        feature);

                            editableFeature.setDefaultGeometry(geom);

                            editingProviderServices

                                .insertFeatureIntoFeatureStore(editableFeature,

                                    featureStore);

                        }

                    }

                    it.dispose();

                    featureStore.getFeatureSelection().deselectAll();

                }

        } catch (DataException e) {

            throw new FinishServiceException(e);

        }

    }

    private Double getDirectedDistance(Point pointInLine, Point distancePoint, Line line)

        throws GeometryOperationNotSupportedException, GeometryOperationException{

        Double distance=distancePoint.distance(pointInLine);

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        Double angle = editingProviderServices.getAngle(pointInLine, distancePoint);

        double angleLine = editingProviderServices.getAngle(line.getVertex(0), line.getVertex(1));

        Double angleDifference=angle - angleLine;

        if ( angleDifference<0 ){

            angleDifference+=2*Math.PI;

        }

        if (angleDifference > Math.PI ) {

            distance = -distance;

        }

        return distance;

    }

    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 OffsetEditingProviderFactory.PROVIDER_NAME;

    }

}