1 files changed, 113 insertions, 0 deletions
diff --git a/src/openvic-simulation/country/Country.hpp b/src/openvic-simulation/country/Country.hpp
new file mode 100644
index 0000000..ce7638f
--- /dev/null
+++ b/src/openvic-simulation/country/Country.hpp
@@ -0,0 +1,113 @@
+#pragma once
+
+#include <filesystem>
+#include <memory>
+#include <string>
+#include <string_view>
+#include <type_traits>
+#include <unordered_map>
+#include <vector>
+
+#include <openvic-dataloader/v2script/AbstractSyntaxTree.hpp>
+
+#include "openvic-simulation/politics/Government.hpp"
+#include "openvic-simulation/politics/Ideology.hpp"
+#include "openvic-simulation/politics/Issue.hpp"
+#include "openvic-simulation/pop/Culture.hpp"
+#include "openvic-simulation/types/Colour.hpp"
+#include "openvic-simulation/types/Date.hpp"
+#include "openvic-simulation/types/IdentifierRegistry.hpp"
+
+namespace OpenVic {
+	struct GameManager;
+	struct CountryManager;
+
+	struct CountryParty {
+		friend struct CountryManager;
+
+	private:
+		const std::string name;
+		const Date start_date;
+		const Date end_date;
+		const Ideology& ideology;
+		const std::vector<const Issue*> policies;
+
+		CountryParty(
+			std::string_view new_name,
+			Date new_start_date,
+			Date new_end_date,
+			const Ideology& new_ideology,
+			std::vector<const Issue*>&& new_policies
+		);
+
+	public:
+		std::string_view get_name() const;
+		const Date& get_start_date() const;
+		const Date& get_end_date() const;
+		const Ideology& get_ideology() const;
+		const std::vector<const Issue*>& get_policies() const;
+	};
+
+	struct UnitNames {
+		friend struct CountryManager;
+
+	private:
+		const std::string identifier;
+		const std::vector<std::string> names;
+
+		UnitNames(std::string_view new_identifier, std::vector<std::string>&& new_names);
+
+	public:
+		std::string_view get_identifier() const;
+		const std::vector<std::string>& get_names() const;
+	};
+
+	struct Country : HasIdentifierAndColour {
+		friend struct CountryManager;
+
+	private:
+		const GraphicalCultureType& graphical_culture;
+		const std::vector<CountryParty> parties;
+		const std::vector<UnitNames> unit_names;
+		bool dynamic_tag;
+		const std::map<const GovernmentType*, colour_t> alternative_colours;
+
+		Country(
+			std::string_view new_identifier,
+			colour_t new_color,
+			const GraphicalCultureType& new_graphical_culture,
+			std::vector<CountryParty>&& new_parties,
+			std::vector<UnitNames>&& new_unit_names,
+			bool new_dynamic_tag,
+			std::map<const GovernmentType*, colour_t>&& new_alternative_colours
+		);
+
+	public:
+		const GraphicalCultureType& get_graphical_culture() const;
+		const std::vector<CountryParty>& get_parties() const;
+		const std::vector<UnitNames>& get_unit_names() const;
+		bool is_dynamic_tag() const;
+		const std::map<const GovernmentType*, colour_t>& get_alternative_colours() const;
+	};
+
+	struct CountryManager {
+	private:
+		IdentifierRegistry<Country> countries;
+
+	public:
+		CountryManager();
+
+		bool add_country(
+			std::string_view identifier,
+			colour_t color,
+			const GraphicalCultureType& graphical_culture,
+			std::vector<CountryParty>&& parties,
+			std::vector<UnitNames>&& unit_names,
+			bool dynamic_tag,
+			std::map<const GovernmentType*, colour_t>&& new_alternative_colours
+		);
+		IDENTIFIER_REGISTRY_ACCESSORS_CUSTOM_PLURAL(country, countries);
+
+		bool load_country_data_file(GameManager& game_manager, std::string_view name, bool is_dynamic, ast::NodeCPtr root);
+	};
+}
+\ No newline at end of file