org.opentripplanner.ext.siri.updater.EstimatedTimetableHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of otp Show documentation

The OpenTripPlanner multimodal journey planning system

The newest version!

package org.opentripplanner.ext.siri.updater;

import java.util.List;
import org.opentripplanner.ext.siri.SiriTimetableSnapshotSource;
import org.opentripplanner.updater.RealTimeUpdateContext;
import org.opentripplanner.updater.spi.UpdateResult;
import org.opentripplanner.updater.trip.UpdateIncrementality;
import uk.org.siri.siri20.EstimatedTimetableDeliveryStructure;

/**
 * A consumer of estimated timetables that applies the real-time updates to the transit model.
 */
public class EstimatedTimetableHandler {

  private final SiriTimetableSnapshotSource snapshotSource;
  private final boolean fuzzyTripMatching;
  /**
   * The ID for the static feed to which these real time updates are applied
   */
  private final String feedId;

  public EstimatedTimetableHandler(
    SiriTimetableSnapshotSource snapshotSource,
    boolean fuzzyTripMatching,
    String feedId
  ) {
    this.snapshotSource = snapshotSource;
    this.fuzzyTripMatching = fuzzyTripMatching;
    this.feedId = feedId;
  }

  /**
   * Apply the update to the transit model.
   */
  public UpdateResult applyUpdate(
    List estimatedTimetableDeliveries,
    UpdateIncrementality updateMode,
    RealTimeUpdateContext context
  ) {
    return snapshotSource.applyEstimatedTimetable(
      fuzzyTripMatching ? context.siriFuzzyTripMatcher() : null,
      context.entityResolver(feedId),
      feedId,
      updateMode,
      estimatedTimetableDeliveries
    );
  }
}