Path: blob/master/scene/animation/animation_blend_tree.cpp
20956 views
/**************************************************************************/1/* animation_blend_tree.cpp */2/**************************************************************************/3/* This file is part of: */4/* GODOT ENGINE */5/* https://godotengine.org */6/**************************************************************************/7/* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */8/* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */9/* */10/* Permission is hereby granted, free of charge, to any person obtaining */11/* a copy of this software and associated documentation files (the */12/* "Software"), to deal in the Software without restriction, including */13/* without limitation the rights to use, copy, modify, merge, publish, */14/* distribute, sublicense, and/or sell copies of the Software, and to */15/* permit persons to whom the Software is furnished to do so, subject to */16/* the following conditions: */17/* */18/* The above copyright notice and this permission notice shall be */19/* included in all copies or substantial portions of the Software. */20/* */21/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */22/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */23/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */24/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */25/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */26/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */27/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */28/**************************************************************************/2930#include "animation_blend_tree.h"3132#include "scene/resources/animation.h"3334void AnimationNodeAnimation::set_animation(const StringName &p_name) {35animation = p_name;36}3738StringName AnimationNodeAnimation::get_animation() const {39return animation;40}4142Vector<String> (*AnimationNodeAnimation::get_editable_animation_list)() = nullptr;4344void AnimationNodeAnimation::get_parameter_list(List<PropertyInfo> *r_list) const {45AnimationNode::get_parameter_list(r_list);46r_list->push_back(PropertyInfo(Variant::BOOL, backward, PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NONE));47}4849Variant AnimationNodeAnimation::get_parameter_default_value(const StringName &p_parameter) const {50Variant ret = AnimationNode::get_parameter_default_value(p_parameter);51if (ret != Variant()) {52return ret;53}54if (p_parameter == backward) {55return false;56}57return 0.0;58}5960AnimationNode::NodeTimeInfo AnimationNodeAnimation::get_node_time_info() const {61NodeTimeInfo nti;62if (!process_state->tree->has_animation(animation)) {63return nti;64}6566if (use_custom_timeline) {67nti.length = timeline_length;68nti.loop_mode = loop_mode;69} else {70Ref<Animation> anim = process_state->tree->get_animation(animation);71nti.length = (double)anim->get_length();72nti.loop_mode = anim->get_loop_mode();73}74nti.position = get_parameter(current_position);7576return nti;77}7879void AnimationNodeAnimation::_validate_property(PropertyInfo &p_property) const {80if (Engine::get_singleton()->is_editor_hint() && p_property.name == "animation" && get_editable_animation_list) {81Vector<String> names = get_editable_animation_list();82String anims;83for (int i = 0; i < names.size(); i++) {84if (i > 0) {85anims += ",";86}87anims += String(names[i]);88}89if (!anims.is_empty()) {90p_property.hint = PROPERTY_HINT_ENUM;91p_property.hint_string = anims;92}93}9495if (!use_custom_timeline) {96if (p_property.name == "timeline_length" || p_property.name == "start_offset" || p_property.name == "loop_mode" || p_property.name == "stretch_time_scale") {97p_property.usage = PROPERTY_USAGE_NONE;98}99}100}101102AnimationNode::NodeTimeInfo AnimationNodeAnimation::process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only) {103process_state->is_testing = p_test_only;104105AnimationMixer::PlaybackInfo pi = p_playback_info;106if (p_playback_info.seeked) {107if (p_playback_info.is_external_seeking) {108pi.delta = get_node_time_info().position - p_playback_info.time;109}110} else {111pi.time = get_node_time_info().position + (get_parameter(backward) ? -p_playback_info.delta : p_playback_info.delta);112}113114NodeTimeInfo nti = _process(pi, p_test_only);115116if (!p_test_only) {117set_node_time_info(nti);118}119120return nti;121}122123AnimationNode::NodeTimeInfo AnimationNodeAnimation::_process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only) {124if (!process_state->tree->has_animation(animation)) {125AnimationNodeBlendTree *tree = Object::cast_to<AnimationNodeBlendTree>(node_state.parent);126if (tree) {127String node_name = tree->get_node_name(Ref<AnimationNodeAnimation>(this));128make_invalid(vformat(RTR("On BlendTree node '%s', animation not found: '%s'"), node_name, animation));129130} else {131make_invalid(vformat(RTR("Animation not found: '%s'"), animation));132}133134return NodeTimeInfo();135}136137Ref<Animation> anim = process_state->tree->get_animation(animation);138double anim_size = (double)anim->get_length();139140NodeTimeInfo cur_nti = get_node_time_info();141double cur_len = cur_nti.length;142double cur_time = p_playback_info.time;143double cur_delta = p_playback_info.delta;144bool cur_backward = get_parameter(backward);145146Animation::LoopMode cur_loop_mode = cur_nti.loop_mode;147double prev_time = cur_nti.position;148149Animation::LoopedFlag looped_flag = Animation::LOOPED_FLAG_NONE;150bool node_backward = play_mode == PLAY_MODE_BACKWARD;151152bool p_seek = p_playback_info.seeked;153bool p_is_external_seeking = p_playback_info.is_external_seeking;154155// 1. Progress for AnimationNode.156bool will_end = Animation::is_greater_or_equal_approx(cur_time + cur_delta, cur_len);157bool is_started = p_seek && !p_is_external_seeking && Math::is_zero_approx(cur_time);158bool immediately_after_start = is_started && advance_on_start;159160// 1. Progress for AnimationNode.161if (immediately_after_start) {162cur_time = cur_delta;163}164if (cur_loop_mode != Animation::LOOP_NONE) {165if (cur_loop_mode == Animation::LOOP_LINEAR) {166if (!Math::is_zero_approx(cur_len)) {167cur_time = Math::fposmod(cur_time, cur_len);168}169cur_backward = false;170} else {171if (!Math::is_zero_approx(cur_len)) {172if (Animation::is_greater_or_equal_approx(prev_time, 0) && Animation::is_less_approx(cur_time, 0)) {173cur_backward = !cur_backward;174} else if (Animation::is_less_or_equal_approx(prev_time, cur_len) && Animation::is_greater_approx(cur_time, cur_len)) {175cur_backward = !cur_backward;176}177cur_time = Math::pingpong(cur_time, cur_len);178}179}180} else {181if (Animation::is_less_approx(cur_time, 0)) {182cur_delta += cur_time;183cur_time = 0;184} else if (Animation::is_greater_approx(cur_time, cur_len)) {185cur_delta += cur_time - cur_len;186cur_time = cur_len;187}188cur_backward = false;189// If ended, don't progress AnimationNode. So set delta to 0.190if (!Math::is_zero_approx(cur_delta)) {191if (play_mode == PLAY_MODE_FORWARD) {192if (Animation::is_greater_or_equal_approx(prev_time, cur_len)) {193cur_delta = 0;194}195} else {196if (Animation::is_less_or_equal_approx(prev_time, 0)) {197cur_delta = 0;198}199}200}201}202203// 2. For return, store "AnimationNode" time info here, not "Animation" time info as below.204NodeTimeInfo nti;205nti.length = cur_len;206nti.position = cur_time;207nti.delta = cur_delta;208nti.loop_mode = cur_loop_mode;209nti.will_end = will_end;210211// 3. Progress for Animation.212double prev_playback_time = prev_time + start_offset;213double cur_playback_time = cur_time + start_offset;214if (stretch_time_scale) {215double mlt = anim_size / cur_len;216prev_playback_time *= mlt;217cur_playback_time *= mlt;218cur_delta *= mlt;219}220if (cur_loop_mode == Animation::LOOP_LINEAR) {221if (!Math::is_zero_approx(anim_size)) {222prev_playback_time = Math::fposmod(prev_playback_time, anim_size);223cur_playback_time = Math::fposmod(cur_playback_time, anim_size);224if (Animation::is_greater_or_equal_approx(prev_playback_time, 0) && Animation::is_less_approx(cur_playback_time, 0)) {225looped_flag = node_backward ? Animation::LOOPED_FLAG_END : Animation::LOOPED_FLAG_START;226}227if (Animation::is_less_or_equal_approx(prev_playback_time, anim_size) && Animation::is_greater_approx(cur_playback_time, anim_size)) {228looped_flag = node_backward ? Animation::LOOPED_FLAG_START : Animation::LOOPED_FLAG_END;229}230}231} else if (cur_loop_mode == Animation::LOOP_PINGPONG) {232if (!Math::is_zero_approx(anim_size)) {233if (Animation::is_greater_or_equal_approx(Math::fposmod(cur_playback_time, anim_size * 2.0), anim_size)) {234cur_delta = -cur_delta; // Needed for retrieving discrete keys correctly.235}236prev_playback_time = Math::pingpong(prev_playback_time, anim_size);237cur_playback_time = Math::pingpong(cur_playback_time, anim_size);238if (Animation::is_greater_or_equal_approx(prev_playback_time, 0) && Animation::is_less_approx(cur_playback_time, 0)) {239looped_flag = node_backward ? Animation::LOOPED_FLAG_END : Animation::LOOPED_FLAG_START;240}241if (Animation::is_less_or_equal_approx(prev_playback_time, anim_size) && Animation::is_greater_approx(cur_playback_time, anim_size)) {242looped_flag = node_backward ? Animation::LOOPED_FLAG_START : Animation::LOOPED_FLAG_END;243}244}245} else {246if (Animation::is_less_approx(cur_playback_time, 0)) {247cur_playback_time = 0;248} else if (Animation::is_greater_approx(cur_playback_time, anim_size)) {249cur_playback_time = anim_size;250}251252// Emit start & finish signal. Internally, the detections are the same for backward.253// We should use call_deferred since the track keys are still being processed.254if (process_state->tree && !p_test_only) {255// AnimationTree uses seek to 0 "internally" to process the first key of the animation, which is used as the start detection.256if (is_started) {257process_state->tree->call_deferred(SNAME("emit_signal"), SceneStringName(animation_started), animation);258}259// Finished.260if (Animation::is_less_approx(prev_playback_time, anim_size) && Animation::is_greater_or_equal_approx(cur_playback_time, anim_size)) {261cur_playback_time = anim_size;262process_state->tree->call_deferred(SNAME("emit_signal"), SceneStringName(animation_finished), animation);263}264}265}266267if (!p_test_only) {268// Force process first key for Discrete/Method/Audio/AnimationPlayback.269if (immediately_after_start) {270AnimationMixer::PlaybackInfo pi = p_playback_info;271pi.start = 0.0;272pi.end = cur_len;273if (play_mode == PLAY_MODE_FORWARD) {274pi.time = 0;275} else {276pi.time = anim_size;277}278pi.delta = 0;279pi.weight = CMP_EPSILON;280blend_animation(animation, pi);281}282283AnimationMixer::PlaybackInfo pi = p_playback_info;284pi.start = 0.0;285pi.end = cur_len;286if (play_mode == PLAY_MODE_FORWARD) {287pi.time = cur_playback_time;288} else {289pi.time = anim_size - cur_playback_time;290}291if (node_backward ? cur_backward : !cur_backward) {292pi.delta = cur_delta;293} else {294pi.delta = -cur_delta;295}296pi.weight = 1.0;297pi.looped_flag = looped_flag;298blend_animation(animation, pi);299300set_parameter(backward, cur_backward);301}302303return nti;304}305306String AnimationNodeAnimation::get_caption() const {307return "Animation";308}309310void AnimationNodeAnimation::set_play_mode(PlayMode p_play_mode) {311play_mode = p_play_mode;312}313314AnimationNodeAnimation::PlayMode AnimationNodeAnimation::get_play_mode() const {315return play_mode;316}317318void AnimationNodeAnimation::set_backward(bool p_backward) {319set_parameter(backward, p_backward);320}321322bool AnimationNodeAnimation::is_backward() const {323return get_parameter(backward);324}325326void AnimationNodeAnimation::set_advance_on_start(bool p_advance_on_start) {327advance_on_start = p_advance_on_start;328}329330bool AnimationNodeAnimation::is_advance_on_start() const {331return advance_on_start;332}333334void AnimationNodeAnimation::set_use_custom_timeline(bool p_use_custom_timeline) {335use_custom_timeline = p_use_custom_timeline;336notify_property_list_changed();337}338339bool AnimationNodeAnimation::is_using_custom_timeline() const {340return use_custom_timeline;341}342343void AnimationNodeAnimation::set_timeline_length(double p_length) {344timeline_length = p_length;345}346347double AnimationNodeAnimation::get_timeline_length() const {348return timeline_length;349}350351void AnimationNodeAnimation::set_stretch_time_scale(bool p_stretch_time_scale) {352stretch_time_scale = p_stretch_time_scale;353notify_property_list_changed();354}355356bool AnimationNodeAnimation::is_stretching_time_scale() const {357return stretch_time_scale;358}359360void AnimationNodeAnimation::set_start_offset(double p_offset) {361start_offset = p_offset;362}363364double AnimationNodeAnimation::get_start_offset() const {365return start_offset;366}367368void AnimationNodeAnimation::set_loop_mode(Animation::LoopMode p_loop_mode) {369loop_mode = p_loop_mode;370}371372Animation::LoopMode AnimationNodeAnimation::get_loop_mode() const {373return loop_mode;374}375376void AnimationNodeAnimation::_bind_methods() {377ClassDB::bind_method(D_METHOD("set_animation", "name"), &AnimationNodeAnimation::set_animation);378ClassDB::bind_method(D_METHOD("get_animation"), &AnimationNodeAnimation::get_animation);379380ClassDB::bind_method(D_METHOD("set_play_mode", "mode"), &AnimationNodeAnimation::set_play_mode);381ClassDB::bind_method(D_METHOD("get_play_mode"), &AnimationNodeAnimation::get_play_mode);382383ClassDB::bind_method(D_METHOD("set_advance_on_start", "advance_on_start"), &AnimationNodeAnimation::set_advance_on_start);384ClassDB::bind_method(D_METHOD("is_advance_on_start"), &AnimationNodeAnimation::is_advance_on_start);385386ClassDB::bind_method(D_METHOD("set_use_custom_timeline", "use_custom_timeline"), &AnimationNodeAnimation::set_use_custom_timeline);387ClassDB::bind_method(D_METHOD("is_using_custom_timeline"), &AnimationNodeAnimation::is_using_custom_timeline);388389ClassDB::bind_method(D_METHOD("set_timeline_length", "timeline_length"), &AnimationNodeAnimation::set_timeline_length);390ClassDB::bind_method(D_METHOD("get_timeline_length"), &AnimationNodeAnimation::get_timeline_length);391392ClassDB::bind_method(D_METHOD("set_stretch_time_scale", "stretch_time_scale"), &AnimationNodeAnimation::set_stretch_time_scale);393ClassDB::bind_method(D_METHOD("is_stretching_time_scale"), &AnimationNodeAnimation::is_stretching_time_scale);394395ClassDB::bind_method(D_METHOD("set_start_offset", "start_offset"), &AnimationNodeAnimation::set_start_offset);396ClassDB::bind_method(D_METHOD("get_start_offset"), &AnimationNodeAnimation::get_start_offset);397398ClassDB::bind_method(D_METHOD("set_loop_mode", "loop_mode"), &AnimationNodeAnimation::set_loop_mode);399ClassDB::bind_method(D_METHOD("get_loop_mode"), &AnimationNodeAnimation::get_loop_mode);400401ADD_PROPERTY(PropertyInfo(Variant::STRING_NAME, "animation"), "set_animation", "get_animation");402ADD_PROPERTY(PropertyInfo(Variant::INT, "play_mode", PROPERTY_HINT_ENUM, "Forward,Backward"), "set_play_mode", "get_play_mode");403ADD_PROPERTY(PropertyInfo(Variant::BOOL, "advance_on_start"), "set_advance_on_start", "is_advance_on_start");404ADD_PROPERTY(PropertyInfo(Variant::BOOL, "use_custom_timeline"), "set_use_custom_timeline", "is_using_custom_timeline");405ADD_PROPERTY(PropertyInfo(Variant::FLOAT, "timeline_length", PROPERTY_HINT_RANGE, "0.001,60,0.001,or_greater,or_less,hide_control,suffix:s"), "set_timeline_length", "get_timeline_length");406ADD_PROPERTY(PropertyInfo(Variant::BOOL, "stretch_time_scale"), "set_stretch_time_scale", "is_stretching_time_scale");407ADD_PROPERTY(PropertyInfo(Variant::FLOAT, "start_offset", PROPERTY_HINT_RANGE, "-60,60,0.001,or_greater,or_less,hide_control,suffix:s"), "set_start_offset", "get_start_offset");408ADD_PROPERTY(PropertyInfo(Variant::INT, "loop_mode", PROPERTY_HINT_ENUM, "None,Linear,Ping-Pong"), "set_loop_mode", "get_loop_mode");409410BIND_ENUM_CONSTANT(PLAY_MODE_FORWARD);411BIND_ENUM_CONSTANT(PLAY_MODE_BACKWARD);412}413414AnimationNodeAnimation::AnimationNodeAnimation() {415}416417////////////////////////////////////////////////////////418419void AnimationNodeSync::_bind_methods() {420ClassDB::bind_method(D_METHOD("set_use_sync", "enable"), &AnimationNodeSync::set_use_sync);421ClassDB::bind_method(D_METHOD("is_using_sync"), &AnimationNodeSync::is_using_sync);422423ADD_PROPERTY(PropertyInfo(Variant::BOOL, "sync"), "set_use_sync", "is_using_sync");424}425426void AnimationNodeSync::set_use_sync(bool p_sync) {427sync = p_sync;428}429430bool AnimationNodeSync::is_using_sync() const {431return sync;432}433434AnimationNodeSync::AnimationNodeSync() {435}436437////////////////////////////////////////////////////////438void AnimationNodeOneShot::get_parameter_list(List<PropertyInfo> *r_list) const {439AnimationNode::get_parameter_list(r_list);440r_list->push_back(PropertyInfo(Variant::BOOL, active, PROPERTY_HINT_NONE, "", PROPERTY_USAGE_EDITOR | PROPERTY_USAGE_STORAGE | PROPERTY_USAGE_READ_ONLY));441r_list->push_back(PropertyInfo(Variant::BOOL, internal_active, PROPERTY_HINT_NONE, "", PROPERTY_USAGE_STORAGE | PROPERTY_USAGE_READ_ONLY));442r_list->push_back(PropertyInfo(Variant::INT, request, PROPERTY_HINT_ENUM, ",Fire,Abort,Fade Out"));443r_list->push_back(PropertyInfo(Variant::FLOAT, fade_in_remaining, PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NONE));444r_list->push_back(PropertyInfo(Variant::FLOAT, fade_out_remaining, PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NONE));445r_list->push_back(PropertyInfo(Variant::FLOAT, time_to_restart, PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NONE));446}447448Variant AnimationNodeOneShot::get_parameter_default_value(const StringName &p_parameter) const {449Variant ret = AnimationNode::get_parameter_default_value(p_parameter);450if (ret != Variant()) {451return ret;452}453454if (p_parameter == request) {455return ONE_SHOT_REQUEST_NONE;456} else if (p_parameter == active || p_parameter == internal_active) {457return false;458} else if (p_parameter == time_to_restart) {459return -1.0;460}461462return 0.0;463}464465bool AnimationNodeOneShot::is_parameter_read_only(const StringName &p_parameter) const {466if (AnimationNode::is_parameter_read_only(p_parameter)) {467return true;468}469470if (p_parameter == active || p_parameter == internal_active) {471return true;472}473return false;474}475476void AnimationNodeOneShot::set_fade_in_time(double p_time) {477fade_in = p_time;478}479480double AnimationNodeOneShot::get_fade_in_time() const {481return fade_in;482}483484void AnimationNodeOneShot::set_fade_out_time(double p_time) {485fade_out = p_time;486}487488double AnimationNodeOneShot::get_fade_out_time() const {489return fade_out;490}491492void AnimationNodeOneShot::set_fade_in_curve(const Ref<Curve> &p_curve) {493fade_in_curve = p_curve;494}495496Ref<Curve> AnimationNodeOneShot::get_fade_in_curve() const {497return fade_in_curve;498}499500void AnimationNodeOneShot::set_fade_out_curve(const Ref<Curve> &p_curve) {501fade_out_curve = p_curve;502}503504Ref<Curve> AnimationNodeOneShot::get_fade_out_curve() const {505return fade_out_curve;506}507508void AnimationNodeOneShot::set_auto_restart_enabled(bool p_enabled) {509auto_restart = p_enabled;510}511512void AnimationNodeOneShot::set_auto_restart_delay(double p_time) {513auto_restart_delay = p_time;514}515516void AnimationNodeOneShot::set_auto_restart_random_delay(double p_time) {517auto_restart_random_delay = p_time;518}519520bool AnimationNodeOneShot::is_auto_restart_enabled() const {521return auto_restart;522}523524double AnimationNodeOneShot::get_auto_restart_delay() const {525return auto_restart_delay;526}527528double AnimationNodeOneShot::get_auto_restart_random_delay() const {529return auto_restart_random_delay;530}531532void AnimationNodeOneShot::set_mix_mode(MixMode p_mix) {533mix = p_mix;534}535536AnimationNodeOneShot::MixMode AnimationNodeOneShot::get_mix_mode() const {537return mix;538}539540void AnimationNodeOneShot::set_break_loop_at_end(bool p_enable) {541break_loop_at_end = p_enable;542}543544bool AnimationNodeOneShot::is_loop_broken_at_end() const {545return break_loop_at_end;546}547548void AnimationNodeOneShot::set_abort_on_reset(bool p_enable) {549abort_on_reset = p_enable;550}551552bool AnimationNodeOneShot::is_aborted_on_reset() const {553return abort_on_reset;554}555556String AnimationNodeOneShot::get_caption() const {557return "OneShot";558}559560bool AnimationNodeOneShot::has_filter() const {561return true;562}563564AnimationNode::NodeTimeInfo AnimationNodeOneShot::_process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only) {565OneShotRequest cur_request = static_cast<OneShotRequest>((int)get_parameter(request));566bool cur_active = get_parameter(active);567bool cur_internal_active = get_parameter(internal_active);568NodeTimeInfo cur_nti = get_node_time_info();569double cur_time_to_restart = get_parameter(time_to_restart);570double cur_fade_in_remaining = get_parameter(fade_in_remaining);571double cur_fade_out_remaining = get_parameter(fade_out_remaining);572573set_parameter(request, ONE_SHOT_REQUEST_NONE);574575bool is_shooting = true;576bool is_fading_out = cur_active == true && cur_internal_active == false;577578double p_time = p_playback_info.time;579double p_delta = p_playback_info.delta;580double abs_delta = Math::abs(p_delta);581bool p_seek = p_playback_info.seeked;582bool p_is_external_seeking = p_playback_info.is_external_seeking;583584bool do_start = cur_request == ONE_SHOT_REQUEST_FIRE;585586bool is_reset = Math::is_zero_approx(p_time) && p_seek && !p_is_external_seeking;587if (is_reset && cur_internal_active) {588do_start = true;589}590591bool is_abort = cur_request == ONE_SHOT_REQUEST_ABORT;592if (is_reset && !do_start && (is_fading_out || (abort_on_reset && cur_active))) {593is_abort = true;594}595596if (is_abort) {597set_parameter(internal_active, false);598set_parameter(active, false);599set_parameter(time_to_restart, -1);600set_parameter(fade_out_remaining, 0);601cur_fade_out_remaining = 0;602is_fading_out = false;603is_shooting = false;604} else if (cur_request == ONE_SHOT_REQUEST_FADE_OUT && !is_fading_out) { // If fading, keep current fade.605if (cur_active) {606// Request fading.607is_fading_out = true;608cur_fade_out_remaining = fade_out;609cur_fade_in_remaining = 0;610} else {611// Shot is ended, do nothing.612is_shooting = false;613}614set_parameter(internal_active, false);615set_parameter(time_to_restart, -1);616} else if (!do_start && !cur_active) {617if (Animation::is_greater_or_equal_approx(cur_time_to_restart, 0) && !p_seek) {618cur_time_to_restart -= abs_delta;619if (Animation::is_less_approx(cur_time_to_restart, 0)) {620do_start = true; // Restart.621}622set_parameter(time_to_restart, cur_time_to_restart);623}624if (!do_start) {625is_shooting = false;626}627}628629bool os_seek = p_seek;630631if (!is_shooting) {632AnimationMixer::PlaybackInfo pi = p_playback_info;633pi.weight = 1.0;634return blend_input(0, pi, FILTER_IGNORE, sync, p_test_only);635}636637if (do_start) {638os_seek = true;639if (!cur_internal_active) {640cur_fade_in_remaining = fade_in; // If already active, don't fade-in again.641}642cur_internal_active = true;643set_parameter(request, ONE_SHOT_REQUEST_NONE);644set_parameter(internal_active, true);645set_parameter(active, true);646}647648real_t blend = 1.0;649bool use_blend = sync;650651if (Animation::is_greater_approx(cur_fade_in_remaining, 0)) {652if (Animation::is_greater_approx(fade_in, 0)) {653use_blend = true;654blend = (fade_in - cur_fade_in_remaining) / fade_in;655if (fade_in_curve.is_valid()) {656blend = fade_in_curve->sample(blend);657}658} else {659blend = 0; // Should not happen.660}661}662663if (is_fading_out) {664use_blend = true;665if (Animation::is_greater_approx(fade_out, 0)) {666blend = cur_fade_out_remaining / fade_out;667if (fade_out_curve.is_valid()) {668blend = 1.0 - fade_out_curve->sample(1.0 - blend);669}670} else {671blend = 0;672}673}674675AnimationMixer::PlaybackInfo pi = p_playback_info;676NodeTimeInfo main_nti;677if (mix == MIX_MODE_ADD) {678pi.weight = 1.0;679main_nti = blend_input(0, pi, FILTER_IGNORE, sync, p_test_only);680} else {681pi.seeked &= use_blend;682pi.weight = 1.0 - blend;683main_nti = blend_input(0, pi, FILTER_BLEND, sync, p_test_only); // Unlike below, processing this edge is a corner case.684}685686pi = p_playback_info;687if (do_start) {688pi.time = 0;689} else if (os_seek) {690pi.time = cur_nti.position;691}692pi.seeked = os_seek;693pi.weight = Math::is_zero_approx(blend) ? CMP_EPSILON : blend;694695NodeTimeInfo os_nti = blend_input(1, pi, FILTER_PASS, true, p_test_only); // Blend values must be more than CMP_EPSILON to process discrete keys in edge.696697if (Animation::is_less_or_equal_approx(cur_fade_in_remaining, 0) && !do_start && !is_fading_out) {698// Predict time scale by difference of delta times to estimate input animation's remain time in self time scale.699// TODO: Time scale should be included into NodeTimeInfo for Godot 5.0.700double abs_os_delta = Math::abs(os_nti.delta);701double tscl = Math::is_zero_approx(abs_delta) || Math::is_zero_approx(abs_os_delta) || Math::is_equal_approx(abs_delta, abs_os_delta) ? 1.0 : (abs_delta / abs_os_delta);702double os_rem = os_nti.get_remain(break_loop_at_end) * tscl;703if (Animation::is_less_or_equal_approx(os_rem, fade_out)) {704is_fading_out = true;705cur_fade_out_remaining = os_rem + abs_delta;706cur_fade_in_remaining = 0;707set_parameter(internal_active, false);708}709}710711if (!p_seek) {712if (Animation::is_less_or_equal_approx(os_nti.get_remain(break_loop_at_end), 0) || (is_fading_out && Animation::is_less_or_equal_approx(cur_fade_out_remaining, 0))) {713set_parameter(internal_active, false);714set_parameter(active, false);715if (auto_restart) {716double restart_sec = auto_restart_delay + Math::randd() * auto_restart_random_delay;717set_parameter(time_to_restart, restart_sec);718}719}720if (!do_start) {721cur_fade_in_remaining = MAX(0, cur_fade_in_remaining - abs_delta); // Don't consider seeked delta by restart.722}723cur_fade_out_remaining = MAX(0, cur_fade_out_remaining - abs_delta);724}725726set_parameter(fade_in_remaining, cur_fade_in_remaining);727set_parameter(fade_out_remaining, cur_fade_out_remaining);728729return cur_internal_active ? os_nti : main_nti;730}731732void AnimationNodeOneShot::_bind_methods() {733ClassDB::bind_method(D_METHOD("set_fadein_time", "time"), &AnimationNodeOneShot::set_fade_in_time);734ClassDB::bind_method(D_METHOD("get_fadein_time"), &AnimationNodeOneShot::get_fade_in_time);735736ClassDB::bind_method(D_METHOD("set_fadein_curve", "curve"), &AnimationNodeOneShot::set_fade_in_curve);737ClassDB::bind_method(D_METHOD("get_fadein_curve"), &AnimationNodeOneShot::get_fade_in_curve);738739ClassDB::bind_method(D_METHOD("set_fadeout_time", "time"), &AnimationNodeOneShot::set_fade_out_time);740ClassDB::bind_method(D_METHOD("get_fadeout_time"), &AnimationNodeOneShot::get_fade_out_time);741742ClassDB::bind_method(D_METHOD("set_fadeout_curve", "curve"), &AnimationNodeOneShot::set_fade_out_curve);743ClassDB::bind_method(D_METHOD("get_fadeout_curve"), &AnimationNodeOneShot::get_fade_out_curve);744745ClassDB::bind_method(D_METHOD("set_break_loop_at_end", "enable"), &AnimationNodeOneShot::set_break_loop_at_end);746ClassDB::bind_method(D_METHOD("is_loop_broken_at_end"), &AnimationNodeOneShot::is_loop_broken_at_end);747748ClassDB::bind_method(D_METHOD("set_abort_on_reset", "enable"), &AnimationNodeOneShot::set_abort_on_reset);749ClassDB::bind_method(D_METHOD("is_aborted_on_reset"), &AnimationNodeOneShot::is_aborted_on_reset);750751ClassDB::bind_method(D_METHOD("set_autorestart", "active"), &AnimationNodeOneShot::set_auto_restart_enabled);752ClassDB::bind_method(D_METHOD("has_autorestart"), &AnimationNodeOneShot::is_auto_restart_enabled);753754ClassDB::bind_method(D_METHOD("set_autorestart_delay", "time"), &AnimationNodeOneShot::set_auto_restart_delay);755ClassDB::bind_method(D_METHOD("get_autorestart_delay"), &AnimationNodeOneShot::get_auto_restart_delay);756757ClassDB::bind_method(D_METHOD("set_autorestart_random_delay", "time"), &AnimationNodeOneShot::set_auto_restart_random_delay);758ClassDB::bind_method(D_METHOD("get_autorestart_random_delay"), &AnimationNodeOneShot::get_auto_restart_random_delay);759760ClassDB::bind_method(D_METHOD("set_mix_mode", "mode"), &AnimationNodeOneShot::set_mix_mode);761ClassDB::bind_method(D_METHOD("get_mix_mode"), &AnimationNodeOneShot::get_mix_mode);762763ADD_PROPERTY(PropertyInfo(Variant::INT, "mix_mode", PROPERTY_HINT_ENUM, "Blend,Add"), "set_mix_mode", "get_mix_mode");764765ADD_PROPERTY(PropertyInfo(Variant::FLOAT, "fadein_time", PROPERTY_HINT_RANGE, "0,60,0.01,or_greater,suffix:s"), "set_fadein_time", "get_fadein_time");766ADD_PROPERTY(PropertyInfo(Variant::OBJECT, "fadein_curve", PROPERTY_HINT_RESOURCE_TYPE, Curve::get_class_static()), "set_fadein_curve", "get_fadein_curve");767ADD_PROPERTY(PropertyInfo(Variant::FLOAT, "fadeout_time", PROPERTY_HINT_RANGE, "0,60,0.01,or_greater,suffix:s"), "set_fadeout_time", "get_fadeout_time");768ADD_PROPERTY(PropertyInfo(Variant::OBJECT, "fadeout_curve", PROPERTY_HINT_RESOURCE_TYPE, Curve::get_class_static()), "set_fadeout_curve", "get_fadeout_curve");769ADD_PROPERTY(PropertyInfo(Variant::BOOL, "break_loop_at_end"), "set_break_loop_at_end", "is_loop_broken_at_end");770ADD_PROPERTY(PropertyInfo(Variant::BOOL, "abort_on_reset"), "set_abort_on_reset", "is_aborted_on_reset");771772ADD_GROUP("Auto Restart", "autorestart_");773ADD_PROPERTY(PropertyInfo(Variant::BOOL, "autorestart"), "set_autorestart", "has_autorestart");774775ADD_PROPERTY(PropertyInfo(Variant::FLOAT, "autorestart_delay", PROPERTY_HINT_RANGE, "0,60,0.01,or_greater,suffix:s"), "set_autorestart_delay", "get_autorestart_delay");776ADD_PROPERTY(PropertyInfo(Variant::FLOAT, "autorestart_random_delay", PROPERTY_HINT_RANGE, "0,60,0.01,or_greater,suffix:s"), "set_autorestart_random_delay", "get_autorestart_random_delay");777778BIND_ENUM_CONSTANT(ONE_SHOT_REQUEST_NONE);779BIND_ENUM_CONSTANT(ONE_SHOT_REQUEST_FIRE);780BIND_ENUM_CONSTANT(ONE_SHOT_REQUEST_ABORT);781BIND_ENUM_CONSTANT(ONE_SHOT_REQUEST_FADE_OUT);782783BIND_ENUM_CONSTANT(MIX_MODE_BLEND);784BIND_ENUM_CONSTANT(MIX_MODE_ADD);785}786787AnimationNodeOneShot::AnimationNodeOneShot() {788add_input("in");789add_input("shot");790}791792////////////////////////////////////////////////793794void AnimationNodeAdd2::get_parameter_list(List<PropertyInfo> *r_list) const {795AnimationNode::get_parameter_list(r_list);796r_list->push_back(PropertyInfo(Variant::FLOAT, add_amount, PROPERTY_HINT_RANGE, "0,1,0.01,or_less,or_greater"));797}798799Variant AnimationNodeAdd2::get_parameter_default_value(const StringName &p_parameter) const {800Variant ret = AnimationNode::get_parameter_default_value(p_parameter);801if (ret != Variant()) {802return ret;803}804805return 0.0;806}807808String AnimationNodeAdd2::get_caption() const {809return "Add2";810}811812bool AnimationNodeAdd2::has_filter() const {813return true;814}815816AnimationNode::NodeTimeInfo AnimationNodeAdd2::_process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only) {817double amount = get_parameter(add_amount);818819AnimationMixer::PlaybackInfo pi = p_playback_info;820pi.weight = 1.0;821NodeTimeInfo nti = blend_input(0, pi, FILTER_IGNORE, sync, p_test_only);822pi.weight = amount;823blend_input(1, pi, FILTER_PASS, sync, p_test_only);824825return nti;826}827828AnimationNodeAdd2::AnimationNodeAdd2() {829add_input("in");830add_input("add");831}832833////////////////////////////////////////////////834835void AnimationNodeAdd3::get_parameter_list(List<PropertyInfo> *r_list) const {836AnimationNode::get_parameter_list(r_list);837r_list->push_back(PropertyInfo(Variant::FLOAT, add_amount, PROPERTY_HINT_RANGE, "-1,1,0.01,or_less,or_greater"));838}839840Variant AnimationNodeAdd3::get_parameter_default_value(const StringName &p_parameter) const {841Variant ret = AnimationNode::get_parameter_default_value(p_parameter);842if (ret != Variant()) {843return ret;844}845846return 0.0;847}848849String AnimationNodeAdd3::get_caption() const {850return "Add3";851}852853bool AnimationNodeAdd3::has_filter() const {854return true;855}856857AnimationNode::NodeTimeInfo AnimationNodeAdd3::_process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only) {858double amount = get_parameter(add_amount);859860AnimationMixer::PlaybackInfo pi = p_playback_info;861pi.weight = MAX(0, -amount);862blend_input(0, pi, FILTER_PASS, sync, p_test_only);863pi.weight = 1.0;864NodeTimeInfo nti = blend_input(1, pi, FILTER_IGNORE, sync, p_test_only);865pi.weight = MAX(0, amount);866blend_input(2, pi, FILTER_PASS, sync, p_test_only);867868return nti;869}870871AnimationNodeAdd3::AnimationNodeAdd3() {872add_input("-add");873add_input("in");874add_input("+add");875}876877/////////////////////////////////////////////878879void AnimationNodeBlend2::get_parameter_list(List<PropertyInfo> *r_list) const {880AnimationNode::get_parameter_list(r_list);881r_list->push_back(PropertyInfo(Variant::FLOAT, blend_amount, PROPERTY_HINT_RANGE, "0,1,0.01,or_less,or_greater"));882}883884Variant AnimationNodeBlend2::get_parameter_default_value(const StringName &p_parameter) const {885Variant ret = AnimationNode::get_parameter_default_value(p_parameter);886if (ret != Variant()) {887return ret;888}889890return 0.0; // For blend amount.891}892893String AnimationNodeBlend2::get_caption() const {894return "Blend2";895}896897AnimationNode::NodeTimeInfo AnimationNodeBlend2::_process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only) {898double amount = get_parameter(blend_amount);899900AnimationMixer::PlaybackInfo pi = p_playback_info;901pi.weight = 1.0 - amount;902NodeTimeInfo nti0 = blend_input(0, pi, FILTER_BLEND, sync, p_test_only);903pi.weight = amount;904NodeTimeInfo nti1 = blend_input(1, pi, FILTER_PASS, sync, p_test_only);905906return amount > 0.5 ? nti1 : nti0; // Hacky but good enough.907}908909bool AnimationNodeBlend2::has_filter() const {910return true;911}912913AnimationNodeBlend2::AnimationNodeBlend2() {914add_input("in");915add_input("blend");916}917918//////////////////////////////////////919920void AnimationNodeBlend3::get_parameter_list(List<PropertyInfo> *r_list) const {921AnimationNode::get_parameter_list(r_list);922r_list->push_back(PropertyInfo(Variant::FLOAT, blend_amount, PROPERTY_HINT_RANGE, "-1,1,0.01,or_less,or_greater"));923}924925Variant AnimationNodeBlend3::get_parameter_default_value(const StringName &p_parameter) const {926Variant ret = AnimationNode::get_parameter_default_value(p_parameter);927if (ret != Variant()) {928return ret;929}930931return 0.0; // For blend amount.932}933934String AnimationNodeBlend3::get_caption() const {935return "Blend3";936}937938AnimationNode::NodeTimeInfo AnimationNodeBlend3::_process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only) {939double amount = get_parameter(blend_amount);940941AnimationMixer::PlaybackInfo pi = p_playback_info;942pi.weight = MAX(0, -amount);943NodeTimeInfo nti0 = blend_input(0, pi, FILTER_IGNORE, sync, p_test_only);944pi.weight = 1.0 - Math::abs(amount);945NodeTimeInfo nti1 = blend_input(1, pi, FILTER_IGNORE, sync, p_test_only);946pi.weight = MAX(0, amount);947NodeTimeInfo nti2 = blend_input(2, pi, FILTER_IGNORE, sync, p_test_only);948949return amount > 0.5 ? nti2 : (amount < -0.5 ? nti0 : nti1); // Hacky but good enough.950}951952AnimationNodeBlend3::AnimationNodeBlend3() {953add_input("-blend");954add_input("in");955add_input("+blend");956}957958////////////////////////////////////////////////959960void AnimationNodeSub2::get_parameter_list(List<PropertyInfo> *r_list) const {961AnimationNode::get_parameter_list(r_list);962r_list->push_back(PropertyInfo(Variant::FLOAT, sub_amount, PROPERTY_HINT_RANGE, "0,1,0.01,or_less,or_greater"));963}964965Variant AnimationNodeSub2::get_parameter_default_value(const StringName &p_parameter) const {966Variant ret = AnimationNode::get_parameter_default_value(p_parameter);967if (ret != Variant()) {968return ret;969}970971return 0.0;972}973974String AnimationNodeSub2::get_caption() const {975return "Sub2";976}977978bool AnimationNodeSub2::has_filter() const {979return true;980}981982AnimationNode::NodeTimeInfo AnimationNodeSub2::_process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only) {983double amount = get_parameter(sub_amount);984985AnimationMixer::PlaybackInfo pi = p_playback_info;986// Out = Sub.Transform3D^(-1) * In.Transform3D987pi.weight = -amount;988blend_input(1, pi, FILTER_PASS, sync, p_test_only);989pi.weight = 1.0;990991return blend_input(0, pi, FILTER_IGNORE, sync, p_test_only);992}993994AnimationNodeSub2::AnimationNodeSub2() {995add_input("in");996add_input("sub");997}998999/////////////////////////////////10001001void AnimationNodeTimeScale::get_parameter_list(List<PropertyInfo> *r_list) const {1002AnimationNode::get_parameter_list(r_list);1003r_list->push_back(PropertyInfo(Variant::FLOAT, scale, PROPERTY_HINT_RANGE, "-32,32,0.01,or_less,or_greater"));1004}10051006Variant AnimationNodeTimeScale::get_parameter_default_value(const StringName &p_parameter) const {1007Variant ret = AnimationNode::get_parameter_default_value(p_parameter);1008if (ret != Variant()) {1009return ret;1010}10111012return 1.0; // Initial timescale.1013}10141015String AnimationNodeTimeScale::get_caption() const {1016return "TimeScale";1017}10181019AnimationNode::NodeTimeInfo AnimationNodeTimeScale::_process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only) {1020double cur_scale = get_parameter(scale);10211022AnimationMixer::PlaybackInfo pi = p_playback_info;1023pi.weight = 1.0;1024if (!pi.seeked) {1025pi.delta *= cur_scale;1026}10271028return blend_input(0, pi, FILTER_IGNORE, true, p_test_only);1029}10301031AnimationNodeTimeScale::AnimationNodeTimeScale() {1032add_input("in");1033}10341035////////////////////////////////////10361037void AnimationNodeTimeSeek::get_parameter_list(List<PropertyInfo> *r_list) const {1038AnimationNode::get_parameter_list(r_list);1039r_list->push_back(PropertyInfo(Variant::FLOAT, seek_pos_request, PROPERTY_HINT_RANGE, "-1,3600,0.01,or_greater")); // It will be reset to -1 after seeking the position immediately.1040}10411042Variant AnimationNodeTimeSeek::get_parameter_default_value(const StringName &p_parameter) const {1043Variant ret = AnimationNode::get_parameter_default_value(p_parameter);1044if (ret != Variant()) {1045return ret;1046}10471048return -1.0; // Initial seek request.1049}10501051String AnimationNodeTimeSeek::get_caption() const {1052return "TimeSeek";1053}10541055void AnimationNodeTimeSeek::set_explicit_elapse(bool p_enable) {1056explicit_elapse = p_enable;1057}10581059bool AnimationNodeTimeSeek::is_explicit_elapse() const {1060return explicit_elapse;1061}10621063AnimationNode::NodeTimeInfo AnimationNodeTimeSeek::_process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only) {1064double cur_seek_pos = get_parameter(seek_pos_request);10651066AnimationMixer::PlaybackInfo pi = p_playback_info;1067pi.weight = 1.0;1068if (Animation::is_greater_or_equal_approx(cur_seek_pos, 0)) {1069pi.time = cur_seek_pos;1070pi.seeked = true;1071pi.is_external_seeking = explicit_elapse;1072set_parameter(seek_pos_request, -1.0); // Reset.1073}10741075return blend_input(0, pi, FILTER_IGNORE, true, p_test_only);1076}10771078AnimationNodeTimeSeek::AnimationNodeTimeSeek() {1079add_input("in");1080}10811082void AnimationNodeTimeSeek::_bind_methods() {1083ClassDB::bind_method(D_METHOD("set_explicit_elapse", "enable"), &AnimationNodeTimeSeek::set_explicit_elapse);1084ClassDB::bind_method(D_METHOD("is_explicit_elapse"), &AnimationNodeTimeSeek::is_explicit_elapse);1085ADD_PROPERTY(PropertyInfo(Variant::BOOL, "explicit_elapse"), "set_explicit_elapse", "is_explicit_elapse");1086}10871088/////////////////////////////////////////////////10891090bool AnimationNodeTransition::_set(const StringName &p_path, const Variant &p_value) {1091String path = p_path;10921093if (!path.begins_with("input_")) {1094return false;1095}10961097int which = path.get_slicec('/', 0).get_slicec('_', 1).to_int();1098String what = path.get_slicec('/', 1);10991100if (which == get_input_count() && what == "name") {1101if (add_input(p_value)) {1102return true;1103}1104return false;1105}11061107ERR_FAIL_INDEX_V(which, get_input_count(), false);11081109if (what == "name") {1110set_input_name(which, p_value);1111} else if (what == "auto_advance") {1112set_input_as_auto_advance(which, p_value);1113} else if (what == "break_loop_at_end") {1114set_input_break_loop_at_end(which, p_value);1115} else if (what == "reset") {1116set_input_reset(which, p_value);1117} else {1118return false;1119}11201121return true;1122}11231124bool AnimationNodeTransition::_get(const StringName &p_path, Variant &r_ret) const {1125String path = p_path;11261127if (!path.begins_with("input_")) {1128return false;1129}11301131int which = path.get_slicec('/', 0).get_slicec('_', 1).to_int();1132String what = path.get_slicec('/', 1);11331134ERR_FAIL_INDEX_V(which, get_input_count(), false);11351136if (what == "name") {1137r_ret = get_input_name(which);1138} else if (what == "auto_advance") {1139r_ret = is_input_set_as_auto_advance(which);1140} else if (what == "break_loop_at_end") {1141r_ret = is_input_loop_broken_at_end(which);1142} else if (what == "reset") {1143r_ret = is_input_reset(which);1144} else {1145return false;1146}11471148return true;1149}11501151void AnimationNodeTransition::get_parameter_list(List<PropertyInfo> *r_list) const {1152AnimationNode::get_parameter_list(r_list);1153String anims;1154for (int i = 0; i < get_input_count(); i++) {1155if (i > 0) {1156anims += ",";1157}1158anims += inputs[i].name;1159}11601161r_list->push_back(PropertyInfo(Variant::STRING, current_state, PROPERTY_HINT_NONE, "", PROPERTY_USAGE_DEFAULT | PROPERTY_USAGE_READ_ONLY)); // For interface.1162r_list->push_back(PropertyInfo(Variant::STRING, transition_request, PROPERTY_HINT_ENUM, anims)); // For transition request. It will be cleared after setting the value immediately.1163r_list->push_back(PropertyInfo(Variant::INT, current_index, PROPERTY_HINT_NONE, "", PROPERTY_USAGE_STORAGE | PROPERTY_USAGE_READ_ONLY)); // To avoid finding the index every frame, use this internally.1164r_list->push_back(PropertyInfo(Variant::INT, prev_index, PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NONE));1165r_list->push_back(PropertyInfo(Variant::FLOAT, prev_xfading, PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NONE));1166}11671168Variant AnimationNodeTransition::get_parameter_default_value(const StringName &p_parameter) const {1169Variant ret = AnimationNode::get_parameter_default_value(p_parameter);1170if (ret != Variant()) {1171return ret;1172}11731174if (p_parameter == prev_xfading) {1175return 0.0;1176} else if (p_parameter == prev_index || p_parameter == current_index) {1177return (int)-1;1178} else {1179return String();1180}1181}11821183bool AnimationNodeTransition::is_parameter_read_only(const StringName &p_parameter) const {1184if (AnimationNode::is_parameter_read_only(p_parameter)) {1185return true;1186}11871188if (p_parameter == current_state || p_parameter == current_index) {1189return true;1190}1191return false;1192}11931194String AnimationNodeTransition::get_caption() const {1195return "Transition";1196}11971198void AnimationNodeTransition::set_input_count(int p_inputs) {1199for (int i = get_input_count(); i < p_inputs; i++) {1200add_input("state_" + itos(i));1201}1202while (get_input_count() > p_inputs) {1203remove_input(get_input_count() - 1);1204}12051206pending_update = true;12071208emit_signal(SNAME("tree_changed")); // For updating connect activity map.1209notify_property_list_changed();1210}12111212bool AnimationNodeTransition::add_input(const String &p_name) {1213if (AnimationNode::add_input(p_name)) {1214input_data.push_back(InputData());1215return true;1216}1217return false;1218}12191220void AnimationNodeTransition::remove_input(int p_index) {1221input_data.remove_at(p_index);1222AnimationNode::remove_input(p_index);1223}12241225bool AnimationNodeTransition::set_input_name(int p_input, const String &p_name) {1226pending_update = true;1227if (!AnimationNode::set_input_name(p_input, p_name)) {1228return false;1229}1230emit_signal(SNAME("tree_changed")); // For updating enum options.1231return true;1232}12331234void AnimationNodeTransition::set_input_as_auto_advance(int p_input, bool p_enable) {1235ERR_FAIL_INDEX(p_input, get_input_count());1236input_data[p_input].auto_advance = p_enable;1237}12381239bool AnimationNodeTransition::is_input_set_as_auto_advance(int p_input) const {1240ERR_FAIL_INDEX_V(p_input, get_input_count(), false);1241return input_data[p_input].auto_advance;1242}12431244void AnimationNodeTransition::set_input_break_loop_at_end(int p_input, bool p_enable) {1245ERR_FAIL_INDEX(p_input, get_input_count());1246input_data[p_input].break_loop_at_end = p_enable;1247}12481249bool AnimationNodeTransition::is_input_loop_broken_at_end(int p_input) const {1250ERR_FAIL_INDEX_V(p_input, get_input_count(), false);1251return input_data[p_input].break_loop_at_end;1252}12531254void AnimationNodeTransition::set_input_reset(int p_input, bool p_enable) {1255ERR_FAIL_INDEX(p_input, get_input_count());1256input_data[p_input].reset = p_enable;1257}12581259bool AnimationNodeTransition::is_input_reset(int p_input) const {1260ERR_FAIL_INDEX_V(p_input, get_input_count(), true);1261return input_data[p_input].reset;1262}12631264void AnimationNodeTransition::set_xfade_time(double p_fade) {1265xfade_time = p_fade;1266}12671268double AnimationNodeTransition::get_xfade_time() const {1269return xfade_time;1270}12711272void AnimationNodeTransition::set_xfade_curve(const Ref<Curve> &p_curve) {1273xfade_curve = p_curve;1274}12751276Ref<Curve> AnimationNodeTransition::get_xfade_curve() const {1277return xfade_curve;1278}12791280void AnimationNodeTransition::set_allow_transition_to_self(bool p_enable) {1281allow_transition_to_self = p_enable;1282}12831284bool AnimationNodeTransition::is_allow_transition_to_self() const {1285return allow_transition_to_self;1286}12871288AnimationNode::NodeTimeInfo AnimationNodeTransition::_process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only) {1289String cur_transition_request = get_parameter(transition_request);1290int cur_current_index = get_parameter(current_index);1291int cur_prev_index = get_parameter(prev_index);12921293NodeTimeInfo cur_nti = get_node_time_info();1294double cur_prev_xfading = get_parameter(prev_xfading);12951296bool switched = false;1297bool restart = false;1298bool clear_remaining_fade = false;12991300if (pending_update) {1301if (cur_current_index < 0 || cur_current_index >= get_input_count()) {1302set_parameter(prev_index, -1);1303if (get_input_count() > 0) {1304set_parameter(current_index, 0);1305set_parameter(current_state, get_input_name(0));1306} else {1307set_parameter(current_index, -1);1308set_parameter(current_state, StringName());1309}1310} else {1311set_parameter(current_state, get_input_name(cur_current_index));1312}1313pending_update = false;1314}13151316double p_time = p_playback_info.time;1317bool p_seek = p_playback_info.seeked;1318bool p_is_external_seeking = p_playback_info.is_external_seeking;13191320if (Math::is_zero_approx(p_time) && p_seek && !p_is_external_seeking) {1321clear_remaining_fade = true; // Reset occurs.1322}13231324if (!cur_transition_request.is_empty()) {1325int new_idx = find_input(cur_transition_request);1326if (new_idx >= 0) {1327if (cur_current_index == new_idx) {1328if (allow_transition_to_self) {1329// Transition to same state.1330restart = input_data[cur_current_index].reset;1331clear_remaining_fade = true;1332}1333} else {1334switched = true;1335cur_prev_index = cur_current_index;1336set_parameter(prev_index, cur_current_index);1337cur_current_index = new_idx;1338set_parameter(current_index, cur_current_index);1339set_parameter(current_state, cur_transition_request);1340}1341} else {1342ERR_PRINT("No such input: '" + cur_transition_request + "'");1343}1344cur_transition_request = String();1345set_parameter(transition_request, cur_transition_request);1346}13471348if (clear_remaining_fade) {1349cur_prev_xfading = 0;1350set_parameter(prev_xfading, 0);1351cur_prev_index = -1;1352set_parameter(prev_index, -1);1353}13541355AnimationMixer::PlaybackInfo pi = p_playback_info;13561357// Special case for restart.1358if (restart) {1359pi.time = 0;1360pi.seeked = true;1361pi.weight = 1.0;1362return blend_input(cur_current_index, pi, FILTER_IGNORE, true, p_test_only);1363}13641365if (switched) {1366cur_prev_xfading = xfade_time;1367}13681369if (cur_current_index < 0 || cur_current_index >= get_input_count() || cur_prev_index >= get_input_count()) {1370return NodeTimeInfo();1371}13721373if (sync) {1374pi.weight = 0;1375for (int i = 0; i < get_input_count(); i++) {1376if (i != cur_current_index && i != cur_prev_index) {1377blend_input(i, pi, FILTER_IGNORE, true, p_test_only);1378}1379}1380}13811382if (cur_prev_index < 0) { // Process current animation, check for transition.1383pi.weight = 1.0;1384cur_nti = blend_input(cur_current_index, pi, FILTER_IGNORE, true, p_test_only);1385if (input_data[cur_current_index].auto_advance && Animation::is_less_or_equal_approx(cur_nti.get_remain(input_data[cur_current_index].break_loop_at_end), xfade_time)) {1386set_parameter(transition_request, get_input_name((cur_current_index + 1) % get_input_count()));1387}1388} else { // Cross-fading from prev to current.13891390real_t blend = 0.0;1391real_t blend_inv = 1.0;1392bool use_blend = sync;1393if (xfade_time > 0) {1394use_blend = true;1395blend = cur_prev_xfading / xfade_time;1396if (xfade_curve.is_valid()) {1397blend = xfade_curve->sample(blend);1398}1399blend_inv = 1.0 - blend;1400blend = Math::is_zero_approx(blend) ? CMP_EPSILON : blend;1401blend_inv = Math::is_zero_approx(blend_inv) ? CMP_EPSILON : blend_inv;1402}14031404// Blend values must be more than CMP_EPSILON to process discrete keys in edge.1405pi.weight = blend_inv;1406if (input_data[cur_current_index].reset && !p_seek && switched) { // Just switched, seek to start of current.1407pi.time = 0;1408pi.seeked = true;1409}1410cur_nti = blend_input(cur_current_index, pi, FILTER_IGNORE, true, p_test_only);14111412pi = p_playback_info;1413pi.seeked &= use_blend;1414pi.weight = blend;1415blend_input(cur_prev_index, pi, FILTER_IGNORE, true, p_test_only);1416if (!p_seek) {1417if (Animation::is_less_or_equal_approx(cur_prev_xfading, 0)) {1418set_parameter(prev_index, -1);1419}1420cur_prev_xfading -= Math::abs(p_playback_info.delta);1421}1422}14231424set_parameter(prev_xfading, cur_prev_xfading);14251426return cur_nti;1427}14281429void AnimationNodeTransition::_get_property_list(List<PropertyInfo> *p_list) const {1430for (int i = 0; i < get_input_count(); i++) {1431p_list->push_back(PropertyInfo(Variant::STRING, "input_" + itos(i) + "/name", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_DEFAULT | PROPERTY_USAGE_INTERNAL));1432p_list->push_back(PropertyInfo(Variant::BOOL, "input_" + itos(i) + "/auto_advance", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_DEFAULT | PROPERTY_USAGE_INTERNAL));1433p_list->push_back(PropertyInfo(Variant::BOOL, "input_" + itos(i) + "/break_loop_at_end", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_DEFAULT | PROPERTY_USAGE_INTERNAL));1434p_list->push_back(PropertyInfo(Variant::BOOL, "input_" + itos(i) + "/reset", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_DEFAULT | PROPERTY_USAGE_INTERNAL));1435}1436}14371438void AnimationNodeTransition::_bind_methods() {1439ClassDB::bind_method(D_METHOD("set_input_count", "input_count"), &AnimationNodeTransition::set_input_count);14401441ClassDB::bind_method(D_METHOD("set_input_as_auto_advance", "input", "enable"), &AnimationNodeTransition::set_input_as_auto_advance);1442ClassDB::bind_method(D_METHOD("is_input_set_as_auto_advance", "input"), &AnimationNodeTransition::is_input_set_as_auto_advance);14431444ClassDB::bind_method(D_METHOD("set_input_break_loop_at_end", "input", "enable"), &AnimationNodeTransition::set_input_break_loop_at_end);1445ClassDB::bind_method(D_METHOD("is_input_loop_broken_at_end", "input"), &AnimationNodeTransition::is_input_loop_broken_at_end);14461447ClassDB::bind_method(D_METHOD("set_input_reset", "input", "enable"), &AnimationNodeTransition::set_input_reset);1448ClassDB::bind_method(D_METHOD("is_input_reset", "input"), &AnimationNodeTransition::is_input_reset);14491450ClassDB::bind_method(D_METHOD("set_xfade_time", "time"), &AnimationNodeTransition::set_xfade_time);1451ClassDB::bind_method(D_METHOD("get_xfade_time"), &AnimationNodeTransition::get_xfade_time);14521453ClassDB::bind_method(D_METHOD("set_xfade_curve", "curve"), &AnimationNodeTransition::set_xfade_curve);1454ClassDB::bind_method(D_METHOD("get_xfade_curve"), &AnimationNodeTransition::get_xfade_curve);14551456ClassDB::bind_method(D_METHOD("set_allow_transition_to_self", "enable"), &AnimationNodeTransition::set_allow_transition_to_self);1457ClassDB::bind_method(D_METHOD("is_allow_transition_to_self"), &AnimationNodeTransition::is_allow_transition_to_self);14581459ADD_PROPERTY(PropertyInfo(Variant::FLOAT, "xfade_time", PROPERTY_HINT_RANGE, "0,120,0.01,suffix:s"), "set_xfade_time", "get_xfade_time");1460ADD_PROPERTY(PropertyInfo(Variant::OBJECT, "xfade_curve", PROPERTY_HINT_RESOURCE_TYPE, Curve::get_class_static()), "set_xfade_curve", "get_xfade_curve");1461ADD_PROPERTY(PropertyInfo(Variant::BOOL, "allow_transition_to_self"), "set_allow_transition_to_self", "is_allow_transition_to_self");1462ADD_PROPERTY(PropertyInfo(Variant::INT, "input_count", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_EDITOR | PROPERTY_USAGE_ARRAY | PROPERTY_USAGE_UPDATE_ALL_IF_MODIFIED, "Inputs,input_"), "set_input_count", "get_input_count");1463}14641465AnimationNodeTransition::AnimationNodeTransition() {1466}14671468/////////////////////14691470String AnimationNodeOutput::get_caption() const {1471return "Output";1472}14731474AnimationNode::NodeTimeInfo AnimationNodeOutput::_process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only) {1475AnimationMixer::PlaybackInfo pi = p_playback_info;1476pi.weight = 1.0;1477return blend_input(0, pi, FILTER_IGNORE, true, p_test_only);1478}14791480AnimationNodeOutput::AnimationNodeOutput() {1481add_input("output");1482}14831484///////////////////////////////////////////////////////1485void AnimationNodeBlendTree::add_node(const StringName &p_name, Ref<AnimationNode> p_node, const Vector2 &p_position) {1486ERR_FAIL_COND(nodes.has(p_name));1487ERR_FAIL_COND(p_node.is_null());1488ERR_FAIL_COND(p_name == SceneStringName(output));1489ERR_FAIL_COND(String(p_name).contains_char('/'));14901491Node n;1492n.node = p_node;1493n.position = p_position;1494n.connections.resize(n.node->get_input_count());1495nodes[p_name] = n;14961497emit_changed();1498emit_signal(SNAME("tree_changed"));14991500p_node->connect(SNAME("tree_changed"), callable_mp(this, &AnimationNodeBlendTree::_tree_changed), CONNECT_REFERENCE_COUNTED);1501p_node->connect(SNAME("animation_node_renamed"), callable_mp(this, &AnimationNodeBlendTree::_animation_node_renamed), CONNECT_REFERENCE_COUNTED);1502p_node->connect(SNAME("animation_node_removed"), callable_mp(this, &AnimationNodeBlendTree::_animation_node_removed), CONNECT_REFERENCE_COUNTED);1503p_node->connect_changed(callable_mp(this, &AnimationNodeBlendTree::_node_changed).bind(p_name), CONNECT_REFERENCE_COUNTED);1504}15051506Ref<AnimationNode> AnimationNodeBlendTree::get_node(const StringName &p_name) const {1507ERR_FAIL_COND_V(!nodes.has(p_name), Ref<AnimationNode>());15081509return nodes[p_name].node;1510}15111512StringName AnimationNodeBlendTree::get_node_name(const Ref<AnimationNode> &p_node) const {1513for (const KeyValue<StringName, Node> &E : nodes) {1514if (E.value.node == p_node) {1515return E.key;1516}1517}15181519ERR_FAIL_V(StringName());1520}15211522void AnimationNodeBlendTree::set_node_position(const StringName &p_node, const Vector2 &p_position) {1523ERR_FAIL_COND(!nodes.has(p_node));1524nodes[p_node].position = p_position;1525}15261527Vector2 AnimationNodeBlendTree::get_node_position(const StringName &p_node) const {1528ERR_FAIL_COND_V(!nodes.has(p_node), Vector2());1529return nodes[p_node].position;1530}15311532void AnimationNodeBlendTree::get_child_nodes(List<ChildNode> *r_child_nodes) {1533Vector<StringName> ns;15341535for (const KeyValue<StringName, Node> &E : nodes) {1536ns.push_back(E.key);1537}15381539for (int i = 0; i < ns.size(); i++) {1540ChildNode cn;1541cn.name = ns[i];1542cn.node = nodes[cn.name].node;1543r_child_nodes->push_back(cn);1544}1545}15461547bool AnimationNodeBlendTree::has_node(const StringName &p_name) const {1548return nodes.has(p_name);1549}15501551Vector<StringName> AnimationNodeBlendTree::get_node_connection_array(const StringName &p_name) const {1552ERR_FAIL_COND_V(!nodes.has(p_name), Vector<StringName>());1553return nodes[p_name].connections;1554}15551556void AnimationNodeBlendTree::remove_node(const StringName &p_name) {1557ERR_FAIL_COND(!nodes.has(p_name));1558ERR_FAIL_COND(p_name == SceneStringName(output)); //can't delete output15591560{1561Ref<AnimationNode> node = nodes[p_name].node;1562node->disconnect(SNAME("tree_changed"), callable_mp(this, &AnimationNodeBlendTree::_tree_changed));1563node->disconnect(SNAME("animation_node_renamed"), callable_mp(this, &AnimationNodeBlendTree::_animation_node_renamed));1564node->disconnect(SNAME("animation_node_removed"), callable_mp(this, &AnimationNodeBlendTree::_animation_node_removed));1565node->disconnect_changed(callable_mp(this, &AnimationNodeBlendTree::_node_changed));1566}15671568nodes.erase(p_name);15691570// Erase connections to name.1571for (KeyValue<StringName, Node> &E : nodes) {1572for (int i = 0; i < E.value.connections.size(); i++) {1573if (E.value.connections[i] == p_name) {1574E.value.connections.write[i] = StringName();1575}1576}1577}15781579emit_signal(SNAME("animation_node_removed"), get_instance_id(), p_name);1580emit_changed();1581emit_signal(SNAME("tree_changed"));1582}15831584void AnimationNodeBlendTree::rename_node(const StringName &p_name, const StringName &p_new_name) {1585ERR_FAIL_COND(!nodes.has(p_name));1586ERR_FAIL_COND(nodes.has(p_new_name));1587ERR_FAIL_COND(p_name == SceneStringName(output));1588ERR_FAIL_COND(p_new_name == SceneStringName(output));15891590nodes[p_name].node->disconnect_changed(callable_mp(this, &AnimationNodeBlendTree::_node_changed));15911592const Node temp_copy = nodes[p_name];1593nodes[p_new_name] = temp_copy; // might realloc1594nodes.erase(p_name);15951596// Rename connections.1597for (KeyValue<StringName, Node> &E : nodes) {1598for (int i = 0; i < E.value.connections.size(); i++) {1599if (E.value.connections[i] == p_name) {1600E.value.connections.write[i] = p_new_name;1601}1602}1603}1604// Connection must be done with new name.1605nodes[p_new_name].node->connect_changed(callable_mp(this, &AnimationNodeBlendTree::_node_changed).bind(p_new_name), CONNECT_REFERENCE_COUNTED);16061607emit_signal(SNAME("animation_node_renamed"), get_instance_id(), p_name, p_new_name);1608emit_signal(SNAME("tree_changed"));1609}16101611void AnimationNodeBlendTree::connect_node(const StringName &p_input_node, int p_input_index, const StringName &p_output_node) {1612ERR_FAIL_COND(!nodes.has(p_output_node));1613ERR_FAIL_COND(!nodes.has(p_input_node));1614ERR_FAIL_COND(p_output_node == SceneStringName(output));1615ERR_FAIL_COND(p_input_node == p_output_node);16161617Ref<AnimationNode> input = nodes[p_input_node].node;1618ERR_FAIL_INDEX(p_input_index, nodes[p_input_node].connections.size());16191620for (KeyValue<StringName, Node> &E : nodes) {1621for (int i = 0; i < E.value.connections.size(); i++) {1622StringName output = E.value.connections[i];1623ERR_FAIL_COND(output == p_output_node);1624}1625}16261627nodes[p_input_node].connections.write[p_input_index] = p_output_node;16281629emit_changed();1630}16311632void AnimationNodeBlendTree::disconnect_node(const StringName &p_node, int p_input_index) {1633ERR_FAIL_COND(!nodes.has(p_node));16341635Ref<AnimationNode> input = nodes[p_node].node;1636ERR_FAIL_INDEX(p_input_index, nodes[p_node].connections.size());16371638nodes[p_node].connections.write[p_input_index] = StringName();1639}16401641AnimationNodeBlendTree::ConnectionError AnimationNodeBlendTree::can_connect_node(const StringName &p_input_node, int p_input_index, const StringName &p_output_node) const {1642if (!nodes.has(p_output_node) || p_output_node == SceneStringName(output)) {1643return CONNECTION_ERROR_NO_OUTPUT;1644}16451646if (!nodes.has(p_input_node)) {1647return CONNECTION_ERROR_NO_INPUT;1648}16491650if (p_input_node == p_output_node) {1651return CONNECTION_ERROR_SAME_NODE;1652}16531654Ref<AnimationNode> input = nodes[p_input_node].node;16551656if (p_input_index < 0 || p_input_index >= nodes[p_input_node].connections.size()) {1657return CONNECTION_ERROR_NO_INPUT_INDEX;1658}16591660if (nodes[p_input_node].connections[p_input_index] != StringName()) {1661return CONNECTION_ERROR_CONNECTION_EXISTS;1662}16631664for (const KeyValue<StringName, Node> &E : nodes) {1665for (int i = 0; i < E.value.connections.size(); i++) {1666const StringName output = E.value.connections[i];1667if (output == p_output_node) {1668return CONNECTION_ERROR_CONNECTION_EXISTS;1669}1670}1671}1672return CONNECTION_OK;1673}16741675void AnimationNodeBlendTree::get_node_connections(List<NodeConnection> *r_connections) const {1676for (const KeyValue<StringName, Node> &E : nodes) {1677for (int i = 0; i < E.value.connections.size(); i++) {1678const StringName output = E.value.connections[i];1679if (output != StringName()) {1680NodeConnection nc;1681nc.input_node = E.key;1682nc.input_index = i;1683nc.output_node = output;1684r_connections->push_back(nc);1685}1686}1687}1688}16891690String AnimationNodeBlendTree::get_caption() const {1691return "BlendTree";1692}16931694AnimationNode::NodeTimeInfo AnimationNodeBlendTree::_process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only) {1695Ref<AnimationNodeOutput> output = nodes[SceneStringName(output)].node;1696node_state.connections = nodes[SceneStringName(output)].connections;1697ERR_FAIL_COND_V(output.is_null(), NodeTimeInfo());16981699AnimationMixer::PlaybackInfo pi = p_playback_info;1700pi.weight = 1.0;17011702return _blend_node(output, "output", this, pi, FILTER_IGNORE, true, p_test_only, nullptr);1703}17041705LocalVector<StringName> AnimationNodeBlendTree::get_node_list() const {1706LocalVector<StringName> list;1707list.reserve(nodes.size());1708for (const KeyValue<StringName, Node> &E : nodes) {1709list.push_back(E.key);1710}1711list.sort_custom<StringName::AlphCompare>();1712return list;1713}17141715TypedArray<StringName> AnimationNodeBlendTree::get_node_list_as_typed_array() const {1716TypedArray<StringName> typed_arr;1717LocalVector<StringName> vec = get_node_list();1718typed_arr.resize(vec.size());1719for (uint32_t i = 0; i < vec.size(); i++) {1720typed_arr[i] = vec[i];1721}1722return typed_arr;1723}17241725void AnimationNodeBlendTree::set_graph_offset(const Vector2 &p_graph_offset) {1726graph_offset = p_graph_offset;1727}17281729Vector2 AnimationNodeBlendTree::get_graph_offset() const {1730return graph_offset;1731}17321733Ref<AnimationNode> AnimationNodeBlendTree::get_child_by_name(const StringName &p_name) const {1734return get_node(p_name);1735}17361737bool AnimationNodeBlendTree::_set(const StringName &p_name, const Variant &p_value) {1738String prop_name = p_name;1739if (prop_name.begins_with("nodes/")) {1740String node_name = prop_name.get_slicec('/', 1);1741String what = prop_name.get_slicec('/', 2);17421743if (what == "node") {1744Ref<AnimationNode> anode = p_value;1745if (anode.is_valid()) {1746add_node(node_name, p_value);1747}1748return true;1749}17501751if (what == "position") {1752if (nodes.has(node_name)) {1753nodes[node_name].position = p_value;1754}1755return true;1756}1757} else if (prop_name == "node_connections") {1758Array conns = p_value;1759ERR_FAIL_COND_V(conns.size() % 3 != 0, false);17601761for (int i = 0; i < conns.size(); i += 3) {1762connect_node(conns[i], conns[i + 1], conns[i + 2]);1763}1764return true;1765}17661767return false;1768}17691770bool AnimationNodeBlendTree::_get(const StringName &p_name, Variant &r_ret) const {1771String prop_name = p_name;1772if (prop_name.begins_with("nodes/")) {1773String node_name = prop_name.get_slicec('/', 1);1774String what = prop_name.get_slicec('/', 2);17751776if (what == "node") {1777if (nodes.has(node_name)) {1778r_ret = nodes[node_name].node;1779return true;1780}1781}17821783if (what == "position") {1784if (nodes.has(node_name)) {1785r_ret = nodes[node_name].position;1786return true;1787}1788}1789} else if (prop_name == "node_connections") {1790List<NodeConnection> nc;1791get_node_connections(&nc);1792Array conns;1793conns.resize(nc.size() * 3);17941795int idx = 0;1796for (const NodeConnection &E : nc) {1797conns[idx * 3 + 0] = E.input_node;1798conns[idx * 3 + 1] = E.input_index;1799conns[idx * 3 + 2] = E.output_node;1800idx++;1801}18021803r_ret = conns;1804return true;1805}18061807return false;1808}18091810void AnimationNodeBlendTree::_get_property_list(List<PropertyInfo> *p_list) const {1811List<StringName> names;1812for (const KeyValue<StringName, Node> &E : nodes) {1813names.push_back(E.key);1814}18151816for (const StringName &E : names) {1817String prop_name = E;1818if (prop_name != "output") {1819p_list->push_back(PropertyInfo(Variant::OBJECT, "nodes/" + prop_name + "/node", PROPERTY_HINT_RESOURCE_TYPE, AnimationNode::get_class_static(), PROPERTY_USAGE_NO_EDITOR));1820}1821p_list->push_back(PropertyInfo(Variant::VECTOR2, "nodes/" + prop_name + "/position", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NO_EDITOR));1822}18231824p_list->push_back(PropertyInfo(Variant::ARRAY, "node_connections", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NO_EDITOR));1825}18261827void AnimationNodeBlendTree::_tree_changed() {1828AnimationRootNode::_tree_changed();1829}18301831void AnimationNodeBlendTree::_animation_node_renamed(const ObjectID &p_oid, const String &p_old_name, const String &p_new_name) {1832AnimationRootNode::_animation_node_renamed(p_oid, p_old_name, p_new_name);1833}18341835void AnimationNodeBlendTree::_animation_node_removed(const ObjectID &p_oid, const StringName &p_node) {1836AnimationRootNode::_animation_node_removed(p_oid, p_node);1837}18381839void AnimationNodeBlendTree::reset_state() {1840graph_offset = Vector2();1841nodes.clear();1842_initialize_node_tree();1843emit_changed();1844emit_signal(SNAME("tree_changed"));1845}18461847void AnimationNodeBlendTree::_node_changed(const StringName &p_node) {1848ERR_FAIL_COND(!nodes.has(p_node));1849nodes[p_node].connections.resize(nodes[p_node].node->get_input_count());1850emit_signal(SNAME("node_changed"), p_node);1851}18521853#ifdef TOOLS_ENABLED1854void AnimationNodeBlendTree::get_argument_options(const StringName &p_function, int p_idx, List<String> *r_options) const {1855const String pf = p_function;1856bool add_node_options = false;1857if (p_idx == 0) {1858add_node_options = (pf == "get_node" || pf == "has_node" || pf == "rename_node" || pf == "remove_node" || pf == "set_node_position" || pf == "get_node_position" || pf == "connect_node" || pf == "disconnect_node");1859} else if (p_idx == 2) {1860add_node_options = (pf == "connect_node" || pf == "disconnect_node");1861}1862if (add_node_options) {1863for (const KeyValue<StringName, Node> &E : nodes) {1864r_options->push_back(String(E.key).quote());1865}1866}1867AnimationRootNode::get_argument_options(p_function, p_idx, r_options);1868}1869#endif18701871void AnimationNodeBlendTree::_bind_methods() {1872ClassDB::bind_method(D_METHOD("add_node", "name", "node", "position"), &AnimationNodeBlendTree::add_node, DEFVAL(Vector2()));1873ClassDB::bind_method(D_METHOD("get_node", "name"), &AnimationNodeBlendTree::get_node);1874ClassDB::bind_method(D_METHOD("remove_node", "name"), &AnimationNodeBlendTree::remove_node);1875ClassDB::bind_method(D_METHOD("rename_node", "name", "new_name"), &AnimationNodeBlendTree::rename_node);1876ClassDB::bind_method(D_METHOD("has_node", "name"), &AnimationNodeBlendTree::has_node);1877ClassDB::bind_method(D_METHOD("connect_node", "input_node", "input_index", "output_node"), &AnimationNodeBlendTree::connect_node);1878ClassDB::bind_method(D_METHOD("disconnect_node", "input_node", "input_index"), &AnimationNodeBlendTree::disconnect_node);1879ClassDB::bind_method(D_METHOD("get_node_list"), &AnimationNodeBlendTree::get_node_list_as_typed_array);18801881ClassDB::bind_method(D_METHOD("set_node_position", "name", "position"), &AnimationNodeBlendTree::set_node_position);1882ClassDB::bind_method(D_METHOD("get_node_position", "name"), &AnimationNodeBlendTree::get_node_position);18831884ClassDB::bind_method(D_METHOD("set_graph_offset", "offset"), &AnimationNodeBlendTree::set_graph_offset);1885ClassDB::bind_method(D_METHOD("get_graph_offset"), &AnimationNodeBlendTree::get_graph_offset);18861887ADD_PROPERTY(PropertyInfo(Variant::VECTOR2, "graph_offset", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NO_EDITOR), "set_graph_offset", "get_graph_offset");18881889BIND_CONSTANT(CONNECTION_OK);1890BIND_CONSTANT(CONNECTION_ERROR_NO_INPUT);1891BIND_CONSTANT(CONNECTION_ERROR_NO_INPUT_INDEX);1892BIND_CONSTANT(CONNECTION_ERROR_NO_OUTPUT);1893BIND_CONSTANT(CONNECTION_ERROR_SAME_NODE);1894BIND_CONSTANT(CONNECTION_ERROR_CONNECTION_EXISTS);18951896ADD_SIGNAL(MethodInfo(SNAME("node_changed"), PropertyInfo(Variant::STRING_NAME, "node_name")));1897}18981899void AnimationNodeBlendTree::_initialize_node_tree() {1900Ref<AnimationNodeOutput> output;1901output.instantiate();1902Node n;1903n.node = output;1904n.position = Vector2(300, 150);1905n.connections.resize(1);1906nodes["output"] = n;1907}19081909AnimationNodeBlendTree::AnimationNodeBlendTree() {1910_initialize_node_tree();1911}19121913AnimationNodeBlendTree::~AnimationNodeBlendTree() {1914}191519161917