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

import java.util.ArrayList;

import java.util.Collection;

import java.util.HashMap;

import java.util.LinkedHashMap;

import java.util.List;

import java.util.Map;

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

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

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

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

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

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.exception.BaseException;

import org.gvsig.tools.i18n.I18nManager;

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

/**

 * 

 * @author llmarques

 *

 */

public class RegularPolygonEditingProvider extends AbstractEditingProvider

    implements EditingProvider {

    protected EditingServiceParameter polygonSides;

    protected EditingServiceParameter center;

    protected EditingServiceParameter mode;

    protected EditingServiceParameter pointOfCircle;

    protected Map<EditingServiceParameter, Object> values;

    protected Map<String, String> options;

    protected FeatureStore featureStore;

    public RegularPolygonEditingProvider(ProviderServices providerServices,

        DynObject parameters) {

        super(providerServices);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        polygonSides =

            new DefaultEditingServiceParameter("sides",

                "sides_of_regular_polygon", EditingServiceParameter.TYPE.VALUE);

        options = new LinkedHashMap<String, String>();

        options.put("I", "inscribed");

        options.put("C", "circumscribed");

        mode =

            new DefaultEditingServiceParameter("mode",

                "inscribed_or_circumscribed", options,

                EditingServiceParameter.TYPE.OPTION);

        center =

            new DefaultEditingServiceParameter("center",

                "center_of_regular_polygon",

                EditingServiceParameter.TYPE.POSITION);

        pointOfCircle =

            new DefaultEditingServiceParameter("point_of_circle",

                "point_of_circle", EditingServiceParameter.TYPE.POSITION);

    }

    public EditingServiceParameter next() {

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

            return polygonSides;

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

            return mode;

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

            return center;

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

            return pointOfCircle;

        }

        return null;

    }

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        DrawingStatus drawingStatus = new DefaultDrawingStatus();

        if (values != null) {

            Double sidesValue = (Double) values.get(polygonSides);

            String modeValue = (String) values.get(mode);

            Point centerValue = (Point) values.get(center);

            if (modeValue != null && centerValue != null) {

                try {

                    EditingProviderServices editingProviderServices =

                        (EditingProviderServices) getProviderServices();

                    int subtype =

                        editingProviderServices.getSubType(featureStore);

                    Curve polygon =

                        calculateRegularPolygon(modeValue, centerValue,

                            mousePosition, sidesValue.intValue());

                    Curve line =

                        editingProviderServices.createLine(centerValue.getX(),

                            centerValue.getY(), mousePosition.getX(),

                            mousePosition.getY(), subtype);

                    double radius = mousePosition.distance(centerValue);

                    Curve circumference =

                        editingProviderServices.createArc(centerValue, radius,

                            0, 2 * Math.PI, subtype);

                    drawingStatus.addGeometry(circumference);

                    drawingStatus.addGeometry(polygon);

                    drawingStatus.addGeometry(centerValue);

                    drawingStatus.addGeometry(line);

                } catch (BaseException e) {

                    throw new DrawServiceException(e);

                }

            }

        }

        return drawingStatus;

    }

    /**

     * Calculates polygon as of mode (inscribed or circumscribed), center of

     * polygon, point of circle and number of sides.

     * 

     * @param modeValue

     *            of polygon (inscribed or circumscribed).

     * @param centerValue

     *            of polygon

     * @param pointOfCircleValue

     *            of circle that inscribe or circumscribe polygon

     * @param sidesValue

     *            number of sides.

     */

    protected Curve calculateRegularPolygon(String modeValue,

        Point centerValue, Point pointOfCircle, int sidesValue)

        throws BaseException {

        GeometryManager geometryManager = GeometryLocator.getGeometryManager();

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        int subtype = editingProviderServices.getSubType(featureStore);

        Curve polygon =

            (Curve) geometryManager.create(Geometry.TYPES.CURVE, subtype);

        double initAngle =

            editingProviderServices.getAngle(centerValue, pointOfCircle);

        double radiusValue = pointOfCircle.distance(centerValue);

        double theta = 2 * Math.PI / sidesValue;

        if (modeValue.equalsIgnoreCase("I")) {

            for (int i = 0; i < sidesValue; ++i) {

                double x =

                    centerValue.getX() + radiusValue

                        * Math.cos(theta * i + initAngle);

                double y =

                    centerValue.getY() + radiusValue

                        * Math.sin(theta * i + initAngle);

                Point createPoint =

                    editingProviderServices.createPoint(x, y, subtype);

                polygon.addVertex(createPoint);

            }

        } else if (modeValue.equalsIgnoreCase("C")) {

            for (int i = 0; i < sidesValue; ++i) {

                double x =

                    centerValue.getX() + radiusValue

                        * Math.cos(theta * i + initAngle);

                double y =

                    centerValue.getY() + radiusValue

                        * Math.sin(theta * i + initAngle);

                Point apothemPoint =

                    editingProviderServices.createPoint(x, y, subtype);

                Double[] apothemParams =

                    editingProviderServices.getLineParams(centerValue,

                        apothemPoint);

                Point[] perpendicular =

                    editingProviderServices.getPerpendicular(apothemParams[0],

                        apothemParams[1], apothemPoint, subtype);

                double x1 =

                    centerValue.getX() + radiusValue

                        * Math.cos(theta * (i + 1) + initAngle);

                double y1 =

                    centerValue.getY() + radiusValue

                        * Math.sin(theta * (i + 1) + initAngle);

                Point nextApothemPoint =

                    editingProviderServices.createPoint(x1, y1, subtype);

                Double[] nextApothemParams =

                    editingProviderServices.getLineParams(centerValue,

                        nextApothemPoint);

                Point[] nextPerpendicular =

                    editingProviderServices.getPerpendicular(

                        nextApothemParams[0], nextApothemParams[1],

                        nextApothemPoint, subtype);

                Point vertex =

                    editingProviderServices.getIntersection(perpendicular,

                        nextPerpendicular, subtype);

                polygon.addVertex(vertex);

            }

        }

        polygon = (Curve) closeGeometry(polygon);

        return polygon;

    }

    /**

     * Close geometry if it is necessary.

     * 

     * @param geometry

     *            to be closed.

     */

    protected Geometry closeGeometry(Geometry geometry) {

        if (!isClose(geometry) && geometry != null) {

            if (geometry instanceof Surface) {

                Surface surface = (Surface) geometry;

                Point firstp = surface.getVertex(0);

                firstp = (Point) firstp.cloneGeometry();

                surface.addVertex(firstp);

                return surface;

            } else if (geometry instanceof Curve) {

                Curve line = (Curve) geometry;

                Point firstp = line.getVertex(0);

                firstp = (Point) firstp.cloneGeometry();

                line.addVertex(firstp);

                return line;

            }

        }

        return geometry;

    }

    /**

     * Gets if the geometry received by parameter is closed or not.

     * 

     * @param geometry

     * @return True if geometry is closed, otherwise false.

     */

    private boolean isClose(Geometry geometry) {

        if (geometry != null) {

            if (geometry instanceof OrientablePrimitive) {

                OrientablePrimitive orientablePrimitive =

                    (OrientablePrimitive) geometry;

                Point firstPoint = orientablePrimitive.getVertex(0);

                Point lastPoint =

                    orientablePrimitive.getVertex(orientablePrimitive

                        .getNumVertices() - 1);

                if (firstPoint.equals(lastPoint)) {

                    return true;

                }

            }

        }

        return false;

    }

    public void stop() throws StopServiceException {

        if (values != null) {

            values.clear();

        }

    }

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> parameters =

            new ArrayList<EditingServiceParameter>();

        parameters.add(polygonSides);

        parameters.add(mode);

        parameters.add(center);

        parameters.add(pointOfCircle);

        return parameters;

    }

    public void setValue(Object value) throws InvalidEntryException {

        EditingServiceParameter parameter = next();

        validateAndInsertValue(parameter, value);

    }

    private void validateAndInsertValue(EditingServiceParameter param,

        Object value) throws InvalidEntryException {

        if (param == polygonSides) {

            if (value instanceof Double) {

                double sides = (Double) value;

                if (sides > 0) {

                    values.put(param, sides);

                    return;

                } else {

                    throw new InvalidEntryException(null);

                }

            }

        } else if (param == mode) {

            if (value instanceof String) {

                String option = (String) value;

                Collection<String> valueOptions = options.keySet();

                if (valueOptions.contains(option)) {

                    values.put(param, value);

                    return;

                } else {

                    throw new InvalidEntryException(null);

                }

            }

        } else if (param == center) {

            if (value instanceof Point) {

                values.put(param, value);

                return;

            }

        } else if (param == pointOfCircle) {

            if (value instanceof Point) {

                values.put(param, value);

                return;

            }

        }

        throw new InvalidEntryException(null);

    }

    public Geometry finish() throws FinishServiceException {

        if (values != null) {

            Double sidesValue = (Double) values.get(polygonSides);

            String modeValue = (String) values.get(mode);

            Point centerValue = (Point) values.get(center);

            Point pointOfCircleValue = (Point) values.get(pointOfCircle);

            try {

                Curve polygon =

                    calculateRegularPolygon(modeValue, centerValue,

                        pointOfCircleValue, sidesValue.intValue());

                EditingProviderServices editingProviderServices =

                    (EditingProviderServices) getProviderServices();

                GeometryType geomType =

                    editingProviderServices.getGeomType(featureStore);

                if (geomType.isTypeOf(Geometry.TYPES.MULTICURVE)) {

                    MultiCurve multiCurve =

                        GeometryLocator.getGeometryManager().createMultiCurve(

                            geomType.getSubType());

                    multiCurve.addCurve(polygon);

                    return multiCurve;

                }

                return polygon;

            } catch (Exception e) {

                throw new FinishServiceException(e);

            }

        }

        return null;

    }

    public void finishAndStore() throws FinishServiceException {

        try {

            Geometry geometry = finish();

            if (geometry != null) {

                EditingProviderServices editingProviderServices =

                    (EditingProviderServices) getProviderServices();

                editingProviderServices.insertGeometryIntoFeatureStore(

                    geometry, featureStore);

            }

        } catch (Exception e) {

            throw new FinishServiceException(e);

        }

    }

    public void start() throws StartServiceException {

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

    }

    public String getName() {

        return RegularPolygonEditingProviderFactory.PROVIDER_NAME;

    }

}