org.opentripplanner.routing.algorithm.mapping.ItinerariesHelper 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.routing.algorithm.mapping;

import java.util.List;
import java.util.OptionalDouble;
import org.opentripplanner.model.plan.Itinerary;
import org.opentripplanner.model.plan.StreetLeg;
import org.opentripplanner.model.plan.WalkStep;
import org.opentripplanner.routing.api.request.preference.WheelchairPreferences;
import org.opentripplanner.street.model.edge.StreetEdge;

public class ItinerariesHelper {

  public static void decorateItinerariesWithRequestData(
    List itineraries,
    boolean wheelchairEnabled,
    WheelchairPreferences wheelchairPreferences
  ) {
    if (!wheelchairEnabled) {
      return;
    }
    for (Itinerary it : itineraries) {
      // Communicate the fact that the only way we were able to get a response
      // was by removing a slope limit.
      OptionalDouble maxSlope = getMaxSlope(it);
      if (maxSlope.isPresent()) {
        it.setTooSloped(maxSlope.getAsDouble() > wheelchairPreferences.maxSlope());
        it.setMaxSlope(maxSlope.getAsDouble());
      }
    }
  }

  private static OptionalDouble getMaxSlope(Itinerary it) {
    return it
      .getLegs()
      .stream()
      .filter(StreetLeg.class::isInstance)
      .map(StreetLeg.class::cast)
      .map(StreetLeg::getWalkSteps)
      .flatMap(List::stream)
      .map(WalkStep::getEdges)
      .filter(StreetEdge.class::isInstance)
      .map(StreetEdge.class::cast)
      .mapToDouble(StreetEdge::getMaxSlope)
      .max();
  }
}