Add getter

ibi-group · Dec 4, 2023 · fe9f32e · fe9f32e
1 parent 6910224
commit fe9f32e
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/onebusaway-gtfs/src/main/java/org/onebusaway/gtfs/impl/StopTimeArray.java b/onebusaway-gtfs/src/main/java/org/onebusaway/gtfs/impl/StopTimeArray.java
@@ -263,6 +263,11 @@ public StopLocation getLocation() {
       return stops[index];
     }
 
+    @Override
+    public StopLocation getLocationGroup() {
+      return stops[index];
+    }
+
     @Override
     public void setStop(StopLocation stop) {
       stops[index] = stop;

diff --git a/onebusaway-gtfs/src/main/java/org/onebusaway/gtfs/model/StopTime.java b/onebusaway-gtfs/src/main/java/org/onebusaway/gtfs/model/StopTime.java
@@ -292,6 +292,13 @@ public StopLocation getLocation() {
     return location;
   }
 
+  public StopLocation getLocationGroup() {
+    if (proxy != null) {
+      return proxy.getLocationGroup();
+    }
+    return locationGroup;
+  }
+
   /**
    * Returns possible entity for the stop location in this order:
    *  - stop

diff --git a/onebusaway-gtfs/src/main/java/org/onebusaway/gtfs/model/StopTimeProxy.java b/onebusaway-gtfs/src/main/java/org/onebusaway/gtfs/model/StopTimeProxy.java
@@ -46,6 +46,7 @@ public interface StopTimeProxy {
   public StopLocation getStop();
 
   public StopLocation getLocation();
+  public StopLocation getLocationGroup();
 
   public void setStop(StopLocation stop);