org.opentripplanner.apis.transmodel.mapping.TripViaLocationMapper 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.apis.transmodel.mapping;

import static java.util.stream.Collectors.toList;

import java.time.Duration;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import javax.annotation.Nullable;
import org.opentripplanner.apis.transmodel.model.framework.CoordinateInputType;
import org.opentripplanner.apis.transmodel.model.plan.TripQuery;
import org.opentripplanner.apis.transmodel.model.plan.ViaLocationInputType;
import org.opentripplanner.apis.transmodel.support.OneOfInputValidator;
import org.opentripplanner.framework.geometry.WgsCoordinate;
import org.opentripplanner.routing.api.request.via.PassThroughViaLocation;
import org.opentripplanner.routing.api.request.via.ViaLocation;
import org.opentripplanner.routing.api.request.via.VisitViaLocation;
import org.opentripplanner.transit.model.framework.FeedScopedId;

@SuppressWarnings("unchecked")
class TripViaLocationMapper {

  static List mapToViaLocations(final List> via) {
    return via.stream().map(TripViaLocationMapper::mapViaLocation).collect(toList());
  }

  /**
   * @deprecated Legacy passThrough, use via instead
   */
  @Deprecated
  static List toLegacyPassThroughLocations(
    final List> passThroughPoints
  ) {
    return passThroughPoints
      .stream()
      .map(TripViaLocationMapper::mapLegacyPassThroughViaLocation)
      .filter(Objects::nonNull)
      .collect(toList());
  }

  private static ViaLocation mapViaLocation(Map inputMap) {
    var fieldName = OneOfInputValidator.validateOneOf(
      inputMap,
      TripQuery.TRIP_VIA_PARAMETER,
      ViaLocationInputType.FIELD_VISIT,
      ViaLocationInputType.FIELD_PASS_THROUGH
    );

    Map value = (Map) inputMap.get(fieldName);

    return switch (fieldName) {
      case ViaLocationInputType.FIELD_VISIT -> mapVisitViaLocation(value);
      case ViaLocationInputType.FIELD_PASS_THROUGH -> mapPassThroughViaLocation(value);
      default -> throw new IllegalArgumentException("Unknown field: " + fieldName);
    };
  }

  private static VisitViaLocation mapVisitViaLocation(Map inputMap) {
    var label = (String) inputMap.get(ViaLocationInputType.FIELD_LABEL);
    var minimumWaitTime = (Duration) inputMap.get(ViaLocationInputType.FIELD_MINIMUM_WAIT_TIME);
    var stopLocationIds = mapStopLocationIds(inputMap);
    var coordinate = mapCoordinate(inputMap);
    return new VisitViaLocation(label, minimumWaitTime, stopLocationIds, coordinate);
  }

  private static PassThroughViaLocation mapPassThroughViaLocation(Map inputMap) {
    var label = (String) inputMap.get(ViaLocationInputType.FIELD_LABEL);
    var stopLocationIds = mapStopLocationIds(inputMap);
    return new PassThroughViaLocation(label, stopLocationIds);
  }

  private static List mapStopLocationIds(Map map) {
    var c = (Collection) map.get(ViaLocationInputType.FIELD_STOP_LOCATION_IDS);
    return c == null ? List.of() : c.stream().map(TransitIdMapper::mapIDToDomain).toList();
  }

  private static List mapCoordinate(Map map) {
    return CoordinateInputType.mapToWgsCoordinate(ViaLocationInputType.FIELD_COORDINATE, map)
      .map(List::of)
      .orElseGet(List::of);
  }

  /**
   * @deprecated Legacy passThrough, use via instead
   */
  @Deprecated
  @Nullable
  private static ViaLocation mapLegacyPassThroughViaLocation(Map inputMap) {
    final String name = (String) inputMap.get("name");
    List placeIds = (List) inputMap.get("placeIds");
    if (placeIds == null || placeIds.isEmpty()) {
      return null;
    }
    final List stopLocationIds = placeIds
      .stream()
      .map(TransitIdMapper::mapIDToDomain)
      .toList();
    return new PassThroughViaLocation(name, stopLocationIds);
  }
}