org.opentripplanner.ext.restapi.mapping.StopMapper 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.ext.restapi.mapping;

import java.util.Collection;
import java.util.List;
import java.util.stream.Collectors;
import org.opentripplanner.ext.restapi.model.ApiStop;
import org.opentripplanner.ext.restapi.model.ApiStopShort;
import org.opentripplanner.framework.i18n.I18NStringMapper;
import org.opentripplanner.transit.model.site.StopLocation;

public class StopMapper {

  public static List mapToApi(Collection domain) {
    if (domain == null) {
      return null;
    }
    return domain.stream().map(StopMapper::mapToApi).collect(Collectors.toList());
  }

  public static ApiStop mapToApi(StopLocation domain) {
    return mapToApi(domain, true);
  }

  public static ApiStop mapToApi(StopLocation domain, boolean extended) {
    if (domain == null) {
      return null;
    }

    ApiStop api = new ApiStop();
    api.id = FeedScopedIdMapper.mapToApi(domain.getId());
    api.lat = domain.getLat();
    api.lon = domain.getLon();
    api.code = domain.getCode();
    api.name = domain.getName().toString();
    if (extended) {
      String desc = domain.getDescription() != null ? domain.getDescription().toString() : null;
      api.desc = desc;
      api.zoneId = domain.getFirstZoneAsString();
      api.url = I18NStringMapper.mapToApi(domain.getUrl(), null);
      api.locationType = 0;
      api.stationId = FeedScopedIdMapper.mapIdToApi(domain.getParentStation());
      api.parentStation = mapToParentStationOldId(domain);
      //api.stopTimezone = stop.getTimezone();
      api.wheelchairBoarding = WheelchairAccessibilityMapper.mapToApi(
        domain.getWheelchairAccessibility()
      );
      //api.direction = stop.getDirection();
    }
    return api;
  }

  public static ApiStopShort mapToApiShort(StopLocation domain) {
    if (domain == null) {
      return null;
    }

    ApiStopShort api = new ApiStopShort();
    api.id = FeedScopedIdMapper.mapToApi(domain.getId());
    api.code = domain.getCode();
    api.name = domain.getName().toString();
    api.lat = domain.getLat();
    api.lon = domain.getLon();
    api.url = I18NStringMapper.mapToApi(domain.getUrl(), null);
    api.stationId = FeedScopedIdMapper.mapIdToApi(domain.getParentStation());
    // parentStation may be missing on the stop returning null.
    // TODO harmonize these names, maybe use "station" everywhere
    api.cluster = mapToParentStationOldId(domain);

    return api;
  }

  /** @param distance in integral meters, to avoid serializing a bunch of decimal places. */
  public static ApiStopShort mapToApiShort(StopLocation domain, double distance) {
    if (domain == null) {
      return null;
    }

    ApiStopShort api = mapToApiShort(domain);
    api.dist = (int) distance;

    return api;
  }

  public static List mapToApiShort(Collection domain) {
    if (domain == null) {
      return null;
    }
    return domain.stream().map(StopMapper::mapToApiShort).toList();
  }

  /**
   * Get the parent station id (without feed-scope) for the given station element, this method
   * should only be used to fetch old ids for beeing backward compatible.
   */
  private static String mapToParentStationOldId(StopLocation stop) {
    return stop.isPartOfStation() ? stop.getParentStation().getId().getId() : null;
  }
}