fr.itinerennes.bundler.integration.onebusaway.GenerateStaticObaApiResults.java Source code

Introduction

Here is the source code for fr.itinerennes.bundler.integration.onebusaway.GenerateStaticObaApiResults.java

Source

package fr.itinerennes.bundler.integration.onebusaway;

/*
 * [license]
 * Itinerennes data resources generator
 * ~~~~
 * Copyright (C) 2013 - 2014 Dudie
 * ~~~~
 * 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 3 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, see
 * <http://www.gnu.org/licenses/gpl-3.0.html>.
 * [/license]
 */

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.Writer;
import java.util.Calendar;
import java.util.HashMap;
import java.util.Map;

import org.apache.http.impl.client.DefaultHttpClient;
import org.onebusaway.gtfs.model.Stop;
import org.onebusaway.gtfs.model.Trip;
import org.onebusaway.gtfs.services.GtfsDao;

import com.google.gson.Gson;

import fr.dudie.onebusaway.client.JsonOneBusAwayClient;
import fr.dudie.onebusaway.gson.OneBusAwayGsonFactory;
import fr.dudie.onebusaway.model.StopSchedule;
import fr.dudie.onebusaway.model.TripSchedule;
import fr.itinerennes.bundler.gtfs.GtfsException;
import fr.itinerennes.bundler.gtfs.GtfsUtils;

public class GenerateStaticObaApiResults {

    private static JsonOneBusAwayClient oba;

    private static Gson gson;

    public static void main(String[] args) throws IOException, GtfsException {

        final String url = args[0];
        final String key = args[1];
        final String gtfsFile = args[2];
        final String out = args[3].replaceAll("/$", "");

        final Map<String, String> agencyMapping = new HashMap<String, String>();
        agencyMapping.put("1", "2");
        final GtfsDao gtfs = GtfsUtils.load(new File(gtfsFile), agencyMapping);

        oba = new JsonOneBusAwayClient(new DefaultHttpClient(), url, key);
        gson = OneBusAwayGsonFactory.newInstance(true);

        final Calendar end = Calendar.getInstance();
        end.set(2013, 11, 22, 0, 0);

        final Calendar start = Calendar.getInstance();
        start.set(2013, 10, 4, 0, 0);

        final Calendar current = Calendar.getInstance();
        current.setTime(start.getTime());

        while (current.before(end) || current.equals(end)) {
            System.out.println(current.getTime());
            for (final Stop stop : gtfs.getAllStops()) {
                final String stopId = stop.getId().toString();
                final String dateDir = String.format("%04d/%02d/%02d", current.get(Calendar.YEAR),
                        current.get(Calendar.MONTH) + 1, current.get(Calendar.DAY_OF_MONTH));
                final String methodDir = String.format("%s/schedule-for-stop", out);

                final File outDir = new File(String.format("%s/%s", methodDir, dateDir));
                outDir.mkdirs();
                final File f = new File(outDir, String.format("%s.json", stopId));

                final StopSchedule ss = oba.getScheduleForStop(stopId, current.getTime());
                final String json = gson.toJson(ss);

                final Writer w = new PrintWriter(f);
                w.write(json);
                w.close();
            }
            current.add(Calendar.DAY_OF_MONTH, 1);
        }

        final File outDir = new File(String.format("%s/trip-details", out));
        outDir.mkdirs();
        for (final Trip trip : gtfs.getAllTrips()) {
            final String tripId = trip.getId().toString();
            final File f = new File(outDir, String.format("%s.json", tripId));

            final TripSchedule ts = oba.getTripDetails(tripId);
            final String json = gson.toJson(ts);

            final Writer w = new PrintWriter(f);
            w.write(json);
            w.close();
        }
    }
}