]> git.tdb.fi Git - libs/gl.git/blob - source/animation/animation.cpp
Clear the empty name flag after emitting a member access
[libs/gl.git] / 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