/**
 * Copyright (C) 2011 Brian Ferris <bdferris@onebusaway.org>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.onebusaway.transit_data_federation.bundle.utilities;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import org.onebusaway.csv_entities.EntityHandler;
import org.onebusaway.gtfs.model.Stop;
import org.onebusaway.gtfs.serialization.GtfsReader;

import com.vividsolutions.jts.geom.Coordinate;
import com.vividsolutions.jts.geom.Geometry;
import com.vividsolutions.jts.geom.GeometryFactory;
import com.vividsolutions.jts.geom.LinearRing;
import com.vividsolutions.jts.geom.Point;

/**
 * Utility script to compute the set of stops from a gtfs feed that fall within
 * a more complex polygon boundary
 * 
 * @author bdferris
 * @see GtfsStopsInRegionMain
 */
public class GtfsStopsInPolygonMain {
  
  private static GeometryFactory _factory = new GeometryFactory();

  @SuppressWarnings("unchecked")
  public static void main(String[] args) throws IOException {

    if (args.length != 2) {
      System.err.println("usage: gtfs_path polygon_path");
      System.exit(-1);
    }

    Geometry points = readPoints(args[1]);
    
    GtfsReader reader = new GtfsReader();
    reader.setDefaultAgencyId("1");
    reader.getEntityClasses().retainAll(Arrays.asList(Stop.class));
    reader.setInputLocation(new File(args[0]));
    reader.addEntityHandler(new EntityHandlerImpl(points));
    reader.run();
  }

  private static Geometry readPoints(String arg) throws NumberFormatException, IOException {
    
    BufferedReader reader = new BufferedReader(new FileReader(arg));
    String line = null;
    
    
    List<Coordinate> points = new ArrayList<Coordinate>();
    
    while((line = reader.readLine()) != null) {
      String[] tokens = line.trim().split("[,\\s]+");
      double lat = Double.parseDouble(tokens[0]);
      double lon = Double.parseDouble(tokens[1]);
      points.add(new Coordinate(lat,lon));
    }
    
    points.add(points.get(0));
    
    LinearRing ring = _factory.createLinearRing(points.toArray(new Coordinate[points.size()]));
    return _factory.createPolygon(ring, new LinearRing[0]);
  }

  private static class EntityHandlerImpl implements EntityHandler {

    private Geometry _geometry;

    public EntityHandlerImpl(Geometry geometry) {
      _geometry = geometry;
    }

    @Override
    public void handleEntity(Object bean) {

      Stop stop = (Stop) bean;

      Point point = _factory.createPoint(new Coordinate(stop.getLat(),stop.getLon()));
      if (_geometry.contains(point))
        System.out.println(stop.getLat() + " " + stop.getLon() + " "
            + stop.getId());
    }

  }
}