Change TrainWrapper to provide Siding

zbx1425 · Dec 31, 2023 · 495d496 · 495d496
1 parent 1ae13ba
commit 495d496
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/common/src/main/java/cn/zbx1425/mtrsteamloco/render/scripting/train/TrainWrapper.java b/common/src/main/java/cn/zbx1425/mtrsteamloco/render/scripting/train/TrainWrapper.java
@@ -56,6 +56,7 @@ private PlatformLookupMap getTrainPlatforms() {
         List<Long> routeIds = train.getRouteIds();
         DataCache dataCache = ClientData.DATA_CACHE;
         PlatformLookupMap result = new PlatformLookupMap();
+        result.siding = dataCache.sidingIdMap.get(train.sidingId);
         if (routeIds.isEmpty()) return result;
 
         int routeIndex = 0;
@@ -169,6 +170,7 @@ public List<PlatformInfo> getDebugThisRoutePlatforms(int count) {
     }
 
     private static class PlatformLookupMap {
+        public Siding siding;
         public final List<PlatformInfo> platforms = new ArrayList<>();
         public final TreeMap<Integer, Integer> pathToPlatformIndex = new TreeMap<>();
         public final TreeMap<Integer, List<PlatformInfo>> pathToRoutePlatforms = new TreeMap<>();
@@ -208,7 +210,7 @@ public boolean shouldRenderDetail() {
     }
 
     @SuppressWarnings("unused") public long id() { return train.id; }
-    @SuppressWarnings("unused") public long sidingId() { return train.sidingId; }
+    @SuppressWarnings("unused") public Siding siding() { return trainPlatforms.siding; }
     @SuppressWarnings("unused") public String trainTypeId() { return train.trainId; }
     @SuppressWarnings("unused") public String baseTrainType() { return train.baseTrainType; }
     @SuppressWarnings("unused") public TransportMode transportMode() { return train.transportMode; }