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

Go to download
package org.opentripplanner.gtfs.mapping;

import org.opentripplanner.graph_builder.DataImportIssueStore;
import org.opentripplanner.model.Route;
import org.opentripplanner.model.TransitMode;
import org.opentripplanner.util.MapUtils;

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

/** Responsible for mapping GTFS Route into the OTP model. */
class RouteMapper {
    private final AgencyMapper agencyMapper;
    private final BrandingMapper brandingMapper;

    private final DataImportIssueStore issueStore;

    private final Map mappedRoutes = new HashMap<>();

    RouteMapper(AgencyMapper agencyMapper, DataImportIssueStore issueStore) {
        this.agencyMapper = agencyMapper;
        this.issueStore = issueStore;
        this.brandingMapper = new BrandingMapper();
    }

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

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

    private Route doMap(org.onebusaway.gtfs.model.Route rhs) {
        Route lhs = new Route(AgencyAndIdMapper.mapAgencyAndId(rhs.getId()));

        lhs.setAgency(agencyMapper.map(rhs.getAgency()));
        lhs.setShortName(rhs.getShortName());
        lhs.setLongName(rhs.getLongName());
        int routeType = rhs.getType();
        lhs.setGtfsType(routeType);
        TransitMode mode = TransitModeMapper.mapMode(routeType);
        if (mode == null) {
            issueStore.add(
                    "RouteMapper", "Treating %s route type for route %s as BUS.", routeType,
                    lhs.getId().toString()
            );
            lhs.setMode(TransitMode.BUS);
        }
        else {
            lhs.setMode(mode);
        }
        lhs.setDesc(rhs.getDesc());
        lhs.setUrl(rhs.getUrl());
        lhs.setColor(rhs.getColor());
        lhs.setTextColor(rhs.getTextColor());
        lhs.setBikesAllowed(BikeAccessMapper.mapForRoute(rhs));
        lhs.setSortOrder(rhs.getSortOrder());
        lhs.setBranding(brandingMapper.map(rhs));

        return lhs;
    }
}