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

import java.util.ArrayList;

import java.util.Iterator;

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

import org.gvsig.fmap.dal.feature.FeatureSelection;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.FeatureType;

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

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

import org.gvsig.fmap.geom.complex.Complex;

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;

/**

 * @author llmarques

 *

 */

public class CombineEditingProvider extends AbstractEditingProvider implements EditingProvider {

    private static final int HAS_POINT = 1;

    private static final int HAS_CURVE = 2;

    private static final int HAS_SURFACE = 4;

    private final EditingServiceParameter selectionParameter;

    private final FeatureStore featureStore;

    private FeatureSelection featureSelection;

    /**

     * Default constructor.

     *

     * @param parameters

     *            of this provider

     * @param services

     */

    public CombineEditingProvider(DynObject parameters, ProviderServices services) {

        super(services);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.selectionParameter =

            new DefaultEditingServiceParameter("selection", "selection",

                TYPE.SELECTION);

    }

    @Override

    public EditingServiceParameter next() {

        if (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();

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

                        // Gets geometry type of first selected feature

                        GeometryType geometryType =

                            feature.getDefaultGeometry().getGeometryType();

                        // Iterate over remaining selected features

                        while (it.hasNext()) {

                            feature = (Feature) it.next();

                            // Remaining selected features must be the same or

                            // the equivalent multigeometry type of first

                            // feature. If one of reamining geometries is

                            // different, selection is not valid.

                            if (geometryType.isTypeOf(CURVE)

                                || geometryType.isTypeOf(MULTICURVE)) {

                                GeometryType featureGeoType =

                                    feature.getDefaultGeometry()

                                        .getGeometryType();

                                if (!(featureGeoType.isTypeOf(CURVE) || featureGeoType

                                    .isTypeOf(MULTICURVE))) {

                                    throw new InvalidEntryException(null);

                                }

                            } else if (geometryType.isTypeOf(SURFACE)

                                || geometryType.isTypeOf(SURFACE)) {

                                GeometryType featureGeoType =

                                    feature.getDefaultGeometry()

                                        .getGeometryType();

                                if (!(featureGeoType.isTypeOf(SURFACE) || featureGeoType

                                    .isTypeOf(MULTISURFACE))) {

                                    throw new InvalidEntryException(null);

                                }

                            }

                        }

                        DisposeUtils.disposeQuietly(featureSelection);

                        ToolsLocator.getDisposableManager().bind(fSelection);

                        featureSelection = fSelection;

                        return;

                    } catch (DataException e) {

                        throw new InvalidEntryException(e);

                    } finally {

                        DisposeUtils.disposeQuietly(it);

                    }

                }

            }

        }

        throw new InvalidEntryException(null);

    }

    @Override

    public Geometry finish() throws FinishServiceException {

        if (featureSelection != null) {

            Geometry combinedGeometry = null;

            DisposableIterator it = null;

            EditingProviderServices editingProviderServices

                    = (EditingProviderServices) getProviderServices();

            int resultType = 0;

            GeometryManager geometryManager = GeometryLocator.getGeometryManager();

            try {

                int subtype = editingProviderServices.getSubType(featureStore);

                MultiPoint multiPoint = geometryManager.createMultiPoint(subtype);

                MultiCurve multiCurve = geometryManager.createMultiCurve(subtype);

                MultiSurface multiSurface = geometryManager.createMultiSurface(subtype);

                it = featureSelection.fastIterator();

                while (it.hasNext()) {

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

                    Geometry geom = tmpFeature.getDefaultGeometry();

                    if(geom.getGeometryType().isTypeOf(POINT) ||

                            geom.getGeometryType().isTypeOf(MULTIPOINT) ) {

                        multiPoint.addPrimitives(geom);

                        resultType |= HAS_POINT;

                    } else if(geom.getGeometryType().isTypeOf(CURVE) ||

                            geom.getGeometryType().isTypeOf(MULTICURVE) ) {

                        multiCurve.addPrimitives(geom);

                        resultType |= HAS_CURVE;

                    } else if(geom.getGeometryType().isTypeOf(SURFACE) ||

                            geom.getGeometryType().isTypeOf(MULTISURFACE) ) {

                        multiSurface.addPrimitives(geom);

                        resultType |= HAS_SURFACE;

                    } else if(geom.getGeometryType().isTypeOf(COMPLEX)) {

                        Complex complex = (Complex)geom;

                        Iterator<Geometry> complexIt = complex.iterator();

                        while (complexIt.hasNext()) {

                            Geometry geom1 = complexIt.next();

                            if (geom1.getGeometryType().isTypeOf(POINT)

                                    || geom1.getGeometryType().isTypeOf(MULTIPOINT)) {

                                multiPoint.addPrimitives(geom1);

                                resultType |= HAS_POINT;

                            } else if (geom1.getGeometryType().isTypeOf(CURVE)

                                    || geom1.getGeometryType().isTypeOf(MULTICURVE)) {

                                multiCurve.addPrimitives(geom1);

                                resultType |= HAS_CURVE;

                            } else if (geom1.getGeometryType().isTypeOf(SURFACE)

                                    || geom1.getGeometryType().isTypeOf(MULTISURFACE)) {

                                multiSurface.addPrimitives(geom1);

                                resultType |= HAS_SURFACE;

                            }

                        }

                    }

                }

                GeometryType storeGeomType = editingProviderServices.getGeomType(featureStore);

                if(resultType > 4 || resultType == 3){ //COMPLEX

                    if(storeGeomType.getType()!=COMPLEX && storeGeomType.getType()!=GEOMETRY){

                        throw new FinishServiceException("_The_resulting_geometry_is_not_valid_for_the_store", null);

                    }

                    combinedGeometry = geometryManager.create(COMPLEX, subtype);

                    Complex complex = (Complex)combinedGeometry;

                    if((resultType & HAS_POINT) == HAS_POINT){

                        complex.addPrimitives(multiPoint.union());

                    } else if((resultType & HAS_CURVE) == HAS_CURVE){

                        complex.addPrimitives(multiCurve.dissolve());

                    } else if((resultType & HAS_SURFACE) == HAS_SURFACE){

                        complex.addPrimitives(multiSurface.union());

                    }

                } else {

                    switch (resultType){

                        case HAS_POINT:

                            if(!storeGeomType.isTypeOf(GEOMETRY) &&

                               !storeGeomType.isTypeOf(POINT) &&

                               !storeGeomType.isTypeOf(MULTIPOINT) ){

                               throw new FinishServiceException("The resulting geometry is not valid for the store.", "_The_resulting_geometry_is_not_valid_for_the_store", null);

                            }

                            combinedGeometry = multiPoint.union();

                            if(storeGeomType.isTypeOf(POINT) && combinedGeometry.getGeometryType().isTypeOf(MULTIPOINT)){

                               throw new FinishServiceException("The resulting geometry is not valid for the store.", "_The_resulting_geometry_is_not_valid_for_the_store", null);

                            }

                            if(storeGeomType.isTypeOf(MULTIPOINT) && combinedGeometry.getGeometryType().isTypeOf(POINT)){

                                MultiPoint tmpMultiPoint = geometryManager.createMultiPoint(subtype);

                                tmpMultiPoint.addPrimitives(combinedGeometry);

                                combinedGeometry = tmpMultiPoint;

                            }

                            break;

                        case HAS_CURVE:

                            if(!storeGeomType.isTypeOf(GEOMETRY) &&

                               !storeGeomType.isTypeOf(CURVE) &&

                               !storeGeomType.isTypeOf(MULTICURVE) ){

                               throw new FinishServiceException("The resulting geometry is not valid for the store.", "_The_resulting_geometry_is_not_valid_for_the_store", null);

                            }

                            combinedGeometry = multiCurve.dissolve();

                            if(storeGeomType.isTypeOf(CURVE) && combinedGeometry.getGeometryType().isTypeOf(MULTICURVE)){

                               throw new FinishServiceException("The resulting geometry is not valid for the store.", "_The_resulting_geometry_is_not_valid_for_the_store", null);

                            }

                            if(storeGeomType.isTypeOf(MULTICURVE) && combinedGeometry.getGeometryType().isTypeOf(CURVE)){

                                MultiCurve tmpMultiCurve = geometryManager.createMultiCurve(subtype);

                                tmpMultiCurve.addPrimitives(combinedGeometry);

                                combinedGeometry = tmpMultiCurve;

                            }

                            break;

                        case HAS_SURFACE:

                            if(!storeGeomType.isTypeOf(GEOMETRY) &&

                               !storeGeomType.isTypeOf(SURFACE) &&

                               !storeGeomType.isTypeOf(MULTISURFACE) ){

                               throw new FinishServiceException("The resulting geometry is not valid for the store.", "_The_resulting_geometry_is_not_valid_for_the_store", null);

                            }

                            combinedGeometry = multiSurface.union();

                            if(storeGeomType.isTypeOf(SURFACE) && combinedGeometry.getGeometryType().isTypeOf(MULTISURFACE)){

                               throw new FinishServiceException("The resulting geometry is not valid for the store.", "_The_resulting_geometry_is_not_valid_for_the_store", null);

                            }

                            if(storeGeomType.isTypeOf(MULTISURFACE) && combinedGeometry.getGeometryType().isTypeOf(SURFACE)){

                                MultiSurface tmpMultiSurface = geometryManager.createMultiSurface(subtype);

                                tmpMultiSurface.addPrimitives(combinedGeometry);

                                combinedGeometry = tmpMultiSurface;

                            }

                            break;

                    }

                }

            } catch (FinishServiceException e) {

                throw e;

            } catch (BaseException e) {

                throw new FinishServiceException(e);

            } finally {

                DisposeUtils.disposeQuietly(it);

            }

            return combinedGeometry;

        }

        return null;

    }

    @SuppressWarnings("rawtypes")

    @Override

    public void finishAndStore() throws FinishServiceException {

        Geometry geometry = finish();

        final EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        FeatureType featureType = null;

        EditableFeature eFeature = null;

        try {

            eFeature = featureStore.createNewFeature();

            featureType = featureStore.getDefaultFeatureType();

        } catch (DataException e) {

            throw new FinishServiceException(e);

        }

        Iterator typeIterator = featureType.iterator();

        // Iterate over feature type

        while (typeIterator.hasNext()) {

            boolean insertValue = true;

            FeatureAttributeDescriptor attribute =

                (FeatureAttributeDescriptor) typeIterator.next();

            // If type is a primary key do nothing

            if (!attribute.isPrimaryKey()) {

                // int type = attribute.getType();

                DisposableIterator featureIterator = null;

                try {

                    featureIterator = featureSelection.fastIterator();

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

                    // Restart iterator and initilize it at second position

                    featureIterator.dispose();

                    featureIterator = featureSelection.fastIterator(1);

                    // Gets the type value of first feature

                    Object value = feature.get(attribute.getName());

                    // Compare first value with the rest of features. If values

                    // are different, break bucle and don't insert value at new

                    // feature.

                    if (value != null) {

                        while (featureIterator.hasNext()) {

                            Feature nextFeature =

                                (Feature) featureIterator.next();

                            Object nextValue =

                                nextFeature.get(attribute.getName());

                            if (nextValue == null || !value.equals(nextValue)) {

                                insertValue = false;

                                break;

                            }

                        }

                        if (insertValue) {

                            eFeature.set(attribute.getName(), value);

                        }

                    }

                } catch (DataException e) {

                    throw new FinishServiceException(e);

                } finally {

                    DisposeUtils.disposeQuietly(featureIterator);

                }

            }

        }

        // Inserts new feature

        eFeature.setDefaultGeometry(geometry);

        editingProviderServices.insertFeatureIntoFeatureStore(eFeature,

            featureStore);

        // Deletes selected features

        try {

            featureSelection.accept((Object obj) -> {

                Feature feature = (Feature) obj;

                editingProviderServices.deleteFeatureFromFeatureStore(

                        feature, featureStore);

            });

            featureStore.getFeatureSelection().deselectAll();

        } catch (BaseException e) {

            throw new FinishServiceException(e);

        }

    }

    @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 CombineEditingProviderFactory.PROVIDER_NAME;

    }

    @Override

    public Object getValue(EditingServiceParameter parameter) {

        if(parameter == selectionParameter){

            return featureSelection;

        }

        return null;

    }

}