Application: gvSIG desktop » gvSIG tools: Vector editing

package org.gvsig.vectorediting.lib.prov.editvertex.operation;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

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

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

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

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

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

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

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

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

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

public class PolygonEditVertexOperation implements EditVertexOperation {

    public PolygonEditVertexOperation() {

        // TODO Auto-generated constructor stub

    }

    public Geometry insertVertex(Primitive geom, int index) throws CreateGeometryException {

        GeometryManager geometryManager = GeometryLocator.getGeometryManager();

        Polygon polygon = (Polygon) geom.cloneGeometry();

        int numVertices = polygon.getNumVertices();

        if (index < 0 || index >= numVertices) {

            return polygon;

        }

        int indexToInsert = index;

        Point antVertex = polygon.getVertex(numVertices - 1);

        if (index == 0 && polygon.getVertex(0).equals(antVertex)) {

            indexToInsert = numVertices - 1;

        }

        if (indexToInsert!=0){

            antVertex = polygon.getVertex(indexToInsert - 1);

        }

        Point point = polygon.getVertex(indexToInsert);

        Point newPoint = geometryManager.createPoint((point.getX() + antVertex.getX()) / 2,

            (point.getY() + antVertex.getY()) / 2, geom.getGeometryType().getSubType());

        polygon.insertVertex(indexToInsert, newPoint);

        return polygon;

    }

    public Geometry removeVertex(Primitive geom, int index)

        throws CreateGeometryException {

        Polygon polygon = (Polygon) geom.cloneGeometry();

        int numVertices = polygon.getNumVertices();

        if (index < 0 || index >= numVertices) {

            return polygon;

        }

        if (polygon.getVertex(0).equals(polygon.getVertex(numVertices - 1))) {

            if (numVertices <= 4) {

                return polygon;

            }

            numVertices--;

        } else {

            if (numVertices <= 3) {

                return polygon;

            }

        }

        polygon.removeVertex(index);

        return polygon;

    }

    public Geometry moveVertex(Primitive geom, int index, Point point) {

        Polygon result = (Polygon) geom.cloneGeometry();

        int numVertices = result.getNumVertices();

        if (result.getVertex(0).equals(result.getVertex(numVertices - 1))) {

            if (index == 0 || index == (result.getNumVertices() - 1)) {

                EditVertexOperationUtils.setVertex(result, 0, point);

                EditVertexOperationUtils.setVertex(result,

                    result.getNumVertices() - 1, (Point)point.cloneGeometry());

                return result;

            }

        }

        EditVertexOperationUtils.setVertex(result, index, point);

        return result;

    }

    public List<Point> getVertexesList(Primitive geom) {

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

        if (geom instanceof OrientablePrimitive) {

            OrientablePrimitive orientable = (OrientablePrimitive) geom;

            orientable.getNumVertices();

            for (int i = 0; i < orientable.getNumVertices(); i++) {

                vertexList.add(orientable.getVertex(i));

            }

        }

        return vertexList;

    }

    public int getNumVertex(Primitive geom) {

        if (geom instanceof OrientablePrimitive) {

            OrientablePrimitive orientable = (OrientablePrimitive) geom;

            return orientable.getNumVertices();

        }

        return 0;

    }

    public boolean canInsertVertex(Primitive geom, int index) {

        return true;

    }

    public boolean canRemoveVertex(Primitive geom, int index) {

        Polygon polygon = (Polygon) geom;

        int numVertices = polygon.getNumVertices();

        if (index < 0) {

            return false;

        } else if (index >= numVertices) {

            return false;

        }

        if (polygon.getVertex(0).equals(polygon.getVertex(numVertices - 1))) {

            if (numVertices <= 4) {

                return false;

            }

        } else {

            if (numVertices <= 3) {

                return false;

            }

        }

        return true;

    }

    public boolean canMoveVertex(int index, Point point) {

        return true;

    }

    public void addToDrawingStatus(DefaultDrawingStatus drawingStatus,

        Primitive primitive) {

        EditingProviderManager editingProviderManager =

            EditingProviderLocator.getProviderManager();

        ISymbol polygonSymbolEditing =

            editingProviderManager.getSymbol("polygon-symbol-editing");

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

    }

}