godot/scene/animation/animation_cache.cpp
2016-06-19 01:43:02 -03:00

387 lines
9.2 KiB
C++

/*************************************************************************/
/* animation_cache.cpp */
/*************************************************************************/
/* This file is part of: */
/* GODOT ENGINE */
/* http://www.godotengine.org */
/*************************************************************************/
/* Copyright (c) 2007-2016 Juan Linietsky, Ariel Manzur. */
/* */
/* Permission is hereby granted, free of charge, to any person obtaining */
/* a copy of this software and associated documentation files (the */
/* "Software"), to deal in the Software without restriction, including */
/* without limitation the rights to use, copy, modify, merge, publish, */
/* distribute, sublicense, and/or sell copies of the Software, and to */
/* permit persons to whom the Software is furnished to do so, subject to */
/* the following conditions: */
/* */
/* The above copyright notice and this permission notice shall be */
/* included in all copies or substantial portions of the Software. */
/* */
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
/*************************************************************************/
#include "animation_cache.h"
void AnimationCache::_node_exit_tree(Node *p_node) {
//it is one shot, so it disconnects upon arrival
ERR_FAIL_COND(!connected_nodes.has(p_node));
connected_nodes.erase(p_node);
for(int i=0;i<path_cache.size();i++) {
if (path_cache[i].node!=p_node)
continue;
path_cache[i].valid=false; //invalidate path cache
}
}
void AnimationCache::_animation_changed() {
_clear_cache();
}
void AnimationCache::_clear_cache() {
while(connected_nodes.size()) {
connected_nodes.front()->get()->disconnect("exit_tree",this,"_node_exit_tree");
connected_nodes.erase(connected_nodes.front());
}
path_cache.clear();;
cache_valid=false;
cache_dirty=true;
}
void AnimationCache::_update_cache() {
cache_valid=false;
ERR_FAIL_COND(!root);
ERR_FAIL_COND(!root->is_inside_tree());
ERR_FAIL_COND(animation.is_null());
for(int i=0;i<animation->get_track_count();i++) {
NodePath np = animation->track_get_path(i);
Node *node = root->get_node(np);
if (!node) {
path_cache.push_back(Path());
ERR_EXPLAIN("Invalid Track Path in Animation: "+np);
ERR_CONTINUE(!node);
}
Path path;
Ref<Resource> res;
if (np.get_subname_count()) {
if (animation->track_get_type(i)==Animation::TYPE_TRANSFORM) {
path_cache.push_back(Path());
ERR_EXPLAIN("Transform tracks can't have a subpath: "+np);
ERR_CONTINUE(animation->track_get_type(i)==Animation::TYPE_TRANSFORM);
}
RES res;
for(int j=0;j<np.get_subname_count();j++) {
res = j==0 ? node->get(np.get_subname(j)) : res->get(np.get_subname(j));
if (res.is_null())
break;
}
if (res.is_null()) {
path_cache.push_back(Path());
ERR_EXPLAIN("Invalid Track SubPath in Animation: "+np);
ERR_CONTINUE(res.is_null());
}
path.resource=res;
path.object=res.ptr();
} else {
if (animation->track_get_type(i)==Animation::TYPE_TRANSFORM) {
StringName property = np.get_property();
String ps = property;
Spatial *sp = node->cast_to<Spatial>();
if (!sp) {
path_cache.push_back(Path());
ERR_EXPLAIN("Transform track not of type Spatial: "+np);
ERR_CONTINUE(!sp);
}
if (ps!="") {
Skeleton *sk = node->cast_to<Skeleton>();
if (!sk) {
path_cache.push_back(Path());
ERR_EXPLAIN("Property defined in Transform track, but not a Skeleton!: "+np);
ERR_CONTINUE(!sk);
}
int idx = sk->find_bone(ps);
if (idx==-1) {
path_cache.push_back(Path());
ERR_EXPLAIN("Property defined in Transform track, but not a Skeleton Bone!: "+np);
ERR_CONTINUE(idx==-1);
}
path.bone_idx=idx;
path.skeleton=sk;
}
path.spatial=sp;
}
path.node=node;
path.object=node;
}
if (animation->track_get_type(i)==Animation::TYPE_VALUE) {
if (np.get_property().operator String()=="") {
path_cache.push_back(Path());
ERR_EXPLAIN("Value Track lacks property: "+np);
ERR_CONTINUE(np.get_property().operator String()=="");
}
path.property=np.get_property();
} else if (animation->track_get_type(i)==Animation::TYPE_METHOD) {
if (np.get_property().operator String()!="") {
path_cache.push_back(Path());
ERR_EXPLAIN("Method Track has property: "+np);
ERR_CONTINUE(np.get_property().operator String()!="");
}
}
path.valid=true;
path_cache.push_back(path);
if (!connected_nodes.has(path.node)) {
connected_nodes.insert(path.node);
path.node->connect("exit_tree",this,"_node_exit_tree",Node::make_binds(path.node),CONNECT_ONESHOT);
}
}
cache_dirty=false;
cache_valid=true;
}
void AnimationCache::set_track_transform(int p_idx,const Transform& p_transform) {
if (cache_dirty)
_update_cache();
ERR_FAIL_COND(!cache_valid);
ERR_FAIL_INDEX(p_idx,path_cache.size());
Path &p = path_cache[p_idx];
if (!p.valid)
return;
ERR_FAIL_COND(!p.node);
ERR_FAIL_COND(!p.spatial);
if (p.skeleton) {
p.skeleton->set_bone_pose(p.bone_idx,p_transform);
} else {
p.spatial->set_transform(p_transform);
}
}
void AnimationCache::set_track_value(int p_idx,const Variant& p_value) {
if (cache_dirty)
_update_cache();
ERR_FAIL_COND(!cache_valid);
ERR_FAIL_INDEX(p_idx,path_cache.size());
Path &p = path_cache[p_idx];
if (!p.valid)
return;
ERR_FAIL_COND(!p.object);
p.object->set(p.property,p_value);
}
void AnimationCache::call_track(int p_idx,const StringName& p_method,const Variant** p_args,int p_argcount,Variant::CallError &r_error) {
if (cache_dirty)
_update_cache();
ERR_FAIL_COND(!cache_valid);
ERR_FAIL_INDEX(p_idx,path_cache.size());
Path &p = path_cache[p_idx];
if (!p.valid)
return;
ERR_FAIL_COND(!p.object);
p.object->call(p_method,p_args,p_argcount,r_error);
}
void AnimationCache::set_all(float p_time, float p_delta) {
if (cache_dirty)
_update_cache();
ERR_FAIL_COND(!cache_valid);
int tc = animation->get_track_count();
for(int i=0;i<tc;i++) {
switch(animation->track_get_type(i)) {
case Animation::TYPE_TRANSFORM: {
Vector3 loc,scale;
Quat rot;
animation->transform_track_interpolate(i,p_time,&loc,&rot,&scale);
Transform tr( Matrix3(rot), loc );
tr.basis.scale(scale);
set_track_transform(i,tr);
} break;
case Animation::TYPE_VALUE: {
if (animation->value_track_get_update_mode(i)==Animation::UPDATE_CONTINUOUS || (animation->value_track_get_update_mode(i)==Animation::UPDATE_DISCRETE && p_delta==0)) {
Variant v = animation->value_track_interpolate(i,p_time);
set_track_value(i,v);
} else {
List<int> indices;
animation->value_track_get_key_indices(i,p_time,p_delta,&indices);
for(List<int>::Element *E=indices.front();E;E=E->next()) {
Variant v = animation->track_get_key_value(i,E->get());
set_track_value(i,v);
}
}
} break;
case Animation::TYPE_METHOD: {
List<int> indices;
animation->method_track_get_key_indices(i,p_time,p_delta,&indices);
for(List<int>::Element *E=indices.front();E;E=E->next()) {
Vector<Variant> args = animation->method_track_get_params(i,E->get());
StringName name = animation->method_track_get_name(i,E->get());
Variant::CallError err;
if (!args.size()) {
call_track(i,name,NULL,0,err);
} else {
Vector<Variant*> argptrs;
argptrs.resize(args.size());
for(int j=0;j<args.size();j++) {
argptrs[j]=&args[j];
}
call_track(i,name,(const Variant**)&argptrs[0],args.size(),err);
}
}
} break;
default: {}
}
}
}
void AnimationCache::set_animation(const Ref<Animation>& p_animation) {
_clear_cache();
if (animation.is_valid())
animation->disconnect("changed",this,"_animation_changed");
animation=p_animation;
if (animation.is_valid())
animation->connect("changed",this,"_animation_changed");
}
void AnimationCache::_bind_methods() {
ObjectTypeDB::bind_method(_MD("_node_exit_tree"),&AnimationCache::_node_exit_tree);
ObjectTypeDB::bind_method(_MD("_animation_changed"),&AnimationCache::_animation_changed);
}
void AnimationCache::set_root(Node* p_root) {
_clear_cache();
root=p_root;
}
AnimationCache::AnimationCache() {
root=NULL;
cache_dirty=true;
cache_valid=false;
}