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

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import org.gvsig.euclidean.EuclideanManager;

import org.gvsig.fmap.dal.exception.DataException;

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

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

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

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

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

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

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

import org.gvsig.fmap.mapcontext.MapContext;

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.fmap.mapcontext.rendering.symbols.ISymbol;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.service.spi.ProviderServices;

import org.gvsig.tools.util.ToolsUtilLocator;

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.spi.AbstractEditingProvider;

import org.gvsig.vectorediting.lib.spi.DefaultDrawingStatus;

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

import org.gvsig.vectorediting.lib.spi.EditingProviderManager;

import org.gvsig.vectorediting.lib.spi.EditingProviderServices;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class CircleTanTanRadiusEditingProvider extends AbstractEditingProvider implements

        EditingProvider {

    protected EditingServiceParameter firstPoint;

    protected EditingServiceParameter secondPoint;

    protected EditingServiceParameter radius;

    protected Geometry firstGeometry;

    protected Geometry secondGeometry;

    protected Map<EditingServiceParameter, Object> values;

    protected FeatureStore featureStore;

    private final DynObject parameters;

    private static final Logger LOGGER = LoggerFactory

            .getLogger(CircleTanTanRadiusEditingProvider.class);

    public CircleTanTanRadiusEditingProvider(ProviderServices providerServices,

            DynObject parameters) {

        super(providerServices);

        this.parameters = parameters;

        this.featureStore

                = (FeatureStore) parameters

                        .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.firstPoint

                = new DefaultEditingServiceParameter("point_near_tangent_in_first_geometry", "point_near_tangent_in_first_geometry",

                        TYPE.POSITION);

        this.secondPoint

                = new DefaultEditingServiceParameter("point_near_tangent_in_second_geometry", "point_near_tangent_in_second_geometry",

                        TYPE.POSITION);

        this.radius

                = new DefaultEditingServiceParameter("radius", "radius",

                        TYPE.VALUE);

    }

    public DefaultDrawingStatus getDrawingStatus(Point mousePosition)

            throws DrawServiceException {

        DefaultDrawingStatus drawingStatus = new DefaultDrawingStatus();

        EditingProviderManager editingProviderManager

                = EditingProviderLocator.getProviderManager();

        ISymbol polygonSymbolEditing = editingProviderManager.getSymbol("polygon-symbol-editing");

        ISymbol auxiliaryLineSymbolEditing = editingProviderManager.getSymbol("auxiliary-line-symbol-editing");

        ISymbol auxiliaryPointSymbolEditing = editingProviderManager.getSymbol("auxiliary-point-symbol-editing");

        EuclideanManager euclideanManager = ToolsUtilLocator.getEuclideanManager();

        GeometryManager geomManager = GeometryLocator.getGeometryManager();

        EditingProviderServices editingProviderServices

                = (EditingProviderServices) getProviderServices();

        if (values != null) {

            Point firstPointValue = (Point) values.get(firstPoint);

            Point secondPointValue = (Point) values.get(secondPoint);

            Double radiusValue = (Double) values.get(radius);

            if (firstPointValue != null && secondPointValue == null) {

                try {

                    int subtype = editingProviderServices.getSubType(featureStore);

                    drawingStatus.addStatus(firstGeometry.toLines(), auxiliaryLineSymbolEditing, "");

                    drawingStatus.addStatus(firstPointValue, auxiliaryPointSymbolEditing, "");

                } catch (Exception e) {

                    throw new DrawServiceException(e);

                }

            } else if (firstPointValue != null && secondPointValue != null && radiusValue == null) {

                try {

                    int subtype = editingProviderServices.getSubType(featureStore);

                    drawingStatus.addStatus(firstGeometry.toLines(), auxiliaryLineSymbolEditing, "");

                    drawingStatus.addStatus(firstPointValue, auxiliaryPointSymbolEditing, "");

                    drawingStatus.addStatus(secondGeometry.toLines(), auxiliaryLineSymbolEditing, "");

                    drawingStatus.addStatus(secondPointValue, auxiliaryPointSymbolEditing, "");

                } catch (Exception e) {

                    throw new DrawServiceException(e);

                }

            }

        }

        return drawingStatus;

    }

    public EditingServiceParameter next() {

        if (values != null) {

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

                return firstPoint;

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

                return secondPoint;

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

                return radius;

            }

        }

        return null;

    }

    public void stop() {

        values.clear();

    }

    private void validateAndInsertValue(EditingServiceParameter param,

            Object value) throws InvalidEntryException {

        if (param == firstPoint) {

            if (value instanceof Point) {

                firstGeometry = getGeometry((Point)value);

                if(firstGeometry != null) {

                    values.put(param, value);

                    return;

                }

            }

        } else if (param == secondPoint) {

            if (value instanceof Point) {

                secondGeometry = getGeometry((Point)value);

                if(secondGeometry != null) {

                    values.put(param, value);

                }

                return;

            }

        } else if (value instanceof Double) {

            double radius = (Double) value;

            if ((radius - 0.01) > 0 && validateRadius(radius)) {

                values.put(param, value);

                return;

            }

        }

        throw new InvalidEntryException(null);

    }

    protected Geometry getGeometry(Point point){

        EditingProviderServices editingProviderServices = (EditingProviderServices) getProviderServices();

        MapContext mapContext = (MapContext) this.parameters.getDynValue(EditingProviderFactory.MAPCONTEXT_FIELD);

        Geometry geometry = editingProviderServices.getGeometryOfVisibleLayers(point, featureStore, mapContext);

//        Geometry geometry = editingProviderServices.getGeometry(point, this.featureStore, mapContext);

//        if(geometry == null){

//            FLayer[] visibleLayers = mapContext.getLayers().getVisibles();//.getActives();

//            for (FLayer visibleLayer : visibleLayers) {

//                if(visibleLayer instanceof FLyrVect){

//                    if(((FLyrVect) visibleLayer).getFeatureStore()!=this.featureStore){

//                        geometry  = editingProviderServices.getGeometry(point, ((FLyrVect)visibleLayer).getFeatureStore(), mapContext);

//                        if(geometry != null){

//                            break;

//                        }

//                    }

//                }

//            }

//        }

        return geometry;

    }

    private boolean validateRadius(double radius) {

        try {

            Point firstPointValue = (Point) values.get(firstPoint);

            Point secondPointValue = (Point) values.get(secondPoint);

            EditingProviderServices editingProviderServices

                    = (EditingProviderServices) getProviderServices();

            int subtype = editingProviderServices.getSubType(featureStore);

            Geometry geometry

                    = editingProviderServices.createCircle(firstGeometry, secondGeometry, radius, firstPointValue, secondPointValue, subtype);

            return geometry != null;

        } catch (DataException | CreateGeometryException ex) {

            LOGGER.warn("Can't create circle with radius "+radius, ex);

        }

        return false;

    }

    @Override

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> list

                = new ArrayList<>();

        list.add(firstPoint);

        list.add(secondPoint);

        list.add(radius);

        return list;

    }

    @Override

    public void setValue(Object value) throws InvalidEntryException {

        EditingServiceParameter param = next();

        validateAndInsertValue(param, value);

    }

    @Override

    public void finishAndStore() throws FinishServiceException {

        EditingProviderServices editingProviderServices

                = (EditingProviderServices) getProviderServices();

        Geometry geometry = finish();

        if(geometry!=null){

            editingProviderServices.insertGeometryIntoFeatureStore(geometry,

                featureStore);

        }

    }

    public Geometry finish() throws FinishServiceException {

        final GeometryManager geomManager = GeometryLocator.getGeometryManager();

        Point firstPointValue = (Point) values.get(firstPoint);

        Point secondPointValue = (Point) values.get(secondPoint);

        double radiusValue = (double) values.get(radius);

        EditingProviderServices editingProviderServices

                = (EditingProviderServices) getProviderServices();

        try {

            int subtype = editingProviderServices.getSubType(featureStore);

            GeometryType storeGeomType

                    = editingProviderServices.getGeomType(featureStore);

            Geometry geometry

                    = editingProviderServices.createCircle(firstGeometry, secondGeometry, radiusValue, firstPointValue, secondPointValue, subtype);

            if (storeGeomType.isTypeOf(MULTISURFACE)) {

                MultiSurface multisurface = null;

                if(geometry != null) {

                    multisurface

                        = geomManager.createMultiSurface(

                                storeGeomType.getSubType());

                    multisurface.addSurface((Circle) geometry);

                }

                return multisurface;

            }

            return geometry;

        } catch (Exception e) {

            throw new FinishServiceException(e);

        }

    }

    @Override

    public void start() throws StartServiceException {

        this.values = new HashMap<>();

    }

    @Override

    public String getName() {

        return CircleTanTanRadiusEditingProviderFactory.PROVIDER_NAME;

    }

}