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

import java.util.ArrayList;

import java.util.HashMap;

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

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

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

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

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

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

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

import org.gvsig.tools.dynobject.DynObject;

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

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

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

public class FilledEllipseEditingProvider extends AbstractEditingProvider implements

EditingProvider {

    protected EditingServiceParameter firstPointAxisA;

    protected EditingServiceParameter secondPointAxisA;

    protected EditingServiceParameter lengthAxisB;

    protected Map<EditingServiceParameter, Object> values;

    protected FeatureStore featureStore;

    public FilledEllipseEditingProvider(ProviderServices providerServices,

        DynObject parameters) {

        super(providerServices);

        this.featureStore =

            (FeatureStore) parameters

            .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.firstPointAxisA =

            new DefaultEditingServiceParameter("first_point_A_axis",

                "first_point_A_axis", EditingServiceParameter.TYPE.POSITION);

        this.secondPointAxisA =

            new DefaultEditingServiceParameter("second_point_A_axis",

                "second_point_A_axis", EditingServiceParameter.TYPE.POSITION);

        this.lengthAxisB =

            new DefaultEditingServiceParameter("length_of_B_axis",

                "length_of_B_axis", EditingServiceParameter.TYPE.POSITION,

                EditingServiceParameter.TYPE.VALUE);

    }

    public EditingServiceParameter next() {

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

            return firstPointAxisA;

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

            return secondPointAxisA;

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

            return lengthAxisB;

        }

        return null;

    }

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        DefaultDrawingStatus drawingStatus = new DefaultDrawingStatus();

        EditingProviderManager editingProviderManager =

            EditingProviderLocator.getProviderManager();

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

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

        if (values != null) {

            EditingProviderServices editingServiceProvider =

                (EditingProviderServices) getProviderServices();

            Point firstPointAxis1Value = (Point) values.get(firstPointAxisA);

            Point secondPointAxis1Value = (Point) values.get(secondPointAxisA);

            if (firstPointAxis1Value != null) {

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

                if (secondPointAxis1Value == null) {

                    try {

                        int subtype =

                            editingServiceProvider.getSubType(featureStore);

                        Curve line =

                            editingServiceProvider.createLine(

                                firstPointAxis1Value, mousePosition, subtype);

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

                        drawingStatus.addStatus(line, auxiliaryLineSymbolEditing, "");

                        return drawingStatus;

                    } catch (Exception e) {

                        throw new DrawServiceException(e);

                    }

                } else {

                    try {

                        int subtype =

                            editingServiceProvider.getSubType(featureStore);

                        Point center =

                            editingServiceProvider.getMidPoint(

                                firstPointAxis1Value, secondPointAxis1Value,

                                subtype);

                        double distance = center.distance(mousePosition);

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

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

                        Curve aAxis =

                            editingServiceProvider.createLine(

                                firstPointAxis1Value, secondPointAxis1Value,

                                subtype);

                        drawingStatus.addStatus(aAxis, auxiliaryLineSymbolEditing, "");

                        addTemporaryGeometryToDrawingStatus(drawingStatus, firstPointAxis1Value, secondPointAxis1Value,

                            subtype, distance);

                        Double[] lineParams =

                            editingServiceProvider.getLineParams(

                                firstPointAxis1Value, secondPointAxis1Value);

                        Point[] bisector =

                            editingServiceProvider.getPerpendicular(

                                lineParams[0], lineParams[1], center, subtype);

                        Double[] bisectorParams =

                            editingServiceProvider.getLineParams(bisector[0],

                                bisector[1]);

                        double m = bisectorParams[0]; // Line slope

                        double y0 = bisectorParams[1]; // Y-Intercept line

                        double mp2 = Math.pow(m, 2); // slope?

                        Curve line;

                        if (!Double.isInfinite(m)) {

                            double a = 1 + mp2;

                            double b = (-2 - (2 * mp2)) * center.getX();

                            double c =

                                ((1 + mp2) * Math.pow(center.getX(), 2))

                                - Math.pow(distance, 2);

                            double x1 =

                                (-b + Math.sqrt((Math.pow(b, 2) - (4 * a * c))))

                                / (2 * a);

                            double y1 = (m * x1) + y0;

                            double x2 =

                                (-b - Math.sqrt((Math.pow(b, 2) - (4 * a * c))))

                                / (2 * a);

                            double y2 = (m * x2) + y0;

                            line =

                                editingServiceProvider.createLine(x1, y1, x2,

                                    y2, subtype);

                            drawingStatus.addStatus(line, auxiliaryLineSymbolEditing, "");

                        } else {

                            double x1 = center.getX();

                            double y1 = center.getY() + distance;

                            double y2 = center.getY() - distance;

                            line =

                                editingServiceProvider.createLine(x1, y1, x1,

                                    y2, subtype);

                            drawingStatus.addStatus(line, auxiliaryLineSymbolEditing, "");

                        }

                        drawingStatus.addStatus(aAxis, auxiliaryLineSymbolEditing, "");

                    } catch (Exception e) {

                        throw new DrawServiceException(e);

                    }

                }

            }

        }

        return drawingStatus;

    }

    /**

     * @param drawingStatus

     * @param firstPointAxis1Value

     * @param secondPointAxis1Value

     * @param subtype

     * @param distance

     * @throws CreateGeometryException

     */

    protected void addTemporaryGeometryToDrawingStatus(DefaultDrawingStatus drawingStatus, Point firstPointAxis1Value,

        Point secondPointAxis1Value, int subtype, double distance) throws CreateGeometryException {

        Ellipse ellipse = (Ellipse) GeometryLocator.getGeometryManager().create(Geometry.TYPES.ELLIPSE, subtype);

        ellipse.setPoints(firstPointAxis1Value, secondPointAxis1Value, distance);

        EditingProviderManager editingProviderManager = EditingProviderLocator.getProviderManager();

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

        drawingStatus.addStatus(ellipse, polygonSymbolEditing, "");

    }

    public void stop() throws StopServiceException {

        if (values != null) {

            values.clear();

        }

    }

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> list =

            new ArrayList<EditingServiceParameter>();

        list.add(firstPointAxisA);

        list.add(secondPointAxisA);

        list.add(lengthAxisB);

        return list;

    }

    public void setValue(Object value) throws InvalidEntryException {

        EditingServiceParameter parameter = this.next();

        validateAndInsertValue(parameter, value);

    }

    private void validateAndInsertValue(EditingServiceParameter param,

        Object value) throws InvalidEntryException {

        if ((param == firstPointAxisA) || (param == secondPointAxisA)) {

            if (value instanceof Point) {

                values.put(param, value);

                return;

            }

        } else if (param == lengthAxisB) {

            if (value instanceof Point) {

                Double length = null;

                try {

                    EditingProviderServices editingProviderServices =

                        (EditingProviderServices) getProviderServices();

                    Point firstPointAxis1Value =

                        (Point) values.get(firstPointAxisA);

                    Point firstPointAxis2Value =

                        (Point) values.get(secondPointAxisA);

                    int subtype =

                        editingProviderServices.getSubType(featureStore);

                    Point center =

                        editingProviderServices

                        .getMidPoint(firstPointAxis1Value,

                            firstPointAxis2Value, subtype);

                    length = ((Point) value).distance(center) * 2;

                } catch (Exception e) {

                    throw new InvalidEntryException(e);

                }

                if (length != null) {

                    values.put(param, length);

                    return;

                }

            } else if (value instanceof Double) {

                if ((((Double) value) - 0.01) > 0) {

                    values.put(param, value);

                    return;

                }

            }

        }

    }

    public Geometry finish() throws FinishServiceException {

        if (values != null) {

            EditingProviderServices editingProviderServices =

                (EditingProviderServices) getProviderServices();

            Point firstPointAxis1Value = (Point) values.get(firstPointAxisA);

            Point secondPointAxis1Value = (Point) values.get(secondPointAxisA);

            double lengthAxis2Value = (Double) values.get(lengthAxisB);

            if ((firstPointAxis1Value != null)

                && (secondPointAxis1Value != null) && (lengthAxis2Value > 0)) {

                try {

                    int subtype = editingProviderServices.getSubType(featureStore);

                    GeometryType storeGeomType = editingProviderServices.getGeomType(featureStore);

                    Ellipse ellipse =

                        (Ellipse) GeometryLocator.getGeometryManager().create(Geometry.TYPES.ELLIPSE, subtype);

                    ellipse.setPoints(firstPointAxis1Value, secondPointAxis1Value, lengthAxis2Value / 2);

//                    if (storeGeomType.isTypeOf(MULTISURFACE)) {

//                        return ellipse.toPolygons();

//                    }

                    if (storeGeomType.isTypeOf(MULTISURFACE)) {

                        MultiSurface multisurface;

                        multisurface =

                            GeometryLocator.getGeometryManager().createMultiSurface(

                                storeGeomType.getSubType());

                        multisurface.addSurface((Ellipse) ellipse);

                        return multisurface;

                    }

                    return ellipse;

                } catch (Exception e) {

                    throw new FinishServiceException(e);

                }

            }

        }

        return null;

    }

    public void finishAndStore() throws FinishServiceException {

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        Geometry geometry = finish();

        editingProviderServices.insertGeometryIntoFeatureStore(geometry,

            featureStore);

    }

    public void start() throws StartServiceException {

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

    }

    public String getName() {

        return FilledEllipseEditingProviderFactory.PROVIDER_NAME;

    }

}