org.opentripplanner.ext.flex.FlexIndex 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.flex;

import com.google.common.collect.ArrayListMultimap;
import com.google.common.collect.HashMultimap;
import com.google.common.collect.Multimap;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import org.opentripplanner.ext.flex.trip.FlexTrip;
import org.opentripplanner.model.PathTransfer;
import org.opentripplanner.routing.api.request.StreetMode;
import org.opentripplanner.transit.model.framework.FeedScopedId;
import org.opentripplanner.transit.model.network.Route;
import org.opentripplanner.transit.model.site.GroupStop;
import org.opentripplanner.transit.model.site.StopLocation;
import org.opentripplanner.transit.service.TimetableRepository;

public class FlexIndex {

  private final Multimap transfersToStop = ArrayListMultimap.create();

  private final Multimap transfersFromStop = ArrayListMultimap.create();

  private final Multimap> flexTripsByStop = HashMultimap.create();

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

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

  public FlexIndex(TimetableRepository timetableRepository) {
    // Flex transfers should only use WALK mode transfers.
    for (PathTransfer transfer : timetableRepository.findTransfers(StreetMode.WALK)) {
      transfersToStop.put(transfer.to, transfer);
      transfersFromStop.put(transfer.from, transfer);
    }
    for (FlexTrip flexTrip : timetableRepository.getAllFlexTrips()) {
      routeById.put(flexTrip.getTrip().getRoute().getId(), flexTrip.getTrip().getRoute());
      tripById.put(flexTrip.getTrip().getId(), flexTrip);
      for (StopLocation stop : flexTrip.getStops()) {
        if (stop instanceof GroupStop groupStop) {
          for (StopLocation stopElement : groupStop.getChildLocations()) {
            flexTripsByStop.put(stopElement, flexTrip);
          }
        } else {
          flexTripsByStop.put(stop, flexTrip);
        }
      }
    }
  }

  public Collection getTransfersToStop(StopLocation stopLocation) {
    return transfersToStop.get(stopLocation);
  }

  public Collection getTransfersFromStop(StopLocation stopLocation) {
    return transfersFromStop.get(stopLocation);
  }

  public Collection> getFlexTripsByStop(StopLocation stopLocation) {
    return flexTripsByStop.get(stopLocation);
  }

  public Route getRouteById(FeedScopedId id) {
    return routeById.get(id);
  }

  public boolean contains(Route route) {
    return routeById.containsKey(route.getId());
  }

  public Collection getAllFlexRoutes() {
    return routeById.values();
  }

  public FlexTrip getTripById(FeedScopedId id) {
    return tripById.get(id);
  }

  public Collection> getAllFlexTrips() {
    return tripById.values();
  }
}