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

import java.awt.geom.AffineTransform;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.Iterator;

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

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

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

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.GeometryUtils;

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

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

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

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

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

import org.gvsig.fmap.geom.exception.CreateGeometryException;

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

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

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

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dataTypes.DataTypes;

import org.gvsig.tools.dispose.DisposableIterator;

import org.gvsig.tools.dispose.DisposeUtils;

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.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.api.exceptions.StopServiceException;

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

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;

/**

 * @author fdiaz

 *

 */

public class MeasureEditingProvider extends AbstractEditingProvider implements

        EditingProvider {

    protected final EditingServiceParameter selection;

    protected final EditingServiceParameter insertionPointParameter;

    protected final EditingServiceParameter geometryBaseParameter;

    private final EditingServiceParameter measureParameter;

    protected final EditingServiceParameter alignParameter;

    protected final EditingServiceParameter rotateFieldParameter;

    protected final EditingServiceParameter deleteOriginalGeometriesParameter;

    protected final FeatureStore featureStore;

    protected final MapContext mapContext;

    protected Map<EditingServiceParameter, Object> values;

    /**

     * Default constructor.

     *

     * @param services available services for this provider

     * @param parameters of this provider

     */

    public MeasureEditingProvider(DynObject parameters,

            ProviderServices services) {

        super(services);

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        EditingProviderServices editingProviderServices

                = (EditingProviderServices) getProviderServices();

        this.featureStore

                = (FeatureStore) parameters

                        .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.mapContext

                = (MapContext) parameters

                        .getDynValue(EditingProviderFactory.MAPCONTEXT_FIELD);

        this.selection

                = new DefaultEditingServiceParameter("selection", "selection",

                        TYPE.SELECTION);

        this.insertionPointParameter

                = new DefaultEditingServiceParameter("_insertion_point", "_insertion_point",

                        TYPE.POSITION);

        this.geometryBaseParameter

                = new DefaultEditingServiceParameter("_geometry_base", "_geometry_base",

                        TYPE.POSITION);

        this.measureParameter

                = new DefaultEditingServiceParameter("_measure",

                        "_measure", EditingServiceParameter.TYPE.VALUE).setDataType(DataTypes.DOUBLE);;

        DefaultEditingServiceParameterOptions alignOptions2 = new DefaultEditingServiceParameterOptions()

                .add("_align", true, i18nManager.getTranslation("_yes"))

                .add("_dont_align", false, i18nManager.getTranslation("_no"));

        String alignConsoleMsg

                = editingProviderServices.makeConsoleMessage(

                        "_align", alignOptions2);

        this.alignParameter

                = new DefaultEditingServiceParameter(

                        "_align",

                        alignConsoleMsg,

                        alignOptions2,

                        i18nManager.getTranslation("_no"),

                        true,

                        TYPE.OPTION

                ).setDataType(DataTypes.BOOLEAN);

        DefaultEditingServiceParameterOptions rotateFieldOptions2 = new DefaultEditingServiceParameterOptions();

        Iterable<FeatureAttributeDescriptor> attrs = featureStore.getDefaultFeatureTypeQuietly().getAllAttributeDescriptors();

        for (FeatureAttributeDescriptor attr : attrs) {

            if(attr.getDataType().isNumeric()) {

                rotateFieldOptions2.add(attr.getLabel(), attr.getName(), attr.getShortLabel());

            }

        }

        this.rotateFieldParameter

                = new DefaultEditingServiceParameter(

                        "_rotate_field",

                        "_rotate_field",

                        rotateFieldOptions2,

                        null,

                        TYPE.OPTION);

        DefaultEditingServiceParameterOptions deleteOriginalGeometriesOptions2 = new DefaultEditingServiceParameterOptions()

                .add("delete_original_geometries", true, i18nManager.getTranslation("_yes"))

                .add("keep_original_geometries", false, i18nManager.getTranslation("_no"));

        String consoleMsg

                = editingProviderServices.makeConsoleMessage(

                        "delete_original_geometries_question", deleteOriginalGeometriesOptions2);

        this.deleteOriginalGeometriesParameter

                = new DefaultEditingServiceParameter(

                        i18nManager.getTranslation("delete_original_geometries"),

                        consoleMsg,

                        deleteOriginalGeometriesOptions2,

                        false,

                        TYPE.OPTION).setDataType(DataTypes.BOOLEAN);;

    }

    @Override

    public EditingServiceParameter next() {

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

            return selection;

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

            return insertionPointParameter;

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

            return geometryBaseParameter;

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

            return measureParameter;

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

            return alignParameter;

        } else if ((boolean) (values.get(alignParameter)) == true && isPointType() && values.get(rotateFieldParameter) == null) {

            return rotateFieldParameter;

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

            return deleteOriginalGeometriesParameter;

        }

        return null;

    }

    protected boolean isPointType() {

        return featureStore.getDefaultFeatureTypeQuietly().getDefaultGeometryAttribute().getGeomType().getType() == Geometry.TYPES.POINT;

    }

    @Override

    public DrawingStatus getDrawingStatus(Point mousePosition)

            throws DrawServiceException {

        DefaultDrawingStatus drawingStatus = new DefaultDrawingStatus();

        GeometryManager geometryManager = GeometryLocator.getGeometryManager();

        EditingProviderManager editingProviderManager

                = EditingProviderLocator.getProviderManager();

        EditingProviderServices editingProviderServices

                = (EditingProviderServices) getProviderServices();

        int subtype;

        try {

            subtype = editingProviderServices.getSubType(featureStore);

        } catch (DataException e2) {

            throw new DrawServiceException(e2);

        }

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

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

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

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

        if (values != null) {

            FeatureSelection featureSelection

                    = (FeatureSelection) values.get(selection);

            Point insertionPointValue = (Point) values.get(insertionPointParameter);

            if (featureSelection != null && insertionPointValue != null) {

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

                Line auxLine;

                try {

                    auxLine = geometryManager.createLine(subtype);

                } catch (CreateGeometryException e1) {

                    throw new DrawServiceException(e1);

                }

                MultiLine geometryBaseValue = (MultiLine) values.get(geometryBaseParameter);

                if (geometryBaseValue != null) {

                    for (Geometry geometry : geometryBaseValue) {

                        Line line = (Line) geometry;

                        List<PointAndRotation> pointsAndRotation;

                        try {

                            pointsAndRotation = getPointsAndRotations(line);

                        } catch (Exception ex) {

                            throw new DrawServiceException(ex);

                        }

                        if (pointsAndRotation != null) {

                            for (PointAndRotation pointAndRotation : pointsAndRotation) {

                                DisposableIterator it = null;

                                try {

                                    it = featureSelection.fastIterator();

                                    while (it.hasNext()) {

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

                                        ISymbol previewSymbol = this.getPreviewSymbol(feature);

                                        Geometry geom

                                                = feature.getDefaultGeometry().cloneGeometry();

                                        AffineTransform at

                                                = getMoveAffineTransform(insertionPointValue,

                                                        pointAndRotation.getPoint());

                                        geom.transform(at);

                                        Boolean alignValue = (Boolean) values.get(alignParameter);

                                        if (alignValue != null && alignValue.equals(Boolean.TRUE)) {

                                            AffineTransform rotateTransform = getRotateAffineTransform(pointAndRotation.getPoint(), pointAndRotation.getAngle()); //pointAndRotation.getAngle()-Math.PI/2);

                                            geom.transform(rotateTransform);

                                        }

                                        ISymbol symbol = null;

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

                                            symbol = lineSymbolEditing;

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

                                            symbol = polygonSymbolEditing;

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

                                            symbol = auxiliaryPointSymbolEditing;

                                        }

                                        if (geom instanceof Aggregate) {

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

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

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

                                                drawingStatus.addStatus(((Aggregate) geom).getPrimitiveAt(i), previewSymbol, "");

                                            }

                                        } else {

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

                                            drawingStatus.addStatus(geom, previewSymbol, "");

                                        }

                                    }

                                } catch (BaseException e) {

                                    throw new DrawServiceException(e);

                                } finally {

                                    DisposeUtils.disposeQuietly(it);

                                }

                            }

                        }

                    }

                }

            }

        }

        return drawingStatus;

    }

    private AffineTransform getMoveAffineTransform(Point p1, Point p2)

            throws GeometryOperationNotSupportedException,

            GeometryOperationException {

        AffineTransform translate

                = AffineTransform.getTranslateInstance(p2.getX() - p1.getX(),

                        p2.getY() - p1.getY());

        return translate;

    }

    private AffineTransform getRotateAffineTransform(Point p, double angle)

            throws GeometryOperationNotSupportedException,

            GeometryOperationException {

        AffineTransform rotate

                = AffineTransform.getRotateInstance(angle, p.getX(), p.getY());

        return rotate;

    }

    @Override

    public void stop() throws StopServiceException {

        if (values != null) {

            values.clear();

        }

    }

    @Override

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> parameters

                = new ArrayList<>();

        parameters.add(selection);

        parameters.add(insertionPointParameter);

        parameters.add(geometryBaseParameter);

        parameters.add(measureParameter);

        parameters.add(alignParameter);

        parameters.add(rotateFieldParameter);

        parameters.add(deleteOriginalGeometriesParameter);

        return parameters;

    }

    @Override

    public boolean isEnabled(EditingServiceParameter parameter) {

        return true;

    }

    @Override

    public void setValue(EditingServiceParameter parameter, Object value) throws InvalidEntryException {

        validateAndInsertValue(parameter, value);

    }

    @Override

    public void setValue(Object value) throws InvalidEntryException {

        EditingServiceParameter parameter = next();

        validateAndInsertValue(parameter, value);

    }

    protected void validateAndInsertValue(EditingServiceParameter parameter,

            Object value) throws InvalidEntryException {

        if (parameter == selection) {

            if (value instanceof FeatureSelection) {

                if (((FeatureSelection) value).getSelectedCount() > 0) {

                    values.put(selection, value);

                }

            }

        } else if (parameter == insertionPointParameter) {

            if (value instanceof Point) {

                values.put(parameter, value);

            }

        } else if (parameter == geometryBaseParameter && value instanceof Point) {

            if (value instanceof Point) {

                Point point = (Point) value;

                Geometry geometry = getGeometry(point); //editingProviderServices.getGeometryOfVisibleLayers(point, featureStore, mapContext);

                if (geometry != null) {

                    try {

                        if (geometry instanceof Curve

                                || geometry instanceof MultiCurve

                                || geometry instanceof Surface

                                || geometry instanceof MultiSurface) {

                            values.put(parameter, geometry.toLines());

                        } else {

                            throw new InvalidEntryException(null);

                        }

                    } catch (Exception ex) {

                        throw new InvalidEntryException(ex);

                    }

                    return;

                }

            }

        } else if (parameter.equals(measureParameter)) {

            if (value instanceof Double) {

                values.put(parameter, value);

            } else {

                throw new InvalidEntryException(null);

            }

        } else if (parameter == rotateFieldParameter) {

            values.put(parameter, parameter.getOptions2().getValue(value, parameter.getDefaultValue()));

        } else if (parameter == alignParameter) {

            values.put(parameter, parameter.getOptions2().getValue(value, parameter.getDefaultValue()));

        } else if (parameter == deleteOriginalGeometriesParameter) {

            values.put(parameter, parameter.getOptions2().getValue(value, parameter.getDefaultValue()));

        }

    }

    private Geometry getGeometry(Point point) {

        EditingProviderServices editingProviderServices = (EditingProviderServices) getProviderServices();

        Geometry geometry = editingProviderServices.getGeometryOfVisibleLayers(point, featureStore, mapContext);

        return geometry;

    }

    @Override

    public Geometry finish() throws FinishServiceException {

        return null;

    }

    @Override

    public void finishAndStore() throws FinishServiceException {

        if (values != null) {

            final Point insertionPointValue = (Point) values.get(insertionPointParameter);

            final MultiLine geometryBaseValue = (MultiLine) values.get(geometryBaseParameter);

//            final Double measureValue = (Double) values.get(measureParameter);

            final Boolean alignValue = (Boolean) values.get(alignParameter);

            final String rotateFieldValue = (String) values.get(rotateFieldParameter);

            final Boolean deleteOriginalGeometriesValue = (Boolean) values.get(deleteOriginalGeometriesParameter);

            FeatureSelection featureSelection

                    = (FeatureSelection) values.get(selection);

            ToolsLocator.getDisposableManager().bind(featureSelection);

            try {

                for (Geometry geometry : geometryBaseValue) {

                    Line line = (Line) geometry;

                    List<PointAndRotation> pointsAndRotations;

                    try {

                        pointsAndRotations = getPointsAndRotations(line);

                    } catch (Exception ex) {

                        throw new DrawServiceException(ex);

                    }

                    if (pointsAndRotations == null) {

                        return;

                    }

                    for (PointAndRotation pointAndRotation : pointsAndRotations) {

                        DisposableIterator it = null;

                        featureSelection.accept((Object obj) -> {

                            Feature feature = (Feature) obj;

                            EditingProviderServices editingProviderServices

                                    = (EditingProviderServices) getProviderServices();

                            EditableFeature editable

                                    = editingProviderServices.getFeatureCopyWithoutUniqueIndex(

                                            featureStore, feature);

                            Geometry geom

                                    = feature.getDefaultGeometry().cloneGeometry();

                            AffineTransform at

                                    = getMoveAffineTransform(insertionPointValue,

                                            pointAndRotation.getPoint());

                            geom.transform(at);

                            if (alignValue != null && alignValue.equals(Boolean.TRUE)) {

                                // We subtract PI/2 because we consider that when a figure is not rotated it points upwards

                                AffineTransform rotateTransform = getRotateAffineTransform(pointAndRotation.getPoint(), pointAndRotation.getAngle()); //-Math.PI/2);

                                geom.transform(rotateTransform);

                                if (rotateFieldValue != null) {

                                    FeatureType featType = featureStore.getDefaultFeatureTypeQuietly();

                                    if (featType != null) {

                                        FeatureAttributeDescriptor attr = featType.getAttributeDescriptor((String) rotateFieldValue);

                                        if (attr != null) {

                                            // We change the sign of the angle because in the MarkerSymbols the angles go in a clockwise direction

                                            editable.set(rotateFieldValue, -(Math.toDegrees(pointAndRotation.getAngle()-Math.PI/2)));

                                        }

                                    }

                                }

                            }

                            editable.setDefaultGeometry(geom);

                            editingProviderServices.insertFeatureIntoFeatureStore(

                                    editable, featureStore);

                        });

                    }

                }

                if(deleteOriginalGeometriesValue) {

                    for(Feature feature : featureStore.getFeatureSelection()) {

                        featureStore.delete(feature);

                    }

                }

                featureStore.getFeatureSelection().deselectAll();

                featureSelection.dispose();

            } catch (BaseException e) {

                throw new FinishServiceException(e);

            }

        }

    }

    @Override

    public void start() throws StartServiceException, InvalidEntryException {

        values = new HashMap<>();

        FeatureSelection selected = null;

        if (featureStore != null) {

            try {

                selected = featureStore.getFeatureSelection();

            } catch (DataException e) {

                throw new StartServiceException(e);

            }

            try {

                setValue(selected);

            } catch (InvalidEntryException e) {

                throw new InvalidEntryException(e);

            }

        }

    }

    @Override

    public String getName() {

        return MeasureEditingProviderFactory.PROVIDER_NAME;

    }

    @Override

    public Object getValue(EditingServiceParameter parameter) {

        return values != null ? values.get(parameter) : null;

    }

    protected List<PointAndRotation> getPointsAndRotations(Line line) throws CreateGeometryException, GeometryOperationNotSupportedException, GeometryOperationException, CloneNotSupportedException {

        Double measureValue = (Double) values.get(measureParameter);

        if (measureValue == null) {

            return null;

        }

        return MeasureEditingProvider.this.getPointsAndRotations(line, measureValue);

    }

    protected List<PointAndRotation> getPointsAndRotations(Line line, double measure) throws CreateGeometryException, GeometryOperationNotSupportedException, GeometryOperationException, CloneNotSupportedException {

        GeometryManager geomManager = GeometryLocator.getGeometryManager();

        List<PointAndRotation> res = new ArrayList<>();

        if (line.getNumVertices() <= 1) {

            return res;

        }

        Point previousPoint = null;

        double previousLength = 0;

        // Insertamos el primer punto solo si la linea es cerrada

        if(line.isClosed()){

            Point v0 = line.getVertex(0);

            int i = 1;

            Point v1 = line.getVertex(1);

            // Este bucle es para evitar un problema con geometr?as que lleguen con puntos repetidos al inicio

//            while (v0.equals(v1) && line.getNumVertices() > i+1 ){

//                v1 = line.getVertex(i++);

//            }

            PointAndRotation rotatedPoint = new PointAndRotation(

                    line.getVertex(0),

                    GeometryUtils.calculateAngle(v0, v1)

            );

            res.add(rotatedPoint);

        }

        boolean advanceToNext = true;

        Iterator<Point> it = line.iterator();

        Point currentPoint = null;

        while (it.hasNext() || !advanceToNext) {

            if (advanceToNext) {

                currentPoint = it.next();

            }

            if (previousPoint == null) {

                previousPoint = currentPoint.clone();

                advanceToNext = true;

                continue;

            }

            double distance = previousPoint.distance(currentPoint);

            if (previousLength + distance < measure) {

                previousLength += distance;

                previousPoint = currentPoint.cloneGeometry();

                advanceToNext = true;

            } else {

                //buscar punto dentro del segmento a una distancia = segmentLengthMeters-previousLength

                Point point = GeometryUtils.calculateLambdaPoint(previousPoint, currentPoint, (measure - previousLength) / distance);

                //agregarlo al currentSegment,

                PointAndRotation rotatedPoint = new PointAndRotation(

                        point,

                        GeometryUtils.calculateAngle(previousPoint, currentPoint)

                );

                res.add(rotatedPoint);

                previousPoint = point.clone();

                previousLength = 0;

                advanceToNext = false;

            }

        }

        return res;

    }

    public static class PointAndRotation {

        private Point point;

        private double angle;

        public PointAndRotation(Point point, double angle) {

            this.point = point;

            this.angle = angle;

        }

        public void setPoint(Point point) {

            this.point = point;

        }

        public Point getPoint() {

            return point;

        }

        public void setAngle(double angle) {

            this.angle = angle;

        }

        public double getAngle() {

            return angle;

        }

    }

}