org.opentripplanner.gtfs.mapping.LocationGroupMapper 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.gtfs.mapping;

import static org.opentripplanner.gtfs.mapping.AgencyAndIdMapper.mapAgencyAndId;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import org.onebusaway.gtfs.model.Location;
import org.onebusaway.gtfs.model.LocationGroup;
import org.onebusaway.gtfs.model.Stop;
import org.opentripplanner.framework.collection.MapUtils;
import org.opentripplanner.framework.i18n.NonLocalizedString;
import org.opentripplanner.transit.model.site.GroupStop;
import org.opentripplanner.transit.model.site.GroupStopBuilder;
import org.opentripplanner.transit.service.StopModelBuilder;

public class LocationGroupMapper {

  private final StopMapper stopMapper;
  private final LocationMapper locationMapper;
  private final StopModelBuilder stopModelBuilder;

  private final Map mappedLocationGroups = new HashMap<>();

  public LocationGroupMapper(
    StopMapper stopMapper,
    LocationMapper locationMapper,
    StopModelBuilder stopModelBuilder
  ) {
    this.stopMapper = stopMapper;
    this.locationMapper = locationMapper;
    this.stopModelBuilder = stopModelBuilder;
  }

  Collection map(Collection allLocationGroups) {
    return MapUtils.mapToList(allLocationGroups, this::map);
  }

  /** Map from GTFS to OTP model, {@code null} safe. */
  GroupStop map(LocationGroup original) {
    return original == null ? null : mappedLocationGroups.computeIfAbsent(original, this::doMap);
  }

  private GroupStop doMap(LocationGroup element) {
    GroupStopBuilder groupStopBuilder = stopModelBuilder
      .groupStop(mapAgencyAndId(element.getId()))
      .withName(new NonLocalizedString(element.getName()));

    for (var location : element.getLocations()) {
      Objects.requireNonNull(
        location,
        "Location group '%s' contains a null element.".formatted(element.getId())
      );
      switch (location) {
        case Stop stop -> groupStopBuilder.addLocation(stopMapper.map(stop));
        case Location loc -> groupStopBuilder.addLocation(locationMapper.map(loc));
        case LocationGroup ignored -> throw new RuntimeException(
          "Nested GroupStops are not allowed"
        );
        default -> throw new RuntimeException(
          "Unknown location type: " + location.getClass().getSimpleName()
        );
      }
    }

    return groupStopBuilder.build();
  }
}