Add implementation for Service Journey filter

Author: bartosz

src/main/java/org/opentripplanner/apis/transmodel/mapping/SelectRequestMapper.java

@@ -32,6 +32,11 @@ static SelectRequest mapSelectRequest(Map<String, List<?>> input) {
       selectRequestBuilder.withGroupOfRoutes(mapIDsToDomainNullSafe(groupOfLines));
     }
 
+    if (input.containsKey("serviceJourneys")) {
+      var serviceJourneys = (List<String>) input.get("serviceJourneys");
+      selectRequestBuilder.withTrips(mapIDsToDomainNullSafe(serviceJourneys));
+    }
+
     if (input.containsKey("transportModes")) {
       var tModes = new ArrayList<MainAndSubMode>();
 

src/main/java/org/opentripplanner/routing/algorithm/raptoradapter/transit/request/RaptorRoutingRequestTransitDataCreator.java

@@ -181,18 +181,24 @@ private static List<TripPatternForDate> filterActiveTripPatterns(
     // TripPatternForDate objects that start on that particular day. This is to prevent duplicates.
     // This was previously a stream, but was unrolled for improved performance.
 
+    var hasTripFilters = filter.hasTripFilters();
     Predicate<TripTimes> tripTimesWithSubmodesPredicate = tripTimes ->
-      filter.tripTimesPredicate(tripTimes, filter.hasSubModeFilters());
+      filter.tripTimesPredicate(tripTimes, filter.hasSubModeFilters() || hasTripFilters);
+
     Predicate<TripTimes> tripTimesWithoutSubmodesPredicate = tripTimes ->
       filter.tripTimesPredicate(tripTimes, false);
+
     Collection<TripPatternForDate> tripPatternsForDate = transitLayer.getTripPatternsForDate(date);
     List<TripPatternForDate> result = new ArrayList<>(tripPatternsForDate.size());
+
     for (TripPatternForDate p : tripPatternsForDate) {
       if (firstDay || p.getStartOfRunningPeriod().equals(date)) {
         if (filter.tripPatternPredicate(p)) {
-          var tripTimesPredicate = p.getTripPattern().getPattern().getContainsMultipleModes()
+          var tripTimesPredicate = hasTripFilters ||
+            p.getTripPattern().getPattern().getContainsMultipleModes()
             ? tripTimesWithSubmodesPredicate
             : tripTimesWithoutSubmodesPredicate;
+
           TripPatternForDate tripPatternForDate = p.newWithFilteredTripTimes(tripTimesPredicate);
           if (tripPatternForDate != null) {
             result.add(tripPatternForDate);

src/main/java/org/opentripplanner/routing/algorithm/raptoradapter/transit/request/RouteRequestTransitDataProviderFilter.java

@@ -38,6 +38,8 @@ public class RouteRequestTransitDataProviderFilter implements TransitDataProvide
 
   private final boolean hasSubModeFilters;
 
+  private final boolean hasTripFilters;
+
   public RouteRequestTransitDataProviderFilter(RouteRequest request) {
     this(
       request.journey().transfer().mode() == StreetMode.BIKE,
@@ -68,13 +70,19 @@ public RouteRequestTransitDataProviderFilter(
     this.bannedTrips = bannedTrips;
     this.filters = filters.toArray(TransitFilter[]::new);
     this.hasSubModeFilters = filters.stream().anyMatch(TransitFilter::isSubModePredicate);
+    this.hasTripFilters = filters.stream().anyMatch(TransitFilter::isTripPredicate);
   }
 
   @Override
   public boolean hasSubModeFilters() {
     return hasSubModeFilters;
   }
 
+  @Override
+  public boolean hasTripFilters() {
+    return hasTripFilters;
+  }
+
   public static BikeAccess bikeAccessForTrip(Trip trip) {
     if (trip.getBikesAllowed() != BikeAccess.UNKNOWN) {
       return trip.getBikesAllowed();

src/main/java/org/opentripplanner/routing/algorithm/raptoradapter/transit/request/TransitDataProviderFilter.java

@@ -20,6 +20,8 @@ public interface TransitDataProviderFilter {
 
   boolean hasSubModeFilters();
 
+  boolean hasTripFilters();
+
   boolean tripTimesPredicate(TripTimes tripTimes, boolean withFilters);
 
   /**

src/main/java/org/opentripplanner/routing/api/request/request/filter/SelectRequest.java

@@ -22,6 +22,7 @@ public static Builder of() {
   private final List<FeedScopedId> agencies;
   private final List<FeedScopedId> groupOfRoutes;
   private final List<FeedScopedId> routes;
+  private final List<FeedScopedId> trips;
 
   public SelectRequest(Builder builder) {
     if (builder.transportModes.isEmpty()) {
@@ -35,7 +36,17 @@ public SelectRequest(Builder builder) {
 
     this.agencies = List.copyOf(builder.agencies);
     this.groupOfRoutes = List.copyOf(builder.groupOfRoutes);
-    this.routes = builder.routes;
+    this.routes = List.copyOf(builder.routes);
+    this.trips = List.copyOf(builder.trips);
+  }
+
+  public boolean isTripsOnly() {
+    return (
+      this.transportModeFilter == null &&
+      this.agencies.isEmpty() &&
+      this.groupOfRoutes.isEmpty() &&
+      this.routes.isEmpty()
+    );
   }
 
   public boolean matches(TripPattern tripPattern) {
@@ -77,21 +88,32 @@ public boolean matches(TripPattern tripPattern) {
   public boolean matchesSelect(TripTimes tripTimes) {
     var trip = tripTimes.getTrip();
 
-    return (
+    var tripFilter = this.getTrips();
+
+    var matchesTrip =
+      (tripFilter == null || tripFilter.isEmpty()) || tripFilter.contains(trip.getId());
+    var matchesTransportMode =
       this.transportModeFilter == null ||
-      this.transportModeFilter.match(trip.getMode(), trip.getNetexSubMode())
-    );
+      this.transportModeFilter.match(trip.getMode(), trip.getNetexSubMode());
+
+    return matchesTrip && matchesTransportMode;
   }
 
   /**
    * Matches the not clause of a transit filter request.
    */
   public boolean matchesNot(TripTimes tripTimes) {
     var trip = tripTimes.getTrip();
-    return (
+
+    var tripFilter = this.getTrips();
+
+    var matchesTrip =
+      (tripFilter != null && !tripFilter.isEmpty()) && tripFilter.contains(trip.getId());
+    var matchesTransportMode =
       this.transportModeFilter != null &&
-      this.transportModeFilter.match(trip.getMode(), trip.getNetexSubMode())
-    );
+      this.transportModeFilter.match(trip.getMode(), trip.getNetexSubMode());
+
+    return matchesTrip || matchesTransportMode;
   }
 
   @Override
@@ -120,6 +142,10 @@ public List<FeedScopedId> routes() {
     return routes;
   }
 
+  public List<FeedScopedId> getTrips() {
+    return trips;
+  }
+
   private String transportModesToString() {
     if (transportModes == null) {
       return null;
@@ -146,6 +172,7 @@ public static class Builder {
     private List<FeedScopedId> agencies = new ArrayList<>();
     private List<FeedScopedId> groupOfRoutes = new ArrayList<>();
     private List<FeedScopedId> routes = new ArrayList<>();
+    private List<FeedScopedId> trips = new ArrayList<>();
 
     public Builder withTransportModes(List<MainAndSubMode> transportModes) {
       this.transportModes = transportModes;
@@ -186,6 +213,11 @@ public Builder withGroupOfRoutes(List<FeedScopedId> groupOfRoutes) {
       return this;
     }
 
+    public Builder withTrips(List<FeedScopedId> trips) {
+      this.trips = trips;
+      return this;
+    }
+
     public SelectRequest build() {
       return new SelectRequest(this);
     }

src/main/java/org/opentripplanner/routing/api/request/request/filter/TransitFilter.java

@@ -17,4 +17,8 @@ public interface TransitFilter {
   default boolean isSubModePredicate() {
     return false;
   }
+
+  default boolean isTripPredicate() {
+    return false;
+  }
 }

src/main/java/org/opentripplanner/routing/api/request/request/filter/TransitFilterRequest.java

@@ -57,9 +57,27 @@ public boolean isSubModePredicate() {
     return false;
   }
 
+  @Override
+  public boolean isTripPredicate() {
+    for (var selectRequest : select) {
+      if (selectRequest.getTrips() != null && !selectRequest.getTrips().isEmpty()) {
+        return true;
+      }
+    }
+
+    for (var selectRequest : not) {
+      if (selectRequest.getTrips() != null && !selectRequest.getTrips().isEmpty()) {
+        return true;
+      }
+    }
+    return false;
+  }
+
   @Override
   public boolean matchTripPattern(TripPattern tripPattern) {
-    if (select.length != 0) {
+    var tripPatternSelect = Arrays.stream(select).filter(s -> !s.isTripsOnly()).toArray();
+
+    if (tripPatternSelect.length != 0) {
       var anyMatch = false;
       for (SelectRequest s : select) {
         if (s.matches(tripPattern)) {
@@ -73,7 +91,8 @@ public boolean matchTripPattern(TripPattern tripPattern) {
     }
 
     for (SelectRequest s : not) {
-      if (s.matches(tripPattern)) {
+      // We'll filter trips in the next step
+      if (!s.isTripsOnly() && s.matches(tripPattern)) {
         return false;
       }
     }

src/test/java/org/opentripplanner/routing/algorithm/raptoradapter/transit/request/RaptorRoutingRequestTransitDataCreatorTest.java

@@ -144,6 +144,11 @@ public boolean hasSubModeFilters() {
       return false;
     }
 
+    @Override
+    public boolean hasTripFilters() {
+      return false;
+    }
+
     @Override
     public BitSet filterAvailableStops(
       RoutingTripPattern tripPattern,