1 files changed, 53 insertions, 0 deletions
diff --git a/extension/src/openvic-extension/classes/GUINode.hpp b/extension/src/openvic-extension/classes/GUINode.hpp
new file mode 100644
index 0000000..1671547
--- /dev/null
+++ b/extension/src/openvic-extension/classes/GUINode.hpp
@@ -0,0 +1,53 @@
+#pragma once
+
+#include <godot_cpp/classes/button.hpp>
+#include <godot_cpp/classes/check_box.hpp>
+#include <godot_cpp/classes/label.hpp>
+#include <godot_cpp/classes/panel.hpp>
+#include <godot_cpp/classes/texture_progress_bar.hpp>
+#include <godot_cpp/classes/texture_rect.hpp>
+
+#include <openvic-simulation/interface/GUI.hpp>
+
+#include "openvic-extension/classes/GFXIconTexture.hpp"
+#include "openvic-extension/classes/GFXMaskedFlagTexture.hpp"
+#include "openvic-extension/classes/GFXPieChartTexture.hpp"
+
+namespace OpenVic {
+	class GUINode : public godot::Control {
+		GDCLASS(GUINode, godot::Control)
+
+		template<std::derived_from<godot::Node> T>
+		T* _get_cast_node(godot::NodePath const& path) const;
+
+		template<std::derived_from<godot::Texture2D> T>
+		godot::Ref<T> _get_cast_texture_from_node(godot::NodePath const& path) const;
+
+	protected:
+		static void _bind_methods();
+
+	public:
+		GUINode() = default;
+
+		godot::Error _add_gui_element(GUI::Element const* element, godot::String const& name);
+		godot::Error add_gui_element(
+			godot::String const& gui_file, godot::String const& gui_element, godot::String const& name = ""
+		);
+
+		godot::Button* get_button_node(godot::NodePath const& path) const;
+		godot::CheckBox* get_check_box_node(godot::NodePath const& path) const;
+		godot::Label* get_label_node(godot::NodePath const& path) const;
+		godot::Panel* get_panel_node(godot::NodePath const& path) const;
+		godot::TextureProgressBar* get_progress_bar_node(godot::NodePath const& path) const;
+		godot::TextureRect* get_texture_rect_node(godot::NodePath const& path) const;
+
+		/* Helper functions to get textures from TextureRects and Buttons. */
+		godot::Ref<godot::Texture2D> get_texture_from_node(godot::NodePath const& path) const;
+		godot::Ref<GFXIconTexture> get_gfx_icon_texture_from_node(godot::NodePath const& path) const;
+		godot::Ref<GFXMaskedFlagTexture> get_gfx_masked_flag_texture_from_node(godot::NodePath const& path) const;
+		godot::Ref<GFXPieChartTexture> get_gfx_pie_chart_texture_from_node(godot::NodePath const& path) const;
+
+		godot::Error hide_node(godot::NodePath const& path) const;
+		godot::Error hide_nodes(godot::Array const& paths) const;
+	};
+}