1 files changed, 79 insertions, 0 deletions

diff --git a/src/openvic-simulation/country/CountryInstance.cpp b/src/openvic-simulation/country/CountryInstance.cpp
new file mode 100644
index 0000000..a14f0f0
--- /dev/null
+++ b/src/openvic-simulation/country/CountryInstance.cpp
@@ -0,0 +1,79 @@
+#include "CountryInstance.hpp"
+
+using namespace OpenVic;
+
+bool CountryInstance::add_accepted_culture(Culture const* new_accepted_culture) {
+    if(std::find(accepted_cultures.begin(), accepted_cultures.end(), new_accepted_culture) != accepted_cultures.end()) {
+        Logger::warning("Attempted to add accepted culture ", new_accepted_culture->get_identifier(), " to country ", base_country->get_identifier(), ": already present!");
+        return false;
+    }
+    accepted_cultures.push_back(new_accepted_culture);
+    return true;
+}
+
+bool CountryInstance::remove_accepted_culture(Culture const* culture_to_remove) {
+    auto existing_entry = std::find(accepted_cultures.begin(), accepted_cultures.end(), culture_to_remove);
+    if (existing_entry == accepted_cultures.end()) {
+        Logger::warning("Attempted to remove accepted culture ", culture_to_remove->get_identifier(), " from country ", base_country->get_identifier(), ": not present!");
+        return false;
+    }
+    accepted_cultures.erase(existing_entry);
+    return true;
+}
+
+void CountryInstance::add_to_upper_house(Ideology const* party, fixed_point_t popularity) {
+    upper_house[party] = popularity;
+}
+
+bool CountryInstance::remove_from_upper_house(Ideology const* party) {
+    return upper_house.erase(party) == 1;
+}
+
+bool CountryInstance::add_reform(Reform const* new_reform) {
+    if (std::find(reforms.begin(), reforms.end(), new_reform) != reforms.end()) {
+        Logger::warning("Attempted to add reform ", new_reform->get_identifier(), " to country ", base_country->get_identifier(), ": already present!");
+        return false;
+    }
+    reforms.push_back(new_reform);
+    return true;
+}
+
+bool CountryInstance::remove_reform(Reform const* reform_to_remove) {
+    auto existing_entry = std::find(reforms.begin(), reforms.end(), reform_to_remove);
+    if (existing_entry == reforms.end()) {
+        Logger::warning("Attempted to remove reform ", reform_to_remove->get_identifier(), " from country ", base_country->get_identifier(), ": not present!");
+        return false;
+    }
+    reforms.erase(existing_entry);
+    return true;
+}
+
+void CountryInstance::apply_history_to_country(CountryHistoryMap const& history, Date date) {
+    auto entries = history.get_entries(date);
+
+    accepted_cultures.clear();
+    upper_house.clear();
+    reforms.clear();
+
+    for (CountryHistoryEntry const* entry : entries) {
+        if (entry->get_primary_culture()) primary_culture = *entry->get_primary_culture();
+        for (const auto culture : entry->get_accepted_cultures()) {
+            add_accepted_culture(culture);
+        }
+        if (entry->get_religion()) religion = *entry->get_religion();
+        if (entry->get_ruling_party()) ruling_party = *entry->get_ruling_party();
+        if (entry->get_last_election()) last_election = *entry->get_last_election();
+        for (const auto& party : entry->get_upper_house()) {
+            add_to_upper_house(party.first, party.second);
+        }
+        if (entry->get_capital()) capital = *entry->get_capital();
+        if (entry->get_government_type()) government_type = *entry->get_government_type();
+        if (entry->get_plurality()) plurality = *entry->get_plurality();
+        if (entry->get_national_value()) national_value = *entry->get_national_value();
+        if (entry->get_civilised()) civilised = *entry->get_civilised();
+        if (entry->get_prestige()) prestige = *entry->get_prestige();
+        for (const auto reform : entry->get_reforms()) {
+            add_reform(reform);
+        }
+    }
+}
\ No newline at end of file