org.opentripplanner.netex.mapping.TransportModeMapper 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.netex.mapping;

import org.opentripplanner.common.model.T2;
import org.opentripplanner.model.TransitMode;
import org.rutebanken.netex.model.AllVehicleModesOfTransportEnumeration;
import org.rutebanken.netex.model.TransportSubmodeStructure;

/**
 * This is a best effort at mapping the NeTEx transport modes to the OTP route codes which are identical to the
 * GTFS extended route types
 */
class TransportModeMapper {

    public T2 map(
        AllVehicleModesOfTransportEnumeration netexMode,
        TransportSubmodeStructure submode
    ) {
        if (submode != null) {
            return getSubmodeAsString(submode);
        }
        return new T2<>(mapAllVehicleModesOfTransport(netexMode), null);
    }

    private TransitMode mapAllVehicleModesOfTransport(AllVehicleModesOfTransportEnumeration mode) {
        switch (mode) {
            case AIR:
                return TransitMode.AIRPLANE;
            case BUS:
            case TAXI:
                return TransitMode.BUS;
            case CABLEWAY:
                return TransitMode.CABLE_CAR;
            case COACH:
                return TransitMode.COACH;
            case FUNICULAR:
                return TransitMode.FUNICULAR;
            case METRO:
                return TransitMode.SUBWAY;
            case RAIL:
                return TransitMode.RAIL;
            case TRAM:
                return TransitMode.TRAM;
            case WATER:
                return TransitMode.FERRY;
            default:
                throw new IllegalArgumentException(mode.toString());
        }
    }

    private T2 getSubmodeAsString(TransportSubmodeStructure submode) {
        if (submode.getAirSubmode() != null) {
            return new T2<>(TransitMode.AIRPLANE, submode.getAirSubmode().value());
        } else if (submode.getBusSubmode() != null) {
            return new T2<>(TransitMode.BUS, submode.getBusSubmode().value());
        } else if (submode.getTelecabinSubmode() != null) {
            return new T2<>(TransitMode.GONDOLA, submode.getTelecabinSubmode().value());
        } else if (submode.getCoachSubmode() != null) {
            return new T2<>(TransitMode.COACH, submode.getCoachSubmode().value());
        } else if (submode.getFunicularSubmode() != null) {
            return new T2<>(TransitMode.FUNICULAR, submode.getFunicularSubmode().value());
        } else if (submode.getMetroSubmode() != null) {
            return new T2<>(TransitMode.SUBWAY, submode.getMetroSubmode().value());
        } else if (submode.getRailSubmode() != null) {
            return new T2<>(TransitMode.RAIL, submode.getRailSubmode().value());
        } else if (submode.getTramSubmode() != null) {
            return new T2<>(TransitMode.TRAM, submode.getTramSubmode().value());
        } else if (submode.getWaterSubmode() != null) {
            return new T2<>(TransitMode.FERRY, submode.getWaterSubmode().value());
        }
        throw new IllegalArgumentException();
    }
}