org.opentripplanner.ext.siri.SiriTransportModeMapper Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of otp Show documentation

The OpenTripPlanner multimodal journey planning system

There is a newer version: 2.5.0

package org.opentripplanner.ext.siri;

import org.opentripplanner.model.TransitMode;
import uk.org.siri.siri20.VehicleModesEnumeration;

import java.util.List;

public class SiriTransportModeMapper {

  /**
   * Maps first SIRI-VehicleMode to OTP-mode
   * @param vehicleModes
   * @return
   */
  static TransitMode mapTransitMainMode(List vehicleModes) {
    if (vehicleModes != null && !vehicleModes.isEmpty()) {
      VehicleModesEnumeration vehicleModesEnumeration = vehicleModes.get(0);
      switch (vehicleModesEnumeration) {
        case RAIL:
          return TransitMode.RAIL;
        case COACH:
          return TransitMode.COACH;
        case BUS:
          return TransitMode.BUS;
        case METRO:
        case UNDERGROUND:
          return TransitMode.SUBWAY;
        case TRAM:
          return TransitMode.TRAM;
        case FERRY:
          return TransitMode.FERRY;
        case AIR:
          return TransitMode.AIRPLANE;
      }
    }
    return TransitMode.BUS;
  }
}