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

import java.util.ArrayList;

import java.util.Iterator;

import java.util.LinkedHashMap;

import java.util.List;

import java.util.Map;

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.primitive.Point;

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

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

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

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.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.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 SplineEditingProvider extends AbstractEditingProvider implements

    EditingProvider {

    protected EditingServiceParameter points;

    protected List<Point> values;

    protected Map<String, String> options;

    protected boolean closeSpline;

    protected boolean finishSpline;

    protected FeatureStore featureStore;

    /**

     * Default constructor.

     *

     * @param parameters

     *            of service.

     * @param providerServices

     *            available services. See {@link EditingProviderServices}.

     */

    public SplineEditingProvider(ProviderServices services, DynObject parameters) {

        super(services);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        options = new LinkedHashMap<String, String>();

        options.put(i18nManager.getTranslation("key_close"), "close_spline");

        options.put(i18nManager.getTranslation("key_finish"), "finish");

        options.put(i18nManager.getTranslation("key_remove_last_point"), "remove_last_point");

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        String consoleMsg =

            editingProviderServices.makeConsoleMessage("indicate_new_point",

                options);

        this.points =

            new DefaultEditingServiceParameter("insert_point", consoleMsg,

                options, EditingServiceParameter.TYPE.LIST_POSITIONS,

                EditingServiceParameter.TYPE.OPTION);

        this.closeSpline = false;

        this.finishSpline = false;

    }

    public EditingServiceParameter next() {

        if (finishSpline) {

            return null;

        }

        return points;

    }

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        DefaultDrawingStatus drawingStatus = new DefaultDrawingStatus();

        EditingProviderManager editingProviderManager =

            EditingProviderLocator.getProviderManager();

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

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

        if (values != null) {

            if (values.size() >= 1) {

                for (Iterator iterator = values.iterator(); iterator.hasNext();) {

                    Point point = (Point) iterator.next();

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

                }

                EditingProviderServices editingProviderServices =

                    (EditingProviderServices) getProviderServices();

                try {

                    int subtype =

                        editingProviderServices.getSubType(featureStore);

                    Spline spline =

                        editingProviderServices.createSpline(values, subtype);

                    spline.addVertex(mousePosition);

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

                    drawingStatus.addStatus(spline, lineSymbolEditing, "");

//                    drawingStatus.addGeometry(spline);

                } catch (BaseException e) {

                    throw new DrawServiceException(e);

                }

            }

        }

        return drawingStatus;

    }

    public void stop() throws StopServiceException {

        values.clear();

    }

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> parameters =

            new ArrayList<EditingServiceParameter>();

        parameters.add(points);

        return parameters;

    }

    @Override

    public void setValue(EditingServiceParameter parameter, Object value) throws InvalidEntryException {

        validateAndInsertValue(parameter, value);

    }

    public void setValue(Object value) throws InvalidEntryException {

        EditingServiceParameter parameter = next();

        validateAndInsertValue(parameter, value);

    }

    /**

     * Checks if value is valid to parameter received as parameter.

     *

     * @param param

     *            of value

     * @param value

     *            to e inserted

     * @throws InvalidEntryException

     *             If values is no valid throw an InvalidEntryException

     */

    private void validateAndInsertValue(EditingServiceParameter parameter,

        Object value) throws InvalidEntryException {

        if (parameter == points) {

            if (value instanceof Point) {

                values.add((Point) value);

                return;

            } else if (value instanceof String) {

                String option = (String) value;

                I18nManager i18nManager = ToolsLocator.getI18nManager();

                if (values.size() > 0) {

                    if (option.equalsIgnoreCase(i18nManager

                        .getTranslation("key_remove_last_point"))) {

                        values.remove(values.size() - 1);

                        return;

                    }

                }

                if (values.size() > 1) {

                    if (option.equalsIgnoreCase(i18nManager

                        .getTranslation("key_close"))

                        || option.equalsIgnoreCase(i18nManager

                            .getTranslation("key_finish"))) {

                        if (option.equalsIgnoreCase(i18nManager

                            .getTranslation("key_close"))) {

                            this.closeSpline = true;

                        }

                        this.finishSpline = true;

                        return;

                    }

                }

            }

        }

        throw new InvalidEntryException(null);

    }

    public Geometry finish() throws FinishServiceException {

        if (values != null) {

            EditingProviderServices editingProviderServices =

                (EditingProviderServices) getProviderServices();

            try {

                if (closeSpline) {

                    values.add(values.get(0));

                }

                int subtype = editingProviderServices.getSubType(featureStore);

                Spline spline =

                    editingProviderServices.createSpline(values, subtype);

                GeometryType geomType =

                    editingProviderServices.getGeomType(featureStore);

                if (geomType.isTypeOf(MULTICURVE)) {

                    GeometryManager geometryManager =

                        GeometryLocator.getGeometryManager();

                    MultiCurve multiCurve =

                        geometryManager.createMultiCurve(geomType.getSubType());

                    multiCurve.addCurve(spline);

                    return multiCurve;

                } else {

                    return spline;

                }

            } catch (BaseException e) {

                throw new FinishServiceException(e);

            }

        }

        return null;

    }

    public void finishAndStore() throws FinishServiceException {

        Geometry geometry = this.finish();

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        editingProviderServices.insertGeometryIntoFeatureStore(geometry,

            featureStore);

    }

    public void start() throws StartServiceException {

        values = new ArrayList<Point>();

        closeSpline = false;

        finishSpline = false;

    }

    public String getName() {

        return SplineEditingProviderFactory.PROVIDER_NAME;

    }

}