org.opentripplanner.routing.TripTimesShortHelper Maven / Gradle / Ivy

Go to download
package org.opentripplanner.routing;

import org.opentripplanner.model.TripTimeOnDate;
import org.opentripplanner.model.Timetable;
import org.opentripplanner.model.TimetableSnapshot;
import org.opentripplanner.model.Trip;
import org.opentripplanner.model.TripPattern;
import org.opentripplanner.model.calendar.ServiceDate;
import org.opentripplanner.routing.core.ServiceDay;
import org.opentripplanner.routing.trippattern.TripTimes;

import java.util.ArrayList;
import java.util.List;

public class TripTimesShortHelper {
  public static List getTripTimesShort(RoutingService routingService, Trip trip, ServiceDate serviceDate) {
    final ServiceDay serviceDay = new ServiceDay(routingService.getServiceCodes(),
        serviceDate,
        routingService.getCalendarService(),
        trip.getRoute().getAgency().getId()
    );
    Timetable timetable = null;
    TimetableSnapshot timetableSnapshot = routingService.getTimetableSnapshot();
    if (timetableSnapshot != null) {
      // Check if realtime-data is available for trip

      TripPattern pattern = timetableSnapshot.getLastAddedTripPattern(trip.getId(), serviceDate);
      if (pattern == null) {
        pattern = routingService.getPatternForTrip().get(trip);
      }
      timetable = timetableSnapshot.resolve(pattern, serviceDate);

      // If realtime moved pattern back to original trip, fetch it instead
      if (timetable.getTripIndex(trip.getId()) == -1) {
        pattern = routingService.getPatternForTrip().get(trip);
        timetable = timetableSnapshot.resolve(pattern, serviceDate);
      }
    }
    if (timetable == null) {
      timetable = routingService.getPatternForTrip().get(trip).getScheduledTimetable();
    }

    // This check is made here to avoid changing TripTimeShort.fromTripTimes
    TripTimes times = timetable.getTripTimes(timetable.getTripIndex(trip.getId()));
    if (!serviceDay.serviceRunning(times.getServiceCode())) {
      return new ArrayList<>();
    }
    else {
      return TripTimeOnDate.fromTripTimes(timetable, trip, serviceDay);
    }
  }

}