fr.lemet.gtfs.GtfsBase.java Source code

Introduction

Here is the source code for fr.lemet.gtfs.GtfsBase.java

Source

package fr.lemet.gtfs;
/**
 * 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.
 */

import java.io.File;
import java.io.IOException;
import java.util.Collection;
import java.util.List;
import java.util.Set;
import java.util.Vector;

import org.hibernate.SessionFactory;
import org.hibernate.cfg.Configuration;
import org.onebusaway.gtfs.model.AgencyAndId;
import org.onebusaway.gtfs.model.Stop;
import org.onebusaway.gtfs.model.calendar.ServiceDate;
import org.onebusaway.gtfs.serialization.GtfsReader;
import org.onebusaway.gtfs.services.GtfsMutableRelationalDao;
import org.onebusaway.gtfs.services.HibernateGtfsFactory;
import org.onebusaway.gtfs.services.calendar.CalendarService;

public class GtfsBase {

    private static final String KEY_CLASSPATH = "classpath:";

    private static final String KEY_FILE = "file:";

    String resource = "classpath:org/onebusaway/gtfs/examples/hibernate-configuration-examples.xml";
    HibernateGtfsFactory factory;

    GtfsBase(String gtfsPath) throws IOException {
        HibernateGtfsFactory factory = createHibernateGtfsFactory(resource);
        GtfsReader reader = new GtfsReader();
        reader.setInputLocation(new File(gtfsPath));

        GtfsMutableRelationalDao dao = factory.getDao();
        reader.setEntityStore(dao);
        reader.run();

        Collection<Stop> stops = dao.getAllStops();

        for (Stop stop : stops)
            System.out.println(stop.getName());
    }

    public List<String> service() {
        List<String> lesservices = new Vector<String>();
        CalendarService calendarService = factory.getCalendarService();
        Set<AgencyAndId> serviceIds = calendarService.getServiceIds();
        for (AgencyAndId serviceId : serviceIds) {
            Set<ServiceDate> dates = calendarService.getServiceDatesForServiceId(serviceId);
            ServiceDate from = null;
            ServiceDate to = null;
            for (ServiceDate date : dates) {
                from = min(from, date);
                to = max(to, date);
            }

            lesservices.add("serviceId=" + serviceId + " from=" + from + " to=" + to);
        }

        return lesservices;
    }

    private static ServiceDate min(ServiceDate a, ServiceDate b) {
        if (a == null)
            return b;
        if (b == null)
            return a;
        return a.compareTo(b) <= 0 ? a : b;
    }

    private static ServiceDate max(ServiceDate a, ServiceDate b) {
        if (a == null)
            return b;
        if (b == null)
            return a;
        return a.compareTo(b) <= 0 ? b : a;
    }

    public static HibernateGtfsFactory createHibernateGtfsFactory(String resource) {

        Configuration config = new Configuration();

        if (resource.startsWith(KEY_CLASSPATH)) {
            resource = resource.substring(KEY_CLASSPATH.length());
            config = config.configure(resource);
        } else if (resource.startsWith(KEY_FILE)) {
            resource = resource.substring(KEY_FILE.length());
            config = config.configure(new File(resource));
        } else {
            config = config.configure(new File(resource));
        }

        SessionFactory sessionFactory = config.buildSessionFactory();
        return new HibernateGtfsFactory(sessionFactory);
    }
}