org.opentripplanner.ext.fares.impl.GtfsFaresService 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.fares.impl;

import java.util.Collection;
import java.util.Objects;
import org.opentripplanner.ext.fares.model.LegProducts;
import org.opentripplanner.model.fare.ItineraryFares;
import org.opentripplanner.model.plan.Itinerary;
import org.opentripplanner.routing.fares.FareService;

public record GtfsFaresService(DefaultFareService faresV1, GtfsFaresV2Service faresV2)
  implements FareService {
  @Override
  public ItineraryFares calculateFares(Itinerary itinerary) {
    var fare = Objects.requireNonNullElse(
      faresV1.calculateFares(itinerary),
      ItineraryFares.empty()
    );
    var products = faresV2.getProducts(itinerary);
    fare.addItineraryProducts(products.itineraryProducts());
    if (products.itineraryProducts().isEmpty()) {
      addLegProducts(products.legProducts(), fare);
    }
    return fare;
  }
  /**
   * Add a complex set of fare products for a specific leg;
   */
  private static void addLegProducts(Collection legProducts, ItineraryFares fares) {
    legProducts.forEach(lp -> {
      lp
        .products()
        .stream()
        .map(LegProducts.ProductWithTransfer::products)
        .forEach(fp -> {
          fares.addFareProduct(lp.leg(), fp);
        });
    });
  }
}