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

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

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

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

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

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

import org.gvsig.tools.ToolsLocator;

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.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 <a href="mailto:lmarques@disid.com">Lluis Marques</a>

 *

 */

public class ReverseDirectionEditingProvider extends AbstractEditingProvider

    implements EditingProvider {

    private EditingServiceParameter selectionParameter;

    private FeatureSelection selection;

    private FeatureStore featureStore;

    public ReverseDirectionEditingProvider(ProviderServices providerServices,

        DynObject parameters) {

        super(providerServices);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.selectionParameter =

            new DefaultEditingServiceParameter("selection", "selection",

                TYPE.SELECTION);

    }

    public EditingServiceParameter next() {

        if (selection == null) {

            return this.selectionParameter;

        }

        return null;

    }

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        return null;

    }

    public void stop() throws StopServiceException {

        ToolsLocator.getDisposableManager().release(selection);

        DisposeUtils.disposeQuietly(selection);

        this.selection = null;

    }

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> parameters =

            new ArrayList<EditingServiceParameter>();

        parameters.add(selectionParameter);

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

            if (value instanceof FeatureSelection) {

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

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

                                    throw new InvalidEntryException(null);

                                }

                            }

                        });

                    } catch (BaseException e) {

                        throw new InvalidEntryException(e);

                    }

                    ToolsLocator.getDisposableManager().bind(

                        (FeatureSelection) value);

                    this.selection = featureSelection;

                } else {

                    throw new InvalidEntryException(null);

                }

            }

        }

    }

    public Geometry finish() throws FinishServiceException {

        return null;

    }

    public void finishAndStore() throws FinishServiceException {

        if (selection != null) {

            final EditingProviderServices editingProviderServices =

                (EditingProviderServices) getProviderServices();

            try {

                selection.accept(new Visitor() {

                    public void visit(Object obj)

                        throws VisitCanceledException, BaseException {

                        Feature feature = (Feature) obj;

                        EditableFeature eFeature = feature.getEditable();

                        Geometry geometry = eFeature.getDefaultGeometry();

                        eFeature.setDefaultGeometry(reverseDirection(geometry));

                        editingProviderServices.updateFeatureInFeatureStore(

                            eFeature, featureStore);

                    }

                });

                ToolsLocator.getDisposableManager().release(selection);

                DisposeUtils.dispose(selection);

                featureStore.getFeatureSelection().deselectAll();

            } catch (BaseException e) {

                // TODO Auto-generated catch block

                e.printStackTrace();

            }

        }

    }

    private Geometry reverseDirection(Geometry geometry)

        throws GeometryException, GeometryOperationNotSupportedException, GeometryOperationException {

        GeometryManager geometryManager = GeometryLocator.getGeometryManager();

        GeometryType geometryType = geometry.getGeometryType();

        if (geometryType.isTypeOf(TYPES.MULTICURVE)) {

            MultiCurve reversedMultiCurve =

                geometryManager.createMultiCurve(geometryType.getSubType());

            MultiCurve multicurve = (MultiCurve) geometry;

            for (int i = 0; i < multicurve.getPrimitivesNumber(); i++) {

                reversedMultiCurve.addCurve(reverseCurve(multicurve.getCurveAt(i)));

            }

            return reversedMultiCurve;

        } else if (geometryType.isTypeOf(TYPES.CURVE)) {

            return reverseCurve((Curve) geometry);

        } else {

            throw new IllegalArgumentException(String.format(

                "Can't reverse geometry with {} type", geometryType.getType()));

        }

    }

    private Curve reverseCurve(Curve curve) throws GeometryException, GeometryOperationNotSupportedException, GeometryOperationException {

        GeometryManager geometryManager = GeometryLocator.getGeometryManager();

        Curve curveToReverse = null;

        Curve reverseCurve = null;

        if(curve instanceof Line || curve instanceof Spline){

            reverseCurve = (Curve) curve.cloneGeometry();

            reverseCurve.flip();

        } else {

            curveToReverse = (Curve) curve.toLines().getPrimitiveAt(0);

            reverseCurve = (Curve) geometryManager.create(Geometry.TYPES.LINE, curve.getGeometryType().getSubType());

            for (int i = curveToReverse.getNumVertices() - 1; i >= 0; i--) {

                reverseCurve.addVertex(curveToReverse.getVertex(i));

            }

        }

        return reverseCurve;

    }

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

                try {

                    setValue(selected);

                } catch (InvalidEntryException e) {

                    throw new InvalidEntryException(e);

                }

            }

        }

    }

    public String getName() {

        return ReverseDirectionEditingProviderFactory.PROVIDER_NAME;

    }

}