1 files changed, 142 insertions, 0 deletions
diff --git a/src/openvic-simulation/vm/Codegen.hpp b/src/openvic-simulation/vm/Codegen.hpp
new file mode 100644
index 0000000..1670718
--- /dev/null
+++ b/src/openvic-simulation/vm/Codegen.hpp
@@ -0,0 +1,142 @@
+#pragma once
+
+#include <openvic-dataloader/v2script/AbstractSyntaxTree.hpp>
+#include <openvic-dataloader/v2script/Parser.hpp>
+
+#include "AsmBuilder.hpp"
+#include "Module.hpp"
+#include <lauf/asm/builder.h>
+
+namespace OpenVic::Vm {
+	struct Codegen {
+		Codegen(const char* module_name, lauf_asm_build_options options = lauf_asm_default_build_options)
+			: _module(module_name), _builder(options) {}
+
+		Codegen(Module&& module, AsmBuilder&& builder) : _module(std::move(module)), _builder(std::move(builder)) {}
+
+		operator lauf_asm_module*() {
+			return _module;
+		}
+
+		operator const lauf_asm_module*() const {
+			return _module;
+		}
+
+		operator lauf_asm_builder*() {
+			return _builder;
+		}
+
+		operator const lauf_asm_builder*() const {
+			return _builder;
+		}
+
+		Module& module() {
+			return _module;
+		}
+
+		Module const& module() const {
+			return _module;
+		}
+
+		AsmBuilder& builder() {
+			return _builder;
+		}
+
+		AsmBuilder const& builder() const {
+			return _builder;
+		}
+
+		lauf_asm_block* create_block(size_t input_count) {
+			return lauf_asm_declare_block(_builder, input_count);
+		}
+
+		void set_working_block(lauf_asm_block* block) {
+			lauf_asm_build_block(*this, block);
+		}
+
+		lauf_asm_function* create_effect_function(ovdl::v2script::Parser const& parser, ovdl::v2script::ast::Node* node);
+		lauf_asm_function* create_condition_function(ovdl::v2script::Parser const& parser, ovdl::v2script::ast::Node* node);
+
+		void generate_effect_from(ovdl::v2script::Parser const& parser, ovdl::v2script::ast::Node* node);
+		void generate_condition_from(ovdl::v2script::Parser const& parser, ovdl::v2script::ast::Node* node);
+
+		// Bytecode instructions //
+		void inst_push_scope_this();
+		void inst_push_scope_from();
+
+		void inst_push_get_country_capital();
+		void inst_push_get_country_cultural_union();
+		void inst_push_get_country_overlord();
+		void inst_push_get_country_sphere_owner();
+
+		void inst_push_get_province_controller();
+		void inst_push_get_province_owner();
+		void inst_push_get_province_state();
+
+		void inst_push_get_pop_location();
+		void inst_push_get_pop_country();
+		void inst_push_get_pop_cultural_union();
+
+		void inst_push_get_random_country();
+		void inst_push_get_random_owned();
+
+		void inst_push_get_random_neighbor_province();
+		void inst_push_get_random_empty_neighbor_province();
+
+		void inst_push_get_this_union();
+
+		void inst_push_get_tag(const char* nation_id);
+		void inst_push_get_province(const char* province_id);
+		void inst_push_get_region(const char* region_id);
+		void inst_push_get_continent(const char* continent_id);
+
+		void inst_push_get_pop_type(const char* pop_type_id);
+		void inst_push_get_unit_type(const char* unit_type_id);
+		void inst_push_get_factory_type(const char* factory_type_id);
+		void inst_push_get_building_type(const char* building_type_id);
+		void inst_push_get_crime_type(const char* crime_type_id);
+
+		void inst_push_get_culture(const char* culture_id);
+		void inst_push_get_religion(const char* religion_id);
+
+		void inst_push_get_economic_reform(const char* economic_reform_id);
+		void inst_push_get_social_reform(const char* social_reform_id);
+		void inst_push_get_political_reform(const char* political_reform_id);
+
+		void inst_push_get_trade_policy(const char* trade_policy_id);
+		void inst_push_get_war_policy(const char* war_policy_id);
+
+		void inst_push_get_issue(const char* issue_id);
+		void inst_push_get_ideology(const char* ideology_id);
+		void inst_push_get_party(const char* party_id);
+
+		void inst_push_get_military_leader(const char* military_leader_id);
+		void inst_push_get_personality_trait(const char* personality_trait_id);
+		void inst_push_get_background_trait(const char* background_trait_id);
+
+		void inst_push_get_event(const char* event_id);
+
+		void inst_push_get_trade_good(const char* trade_good_id);
+		void inst_push_get_resource(const char* resource_id);
+
+		void inst_push_get_technology(const char* technology_id);
+		void inst_push_get_government(const char* government_id);
+		void inst_push_get_casus_belli(const char* casus_belli_id);
+		void inst_push_get_national_value(const char* national_value_id);
+
+		void inst_push_get_variable(const char* variable_id);
+
+		void inst_push_get_global_flag(const char* flag_id);
+
+		void inst_push_get_country_flag(const char* flag_id);
+		void inst_push_get_country_modifier(const char* modifier_id);
+
+		void inst_push_get_province_flag(const char* flag_id);
+		void inst_push_get_province_modifier(const char* modifier_id);
+		// Bytecode instructions //
+
+	private:
+		Module _module;
+		AsmBuilder _builder;
+	};
+}