Add type to geocoding results

ibi-group · May 29, 2024 · c27e8ce · c27e8ce
1 parent f78d9c9
commit c27e8ce
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/ext/java/org/opentripplanner/ext/geocoder/StopCluster.java b/src/ext/java/org/opentripplanner/ext/geocoder/StopCluster.java
@@ -31,9 +31,15 @@ public record Agency(FeedScopedId id, String name) {}
    */
   public record FeedPublisher(String name) {}
 
+  public enum LocationType {
+    STATION,
+    STOP,
+  }
+
   public record Location(
     FeedScopedId id,
     @Nullable String code,
+    LocationType type,
     String name,
     Coordinate coordinate,
     Collection<String> modes,
@@ -43,6 +49,7 @@ public record Location(
     public Location {
       Objects.requireNonNull(id);
       Objects.requireNonNull(name);
+      Objects.requireNonNull(type);
       Objects.requireNonNull(coordinate);
       Objects.requireNonNull(modes);
       Objects.requireNonNull(agencies);

diff --git a/src/ext/java/org/opentripplanner/ext/geocoder/StopClusterMapper.java b/src/ext/java/org/opentripplanner/ext/geocoder/StopClusterMapper.java
@@ -1,5 +1,8 @@
 package org.opentripplanner.ext.geocoder;
 
+import static org.opentripplanner.ext.geocoder.StopCluster.LocationType.STATION;
+import static org.opentripplanner.ext.geocoder.StopCluster.LocationType.STOP;
+
 import com.google.common.collect.Iterables;
 import java.util.Collection;
 import java.util.List;
@@ -129,6 +132,7 @@ StopCluster.Location toLocation(FeedScopedId id) {
       return new StopCluster.Location(
         loc.getId(),
         loc.getCode(),
+        STOP,
         loc.getName().toString(),
         new StopCluster.Coordinate(loc.getLat(), loc.getLon()),
         modes,
@@ -150,6 +154,7 @@ StopCluster.Location toLocation(FeedScopedId id) {
       return new StopCluster.Location(
         group.getId(),
         group.getCode(),
+        STATION,
         group.getName().toString(),
         new StopCluster.Coordinate(group.getLat(), group.getLon()),
         modes,