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

import java.util.ArrayList;

import java.util.HashMap;

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.aggregate.MultiCurve;

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

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

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

import org.gvsig.tools.dynobject.DynObject;

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

/**

 * @author llmarques

 *

 */

public class ArcEditingProvider extends AbstractEditingProvider implements

    EditingProvider {

    private EditingServiceParameter startPoint;

    private EditingServiceParameter endPoint;

    private EditingServiceParameter middlePoint;

    private Map<EditingServiceParameter, Object> values;

    private FeatureStore featureStore;

    /**

     * Default constructor.

     * 

     * @param providerServices

     *            available services for this provider

     * @param parameters

     *            of this provider

     */

    public ArcEditingProvider(ProviderServices providerServices,

        DynObject parameters) {

        super(providerServices);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.startPoint =

            new DefaultEditingServiceParameter("Start point", "start_point",

                TYPE.POSITION);

        this.endPoint =

            new DefaultEditingServiceParameter("End point", "end_point",

                TYPE.POSITION);

        this.middlePoint =

            new DefaultEditingServiceParameter("Middle point", "middle_point",

                TYPE.POSITION);

    }

    /*

     * (non-Javadoc)

     * 

     * @see org.gvsig.vectorediting.lib.spi.EditingProvider#next()

     */

    public EditingServiceParameter next() {

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

            return this.startPoint;

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

            return this.endPoint;

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

            return this.middlePoint;

        }

        return null;

    }

    /*

     * (non-Javadoc)

     * 

     * @see

     * org.gvsig.vectorediting.lib.spi.EditingProvider#getDrawingStatus(org.

     * gvsig.fmap.geom.primitive.Point)

     */

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        DefaultDrawingStatus geometries = new DefaultDrawingStatus();

        if (values != null && values.get(startPoint) != null) {

            Point startPointValue = (Point) values.get(startPoint);

            EditingProviderServices editingProviderServices =

                (EditingProviderServices) getProviderServices();

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

                try {

                    int subtype =

                        editingProviderServices.getSubType(featureStore);

                    Curve line =

                        editingProviderServices.createLine(startPointValue,

                            mousePosition, subtype);

                    geometries.addGeometry(line);

                } catch (Exception e) {

                    throw new DrawServiceException(e);

                }

                return geometries;

            } else {

                Point endPointValue = (Point) values.get(endPoint);

                try {

                    int subtype =

                        editingProviderServices.getSubType(featureStore);

                    Curve arc =

                        editingProviderServices.createArc(startPointValue,

                            mousePosition, endPointValue, subtype);

                    geometries.addGeometry(arc);

                    Point center =

                        editingProviderServices.getCenter(startPointValue,

                            mousePosition, endPointValue, subtype);

                    geometries.addGeometry(center);

                    geometries.addGeometry(startPointValue);

                    geometries.addGeometry(endPointValue);

                } catch (Exception e) {

                    throw new DrawServiceException(e);

                }

                return geometries;

            }

        }

        return null;

    }

    /*

     * (non-Javadoc)

     * 

     * @see org.gvsig.vectorediting.lib.spi.EditingProvider#stop()

     */

    public void stop() throws StopServiceException {

        values.clear();

    }

    /*

     * (non-Javadoc)

     * 

     * @see org.gvsig.vectorediting.lib.spi.EditingProvider#getParameters()

     */

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> parameters =

            new ArrayList<EditingServiceParameter>();

        parameters.add(startPoint);

        parameters.add(endPoint);

        parameters.add(middlePoint);

        return parameters;

    }

    /*

     * (non-Javadoc)

     * 

     * @see

     * org.gvsig.vectorediting.lib.spi.EditingProvider#setValue(java.lang.Object

     * )

     */

    public void setValue(Object value) throws InvalidEntryException {

        EditingServiceParameter param = next();

        validateAndInsertValue(param, value);

    }

    private void validateAndInsertValue(EditingServiceParameter param,

        Object value) throws InvalidEntryException {

        if (param == startPoint) {

            if (value instanceof Point) {

                values.put(param, value);

                return;

            }

        } else if (param == endPoint) {

            if (value instanceof Point) {

                values.put(param, value);

                return;

            }

        } else if (param == middlePoint) {

            if (value instanceof Point) {

                values.put(param, value);

                return;

            }

        }

        throw new InvalidEntryException(null);

    }

    /*

     * (non-Javadoc)

     * 

     * @see org.gvsig.vectorediting.lib.spi.EditingProvider#finish()

     */

    public Geometry finish() throws FinishServiceException {

        Point startPointValue = (Point) values.get(startPoint);

        Point middlePointValue = (Point) values.get(middlePoint);

        Point endPointValue = (Point) values.get(endPoint);

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        try {

            GeometryType storeGeomType =

                editingProviderServices.getGeomType(featureStore);

            Geometry geometry = null;

            int subtype = editingProviderServices.getSubType(featureStore);

            geometry =

                editingProviderServices.createArc(startPointValue,

                    middlePointValue, endPointValue, subtype);

            if (storeGeomType.isTypeOf(MULTICURVE)) {

                MultiCurve multiCurve;

                multiCurve =

                    GeometryLocator.getGeometryManager().createMultiCurve(

                        storeGeomType.getSubType());

                multiCurve.addCurve((Curve) geometry);

                return multiCurve;

            }

            return geometry;

        } catch (Exception e) {

            throw new FinishServiceException(e);

        }

    }

    /*

     * (non-Javadoc)

     * 

     * @see org.gvsig.vectorediting.lib.spi.EditingProvider#finishAndStore()

     */

    public void finishAndStore() throws FinishServiceException {

        Geometry geometry = finish();

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        editingProviderServices.insertGeometryIntoFeatureStore(geometry,

            featureStore);

    }

    /*

     * (non-Javadoc)

     * 

     * @see org.gvsig.vectorediting.lib.spi.EditingProvider#start()

     */

    public void start() throws StartServiceException {

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

    }

    /*

     * (non-Javadoc)

     * 

     * @see org.gvsig.vectorediting.lib.spi.EditingProvider#getName()

     */

    public String getName() {

        return ArcEditingProviderFactory.PROVIDER_NAME;

    }

}