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

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

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

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

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

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.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 InternalPolygonEditingProvider extends AbstractEditingProvider

    implements EditingProvider {

    private EditingServiceParameter selectionParameter;

    private EditingServiceParameter geometryParameter;

    private FeatureStore featureStore;

    private Map<EditingServiceParameter, Object> values;

    public InternalPolygonEditingProvider(ProviderServices services,

        DynObject parameters) {

        super(services);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.selectionParameter =

            new DefaultEditingServiceParameter("selection", "selection",

                TYPE.SELECTION);

        this.geometryParameter =

            new DefaultEditingServiceParameter("Draw geometry",

                "draw_geometry_to_internal_polygon", Geometry.TYPES.MULTISURFACE, TYPE.GEOMETRY);

    }

    public EditingServiceParameter next() {

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

            return selectionParameter;

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

            return geometryParameter;

        }

        return null;

    }

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        return null;

    }

    public void stop() throws StopServiceException {

        values.clear();

    }

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> list =

            new ArrayList<EditingServiceParameter>();

        list.add(selectionParameter);

        list.add(geometryParameter);

        return list;

    }

    public void setValue(Object value) throws InvalidEntryException {

        EditingServiceParameter param = next();

        validateAndInsertValue(param, value);

    }

    private void validateAndInsertValue(EditingServiceParameter param,

        Object value) throws InvalidEntryException {

        if (param == selectionParameter) {

            if (value instanceof FeatureSelection) {

                // Before set value checks if all geometries selected are

                // instance of surface or multisurface

                FeatureSelection selection = (FeatureSelection) value;

                DisposableIterator it = null;

                if (selection.getSelectedCount() > 0) {

                    try {

                        it = selection.fastIterator();

                        while (it.hasNext()) {

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

                            Geometry geometry = feature.getDefaultGeometry();

                            GeometryType geometryType =

                                geometry.getGeometryType();

                            if (!geometryType

                                .isTypeOf(Geometry.TYPES.MULTISURFACE)

                                && !geometryType

                                    .isTypeOf(Geometry.TYPES.SURFACE)

                                && !geometryType

                                    .isTypeOf(Geometry.TYPES.MULTIPOLYGON)

                                && !geometryType

                                    .isTypeOf(Geometry.TYPES.POLYGON)

                                ) {

                                throw new InvalidEntryException(null);

                            }

                        }

                        values.put(selectionParameter, selection);

                    } catch (BaseException e) {

                        throw new InvalidEntryException(e);

                    } finally {

                        it.dispose();

                    }

                } else {

                    throw new InvalidEntryException(null);

                }

            }

        } else if (param == geometryParameter) {

            if (value instanceof Geometry && !(value instanceof Point)) {

                // Before set value checks if some geometry contains the

                // geometry received as parameter.

                Geometry geometry = (Geometry) value;

                FeatureSelection selection =

                    (FeatureSelection) values.get(selectionParameter);

                DisposableIterator it = null;

                try {

                    it = selection.fastIterator();

                    while (it.hasNext()) {

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

                        if (feature.getDefaultGeometry().contains(geometry)) {

                            values.put(param, value);

                            return;

                        }

                    }

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

                        throw new InvalidEntryException(null);

                    }

                } catch (BaseException e) {

                    throw new InvalidEntryException(e);

                } finally {

                    it.dispose();

                }

            }

        }

    }

    public void finishAndStore() throws FinishServiceException {

        FeatureSelection selected =

            (FeatureSelection) values.get(selectionParameter);

        Geometry geometry = (Geometry) values.get(geometryParameter);

        DisposableIterator it = null;

        try {

            if (!selected.isEmpty()) {

                it = selected.fastIterator();

                while (it.hasNext()) {

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

                    Geometry geometryFeature =

                        feature.getDefaultGeometry().cloneGeometry();

                    if (geometryFeature.contains(geometry)) {

                        Geometry newGeom = null;

                        if(geometryFeature instanceof Polygon){

                            newGeom = geometryFeature.cloneGeometry();

                        } else {

                            newGeom = geometryFeature.toPolygons();

                        }

                        if(newGeom instanceof MultiPolygon){

                            MultiPolygon newMultiPolygon = (MultiPolygon)newGeom;

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

                                Polygon primitive = (Polygon) newMultiPolygon.getPrimitiveAt(i);

                                if(primitive.contains(geometry)){

                                    addInteriorRings(primitive, geometry);

                                }

                            }

                        } else {

                            if(newGeom.contains(geometry)){

                                addInteriorRings((Polygon)newGeom, geometry);

                            }

                        }

                        EditingProviderServices editingProviderServices =

                            (EditingProviderServices) getProviderServices();

                        EditableFeature editableFeature = feature.getEditable();

                        editableFeature.setDefaultGeometry(newGeom);

                        editingProviderServices.updateFeatureInFeatureStore(

                            editableFeature, featureStore);

                    }

                }

            }

        } catch (BaseException e) {

            throw new FinishServiceException(

                "Can't finalize " + this.getName(), e);

        } finally {

            it.dispose();

        }

    }

    private void addInteriorRings(Polygon polygon, Geometry geometry) throws GeometryException {

        if (geometry instanceof Polygon) {

            polygon.addInteriorRing((Polygon) geometry);

        } else {

            MultiPolygon multiPolygon = geometry.toPolygons();

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

                Polygon surface = (Polygon) multiPolygon.getPrimitiveAt(i);

                polygon.addInteriorRing(surface);

            }

        }

    }

    public Geometry finish() throws FinishServiceException {

        return null;

    }

    public void start() throws StartServiceException, InvalidEntryException {

        this.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) {

                try {

                    setValue(selected);

                } catch (InvalidEntryException e) {

                    throw new InvalidEntryException(e);

                }

            }

        }

    }

    public String getName() {

        return InternalPolygonEditingProviderFactory.PROVIDER_NAME;

    }

}