source/animation/animation.cpp

   1 #include <cmath>
   2 #include <msp/core/maputils.h>
   3 #include <msp/datafile/collection.h>
   4 #include <msp/interpolate/bezierspline.h>
   5 #include "animation.h"
   6 #include "animationeventobserver.h"
   7 #include "armature.h"
   8 #include "error.h"
   9 #include "pose.h"
  10
  11 using namespace std;
  12
  13 namespace Msp {
  14 namespace GL {
  15
  16 Animation::Animation():
  17         armature(0),
  18         looping(false)
  19 { }
  20
  21 Animation::~Animation()
  22 {
  23         for(vector<Curve *>::iterator i=curves.begin(); i!=curves.end(); ++i)
  24                 delete *i;
  25 }
  26
  27 void Animation::set_armature(const Armature &a)
  28 {
  29         if(!keyframes.empty() && &a!=armature)
  30                 throw invalid_operation("Animation::set_armature");
  31         armature = &a;
  32 }
  33
  34 unsigned Animation::get_slot_for_uniform(const string &n) const
  35 {
  36         for(unsigned i=0; i<uniforms.size(); ++i)
  37                 if(uniforms[i].name==n)
  38                         return i;
  39         throw key_error(n);
  40 }
  41
  42 const string &Animation::get_uniform_name(unsigned i) const
  43 {
  44         if(i>=uniforms.size())
  45                 throw out_of_range("Animation::get_uniform_name");
  46         return uniforms[i].name;
  47 }
  48
  49 void Animation::add_keyframe(const Time::TimeDelta &t, const KeyFrame &kf)
  50 {
  51         add_keyframe(t, &kf, false, false);
  52         create_curves();
  53 }
  54
  55 void Animation::add_keyframe_owned(const Time::TimeDelta &t, const KeyFrame *kf)
  56 {
  57         add_keyframe(t, kf, false, true);
  58         create_curves();
  59 }
  60
  61 void Animation::add_keyframe(const Time::TimeDelta &t, const KeyFrame &kf, float slope)
  62 {
  63         add_keyframe(t, &kf, slope, slope, false);
  64         create_curves();
  65 }
  66
  67 void Animation::add_keyframe(const Time::TimeDelta &t, const KeyFrame &kf, float ss, float es)
  68 {
  69         add_keyframe(t, &kf, ss, es, false);
  70         create_curves();
  71 }
  72
  73 void Animation::add_control_keyframe(const KeyFrame &kf)
  74 {
  75         if(keyframes.empty())
  76                 throw invalid_operation("Animation::add_control_keyframe");
  77
  78         add_keyframe(keyframes.back().time, &kf, true, false);
  79 }
  80
  81 void Animation::add_control_keyframe_owned(const KeyFrame *kf)
  82 {
  83         if(keyframes.empty())
  84                 throw invalid_operation("Animation::add_control_keyframe_owned");
  85
  86         add_keyframe(keyframes.back().time, kf, true, true);
  87 }
  88
  89 void Animation::add_keyframe(const Time::TimeDelta &t, const KeyFrame *kf, float ss, float es, bool owned)
  90 {
  91         if(keyframes.empty())
  92                 return add_keyframe(t, kf, false, owned);
  93
  94         if(keyframes.back().control)
  95                 throw invalid_operation("Animation::add_keyframe");
  96
  97         const KeyFrame &last = *keyframes.back().keyframe;
  98         const Transform &trn = kf->get_transform();
  99         const Transform &last_trn = last.get_transform();
 100         const KeyFrame::UniformMap &kf_unis = kf->get_uniforms();
 101         const KeyFrame::UniformMap &last_unis = last.get_uniforms();
 102         for(unsigned i=1; i<=2; ++i)
 103         {
 104                 float x = (i==1 ? ss/3 : 1-es/3);
 105                 KeyFrame *ckf = new KeyFrame;
 106                 Transform ctrn;
 107                 ctrn.set_position(last_trn.get_position()*(1-x)+trn.get_position()*x);
 108                 const Transform::AngleVector3 &e1 = last_trn.get_euler();
 109                 const Transform::AngleVector3 &e2 = trn.get_euler();
 110                 ctrn.set_euler(Transform::AngleVector3(e1.x*(1-x)+e2.x*x, e1.y*(1-x)+e2.y*x, e1.z*(1-x)+e2.z*x));
 111                 ctrn.set_scale(last_trn.get_scale()*(1-x)+trn.get_scale()*x);
 112                 ckf->set_transform(ctrn);
 113
 114                 for(KeyFrame::UniformMap::const_iterator j=kf_unis.begin(); j!=kf_unis.end(); ++j)
 115                 {
 116                         KeyFrame::UniformMap::const_iterator k = last_unis.find(j->first);
 117                         if(k==last_unis.end())
 118                                 continue;
 119
 120                         KeyFrame::AnimatedUniform uni(j->second.size, 0.0f);
 121                         for(unsigned c=0; c<uni.size; ++c)
 122                                 uni.values[c] = k->second.values[c]*(1-x)+j->second.values[c]*x;
 123
 124                         ckf->set_uniform(j->first, uni);
 125                 }
 126
 127                 add_keyframe(t, ckf, true, true);
 128         }
 129
 130         add_keyframe(t, kf, false, owned);
 131 }
 132
 133 void Animation::add_keyframe(const Time::TimeDelta &t, const KeyFrame *kf, bool c, bool owned)
 134 {
 135         if(c && keyframes.empty())
 136                 throw invalid_argument("Animation::add_keyframe");
 137         if(keyframes.empty() && t!=Time::zero)
 138                 throw invalid_argument("Animation::add_keyframe");
 139         if(!keyframes.empty() && t<keyframes.back().time)
 140                 throw invalid_argument("Animation::add_keyframe");
 141         if(kf->get_pose() && armature && kf->get_pose()->get_armature()!=armature)
 142                 throw invalid_argument("Animation::add_keyframe");
 143
 144         const KeyFrame::UniformMap &kf_uniforms = kf->get_uniforms();
 145         for(vector<UniformInfo>::const_iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
 146         {
 147                 KeyFrame::UniformMap::const_iterator j = kf_uniforms.find(i->name);
 148                 if(j!=kf_uniforms.end() && j->second.size!=i->size)
 149                         throw invalid_argument("Animation::add_keyframe");
 150         }
 151
 152         if(kf->get_pose() && !armature)
 153                 armature = kf->get_pose()->get_armature();
 154
 155         TimedKeyFrame tkf;
 156         tkf.time = t;
 157         tkf.keyframe = kf;
 158         if(!owned)
 159                 tkf.keyframe.keep();
 160         tkf.control = c;
 161
 162         keyframes.push_back(tkf);
 163
 164         for(KeyFrame::UniformMap::const_iterator i=kf_uniforms.begin(); i!=kf_uniforms.end(); ++i)
 165         {
 166                 bool found = false;
 167                 for(vector<UniformInfo>::const_iterator j=uniforms.begin(); (!found && j!=uniforms.end()); ++j)
 168                         found = (j->name==i->first);
 169
 170                 if(!found)
 171                         uniforms.push_back(UniformInfo(i->first, i->second.size));
 172         }
 173 }
 174
 175 void Animation::create_curves()
 176 {
 177         for(vector<Curve *>::iterator i=curves.begin(); i!=curves.end(); ++i)
 178                 delete *i;
 179         curves.clear();
 180
 181         curves.reserve(6+uniforms.size());
 182         create_curve(POSITION, Transform::POSITION, &extract_position);
 183         create_curve(EULER, Transform::EULER, &extract_euler);
 184         create_curve(SCALE, Transform::SCALE, &extract_scale);
 185
 186         uniform_curve_offset = curves.size();
 187         for(vector<UniformInfo>::const_iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
 188         {
 189                 if(i->size==1)
 190                         create_curve<1>(UNIFORM, -1, ExtractUniform<1>(i->name));
 191                 else if(i->size==2)
 192                         create_curve<2>(UNIFORM, -1, ExtractUniform<2>(i->name));
 193                 else if(i->size==3)
 194                         create_curve<3>(UNIFORM, -1, ExtractUniform<3>(i->name));
 195                 else if(i->size==4)
 196                         create_curve<4>(UNIFORM, -1, ExtractUniform<4>(i->name));
 197         }
 198 }
 199
 200 void Animation::create_curve(CurveTarget target, Transform::ComponentMask mask, ExtractComponent::Extract extract)
 201 {
 202         Transform::ComponentMask all = mask;
 203         Transform::ComponentMask any = Transform::NONE;
 204         for(vector<TimedKeyFrame>::const_iterator i=keyframes.begin(); i!=keyframes.end(); ++i)
 205         {
 206                 all = all&i->keyframe->get_transform().get_mask();
 207                 any = any|i->keyframe->get_transform().get_mask();
 208         }
 209
 210         if(all==mask)
 211                 create_curve<3>(target, -1, extract);
 212         else if(any&mask)
 213         {
 214                 unsigned low_bit = mask&(mask>>2);
 215                 for(unsigned i=3; i-->0; )
 216                 {
 217                         Transform::ComponentMask bit = static_cast<Transform::ComponentMask>(low_bit<<i);
 218                         if(any&bit)
 219                                 create_curve<1>(target, i, ExtractComponent(extract, i, bit));
 220                 }
 221         }
 222 }
 223
 224 template<unsigned N, typename T>
 225 void Animation::create_curve(CurveTarget target, int component, const T &extract)
 226 {
 227         typedef typename ValueCurve<N>::Knot Knot;
 228
 229         vector<Knot> knots;
 230         unsigned n_control = 0;
 231         for(vector<TimedKeyFrame>::const_iterator i=keyframes.begin(); i!=keyframes.end(); ++i)
 232         {
 233                 if(i->control && knots.empty())
 234                         continue;
 235
 236                 typename Interpolate::SplineValue<float, N>::Type value;
 237                 if(extract(*i->keyframe, value))
 238                 {
 239                         typename Knot::Value dvalue = value;
 240                         float x = i->time/Time::sec;
 241                         if(i->control)
 242                         {
 243                                 ++n_control;
 244                                 if(n_control>2)
 245                                         throw logic_error("too many control keyframes");
 246                         }
 247                         else
 248                         {
 249                                 if(n_control==1)
 250                                 {
 251                                         typename Knot::Value cv = knots.back().y;
 252                                         knots.back().y = (knots[knots.size()-2].y+cv*2.0)/3.0;
 253                                         knots.push_back(Knot(x, (dvalue+cv*2.0)/3.0));
 254                                 }
 255                                 else if(n_control==0 && !knots.empty())
 256                                 {
 257                                         typename Knot::Value prev = knots.back().y;
 258                                         knots.push_back(Knot(knots.back().x, (prev*2.0+dvalue)/3.0));
 259                                         knots.push_back(Knot(x, (prev+dvalue*2.0)/3.0));
 260                                 }
 261                                 n_control = 0;
 262                         }
 263                         knots.push_back(Knot(x, value));
 264                 }
 265         }
 266
 267         while(n_control--)
 268                 knots.pop_back();
 269
 270         if(knots.size()==1)
 271         {
 272                 knots.push_back(knots.back());
 273                 knots.push_back(knots.back());
 274                 knots.back().x += 1;
 275                 knots.push_back(knots.back());
 276         }
 277
 278         curves.push_back(new ValueCurve<N>(target, component, knots));
 279 }
 280
 281 bool Animation::extract_position(const KeyFrame &kf, Vector3 &value)
 282 {
 283         value = kf.get_transform().get_position();
 284         return true;
 285 }
 286
 287 bool Animation::extract_euler(const KeyFrame &kf, Vector3 &value)
 288 {
 289         const Transform::AngleVector3 &euler = kf.get_transform().get_euler();
 290         value = Vector3(euler.x.radians(), euler.y.radians(), euler.z.radians());
 291         return true;
 292 }
 293
 294 bool Animation::extract_scale(const KeyFrame &kf, Vector3 &value)
 295 {
 296         value = kf.get_transform().get_scale();
 297         return true;
 298 }
 299
 300 void Animation::add_event(const Time::TimeDelta &t, const string &n, const Variant &v)
 301 {
 302         Event event;
 303         event.time = t;
 304         event.name = n;
 305         event.value = v;
 306         events.push_back(event);
 307 }
 308
 309 const Time::TimeDelta &Animation::get_duration() const
 310 {
 311         if(keyframes.empty())
 312                 return Time::zero;
 313
 314         return keyframes.back().time;
 315 }
 316
 317 void Animation::set_looping(bool l)
 318 {
 319         looping = l;
 320 }
 321
 322
 323 Animation::Curve::Curve(CurveTarget t, int c):
 324         target(t),
 325         component(c)
 326 { }
 327
 328
 329 template<unsigned N>
 330 Animation::ValueCurve<N>::ValueCurve(CurveTarget t, int c, const vector<Knot> &k):
 331         Curve(t, c),
 332         spline(Interpolate::BezierSpline<double, 3, N>(k))
 333 { }
 334
 335 template<unsigned N>
 336 void Animation::ValueCurve<N>::apply(float, Matrix &) const
 337 {
 338         throw invalid_operation("ValueCurve::apply");
 339 }
 340
 341 template<>
 342 void Animation::ValueCurve<1>::apply(float x, Matrix &matrix) const
 343 {
 344         float value = spline(x);
 345         if(target==POSITION || target==SCALE)
 346         {
 347                 if(target==POSITION)
 348                 {
 349                         Vector3 vec;
 350                         vec[component] = value;
 351                         matrix.translate(vec);
 352                 }
 353                 else
 354                 {
 355                         Vector3 vec(1.0f, 1.0f, 1.0f);
 356                         vec[component] = value;
 357                         matrix.scale(vec);
 358                 }
 359         }
 360         else if(target==EULER)
 361         {
 362                 Vector3 vec;
 363                 vec[component] = 1.0f;
 364                 matrix.rotate(Geometry::Angle<float>::from_radians(value), vec);
 365         }
 366         else
 367                 throw invalid_operation("ValueCurve::apply");
 368 }
 369
 370 template<>
 371 void Animation::ValueCurve<3>::apply(float x, Matrix &matrix) const
 372 {
 373         Vector3 value = spline(x);
 374         if(target==POSITION)
 375                 matrix.translate(value);
 376         else if(target==EULER)
 377         {
 378                 matrix.rotate(Geometry::Angle<float>::from_radians(value.z), Vector3(0, 0, 1));
 379                 matrix.rotate(Geometry::Angle<float>::from_radians(value.y), Vector3(0, 1, 0));
 380                 matrix.rotate(Geometry::Angle<float>::from_radians(value.x), Vector3(1, 0, 0));
 381         }
 382         else if(target==SCALE)
 383                 matrix.scale(value);
 384         else
 385                 throw invalid_operation("ValueCurve::apply");
 386 }
 387
 388 template<unsigned N>
 389 void Animation::ValueCurve<N>::apply(float x, KeyFrame::AnimatedUniform &uni) const
 390 {
 391         uni.size = N;
 392         typename Interpolate::Spline<double, 3, N>::Value value = spline(x);
 393         for(unsigned i=0; i<N; ++i)
 394                 uni.values[i] = Interpolate::SplineValue<double, N>::get(value, i);
 395 }
 396
 397
 398 bool Animation::ExtractComponent::operator()(const KeyFrame &kf, float &value) const
 399 {
 400         Vector3 vec;
 401         if(!extract(kf, vec))
 402                 return false;
 403
 404         value = vec[index];
 405         return kf.get_transform().get_mask()&mask;
 406 }
 407
 408
 409 template<unsigned N>
 410 bool Animation::ExtractUniform<N>::operator()(const KeyFrame &kf, typename Interpolate::SplineValue<float, N>::Type &value) const
 411 {
 412         const KeyFrame::UniformMap &kf_uniforms = kf.get_uniforms();
 413         const KeyFrame::UniformMap::const_iterator i = kf_uniforms.find(name);
 414         if(i==kf_uniforms.end())
 415                 return false;
 416
 417         value = Interpolate::SplineValue<float, N>::make(i->second.values);
 418         return true;
 419 }
 420
 421
 422 Animation::UniformInfo::UniformInfo(const string &n, unsigned s):
 423         name(n),
 424         size(s)
 425 { }
 426
 427
 428 Animation::Iterator::Iterator(const Animation &a):
 429         animation(&a),
 430         event_iter(animation->events.begin()),
 431         end(false)
 432 {
 433 }
 434
 435 Animation::Iterator &Animation::Iterator::operator+=(const Time::TimeDelta &t)
 436 {
 437         const Time::TimeDelta &duration = animation->get_duration();
 438         if(!duration)
 439                 return *this;
 440
 441         elapsed += t;
 442         if(animation->looping)
 443         {
 444                 while(elapsed>=duration)
 445                         elapsed -= duration;
 446         }
 447         else if(elapsed>=duration)
 448         {
 449                 end = true;
 450                 elapsed = duration;
 451         }
 452
 453         return *this;
 454 }
 455
 456 void Animation::Iterator::dispatch_events(AnimationEventObserver &observer)
 457 {
 458         for(; (event_iter!=animation->events.end() && event_iter->time<=elapsed); ++event_iter)
 459                 observer.animation_event(0, event_iter->name, event_iter->value);
 460 }
 461
 462 Matrix Animation::Iterator::get_matrix() const
 463 {
 464         Matrix matrix;
 465         for(unsigned i=0; i<animation->uniform_curve_offset; ++i)
 466                 animation->curves[i]->apply(elapsed/Time::sec, matrix);
 467         return matrix;
 468 }
 469
 470 KeyFrame::AnimatedUniform Animation::Iterator::get_uniform(unsigned i) const
 471 {
 472         if(i>=animation->uniforms.size())
 473                 throw out_of_range("Animation::Iterator::get_uniform");
 474
 475         KeyFrame::AnimatedUniform uni(animation->uniforms[i].size, 0.0f);
 476         animation->curves[animation->uniform_curve_offset+i]->apply(elapsed/Time::sec, uni);
 477         return uni;
 478 }
 479
 480 Matrix Animation::Iterator::get_pose_matrix(unsigned link) const
 481 {
 482         if(!animation->armature)
 483                 throw invalid_operation("Animation::Iterator::get_pose_matrix");
 484         if(link>animation->armature->get_max_link_index())
 485                 throw out_of_range("Animation::Iterator::get_pose_matrix");
 486
 487         throw logic_error("pose animations are currently unimplemented");
 488 }
 489
 490
 491 Animation::Loader::Loader(Animation &a):
 492         DataFile::CollectionObjectLoader<Animation>(a, 0)
 493 {
 494         init();
 495 }
 496
 497 Animation::Loader::Loader(Animation &a, Collection &c):
 498         DataFile::CollectionObjectLoader<Animation>(a, &c)
 499 {
 500         init();
 501 }
 502
 503 void Animation::Loader::init()
 504 {
 505         start_slope = 1;
 506         end_slope = 1;
 507         slopes_set = 0;
 508         add("armature", &Animation::armature);
 509         add("control_keyframe", &Loader::control_keyframe);
 510         add("control_keyframe", &Loader::control_keyframe_inline);
 511         add("event", &Loader::event);
 512         add("event", &Loader::event1i);
 513         add("event", &Loader::event1f);
 514         add("event", &Loader::event2f);
 515         add("event", &Loader::event3f);
 516         add("event", &Loader::event4f);
 517         add("interval", &Loader::interval);
 518         add("keyframe", &Loader::keyframe);
 519         add("keyframe", &Loader::keyframe_inline);
 520         add("looping", &Animation::looping);
 521         add("slopes", &Loader::slopes);
 522 }
 523
 524 void Animation::Loader::finish()
 525 {
 526         obj.create_curves();
 527 }
 528
 529 void Animation::Loader::check_slopes_and_control(bool s, bool c)
 530 {
 531         if(s && c)
 532                 throw logic_error("can't use both slopes and control keyframes in same segment");
 533 }
 534
 535 void Animation::Loader::add_kf(const KeyFrame *kf, bool c, bool owned)
 536 {
 537         if(slopes_set && !c)
 538                 obj.add_keyframe(current_time, kf, start_slope, end_slope, owned);
 539         else
 540                 obj.add_keyframe(current_time, kf, c, owned);
 541
 542         start_slope = end_slope;
 543         end_slope = 1;
 544         slopes_set = (slopes_set<<1)&3;
 545 }
 546
 547 void Animation::Loader::load_kf(const string &n, bool c)
 548 {
 549         add_kf(&get_collection().get<KeyFrame>(n), c, false);
 550 }
 551
 552 void Animation::Loader::load_kf_inline(bool c)
 553 {
 554         RefPtr<KeyFrame> kf = new KeyFrame;
 555         if(coll)
 556                 load_sub(*kf, get_collection());
 557         else
 558                 load_sub(*kf);
 559
 560         add_kf(kf.get(), c, true);
 561         kf.release();
 562 }
 563
 564 void Animation::Loader::control_keyframe(const string &n)
 565 {
 566         slopes_set &= 1;
 567         check_slopes_and_control(slopes_set, true);
 568         load_kf(n, true);
 569 }
 570
 571 void Animation::Loader::control_keyframe_inline()
 572 {
 573         slopes_set &= 1;
 574         check_slopes_and_control(slopes_set, true);
 575         load_kf_inline(true);
 576 }
 577
 578 void Animation::Loader::event(const string &n)
 579 {
 580         obj.add_event(current_time, n);
 581 }
 582
 583 void Animation::Loader::event1i(const string &n, int v)
 584 {
 585         obj.add_event(current_time, n, v);
 586 }
 587
 588 void Animation::Loader::event1f(const string &n, float v)
 589 {
 590         obj.add_event(current_time, n, v);
 591 }
 592
 593 void Animation::Loader::event2f(const string &n, float v0, float v1)
 594 {
 595         obj.add_event(current_time, n, LinAl::Vector<float, 2>(v0, v1));
 596 }
 597
 598 void Animation::Loader::event3f(const string &n, float v0, float v1, float v2)
 599 {
 600         obj.add_event(current_time, n, Vector3(v0, v1, v2));
 601 }
 602
 603 void Animation::Loader::event4f(const string &n, float v0, float v1, float v2, float v3)
 604 {
 605         obj.add_event(current_time, n, Vector4(v0, v1, v2, v3));
 606 }
 607
 608 void Animation::Loader::interval(float t)
 609 {
 610         current_time += t*Time::sec;
 611 }
 612
 613 void Animation::Loader::keyframe(const string &n)
 614 {
 615         load_kf(n, false);
 616 }
 617
 618 void Animation::Loader::keyframe_inline()
 619 {
 620         load_kf_inline(false);
 621 }
 622
 623 void Animation::Loader::slopes(float s, float e)
 624 {
 625         check_slopes_and_control(true, (!obj.keyframes.empty() && obj.keyframes.back().control));
 626
 627         start_slope = s;
 628         end_slope = e;
 629         slopes_set = 1;
 630 }
 631
 632 } // namespace GL
 633 } // namespace Msp