/* limbo_ai_editor_plugin.cpp */ #ifdef TOOLS_ENABLED #include "limbo_ai_editor_plugin.h" #include "core/array.h" #include "core/class_db.h" #include "core/dictionary.h" #include "core/error_list.h" #include "core/error_macros.h" #include "core/io/config_file.h" #include "core/io/image_loader.h" #include "core/io/resource_loader.h" #include "core/io/resource_saver.h" #include "core/list.h" #include "core/math/math_defs.h" #include "core/object.h" #include "core/os/dir_access.h" #include "core/os/memory.h" #include "core/print_string.h" #include "core/project_settings.h" #include "core/resource.h" #include "core/script_language.h" #include "core/string_name.h" #include "core/typedefs.h" #include "core/ustring.h" #include "core/variant.h" #include "core/vector.h" #include "editor/editor_inspector.h" #include "editor/editor_node.h" #include "editor/editor_plugin.h" #include "editor/editor_scale.h" #include "editor/editor_settings.h" #include "modules/limboai/bt/actions/bt_action.h" #include "modules/limboai/bt/behavior_tree.h" #include "modules/limboai/bt/bt_task.h" #include "modules/limboai/bt/composites/bt_parallel.h" #include "modules/limboai/bt/composites/bt_selector.h" #include "modules/limboai/bt/composites/bt_sequence.h" #include "scene/gui/box_container.h" #include "scene/gui/button.h" #include "scene/gui/control.h" #include "scene/gui/file_dialog.h" #include "scene/gui/flow_container.h" #include "scene/gui/label.h" #include "scene/gui/line_edit.h" #include "scene/gui/popup_menu.h" #include "scene/gui/scroll_container.h" #include "scene/gui/separator.h" #include "scene/gui/split_container.h" #include "scene/gui/tree.h" #include ////////////////////////////// TaskTree ////////////////////////////////////// TreeItem *TaskTree::_create_tree(const Ref &p_task, TreeItem *p_parent, int p_idx) { ERR_FAIL_COND_V(p_task.is_null(), nullptr); TreeItem *item = tree->create_item(p_parent, p_idx); item->set_metadata(0, p_task); // p_task->connect("changed"...) for (int i = 0; i < p_task->get_child_count(); i++) { _create_tree(p_task->get_child(i), item); } _update_item(item); return item; } void TaskTree::_update_item(TreeItem *p_item) { ERR_FAIL_COND_MSG(p_item == nullptr, "Argument \"p_item\" is null."); Ref task = p_item->get_metadata(0); ERR_FAIL_COND_MSG(!task.is_valid(), "Invalid task reference in metadata."); p_item->set_text(0, task->get_task_name()); if (task->get_script_instance() && !task->get_script_instance()->get_script()->get_path().empty()) { p_item->set_icon(0, LimboAIEditor::get_task_icon(task->get_script_instance()->get_script()->get_path())); } else { p_item->set_icon(0, LimboAIEditor::get_task_icon(task->get_class())); } p_item->set_editable(0, false); for (int i = 0; i < p_item->get_button_count(0); i++) { p_item->erase_button(0, i); } String warning = task->get_configuration_warning(); if (!warning.empty()) { p_item->add_button(0, get_icon("NodeWarning", "EditorIcons"), 0, false, warning); } // TODO: Update probabilities. } void TaskTree::_update_tree() { Ref sel; if (tree->get_selected()) { sel = tree->get_selected()->get_metadata(0); } tree->clear(); if (bt->get_root_task().is_valid()) { _create_tree(bt->get_root_task(), nullptr); } TreeItem *item = _find_item(sel); if (item) { item->select(0); } } TreeItem *TaskTree::_find_item(const Ref &p_task) const { if (p_task.is_null()) { return nullptr; } TreeItem *item = tree->get_root(); List stack; while (item && item->get_metadata(0) != p_task) { if (item->get_children()) { stack.push_back(item->get_children()); } item = item->get_next(); if (item == nullptr && !stack.empty()) { item = stack.front()->get(); stack.pop_front(); } } return item; } void TaskTree::_on_item_rmb_selected(const Vector2 &p_pos) { emit_signal("rmb_pressed", tree->get_global_transform().xform(p_pos)); } void TaskTree::_on_item_selected() { if (last_selected.is_valid()) { update_task(last_selected); if (last_selected->is_connected("changed", this, "_on_task_changed")) { last_selected->disconnect("changed", this, "_on_task_changed"); } } last_selected = get_selected(); last_selected->connect("changed", this, "_on_task_changed"); emit_signal("task_selected", last_selected); } void TaskTree::_on_task_changed() { _update_item(tree->get_selected()); } void TaskTree::load_bt(const Ref &p_behavior_tree) { ERR_FAIL_COND_MSG(p_behavior_tree.is_null(), "Tried to load a null tree."); if (last_selected.is_valid() and last_selected->is_connected("changed", this, "_on_task_changed")) { last_selected->disconnect("changed", this, "_on_task_changed"); } bt = p_behavior_tree; tree->clear(); if (bt->get_root_task().is_valid()) { _create_tree(bt->get_root_task(), nullptr); } } void TaskTree::update_task(const Ref &p_task) { ERR_FAIL_COND(p_task.is_null()); TreeItem *item = _find_item(p_task); if (item) { _update_item(item); } } Ref TaskTree::get_selected() const { if (tree->get_selected()) { return tree->get_selected()->get_metadata(0); } return nullptr; } void TaskTree::deselect() { TreeItem *sel = tree->get_selected(); if (sel) { sel->deselect(0); } } Variant TaskTree::get_drag_data_fw(const Point2 &p_point, Control *p_from) { if (editable && tree->get_item_at_position(p_point)) { Dictionary drag_data; drag_data["type"] = "task"; drag_data["task"] = tree->get_item_at_position(p_point)->get_metadata(0); tree->set_drop_mode_flags(Tree::DROP_MODE_INBETWEEN | Tree::DROP_MODE_ON_ITEM); return drag_data; } return Variant(); } bool TaskTree::can_drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from) const { if (!editable) { return false; } Dictionary d = p_data; if (!d.has("type") || !d.has("task")) { return false; } int section = tree->get_drop_section_at_position(p_point); TreeItem *item = tree->get_item_at_position(p_point); if (!item || section < -1 || (section == -1 && !item->get_parent())) { return false; } if (String(d["type"]) == "task") { Ref task = d["task"]; const Ref to_task = item->get_metadata(0); if (task != to_task && !to_task->is_descendant_of(task)) { return true; } } return false; } void TaskTree::drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from) { Dictionary d = p_data; TreeItem *item = tree->get_item_at_position(p_point); if (item && d.has("task")) { Ref task = d["task"]; emit_signal("task_dragged", task, item->get_metadata(0), tree->get_drop_section_at_position(p_point)); } } void TaskTree::_bind_methods() { ClassDB::bind_method(D_METHOD("_on_item_rmb_selected"), &TaskTree::_on_item_rmb_selected); ClassDB::bind_method(D_METHOD("_on_item_selected"), &TaskTree::_on_item_selected); ClassDB::bind_method(D_METHOD("_on_task_changed"), &TaskTree::_on_task_changed); ClassDB::bind_method(D_METHOD("load_bt", "p_behavior_tree"), &TaskTree::load_bt); ClassDB::bind_method(D_METHOD("get_bt"), &TaskTree::get_bt); ClassDB::bind_method(D_METHOD("update_tree"), &TaskTree::update_tree); ClassDB::bind_method(D_METHOD("update_task", "p_task"), &TaskTree::update_task); ClassDB::bind_method(D_METHOD("get_selected"), &TaskTree::get_selected); ClassDB::bind_method(D_METHOD("deselect"), &TaskTree::deselect); ClassDB::bind_method(D_METHOD("get_drag_data_fw"), &TaskTree::get_drag_data_fw); ClassDB::bind_method(D_METHOD("can_drop_data_fw"), &TaskTree::can_drop_data_fw); ClassDB::bind_method(D_METHOD("drop_data_fw"), &TaskTree::drop_data_fw); ADD_SIGNAL(MethodInfo("rmb_pressed")); ADD_SIGNAL(MethodInfo("task_selected")); ADD_SIGNAL(MethodInfo("task_dragged", PropertyInfo(Variant::OBJECT, "p_task", PROPERTY_HINT_RESOURCE_TYPE, "BTTask"), PropertyInfo(Variant::OBJECT, "p_to_task", PROPERTY_HINT_RESOURCE_TYPE, "BTTask"), PropertyInfo(Variant::INT, "p_type"))); } TaskTree::TaskTree() { editable = true; tree = memnew(Tree); add_child(tree); tree->set_columns(2); tree->set_column_expand(0, true); tree->set_column_expand(1, false); tree->set_column_min_width(1, 64); tree->set_anchor(MARGIN_RIGHT, ANCHOR_END); tree->set_anchor(MARGIN_BOTTOM, ANCHOR_END); tree->set_allow_rmb_select(true); tree->connect("item_rmb_selected", this, "_on_item_rmb_selected"); tree->connect("item_selected", this, "_on_item_selected"); tree->set_drag_forwarding(this); } TaskTree::~TaskTree() { if (last_selected.is_valid() and last_selected->is_connected("changed", this, "_on_task_changed")) { last_selected->disconnect("changed", this, "_on_task_changed"); } } ////////////////////////////// TaskTree ////////////////////////////////////// ////////////////////////////// TaskSection //////////////////////////////////// void TaskSection::_on_task_button_pressed(const StringName &p_task) { emit_signal("task_button_pressed", p_task); } void TaskSection::_on_header_pressed() { tasks_container->set_visible(!tasks_container->is_visible()); section_header->set_icon(tasks_container->is_visible() ? get_icon("GuiTreeArrowDown", "EditorIcons") : get_icon("GuiTreeArrowRight", "EditorIcons")); } void TaskSection::set_filter(String p_filter_text) { int num_hidden = 0; if (p_filter_text.empty()) { for (int i = 0; i < tasks_container->get_child_count(); i++) { Object::cast_to