org.opentripplanner.gtfs.mapping.StationMapper Maven / Gradle / Ivy

package org.opentripplanner.gtfs.mapping;

import org.opentripplanner.model.Station;

import java.util.HashMap;
import java.util.Map;
import java.util.TimeZone;

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

/**
 * Responsible for mapping GTFS Stop into the OTP model.
 * 
 * NOTE! This class has state. This class also holds a index of all mapped stops to avoid
 * mapping the same stop twice. We do this because the library (onebusaway) return transfers with
 * Stop object references, not stop ids. Instead of looking up the Stops by id in the {@link
 * TransferMapper} we just use the this class to cache stops. This way, the order of which stops
 * and transfers are mapped does not matter.
 */
class StationMapper {

  /** @see StationMapper (this class JavaDoc) for way we need this. */
  private Map mappedStops = new HashMap<>();

  /** Map from GTFS to OTP model, {@code null} safe. */
  Station map(org.onebusaway.gtfs.model.Stop orginal) {
    return orginal == null ? null : mappedStops.computeIfAbsent(orginal, this::doMap);
  }

  private Station doMap(org.onebusaway.gtfs.model.Stop rhs) {
    if (rhs.getLocationType() != org.onebusaway.gtfs.model.Stop.LOCATION_TYPE_STATION) {
      throw new IllegalArgumentException(
          "Expected type " + org.onebusaway.gtfs.model.Stop.LOCATION_TYPE_STATION + ", but got "
              + rhs.getLocationType());
    }

    return new Station(
        mapAgencyAndId(rhs.getId()),
        rhs.getName(),
        WgsCoordinateMapper.mapToDomain(rhs),
        rhs.getCode(),
        rhs.getDesc(),
        rhs.getUrl(),
        rhs.getTimezone() == null ? null : TimeZone.getTimeZone(rhs.getTimezone()),
        // Use default cost priority
        null
    );
  }
}