org.opentripplanner.model.plan.FrequencyTransitLeg Maven / Gradle / Ivy

Go to download
package org.opentripplanner.model.plan;

import java.time.LocalDate;
import java.time.ZoneId;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.GregorianCalendar;
import java.util.List;
import org.opentripplanner.model.StopLocation;
import org.opentripplanner.model.TripPattern;
import org.opentripplanner.model.transfer.ConstrainedTransfer;
import org.opentripplanner.routing.trippattern.TripTimes;

/**
 * One leg of a trip -- that is, a temporally continuous piece of the journey that takes place on a
 * particular vehicle, which does run using a frequency-based schedule, rather than a timetable-
 * based schedule.
 */
public class FrequencyTransitLeg extends ScheduledTransitLeg {

    private final int frequencyHeadwayInSeconds;

    public FrequencyTransitLeg(
            TripTimes tripTimes,
            TripPattern tripPattern,
            int boardStopIndexInPattern,
            int alightStopIndexInPattern,
            Calendar startTime,
            Calendar endTime,
            LocalDate serviceDate,
            ZoneId zoneId,
            ConstrainedTransfer transferFromPreviousLeg,
            ConstrainedTransfer transferToNextLeg,
            int generalizedCost,
            int frequencyHeadwayInSeconds
    ) {
        super(tripTimes,
                tripPattern,
                boardStopIndexInPattern,
                alightStopIndexInPattern,
                startTime,
                endTime,
                serviceDate,
                zoneId,
                transferFromPreviousLeg,
                transferToNextLeg,
                generalizedCost
        );
        this.frequencyHeadwayInSeconds = frequencyHeadwayInSeconds;
    }

    @Override
    public Boolean getNonExactFrequency() {
        return frequencyHeadwayInSeconds != 0;
    }

    @Override
    public Integer getHeadway() {
        return frequencyHeadwayInSeconds;
    }

    @Override
    public List getIntermediateStops() {
        List visits = new ArrayList<>();

        for (int i = boardStopPosInPattern + 1; i < alightStopPosInPattern; i++) {
            StopLocation stop = tripPattern.getStop(i);

            int arrivalTime = tripTimes.getArrivalTime(i);
            int departureTime = tripTimes.getDepartureTime(i) + frequencyHeadwayInSeconds;

            StopArrival visit = new StopArrival(
                    Place.forStop(stop),
                    GregorianCalendar.from(serviceDate.toZonedDateTime(zoneId, arrivalTime)),
                    GregorianCalendar.from(serviceDate.toZonedDateTime(zoneId, departureTime)),
                    i,
                    tripTimes.getOriginalGtfsStopSequence(i)
            );
            visits.add(visit);
        }
        return visits;
    }
}