787 lines
24 KiB
C++
787 lines
24 KiB
C++
#pragma once
|
|
|
|
#include "scene/animation/animation_player.h"
|
|
|
|
#include "core/io/resource.h"
|
|
#include "scene/3d/skeleton_3d.h"
|
|
#include "sync_track.h"
|
|
|
|
#include <cassert>
|
|
|
|
/**
|
|
* @class AnimationData
|
|
* Represents data that is transported via animation connections in the SyncedAnimationGraph.
|
|
*
|
|
* Essentially, it is a hash map for all Animation::Track values that can are sampled from an Animation.
|
|
*/
|
|
struct AnimationData {
|
|
enum TrackType : uint8_t {
|
|
TYPE_VALUE, // Set a value in a property, can be interpolated.
|
|
TYPE_POSITION_3D, // Position 3D track, can be compressed.
|
|
TYPE_ROTATION_3D, // Rotation 3D track, can be compressed.
|
|
TYPE_SCALE_3D, // Scale 3D track, can be compressed.
|
|
TYPE_BLEND_SHAPE, // Blend Shape track, can be compressed.
|
|
TYPE_METHOD, // Call any method on a specific node.
|
|
TYPE_BEZIER, // Bezier curve.
|
|
TYPE_AUDIO,
|
|
TYPE_ANIMATION,
|
|
};
|
|
|
|
struct TrackValue {
|
|
Animation::Track *track = nullptr;
|
|
TrackType type = TYPE_ANIMATION;
|
|
|
|
virtual ~TrackValue() = default;
|
|
|
|
virtual void blend(const TrackValue &to_value, const float lambda) {
|
|
print_error(vformat("Blending of TrackValue of type %d with TrackValue of type %d not yet implemented.", type, to_value.type));
|
|
}
|
|
|
|
virtual bool operator==(const TrackValue &other_value) const {
|
|
print_error(vformat("Comparing TrackValue of type %d with TrackValue of type %d not yet implemented.", type, other_value.type));
|
|
return false;
|
|
}
|
|
bool operator!=(const TrackValue &other_value) const {
|
|
return !(*this == other_value);
|
|
}
|
|
|
|
virtual TrackValue *clone() const {
|
|
print_error(vformat("Cannot clone TrackValue of type %d: not yet implemented.", type));
|
|
return nullptr;
|
|
}
|
|
};
|
|
|
|
struct TransformTrackValue : public TrackValue {
|
|
int bone_idx = -1;
|
|
|
|
bool loc_used = false;
|
|
bool rot_used = false;
|
|
bool scale_used = false;
|
|
Vector3 init_loc = Vector3(0, 0, 0);
|
|
Quaternion init_rot = Quaternion(0, 0, 0, 1);
|
|
Vector3 init_scale = Vector3(1, 1, 1);
|
|
Vector3 loc;
|
|
Quaternion rot;
|
|
Vector3 scale;
|
|
|
|
TransformTrackValue() { type = TYPE_POSITION_3D; }
|
|
|
|
void blend(const TrackValue &to_value, const float lambda) override {
|
|
const TransformTrackValue *to_value_casted = &static_cast<const TransformTrackValue &>(to_value);
|
|
assert(bone_idx == to_value_casted->bone_idx);
|
|
if (loc_used) {
|
|
loc = (1. - lambda) * loc + lambda * to_value_casted->loc;
|
|
}
|
|
|
|
if (rot_used) {
|
|
rot = rot.slerp(to_value_casted->rot, lambda);
|
|
}
|
|
|
|
if (scale_used) {
|
|
scale = (1. - lambda) * scale + lambda * to_value_casted->scale;
|
|
}
|
|
}
|
|
|
|
bool operator==(const TrackValue &other_value) const override {
|
|
if (type != other_value.type) {
|
|
return false;
|
|
}
|
|
|
|
const TransformTrackValue *other_value_casted = &static_cast<const TransformTrackValue &>(other_value);
|
|
return bone_idx == other_value_casted->bone_idx && loc == other_value_casted->loc && rot == other_value_casted->rot && scale == other_value_casted->scale;
|
|
}
|
|
|
|
TrackValue *clone() const override {
|
|
TransformTrackValue *result = memnew(TransformTrackValue);
|
|
result->track = track;
|
|
result->bone_idx = bone_idx;
|
|
|
|
result->loc_used = loc_used;
|
|
result->rot_used = rot_used;
|
|
result->scale_used = scale_used;
|
|
|
|
result->init_loc = init_loc;
|
|
result->init_rot = init_rot;
|
|
result->init_scale = init_scale;
|
|
|
|
result->loc = loc;
|
|
result->rot = rot;
|
|
result->scale = scale;
|
|
|
|
return result;
|
|
}
|
|
};
|
|
|
|
AnimationData() = default;
|
|
~AnimationData() {
|
|
_clear_values();
|
|
}
|
|
AnimationData(const AnimationData &other) {
|
|
for (const KeyValue<Animation::TypeHash, TrackValue *> &K : other.track_values) {
|
|
track_values.insert(K.key, K.value->clone());
|
|
}
|
|
}
|
|
AnimationData(AnimationData &&other) noexcept :
|
|
track_values(std::exchange(other.track_values, AHashMap<Animation::TypeHash, TrackValue *, HashHasher>())) {
|
|
}
|
|
AnimationData &operator=(const AnimationData &other) {
|
|
AnimationData temp(other);
|
|
std::swap(track_values, temp.track_values);
|
|
return *this;
|
|
}
|
|
AnimationData &operator=(AnimationData &&other) noexcept {
|
|
std::swap(track_values, other.track_values);
|
|
return *this;
|
|
}
|
|
|
|
void
|
|
set_value(const Animation::TypeHash &thash, TrackValue *value) {
|
|
if (!track_values.has(thash)) {
|
|
track_values.insert(thash, value);
|
|
} else {
|
|
track_values[thash] = value;
|
|
}
|
|
}
|
|
|
|
void clear() {
|
|
_clear_values();
|
|
}
|
|
|
|
bool has_same_tracks(const AnimationData &other) const {
|
|
HashSet<Animation::TypeHash> valid_track_hashes;
|
|
for (const KeyValue<Animation::TypeHash, TrackValue *> &K : track_values) {
|
|
valid_track_hashes.insert(K.key);
|
|
}
|
|
|
|
for (const KeyValue<Animation::TypeHash, TrackValue *> &K : other.track_values) {
|
|
if (HashSet<Animation::TypeHash>::Iterator entry = valid_track_hashes.find(K.key)) {
|
|
valid_track_hashes.remove(entry);
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return valid_track_hashes.size() == 0;
|
|
}
|
|
|
|
void blend(const AnimationData &to_data, const float lambda) {
|
|
if (!has_same_tracks(to_data)) {
|
|
print_error("Cannot blend AnimationData: tracks do not match.");
|
|
return;
|
|
}
|
|
|
|
for (const KeyValue<Animation::TypeHash, TrackValue *> &K : track_values) {
|
|
TrackValue *track_value = K.value;
|
|
TrackValue *other_track_value = to_data.track_values[K.key];
|
|
|
|
track_value->blend(*other_track_value, lambda);
|
|
}
|
|
}
|
|
|
|
void sample_from_animation(const Ref<Animation> &animation, const Skeleton3D *skeleton_3d, double p_time);
|
|
|
|
AHashMap<Animation::TypeHash, TrackValue *, HashHasher> track_values; // Animation::Track to TrackValue
|
|
|
|
protected:
|
|
void _clear_values() {
|
|
for (KeyValue<Animation::TypeHash, TrackValue *> &K : track_values) {
|
|
memdelete(K.value);
|
|
}
|
|
}
|
|
};
|
|
|
|
struct GraphEvaluationContext {
|
|
AnimationPlayer *animation_player = nullptr;
|
|
Skeleton3D *skeleton_3d = nullptr;
|
|
};
|
|
|
|
/**
|
|
* @class SyncedAnimationNode
|
|
* Base class for all nodes in an SyncedAnimationGraph including BlendTree nodes and StateMachine states.
|
|
*/
|
|
class SyncedAnimationNode : public Resource {
|
|
GDCLASS(SyncedAnimationNode, Resource);
|
|
|
|
friend class SyncedAnimationGraph;
|
|
|
|
protected:
|
|
static void _bind_methods();
|
|
|
|
virtual void get_parameter_list(List<PropertyInfo> *r_list) const;
|
|
virtual Variant get_parameter_default_value(const StringName &p_parameter) const;
|
|
virtual bool is_parameter_read_only(const StringName &p_parameter) const;
|
|
|
|
virtual void set_parameter(const StringName &p_name, const Variant &p_value);
|
|
virtual Variant get_parameter(const StringName &p_name) const;
|
|
|
|
virtual void _tree_changed();
|
|
virtual void _animation_node_renamed(const ObjectID &p_oid, const String &p_old_name, const String &p_new_name);
|
|
virtual void _animation_node_removed(const ObjectID &p_oid, const StringName &p_node);
|
|
|
|
public:
|
|
struct NodeTimeInfo {
|
|
double length = 0.0;
|
|
double position = 0.0;
|
|
double sync_position = 0.0;
|
|
double delta = 0.0;
|
|
double sync_delta = 0.0;
|
|
bool is_synced = false;
|
|
|
|
Animation::LoopMode loop_mode = Animation::LOOP_LINEAR;
|
|
SyncTrack sync_track;
|
|
};
|
|
NodeTimeInfo node_time_info;
|
|
bool active = false;
|
|
|
|
StringName name;
|
|
Vector2 position;
|
|
|
|
virtual ~SyncedAnimationNode() override = default;
|
|
virtual bool initialize(GraphEvaluationContext &context) { return true; }
|
|
|
|
virtual void activate_inputs(Vector<Ref<SyncedAnimationNode>> input_nodes) {
|
|
// By default, all inputs nodes are activated.
|
|
for (const Ref<SyncedAnimationNode> &node : input_nodes) {
|
|
node->active = true;
|
|
node->node_time_info.is_synced = node_time_info.is_synced;
|
|
}
|
|
}
|
|
virtual void calculate_sync_track(Vector<Ref<SyncedAnimationNode>> input_nodes) {
|
|
// By default, use the SyncTrack of the first input.
|
|
if (input_nodes.size() > 0) {
|
|
node_time_info.sync_track = input_nodes[0]->node_time_info.sync_track;
|
|
}
|
|
}
|
|
virtual void update_time(double p_time) {
|
|
if (node_time_info.is_synced) {
|
|
node_time_info.sync_position = p_time;
|
|
} else {
|
|
node_time_info.delta = p_time;
|
|
node_time_info.position += p_time;
|
|
if (node_time_info.position > node_time_info.length) {
|
|
switch (node_time_info.loop_mode) {
|
|
case Animation::LOOP_NONE: {
|
|
node_time_info.position = node_time_info.length;
|
|
break;
|
|
}
|
|
case Animation::LOOP_LINEAR: {
|
|
assert(node_time_info.length > 0.0);
|
|
while (node_time_info.position > node_time_info.length) {
|
|
node_time_info.position -= node_time_info.length;
|
|
}
|
|
break;
|
|
}
|
|
case Animation::LOOP_PINGPONG: {
|
|
assert(false && !"Not yet implemented.");
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
virtual void evaluate(GraphEvaluationContext &context, const LocalVector<AnimationData *> &input_datas, AnimationData &output_data) {
|
|
// By default, use the AnimationData of the first input.
|
|
if (input_datas.size() > 0) {
|
|
output_data = *input_datas[0];
|
|
}
|
|
}
|
|
|
|
bool set_input_node(const StringName &socket_name, SyncedAnimationNode *node);
|
|
virtual void get_input_names(Vector<StringName> &inputs) const {}
|
|
|
|
int get_input_index(const StringName &port_name) const {
|
|
Vector<StringName> inputs;
|
|
get_input_names(inputs);
|
|
return inputs.find(port_name);
|
|
}
|
|
int get_input_count() const {
|
|
Vector<StringName> inputs;
|
|
get_input_names(inputs);
|
|
return inputs.size();
|
|
}
|
|
|
|
// Creates a list of nodes nested within the current node. E.g. all nodes within a BlendTree node.
|
|
virtual void get_child_nodes(List<Ref<SyncedAnimationNode>> *r_child_nodes) const {}
|
|
};
|
|
|
|
class AnimationSamplerNode : public SyncedAnimationNode {
|
|
GDCLASS(AnimationSamplerNode, SyncedAnimationNode);
|
|
|
|
public:
|
|
StringName animation_name;
|
|
|
|
void set_animation(const StringName &p_name);
|
|
StringName get_animation() const;
|
|
|
|
private:
|
|
Ref<Animation> animation;
|
|
|
|
bool initialize(GraphEvaluationContext &context) override;
|
|
void evaluate(GraphEvaluationContext &context, const LocalVector<AnimationData *> &inputs, AnimationData &output) override;
|
|
|
|
protected:
|
|
static void _bind_methods();
|
|
};
|
|
|
|
class OutputNode : public SyncedAnimationNode {
|
|
GDCLASS(OutputNode, SyncedAnimationNode);
|
|
|
|
public:
|
|
void get_input_names(Vector<StringName> &inputs) const override {
|
|
inputs.push_back("Input");
|
|
}
|
|
};
|
|
|
|
class AnimationBlend2Node : public SyncedAnimationNode {
|
|
GDCLASS(AnimationBlend2Node, SyncedAnimationNode);
|
|
|
|
public:
|
|
StringName blend_amount = PNAME("blend_amount");
|
|
float blend_weight = 0.0f;
|
|
bool sync = true;
|
|
|
|
void get_input_names(Vector<StringName> &inputs) const override {
|
|
inputs.push_back("Input0");
|
|
inputs.push_back("Input1");
|
|
}
|
|
void activate_inputs(Vector<Ref<SyncedAnimationNode>> input_nodes) override {
|
|
for (const Ref<SyncedAnimationNode> &node : input_nodes) {
|
|
node->active = true;
|
|
|
|
// If this Blend2 node is already synced then inputs are also synced. Otherwise, inputs are only set to synced if synced blending is active in this node.
|
|
node->node_time_info.is_synced = node_time_info.is_synced || sync;
|
|
}
|
|
}
|
|
void calculate_sync_track(Vector<Ref<SyncedAnimationNode>> input_nodes) override {
|
|
if (node_time_info.is_synced || sync) {
|
|
node_time_info.sync_track = SyncTrack::blend(blend_weight, input_nodes[0]->node_time_info.sync_track, input_nodes[1]->node_time_info.sync_track);
|
|
node_time_info.length = node_time_info.sync_track.duration;
|
|
}
|
|
}
|
|
void update_time(double p_delta) override {
|
|
SyncedAnimationNode::update_time(p_delta);
|
|
|
|
if (sync && !node_time_info.is_synced) {
|
|
node_time_info.sync_position = node_time_info.sync_track.calc_sync_from_abs_time(node_time_info.position);
|
|
}
|
|
}
|
|
void evaluate(GraphEvaluationContext &context, const LocalVector<AnimationData *> &inputs, AnimationData &output) override;
|
|
|
|
void set_use_sync(bool p_sync);
|
|
bool is_using_sync() const;
|
|
|
|
protected:
|
|
static void _bind_methods();
|
|
|
|
void get_parameter_list(List<PropertyInfo> *p_list) const override;
|
|
Variant get_parameter_default_value(const StringName &p_parameter) const override;
|
|
void set_parameter(const StringName &p_name, const Variant &p_value) override;
|
|
Variant get_parameter(const StringName &p_name) const override;
|
|
|
|
void _get_property_list(List<PropertyInfo> *p_list) const;
|
|
bool _get(const StringName &p_name, Variant &r_value) const;
|
|
bool _set(const StringName &p_name, const Variant &p_value);
|
|
};
|
|
|
|
struct BlendTreeConnection {
|
|
const Ref<SyncedAnimationNode> source_node = nullptr;
|
|
const Ref<SyncedAnimationNode> target_node = nullptr;
|
|
const StringName target_port_name = "";
|
|
};
|
|
|
|
/**
|
|
* @class BlendTreeGraph
|
|
* Helper class that is used to build runtime blend trees and also to validate connections.
|
|
*/
|
|
struct BlendTreeGraph {
|
|
struct NodeConnectionInfo {
|
|
int parent_node_index = -1;
|
|
HashSet<int> input_subtree_node_indices; // Contains all nodes down to the tree leaves that influence this node.
|
|
LocalVector<int> connected_child_node_index_at_port; // Contains for each input port the index of the node that is connected to it.
|
|
|
|
NodeConnectionInfo() = default;
|
|
|
|
explicit NodeConnectionInfo(const SyncedAnimationNode *node) {
|
|
parent_node_index = -1;
|
|
for (int i = 0; i < node->get_input_count(); i++) {
|
|
connected_child_node_index_at_port.push_back(-1);
|
|
}
|
|
}
|
|
|
|
void _print_subtree() const {
|
|
String result = vformat("subtree node indices (%d): ", input_subtree_node_indices.size());
|
|
bool is_first = true;
|
|
for (int index : input_subtree_node_indices) {
|
|
if (is_first) {
|
|
result += vformat("%d", index);
|
|
is_first = false;
|
|
} else {
|
|
result += vformat(", %d", index);
|
|
}
|
|
}
|
|
print_line(result);
|
|
}
|
|
|
|
void apply_node_mapping(const LocalVector<int> &node_index_mapping) {
|
|
// Map connected node indices
|
|
for (unsigned int j = 0; j < connected_child_node_index_at_port.size(); j++) {
|
|
int connected_node_index = connected_child_node_index_at_port[j];
|
|
connected_child_node_index_at_port[j] = node_index_mapping.find(connected_node_index);
|
|
}
|
|
|
|
// Map connected subtrees
|
|
HashSet<int> old_indices = input_subtree_node_indices;
|
|
input_subtree_node_indices.clear();
|
|
for (int old_index : old_indices) {
|
|
input_subtree_node_indices.insert(node_index_mapping.find(old_index));
|
|
}
|
|
}
|
|
};
|
|
|
|
Vector<Ref<SyncedAnimationNode>> nodes; // All added nodes
|
|
LocalVector<NodeConnectionInfo> node_connection_info;
|
|
LocalVector<BlendTreeConnection> connections;
|
|
|
|
BlendTreeGraph() {
|
|
Ref<OutputNode> output_node;
|
|
output_node.instantiate();
|
|
output_node->name = "Output";
|
|
add_node(output_node);
|
|
}
|
|
|
|
Ref<SyncedAnimationNode> get_output_node() const {
|
|
return nodes[0];
|
|
}
|
|
|
|
int find_node_index(const Ref<SyncedAnimationNode> &node) const {
|
|
for (int i = 0; i < nodes.size(); i++) {
|
|
if (nodes[i] == node) {
|
|
return i;
|
|
}
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
int find_node_index_by_name(const StringName &name) const {
|
|
for (int i = 0; i < nodes.size(); i++) {
|
|
if (nodes[i]->name == name) {
|
|
return i;
|
|
}
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
void add_node(const Ref<SyncedAnimationNode> &node) {
|
|
StringName node_base_name = node->name;
|
|
if (node_base_name.is_empty()) {
|
|
node_base_name = node->get_class_name();
|
|
}
|
|
node->name = node_base_name;
|
|
|
|
int number_suffix = 1;
|
|
while (find_node_index_by_name(node->name) != -1) {
|
|
node->name = vformat("%s %d", node_base_name, number_suffix);
|
|
number_suffix++;
|
|
}
|
|
|
|
nodes.push_back(node);
|
|
node_connection_info.push_back(NodeConnectionInfo(node.ptr()));
|
|
}
|
|
|
|
void sort_nodes_and_references() {
|
|
LocalVector<int> sorted_node_indices = get_sorted_node_indices();
|
|
|
|
Vector<Ref<SyncedAnimationNode>> sorted_nodes;
|
|
Vector<NodeConnectionInfo> old_node_connection_info = node_connection_info;
|
|
for (unsigned int i = 0; i < sorted_node_indices.size(); i++) {
|
|
int node_index = sorted_node_indices[i];
|
|
sorted_nodes.push_back(nodes[node_index]);
|
|
node_connection_info[i] = old_node_connection_info[node_index];
|
|
}
|
|
nodes = sorted_nodes;
|
|
|
|
for (NodeConnectionInfo &connection_info : node_connection_info) {
|
|
if (connection_info.parent_node_index != -1) {
|
|
connection_info.parent_node_index = sorted_node_indices[connection_info.parent_node_index];
|
|
}
|
|
connection_info.apply_node_mapping(sorted_node_indices);
|
|
}
|
|
}
|
|
|
|
LocalVector<int> get_sorted_node_indices() {
|
|
LocalVector<int> result;
|
|
|
|
sort_nodes_recursive(0, result);
|
|
result.reverse();
|
|
|
|
return result;
|
|
}
|
|
|
|
void sort_nodes_recursive(int node_index, LocalVector<int> &result) {
|
|
for (int input_node_index : node_connection_info[node_index].connected_child_node_index_at_port) {
|
|
if (input_node_index >= 0) {
|
|
sort_nodes_recursive(input_node_index, result);
|
|
}
|
|
}
|
|
result.push_back(node_index);
|
|
}
|
|
|
|
void add_index_and_update_subtrees_recursive(int node, int node_parent) {
|
|
if (node_parent == -1) {
|
|
return;
|
|
}
|
|
|
|
node_connection_info[node_parent].input_subtree_node_indices.insert(node);
|
|
|
|
for (int index : node_connection_info[node].input_subtree_node_indices) {
|
|
node_connection_info[node_parent].input_subtree_node_indices.insert(index);
|
|
}
|
|
|
|
add_index_and_update_subtrees_recursive(node_parent, node_connection_info[node_parent].parent_node_index);
|
|
}
|
|
|
|
bool add_connection(const Ref<SyncedAnimationNode> &source_node, const Ref<SyncedAnimationNode> &target_node, const StringName &target_port_name) {
|
|
if (!is_connection_valid(source_node, target_node, target_port_name)) {
|
|
return false;
|
|
}
|
|
|
|
int source_node_index = find_node_index(source_node);
|
|
int target_node_index = find_node_index(target_node);
|
|
int target_input_port_index = target_node->get_input_index(target_port_name);
|
|
|
|
node_connection_info[source_node_index].parent_node_index = target_node_index;
|
|
node_connection_info[target_node_index].connected_child_node_index_at_port[target_input_port_index] = source_node_index;
|
|
connections.push_back(BlendTreeConnection{ source_node, target_node, target_port_name });
|
|
|
|
add_index_and_update_subtrees_recursive(source_node_index, target_node_index);
|
|
|
|
return true;
|
|
}
|
|
|
|
bool is_connection_valid(const Ref<SyncedAnimationNode> &source_node, const Ref<SyncedAnimationNode> &target_node, StringName target_port_name) {
|
|
int source_node_index = find_node_index(source_node);
|
|
if (source_node_index == -1) {
|
|
print_error("Cannot connect nodes: source node not found.");
|
|
return false;
|
|
}
|
|
|
|
if (node_connection_info[source_node_index].parent_node_index != -1) {
|
|
print_error("Cannot connect node: source node already has a parent.");
|
|
return false;
|
|
}
|
|
|
|
int target_node_index = find_node_index(target_node);
|
|
if (target_node_index == -1) {
|
|
print_error("Cannot connect nodes: target node not found.");
|
|
return false;
|
|
}
|
|
|
|
Vector<StringName> target_inputs;
|
|
target_node->get_input_names(target_inputs);
|
|
|
|
if (!target_inputs.has(target_port_name)) {
|
|
print_error("Cannot connect nodes: target port not found.");
|
|
return false;
|
|
}
|
|
|
|
int target_input_port_index = target_node->get_input_index(target_port_name);
|
|
if (node_connection_info[target_node_index].connected_child_node_index_at_port[target_input_port_index] != -1) {
|
|
print_error("Cannot connect node: target port already connected");
|
|
return false;
|
|
}
|
|
|
|
if (node_connection_info[source_node_index].input_subtree_node_indices.has(target_node_index)) {
|
|
print_error("Cannot connect node: connection would create loop.");
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
};
|
|
|
|
class SyncedBlendTree : public SyncedAnimationNode {
|
|
GDCLASS(SyncedBlendTree, SyncedAnimationNode);
|
|
|
|
BlendTreeGraph tree_graph;
|
|
bool tree_initialized = false;
|
|
|
|
void sort_nodes() {
|
|
_node_runtime_data.clear();
|
|
tree_graph.sort_nodes_and_references();
|
|
}
|
|
|
|
void setup_runtime_data() {
|
|
// Add nodes and allocate runtime data
|
|
for (int i = 0; i < tree_graph.nodes.size(); i++) {
|
|
const Ref<SyncedAnimationNode> node = tree_graph.nodes[i];
|
|
|
|
NodeRuntimeData node_runtime_data;
|
|
for (int ni = 0; ni < node->get_input_count(); ni++) {
|
|
node_runtime_data.input_data.push_back(nullptr);
|
|
}
|
|
|
|
node_runtime_data.output_data = nullptr;
|
|
_node_runtime_data.push_back(node_runtime_data);
|
|
}
|
|
|
|
// Populate runtime data (only now is this.nodes populated to retrieve the nodes)
|
|
for (int i = 0; i < tree_graph.nodes.size(); i++) {
|
|
Ref<SyncedAnimationNode> node = tree_graph.nodes[i];
|
|
NodeRuntimeData &node_runtime_data = _node_runtime_data[i];
|
|
|
|
for (int port_index = 0; port_index < node->get_input_count(); port_index++) {
|
|
const int connected_node_index = tree_graph.node_connection_info[i].connected_child_node_index_at_port[port_index];
|
|
node_runtime_data.input_nodes.push_back(tree_graph.nodes[connected_node_index]);
|
|
}
|
|
}
|
|
}
|
|
|
|
protected:
|
|
void _get_property_list(List<PropertyInfo> *p_list) const;
|
|
bool _get(const StringName &p_name, Variant &r_value) const;
|
|
bool _set(const StringName &p_name, const Variant &p_value);
|
|
|
|
public:
|
|
struct NodeRuntimeData {
|
|
Vector<Ref<SyncedAnimationNode>> input_nodes;
|
|
LocalVector<AnimationData *> input_data;
|
|
AnimationData *output_data = nullptr;
|
|
};
|
|
LocalVector<NodeRuntimeData> _node_runtime_data;
|
|
|
|
Ref<SyncedAnimationNode> get_output_node() const {
|
|
return tree_graph.nodes[0];
|
|
}
|
|
|
|
int find_node_index(const Ref<SyncedAnimationNode> &node) const {
|
|
return tree_graph.find_node_index(node);
|
|
}
|
|
|
|
int find_node_index_by_name(const StringName &name) const {
|
|
return tree_graph.find_node_index_by_name(name);
|
|
}
|
|
|
|
void add_node(const Ref<SyncedAnimationNode> &node) {
|
|
if (tree_initialized) {
|
|
print_error("Cannot add node to BlendTree: BlendTree already initialized.");
|
|
return;
|
|
}
|
|
|
|
tree_graph.add_node(node);
|
|
}
|
|
|
|
bool add_connection(const Ref<SyncedAnimationNode> &source_node, const Ref<SyncedAnimationNode> &target_node, const StringName &target_port_name) {
|
|
if (tree_initialized) {
|
|
print_error("Cannot add connection to BlendTree: BlendTree already initialized.");
|
|
return false;
|
|
}
|
|
|
|
return tree_graph.add_connection(source_node, target_node, target_port_name);
|
|
}
|
|
|
|
// overrides from SyncedAnimationNode
|
|
bool initialize(GraphEvaluationContext &context) override {
|
|
sort_nodes();
|
|
setup_runtime_data();
|
|
|
|
for (const Ref<SyncedAnimationNode> &node : tree_graph.nodes) {
|
|
if (!node->initialize(context)) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
tree_initialized = true;
|
|
|
|
return true;
|
|
}
|
|
|
|
void activate_inputs(Vector<Ref<SyncedAnimationNode>> input_nodes) override {
|
|
tree_graph.nodes[0]->active = true;
|
|
for (int i = 0; i < tree_graph.nodes.size(); i++) {
|
|
const Ref<SyncedAnimationNode> &node = tree_graph.nodes[i];
|
|
|
|
if (!node->active) {
|
|
continue;
|
|
}
|
|
|
|
const NodeRuntimeData &node_runtime_data = _node_runtime_data[i];
|
|
node->activate_inputs(node_runtime_data.input_nodes);
|
|
}
|
|
}
|
|
|
|
void calculate_sync_track(Vector<Ref<SyncedAnimationNode>> input_nodes) override {
|
|
for (int i = tree_graph.nodes.size() - 1; i > 0; i--) {
|
|
const Ref<SyncedAnimationNode> &node = tree_graph.nodes[i];
|
|
|
|
if (!node->active) {
|
|
continue;
|
|
}
|
|
|
|
const NodeRuntimeData &node_runtime_data = _node_runtime_data[i];
|
|
|
|
node->calculate_sync_track(node_runtime_data.input_nodes);
|
|
}
|
|
}
|
|
|
|
void update_time(double p_delta) override {
|
|
tree_graph.nodes[0]->node_time_info.delta = p_delta;
|
|
tree_graph.nodes[0]->node_time_info.position += p_delta;
|
|
|
|
for (int i = 1; i < tree_graph.nodes.size(); i++) {
|
|
const Ref<SyncedAnimationNode> &node = tree_graph.nodes[i];
|
|
|
|
if (!node->active) {
|
|
continue;
|
|
}
|
|
|
|
const Ref<SyncedAnimationNode> &node_parent = tree_graph.nodes[tree_graph.node_connection_info[i].parent_node_index];
|
|
|
|
if (node->node_time_info.is_synced) {
|
|
node->update_time(node_parent->node_time_info.sync_position);
|
|
} else {
|
|
node->update_time(node_parent->node_time_info.delta);
|
|
}
|
|
}
|
|
}
|
|
|
|
void evaluate(GraphEvaluationContext &context, const LocalVector<AnimationData *> &input_datas, AnimationData &output_data) override {
|
|
for (int i = tree_graph.nodes.size() - 1; i > 0; i--) {
|
|
const Ref<SyncedAnimationNode> &node = tree_graph.nodes[i];
|
|
|
|
if (!node->active) {
|
|
continue;
|
|
}
|
|
|
|
NodeRuntimeData &node_runtime_data = _node_runtime_data[i];
|
|
|
|
// Populate the inputs
|
|
for (unsigned int j = 0; j < node_runtime_data.input_data.size(); j++) {
|
|
int child_index = tree_graph.node_connection_info[i].connected_child_node_index_at_port[j];
|
|
node_runtime_data.input_data[j] = _node_runtime_data[child_index].output_data;
|
|
}
|
|
|
|
// Set output pointer
|
|
if (i == 1) {
|
|
node_runtime_data.output_data = &output_data;
|
|
} else {
|
|
node_runtime_data.output_data = memnew(AnimationData);
|
|
}
|
|
|
|
node->evaluate(context, node_runtime_data.input_data, *node_runtime_data.output_data);
|
|
|
|
// All inputs have been consumed and can now be freed.
|
|
for (const int child_index : tree_graph.node_connection_info[i].connected_child_node_index_at_port) {
|
|
memfree(_node_runtime_data[child_index].output_data);
|
|
}
|
|
}
|
|
}
|
|
|
|
void get_child_nodes(List<Ref<SyncedAnimationNode>> *r_child_nodes) const override {
|
|
for (const Ref<SyncedAnimationNode> &node : tree_graph.nodes) {
|
|
r_child_nodes->push_back(node.ptr());
|
|
}
|
|
}
|
|
};
|