]> git.tdb.fi Git - libs/gl.git/blobdiff - source/animation.cpp
Refactor KeyFrame ownership management in Animation
[libs/gl.git] / source / animation.cpp
index 61c21ce8611c88f40694efedea9a35f2d687a65e..900542dd04bded26a97186b41c98ee949be4dc4f 100644 (file)
@@ -1,8 +1,9 @@
 #include <cmath>
 #include <msp/core/maputils.h>
 #include <msp/datafile/collection.h>
-#include <msp/time/units.h>
+#include <msp/interpolate/bezierspline.h>
 #include "animation.h"
+#include "animationeventobserver.h"
 #include "armature.h"
 #include "error.h"
 #include "pose.h"
@@ -23,6 +24,8 @@ Animation::~Animation()
 
 void Animation::set_armature(const Armature &a)
 {
+       if(!keyframes.empty() && &a!=armature)
+               throw invalid_operation("Animation::set_armature");
        armature = &a;
 }
 
@@ -43,171 +46,239 @@ const string &Animation::get_uniform_name(unsigned i) const
 
 void Animation::add_keyframe(const Time::TimeDelta &t, const KeyFrame &kf)
 {
-       if(!keyframes.empty() && t<keyframes.back().time)
-               throw invalid_argument("Animation::add_keyframe");
+       add_keyframe(t, &kf, false, false);
+       create_curves();
+}
 
-       TimedKeyFrame tkf(*this);
-       tkf.time = t;
-       tkf.keyframe = &kf;
-       tkf.keyframe.keep();
-       prepare_keyframe(tkf);
-       keyframes.push_back(tkf);
+void Animation::add_keyframe(const Time::TimeDelta &t, const KeyFrame &kf, float slope)
+{
+       add_keyframe(t, kf, slope, slope);
 }
 
-void Animation::set_looping(bool l)
+void Animation::add_keyframe(const Time::TimeDelta &t, const KeyFrame &kf, float, float)
 {
-       looping = l;
+       add_keyframe(t, kf);
 }
 
-void Animation::prepare_keyframe(TimedKeyFrame &tkf)
+void Animation::add_control_keyframe(const KeyFrame &kf)
 {
-       tkf.prev = (keyframes.empty() ? 0 : &keyframes.back());
+       if(keyframes.empty())
+               throw invalid_operation("Animation::add_control_keyframe");
+
+       add_keyframe(keyframes.back().time, &kf, true, false);
+}
+
+void Animation::add_keyframe(const Time::TimeDelta &t, const KeyFrame *kf, bool c, bool owned)
+{
+       if(c && keyframes.empty())
+               throw invalid_argument("Animation::add_keyframe");
+       if(keyframes.empty() && t!=Time::zero)
+               throw invalid_argument("Animation::add_keyframe");
+       if(!keyframes.empty() && t<keyframes.back().time)
+               throw invalid_argument("Animation::add_keyframe");
+       if(kf->get_pose() && armature && kf->get_pose()->get_armature()!=armature)
+               throw invalid_argument("Animation::add_keyframe");
+
+       const KeyFrame::UniformMap &kf_uniforms = kf->get_uniforms();
+       for(vector<UniformInfo>::const_iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
+       {
+               KeyFrame::UniformMap::const_iterator j = kf_uniforms.find(i->name);
+               if(j!=kf_uniforms.end() && j->second.size!=i->size)
+                       throw invalid_argument("Animation::add_keyframe");
+       }
+
+       if(kf->get_pose() && !armature)
+               armature = kf->get_pose()->get_armature();
+
+       TimedKeyFrame tkf;
+       tkf.time = t;
+       tkf.keyframe = kf;
+       if(!owned)
+               tkf.keyframe.keep();
+       tkf.control = c;
+
+       keyframes.push_back(tkf);
 
-       const KeyFrame::UniformMap &kf_uniforms = tkf.keyframe->get_uniforms();
        for(KeyFrame::UniformMap::const_iterator i=kf_uniforms.begin(); i!=kf_uniforms.end(); ++i)
        {
                bool found = false;
-               for(unsigned j=0; (!found && j<uniforms.size()); ++j)
-                       if(uniforms[j].name==i->first)
-                       {
-                               if(uniforms[j].size!=i->second.size)
-                                       throw invalid_operation("Animation::prepare_keyframe");
-                               found = true;
-                       }
+               for(vector<UniformInfo>::const_iterator j=uniforms.begin(); (!found && j!=uniforms.end()); ++j)
+                       found = (j->name==i->first);
 
                if(!found)
                        uniforms.push_back(UniformInfo(i->first, i->second.size));
        }
-
-       tkf.prepare();
 }
 
+void Animation::create_curves()
+{
+       for(vector<Curve *>::iterator i=curves.begin(); i!=curves.end(); ++i)
+               delete *i;
+       curves.clear();
 
-Animation::AxisInterpolation::AxisInterpolation():
-       slope(0),
-       scale(0)
-{ }
+       curves.reserve(3+uniforms.size());
+       create_curve<3>(POSITION, &extract_position);
+       create_curve<3>(EULER, &extract_euler);
+       create_curve<3>(SCALE, &extract_scale);
 
-Animation::AxisInterpolation::AxisInterpolation(const float *axis1, const float *axis2)
+       for(vector<UniformInfo>::const_iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
+       {
+               if(i->size==1)
+                       create_curve<1>(UNIFORM, ExtractUniform<1>(i->name));
+               else if(i->size==2)
+                       create_curve<2>(UNIFORM, ExtractUniform<2>(i->name));
+               else if(i->size==3)
+                       create_curve<3>(UNIFORM, ExtractUniform<3>(i->name));
+               else if(i->size==4)
+                       create_curve<4>(UNIFORM, ExtractUniform<4>(i->name));
+       }
+}
+
+template<unsigned N, typename T>
+void Animation::create_curve(CurveTarget target, const T &extract)
 {
-       // Compute a normalized vector halfway between the two endpoints
-       float half[3];
-       float a1_len = 0;
-       float h_len = 0;
-       for(unsigned i=0; i<3; ++i)
+       typedef typename ValueCurve<N>::Knot Knot;
+
+       vector<Knot> knots;
+       unsigned n_control = 0;
+       for(vector<TimedKeyFrame>::const_iterator i=keyframes.begin(); i!=keyframes.end(); ++i)
        {
-               half[i] = (axis1[i]+axis2[i])/2;
-               a1_len += axis1[i]*axis1[i];
-               h_len += half[i]*half[i];
+               if(i->control && knots.empty())
+                       continue;
+
+               typename Interpolate::SplineValue<float, N>::Type value;
+               if(extract(*i->keyframe, value))
+               {
+                       float x = i->time/Time::sec;
+                       if(i->control)
+                       {
+                               ++n_control;
+                               if(n_control>2)
+                                       throw logic_error("too many control keyframes");
+                       }
+                       else
+                       {
+                               if(n_control==1)
+                               {
+                                       typename Knot::Value cv = knots.back().y;
+                                       knots.back().y = (knots[knots.size()-2].y+cv*2.0f)/3.0f;
+                                       knots.push_back(Knot(x, (value+cv*2.0f)/3.0f));
+                               }
+                               else if(n_control==0 && !knots.empty())
+                               {
+                                       typename Knot::Value prev = knots.back().y;
+                                       knots.push_back(Knot(knots.back().x, (prev*2.0f+value)/3.0f));
+                                       knots.push_back(Knot(x, (prev+value*2.0f)/3.0f));
+                               }
+                               n_control = 0;
+                       }
+                       knots.push_back(Knot(x, value));
+               }
        }
+       
+       while(n_control--)
+               knots.pop_back();
 
-       // Compute correction factors for smooth interpolation
-       float cos_half = (axis1[0]*half[0]+axis1[1]*half[1]+axis1[2]*half[2])/sqrt(a1_len*h_len);
-       float angle = acos(cos_half);
-       slope = (angle ? angle/tan(angle) : 1);
-       scale = cos_half;
+       curves.push_back(new ValueCurve<N>(target, knots));
 }
 
+bool Animation::extract_position(const KeyFrame &kf, Vector3 &value)
+{
+       value = kf.get_transform().get_position();
+       return true;
+}
 
-Animation::MatrixInterpolation::MatrixInterpolation():
-       matrix1(0),
-       matrix2(0)
-{ }
-
-Animation::MatrixInterpolation::MatrixInterpolation(const Matrix &m1, const Matrix &m2):
-       matrix1(&m1),
-       matrix2(&m2)
+bool Animation::extract_euler(const KeyFrame &kf, Vector3 &value)
 {
-       const float *m1_data = matrix1->data();
-       const float *m2_data = matrix2->data();
-       for(unsigned i=0; i<3; ++i)
-               axes[i] = AxisInterpolation(m1_data+i*4, m2_data+i*4);
+       const Transform::AngleVector3 &euler = kf.get_transform().get_euler();
+       value = Vector3(euler.x.radians(), euler.y.radians(), euler.z.radians());
+       return true;
 }
 
-Matrix Animation::MatrixInterpolation::get(float t) const
+bool Animation::extract_scale(const KeyFrame &kf, Vector3 &value)
 {
-       float u = t*2.0f-1.0f;
+       value = kf.get_transform().get_scale();
+       return true;
+}
 
-       float matrix[16];
-       for(unsigned i=0; i<4; ++i)
-       {
-               const float *m1_col = matrix1->data()+i*4;
-               const float *m2_col = matrix2->data()+i*4;
-               float *out_col = matrix+i*4;
+void Animation::add_event(const Time::TimeDelta &t, const string &n, const Variant &v)
+{
+       Event event;
+       event.time = t;
+       event.name = n;
+       event.value = v;
+       events.push_back(event);
+}
 
-               if(i<3)
-               {
-                       /* Linear interpolation will produce vectors that fall on the line
-                       between the two endpoints, and has a higher angular velocity near the
-                       middle.  We compensate for the velocity by interpolating the angle
-                       around the halfway point and computing its tangent.  This is
-                       approximated by a third degree polynomial, scaled so that the result
-                       will be in the range [-1, 1]. */
-                       float w = (axes[i].slope+(1-axes[i].slope)*u*u)*u*0.5f+0.5f;
-
-                       /* The interpolated vectors will also be shorter than unit length.  At
-                       the halfway point the length will be equal to the cosine of half the
-                       angle, which was computed earlier.  Use a second degree polynomial to
-                       approximate. */
-                       float n = (axes[i].scale+(1-axes[i].scale)*u*u);
-
-                       for(unsigned j=0; j<3; ++j)
-                               out_col[j] = ((1-w)*m1_col[j]+w*m2_col[j])/n;
-               }
-               else
-               {
-                       for(unsigned j=0; j<3; ++j)
-                               out_col[j] = (1-t)*m1_col[j]+t*m2_col[j];
-               }
-       }
+const Time::TimeDelta &Animation::get_duration() const
+{
+       if(keyframes.empty())
+               return Time::zero;
 
-       matrix[3] = 0;
-       matrix[7] = 0;
-       matrix[11] = 0;
-       matrix[15] = 1;
+       return keyframes.back().time;
+}
 
-       return matrix;
+void Animation::set_looping(bool l)
+{
+       looping = l;
 }
 
 
-Animation::TimedKeyFrame::TimedKeyFrame(const Animation &a):
-       animation(a),
-       prev(0)
+Animation::Curve::Curve(CurveTarget t):
+       target(t)
 { }
 
-void Animation::TimedKeyFrame::prepare()
+
+template<unsigned N>
+Animation::ValueCurve<N>::ValueCurve(CurveTarget t, const vector<Knot> &k):
+       Curve(t),
+       spline(Interpolate::BezierSpline<float, 3, N>(k))
+{ }
+
+template<unsigned N>
+void Animation::ValueCurve<N>::apply(float, Matrix &) const
 {
-       const KeyFrame::UniformMap &kf_uniforms = keyframe->get_uniforms();
-       for(KeyFrame::UniformMap::const_iterator i=kf_uniforms.begin(); i!=kf_uniforms.end(); ++i)
-       {
-               unsigned j = animation.get_slot_for_uniform(i->first);
-               uniforms.reserve(j+1);
-               for(unsigned k=uniforms.size(); k<=j; ++k)
-                       uniforms.push_back(KeyFrame::AnimatedUniform(animation.uniforms[k].size, 0.0f));
+       throw invalid_operation("ValueCurve::apply");
+}
 
-               uniforms[j] = i->second;
+template<>
+void Animation::ValueCurve<3>::apply(float x, Matrix &matrix) const
+{
+       Vector3 value = spline(x);
+       if(target==POSITION)
+               matrix.translate(value);
+       else if(target==EULER)
+       {
+               matrix.rotate(value.z, Vector3(0, 0, 1));
+               matrix.rotate(value.y, Vector3(0, 1, 0));
+               matrix.rotate(value.x, Vector3(1, 0, 0));
        }
+       else if(target==SCALE)
+               matrix.scale(value);
+       else
+               throw invalid_operation("ValueCurve::apply");
+}
 
-       if(!prev)
-               return;
+template<unsigned N>
+void Animation::ValueCurve<N>::apply(float x, KeyFrame::AnimatedUniform &uni) const
+{
+       uni.size = N;
+       typename Interpolate::Spline<float, 3, N>::Value value = spline(x);
+       for(unsigned i=0; i<N; ++i)
+               uni.values[i] = Interpolate::SplineValue<float, N>::get(value, i);
+}
 
-       delta_t = time-prev->time;
-       matrix = MatrixInterpolation(prev->keyframe->get_matrix(), keyframe->get_matrix());
 
-       if(animation.armature)
-       {
-               unsigned max_index = animation.armature->get_max_link_index();
-               pose_matrices.resize(max_index+1);
-               const Pose *pose1 = prev->keyframe->get_pose();
-               const Pose *pose2 = keyframe->get_pose();
-               static Matrix identity;
-               for(unsigned i=0; i<=max_index; ++i)
-               {
-                       const Matrix &matrix1 = (pose1 ? pose1->get_link_matrix(i) : identity);
-                       const Matrix &matrix2 = (pose2 ? pose2->get_link_matrix(i) : identity);
-                       pose_matrices[i] = MatrixInterpolation(matrix1, matrix2);
-               }
-       }
+template<unsigned N>
+bool Animation::ExtractUniform<N>::operator()(const KeyFrame &kf, typename Interpolate::SplineValue<float, N>::Type &value) const
+{
+       const KeyFrame::UniformMap &kf_uniforms = kf.get_uniforms();
+       const KeyFrame::UniformMap::const_iterator i = kf_uniforms.find(name);
+       if(i==kf_uniforms.end())
+               return false;
+
+       value = Interpolate::SplineValue<float, N>::make(i->second.values);
+       return true;
 }
 
 
@@ -219,60 +290,54 @@ Animation::UniformInfo::UniformInfo(const string &n, unsigned s):
 
 Animation::Iterator::Iterator(const Animation &a):
        animation(&a),
-       iter(animation->keyframes.begin()),
+       event_iter(animation->events.begin()),
        end(false)
-{ }
+{
+}
 
 Animation::Iterator &Animation::Iterator::operator+=(const Time::TimeDelta &t)
 {
-       time_since_keyframe += t;
-       while(time_since_keyframe>iter->delta_t)
-       {
-               vector<TimedKeyFrame>::const_iterator next = iter;
-               ++next;
-               if(next==animation->keyframes.end())
-               {
-                       if(animation->looping)
-                               next = animation->keyframes.begin();
-                       else
-                       {
-                               end = true;
-                               time_since_keyframe = iter->delta_t;
-                               break;
-                       }
-               }
+       const Time::TimeDelta &duration = animation->get_duration();
+       if(!duration)
+               return *this;
 
-               time_since_keyframe -= iter->delta_t;
-               iter = next;
+       elapsed += t;
+       if(animation->looping)
+       {
+               while(elapsed>=duration)
+                       elapsed -= duration;
+       }
+       else if(elapsed>=duration)
+       {
+               end = true;
+               elapsed = duration;
        }
 
        return *this;
 }
 
-Matrix Animation::Iterator::get_matrix() const
+void Animation::Iterator::dispatch_events(AnimationEventObserver &observer)
 {
-       if(!iter->prev)
-               return iter->keyframe->get_matrix();
+       for(; (event_iter!=animation->events.end() && event_iter->time<=elapsed); ++event_iter)
+               observer.animation_event(0, event_iter->name, event_iter->value);
+}
 
-       return iter->matrix.get(time_since_keyframe/iter->delta_t);
+Matrix Animation::Iterator::get_matrix() const
+{
+       Matrix matrix;
+       for(unsigned i=0; i<3; ++i)
+               animation->curves[i]->apply(elapsed/Time::sec, matrix);
+       return matrix;
 }
 
 KeyFrame::AnimatedUniform Animation::Iterator::get_uniform(unsigned i) const
 {
-       if(!iter->prev)
-       {
-               if(iter->uniforms.size()>i)
-                       return iter->uniforms[i];
-               else
-                       return KeyFrame::AnimatedUniform(animation->uniforms[i].size, 0.0f);
-       }
+       if(i>=animation->uniforms.size())
+               throw out_of_range("Animation::Iterator::get_uniform");
 
-       unsigned size = animation->uniforms[i].size;
-       float t = time_since_keyframe/iter->delta_t;
-       KeyFrame::AnimatedUniform result(size, 0.0f);
-       for(unsigned j=0; j<size; ++j)
-               result.values[j] = iter->prev->uniforms[i].values[j]*(1-t)+iter->uniforms[i].values[j]*t;
-       return result;
+       KeyFrame::AnimatedUniform uni(animation->uniforms[i].size, 0.0f);
+       animation->curves[3+i]->apply(elapsed/Time::sec, uni);
+       return uni;
 }
 
 Matrix Animation::Iterator::get_pose_matrix(unsigned link) const
@@ -282,21 +347,7 @@ Matrix Animation::Iterator::get_pose_matrix(unsigned link) const
        if(link>animation->armature->get_max_link_index())
                throw out_of_range("Animation::Iterator::get_pose_matrix");
 
-       if(!iter->prev)
-       {
-               if(const Pose *pose = iter->keyframe->get_pose())
-                       return pose->get_link_matrix(link);
-               else
-                       return Matrix();
-       }
-
-       // We must redo the base point correction since interpolation throws if off
-       // XXX This should probably be done on local matrices
-       Matrix result = iter->pose_matrices[link].get(time_since_keyframe/iter->delta_t);
-       const Vector3 &base = animation->armature->get_link(link).get_base();
-       Vector3 new_base = result*base;
-       result = Matrix::translation(base-new_base)*result;
-       return result;
+       throw logic_error("pose animations are currently unimplemented");
 }
 
 
@@ -314,19 +365,35 @@ Animation::Loader::Loader(Animation &a, Collection &c):
 
 void Animation::Loader::init()
 {
+       start_slope = 1;
+       end_slope = 1;
        add("armature", &Animation::armature);
+       add("control_keyframe", &Loader::control_keyframe);
+       add("control_keyframe", &Loader::control_keyframe_inline);
+       add("event", &Loader::event);
+       add("event", &Loader::event1i);
+       add("event", &Loader::event1f);
+       add("event", &Loader::event2f);
+       add("event", &Loader::event3f);
+       add("event", &Loader::event4f);
        add("interval", &Loader::interval);
        add("keyframe", &Loader::keyframe);
        add("keyframe", &Loader::keyframe_inline);
        add("looping", &Animation::looping);
+       add("slopes", &Loader::slopes);
 }
 
-void Animation::Loader::keyframe(const string &n)
+void Animation::Loader::finish()
 {
-       obj.add_keyframe(current_time, get_collection().get<KeyFrame>(n));
+       obj.create_curves();
 }
 
-void Animation::Loader::keyframe_inline()
+void Animation::Loader::load_kf(const string &n, bool c)
+{
+       obj.add_keyframe(current_time, &get_collection().get<KeyFrame>(n), c, false);
+}
+
+void Animation::Loader::load_kf_inline(bool c)
 {
        RefPtr<KeyFrame> kf = new KeyFrame;
        if(coll)
@@ -334,11 +401,48 @@ void Animation::Loader::keyframe_inline()
        else
                load_sub(*kf);
 
-       TimedKeyFrame tkf(obj);
-       tkf.time = current_time;
-       tkf.keyframe = kf;
-       obj.prepare_keyframe(tkf);
-       obj.keyframes.push_back(tkf);
+       obj.add_keyframe(current_time, *kf, c, true);
+       kf.release();
+}
+
+void Animation::Loader::control_keyframe(const string &n)
+{
+       load_kf(n, true);
+}
+
+void Animation::Loader::control_keyframe_inline()
+{
+       load_kf_inline(true);
+}
+
+void Animation::Loader::event(const string &n)
+{
+       obj.add_event(current_time, n);
+}
+
+void Animation::Loader::event1i(const string &n, int v)
+{
+       obj.add_event(current_time, n, v);
+}
+
+void Animation::Loader::event1f(const string &n, float v)
+{
+       obj.add_event(current_time, n, v);
+}
+
+void Animation::Loader::event2f(const string &n, float v0, float v1)
+{
+       obj.add_event(current_time, n, LinAl::Vector<float, 2>(v0, v1));
+}
+
+void Animation::Loader::event3f(const string &n, float v0, float v1, float v2)
+{
+       obj.add_event(current_time, n, Vector3(v0, v1, v2));
+}
+
+void Animation::Loader::event4f(const string &n, float v0, float v1, float v2, float v3)
+{
+       obj.add_event(current_time, n, Vector4(v0, v1, v2, v3));
 }
 
 void Animation::Loader::interval(float t)
@@ -346,5 +450,21 @@ void Animation::Loader::interval(float t)
        current_time += t*Time::sec;
 }
 
+void Animation::Loader::keyframe(const string &n)
+{
+       load_kf(n, false);
+}
+
+void Animation::Loader::keyframe_inline()
+{
+       load_kf_inline(false);
+}
+
+void Animation::Loader::slopes(float s, float e)
+{
+       start_slope = s;
+       end_slope = e;
+}
+
 } // namespace GL
 } // namespace Msp