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

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

import org.gvsig.fmap.geom.GeometryLocator;

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

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

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

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

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.spi.AbstractEditingProvider;

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

/**

 * @author llmarques

 *

 */

public class SimplifyEditingProvider extends AbstractEditingProvider implements

    EditingProvider {

    private EditingServiceParameter selection;

    private EditingServiceParameter tolerance;

    private FeatureStore featureStore;

    private Map<EditingServiceParameter, Object> values;

    /**

     * Default constructor.

     *

     * @param providerServices

     *            available services for this provider

     * @param parameters

     *            of this provider

     */

    public SimplifyEditingProvider(DynObject parameters,

        ProviderServices services) {

        super(services);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.selection =

            new DefaultEditingServiceParameter("selection", "selection",

                TYPE.SELECTION);

        this.tolerance =

            new DefaultEditingServiceParameter("tolerance", "tolerance",

                TYPE.VALUE);

    }

    public EditingServiceParameter next() {

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

            return selection;

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

            return tolerance;

        }

        return null;

    }

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        return null;

    }

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

        return parameters;

    }

    public void setValue(Object value) throws InvalidEntryException {

        EditingServiceParameter parameter = next();

        validateAndInsertValue(parameter, value);

    }

    private void validateAndInsertValue(EditingServiceParameter parameter,

        Object value) throws InvalidEntryException {

        if (parameter == selection) {

            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 geometryType =

                                geometry.getGeometryType();

                            if (!geometryType

                                .isTypeOf(Geometry.TYPES.MULTISURFACE)

                                && !geometryType

                                    .isTypeOf(Geometry.TYPES.SURFACE)

                                && !geometryType.isTypeOf(Geometry.TYPES.CURVE)

                                && !geometryType

                                    .isTypeOf(Geometry.TYPES.MULTICURVE)) {

                                throw new InvalidEntryException(null);

                            }

                        }

                    });

                    values.put(selection, value);

                    return;

                } catch (BaseException e) {

                    throw new InvalidEntryException(e);

                }

            }

        } else if (parameter == tolerance) {

            if (value instanceof Double) {

                Double toleranceValue = (Double) value;

                if (toleranceValue >= 0) {

                    values.put(tolerance, value);

                    return;

                }

            }

        }

        throw new InvalidEntryException(null);

    }

    public Geometry finish() throws FinishServiceException {

        return null;

    }

    public void finishAndStore() throws FinishServiceException {

        if (values != null) {

            FeatureSelection featureSelection =

                (FeatureSelection) values.get(selection);

            final Double toleranceValue = (Double) values.get(tolerance);

            final EditingProviderServices editingProviderServices =

                (EditingProviderServices) getProviderServices();

            try {

                final int subtype =

                    editingProviderServices.getSubType(featureStore);

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

                            || (geoType.isTypeOf(SURFACE))) {

                            if (geoType.isTypeOf(CURVE)) {

                                Curve curveToSimplify = (Curve) geometry;

                                curveToSimplify =

                                    simplifyLine(curveToSimplify,

                                        toleranceValue);

                                geometry = curveToSimplify;

                            } else {

                                Surface surfaceToSimplify = (Surface) geometry;

                                surfaceToSimplify =

                                    simplifySurface(surfaceToSimplify,

                                        toleranceValue);

                                geometry = surfaceToSimplify;

                            }

                            EditableFeature eFeature = feature.getEditable();

                            eFeature.setDefaultGeometry(geometry);

                            editingProviderServices

                                .updateFeatureInFeatureStore(eFeature,

                                    featureStore);

                        } else if (geoType.isTypeOf(MULTICURVE)

                            || geoType.isTypeOf(MULTISURFACE)) {

                            if (geoType.isTypeOf(MULTICURVE)) {

                                MultiCurve multiCurveToSimplify =

                                    (MultiCurve) geometry;

                                MultiCurve multiCurveSimplified =

                                    GeometryLocator.getGeometryManager()

                                        .createMultiCurve(subtype);

                                for (int i = 0; i < multiCurveToSimplify

                                    .getPrimitivesNumber(); i++) {

                                    Curve curveToSimplify =

                                        (Curve) multiCurveToSimplify

                                            .getPrimitiveAt(i);

                                    curveToSimplify =

                                        simplifyLine(curveToSimplify,

                                            toleranceValue);

                                    multiCurveSimplified

                                        .addCurve(curveToSimplify);

                                    geometry = multiCurveSimplified;

                                }

                            } else {

                                MultiSurface multiSurfaceToSimplify =

                                    (MultiSurface) geometry;

                                MultiSurface multiSurfaceSimplified =

                                    GeometryLocator.getGeometryManager()

                                        .createMultiSurface(subtype);

                                for (int i = 0; i < multiSurfaceToSimplify

                                    .getPrimitivesNumber(); i++) {

                                    Surface surfaceToSimplify =

                                        (Surface) multiSurfaceToSimplify

                                            .getPrimitiveAt(i);

                                    surfaceToSimplify =

                                        simplifySurface(surfaceToSimplify,

                                            toleranceValue);

                                    multiSurfaceSimplified

                                        .addSurface(surfaceToSimplify);

                                    geometry = multiSurfaceSimplified;

                                }

                            }

                            EditableFeature eFeature = feature.getEditable();

                            eFeature.setDefaultGeometry(geometry);

                            editingProviderServices

                                .updateFeatureInFeatureStore(eFeature,

                                    featureStore);

                        }

                    }

                });

                featureStore.getFeatureSelection().deselectAll();

            } catch (BaseException e) {

                throw new FinishServiceException(e);

            }

        }

    }

    private Surface simplifySurface(Surface surfaceToSimplify,

        Double toleranceValue) throws GeometryException, DataException,

        GeometryOperationNotSupportedException, GeometryOperationException {

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        int subtype = editingProviderServices.getSubType(featureStore);

        Line lineToSimplify =

            GeometryLocator.getGeometryManager().createLine(subtype);

        for (int i = 0; i < surfaceToSimplify.getNumVertices(); i++) {

            lineToSimplify.addVertex(surfaceToSimplify.getVertex(i));

        }

        Line simplifiedLine = simplifyLine(lineToSimplify, toleranceValue);

        Polygon simplifiedPolygon =

            GeometryLocator.getGeometryManager().createPolygon(subtype);

        for (int i = 0; i < simplifiedLine.getNumVertices(); i++) {

            simplifiedPolygon.addVertex(simplifiedLine.getVertex(i));

        }

        // Close polygon

        simplifiedPolygon.addVertex(simplifiedLine.getVertex(0));

        return simplifiedPolygon;

    }

    private Line simplifyLine(Curve curveToSimplify, Double toleranceValue)

        throws DataException, CreateGeometryException,

        GeometryOperationNotSupportedException, GeometryOperationException {

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        int subtype = editingProviderServices.getSubType(featureStore);

        Line simplifiedLine =

            (Line) GeometryLocator.getGeometryManager().createLine(subtype);

        // Add first vertex

        simplifiedLine.addVertex(curveToSimplify.getVertex(0));

        // Iterate over vertices

        for (int i = 0; i < curveToSimplify.getNumVertices() - 2; i++) {

            Point vertex = curveToSimplify.getVertex(i);

            Point nextVertex = curveToSimplify.getVertex(i + 1);

            Point nextNextVertex = curveToSimplify.getVertex(i + 2);

            // Creates line between i vertex and i+2 vertex

            Line tmpLine =

                editingProviderServices.createLine(vertex, nextNextVertex,

                    subtype);

            // If distance between line and i+1 vertex is bigger than tolerance

            // add point to simplifiedLine, else skip vertex

            if (tmpLine.distance(nextVertex) > toleranceValue) {

                simplifiedLine.addVertex(nextVertex);

            } else {

                i++;

            }

        }

        // Add last vertex

        simplifiedLine.addVertex(curveToSimplify.getVertex(curveToSimplify

            .getNumVertices() - 1));

        return simplifiedLine;

    }

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

                try {

                    setValue(selected);

                } catch (InvalidEntryException e) {

                    throw new InvalidEntryException(e);

                }

            }

        }

    }

    public String getName() {

        return SimplifyEditingProviderFactory.PROVIDER_NAME;

    }

}