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

import java.util.ArrayList;

import java.util.List;

import org.cresques.cts.IProjection;

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

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

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

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.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.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.layers.vectorial.IntersectsGeometryEvaluator;

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.extendline.operation.ExtendLineOperation;

import org.gvsig.vectorediting.lib.prov.extendline.operation.ExtendLineOperationUtils;

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;

/**

 * @author llmarques

 *

 */

public class ExtendLineEditingProvider extends AbstractEditingProvider

    implements EditingProvider {

    private final int TOLERANCE_PIXELS = 3;

    private Point insertedPoint;

    private EditingServiceParameter selectionParameter;

    private EditingServiceParameter linesToExtendParameter;

    private FeatureSelection selection;

    private FeatureSet linesToExtend;

    private FeatureStore featureStore;

    private MapContext mapContext;

    /**

     * Default constructor.

     *

     * @param providerServices

     *            available services for this provider

     * @param parameters

     * @param parameters

     *            of this provider

     */

    public ExtendLineEditingProvider(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.linesToExtendParameter =

            new DefaultEditingServiceParameter("line_to_extend",

                "line_to_extend", TYPE.POSITION);

    }

    public EditingServiceParameter next() {

        if (selection == null) {

            return selectionParameter;

        } else if (linesToExtend == null) {

            return linesToExtendParameter;

        }

        return null;

    }

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        if (selection != null) {

            DefaultDrawingStatus drawingStatus = new DefaultDrawingStatus();

            double tolerance =

                mapContext.getViewPort().toMapDistance(TOLERANCE_PIXELS);

            Geometry buffer;

            FeatureSet geometries;

            DisposableIterator it = null;

            try {

                // Create buffer with tolerance to get the geometry to be

                // extended

                buffer = mousePosition.buffer(tolerance);

                geometries = getGeometryByBuffer(buffer);

                it = geometries.fastIterator();

                while (it.hasNext()) {

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

                    if (!selection.isSelected(feature)) {

                        Geometry geometry = feature.getDefaultGeometry();

                        if (geometry instanceof MultiCurve) {

                            MultiCurve multiCurve = (MultiCurve) geometry;

                            for (int i = 0; i < multiCurve

                                .getPrimitivesNumber(); i++) {

                                Curve curve = multiCurve.getCurveAt(i);

                                if (!isClosed(curve)) {

                                    Curve extendedCurve =

                                        extendLine(mousePosition, curve);

                                    drawingStatus.addGeometry(extendedCurve);

                                }

                            }

                        } else if (geometry instanceof Curve) {

                            if (!isClosed((Curve) geometry)) {

                                Curve extendedCurve =

                                    extendLine(mousePosition, (Curve) geometry);

                                drawingStatus.addGeometry(extendedCurve);

                            }

                        }

                    }

                }

            } catch (BaseException e) {

                throw new DrawServiceException(e);

            } finally {

                it.dispose();

            }

            return drawingStatus;

        }

        return null;

    }

    private boolean isClosed(Curve curve) {

        if (curve instanceof Arc) {

            Arc arc = (Arc) curve;

            if (arc.getInitPoint().equals(arc.getEndPoint())) {

                return true;

            }

        } else if (curve instanceof Curve) {

            Point firstPoint = curve.getVertex(0);

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

            if (firstPoint.equals(lastPoint)) {

                return true;

            }

        }

        return false;

    }

    private Curve extendLine(Point mousePosition, Curve curve)

        throws GeometryOperationNotSupportedException,

        GeometryOperationException, DataException, CreateGeometryException {

        ExtendLineOperation operation =

            ExtendLineOperationUtils.getOperation((Primitive) curve);

        return operation.extendLine(curve, mousePosition, selection);

    }

    public void stop() throws StopServiceException {

        selection = null;

        linesToExtend = null;

    }

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> parameters =

            new ArrayList<EditingServiceParameter>();

        parameters.add(selectionParameter);

        parameters.add(linesToExtendParameter);

        return parameters;

    }

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

                                GeometryType geoType =

                                    geometry.getGeometryType();

                                if (!geoType.isTypeOf(SURFACE)

                                    && !geoType.isTypeOf(MULTISURFACE)

                                    && !geoType.isTypeOf(CURVE)

                                    && !geoType.isTypeOf(MULTICURVE)) {

                                    throw new InvalidEntryException(null);

                                }

                            }

                        });

                        selection = featureSelection;

                        return;

                    } catch (BaseException e) {

                        throw new InvalidEntryException(e);

                    }

                }

            }

        } else if (parameter == linesToExtendParameter) {

            if (value instanceof Point) {

                Point point = (Point) value;

                try {

                    double tolerance =

                        mapContext.getViewPort()

                            .toMapDistance(TOLERANCE_PIXELS);

                    // Create buffer with tolerance to get geometries to be

                    // extended

                    Geometry buffer = point.buffer(tolerance);

                    FeatureSet geometries = getGeometryByBuffer(buffer);

                    if (geometries.getSize() > 0) {

                        geometries.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(CURVE)

                                    && !geoType.isTypeOf(MULTICURVE)) {

                                    throw new InvalidEntryException(null);

                                }

                            }

                        });

                        insertedPoint = point;

                        linesToExtend = geometries;

                        return;

                    }

                } catch (BaseException e) {

                    throw new InvalidEntryException(e);

                }

            }

        }

        throw new InvalidEntryException(null);

    }

    public FeatureSet getGeometryByBuffer(Geometry buffer) throws DataException {

        FeatureQuery queryByGeometry = featureStore.createFeatureQuery();

        // Get default SRS of default feature type

        IProjection defaultSRS =

            this.featureStore.getDefaultFeatureType().getDefaultSRS();

        // Name of default geometry type

        String geomName =

            featureStore.getDefaultFeatureType()

                .getDefaultGeometryAttributeName();

        IntersectsGeometryEvaluator iee =

            new IntersectsGeometryEvaluator(buffer, defaultSRS,

                featureStore.getDefaultFeatureType(), geomName);

        queryByGeometry.setFilter(iee);

        queryByGeometry.setAttributeNames(null);

        return this.featureStore.getFeatureSet(queryByGeometry);

    }

    public Geometry finish() throws FinishServiceException {

        return null;

    }

    public void finishAndStore() throws FinishServiceException {

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

            try {

                linesToExtend.accept(new Visitor() {

                    public void visit(Object obj)

                        throws VisitCanceledException, BaseException {

                        Feature feature = (Feature) obj;

                        Geometry geometry = feature.getDefaultGeometry();

                        EditableFeature eFeature = feature.getEditable();

                        if (geometry instanceof MultiCurve) {

                            // Create new multicurve geometry

                            MultiCurve extendedMultiCurve =

                                createMultiCurve(geometry);

                            // Iterate overs primitives and execute extend

                            // line operation

                            MultiCurve multiCurve = (MultiCurve) geometry;

                            for (int i = 0; i < multiCurve

                                .getPrimitivesNumber(); i++) {

                                Curve curve = multiCurve.getCurveAt(i);

                                Curve extendedCurve =

                                    extendLine(insertedPoint, curve);

                                extendedMultiCurve.addCurve(extendedCurve);

                            }

                            eFeature.setDefaultGeometry(extendedMultiCurve);

                        } else if (geometry instanceof Curve) {

                            Curve extendedCurve =

                                extendLine(insertedPoint, (Curve) geometry);

                            eFeature.setDefaultGeometry(extendedCurve);

                        }

                        linesToExtend.update(eFeature);

                    }

                });

                linesToExtend.dispose();

            } catch (BaseException e) {

                throw new FinishServiceException(e);

            }

        }

    }

    private MultiCurve createMultiCurve(Geometry geometry)

        throws FinishServiceException {

        GeometryManager geoManager = GeometryLocator.getGeometryManager();

        int subtype = geometry.getGeometryType().getSubType();

        MultiCurve extendedMultiCurve = null;

        try {

            extendedMultiCurve = geoManager.createMultiCurve(subtype);

        } catch (CreateGeometryException e) {

            throw new FinishServiceException(e);

        }

        return extendedMultiCurve;

    }

    public void start() throws StartServiceException, InvalidEntryException {

        selection = null;

        linesToExtend = null;

        FeatureSelection selected = null;

        if (featureStore != 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);

                }

            }

        }

    }

    public String getName() {

        return ExtendLineEditingProviderFactory.PROVIDER_NAME;

    }

}