1 files changed, 14 insertions, 10 deletions

@@ -82,17 +82,21 @@ void SubwayMap::OnConnect() {
 }
 
 void SubwayMap::UpdateIndicators() {
-  Redraw();
-}
-
-void SubwayMap::UpdatePainting(std::string from_painting_id,
-                               std::optional<std::string> to_painting_id) {
-  checked_paintings_.insert(from_painting_id);
-
-  if (to_painting_id) {
-    networks_.AddLink(GD_GetSubwayItemForPainting(from_painting_id),
-                      GD_GetSubwayItemForPainting(*to_painting_id));
+  if (AP_IsPaintingShuffle()) {
+    for (const std::string &painting_id : AP_GetCheckedPaintings()) {
+      if (!checked_paintings_.count(painting_id)) {
+        checked_paintings_.insert(painting_id);
+
+        if (AP_GetPaintingMapping().count(painting_id)) {
+          networks_.AddLink(GD_GetSubwayItemForPainting(painting_id),
+                            GD_GetSubwayItemForPainting(
+                                AP_GetPaintingMapping().at(painting_id)));
+        }
+      }
+    }
   }
+
+  Redraw();
 }
 
 void SubwayMap::UpdateSunwarp(SubwaySunwarp from_sunwarp,