Path: blob/master/scene/animation/animation_blend_tree.h
9896 views
/**************************************************************************/1/* animation_blend_tree.h */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#pragma once3132#include "scene/animation/animation_tree.h"3334class AnimationNodeAnimation : public AnimationRootNode {35GDCLASS(AnimationNodeAnimation, AnimationRootNode);3637StringName backward = "backward"; // Only used by pingpong animation.3839StringName animation;4041bool advance_on_start = false;4243bool use_custom_timeline = false;44double timeline_length = 1.0;45Animation::LoopMode loop_mode = Animation::LOOP_NONE;46bool stretch_time_scale = true;47double start_offset = 0.0;4849uint64_t last_version = 0;50bool skip = false;5152public:53enum PlayMode {54PLAY_MODE_FORWARD,55PLAY_MODE_BACKWARD56};5758void get_parameter_list(List<PropertyInfo> *r_list) const override;59virtual Variant get_parameter_default_value(const StringName &p_parameter) const override;6061virtual NodeTimeInfo get_node_time_info() const override; // Wrapper of get_parameter().6263static Vector<String> (*get_editable_animation_list)();6465virtual String get_caption() const override;66virtual NodeTimeInfo process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only = false) override;67virtual NodeTimeInfo _process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only = false) override;6869void set_animation(const StringName &p_name);70StringName get_animation() const;7172void set_play_mode(PlayMode p_play_mode);73PlayMode get_play_mode() const;7475void set_backward(bool p_backward);76bool is_backward() const;7778void set_advance_on_start(bool p_advance_on_start);79bool is_advance_on_start() const;8081void set_use_custom_timeline(bool p_use_custom_timeline);82bool is_using_custom_timeline() const;8384void set_timeline_length(double p_length);85double get_timeline_length() const;8687void set_stretch_time_scale(bool p_stretch_time_scale);88bool is_stretching_time_scale() const;8990void set_start_offset(double p_offset);91double get_start_offset() const;9293void set_loop_mode(Animation::LoopMode p_loop_mode);94Animation::LoopMode get_loop_mode() const;9596AnimationNodeAnimation();9798protected:99void _validate_property(PropertyInfo &p_property) const;100static void _bind_methods();101102private:103PlayMode play_mode = PLAY_MODE_FORWARD;104};105106VARIANT_ENUM_CAST(AnimationNodeAnimation::PlayMode)107108class AnimationNodeSync : public AnimationNode {109GDCLASS(AnimationNodeSync, AnimationNode);110111protected:112bool sync = false;113114static void _bind_methods();115116public:117void set_use_sync(bool p_sync);118bool is_using_sync() const;119120AnimationNodeSync();121};122123class AnimationNodeOneShot : public AnimationNodeSync {124GDCLASS(AnimationNodeOneShot, AnimationNodeSync);125126public:127enum OneShotRequest {128ONE_SHOT_REQUEST_NONE,129ONE_SHOT_REQUEST_FIRE,130ONE_SHOT_REQUEST_ABORT,131ONE_SHOT_REQUEST_FADE_OUT,132};133134enum MixMode {135MIX_MODE_BLEND,136MIX_MODE_ADD137};138139private:140double fade_in = 0.0;141Ref<Curve> fade_in_curve;142double fade_out = 0.0;143Ref<Curve> fade_out_curve;144145bool auto_restart = false;146double auto_restart_delay = 1.0;147double auto_restart_random_delay = 0.0;148MixMode mix = MIX_MODE_BLEND;149bool break_loop_at_end = false;150151StringName request = PNAME("request");152StringName active = PNAME("active");153StringName internal_active = PNAME("internal_active");154StringName fade_in_remaining = "fade_in_remaining";155StringName fade_out_remaining = "fade_out_remaining";156StringName time_to_restart = "time_to_restart";157158protected:159static void _bind_methods();160161public:162virtual void get_parameter_list(List<PropertyInfo> *r_list) const override;163virtual Variant get_parameter_default_value(const StringName &p_parameter) const override;164virtual bool is_parameter_read_only(const StringName &p_parameter) const override;165166virtual String get_caption() const override;167168void set_fade_in_time(double p_time);169double get_fade_in_time() const;170171void set_fade_in_curve(const Ref<Curve> &p_curve);172Ref<Curve> get_fade_in_curve() const;173174void set_fade_out_time(double p_time);175double get_fade_out_time() const;176177void set_fade_out_curve(const Ref<Curve> &p_curve);178Ref<Curve> get_fade_out_curve() const;179180void set_auto_restart_enabled(bool p_enabled);181void set_auto_restart_delay(double p_time);182void set_auto_restart_random_delay(double p_time);183184bool is_auto_restart_enabled() const;185double get_auto_restart_delay() const;186double get_auto_restart_random_delay() const;187188void set_mix_mode(MixMode p_mix);189MixMode get_mix_mode() const;190191void set_break_loop_at_end(bool p_enable);192bool is_loop_broken_at_end() const;193194virtual bool has_filter() const override;195virtual NodeTimeInfo _process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only = false) override;196197AnimationNodeOneShot();198};199200VARIANT_ENUM_CAST(AnimationNodeOneShot::OneShotRequest)201VARIANT_ENUM_CAST(AnimationNodeOneShot::MixMode)202203class AnimationNodeAdd2 : public AnimationNodeSync {204GDCLASS(AnimationNodeAdd2, AnimationNodeSync);205206StringName add_amount = PNAME("add_amount");207208public:209void get_parameter_list(List<PropertyInfo> *r_list) const override;210virtual Variant get_parameter_default_value(const StringName &p_parameter) const override;211212virtual String get_caption() const override;213214virtual bool has_filter() const override;215virtual NodeTimeInfo _process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only = false) override;216217AnimationNodeAdd2();218};219220class AnimationNodeAdd3 : public AnimationNodeSync {221GDCLASS(AnimationNodeAdd3, AnimationNodeSync);222223StringName add_amount = PNAME("add_amount");224225public:226void get_parameter_list(List<PropertyInfo> *r_list) const override;227virtual Variant get_parameter_default_value(const StringName &p_parameter) const override;228229virtual String get_caption() const override;230231virtual bool has_filter() const override;232virtual NodeTimeInfo _process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only = false) override;233234AnimationNodeAdd3();235};236237class AnimationNodeBlend2 : public AnimationNodeSync {238GDCLASS(AnimationNodeBlend2, AnimationNodeSync);239240StringName blend_amount = PNAME("blend_amount");241242public:243virtual void get_parameter_list(List<PropertyInfo> *r_list) const override;244virtual Variant get_parameter_default_value(const StringName &p_parameter) const override;245246virtual String get_caption() const override;247virtual NodeTimeInfo _process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only = false) override;248249virtual bool has_filter() const override;250AnimationNodeBlend2();251};252253class AnimationNodeBlend3 : public AnimationNodeSync {254GDCLASS(AnimationNodeBlend3, AnimationNodeSync);255256StringName blend_amount = PNAME("blend_amount");257258public:259virtual void get_parameter_list(List<PropertyInfo> *r_list) const override;260virtual Variant get_parameter_default_value(const StringName &p_parameter) const override;261262virtual String get_caption() const override;263264virtual NodeTimeInfo _process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only = false) override;265AnimationNodeBlend3();266};267268class AnimationNodeSub2 : public AnimationNodeSync {269GDCLASS(AnimationNodeSub2, AnimationNodeSync);270271StringName sub_amount = PNAME("sub_amount");272273public:274void get_parameter_list(List<PropertyInfo> *r_list) const override;275virtual Variant get_parameter_default_value(const StringName &p_parameter) const override;276277virtual String get_caption() const override;278279virtual bool has_filter() const override;280virtual NodeTimeInfo _process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only = false) override;281282AnimationNodeSub2();283};284285class AnimationNodeTimeScale : public AnimationNode {286GDCLASS(AnimationNodeTimeScale, AnimationNode);287288StringName scale = PNAME("scale");289290public:291virtual void get_parameter_list(List<PropertyInfo> *r_list) const override;292virtual Variant get_parameter_default_value(const StringName &p_parameter) const override;293294virtual String get_caption() const override;295296virtual NodeTimeInfo _process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only = false) override;297298AnimationNodeTimeScale();299};300301class AnimationNodeTimeSeek : public AnimationNode {302GDCLASS(AnimationNodeTimeSeek, AnimationNode);303304StringName seek_pos_request = PNAME("seek_request");305bool explicit_elapse = true;306307protected:308static void _bind_methods();309310public:311virtual void get_parameter_list(List<PropertyInfo> *r_list) const override;312virtual Variant get_parameter_default_value(const StringName &p_parameter) const override;313314virtual String get_caption() const override;315316virtual NodeTimeInfo _process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only = false) override;317318void set_explicit_elapse(bool p_enable);319bool is_explicit_elapse() const;320321AnimationNodeTimeSeek();322};323324class AnimationNodeTransition : public AnimationNodeSync {325GDCLASS(AnimationNodeTransition, AnimationNodeSync);326327struct InputData {328bool auto_advance = false;329bool break_loop_at_end = false;330bool reset = true;331};332LocalVector<InputData> input_data;333334StringName prev_xfading = "prev_xfading";335StringName prev_index = "prev_index";336StringName current_index = PNAME("current_index");337StringName current_state = PNAME("current_state");338StringName transition_request = PNAME("transition_request");339340StringName prev_frame_current = "pf_current";341StringName prev_frame_current_idx = "pf_current_idx";342343double xfade_time = 0.0;344Ref<Curve> xfade_curve;345bool allow_transition_to_self = false;346347bool pending_update = false;348349protected:350bool _get(const StringName &p_path, Variant &r_ret) const;351bool _set(const StringName &p_path, const Variant &p_value);352static void _bind_methods();353void _get_property_list(List<PropertyInfo> *p_list) const;354355public:356virtual void get_parameter_list(List<PropertyInfo> *r_list) const override;357virtual Variant get_parameter_default_value(const StringName &p_parameter) const override;358virtual bool is_parameter_read_only(const StringName &p_parameter) const override;359360virtual String get_caption() const override;361362void set_input_count(int p_inputs);363364virtual bool add_input(const String &p_name) override;365virtual void remove_input(int p_index) override;366virtual bool set_input_name(int p_input, const String &p_name) override;367368void set_input_as_auto_advance(int p_input, bool p_enable);369bool is_input_set_as_auto_advance(int p_input) const;370371void set_input_break_loop_at_end(int p_input, bool p_enable);372bool is_input_loop_broken_at_end(int p_input) const;373374void set_input_reset(int p_input, bool p_enable);375bool is_input_reset(int p_input) const;376377void set_xfade_time(double p_fade);378double get_xfade_time() const;379380void set_xfade_curve(const Ref<Curve> &p_curve);381Ref<Curve> get_xfade_curve() const;382383void set_allow_transition_to_self(bool p_enable);384bool is_allow_transition_to_self() const;385386virtual NodeTimeInfo _process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only = false) override;387388AnimationNodeTransition();389};390391class AnimationNodeOutput : public AnimationNode {392GDCLASS(AnimationNodeOutput, AnimationNode);393394public:395virtual String get_caption() const override;396virtual NodeTimeInfo _process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only = false) override;397AnimationNodeOutput();398};399400/////401402class AnimationNodeBlendTree : public AnimationRootNode {403GDCLASS(AnimationNodeBlendTree, AnimationRootNode);404405struct Node {406Ref<AnimationNode> node;407Vector2 position;408Vector<StringName> connections;409};410411AHashMap<StringName, Node> nodes;412413Vector2 graph_offset;414415void _node_changed(const StringName &p_node);416417void _initialize_node_tree();418419protected:420static void _bind_methods();421bool _set(const StringName &p_name, const Variant &p_value);422bool _get(const StringName &p_name, Variant &r_ret) const;423void _get_property_list(List<PropertyInfo> *p_list) const;424425virtual void _tree_changed() override;426virtual void _animation_node_renamed(const ObjectID &p_oid, const String &p_old_name, const String &p_new_name) override;427virtual void _animation_node_removed(const ObjectID &p_oid, const StringName &p_node) override;428429virtual void reset_state() override;430431public:432enum ConnectionError {433CONNECTION_OK,434CONNECTION_ERROR_NO_INPUT,435CONNECTION_ERROR_NO_INPUT_INDEX,436CONNECTION_ERROR_NO_OUTPUT,437CONNECTION_ERROR_SAME_NODE,438CONNECTION_ERROR_CONNECTION_EXISTS,439//no need to check for cycles due to tree topology440};441442void add_node(const StringName &p_name, Ref<AnimationNode> p_node, const Vector2 &p_position = Vector2());443Ref<AnimationNode> get_node(const StringName &p_name) const;444void remove_node(const StringName &p_name);445void rename_node(const StringName &p_name, const StringName &p_new_name);446bool has_node(const StringName &p_name) const;447StringName get_node_name(const Ref<AnimationNode> &p_node) const;448Vector<StringName> get_node_connection_array(const StringName &p_name) const;449450void set_node_position(const StringName &p_node, const Vector2 &p_position);451Vector2 get_node_position(const StringName &p_node) const;452453virtual void get_child_nodes(List<ChildNode> *r_child_nodes) override;454455void connect_node(const StringName &p_input_node, int p_input_index, const StringName &p_output_node);456void disconnect_node(const StringName &p_node, int p_input_index);457458struct NodeConnection {459StringName input_node;460int input_index = 0;461StringName output_node;462};463464ConnectionError can_connect_node(const StringName &p_input_node, int p_input_index, const StringName &p_output_node) const;465void get_node_connections(List<NodeConnection> *r_connections) const;466467virtual String get_caption() const override;468virtual NodeTimeInfo _process(const AnimationMixer::PlaybackInfo p_playback_info, bool p_test_only = false) override;469470LocalVector<StringName> get_node_list() const;471TypedArray<StringName> get_node_list_as_typed_array() const;472473void set_graph_offset(const Vector2 &p_graph_offset);474Vector2 get_graph_offset() const;475476virtual Ref<AnimationNode> get_child_by_name(const StringName &p_name) const override;477478#ifdef TOOLS_ENABLED479virtual void get_argument_options(const StringName &p_function, int p_idx, List<String> *r_options) const override;480#endif481482AnimationNodeBlendTree();483~AnimationNodeBlendTree();484};485486VARIANT_ENUM_CAST(AnimationNodeBlendTree::ConnectionError)487488489