Application: gvSIG desktop » gvSIG tools: Vector editing

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2015 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.trimline;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.logging.Level;

import java.util.logging.Logger;

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 static org.gvsig.fmap.geom.Geometry.TYPES.CURVE;

import static org.gvsig.fmap.geom.Geometry.TYPES.MULTICURVE;

import org.gvsig.fmap.geom.GeometryException;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

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

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

import org.gvsig.fmap.geom.type.GeometryType;

import org.gvsig.fmap.mapcontext.MapContext;

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

import org.gvsig.tools.dispose.DisposableIterator;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.exception.BaseException;

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

import org.gvsig.vectorediting.lib.prov.trimline.operation.TrimLineOperation;

import org.gvsig.vectorediting.lib.prov.trimline.operation.TrimLineOperationUtils;

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;

/**

 * @author llmarques

 *

 */

public class TrimLineEditingProvider extends AbstractEditingProvider implements

    EditingProvider {

    private final int TOLERANCE_PIXELS = 3;

    private final EditingServiceParameter selectionParameter;

    private final EditingServiceParameter geometryToTrimWithParameter;

    private final EditingServiceParameter shearingPointParameter;

    protected Map<EditingServiceParameter, Object> values;

    private final FeatureStore featureStore;

    private final MapContext mapContext;

    /**

     * Default constructor.

     *

     * @param services

     *            available services for this provider

     * @param parameters

     *            of this provider

     */

    public TrimLineEditingProvider(ProviderServices services,

        DynObject parameters) {

        super(services);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.mapContext =

            (MapContext) parameters

                .getDynValue(EditingProviderFactory.MAPCONTEXT_FIELD);

        this.selectionParameter =

            new DefaultEditingServiceParameter("selection", "selection",

                TYPE.SELECTION);

        this.geometryToTrimWithParameter =

            new DefaultEditingServiceParameter("geometry_to_trim_with", "geometry_to_trim_with",

                TYPE.POSITION);

        this.shearingPointParameter

                = new DefaultEditingServiceParameter("shearing_point",

                        "shearing_point", TYPE.POSITION);

    }

    @Override

    public EditingServiceParameter next() {

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

            return selectionParameter;

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

            return geometryToTrimWithParameter;

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

            return shearingPointParameter;

        }

        return null;

    }

    @Override

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        FeatureSelection selection = (FeatureSelection) values.get(selectionParameter);

        Geometry shears = (Geometry) values.get(geometryToTrimWithParameter);

        if (selection != null && shears != null) {

            DefaultDrawingStatus drawingStatus = new DefaultDrawingStatus();

            EditingProviderManager editingProviderManager =

                EditingProviderLocator.getProviderManager();

            ISymbol lineSymbolEditing =

                editingProviderManager.getSymbol("line-symbol-editing");

            try {

                DisposableIterator selectionIterator = null;

                selectionIterator = selection.fastIterator();

                while (selectionIterator.hasNext()) {

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

                    if (feature != null) {

                        Geometry geometry = feature.getDefaultGeometry();

                        if(geometry != null && geometry.intersects(shears)) {

                            if (geometry instanceof MultiCurve) {

                                MultiCurve multiCurve = (MultiCurve) geometry;

                                for (int i = 0; i < multiCurve

                                        .getPrimitivesNumber(); i++) {

                                    Curve curve = multiCurve.getCurveAt(i);

                                    MultiCurve trimmedCurve

                                            = trimLine(mousePosition, curve, shears);

                                    for (int j = 0; j < trimmedCurve

                                            .getPrimitivesNumber(); j++) {

                                        drawingStatus.addStatus(trimmedCurve.getCurveAt(j),

                                                lineSymbolEditing, "");

                                    }

                                }

                            } else if (geometry instanceof Curve) {

                                Curve curve = (Curve) geometry;

                                MultiCurve extendedCurve

                                        = trimLine(mousePosition, curve, shears);

                                for (int j = 0; j < extendedCurve

                                        .getPrimitivesNumber(); j++) {

                                    drawingStatus.addStatus(

                                            extendedCurve.getCurveAt(j),

                                            lineSymbolEditing, "");

                                }

                            }

                        }

                    }

                }

            } catch (BaseException ex) {

                throw new DrawServiceException(ex);

            }

            return drawingStatus;

        }

        return null;

    }

    private MultiCurve trimLine(Point shearingPoint, Curve curve, Geometry shearsGeometry)

        throws GeometryOperationNotSupportedException,

        GeometryOperationException, DataException, CreateGeometryException {

        TrimLineOperation operation =

            TrimLineOperationUtils.getOperation((Primitive) curve);

        return operation.trimLine(curve, shearingPoint, shearsGeometry);

    }

    @Override

    public void stop() throws StopServiceException {

        values.clear();

    }

    @Override

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> parameters =

            new ArrayList<>();

        parameters.add(selectionParameter);

        parameters.add(geometryToTrimWithParameter);

        parameters.add(shearingPointParameter);

        return parameters;

    }

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

    }

    private void validateAndInsertValue(

        final EditingServiceParameter parameter, Object value)

        throws InvalidEntryException {

        if (parameter == selectionParameter) {

            if (value instanceof FeatureSelection) {

                FeatureSelection featureSelection = (FeatureSelection) value;

                if (featureSelection.getSelectedCount() > 0) {

                    try {

                        featureSelection.accept(new Visitor() {

                            public void visit(Object obj)

                                    throws VisitCanceledException, BaseException {

                                Feature feature = (Feature) obj;

                                Geometry geometry

                                        = feature.getDefaultGeometry();

                                if (!isValidGeometryToSelect(geometry)) {

                                    throw new InvalidEntryException(null);

                                }

                            }

                        });

                        values.put(parameter, featureSelection);

                        return;

                    } catch (BaseException e) {

                        throw new InvalidEntryException(e);

                    }

                }

            }

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

            if (value instanceof Point) {

                Point point = (Point) value;

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

                if (geometry != null) {

                    try {

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

                    } catch (GeometryException ex) {

                        throw new InvalidEntryException(ex);

                    }

                    return;

                }

            }

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

            if (value instanceof Point) {

                Point point = (Point) value;

                values.put(parameter, point);

                return;

            }

        }

        throw new InvalidEntryException(null);

    }

    private boolean isValidGeometryToSelect(Geometry geometry) {

        GeometryType geoType

                = geometry.getGeometryType();

        if (!geoType.isTypeOf(CURVE)

                && !geoType.isTypeOf(MULTICURVE)) {

            return false;

        }

        if (geoType.isTypeOf(CURVE)) {

            if (geometry instanceof Line ) {

                Line line = (Line) geometry;

                return !line.isClosed();

            }

            if (geometry instanceof Arc ) {

                Arc arc = (Arc) geometry;

                return !arc.getInitPoint().equals(arc.getEndPoint());

            }

            return false;

        }

        if (geoType.isTypeOf(MULTICURVE)) {

            MultiCurve multiCurve = (MultiCurve) geometry;

            for (Geometry geom : multiCurve) {

                if (!isValidGeometryToSelect(geom)) {

                    return false;

                }

            }

        }

        return true;

    }

    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 {

        FeatureSelection selection = (FeatureSelection) values.get(selectionParameter);

        Geometry shearsGeometry = (Geometry) values.get(geometryToTrimWithParameter);

        Point point = (Point) values.get(shearingPointParameter);

        if (selection != null && shearsGeometry != null && point != null) {

            try {

                DisposableIterator selectionIterator = null;

                selectionIterator = selection.fastIterator();

                while (selectionIterator.hasNext()) {

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

                    Geometry geometry = feature.getDefaultGeometry();

                    EditableFeature eFeature = feature.getEditable();

                    if(geometry.intersects(shearsGeometry)){

                        if (geometry instanceof Curve) {

                            Curve curveToTrim = (Curve) geometry;

                            MultiCurve result =

                                trimLine(point, curveToTrim, shearsGeometry);

                            eFeature.setDefaultGeometry(result);

                            selection.update(eFeature);

                        } else if (geometry instanceof MultiCurve) {

                            int subType =

                                geometry.getGeometryType().getSubType();

                            MultiCurve trimmedMultiCurve =

                                createMultiCurve(subType);

                            MultiCurve multiCurveToTrim = (MultiCurve) geometry;

                            double minDistance = Double.POSITIVE_INFINITY;

                            int nearestCurveIndex = -1;

                            for (int i = 0; i < multiCurveToTrim

                                .getPrimitivesNumber(); i++) {

                                Curve primitive =

                                    multiCurveToTrim.getCurveAt(i);

                                double distance =

                                    primitive.distance(point);

                                if (distance < minDistance) {

                                    nearestCurveIndex = i;

                                    minDistance = distance;

                                }

                            }

                            MultiCurve result =

                                trimLine(point, multiCurveToTrim

                                    .getCurveAt(nearestCurveIndex), shearsGeometry);

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

                                trimmedMultiCurve.addCurve(result.getCurveAt(i));

                            }

                            for (int i = 0; i < multiCurveToTrim

                                .getPrimitivesNumber(); i++) {

                                if (i != nearestCurveIndex) {

                                    trimmedMultiCurve.addCurve(multiCurveToTrim

                                        .getCurveAt(i));

                                }

                            }

                            eFeature.setDefaultGeometry(trimmedMultiCurve);

                            selection.update(eFeature);

                        }

                    }

                }  

            } catch (BaseException ex) {

                throw new FinishServiceException(ex);

            }

        }

    }

    private MultiCurve createMultiCurve(int subtype)

        throws FinishServiceException, CreateGeometryException {

        GeometryManager geoManager = GeometryLocator.getGeometryManager();

        MultiCurve extendedMultiCurve;

        extendedMultiCurve = geoManager.createMultiCurve(subtype);

        return extendedMultiCurve;

    }

    @Override

    public void start() throws StartServiceException, InvalidEntryException {

        this.values = new HashMap<>();

        FeatureSelection selected = null;

        if (featureStore != null && values.get(selectionParameter) == null) {

            try {

                selected = featureStore.getFeatureSelection();

            } catch (DataException e) {

                throw new StartServiceException(e);

            }

            if (selected.getSelectedCount() > 0) {

                try {

                    setValue(selected);

                } catch (InvalidEntryException e) {

                    throw new InvalidEntryException(e);

                }

            }

        }

    }

    @Override

    public String getName() {

        return TrimLineEditingProviderFactory.PROVIDER_NAME;

    }

    @Override

    public Object getValue(EditingServiceParameter parameter) {

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

    }

}