From ec8c12693b6a965f9cd9120bad79e61094816b29 Mon Sep 17 00:00:00 2001
From: LO Kam Tao Leo <leolo@leolo.org>
Date: Sat, 31 Dec 2022 23:27:49 +0000
Subject: [PATCH] Updating more data

---
 src/main/java/org/leolo/nrdatad/model/TrainScheduleLocation.java | 8 ++++++++
 src/main/java/org/leolo/nrdatad/model/TrainScheduleSector.java   | 5 ++++-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/main/java/org/leolo/nrdatad/model/TrainScheduleLocation.java b/src/main/java/org/leolo/nrdatad/model/TrainScheduleLocation.java
index 319907c..a6b86c8 100644
--- a/src/main/java/org/leolo/nrdatad/model/TrainScheduleLocation.java
+++ b/src/main/java/org/leolo/nrdatad/model/TrainScheduleLocation.java
@@ -164,6 +164,14 @@ public class TrainScheduleLocation {
         this.performanceAllowance = performanceAllowance;
     }
 
+    public TrainScheduleSector getRecordSector() {
+        return recordSector;
+    }
+
+    public void setRecordSector(TrainScheduleSector recordSector) {
+        this.recordSector = recordSector;
+    }
+
     public static TrainScheduleLocation parseJSON(String json){
         return parseJSON(new JSONObject(json), UNKNOWN_RECORD_SEQUENCE);
     }
diff --git a/src/main/java/org/leolo/nrdatad/model/TrainScheduleSector.java b/src/main/java/org/leolo/nrdatad/model/TrainScheduleSector.java
index e5c6408..72e9376 100644
--- a/src/main/java/org/leolo/nrdatad/model/TrainScheduleSector.java
+++ b/src/main/java/org/leolo/nrdatad/model/TrainScheduleSector.java
@@ -241,7 +241,10 @@ public class TrainScheduleSector {
         JSONArray locations = object.getJSONArray("schedule_location");
         for(int i=0;i<locations.length();i++){
             JSONObject location = locations.getJSONObject(i);
-            tss.scheduleLocations.add(TrainScheduleLocation.parseJSON(location));
+            TrainScheduleLocation tsl = TrainScheduleLocation.parseJSON(location, i+1);
+            tsl.setRecordSector(tss);
+
+            tss.scheduleLocations.add(tsl);
         }
         return tss;
     }