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.offset.lib.prov.offset;

import java.awt.geom.AffineTransform;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.LinkedHashMap;

import java.util.List;

import java.util.Map;

import org.apache.commons.lang3.mutable.MutableDouble;

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

import org.gvsig.fmap.geom.GeometryManager;

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

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.exception.CreateEnvelopeException;

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

import org.gvsig.fmap.geom.operation.GeometryOperationException;

import org.gvsig.fmap.geom.operation.GeometryOperationNotSupportedException;

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

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

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

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

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

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

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

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

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

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

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

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposableIterator;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.locator.LocatorException;

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.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 OffsetEditingProvider extends AbstractEditingProvider implements

    EditingProvider {

    private final Double PRECISION=new Double(1.0e-5);

    private I18nManager i18nManager = ToolsLocator.getI18nManager();

    private EditingServiceParameter selectionParameter;

    private EditingServiceParameter pointParameter;

    private Map<EditingServiceParameter, Object> values;

    private FeatureStore featureStore;

    public OffsetEditingProvider(ProviderServices providerServices,

        DynObject parameters) {

        super(providerServices);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.selectionParameter =

            new DefaultEditingServiceParameter("selection",

                i18nManager.getTranslation("selection"), TYPE.SELECTION);

        this.pointParameter =

            new DefaultEditingServiceParameter("offset_distance",

                i18nManager.getTranslation("offset_distance"),

                TYPE.POSITION);

    }

    public EditingServiceParameter next() {

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

            return selectionParameter;

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

            return pointParameter;

        }

        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");

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

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

        FeatureSelection selected =

            (FeatureSelection) values.get(selectionParameter);

        try {

            if ((selected != null) && !selected.isEmpty()) {

                Point point=null;

                Object pointValue = (Point) values.get(pointParameter);

                if ((pointValue != null) && (pointValue instanceof Point)) {

                    point=(Point)pointValue;

                }else{

                    point=mousePosition;

                    Line line;

                    line =

                        GeometryLocator.getGeometryManager().createLine(

                            featureStore.getDefaultFeatureType()

                                .getDefaultGeometryAttribute().getGeomType()

                                .getSubType());

                    Point closestPoint=getClosestPoint(point,selected);

                    //TODO

                    Point perpendicularPoint=null;

                    MutableDouble normalModule=new MutableDouble();

                    try {

                        perpendicularPoint=getPerpendicularPoint(closestPoint, point,selected,normalModule);

                        //normalModule=getNormalModule(closestPoint, point,selected);

                    } catch (Exception e) {

                        throw new DrawServiceException(e);

                    }

                    line.addVertex(closestPoint);

                    //line.addVertex(point);

                    line.addVertex(perpendicularPoint);

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

                    AffineTransform at;

                    try {

                        at = getSymmetryAffineTransform(normalModule.getValue());

                    } catch (Exception e) {

                        throw new DrawServiceException(e);

                    }

                    DisposableIterator it;

                    it = selected.fastIterator();

                    while (it.hasNext()) {

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

                        Geometry transformedGeometry = feat.getDefaultGeometry().cloneGeometry();

                        transformedGeometry.transform(at);

                        ISymbol symbol=null;

                        if(transformedGeometry instanceof Curve || transformedGeometry instanceof MultiCurve){

                            symbol = lineSymbolEditing;

                        } else if(transformedGeometry instanceof Surface || transformedGeometry instanceof MultiSurface){

                            symbol = polygonSymbolEditing;

                        } else if(transformedGeometry instanceof Point || transformedGeometry instanceof MultiPoint){

                            symbol = auxiliaryPointSymbolEditing;

                        }

                        if(transformedGeometry instanceof Aggregate){

                            int primitivesNumber = ((Aggregate)transformedGeometry).getPrimitivesNumber();

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

                                drawingStatus.addStatus(((Aggregate)transformedGeometry).getPrimitiveAt(i), symbol, "");

                            }

                        } else {

                            drawingStatus.addStatus(transformedGeometry, symbol, "");

                        }

                    }

                    it.dispose();

                }

                return drawingStatus;

            }

        } catch (Exception e) {

            throw new DrawServiceException(e);

        }

        return null;

    }

    private AffineTransform getSymmetryAffineTransform(Double distance)

        throws GeometryOperationNotSupportedException,

        GeometryOperationException {

        AffineTransform translate =

            AffineTransform.getTranslateInstance(distance,

                0);

        return translate;

    }

    public void stop() {

        values.clear();

    }

    private void validateAndInsertValue(EditingServiceParameter param,

        Object value) throws InvalidEntryException {

        if (param == selectionParameter) {

            if (value instanceof FeatureSelection) {

                values.put(param, value);

                return;

            }

        } else if (param == pointParameter) {

            if (value instanceof Point) {

                values.put(param, value);

                return;

            }

        }

    }

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> list =

            new ArrayList<EditingServiceParameter>();

        list.add(selectionParameter);

        list.add(pointParameter);

        return list;

    }

    public void setValue(Object value) throws InvalidEntryException {

        EditingServiceParameter param = next();

        validateAndInsertValue(param, value);

    }

    public void finishAndStore() throws FinishServiceException {

        FeatureSelection selected =

            (FeatureSelection) values.get(selectionParameter);

        try {

            if (!selected.isEmpty()) {

                Point point = (Point) values.get(pointParameter);

                if ((point != null)) {

                    Double distance;

                    try {

                        Point closestPoint=getClosestPoint(point,selected);

                        distance = point.distance(closestPoint);

                    } catch (GeometryOperationNotSupportedException e1) {

                        throw new FinishServiceException(e1);

                    } catch (GeometryOperationException e1) {

                        throw new FinishServiceException(e1);

                    }

                    AffineTransform at;

                    try {

                        at = getSymmetryAffineTransform(distance);

                    } catch (GeometryOperationNotSupportedException e) {

                        throw new FinishServiceException(e);

                    } catch (GeometryOperationException e) {

                        throw new FinishServiceException(e);

                    }

                    DisposableIterator it;

                    it = selected.fastIterator();

                    while (it.hasNext()) {

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

                        Geometry geom =

                            feature.getDefaultGeometry().cloneGeometry();

                        geom.transform(at);

                        // Se crea una feature nueva copiando los valores de

                        // la feature original excepto aquellos que sean PK

                        EditingProviderServices editingProviderServices =

                            (EditingProviderServices) getProviderServices();

                        EditableFeature editableFeature =

                            editingProviderServices

                                .getFeatureCopyWithoutPK(featureStore,

                                    feature);

                        editableFeature.setDefaultGeometry(geom);

                        editingProviderServices

                            .insertFeatureIntoFeatureStore(editableFeature,

                                featureStore);

                    }

                    it.dispose();

                    featureStore.getFeatureSelection().deselectAll();

                }

            }

        } catch (DataException e) {

            throw new FinishServiceException(e);

        }

    }

    private Double getNormalModule(Point closestPoint, Point distancePoint,FeatureSelection selected)

        throws DataException, CreateEnvelopeException, GeometryOperationNotSupportedException, GeometryOperationException, CreateGeometryException, LocatorException{

        MutableDouble distance=null;

        getPerpendicularPoint(closestPoint, distancePoint, selected, distance);

        return distance.getValue();

    }

    private Point getPerpendicularPoint(Point closestPoint, Point distancePoint,FeatureSelection selected)

        throws DataException, CreateEnvelopeException, GeometryOperationNotSupportedException, GeometryOperationException, CreateGeometryException, LocatorException{

        return getPerpendicularPoint(closestPoint,distancePoint,selected,null);

    }

    private Point getPerpendicularPoint(Point closestPoint, Point distancePoint,FeatureSelection selected, MutableDouble normalModule)

        throws DataException, CreateEnvelopeException, GeometryOperationNotSupportedException, GeometryOperationException, CreateGeometryException, LocatorException{

        int subType = featureStore.getDefaultFeatureType()

            .getDefaultGeometryAttribute().getGeomType()

            .getSubType();

        Geometry containingGeometry=getContainingGeometry(selected, closestPoint);

        if (containingGeometry instanceof Circle || containingGeometry instanceof Ellipse ||

            containingGeometry instanceof Circumference || containingGeometry instanceof PeriEllipse){

            if (containingGeometry instanceof Circle){

                Circle circle=(Circle)containingGeometry;

                Circumference circumference =(Circumference)GeometryLocator.

                getGeometryManager().create(Geometry.TYPES.CIRCUMFERENCE, subType);

                circumference.setPoints(circle.getCenter(), circle.getRadious());

                containingGeometry=circumference;

            }

            if (containingGeometry instanceof Ellipse || containingGeometry instanceof Circumference){

                Ellipse ellipse=(Ellipse)containingGeometry;

                PeriEllipse periellipse =(PeriEllipse)GeometryLocator.

                getGeometryManager().create(Geometry.TYPES.PERIELLIPSE, subType);

                periellipse.setPoints(ellipse.getAxis1Start(), ellipse.getAxis1End(), ellipse.getAxis2Dist());

                containingGeometry=periellipse;

            }

            if (containingGeometry instanceof PeriEllipse){

                Integer direction;

                if (containingGeometry.contains(distancePoint)){

                    direction=-1;

                }else{

                    direction=1;

                }

                normalModule.setValue(closestPoint.distance(distancePoint)*direction);

                return distancePoint;

            }

        }

        Line containingLine=null;

        if (containingGeometry!=null){

            containingLine = getContainingLine(containingGeometry, closestPoint, subType);

        }

        if (containingLine!=null){

            Double module=containingLine.perimeter();

            Point point0=containingLine.getVertex(0);

            Point point1=containingLine.getVertex(1);

            Double unitarianX=(point1.getX()-point0.getX())/module;

            Double unitarianY=(point1.getY()-point0.getY())/module;

            Point unitarianPoint=GeometryLocator.getGeometryManager().createPoint(

                unitarianX,

                unitarianY,

                subType);

            Double perpendicularAngle=Math.PI/2;

            AffineTransform rotate = AffineTransform.getRotateInstance(perpendicularAngle);

            unitarianPoint.transform(rotate);

            Double distance=getDirectedDistance(closestPoint, distancePoint, containingLine);

            unitarianPoint.setX(unitarianPoint.getX()*distance);

            unitarianPoint.setY(unitarianPoint.getY()*distance);

            unitarianPoint.setX(unitarianPoint.getX()+closestPoint.getX());

            unitarianPoint.setY(unitarianPoint.getY()+closestPoint.getY());

            if (distance>0){

                normalModule.setValue(unitarianPoint.distance(closestPoint));

            }else{

                normalModule.setValue(-unitarianPoint.distance(closestPoint));

            }

            return unitarianPoint;

        }

        return null;

    }

    private Double getDirectedDistance(Point pointInLine, Point distancePoint, Line line)

        throws GeometryOperationNotSupportedException, GeometryOperationException{

        Double distance=distancePoint.distance(pointInLine);

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        Double angle = editingProviderServices.getAngle(pointInLine, distancePoint);

        double angleLine = editingProviderServices.getAngle(line.getVertex(0), line.getVertex(1));

        Double angleDifference=angle - angleLine;

        if ( angleDifference<0 ){

            angleDifference+=2*Math.PI;

        }

        if (angleDifference > Math.PI ) {

            distance = -distance; // perpendicularAngle=3*Math.PI/2;

        }

        return distance;

    }

    private Line getContainingLine(Geometry containingGeometry,

        Point containedPoint, int subType)

            throws CreateGeometryException,

            GeometryOperationNotSupportedException, GeometryOperationException {

        if (containingGeometry instanceof Line ||containingGeometry instanceof Polygon ){

            OrientablePrimitive orientableGeometry=(OrientablePrimitive)containingGeometry;

            for (int i=0; i<orientableGeometry.getNumVertices()-1;i++){

                Point point=orientableGeometry.getVertex(i);

                Point nextPoint=orientableGeometry.getVertex(i+1);

                Line line =

                    GeometryLocator.getGeometryManager().createLine(

                        subType);

                 line.addVertex(point);

                 line.addVertex(nextPoint);

                 //FIXME

                 //if (line.contains(closestPoint)){

                 if (line.isWithinDistance(containedPoint, PRECISION)){

                     return line;

                 }

            }

        }

        return null;

    }

    private Geometry getContainingGeometry (FeatureSelection selected,Point point)

        throws DataException, GeometryOperationNotSupportedException, GeometryOperationException{

        DisposableIterator it;

        it = selected.fastIterator();

        while (it.hasNext()) {

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

            Geometry geometry = feat.getDefaultGeometry();

            if(geometry instanceof Aggregate){

                int primitivesNumber = ((Aggregate)geometry).getPrimitivesNumber();

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

                    Geometry primitiveGeometry=((Aggregate)geometry).getPrimitiveAt(i);

                    if (primitiveGeometry instanceof Polygon) {

                        Polygon geomPoly = (Polygon)primitiveGeometry;

                        //FIXME

                        if (geomPoly.isWithinDistance(point, PRECISION)){

                        //if (geomPoly.touches(point)){

                            return primitiveGeometry;

                        }

                      //FIXME

                      //} else if(primitiveGeometry.contains(closestPoint)){

                      } else if(primitiveGeometry.isWithinDistance(point, PRECISION)){

                            return primitiveGeometry;

                    }

                }

            } else {

                if (geometry instanceof Polygon) {

                    Polygon geomPoly = (Polygon)geometry;

                  //FIXME

                    if (geomPoly.isWithinDistance(point, PRECISION)){

                    //if (geomPoly.touches(point)){

                        return geometry;

                    }

                //FIXME

                //} else if(geometry.contains(closestPoint)){

                } else if(geometry.isWithinDistance(point, PRECISION)){

                    return geometry;

                }

            }

        }

        return null;

    }

    private Point getClosestPoint(Point point, FeatureSelection selected)

        throws DataException, GeometryOperationNotSupportedException,

        GeometryOperationException{

        DisposableIterator it;

        it = selected.fastIterator();

        ArrayList<Point> closestPoints = new ArrayList<Point>();

        while (it.hasNext()) {

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

            Geometry geometry = feat.getDefaultGeometry();

            Geometry[] geoms = point.closestPoints(geometry);

            if (geoms!=null){

                for (int i=0;i<geoms.length;i++){

                    Geometry closePoint=geoms[i];

                    if (closePoint!=null && closePoint instanceof Point){

                        closestPoints.add((Point)closePoint);

                    }

                }

            }

        }

        it.dispose();

        double minorDistance=Double.POSITIVE_INFINITY;

        Point closestPoint=null;

        for (Point closePoint:closestPoints){

            double distance=point.distance(closePoint);

            if (distance<minorDistance && distance>Double.valueOf(0)){

                minorDistance=distance;

                closestPoint=closePoint;

            }

        }

        return closestPoint;

    }

    public Geometry finish() throws FinishServiceException {

        return null;

    }

    public void start() throws StartServiceException {

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

        FeatureSelection selected = null;

        if (featureStore != null) {

            try {

                selected =

                    (FeatureSelection) featureStore.getFeatureSelection()

                        .clone();

            } catch (DataException e) {

                throw new StartServiceException(e);

            } catch (CloneNotSupportedException e) {

                // Do nothing

            }

            if ((selected != null) && (selected.getSelectedCount() > 0)) {

                values.put(selectionParameter, selected);

            }

        }

    }

    public String getName() {

        return OffsetEditingProviderFactory.PROVIDER_NAME;

    }

}