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

import java.util.ArrayList;

import java.util.List;

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.Geometry.TYPES;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.exception.CreateGeometryException;

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.type.GeometryType;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposableIterator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.exception.BaseException;

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;

public class ConvertToPolylineEditingProvider extends AbstractEditingProvider

    implements EditingProvider {

    private final EditingServiceParameter selectionParameter;

    private FeatureSelection featureSelection;

    private final FeatureStore featureStore;

    public ConvertToPolylineEditingProvider(ProviderServices providerServices,

        DynObject parameters) {

        super(providerServices);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.selectionParameter =

            new DefaultEditingServiceParameter("selection", "selection",

                TYPE.SELECTION);

    }

    @Override

    public EditingServiceParameter next() {

        if (this.featureSelection == null) {

            return selectionParameter;

        }

        return null;

    }

    @Override

    public boolean mustRestartAtFinish() {

        return false;

    }

    @Override

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        return null;

    }

    @Override

    public void stop() throws StopServiceException {

        DisposeUtils.disposeQuietly(featureSelection);

        featureSelection = null;

    }

    @Override

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> parameters =

            new ArrayList<>();

        parameters.add(selectionParameter);

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

            if (value instanceof FeatureSelection) {

                FeatureSelection fSelection = (FeatureSelection) value;

                if (fSelection.getSelectedCount() >= 1) {

                    DisposableIterator it = null;

                    try {

                        it = fSelection.fastIterator();

                        while (it.hasNext()) {

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

                            // Gets geometry type of first selected feature

                            GeometryType geometryType =

                                feature.getDefaultGeometry().getGeometryType();

                            if (!geometryType.isTypeOf(TYPES.CURVE)

                                && !geometryType.isTypeOf(TYPES.MULTICURVE)) {

                                throw new InvalidEntryException(null);

                            }

                        }

                        DisposeUtils.disposeQuietly(featureSelection);

                        ToolsLocator.getDisposableManager().bind(fSelection);

                        featureSelection = fSelection;

                    } catch (DataException e) {

                        throw new InvalidEntryException(e);

                    } finally {

                        DisposeUtils.disposeQuietly(it);

                    }

                }

            }

        }

    }

    @Override

    public Geometry finish() throws FinishServiceException {

        return null;

    }

    @Override

    public void finishAndStore() throws FinishServiceException {

        if (featureSelection != null) {

            GeometryManager geometryManager =

                GeometryLocator.getGeometryManager();

            EditingProviderServices editingProviderServices =

                (EditingProviderServices) getProviderServices();

            DisposableIterator it = null;

            try {

                it = featureSelection.fastIterator();

                while (it.hasNext()) {

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

                    EditableFeature eFeature = feature.getEditable();

                    Geometry geometry = feature.getDefaultGeometry();

                    if (geometry.getGeometryType().isTypeOf(TYPES.CURVE)) {

                        eFeature.setDefaultGeometry(geometry.toLines().getPrimitiveAt(0));

                    } else if (geometry.getGeometryType().isTypeOf(TYPES.MULTICURVE)) {

                        eFeature.setDefaultGeometry(geometry.toLines());

                    }

                    DisposeUtils.disposeQuietly(featureSelection);

                    editingProviderServices.updateFeatureInFeatureStore(

                        eFeature, featureStore);

                }

                featureStore.getFeatureSelection().deselectAll();

            } catch (BaseException e) {

                throw new FinishServiceException(e);

            } finally {

                DisposeUtils.disposeQuietly(it);

            }

        }

    }

    private Line convertCurveToLine(Curve curve) throws CreateGeometryException {

        GeometryManager geometryManager = GeometryLocator.getGeometryManager();

        Line newLine =

            geometryManager.createLine(curve.getGeometryType().getSubType());

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

            newLine.addVertex(curve.getVertex(i));

        }

        return newLine;

    }

    @Override

    public void start() throws StartServiceException, InvalidEntryException {

        FeatureSelection selected = null;

        if (featureStore != null) {

            try {

                selected = featureStore.getFeatureSelection();

            } catch (DataException e) {

                throw new StartServiceException(e);

            }

            if (selected.getSelectedCount() >= 1) {

                try {

                    setValue(selected);

                } catch (InvalidEntryException e) {

                    throw new InvalidEntryException(e);

                }

            }

        }

    }

    @Override

    public String getName() {

        return ConvertToPolylineEditingProviderFactory.PROVIDER_NAME;

    }

    @Override

    public Object getValue(EditingServiceParameter parameter) {

        if(parameter == this.selectionParameter){

            return featureSelection;

        }

        return null;

    }

}