GeometryUtils.java

Go to the documentation of this file.

/* *********************************************************************** *
 * project: org.matsim.*
 * GeometryUtils.java
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 * copyright       : (C) 2019 by the members listed in the COPYING,        *
 *                   LICENSE and WARRANTY file.                            *
 * email           : info at matsim dot org                                *
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 *   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.                                   *
 *   See also COPYING, LICENSE and WARRANTY file                           *
 *                                                                         *
 * *********************************************************************** */

package org.matsim.core.utils.geometry;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Random;

import org.geotools.api.feature.simple.SimpleFeature;
import org.geotools.geometry.jts.GeometryBuilder;
import org.locationtech.jts.geom.Coordinate;
import org.locationtech.jts.geom.Geometry;
import org.locationtech.jts.geom.GeometryFactory;
import org.locationtech.jts.geom.LineString;
import org.locationtech.jts.geom.Point;
import org.locationtech.jts.geom.Polygon;
import org.matsim.api.core.v01.Coord;
import org.matsim.api.core.v01.network.Link;
import org.matsim.api.core.v01.network.Network;
import org.matsim.core.gbl.Gbl;
import org.matsim.core.utils.geometry.geotools.MGC;

public class GeometryUtils {

    private static final GeometryFactory GEOMETRY_FACTORY = new GeometryFactory();

    private GeometryUtils() {} // do not instantiate

    public static List<Link> findIntersectingLinks( Link link, final Network network ) {
        LineString segment = GeometryUtils.createGeotoolsLineString(link) ;
        return GeometryUtils.findIntersectingLinks(segment, network) ;
    }

    public static List<Link> findIntersectingLinks(LineString lineString, final Network network) {
        // yy One could probably improve this method by using the (already existing) link quadtree to look only at
        // those links that are in the bounding box.  kai, oct'17

        // convert matsim links into geotools line strings:
        Map<Link,LineString> segments = new LinkedHashMap<>() ;
        for ( Link link : network.getLinks().values() ) {
            LineString theSegment = GeometryUtils.createGeotoolsLineString(link);
            segments.put( link, theSegment ) ;
        }

        // find the intersecting segments:
        List<Link> resultList = new ArrayList<>();
        for ( Entry<Link, LineString> entry : segments.entrySet() ) {
            if( lineString.intersects( entry.getValue() ) ){
                resultList.add( entry.getKey() ) ;
            }
        }
        return resultList;
    }

    public static LineString createGeotoolsLineString(Link link) {
        Coordinate fromCoord = MGC.coord2Coordinate( link.getFromNode().getCoord() ) ;
        Coordinate toCoord = MGC.coord2Coordinate( link.getToNode().getCoord() ) ;
        LineString theSegment = GEOMETRY_FACTORY.createLineString(new Coordinate[] { fromCoord, toCoord });
        return theSegment;
    }

    public static Point createGeotoolsPoint(Coord coord ) {
        Coordinate coordinate = MGC.coord2Coordinate(coord) ;
        Point point = GEOMETRY_FACTORY.createPoint(coordinate);
        return point ;
    }

    public static Polygon createGeotoolsPolygon(List<Coord> coords ) {

        // better way to do this is welcome.  kai, dec'17
        double [] flatArray = new double[coords.size()*2] ;
        int ii=0 ;
        for ( Coord coord : coords ) {
            flatArray[ii] = coord.getX() ;
            ii++ ;
            flatArray[ii] = coord.getY() ;
            ii++ ;
        }
        return new GeometryBuilder().polygon( flatArray ) ;

        // the following yields some failing tests in the minibus contrib. ihab, feb'19

//      Coordinate[] coordinates = new Coordinate[coords.size()] ;
//      int ii=0 ;
//      for ( Coord coord : coords ) {
//          coordinates[ii] = new Coordinate(coord.getX(), coord.getY()); ;
//          ii++ ;
//      }
//      return new GeometryFactory().createPolygon(coordinates);

    }

    public static Point getRandomPointInFeature( Random rnd, SimpleFeature ft ) {
        Gbl.assertNotNull(ft );
        Point p = null;
        double x, y;
        // generate a random point until a point inside the feature geometry is found
        do {
            x = ft.getBounds().getMinX() + rnd.nextDouble() * (ft.getBounds().getMaxX() - ft.getBounds().getMinX());
            y = ft.getBounds().getMinY() + rnd.nextDouble() * (ft.getBounds().getMaxY() - ft.getBounds().getMinY());
            p = MGC.xy2Point(x, y);
        } while ( ! (((Geometry) ft.getDefaultGeometry()).contains(p)) );
        return p;
    }


}