org.opentripplanner.geocoder.reverse.ShapefileBoundaryResolver Maven / Gradle / Ivy

Go to download
package org.opentripplanner.geocoder.reverse;

import java.io.File;
import java.io.IOException;

import org.geotools.data.FeatureSource;
import org.geotools.data.shapefile.ShapefileDataStore;
import org.geotools.feature.FeatureCollection;
import org.geotools.feature.FeatureIterator;
import org.geotools.geometry.jts.JTSFactoryFinder;
import org.opengis.feature.Feature;
import org.opengis.feature.simple.SimpleFeature;

import org.locationtech.jts.geom.Coordinate;
import org.locationtech.jts.geom.Geometry;
import org.locationtech.jts.geom.GeometryFactory;
import org.locationtech.jts.geom.Point;


public class ShapefileBoundaryResolver implements BoundaryResolver {

    private String shapefile, nameField;
    private FeatureCollection collection;
    
    public void setShapefile(String shapefile) {
        this.shapefile = shapefile;
    }
    
    public void setNameField(String nameField) {
        this.nameField = nameField;
    }
    
    public void initialize() throws IOException {
        if (shapefile == null) {
            throw new IllegalStateException("shapefile path is not set");
        }

        // get feature results
        ShapefileDataStore store = new ShapefileDataStore(new File(shapefile).toURI().toURL());
        String name = store.getTypeNames()[0];
        FeatureSource source = store.getFeatureSource(name);
        collection = source.getFeatures();
    }
    
    @Override
    public String resolve(double x, double y) {
        System.out.println("x="+x+", y="+y);
        FeatureIterator iterator = collection.features();
        while( iterator.hasNext() ){
             SimpleFeature feature = (SimpleFeature) iterator.next();

             Geometry geom = (Geometry) feature.getDefaultGeometry();
             
             GeometryFactory geometryFactory = JTSFactoryFinder.getGeometryFactory();             
             Coordinate coord = new Coordinate(x, y);
             Point point = geometryFactory.createPoint(coord);
             
             //System.out.println("checking "+point.toString());
             if(geom.contains(point)) {
                 return feature.getAttribute(this.nameField).toString();
             }
        }
        return null;
        		
        
    }

}