146 lines
3.8 KiB
C++
146 lines
3.8 KiB
C++
/* limbo_debugger.cpp */
|
|
|
|
#include "limbo_debugger.h"
|
|
#include "behavior_tree_data.h"
|
|
#include "core/debugger/engine_debugger.h"
|
|
#include "core/error/error_macros.h"
|
|
#include "core/string/node_path.h"
|
|
#include "modules/limboai/bt/bt_task.h"
|
|
#include "scene/main/scene_tree.h"
|
|
#include "scene/main/window.h"
|
|
|
|
//// LimboDebugger
|
|
|
|
LimboDebugger *LimboDebugger::singleton = nullptr;
|
|
LimboDebugger *LimboDebugger::get_singleton() {
|
|
return singleton;
|
|
}
|
|
|
|
LimboDebugger::LimboDebugger() {
|
|
singleton = this;
|
|
#ifdef DEBUG_ENABLED
|
|
EngineDebugger::register_message_capture("limboai", EngineDebugger::Capture(nullptr, LimboDebugger::parse_message));
|
|
#endif
|
|
}
|
|
|
|
LimboDebugger::~LimboDebugger() {
|
|
singleton = nullptr;
|
|
}
|
|
|
|
void LimboDebugger::initialize() {
|
|
if (EngineDebugger::is_active()) {
|
|
memnew(LimboDebugger);
|
|
}
|
|
}
|
|
|
|
void LimboDebugger::deinitialize() {
|
|
if (singleton) {
|
|
memdelete(singleton);
|
|
}
|
|
}
|
|
|
|
#ifdef DEBUG_ENABLED
|
|
Error LimboDebugger::parse_message(void *p_user, const String &p_msg, const Array &p_args, bool &r_captured) {
|
|
r_captured = true;
|
|
if (p_msg == "track_bt_player") {
|
|
singleton->_track_tree(p_args[0]);
|
|
} else if (p_msg == "untrack_bt_player") {
|
|
singleton->_untrack_tree();
|
|
} else if (p_msg == "start_session") {
|
|
singleton->session_active = true;
|
|
singleton->_send_active_bt_players();
|
|
} else if (p_msg == "stop_session") {
|
|
singleton->session_active = false;
|
|
} else {
|
|
r_captured = false;
|
|
}
|
|
return OK;
|
|
}
|
|
|
|
void LimboDebugger::register_bt_instance(Ref<BTTask> p_instance, NodePath p_player_path) {
|
|
ERR_FAIL_COND(active_trees.has(p_player_path));
|
|
|
|
active_trees.insert(p_player_path, p_instance);
|
|
if (session_active) {
|
|
_send_active_bt_players();
|
|
}
|
|
}
|
|
|
|
void LimboDebugger::unregister_bt_instance(Ref<BTTask> p_instance, NodePath p_player_path) {
|
|
ERR_FAIL_COND(p_player_path.is_empty());
|
|
ERR_FAIL_COND(!active_trees.has(p_player_path));
|
|
|
|
if (tracked_tree == p_player_path) {
|
|
_untrack_tree();
|
|
}
|
|
active_trees.erase(p_player_path);
|
|
|
|
if (session_active) {
|
|
_send_active_bt_players();
|
|
}
|
|
}
|
|
|
|
void LimboDebugger::_track_tree(NodePath p_path) {
|
|
ERR_FAIL_COND(!active_trees.has(p_path));
|
|
|
|
if (!tracked_tree.is_empty()) {
|
|
_untrack_tree();
|
|
}
|
|
|
|
Node *node = SceneTree::get_singleton()->get_root()->get_node(p_path);
|
|
ERR_FAIL_COND(node == nullptr);
|
|
|
|
tracked_tree = p_path;
|
|
if (node->is_class("BTPlayer")) {
|
|
node->connect(SNAME("updated"), callable_mp(this, &LimboDebugger::_on_bt_updated).bind(p_path));
|
|
} else if (node->is_class("BTState")) {
|
|
node->connect(SNAME("updated"), callable_mp(this, &LimboDebugger::_on_state_updated).bind(p_path));
|
|
}
|
|
}
|
|
|
|
void LimboDebugger::_untrack_tree() {
|
|
if (tracked_tree.is_empty()) {
|
|
return;
|
|
}
|
|
|
|
NodePath was_tracking = tracked_tree;
|
|
tracked_tree = NodePath();
|
|
|
|
Node *node = SceneTree::get_singleton()->get_root()->get_node(was_tracking);
|
|
ERR_FAIL_COND(node == nullptr);
|
|
|
|
if (node->is_class("BTPlayer")) {
|
|
node->disconnect(SNAME("updated"), callable_mp(this, &LimboDebugger::_on_bt_updated));
|
|
} else if (node->is_class("BTState")) {
|
|
node->disconnect(SNAME("updated"), callable_mp(this, &LimboDebugger::_on_state_updated));
|
|
}
|
|
}
|
|
|
|
void LimboDebugger::_send_active_bt_players() {
|
|
Array arr;
|
|
for (KeyValue<NodePath, Ref<BTTask>> kv : active_trees) {
|
|
arr.append(kv.key);
|
|
}
|
|
EngineDebugger::get_singleton()->send_message("limboai:active_bt_players", arr);
|
|
}
|
|
|
|
void LimboDebugger::_on_bt_updated(int _status, NodePath p_path) {
|
|
if (p_path != tracked_tree) {
|
|
return;
|
|
}
|
|
Array arr;
|
|
BehaviorTreeData(active_trees.get(tracked_tree), tracked_tree).serialize(arr);
|
|
EngineDebugger::get_singleton()->send_message("limboai:bt_update", arr);
|
|
}
|
|
|
|
void LimboDebugger::_on_state_updated(float _delta, NodePath p_path) {
|
|
if (p_path != tracked_tree) {
|
|
return;
|
|
}
|
|
Array arr;
|
|
BehaviorTreeData(active_trees.get(tracked_tree), tracked_tree).serialize(arr);
|
|
EngineDebugger::get_singleton()->send_message("limboai:bt_update", arr);
|
|
}
|
|
|
|
#endif // DEBUG_ENABLED
|