240 lines
8.4 KiB
C++
240 lines
8.4 KiB
C++
/*************************************************************************/
|
|
/* root_motion_editor_plugin.cpp */
|
|
/*************************************************************************/
|
|
/* This file is part of: */
|
|
/* GODOT ENGINE */
|
|
/* https://godotengine.org */
|
|
/*************************************************************************/
|
|
/* Copyright (c) 2007-2022 Juan Linietsky, Ariel Manzur. */
|
|
/* Copyright (c) 2014-2022 Godot Engine contributors (cf. AUTHORS.md). */
|
|
/* */
|
|
/* Permission is hereby granted, free of charge, to any person obtaining */
|
|
/* a copy of this software and associated documentation files (the */
|
|
/* "Software"), to deal in the Software without restriction, including */
|
|
/* without limitation the rights to use, copy, modify, merge, publish, */
|
|
/* distribute, sublicense, and/or sell copies of the Software, and to */
|
|
/* permit persons to whom the Software is furnished to do so, subject to */
|
|
/* the following conditions: */
|
|
/* */
|
|
/* The above copyright notice and this permission notice shall be */
|
|
/* included in all copies or substantial portions of the Software. */
|
|
/* */
|
|
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
|
|
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
|
|
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
|
|
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
|
|
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
|
|
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
|
|
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
|
|
/*************************************************************************/
|
|
|
|
#include "root_motion_editor_plugin.h"
|
|
#include "editor/editor_node.h"
|
|
#include "scene/animation/animation_player.h"
|
|
#include "scene/animation/animation_tree.h"
|
|
#include "scene/main/window.h"
|
|
|
|
void EditorPropertyRootMotion::_confirmed() {
|
|
TreeItem *ti = filters->get_selected();
|
|
if (!ti) {
|
|
return;
|
|
}
|
|
|
|
NodePath path = ti->get_metadata(0);
|
|
emit_changed(get_edited_property(), path);
|
|
update_property();
|
|
filter_dialog->hide(); //may come from activated
|
|
}
|
|
|
|
void EditorPropertyRootMotion::_node_assign() {
|
|
AnimationTree *atree = Object::cast_to<AnimationTree>(get_edited_object());
|
|
if (!atree->has_node(atree->get_animation_player())) {
|
|
EditorNode::get_singleton()->show_warning(TTR("AnimationTree has no path set to an AnimationPlayer"));
|
|
return;
|
|
}
|
|
AnimationPlayer *player = Object::cast_to<AnimationPlayer>(atree->get_node(atree->get_animation_player()));
|
|
if (!player) {
|
|
EditorNode::get_singleton()->show_warning(TTR("Path to AnimationPlayer is invalid"));
|
|
return;
|
|
}
|
|
|
|
Node *base = player->get_node(player->get_root());
|
|
|
|
if (!base) {
|
|
EditorNode::get_singleton()->show_warning(TTR("Animation player has no valid root node path, so unable to retrieve track names."));
|
|
return;
|
|
}
|
|
|
|
HashSet<String> paths;
|
|
{
|
|
List<StringName> animations;
|
|
player->get_animation_list(&animations);
|
|
|
|
for (const StringName &E : animations) {
|
|
Ref<Animation> anim = player->get_animation(E);
|
|
for (int i = 0; i < anim->get_track_count(); i++) {
|
|
String pathname = anim->track_get_path(i).get_concatenated_names();
|
|
if (!paths.has(pathname)) {
|
|
paths.insert(pathname);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
filters->clear();
|
|
TreeItem *root = filters->create_item();
|
|
|
|
HashMap<String, TreeItem *> parenthood;
|
|
|
|
for (const String &E : paths) {
|
|
NodePath path = E;
|
|
TreeItem *ti = nullptr;
|
|
String accum;
|
|
for (int i = 0; i < path.get_name_count(); i++) {
|
|
String name = path.get_name(i);
|
|
if (!accum.is_empty()) {
|
|
accum += "/";
|
|
}
|
|
accum += name;
|
|
if (!parenthood.has(accum)) {
|
|
if (ti) {
|
|
ti = filters->create_item(ti);
|
|
} else {
|
|
ti = filters->create_item(root);
|
|
}
|
|
parenthood[accum] = ti;
|
|
ti->set_text(0, name);
|
|
ti->set_selectable(0, false);
|
|
ti->set_editable(0, false);
|
|
|
|
if (base->has_node(accum)) {
|
|
Node *node = base->get_node(accum);
|
|
ti->set_icon(0, EditorNode::get_singleton()->get_object_icon(node, "Node"));
|
|
}
|
|
|
|
} else {
|
|
ti = parenthood[accum];
|
|
}
|
|
}
|
|
|
|
Node *node = nullptr;
|
|
if (base->has_node(accum)) {
|
|
node = base->get_node(accum);
|
|
}
|
|
if (!node) {
|
|
continue; //no node, can't edit
|
|
}
|
|
|
|
Skeleton3D *skeleton = Object::cast_to<Skeleton3D>(node);
|
|
if (skeleton) {
|
|
HashMap<int, TreeItem *> items;
|
|
items.insert(-1, ti);
|
|
Ref<Texture> bone_icon = get_theme_icon(SNAME("BoneAttachment3D"), SNAME("EditorIcons"));
|
|
Vector<int> bones_to_process = skeleton->get_parentless_bones();
|
|
while (bones_to_process.size() > 0) {
|
|
int current_bone_idx = bones_to_process[0];
|
|
bones_to_process.erase(current_bone_idx);
|
|
|
|
Vector<int> current_bone_child_bones = skeleton->get_bone_children(current_bone_idx);
|
|
int child_bone_size = current_bone_child_bones.size();
|
|
for (int i = 0; i < child_bone_size; i++) {
|
|
bones_to_process.push_back(current_bone_child_bones[i]);
|
|
}
|
|
|
|
const int parent_idx = skeleton->get_bone_parent(current_bone_idx);
|
|
TreeItem *parent_item = items.find(parent_idx)->value;
|
|
|
|
TreeItem *joint_item = filters->create_item(parent_item);
|
|
items.insert(current_bone_idx, joint_item);
|
|
|
|
joint_item->set_text(0, skeleton->get_bone_name(current_bone_idx));
|
|
joint_item->set_icon(0, bone_icon);
|
|
joint_item->set_selectable(0, true);
|
|
joint_item->set_metadata(0, accum + ":" + skeleton->get_bone_name(current_bone_idx));
|
|
joint_item->set_collapsed(true);
|
|
}
|
|
}
|
|
}
|
|
|
|
filters->ensure_cursor_is_visible();
|
|
filter_dialog->popup_centered_ratio();
|
|
}
|
|
|
|
void EditorPropertyRootMotion::_node_clear() {
|
|
emit_changed(get_edited_property(), NodePath());
|
|
update_property();
|
|
}
|
|
|
|
void EditorPropertyRootMotion::update_property() {
|
|
NodePath p = get_edited_object()->get(get_edited_property());
|
|
assign->set_tooltip_text(p);
|
|
if (p == NodePath()) {
|
|
assign->set_icon(Ref<Texture2D>());
|
|
assign->set_text(TTR("Assign..."));
|
|
assign->set_flat(false);
|
|
return;
|
|
}
|
|
|
|
assign->set_icon(Ref<Texture2D>());
|
|
assign->set_text(p);
|
|
}
|
|
|
|
void EditorPropertyRootMotion::setup(const NodePath &p_base_hint) {
|
|
base_hint = p_base_hint;
|
|
}
|
|
|
|
void EditorPropertyRootMotion::_notification(int p_what) {
|
|
switch (p_what) {
|
|
case NOTIFICATION_ENTER_TREE:
|
|
case NOTIFICATION_THEME_CHANGED: {
|
|
Ref<Texture2D> t = get_theme_icon(SNAME("Clear"), SNAME("EditorIcons"));
|
|
clear->set_icon(t);
|
|
} break;
|
|
}
|
|
}
|
|
|
|
void EditorPropertyRootMotion::_bind_methods() {
|
|
}
|
|
|
|
EditorPropertyRootMotion::EditorPropertyRootMotion() {
|
|
HBoxContainer *hbc = memnew(HBoxContainer);
|
|
add_child(hbc);
|
|
assign = memnew(Button);
|
|
assign->set_h_size_flags(SIZE_EXPAND_FILL);
|
|
assign->set_clip_text(true);
|
|
assign->connect("pressed", callable_mp(this, &EditorPropertyRootMotion::_node_assign));
|
|
hbc->add_child(assign);
|
|
|
|
clear = memnew(Button);
|
|
clear->connect("pressed", callable_mp(this, &EditorPropertyRootMotion::_node_clear));
|
|
hbc->add_child(clear);
|
|
|
|
filter_dialog = memnew(ConfirmationDialog);
|
|
add_child(filter_dialog);
|
|
filter_dialog->set_title(TTR("Edit Filtered Tracks:"));
|
|
filter_dialog->connect("confirmed", callable_mp(this, &EditorPropertyRootMotion::_confirmed));
|
|
|
|
filters = memnew(Tree);
|
|
filter_dialog->add_child(filters);
|
|
filters->set_v_size_flags(SIZE_EXPAND_FILL);
|
|
filters->set_hide_root(true);
|
|
filters->connect("item_activated", callable_mp(this, &EditorPropertyRootMotion::_confirmed));
|
|
//filters->connect("item_edited", this, "_filter_edited");
|
|
}
|
|
|
|
//////////////////////////
|
|
|
|
bool EditorInspectorRootMotionPlugin::can_handle(Object *p_object) {
|
|
return true; // Can handle everything.
|
|
}
|
|
|
|
bool EditorInspectorRootMotionPlugin::parse_property(Object *p_object, const Variant::Type p_type, const String &p_path, const PropertyHint p_hint, const String &p_hint_text, const uint32_t p_usage, const bool p_wide) {
|
|
if (p_path == "root_motion_track" && p_object->is_class("AnimationTree") && p_type == Variant::NODE_PATH) {
|
|
EditorPropertyRootMotion *editor = memnew(EditorPropertyRootMotion);
|
|
add_property_editor(p_path, editor);
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|