1 files changed, 23 insertions, 1 deletions

@@ -103,6 +103,7 @@ struct APState {
       }
 
       tracked_data_storage_keys.push_back("PlayerPos");
+      tracked_data_storage_keys.push_back("Paintings");
 
       initialized = true;
     }
@@ -384,6 +385,14 @@ struct APState {
       }
 
       wxLogMessage("Data storage %s retrieved as null", key);
+    } else if (value.is_array()) {
+      if (key.ends_with("Paintings")) {
+        data_storage[key] = value.get<std::set<std::string>>();
+      } else {
+        data_storage[key] = value.get<std::vector<std::string>>();
+      }
+
+      wxLogMessage("Data storage %s retrieved as list", key);
     }
   }
 
@@ -406,6 +415,15 @@ struct APState {
     return data_storage.count(key) && std::any_cast<bool>(data_storage.at(key));
   }
 
+  const std::set<std::string>& GetCheckedPaintings() {
+    std::string key = data_storage_prefix + "Paintings";
+    if (!data_storage.count(key)) {
+      data_storage[key] = std::set<std::string>();
+    }
+
+    return std::any_cast<const std::set<std::string>&>(data_storage.at(key));
+  }
+
   void RefreshTracker(bool reset) {
     wxLogMessage("Refreshing display...");
 
@@ -471,10 +489,14 @@ bool AP_IsColorShuffle() { return GetState().color_shuffle; }
 
 bool AP_IsPaintingShuffle() { return GetState().painting_shuffle; }
 
-const std::map<std::string, std::string> AP_GetPaintingMapping() {
+const std::map<std::string, std::string>& AP_GetPaintingMapping() {
   return GetState().painting_mapping;
 }
 
+const std::set<std::string>& AP_GetCheckedPaintings() {
+  return GetState().GetCheckedPaintings();
+}
+
 int AP_GetMasteryRequirement() { return GetState().mastery_requirement; }
 
 int AP_GetLevel2Requirement() { return GetState().level_2_requirement; }