5 files changed, 56 insertions, 24 deletions

@@ -241,8 +241,11 @@ struct APState {
       pilgrimage_allows_paintings =
           slot_data.contains("pilgrimage_allows_paintings") &&
           slot_data["pilgrimage_allows_paintings"].get<int>() == 1;
-      sunwarp_access = slot_data["sunwarp_access"].get<SunwarpAccess>();
-      sunwarp_shuffle = slot_data["shuffle_sunwarps"].get<int>() == 1;
+      sunwarp_access = slot_data.contains("sunwarp_access")
+                           ? slot_data["sunwarp_access"].get<SunwarpAccess>()
+                           : kSUNWARP_ACCESS_NORMAL;
+      sunwarp_shuffle = slot_data.contains("shuffle_sunwarps") &&
+                        slot_data["shuffle_sunwarps"].get<int>() == 1;
 
       if (painting_shuffle && slot_data.contains("painting_entrance_to_exit")) {
         painting_mapping.clear();
@@ -380,7 +383,7 @@ struct APState {
       } else {
         data_storage.erase(key);
       }
-      
+
       TrackerLog("Data storage " + key + " retrieved as null");
     }
   }
@@ -254,6 +254,11 @@ struct GameData {
             achievement_panels_.push_back(panel_id);
           }
 
+          if (panel_it.second["location_name"]) {
+            panels_[panel_id].location_name =
+                panel_it.second["location_name"].as<std::string>();
+          }
+
           if (panel_it.second["hunt"]) {
             panels_[panel_id].hunt = panel_it.second["hunt"].as<bool>();
           }
@@ -521,8 +526,21 @@ struct GameData {
       std::string room_name = rooms_[room_id].name;
 
       std::string area_name = room_name;
-      if (fold_areas.count(room_name)) {
-        int fold_area_id = fold_areas[room_name];
+      std::string section_name = panel.name;
+      std::string location_name = room_name + " - " + panel.name;
+
+      if (!panel.location_name.empty()) {
+        location_name = panel.location_name;
+
+        size_t divider_pos = location_name.find(" - ");
+        if (divider_pos != std::string::npos) {
+          area_name = location_name.substr(0, divider_pos);
+          section_name = location_name.substr(divider_pos + 3);
+        }
+      }
+
+      if (fold_areas.count(area_name)) {
+        int fold_area_id = fold_areas[area_name];
         area_name = map_areas_[fold_area_id].name;
       }
 
@@ -542,15 +560,15 @@ struct GameData {
       MapArea &map_area = map_areas_[area_id];
       // room field should be the original room ID
       map_area.locations.push_back(
-          {.name = panel.name,
-           .ap_location_name = room_name + " - " + panel.name,
+          {.name = section_name,
+           .ap_location_name = location_name,
            .ap_location_id = panel.ap_location_id,
            .room = panel.room,
            .panels = {panel.id},
            .classification = classification,
            .hunt = panel.hunt});
-      locations_by_name[map_area.locations.back().ap_location_name] = {
-          area_id, map_area.locations.size() - 1};
+      locations_by_name[location_name] = {area_id,
+                                          map_area.locations.size() - 1};
     }
 
     for (int door_id : door_definition_order_) {
@@ -51,6 +51,7 @@ struct Panel {
   bool exclude_reduce = false;
   bool achievement = false;
   std::string achievement_name;
+  std::string location_name;
   bool non_counting = false;
   int ap_location_id = -1;
   bool hunt = false;
@@ -27,7 +27,7 @@ TrackerState& GetState() {
 class StateCalculator;
 
 struct StateCalculatorOptions {
-  std::string start = "Menu";
+  int start;
   bool pilgrimage = false;
   StateCalculator* parent = nullptr;
 };
@@ -42,8 +42,7 @@ class StateCalculator {
   void Calculate() {
     std::list<int> panel_boundary;
     std::list<Exit> flood_boundary;
-    flood_boundary.push_back(
-        {.destination_room = GD_GetRoomByName(options_.start)});
+    flood_boundary.push_back({.destination_room = options_.start});
 
     bool reachable_changed = true;
     while (reachable_changed) {
@@ -328,21 +327,32 @@ class StateCalculator {
         }
       }
 
-      static const std::vector<std::tuple<std::string, std::string>>
-          pilgrimage_pairs = {
-              {"Crossroads", "Hot Crusts Area"},
-              // {"Orange Tower Third Floor", "Orange Tower Third Floor"},
-              {"Outside The Initiated", "Orange Tower First Floor"},
-              {"Outside The Undeterred", "Orange Tower Fourth Floor"},
-              {"Color Hunt", "Outside The Agreeable"}};
+      std::vector<std::tuple<int, int>> pilgrimage_pairs;
+      if (AP_IsSunwarpShuffle()) {
+        pilgrimage_pairs = std::vector<std::tuple<int, int>>(5);
 
-      for (const auto& [from_room, to_room] : pilgrimage_pairs) {
+        for (const auto& [start_index, mapping] : AP_GetSunwarpMapping()) {
+          if (mapping.dots > 1) {
+            std::get<1>(pilgrimage_pairs[mapping.dots - 2]) = start_index;
+          }
+          if (mapping.dots < 6) {
+            std::get<0>(pilgrimage_pairs[mapping.dots - 1]) =
+                mapping.exit_index;
+          }
+        }
+      } else {
+        pilgrimage_pairs = {{6, 1}, {8, 3}, {9, 4}, {10, 5}};
+      }
+
+      for (const auto& [from_sunwarp, to_sunwarp] : pilgrimage_pairs) {
         StateCalculator pilgrimage_calculator(
-            {.start = from_room, .pilgrimage = true, .parent = this});
+            {.start = GD_GetRoomForSunwarp(from_sunwarp),
+             .pilgrimage = true,
+             .parent = this});
         pilgrimage_calculator.Calculate();
 
         if (!pilgrimage_calculator.GetReachableRooms().count(
-                GD_GetRoomByName(to_room))) {
+                GD_GetRoomForSunwarp(to_sunwarp))) {
           return kMaybe;
         }
       }
@@ -390,7 +400,7 @@ class StateCalculator {
 }  // namespace
 
 void RecalculateReachability() {
-  StateCalculator state_calculator;
+  StateCalculator state_calculator({.start = GD_GetRoomByName("Menu")});
   state_calculator.Calculate();
 
   const std::set<int>& reachable_rooms = state_calculator.GetReachableRooms();
@@ -35,6 +35,6 @@ std::ostream& operator<<(std::ostream& out, const Version& ver) {
   return out << "v" << ver.major << "." << ver.minor << "." << ver.revision;
 }
 
-constexpr const Version kTrackerVersion = Version(0, 8, 0);
+constexpr const Version kTrackerVersion = Version(0, 9, 1);
 
 #endif /* end of include guard: VERSION_H_C757E53C */
\ No newline at end of file