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

import java.util.ArrayList;

import java.util.HashMap;

import java.util.LinkedHashMap;

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

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

import org.gvsig.tools.ToolsLocator;

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

import org.gvsig.vectorediting.lib.spi.EditingProvider;

import org.gvsig.vectorediting.lib.spi.EditingProviderFactory;

import org.gvsig.vectorediting.lib.spi.EditingProviderServices;

/**

 * @author fdiaz

 *

 */

public class SmoothLineEditingProvider extends AbstractEditingProvider implements

    EditingProvider {

    private final EditingServiceParameter selection;

    private final EditingServiceParameter steps;

    private final EditingServiceParameter algorithm;

    private final FeatureStore featureStore;

    private Map<EditingServiceParameter, Object> values;

    protected Map<String, String> options;

    public static final String    NATURAL_CUBIC_SPLINES = "natural_cubic_splines";

    public static final String    BEZIER_CURVES         = "bezier_curves";

    public static final String    BSPLINES              = "b_splines";

    /**

     * Default constructor.

     *

     * @param services

     *            available services for this provider

     * @param parameters

     *            of this provider

     */

    public SmoothLineEditingProvider(DynObject parameters,

        ProviderServices services) {

        super(services);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.selection =

            new DefaultEditingServiceParameter("selection", "selection",

                TYPE.SELECTION);

        this.steps =

            new DefaultEditingServiceParameter("steps", "intermediate_steps_1_9",

                TYPE.VALUE);

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        options = new LinkedHashMap<>();

        options.put(i18nManager.getTranslation("key_natural_cubic_splines"), NATURAL_CUBIC_SPLINES);

        options.put(i18nManager.getTranslation("key_bezier_curves"),BEZIER_CURVES);

        options.put(i18nManager.getTranslation("key_b_splines"),BSPLINES);

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        String consoleMsg =

            editingProviderServices.makeConsoleMessage(null, options);

        this.algorithm =

            new DefaultEditingServiceParameter("algorithm", consoleMsg, options,

                EditingServiceParameter.TYPE.OPTION);

}

    @Override

    public EditingServiceParameter next() {

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

            return selection;

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

            return steps;

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

            return algorithm;

        }

        return null;

    }

    @Override

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        return null;

    }

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

        parameters.add(algorithm);

        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(EditingServiceParameter parameter,

        Object value) throws InvalidEntryException {

        if (parameter == selection) {

            FeatureSelection featureSelection = (FeatureSelection) value;

            if (featureSelection.getSelectedCount() > 0) {

                try {

                    featureSelection.accept((Object obj) -> {

                        Feature feature = (Feature) obj;

                        Geometry geometry = feature.getDefaultGeometry();

                        GeometryType geometryType =

                                geometry.getGeometryType();

                        if (!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 == steps) {

            if (value instanceof Double) {

                Integer stepsValue = ((Double) value).intValue();

                if (stepsValue > 0 && stepsValue < 10 ) {

                    values.put(steps, stepsValue);

                    return;

                }

            }

        }  else if (parameter == algorithm) {

            if (value instanceof String) {

                I18nManager i18nManager = ToolsLocator.getI18nManager();

                String option = (String) value;

                if (option.equalsIgnoreCase(i18nManager.getTranslation("key_natural_cubic_splines"))){

                    values.put(parameter, "key_natural_cubic_splines");

                    return;

                }

                if (option.equalsIgnoreCase(i18nManager.getTranslation("key_bezier_curves"))){

                    values.put(parameter, "key_bezier_curves");

                    return;

                }

                if (option.equalsIgnoreCase(i18nManager.getTranslation("key_b_splines"))){

                    values.put(parameter, "key_b_splines");

                    return;

                }

            }

            throw new InvalidEntryException(null);

        }

        throw new InvalidEntryException(null);

    }

    @Override

    public Geometry finish() throws FinishServiceException {

        return null;

    }

    @Override

    public void finishAndStore() throws FinishServiceException {

        if (values != null) {

            FeatureSelection featureSelection =

                (FeatureSelection) values.get(selection);

            ToolsLocator.getDisposableManager().bind(featureSelection);

            final Integer stepsValue = (Integer) values.get(steps);

            final String algorithmValue = (String)values.get(algorithm);

            final EditingProviderServices editingProviderServices =

                (EditingProviderServices) getProviderServices();

            try {

                final int subtype =

                    editingProviderServices.getSubType(featureStore);

                featureSelection.accept((Object obj) -> {

                    Feature feature = (Feature) obj;

                    Geometry geometry = feature.getDefaultGeometry();

                    GeometryType geoType = geometry.getGeometryType();

                    if (geoType.isTypeOf(AGGREGATE)) {

                        MultiCurve multiCurveToSmooth =

                                (MultiCurve) geometry;

                        MultiCurve multiCurveSmoothed =

                                GeometryLocator.getGeometryManager()

                                        .createMultiCurve(subtype);

                        for (int i = 0; i < multiCurveToSmooth

                                .getPrimitivesNumber(); i++) {

                            Curve curveToSmooth =

                                    (Curve) multiCurveToSmooth

                                            .getPrimitiveAt(i);

                            curveToSmooth =

                                    smoothLine(curveToSmooth, stepsValue, algorithmValue);

                            multiCurveSmoothed.addCurve(curveToSmooth);

                            geometry = multiCurveSmoothed;

                        }

                    } else {

                        Curve curveToSmooth = (Curve) geometry;

                        curveToSmooth =

                                smoothLine(curveToSmooth, stepsValue, algorithmValue);

                        geometry = curveToSmooth;

                    }

                    EditableFeature eFeature = feature.getEditable();

                    eFeature.setDefaultGeometry(geometry);

                    editingProviderServices.updateFeatureInFeatureStore(

                            eFeature, featureStore);

                });

                featureStore.getFeatureSelection().deselectAll();

                featureSelection.dispose();

            } catch (BaseException e) {

                throw new FinishServiceException(e);

            }

        }

    }

    private Line smoothLine(Curve curveToSmooth, Integer steps, String algorithm) throws DataException, CreateGeometryException{

        GeometryManager geometryManager = GeometryLocator.getGeometryManager();

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        int subtype = editingProviderServices.getSubType(featureStore);

        Line lineToSmooth;

        if (curveToSmooth instanceof Arc || curveToSmooth instanceof Spline) {

            lineToSmooth = geometryManager.createLine(curveToSmooth.getGeometryType().getSubType());

            lineToSmooth.setGeneralPath(curveToSmooth.getGeneralPath());

        } else {

            lineToSmooth = (Line)curveToSmooth;

        }

        ControlCurve operation;

        if (algorithm.equalsIgnoreCase("key_natural_cubic_splines")){

            operation = new NatCubic(lineToSmooth);

        } else if (algorithm.equalsIgnoreCase("key_bezier_curves")){

            operation = new Bezier(lineToSmooth);

        } else if (algorithm.equalsIgnoreCase("key_b_splines")){

            operation = new BSpline(lineToSmooth);

        } else {

            return null;

        }

        return operation.getSmoothedLine(steps, subtype);

    }

    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;

    }

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

            }

            if (selected.getSelectedCount() > 0) {

                try {

                    setValue(selected);

                } catch (InvalidEntryException e) {

                    throw new InvalidEntryException(e);

                }

            }

        }

    }

    @Override

    public String getName() {

        return SmoothLineEditingProviderFactory.PROVIDER_NAME;

    }

    @Override

    public Object getValue(EditingServiceParameter parameter) {

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

    }

}