#pragma once

#include <variant>
#include <vector>

#include "openvic-simulation/dataloader/NodeTools.hpp"
#include "openvic-simulation/scripts/ConditionScript.hpp"
#include "openvic-simulation/types/fixed_point/FixedPoint.hpp"

namespace OpenVic {
	struct ConditionalWeight {
		using condition_weight_t = std::pair<fixed_point_t, ConditionScript>;
		using condition_weight_group_t = std::vector<condition_weight_t>;
		using condition_weight_item_t = std::variant<condition_weight_t, condition_weight_group_t>;

		enum class base_key_t : uint8_t {
			BASE, FACTOR, TIME
		};
		using enum base_key_t;

	private:
		fixed_point_t PROPERTY(base);
		std::vector<condition_weight_item_t> PROPERTY(condition_weight_items);
		scope_type_t PROPERTY(initial_scope);
		scope_type_t PROPERTY(this_scope);
		scope_type_t PROPERTY(from_scope);

		struct parse_scripts_visitor_t;

	public:
		ConditionalWeight(scope_type_t new_initial_scope, scope_type_t new_this_scope, scope_type_t new_from_scope);
		ConditionalWeight(ConditionalWeight&&) = default;

		NodeTools::node_callback_t expect_conditional_weight(base_key_t base_key);

		bool parse_scripts(DefinitionManager const& definition_manager);

		fixed_point_t execute(
			InstanceManager const& instance_manager,
			ConditionNode::scope_t const& initial_scope,
			ConditionNode::scope_t const& this_scope,
			ConditionNode::scope_t const& from_scope
		) const;
	};
}