From 81e00e3291834dbae6e70224651bf66198ba154e Mon Sep 17 00:00:00 2001
From: hop311 <hop3114@gmail.com>
Date: Fri, 13 Sep 2024 00:11:33 +0100
Subject: Calculate country and province modifier sums (WIP)

---
 src/openvic-simulation/country/CountryInstance.cpp | 78 ++++++++++++++++++++++
 1 file changed, 78 insertions(+)

(limited to 'src/openvic-simulation/country/CountryInstance.cpp')

diff --git a/src/openvic-simulation/country/CountryInstance.cpp b/src/openvic-simulation/country/CountryInstance.cpp
index 183b0c8..b777cf7 100644
--- a/src/openvic-simulation/country/CountryInstance.cpp
+++ b/src/openvic-simulation/country/CountryInstance.cpp
@@ -41,6 +41,10 @@ CountryInstance::CountryInstance(
 	controlled_provinces {},
 	core_provinces {},
 	states {},
+	base_modifier_sum {},
+	event_modifiers {},
+	static_modifiers {},
+	triggered_modifiers {},
 
 	/* Production */
 	industrial_power { 0 },
@@ -923,6 +927,74 @@ bool CountryInstance::update_rule_set() {
 	return rule_set.trim_and_resolve_conflicts(true);
 }
 
+void CountryInstance::update_base_modifier_sum(Date today) {
+	std::erase_if(event_modifiers, [today](ModifierInstance const& modifier) -> bool {
+		return modifier.get_expiry_date() < today;
+	});
+
+	// TODO - add/remove static modifiers
+	// TODO - add/remove triggered modifiers
+
+	// Update sum of national modifiers
+	base_modifier_sum.clear();
+
+	for (ModifierInstance const& event_modifier : event_modifiers) {
+		base_modifier_sum += *event_modifier.get_modifier();
+	}
+
+	for (Modifier const* static_modifier : static_modifiers) {
+		base_modifier_sum += *static_modifier;
+	}
+
+	for (TriggeredModifier const* triggered_modifier : triggered_modifiers) {
+		base_modifier_sum += *triggered_modifier;
+	}
+
+	if (ruling_party != nullptr) {
+		for (Issue const* issue : ruling_party->get_policies()) {
+			if (issue != nullptr) {
+				base_modifier_sum += *issue;
+			}
+		}
+	}
+
+	for (Reform const* reform : reforms) {
+		if (reform != nullptr) {
+			base_modifier_sum += *reform;
+		}
+	}
+
+	if (national_value != nullptr) {
+		base_modifier_sum += *national_value;
+	}
+
+	if (tech_school != nullptr) {
+		base_modifier_sum += *tech_school;
+	}
+
+	for (Technology const& technology : *unlocked_technologies.get_keys()) {
+		if (is_technology_unlocked(technology)) {
+			base_modifier_sum += technology;
+		}
+	}
+
+	for (Invention const& invention : *unlocked_inventions.get_keys()) {
+		if (is_invention_unlocked(invention)) {
+			base_modifier_sum += invention;
+		}
+	}
+
+	// Update sum of province base modifiers
+	owned_provinces_modifier_sum.clear();
+
+	for (ProvinceInstance const* province : owned_provinces) {
+		// TODO - filter out local province modifiers before adding them to this sum
+		owned_provinces_modifier_sum += province->get_base_modifier_sum();
+	}
+
+	// TODO - calculate stats for each unit type (locked and unlocked)
+}
+
 void CountryInstance::update_gamestate(DefineManager const& define_manager, UnitTypeManager const& unit_type_manager) {
 	// Order of updates might need to be changed/functions split up to account for dependencies
 	_update_production(define_manager);
@@ -1167,6 +1239,12 @@ bool CountryInstanceManager::apply_history_to_countries(
 	return ret;
 }
 
+void CountryInstanceManager::update_base_modifier_sums(Date today) {
+	for (CountryInstance& country : country_instances.get_items()) {
+		country.update_base_modifier_sum(today);
+	}
+}
+
 void CountryInstanceManager::update_gamestate(
 	Date today, DefineManager const& define_manager, UnitTypeManager const& unit_type_manager
 ) {
-- 
cgit v1.2.3-56-ga3b1