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

import java.util.ArrayList;

import java.util.HashMap;

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

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

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

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

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

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

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

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

import org.gvsig.tools.dispose.DisposableIterator;

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.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.api.exceptions.VectorEditingException;

import org.gvsig.vectorediting.lib.prov.split.operation.SplitOperation;

import org.gvsig.vectorediting.lib.prov.split.operation.SplitOperationUtils;

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 llmarques

 *

 */

public class SplitEditingProvider extends AbstractEditingProvider implements

    EditingProvider {

    private EditingServiceParameter selection;

    private EditingServiceParameter splitGeometry;

    private FeatureStore featureStore;

    private Map<EditingServiceParameter, Object> values;

    /**

     * Default constructor.

     *

     * @param providerServices

     *            available services for this provider

     * @param parameters

     *            of this provider

     */

    public SplitEditingProvider(ProviderServices services, DynObject parameters) {

        super(services);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.selection =

            new DefaultEditingServiceParameter("selection", "selection",

                EditingServiceParameter.TYPE.SELECTION);

        this.splitGeometry =

            new DefaultEditingServiceParameter("draw_geometry_to_split",

                "draw_geometry_to_split", Geometry.TYPES.CURVE,

                EditingServiceParameter.TYPE.GEOMETRY);

    }

    public EditingServiceParameter next() {

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

            return selection;

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

            return splitGeometry;

        }

        return null;

    }

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        return null;

    }

    public void stop() throws StopServiceException {

        if (values != null) {

            values.clear();

        }

    }

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> parameters =

            new ArrayList<EditingServiceParameter>();

        parameters.add(selection);

        parameters.add(splitGeometry);

        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 (values != null && value != null) {

            if (value instanceof FeatureSelection) {

                FeatureSelection featureSelection = (FeatureSelection) value;

                try {

                    featureSelection.accept(new Visitor() {

                        public void visit(Object obj)

                            throws VisitCanceledException, BaseException {

                            Feature feature = (Feature) obj;

                            Geometry geometry = feature.getDefaultGeometry();

                            GeometryType geoType = geometry.getGeometryType();

                            if (geoType.isTypeOf(POINT)

                                || geoType.isTypeOf(MULTIPOINT)) {

                                throw new InvalidEntryException(null);

                            }

                        }

                    });

                    values.put(selection, value);

                    return;

                } catch (BaseException e) {

                    throw new InvalidEntryException(e);

                }

            } else if (value instanceof Geometry) {

                Geometry geometry = (Geometry) value;

                if (geometry instanceof MultiCurve) {

                    MultiCurve multiCurve = (MultiCurve) geometry;

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

                        if (isValid((Curve) multiCurve.getPrimitiveAt(i))) {

                            values.put(splitGeometry, geometry);

                            return;

                        }

                    }

                } else if (geometry instanceof Curve) {

                    if (isValid((Curve) geometry)) {

                        values.put(splitGeometry, geometry);

                        return;

                    }

                }

            }

        }

        throw new InvalidEntryException(null);

    }

    private boolean isValid(final Curve curve) {

        FeatureSelection featureSelection =

            (FeatureSelection) values.get(selection);

        try {

            featureSelection.accept(new Visitor() {

                public void visit(Object obj) throws VisitCanceledException,

                    BaseException {

                    Feature feature = (Feature) obj;

                    Geometry geometry = feature.getDefaultGeometry();

                    Geometry intersection = geometry.intersection(curve);

                    if (intersection == null

                        || (!intersection.getGeometryType().isTypeOf(POINT)

                        && !intersection.getGeometryType().isTypeOf(MULTIPOINT)

                        && !intersection.getGeometryType().isTypeOf(CURVE)

                        && !intersection.getGeometryType().isTypeOf(MULTICURVE))) {

                        throw new VectorEditingException();

                    }

                }

            });

            return true;

        } catch (BaseException e) {

            return false;

        }

    }

    public Geometry finish() throws FinishServiceException {

        return null;

    }

    public void finishAndStore() throws FinishServiceException {

        if (values != null) {

            EditingProviderServices editingProviderServices =

                (EditingProviderServices) getProviderServices();

            FeatureSelection featureSelection =

                (FeatureSelection) values.get(selection);

            Geometry splitter = (Geometry) values.get(splitGeometry);

            DisposableIterator it = null;

            try {

                it = featureSelection.fastIterator();

                while (it.hasNext()) {

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

                    Geometry geometryToBeSplitted =

                        feature.getDefaultGeometry();

                    GeometryType geoType =

                        geometryToBeSplitted.getGeometryType();

                    if (geoType.isTypeOf(TYPES.SURFACE)

                        || geoType.isTypeOf(TYPES.CURVE)) {

                        MultiPrimitive splittedGeometries =

                            (MultiPrimitive) split(

                                (Primitive) geometryToBeSplitted, splitter);

                        for (int j = 0; j < splittedGeometries

                            .getPrimitivesNumber(); j++) {

                            copyAlfanumericDataAndInsert(

                                splittedGeometries.getPrimitiveAt(j), feature);

                        }

                    } else if (geoType.isTypeOf(TYPES.MULTISURFACE)

                        || geoType.isTypeOf(TYPES.MULTICURVE)) {

                        MultiPrimitive multiPrimitive =

                            (MultiPrimitive) geometryToBeSplitted;

                        // Multigeometry composed by no splitted geometries.

                        GeometryManager geoManager = GeometryLocator.getGeometryManager();

                        MultiPrimitive multiNoSplitted;

                        if(geoType.isTypeOf(TYPES.MULTISURFACE)){

                            multiNoSplitted =

                                geoManager.createMultiSurface(editingProviderServices

                                    .getSubType(featureStore));

                        } else {

                            multiNoSplitted =

                                geoManager.createMultiCurve(editingProviderServices

                                    .getSubType(featureStore));

                        }

                        for (int i = 0; i < multiPrimitive

                            .getPrimitivesNumber(); i++) {

                            Geometry geoSplitted =

                                split(multiPrimitive.getPrimitiveAt(i),

                                    splitter);

                            if (geoSplitted instanceof MultiPrimitive) {

                                // If the result of split is a multiprimitive,

                                // it has splitted

                                MultiPrimitive multiSplitted = (MultiPrimitive) geoSplitted;

                                // Splitted geometries should be saved as

                                // primitives

                                for (int j = 0; j < multiSplitted

                                    .getPrimitivesNumber(); j++) {

                                    copyAlfanumericDataAndInsert(

                                        multiSplitted.getPrimitiveAt(j),

                                        feature);

                                }

                            } else {

                                // If the result of split is a primitive, it has

                                // not splitted. Add it to mulriNoSplitted to save it later.

                                multiNoSplitted

                                    .addPrimitive((Primitive) geoSplitted);

                            }

                        }

                        if(multiNoSplitted.getPrimitivesNumber() > 0){

                            // Not splitted geometries should be saved as multigeometry

                            copyAlfanumericDataAndInsert(multiNoSplitted,

                                feature);

                        }

                    }

                    editingProviderServices.deleteFeatureFromFeatureStore(

                        feature, featureStore);

                }

                featureSelection.deselectAll();

                featureStore.getFeatureSelection().deselectAll();

            } catch (BaseException e) {

                throw new FinishServiceException(e);

            } finally {

                it.dispose();

            }

        }

    }

    private void copyAlfanumericDataAndInsert(Geometry geometry, Feature feature)

        throws DataException {

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        EditableFeature eFeature =

            editingProviderServices.getFeatureCopyWithoutPK(featureStore,

                feature);

        eFeature.setDefaultGeometry(geometry);

        editingProviderServices.insertFeatureIntoFeatureStore(eFeature,

            featureStore);

    }

    private Geometry split(Primitive geometry, Geometry splitter)

        throws CreateGeometryException, GeometryOperationNotSupportedException,

        GeometryOperationException {

        SplitOperation operation = SplitOperationUtils.getOperation(geometry);

        if (operation != null) {

            return operation.split(geometry, splitter);

        }

        return null;

    }

    public void start() throws StartServiceException, InvalidEntryException {

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

        FeatureSelection selected = null;

        if (featureStore != null) {

            try {

                selected = featureStore.getFeatureSelection();

            } catch (DataException e) {

                throw new StartServiceException(e);

            }

            if (selected.getSelectedCount() > 0) {

                setValue(selected);

            }

        }

    }

    public String getName() {

        return SplitEditingProviderFactory.PROVIDER_NAME;

    }

}