1 files changed, 64 insertions, 0 deletions
diff --git a/src/openvic-simulation/scripts/ConditionalWeight.cpp b/src/openvic-simulation/scripts/ConditionalWeight.cpp
new file mode 100644
index 0000000..29dc93b
--- /dev/null
+++ b/src/openvic-simulation/scripts/ConditionalWeight.cpp
@@ -0,0 +1,64 @@
+#include "ConditionalWeight.hpp"
+
+using namespace OpenVic;
+using namespace OpenVic::NodeTools;
+
+template<typename T>
+static NodeCallback auto expect_modifier(std::vector<T>& items) {
+	return [&items](ast::NodeCPtr node) -> bool {
+		fixed_point_t weight = 0;
+		bool successful = false;
+		bool ret = expect_key("factor", expect_fixed_point(assign_variable_callback(weight)), &successful)(node);
+		if (!successful) {
+			Logger::info("ConditionalWeight modifier missing factor key!");
+			return false;
+		}
+		ConditionScript condition;
+		ret &= condition.expect_script()(node);
+		items.emplace_back(std::make_pair(weight, std::move(condition)));
+		return ret;
+	};
+}
+
+node_callback_t ConditionalWeight::expect_conditional_weight(base_key_t base_key) {
+	return expect_dictionary_keys(
+		// TODO - add days and years as options with a shared expected count of ONE_EXACTLY
+		base_key_to_string(base_key), ONE_EXACTLY, expect_fixed_point(assign_variable_callback(base)),
+		"modifier", ZERO_OR_MORE, expect_modifier(condition_weight_items),
+		"group", ZERO_OR_MORE, [this](ast::NodeCPtr node) -> bool {
+			condition_weight_group_t items;
+			const bool ret = expect_dictionary_keys(
+				"modifier", ONE_OR_MORE, expect_modifier(items)
+			)(node);
+			if (!items.empty()) {
+				condition_weight_items.emplace_back(std::move(items));
+				return ret;
+			}
+			Logger::error("ConditionalWeight group must have at least one modifier!");
+			return false;
+		}
+	);
+}
+
+struct ConditionalWeight::parse_scripts_visitor_t {
+	GameManager const& game_manager;
+
+	bool operator()(condition_weight_t& condition_weight) const {
+		return condition_weight.second.parse_script(false, game_manager);
+	}
+	bool operator()(condition_weight_item_t& item) const {
+		return std::visit(*this, item);
+	}
+	template<typename T>
+	bool operator()(std::vector<T>& items) const {
+		bool ret = true;
+		for (T& item : items) {
+			ret &= (*this)(item);
+		}
+		return ret;
+	}
+};
+
+bool ConditionalWeight::parse_scripts(GameManager const& game_manager) {
+	return parse_scripts_visitor_t { game_manager }(condition_weight_items);
+}