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 application Show documentation

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
  ) {
    return new ScheduledTransitLegBuilder<>(reference)
      .withTransferFromPreviousLeg(original.getTransferFromPrevLeg())
      .withTransferToNextLeg(original.getTransferToNextLeg())
      .withGeneralizedCost(original.getGeneralizedCost())
      .withAccessibilityScore(original.accessibilityScore())
      .build();
  }
}