From d67cd2d08fe2859809e35b93f5a8358c48a3705e Mon Sep 17 00:00:00 2001
From: hop311 <hop3114@gmail.com>
Date: Fri, 6 Sep 2024 23:13:47 +0100
Subject: Calculate country military power (still needs modifiers + soldier pop
 supported regiment count)

---
 src/openvic-simulation/military/UnitType.cpp | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

(limited to 'src/openvic-simulation/military/UnitType.cpp')

diff --git a/src/openvic-simulation/military/UnitType.cpp b/src/openvic-simulation/military/UnitType.cpp
index 11f7221..d4ffbee 100644
--- a/src/openvic-simulation/military/UnitType.cpp
+++ b/src/openvic-simulation/military/UnitType.cpp
@@ -1,5 +1,6 @@
 #include "UnitType.hpp"
 
+#include "openvic-simulation/country/CountryInstance.hpp"
 #include "openvic-simulation/map/TerrainType.hpp"
 
 using namespace OpenVic;
@@ -30,6 +31,29 @@ UnitType::UnitType(
 	supply_cost { std::move(unit_args.supply_cost) },
 	terrain_modifiers { std::move(unit_args.terrain_modifiers) } {}
 
+bool UnitTypeBranched<LAND>::allowed_cultures_check_culture_in_country(
+	allowed_cultures_t allowed_cultures, Culture const& culture, CountryInstance const& country
+) {
+	using enum allowed_cultures_t;
+
+	switch (allowed_cultures) {
+		case ALL_CULTURES:
+			return true;
+		case ACCEPTED_CULTURES:
+			return country.is_primary_or_accepted_culture(culture);
+		case PRIMARY_CULTURE:
+			return country.is_primary_culture(culture);
+		case NO_CULTURES:
+			return false;
+		default:
+			Logger::error(
+				"Unknown allowed cultures value ", static_cast<uint32_t>(allowed_cultures), " for culture ",
+				culture.get_identifier(), " and country ", country.get_identifier()
+			);
+			return false;
+	}
+}
+
 UnitTypeBranched<LAND>::UnitTypeBranched(
 	std::string_view new_identifier, unit_type_args_t& unit_args, regiment_type_args_t const& regiment_type_args
 ) : UnitType { new_identifier, LAND, unit_args },
-- 
cgit v1.2.3-56-ga3b1