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

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.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.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.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.locator.LocatorException;

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.splitline.operation.SplitLineOperation;

import org.gvsig.vectorediting.lib.prov.splitline.operation.SplitLineOperationUtils;

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

    EditingProvider {

    private EditingServiceParameter selection;

    private EditingServiceParameter point;

    private FeatureStore featureStore;

    private Map<EditingServiceParameter, Object> values;

    /**

     * Default constructor.

     *

     * @param providerServices

     *            available services for this provider

     * @param parameters

     *            of this provider

     */

    public SplitLineEditingProvider(DynObject parameters,

        ProviderServices services) {

        super(services);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.selection =

            new DefaultEditingServiceParameter("selection", "selection",

                TYPE.SELECTION);

        this.point =

            new DefaultEditingServiceParameter("indicate_point",

                "indicate_point", TYPE.POSITION);

    }

    public EditingServiceParameter next() {

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

            return selection;

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

            return point;

        }

        return null;

    }

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        if (values != null) {

            DefaultDrawingStatus drawingStatus = new DefaultDrawingStatus();

            EditingProviderManager editingProviderManager =

                EditingProviderLocator.getProviderManager();

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

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

            FeatureSelection featureSelection =

                (FeatureSelection) values.get(selection);

            DisposableIterator it = null;

            try {

                EditingProviderServices editingProviderService =

                    (EditingProviderServices) getProviderServices();

                int subtype = editingProviderService.getSubType(featureStore);

                it = featureSelection.fastIterator();

                while (it.hasNext()) {

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

                    Geometry lineToSplit =

                        getLineToSplit(feature.getDefaultGeometry());

                    Point projectedPoint =

                        getNearestPoint(lineToSplit, mousePosition);

                    Line line =

                        editingProviderService.createLine(mousePosition,

                            projectedPoint, subtype);

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

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

                }

            } catch (Exception e) {

                throw new DrawServiceException(e);

            } finally {

                it.dispose();

            }

            return drawingStatus;

        }

        return null;

    }

    private Geometry getLineToSplit(Geometry geometry) {

        Geometry line = null;

        GeometryType geoType = geometry.getGeometryType();

        if (geoType.isTypeOf(Geometry.TYPES.CURVE)) {

            line = (Curve) geometry;

        } else if (geoType.isTypeOf(Geometry.TYPES.MULTICURVE)) {

            line = (MultiCurve) geometry;

        }

        return line;

    }

    private Point getNearestPoint(Geometry lineToSplit, Point point)

        throws GeometryOperationNotSupportedException,

        GeometryOperationException {

        return (Point) lineToSplit.closestPoints(point)[0];

    }

    public void stop() throws StopServiceException {

        if (values != null) {

            values.clear();

        }

    }

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> parameters =

            new ArrayList<EditingServiceParameter>();

        parameters.add(selection);

        parameters.add(point);

        return parameters;

    }

    public void setValue(Object value) throws InvalidEntryException {

        EditingServiceParameter parameter = next();

        validateAndInsertValue(parameter, value);

    }

    private void validateAndInsertValue(EditingServiceParameter parameter,

        final Object value) throws InvalidEntryException {

        if (values != null && value != null) {

            if (value instanceof FeatureSelection) {

                FeatureSelection feautureSelection = (FeatureSelection) value;

                if (feautureSelection.getSelectedCount() == 1) {

                    try {

                        feautureSelection.accept(new Visitor() {

                            public void visit(Object obj)

                                throws VisitCanceledException, BaseException {

                                Feature feature = (Feature) obj;

                                Geometry geometry =

                                    feature.getDefaultGeometry();

                                GeometryType geoType =

                                    geometry.getGeometryType();

                                if (geoType.isTypeOf(Geometry.TYPES.CURVE)) {

                                    if (isClosed((Curve) geometry)) {

                                        throw new InvalidEntryException(null);

                                    } else {

                                        values.put(selection, value);

                                        return;

                                    }

                                } else if (geoType

                                    .isTypeOf(Geometry.TYPES.MULTICURVE)) {

                                    MultiCurve multiCurve =

                                        (MultiCurve) geometry;

                                    for (int i = 0; i < multiCurve

                                        .getPrimitivesNumber(); i++) {

                                        if (isClosed((Curve) multiCurve

                                            .getPrimitiveAt(i))) {

                                            throw new InvalidEntryException(

                                                null);

                                        }

                                    }

                                    values.put(selection, value);

                                    return;

                                }

                            }

                        });

                        return;

                    } catch (BaseException e) {

                        throw new InvalidEntryException(e);

                    }

                }

            } else if (value instanceof Point) {

                values.put(parameter, value);

                return;

            }

        }

        throw new InvalidEntryException(null);

    }

    private boolean isClosed(Curve curve) {

        Point firstPoint = curve.getVertex(0);

        Point lastPoint = curve.getVertex(curve.getNumVertices() - 1);

        if (firstPoint.equals(lastPoint)) {

            return true;

        }

        return false;

    }

    public Geometry finish() throws FinishServiceException {

        return null;

    }

    public void finishAndStore() throws FinishServiceException {

        if (values != null) {

            final EditingProviderServices editingProviderService =

                (EditingProviderServices) getProviderServices();

            FeatureSelection featureSelection =

                (FeatureSelection) values.get(selection);

            final Point pointValue = (Point) values.get(point);

            try {

                featureSelection.accept(new Visitor() {

                    public void visit(Object obj)

                        throws VisitCanceledException, BaseException {

                        Feature feature = (Feature) obj;

                        Geometry lineToSplit =

                            getLineToSplit(feature.getDefaultGeometry());

                        GeometryType geomType = lineToSplit.getGeometryType();

                        Point projectedPoint =

                            getNearestPoint(lineToSplit, pointValue);

                        if (geomType.isTypeOf(Geometry.TYPES.CURVE)) {

                            Curve[] splittedCurves =

                                splitCurve((Curve) lineToSplit, projectedPoint);

                            // Deletes old feature

                            editingProviderService

                                .deleteFeatureFromFeatureStore(feature,

                                    featureStore);

                            // Copy alfanumeric data and insert new feature

                            EditableFeature eFeature =

                                editingProviderService.getFeatureCopyWithoutPK(

                                    featureStore, feature);

                            eFeature.setDefaultGeometry(splittedCurves[0]);

                            editingProviderService

                                .insertFeatureIntoFeatureStore(eFeature,

                                    featureStore);

                            eFeature =

                                editingProviderService.getFeatureCopyWithoutPK(

                                    featureStore, feature);

                            eFeature.setDefaultGeometry(splittedCurves[1]);

                            editingProviderService

                                .insertFeatureIntoFeatureStore(eFeature,

                                    featureStore);

                        } else if (geomType.isTypeOf(Geometry.TYPES.MULTICURVE)) {

                            MultiCurve multiCurve = (MultiCurve) lineToSplit;

                            List<Geometry> result = splitMultiCurve(multiCurve, projectedPoint);

                            if (result != null && result.size() > 0) {

                                // Deletes old feature

                                editingProviderService

                                    .deleteFeatureFromFeatureStore(feature,

                                        featureStore);

                                for (Geometry geometry : result) {

                                    // Copy alfanumeric data and insert new feature

                                    EditableFeature eFeature =

                                        editingProviderService.getFeatureCopyWithoutPK(

                                            featureStore, feature);

                                    eFeature.setDefaultGeometry(geometry);

                                    editingProviderService

                                        .insertFeatureIntoFeatureStore(eFeature,

                                            featureStore);

                                }

                            }

                        }

                    }

                });

                featureStore.getFeatureSelection().deselectAll();

            } catch (BaseException e) {

                throw new FinishServiceException(e);

            }

        }

    }

    private List<Geometry> splitMultiCurve(MultiCurve multiCurveToSplit,

        Point projectedPoint) throws GeometryOperationNotSupportedException,

        GeometryOperationException, CreateGeometryException, LocatorException,

        DataException {

        List<Geometry> result = new ArrayList<Geometry>();

        Curve curveToSplit = null;

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

            if (intersects(multiCurveToSplit.getCurveAt(i), projectedPoint)) {

                curveToSplit = multiCurveToSplit.getCurveAt(i);

            }

        }

        Curve[] splitedCurves = splitCurve(curveToSplit, projectedPoint);

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

            result.add(splitedCurves[i]);

        }

        EditingProviderServices editingProviderService =

            (EditingProviderServices) getProviderServices();

        int subtype = editingProviderService.getSubType(featureStore);

        MultiCurve multiCurve =

            GeometryLocator.getGeometryManager().createMultiCurve(subtype);

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

            if (!intersects(multiCurveToSplit.getCurveAt(i), projectedPoint)) {

                multiCurve.addCurve(multiCurveToSplit.getCurveAt(i));

            }

        }

        if(multiCurve.getPrimitivesNumber() > 0){

            result.add(multiCurve);

        }

        return result;

    }

    private Curve[] splitCurve(Curve curveToSplit, Point projectedPoint)

        throws DataException, CreateGeometryException, LocatorException,

        GeometryOperationNotSupportedException, GeometryOperationException {

        SplitLineOperation operation =

            SplitLineOperationUtils.getOperation((Primitive) curveToSplit);

        if (operation != null) {

            return operation.split(curveToSplit, projectedPoint);

        }

        return null;

    }

    /**

     * Method use to know what segment intersect with projected point. Due to

     * accuracy of doubles it is necessary to create a buffer of line to know if

     * it intersects.

     *

     * @param curve

     *            segment of line

     * @param projectedPoint

     *            of line

     * @return true if segment intersects with projected point, else false.

     */

    private boolean intersects(Curve curve, Point projectedPoint)

        throws GeometryOperationNotSupportedException,

        GeometryOperationException {

        double tolerance = 1;

        return curve.buffer(tolerance).intersects(projectedPoint);

    }

    public void start() throws StartServiceException, InvalidEntryException {

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

        FeatureSelection selected = null;

        if (featureStore != null) {

            try {

                selected = featureStore.getFeatureSelection();

            } catch (DataException e) {

                throw new StartServiceException(e);

            }

            if (selected.getSelectedCount() > 0) {

                setValue(selected);

            }

        }

    }

    public String getName() {

        return SplitLineEditingProviderFactory.PROVIDER_NAME;

    }

}