expose BeehaveTree on context

bitbrain · Jan 13, 2024 · 482fab8 · 482fab8
1 parent 4ad288e
commit 482fab8
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/addons/beehave/libs/windows/beehave.windows.editor.x86_64.dll b/addons/beehave/libs/windows/beehave.windows.editor.x86_64.dll
diff --git a/extension/src/beehave_context.cpp b/extension/src/beehave_context.cpp
@@ -33,6 +33,8 @@ using namespace godot;
 
 void BeehaveContext::_bind_methods()
 {
+    ClassDB::bind_method(D_METHOD("set_tree", "tree"), &BeehaveContext::set_tree);
+    ClassDB::bind_method(D_METHOD("get_tree"), &BeehaveContext::get_tree);
     ClassDB::bind_method(D_METHOD("set_blackboard", "blackboard"), &BeehaveContext::set_blackboard);
     ClassDB::bind_method(D_METHOD("get_blackboard"), &BeehaveContext::get_blackboard);
     ClassDB::bind_method(D_METHOD("set_actor", "actor"), &BeehaveContext::set_actor);
@@ -41,6 +43,16 @@ void BeehaveContext::_bind_methods()
     ClassDB::bind_method(D_METHOD("get_delta"), &BeehaveContext::get_delta);
 }
 
+BeehaveTree* BeehaveContext::get_tree() const
+{
+    return this->tree;
+}
+
+void BeehaveContext::set_tree(BeehaveTree* tree)
+{
+    this->tree = tree;
+}
+
 BeehaveBlackboard* BeehaveContext::get_blackboard() const
 {
     return this->blackboard;

diff --git a/extension/src/beehave_context.h b/extension/src/beehave_context.h
@@ -33,19 +33,24 @@
 #include <classes/ref_counted.hpp>
 #include <classes/node.hpp>
 #include "nodes/beehave_blackboard.h"
+#include "nodes/beehave_tree.h"
 
 namespace godot {
 
     class BeehaveContext : public RefCounted
     {
         GDCLASS(BeehaveContext, RefCounted);
 
+        BeehaveTree* tree;
         BeehaveBlackboard* blackboard;
         Node* actor;
         float delta;
     protected:
         static void _bind_methods();
     public:
+        BeehaveTree* get_tree() const;
+        void set_tree(BeehaveTree* tree);
+
         BeehaveBlackboard* get_blackboard() const;
         void set_blackboard(BeehaveBlackboard* blackboard);