| gvSIG desktop 1

/*

 * Created on 10-abr-2006

 *

 * gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2004 IVER T.I. and Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 *

 * For more information, contact:

 *

 *  Generalitat Valenciana

 *   Conselleria d'Infraestructures i Transport

 *   Av. Blasco Ib??ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *      +34 963862235

 *   gvsig@gva.es

 *      www.gvsig.gva.es

 *

 *    or

 *

 *   IVER T.I. S.A

 *   Salamanca 50

 *   46005 Valencia

 *   Spain

 *

 *   +34 963163400

 *   dac@iver.es

 */

/* CVS MESSAGES:

 *

 * $Id: 

 * $Log: 

 */

package org.gvsig.jts.voronoi;

import java.awt.geom.Point2D;

import java.rmi.server.UID;

import java.util.ArrayList;

import java.util.Collection;

import java.util.Enumeration;

import java.util.List;

import org.gvsig.exceptions.BaseException;

import org.gvsig.fmap.core.NewFConverter;

import org.gvsig.jts.JtsUtil;

import org.gvsig.jts.voronoi.Voronoier.VoronoiStrategy;

import org.gvsig.jts.voronoi.chen.Coord;

import org.gvsig.jts.voronoi.chen.Vertex;

import org.gvsig.jts.voronoi.chen.Voronoi;

import org.gvsig.jts.voronoi.chen.VoronoiSegment;

import com.hardcode.gdbms.driver.exceptions.ReadDriverException;

import com.hardcode.gdbms.engine.values.Value;

import com.hardcode.gdbms.engine.values.ValueFactory;

import com.iver.cit.gvsig.fmap.core.DefaultFeature;

import com.iver.cit.gvsig.fmap.core.IFeature;

import com.iver.cit.gvsig.fmap.core.IGeometry;

import com.iver.utiles.swing.threads.CancellableProgressTask;

import com.vividsolutions.jts.geom.Coordinate;

import com.vividsolutions.jts.geom.Geometry;

import com.vividsolutions.jts.geom.GeometryCollection;

import com.vividsolutions.jts.geom.LineString;

import com.vividsolutions.jts.operation.polygonize.Polygonizer;

/**

 * It uses voronoi's classes from 1998 Ping-Che Chen's 1998 applet

 * 

 * @author Alvaro Zabala

 * 

 */

public class ChenVoronoiStrategy implements VoronoiStrategy {

        /**

         * Creates a voronoi diagram in form of a collection of features.

         * 

         * Each feature has the next attributes: fid- voronoi feature idenfifier

         * associatedVertex- fid of the vertex which generates the voronoi polygon

         * 

         * @param inputLyr

         * @return

         * @throws ReadDriverException

         */

        public List<IFeature> createTin(VoronoiAndTinInputLyr inputLyr,

                                                                                        boolean onlySelection,

                                                                CancellableProgressTask progressTask) throws ReadDriverException {

                List<IFeature> solution = new ArrayList<IFeature>();

                int numPoints = inputLyr.getSource().getShapeCount();

                List<Coord> coordsList = new ArrayList<Coord>();

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

                        if(onlySelection){

                                if(! inputLyr.getRecordset().getSelection().get(i))

                                        continue;

                        }

                        Point2D point = inputLyr.getPoint(i);

                        Coord coord = new Coord();

                        coord.x = point.getX();

                        coord.y = point.getY();

                        coordsList.add(coord);

                }

                Coord[] coords = new Coord[coordsList.size()];

                coordsList.toArray(coords);

                Vertex[] vertices = Voronoi.generate(coords);

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

                        Vertex vertex = vertices[i];

                        Geometry jtsGeom = createJtsGeom(vertex);

                        if (jtsGeom == null) {

                                System.err.println(i

                                                + "nth vertex has associated a null voronoi polygon");

                                continue;

                        }

                        IGeometry geom = NewFConverter.toFMap(jtsGeom);

                        Value fid = ValueFactory.createValue(i);

                        Value associatedVertex = ValueFactory

                                        .createValue(vertex.originalOrder);

                        Value[] values = new Value[] { fid, associatedVertex };

                        DefaultFeature feature = new DefaultFeature(geom, values, new UID()

                                        .toString());

                        solution.add(feature);

                }

                return solution;

        }

        /**

         * Receives a voronoi vertex and builds a geometry collection with all

         * polygons and dangling lines associated to this voronoi vertex.

         * 

         * TODO SI NO SE CONSIGUE FORMAR POLIGONO, FUSIONAR LOS LINESTRINGS CON UN

         * LINESTRINGSEWER

         * 

         * @param voronoiVertex

         * @return

         */

        private Geometry createJtsGeom(Vertex voronoiVertex) {

                GeometryCollection solution = null;

                List<Geometry> geometries = new ArrayList<Geometry>();

                Polygonizer polygonizer = new Polygonizer();

                List lineStringCollection = new ArrayList();

                Enumeration i = voronoiVertex.edges.elements();

                while (i.hasMoreElements()) {

                        VoronoiSegment s = (VoronoiSegment) i.nextElement();

                        Vertex v1 = s.v1;

                        Coordinate c1 = new Coordinate(v1.x, v1.y);

                        Vertex v2 = s.v2;

                        Coordinate c2 = new Coordinate(v2.x, v2.y);

                        if (c1.equals2D(c2)) {

                                System.out.println("El segmento " + s.toString()

                                                + " tiene los dos vertices iguales");

                                continue;

                        }

                        LineString line = JtsUtil.GEOMETRY_FACTORY

                                        .createLineString(new Coordinate[] { c1, c2 });

                        lineStringCollection.add(line);

                }// while

                if (lineStringCollection.size() == 0)

                        return null;

                polygonizer.add(lineStringCollection);

                Collection polygons = polygonizer.getPolygons();

                geometries.addAll(polygons);

                Collection danglingLines = polygonizer.getDangles();

                geometries.addAll(danglingLines);

                Geometry[] geomsArray = new Geometry[geometries.size()];

                geometries.toArray(geomsArray);

                solution = JtsUtil.GEOMETRY_FACTORY

                                .createGeometryCollection(geomsArray);

                return solution;

        }

        public List<IFeature> createThiessenPolygons(

                        VoronoiAndTinInputLyr inputLyr, 

                        boolean onlySelection,

                        CancellableProgressTask progressTask)

                        throws BaseException {

                List<IFeature> solution = new ArrayList<IFeature>();

                int numPoints = inputLyr.getSource().getShapeCount();

                List<Coord> coordsList = new ArrayList<Coord>();

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

                        if(onlySelection){

                                if(! inputLyr.getRecordset().getSelection().get(i))

                                        continue;

                        }

                        Point2D point = inputLyr.getPoint(i);

                        Coord coord = new Coord();

                        coord.x = point.getX();

                        coord.y = point.getY();

                        coordsList.add(coord);

                }

                Coord[] coords = new Coord[coordsList.size()];

                coordsList.toArray(coords);

                Vertex[] vertices = Voronoi.generate(coords);

                List<Geometry> geometries = new ArrayList<Geometry>();

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

                        Vertex vertex = vertices[i];

                        Geometry jtsGeom = createJtsGeom(vertex);

                        geometries.add(jtsGeom);

                }

                List<Geometry> thiessen = Voronoier.getThiessenPolygons(geometries, progressTask);

                for (int i = 0; i < thiessen.size(); i++) {

                        Geometry jtsGeom = thiessen.get(i);

                        IGeometry geom = NewFConverter.toFMap(jtsGeom);

                        Value fid = ValueFactory.createValue(i);

                        Value associatedVertex = ValueFactory

                                        .createValue(i);

                        Value[] values = new Value[] { fid, associatedVertex };

                        DefaultFeature feature = new DefaultFeature(geom, values, new UID()

                                        .toString());

                        solution.add(feature);

                }

                return solution;

        }

        public List<IFeature> createThiessenPolygons(VoronoiAndTinInputLyr inputLyr, boolean onlySelection)

                        throws BaseException {

                return createThiessenPolygons(inputLyr, onlySelection, null);

        }

        public List<IFeature> createTin(VoronoiAndTinInputLyr inputLyr, boolean onlySelection)

                        throws BaseException {

                return createTin(inputLyr, onlySelection, null);

        }

}