fix minsk bug

Daoortor · Dec 9, 2024 · 5cbf874 · 5cbf874
1 parent dd89ccd
commit 5cbf874
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/models/include/transportSystem.h b/models/include/transportSystem.h
@@ -23,6 +23,8 @@ struct TransportSystem {
     TransportSystem() = default;
     // Construct a transport system from JSON data
     explicit TransportSystem(const QDir &sourceDir);
+    void initStopById();
+    void initStopByName();
     void initStopNamesAutomaton();
     // Check if all pointers point inside corresponding containers
     [[nodiscard]] bool isValid() const;

diff --git a/models/src/transportSystem.cpp b/models/src/transportSystem.cpp
@@ -154,6 +154,18 @@ TransportSystem::TransportSystem(const QDir &sourceDir) {
     cout << "Transport system loaded" << endl;
 }
 
+void TransportSystem::initStopById() {
+    for (auto &stop : stops) {
+        stopById[stop.id] = &stop;
+    }
+}
+
+void TransportSystem::initStopByName() {
+    for (auto &stop : stops) {
+        stopByName[stop.name] = &stop;
+    }
+}
+
 void TransportSystem::initStopNamesAutomaton() {
     std::vector<QString> stationNames;
     stationNames.reserve(stops.size());
@@ -163,7 +175,6 @@ void TransportSystem::initStopNamesAutomaton() {
     stopNamesAutomaton = SuffixAutomaton(stationNames);
 }
 
-
 bool TransportSystem::isValid() const {
     bool routesAreValid = ranges::all_of(routes.begin(), routes.end(), [&](const Route &route) {
         return route.stopTimes >= &*stopTimes.begin() && route.stopTimes < &*stopTimes.end() &&

diff --git a/test/test-data/minsk-subway.cpp b/test/test-data/minsk-subway.cpp
@@ -95,6 +95,9 @@ TransportSystem makeMinskSubway() {
         }
         start += route.tripCount * route.stopCount;
     }
+    subway.initStopById();
+    subway.initStopByName();
+    subway.initStopNamesAutomaton();
 
     return subway;
 }