org.opentripplanner.gtfs.mapping.PathwayMapper 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
There is a newer version: 2.6.0
package org.opentripplanner.gtfs.mapping;

import org.opentripplanner.model.Pathway;
import org.opentripplanner.util.MapUtils;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

/** Responsible for mapping GTFS Pathway into the OTP model. */
class PathwayMapper {

    private final StopMapper stopMapper;

    private EntranceMapper entranceMapper;

    private PathwayNodeMapper nodeMapper;

    private BoardingAreaMapper boardingAreaMapper;

    private Map mappedPathways = new HashMap<>();

    PathwayMapper(
        StopMapper stopMapper,
        EntranceMapper entranceMapper,
        PathwayNodeMapper nodeMapper,
        BoardingAreaMapper boardingAreaMapper
    ) {
        this.stopMapper = stopMapper;
        this.entranceMapper = entranceMapper;
        this.nodeMapper = nodeMapper;
        this.boardingAreaMapper = boardingAreaMapper;
    }

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

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

    private Pathway doMap(org.onebusaway.gtfs.model.Pathway rhs) {
        Pathway lhs = new Pathway();

        lhs.setId(AgencyAndIdMapper.mapAgencyAndId(rhs.getId()));
        lhs.setPathwayMode(rhs.getPathwayMode());
        if (rhs.isTraversalTimeSet()) { lhs.setTraversalTime(rhs.getTraversalTime()); }
        lhs.setName(rhs.getSignpostedAs());
        lhs.setReversedName(rhs.getReversedSignpostedAs());
        if (rhs.isLengthSet()) { lhs.setLength(rhs.getLength()); }
        if (rhs.isStairCountSet()) { lhs.setStairCount(rhs.getStairCount()); }
        if (rhs.isMaxSlopeSet()) { lhs.setSlope(rhs.getMaxSlope()); }
        lhs.setBidirectional(rhs.getIsBidirectional() == 1);

        org.onebusaway.gtfs.model.Stop fromStop = rhs.getFromStop();
        if (fromStop != null) {
            switch (rhs.getFromStop().getLocationType()) {
                case org.onebusaway.gtfs.model.Stop.LOCATION_TYPE_STOP:
                    lhs.setFromStop(stopMapper.map(fromStop));
                    break;
                case org.onebusaway.gtfs.model.Stop.LOCATION_TYPE_ENTRANCE_EXIT:
                    lhs.setFromStop(entranceMapper.map(fromStop));
                    break;
                case org.onebusaway.gtfs.model.Stop.LOCATION_TYPE_NODE:
                    lhs.setFromStop(nodeMapper.map(fromStop));
                    break;
                case org.onebusaway.gtfs.model.Stop.LOCATION_TYPE_BOARDING_AREA:
                    lhs.setFromStop(boardingAreaMapper.map(fromStop));
                    break;
            }
        }

        org.onebusaway.gtfs.model.Stop toStop = rhs.getToStop();
        if (toStop != null) {
            switch (toStop.getLocationType()) {
                case org.onebusaway.gtfs.model.Stop.LOCATION_TYPE_STOP:
                    lhs.setToStop(stopMapper.map(toStop));
                    break;
                case org.onebusaway.gtfs.model.Stop.LOCATION_TYPE_ENTRANCE_EXIT:
                    lhs.setToStop(entranceMapper.map(toStop));
                    break;
                case org.onebusaway.gtfs.model.Stop.LOCATION_TYPE_NODE:
                    lhs.setToStop(nodeMapper.map(toStop));
                    break;
                case org.onebusaway.gtfs.model.Stop.LOCATION_TYPE_BOARDING_AREA:
                    lhs.setToStop(boardingAreaMapper.map(toStop));
                    break;
            }
        }

        return lhs;
    }
}