Merge pull request #90 from OpenVicProject/dataloading-decisions

Decision dataloading (kind of)

2 files changed, 122 insertions, 0 deletions

diff --git a/src/openvic-simulation/misc/Decision.cpp b/src/openvic-simulation/misc/Decision.cpp
new file mode 100644
index 0000000..861a584
--- /dev/null
+++ b/src/openvic-simulation/misc/Decision.cpp
@@ -0,0 +1,75 @@
+#include "Decision.hpp"
+
+using namespace OpenVic;
+using namespace OpenVic::NodeTools;
+
+Decision::Decision(
+	std::string_view new_identifier, bool new_alert, bool new_news,
+	std::string_view new_news_title, std::string_view new_news_desc_long,
+	std::string_view new_news_desc_medium, std::string_view new_news_desc_short,
+	std::string_view new_picture
+) : HasIdentifier { new_identifier }, alert { new_alert }, news { new_news }, news_title { new_news_title },
+	news_desc_long { new_news_desc_long }, news_desc_medium { new_news_desc_medium },
+	news_desc_short { new_news_desc_short }, picture { new_picture } {}
+
+DecisionManager::DecisionManager() : decisions { "decisions" } {}
+
+bool DecisionManager::add_decision(
+	std::string_view identifier, bool alert, bool news, std::string_view news_title,
+	std::string_view news_desc_long, std::string_view news_desc_medium,
+	std::string_view news_desc_short, std::string_view picture
+) {
+	if (identifier.empty()) {
+		Logger::error("Invalid decision identifier - empty!");
+		return false;
+	}
+
+	if (news) {
+		if (news_desc_long.empty() || news_desc_medium.empty() || news_desc_short.empty()) {
+			Logger::warning(
+				"Decision with ID ", identifier, " is a news decision but doesn't have long, medium and short descriptions!"
+			);
+		}
+	} else {
+		if (!news_title.empty() || !news_desc_long.empty() || !news_desc_medium.empty() || !news_desc_short.empty()) {
+			Logger::warning("Decision with ID ", identifier, " is not a news decision but has news strings specified!");
+		}
+	}
+
+	return decisions.add_item({ 
+		identifier,
+		alert,
+		news,
+		news_title,
+		news_desc_long,
+		news_desc_medium,
+		news_desc_short,
+		picture 
+	}, duplicate_warning_callback);
+}
+
+bool DecisionManager::load_decision_file(ast::NodeCPtr root) {
+	return expect_dictionary_keys(
+		"political_decisions", ZERO_OR_ONE, expect_dictionary(
+			[this](std::string_view identifier, ast::NodeCPtr node) -> bool {
+				bool alert = true, news = false;
+				std::string_view news_title, news_desc_long, news_desc_medium, news_desc_short, picture;
+				bool ret = expect_dictionary_keys(
+					"alert", ZERO_OR_ONE, expect_bool(assign_variable_callback(alert)),
+					"news", ZERO_OR_ONE, expect_bool(assign_variable_callback(news)),
+					"news_title", ZERO_OR_ONE, expect_string(assign_variable_callback(news_title)),
+					"news_desc_long", ZERO_OR_ONE, expect_string(assign_variable_callback(news_desc_long)),
+					"news_desc_medium", ZERO_OR_ONE, expect_string(assign_variable_callback(news_desc_medium)),
+					"news_desc_short", ZERO_OR_ONE, expect_string(assign_variable_callback(news_desc_short)),
+					"picture", ZERO_OR_ONE, expect_identifier_or_string(assign_variable_callback(picture)),
+					"potential", ONE_EXACTLY, success_callback, //TODO
+					"allow", ONE_EXACTLY, success_callback, //TODO
+					"effect", ONE_EXACTLY, success_callback, //TODO
+					"ai_will_do", ZERO_OR_ONE, success_callback //TODO
+				)(node);
+				ret &= add_decision(identifier, alert, news, news_title, news_desc_long, news_desc_medium, news_desc_short, picture);
+				return ret;
+			}
+		)
+	)(root);
+}
\ No newline at end of file
diff --git a/src/openvic-simulation/misc/Decision.hpp b/src/openvic-simulation/misc/Decision.hpp
new file mode 100644
index 0000000..e477f64
--- /dev/null
+++ b/src/openvic-simulation/misc/Decision.hpp
@@ -0,0 +1,47 @@
+#pragma once
+
+#include "openvic-simulation/types/IdentifierRegistry.hpp"
+
+namespace OpenVic {
+	struct DecisionManager;
+
+	struct Decision : HasIdentifier {
+		friend struct DecisionManager;
+
+	private:
+		const bool PROPERTY_CUSTOM_PREFIX(alert, has);
+		const bool PROPERTY_CUSTOM_PREFIX(news, is);
+		const std::string PROPERTY(news_title);
+		const std::string PROPERTY(news_desc_long);
+		const std::string PROPERTY(news_desc_medium);
+		const std::string PROPERTY(news_desc_short);
+		const std::string PROPERTY(picture);
+
+		Decision(
+			std::string_view new_identifier, bool new_alert, bool new_news,
+			std::string_view new_news_title, std::string_view new_news_desc_long,
+			std::string_view new_news_desc_medium, std::string_view new_news_desc_short,
+			std::string_view new_picture
+		);
+
+	public:
+		Decision(Decision&&) = default;
+	};
+
+	struct DecisionManager {
+	private:
+		IdentifierRegistry<Decision> decisions;
+
+	public:
+		DecisionManager();
+
+		bool add_decision(
+			std::string_view identifier, bool alert, bool news, std::string_view news_title,
+			std::string_view news_desc_long, std::string_view news_desc_medium,
+			std::string_view news_desc_short, std::string_view picture
+		);
+		IDENTIFIER_REGISTRY_ACCESSORS(decision)
+
+		bool load_decision_file(ast::NodeCPtr root);
+	};
+}