Create combinations

src/hoi4_world/roles/stories_creator.cpp

@@ -1,13 +1,58 @@
 #include "src/hoi4_world/roles/stories_creator.h"
 
+#include <ranges>
+
 #include "external/fmt/include/fmt/format.h"
 #include "src/hoi4_world/roles/roles_importer.h"
 
 
 
-void hoi4::CreateStories()
+namespace
+{
+
+using Tag = std::string;
+using CombinationName = std::string;
+
+
+std::vector<std::pair<Tag, CombinationName>> MakeCombinations(const std::map<std::string, hoi4::Role>& roles,
+    const std::map<std::string, hoi4::Country>& countries)
+{
+   std::vector<std::pair<Tag, CombinationName>> combinations;
+
+   for (const auto& [role_name, role]: roles)
+   {
+      // scan for 'tag=TAG' constructs
+      std::regex tag_match_regex(R"([\s\S]+tag=(.+)[\s\S]+)");
+      std::smatch match;
+      if (!std::regex_match(role.GetRequirements(), match, tag_match_regex))
+      {
+         continue;
+      }
+      std::string required_tag = match[1];
+
+      for (const std::string& country_tag: countries | std::views::keys)
+      {
+         if (required_tag == country_tag)
+         {
+            combinations.emplace_back(country_tag, role_name);
+         }
+      }
+   }
+
+   return combinations;
+}
+
+}  // namespace
+
+
+
+void hoi4::CreateStories(const std::map<std::string, hoi4::Country>& countries)
 {
    Log(LogLevel::Info) << "Writing stories";
-   std::map<std::string, Role> roles = ImportRoles();
+
+   const std::map<std::string, Role> roles = ImportRoles();
    Log(LogLevel::Info) << fmt::format("\tImported {} roles.", roles.size());
+
+   const std::vector<std::pair<Tag, CombinationName>> role_combinations = MakeCombinations(roles, countries);
+   Log(LogLevel::Info) << fmt::format("\tCreated {} role combinations.", role_combinations.size());
 }

src/hoi4_world/roles/stories_creator.h

@@ -2,11 +2,17 @@
 #define SRC_HOI4WORLD_ROLES_STORIESCREATOR_H
 
 
+#include <map>
+#include <string>
+
+#include "src/hoi4_world/countries/hoi4_country.h"
+
+
 
 namespace hoi4
 {
 
-void CreateStories();
+void CreateStories(const std::map<std::string, hoi4::Country>& countries);
 
 }  // namespace hoi4
 

src/hoi4_world/world/hoi4_world_converter.cpp

@@ -320,7 +320,7 @@ hoi4::World hoi4::ConvertWorld(const commonItems::ModFilesystem& hoi4_mod_filesy
    hoi4::Railways railways = railways_future.get();
    hoi4::Buildings buildings = buildings_future.get();
 
-   CreateStories();
+   CreateStories(countries);
 
    return hoi4::World(hoi4::WorldOptions{.countries = countries,
        .great_powers = great_powers,