org.opentripplanner.ext.realtimeresolver.RealtimeResolver 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.realtimeresolver;

import java.util.List;
import java.util.stream.Collectors;
import org.opentripplanner.model.plan.Itinerary;
import org.opentripplanner.model.plan.Leg;
import org.opentripplanner.model.plan.ScheduledTransitLeg;
import org.opentripplanner.model.plan.ScheduledTransitLegBuilder;
import org.opentripplanner.transit.service.TransitService;

public class RealtimeResolver {

  /**
   * Loop through all itineraries and populate legs with real-time data using legReference from the original leg
   */
  public static void populateLegsWithRealtime(
    List itineraries,
    TransitService transitService
  ) {
    itineraries.forEach(it -> {
      if (it.isFlaggedForDeletion()) {
        return;
      }
      var legs = it
        .getLegs()
        .stream()
        .map(leg -> {
          var ref = leg.getLegReference();
          if (ref == null) {
            return leg;
          }

          // Only ScheduledTransitLeg has leg references atm, so this check is just to be future-proof
          if (!(leg.isScheduledTransitLeg())) {
            return leg;
          }

          var realTimeLeg = ref.getLeg(transitService);
          if (realTimeLeg != null) {
            return combineReferenceWithOriginal(
              realTimeLeg.asScheduledTransitLeg(),
              leg.asScheduledTransitLeg()
            );
          }
          return leg;
        })
        .collect(Collectors.toList());

      it.setLegs(legs);
    });
  }

  private static Leg combineReferenceWithOriginal(
    ScheduledTransitLeg reference,
    ScheduledTransitLeg original
  ) {
    var leg = new ScheduledTransitLegBuilder<>(reference)
      .withTransferFromPreviousLeg(original.getTransferFromPrevLeg())
      .withTransferToNextLeg(original.getTransferToNextLeg())
      .withGeneralizedCost(original.getGeneralizedCost())
      .withAccessibilityScore(original.accessibilityScore())
      .build();
    reference.getTransitAlerts().forEach(leg::addAlert);
    return leg;
  }
}