Depsgraph: remove EvaluationContext, pass Depsgraph instead.
[blender.git] / source / blender / blenkernel / intern / anim.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): none yet.
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/blenkernel/intern/anim.c
29  *  \ingroup bke
30  */
31
32 #include "MEM_guardedalloc.h"
33
34 #include <stdlib.h>
35
36 #include "BLI_listbase.h"
37 #include "BLI_math.h"
38
39 #include "BLT_translation.h"
40
41 #include "DNA_anim_types.h"
42 #include "DNA_armature_types.h"
43 #include "DNA_key_types.h"
44 #include "DNA_scene_types.h"
45
46 #include "BKE_context.h"
47 #include "BKE_curve.h"
48 #include "BKE_global.h"
49 #include "BKE_key.h"
50 #include "BKE_main.h"
51 #include "BKE_object.h"
52 #include "BKE_particle.h"
53 #include "BKE_scene.h"
54 #include "BKE_anim.h"
55 #include "BKE_report.h"
56
57 #include "DEG_depsgraph.h"
58 #include "DEG_depsgraph_build.h"
59
60 // XXX bad level call...
61
62 /* --------------------- */
63 /* forward declarations */
64
65 /* ******************************************************************** */
66 /* Animation Visualization */
67
68 /* Initialize the default settings for animation visualization */
69 void animviz_settings_init(bAnimVizSettings *avs)
70 {
71         /* sanity check */
72         if (avs == NULL)
73                 return;
74
75         /* ghosting settings */
76         avs->ghost_bc = avs->ghost_ac = 10;
77
78         avs->ghost_sf = 1; /* xxx - take from scene instead? */
79         avs->ghost_ef = 250; /* xxx - take from scene instead? */
80
81         avs->ghost_step = 1;
82
83
84         /* path settings */
85         avs->path_bc = avs->path_ac = 10;
86
87         avs->path_sf = 1; /* xxx - take from scene instead? */
88         avs->path_ef = 250; /* xxx - take from scene instead? */
89
90         avs->path_viewflag = (MOTIONPATH_VIEW_KFRAS | MOTIONPATH_VIEW_KFNOS);
91
92         avs->path_step = 1;
93 }
94
95 /* ------------------- */
96
97 /* Free the given motion path's cache */
98 void animviz_free_motionpath_cache(bMotionPath *mpath) 
99 {
100         /* sanity check */
101         if (mpath == NULL) 
102                 return;
103                 
104         /* free the path if necessary */
105         if (mpath->points)
106                 MEM_freeN(mpath->points);
107         
108         /* reset the relevant parameters */
109         mpath->points = NULL;
110         mpath->length = 0;
111 }
112
113 /* Free the given motion path instance and its data 
114  * NOTE: this frees the motion path given!
115  */
116 void animviz_free_motionpath(bMotionPath *mpath)
117 {
118         /* sanity check */
119         if (mpath == NULL)
120                 return;
121         
122         /* free the cache first */
123         animviz_free_motionpath_cache(mpath);
124         
125         /* now the instance itself */
126         MEM_freeN(mpath);
127 }
128
129 /* ------------------- */
130
131 /**
132  * Setup motion paths for the given data.
133  * \note Only used when explicitly calculating paths on bones which may/may not be consider already
134  *
135  * \param scene Current scene (for frame ranges, etc.)
136  * \param ob Object to add paths for (must be provided)
137  * \param pchan Posechannel to add paths for (optional; if not provided, object-paths are assumed)
138  */
139 bMotionPath *animviz_verify_motionpaths(ReportList *reports, Scene *scene, Object *ob, bPoseChannel *pchan)
140 {
141         bAnimVizSettings *avs;
142         bMotionPath *mpath, **dst;
143         
144         /* sanity checks */
145         if (ELEM(NULL, scene, ob))
146                 return NULL;
147                 
148         /* get destination data */
149         if (pchan) {
150                 /* paths for posechannel - assume that posechannel belongs to the object */
151                 avs = &ob->pose->avs;
152                 dst = &pchan->mpath;
153         }
154         else {
155                 /* paths for object */
156                 avs = &ob->avs;
157                 dst = &ob->mpath;
158         }
159
160         /* avoid 0 size allocs */
161         if (avs->path_sf >= avs->path_ef) {
162                 BKE_reportf(reports, RPT_ERROR,
163                             "Motion path frame extents invalid for %s (%d to %d)%s",
164                             (pchan) ? pchan->name : ob->id.name,
165                             avs->path_sf, avs->path_ef,
166                             (avs->path_sf == avs->path_ef) ? TIP_(", cannot have single-frame paths") : "");
167                 return NULL;
168         }
169
170         /* if there is already a motionpath, just return that,
171          * provided it's settings are ok (saves extra free+alloc)
172          */
173         if (*dst != NULL) {
174                 int expected_length = avs->path_ef - avs->path_sf;
175                 
176                 mpath = *dst;
177                 
178                 /* path is "valid" if length is valid, but must also be of the same length as is being requested */
179                 if ((mpath->start_frame != mpath->end_frame) && (mpath->length > 0)) {
180                         /* outer check ensures that we have some curve data for this path */
181                         if (mpath->length == expected_length) {
182                                 /* return/use this as it is already valid length */
183                                 return mpath;
184                         }
185                         else {
186                                 /* clear the existing path (as the range has changed), and reallocate below */
187                                 animviz_free_motionpath_cache(mpath);
188                         }
189                 }
190         }
191         else {
192                 /* create a new motionpath, and assign it */
193                 mpath = MEM_callocN(sizeof(bMotionPath), "bMotionPath");
194                 *dst = mpath;
195         }
196         
197         /* set settings from the viz settings */
198         mpath->start_frame = avs->path_sf;
199         mpath->end_frame = avs->path_ef;
200         
201         mpath->length = mpath->end_frame - mpath->start_frame;
202         
203         if (avs->path_bakeflag & MOTIONPATH_BAKE_HEADS)
204                 mpath->flag |= MOTIONPATH_FLAG_BHEAD;
205         else
206                 mpath->flag &= ~MOTIONPATH_FLAG_BHEAD;
207
208         /* set default custom values */
209         mpath->color[0] = 1.0;    /* Red */
210         mpath->color[1] = 0.0;
211         mpath->color[2] = 0.0;
212
213         mpath->line_thickness = 1;
214         mpath->flag |= MOTIONPATH_FLAG_LINES;  /* draw lines by default */
215
216         /* allocate a cache */
217         mpath->points = MEM_callocN(sizeof(bMotionPathVert) * mpath->length, "bMotionPathVerts");
218         
219         /* tag viz settings as currently having some path(s) which use it */
220         avs->path_bakeflag |= MOTIONPATH_BAKE_HAS_PATHS;
221         
222         /* return it */
223         return mpath;
224 }
225
226 /* ------------------- */
227
228 /* Motion path needing to be baked (mpt) */
229 typedef struct MPathTarget {
230         struct MPathTarget *next, *prev;
231         
232         bMotionPath *mpath;         /* motion path in question */
233         
234         Object *ob;                 /* source object */
235         bPoseChannel *pchan;        /* source posechannel (if applicable) */
236 } MPathTarget;
237
238 /* ........ */
239
240 /* get list of motion paths to be baked for the given object
241  *  - assumes the given list is ready to be used
242  */
243 /* TODO: it would be nice in future to be able to update objects dependent on these bones too? */
244 void animviz_get_object_motionpaths(Object *ob, ListBase *targets)
245 {
246         MPathTarget *mpt;
247         
248         /* object itself first */
249         if ((ob->avs.recalc & ANIMVIZ_RECALC_PATHS) && (ob->mpath)) {
250                 /* new target for object */
251                 mpt = MEM_callocN(sizeof(MPathTarget), "MPathTarget Ob");
252                 BLI_addtail(targets, mpt);
253                 
254                 mpt->mpath = ob->mpath;
255                 mpt->ob = ob;
256         }
257         
258         /* bones */
259         if ((ob->pose) && (ob->pose->avs.recalc & ANIMVIZ_RECALC_PATHS)) {
260                 bArmature *arm = ob->data;
261                 bPoseChannel *pchan;
262                 
263                 for (pchan = ob->pose->chanbase.first; pchan; pchan = pchan->next) {
264                         if ((pchan->bone) && (arm->layer & pchan->bone->layer) && (pchan->mpath)) {
265                                 /* new target for bone */
266                                 mpt = MEM_callocN(sizeof(MPathTarget), "MPathTarget PoseBone");
267                                 BLI_addtail(targets, mpt);
268                                 
269                                 mpt->mpath = pchan->mpath;
270                                 mpt->ob = ob;
271                                 mpt->pchan = pchan;
272                         }
273                 }
274         }
275 }
276
277 /* ........ */
278
279 /* update scene for current frame */
280 static void motionpaths_calc_update_scene(Main *bmain,
281                                           struct Depsgraph *depsgraph)
282 {
283         /* Do all updates
284          *  - if this is too slow, resort to using a more efficient way
285          *    that doesn't force complete update, but for now, this is the
286          *    most accurate way!
287          *
288          * TODO(segey): Bring back partial updates, which became impossible
289          * with the new depsgraph due to unsorted nature of bases.
290          *
291          * TODO(sergey): Use evaluation context dedicated to motion paths.
292          */
293         BKE_scene_graph_update_for_newframe(depsgraph, bmain);
294 }
295
296 /* ........ */
297
298 /* perform baking for the targets on the current frame */
299 static void motionpaths_calc_bake_targets(Scene *scene, ListBase *targets)
300 {
301         MPathTarget *mpt;
302         
303         /* for each target, check if it can be baked on the current frame */
304         for (mpt = targets->first; mpt; mpt = mpt->next) {
305                 bMotionPath *mpath = mpt->mpath;
306                 bMotionPathVert *mpv;
307                 
308                 /* current frame must be within the range the cache works for 
309                  *      - is inclusive of the first frame, but not the last otherwise we get buffer overruns
310                  */
311                 if ((CFRA < mpath->start_frame) || (CFRA >= mpath->end_frame))
312                         continue;
313                 
314                 /* get the relevant cache vert to write to */
315                 mpv = mpath->points + (CFRA - mpath->start_frame);
316                 
317                 /* pose-channel or object path baking? */
318                 if (mpt->pchan) {
319                         /* heads or tails */
320                         if (mpath->flag & MOTIONPATH_FLAG_BHEAD) {
321                                 copy_v3_v3(mpv->co, mpt->pchan->pose_head);
322                         }
323                         else {
324                                 copy_v3_v3(mpv->co, mpt->pchan->pose_tail);
325                         }
326                         
327                         /* result must be in worldspace */
328                         mul_m4_v3(mpt->ob->obmat, mpv->co);
329                 }
330                 else {
331                         /* worldspace object location */
332                         copy_v3_v3(mpv->co, mpt->ob->obmat[3]);
333                 }
334         }
335 }
336
337 /* Perform baking of the given object's and/or its bones' transforms to motion paths 
338  *      - scene: current scene
339  *      - ob: object whose flagged motionpaths should get calculated
340  *      - recalc: whether we need to
341  */
342 /* TODO: include reports pointer? */
343  void animviz_calc_motionpaths(Depsgraph *depsgraph, Main *bmain, Scene *scene, ListBase *targets)
344 {
345         MPathTarget *mpt;
346         int sfra, efra;
347         int cfra;
348         
349         /* sanity check */
350         if (ELEM(NULL, targets, targets->first))
351                 return;
352         
353         /* set frame values */
354         cfra = CFRA;
355         sfra = efra = cfra;
356         
357         /* TODO: this method could be improved...
358          * 1) max range for standard baking
359          * 2) minimum range for recalc baking (i.e. between keyframes, but how?) */
360         for (mpt = targets->first; mpt; mpt = mpt->next) {
361                 /* try to increase area to do (only as much as needed) */
362                 sfra = MIN2(sfra, mpt->mpath->start_frame);
363                 efra = MAX2(efra, mpt->mpath->end_frame);
364         }
365         if (efra <= sfra) return;
366         
367         /* calculate path over requested range */
368         for (CFRA = sfra; CFRA <= efra; CFRA++) {
369                 /* update relevant data for new frame */
370                 motionpaths_calc_update_scene(bmain, depsgraph);
371                 
372                 /* perform baking for targets */
373                 motionpaths_calc_bake_targets(scene, targets);
374         }
375         
376         /* reset original environment */
377         CFRA = cfra;
378         motionpaths_calc_update_scene(bmain, depsgraph);
379         
380         /* clear recalc flags from targets */
381         for (mpt = targets->first; mpt; mpt = mpt->next) {
382                 bAnimVizSettings *avs;
383                 
384                 /* get pointer to animviz settings for each target */
385                 if (mpt->pchan)
386                         avs = &mpt->ob->pose->avs;
387                 else
388                         avs = &mpt->ob->avs;
389                 
390                 /* clear the flag requesting recalculation of targets */
391                 avs->recalc &= ~ANIMVIZ_RECALC_PATHS;
392         }
393 }
394
395 /* ******************************************************************** */
396 /* Curve Paths - for curve deforms and/or curve following */
397
398 /* free curve path data 
399  * NOTE: frees the path itself!
400  * NOTE: this is increasingly inaccurate with non-uniform BevPoint subdivisions [#24633]
401  */
402 void free_path(Path *path)
403 {
404         if (path->data) MEM_freeN(path->data);
405         MEM_freeN(path);
406 }
407
408 /* calculate a curve-deform path for a curve 
409  *  - only called from displist.c -> do_makeDispListCurveTypes
410  */
411 void calc_curvepath(Object *ob, ListBase *nurbs)
412 {
413         BevList *bl;
414         BevPoint *bevp, *bevpn, *bevpfirst, *bevplast;
415         PathPoint *pp;
416         Nurb *nu;
417         Path *path;
418         float *fp, *dist, *maxdist, xyz[3];
419         float fac, d = 0, fac1, fac2;
420         int a, tot, cycl = 0;
421         
422         /* in a path vertices are with equal differences: path->len = number of verts */
423         /* NOW WITH BEVELCURVE!!! */
424         
425         if (ob == NULL || ob->type != OB_CURVE) {
426                 return;
427         }
428
429         if (ob->curve_cache->path) free_path(ob->curve_cache->path);
430         ob->curve_cache->path = NULL;
431         
432         /* weak! can only use first curve */
433         bl = ob->curve_cache->bev.first;
434         if (bl == NULL || !bl->nr) {
435                 return;
436         }
437
438         nu = nurbs->first;
439
440         ob->curve_cache->path = path = MEM_callocN(sizeof(Path), "calc_curvepath");
441         
442         /* if POLY: last vertice != first vertice */
443         cycl = (bl->poly != -1);
444         
445         tot = cycl ? bl->nr : bl->nr - 1;
446         
447         path->len = tot + 1;
448         /* exception: vector handle paths and polygon paths should be subdivided at least a factor resolu */
449         if (path->len < nu->resolu * SEGMENTSU(nu)) {
450                 path->len = nu->resolu * SEGMENTSU(nu);
451         }
452         
453         dist = (float *)MEM_mallocN(sizeof(float) * (tot + 1), "calcpathdist");
454
455         /* all lengths in *dist */
456         bevp = bevpfirst = bl->bevpoints;
457         fp = dist;
458         *fp = 0.0f;
459         for (a = 0; a < tot; a++) {
460                 fp++;
461                 if (cycl && a == tot - 1)
462                         sub_v3_v3v3(xyz, bevpfirst->vec, bevp->vec);
463                 else
464                         sub_v3_v3v3(xyz, (bevp + 1)->vec, bevp->vec);
465                 
466                 *fp = *(fp - 1) + len_v3(xyz);
467                 bevp++;
468         }
469         
470         path->totdist = *fp;
471         
472         /* the path verts  in path->data */
473         /* now also with TILT value */
474         pp = path->data = (PathPoint *)MEM_callocN(sizeof(PathPoint) * path->len, "pathdata");
475
476         bevp = bevpfirst;
477         bevpn = bevp + 1;
478         bevplast = bevpfirst + (bl->nr - 1);
479         if (UNLIKELY(bevpn > bevplast)) {
480                 bevpn = cycl ? bevpfirst : bevplast;
481         }
482         fp = dist + 1;
483         maxdist = dist + tot;
484         fac = 1.0f / ((float)path->len - 1.0f);
485         fac = fac * path->totdist;
486         
487         for (a = 0; a < path->len; a++) {
488                 
489                 d = ((float)a) * fac;
490                 
491                 /* we're looking for location (distance) 'd' in the array */
492                 if (LIKELY(tot > 0)) {
493                         while ((fp < maxdist) && (d >= *fp)) {
494                                 fp++;
495                                 if (bevp < bevplast) bevp++;
496                                 bevpn = bevp + 1;
497                                 if (UNLIKELY(bevpn > bevplast)) {
498                                         bevpn = cycl ? bevpfirst : bevplast;
499                                 }
500                         }
501
502                         fac1 = (*(fp) - d) / (*(fp) - *(fp - 1));
503                         fac2 = 1.0f - fac1;
504                 }
505                 else {
506                         fac1 = 1.0f;
507                         fac2 = 0.0f;
508                 }
509
510                 interp_v3_v3v3(pp->vec, bevp->vec, bevpn->vec, fac2);
511                 pp->vec[3] = fac1 * bevp->alfa   + fac2 * bevpn->alfa;
512                 pp->radius = fac1 * bevp->radius + fac2 * bevpn->radius;
513                 pp->weight = fac1 * bevp->weight + fac2 * bevpn->weight;
514                 interp_qt_qtqt(pp->quat, bevp->quat, bevpn->quat, fac2);
515                 normalize_qt(pp->quat);
516                 
517                 pp++;
518         }
519         
520         MEM_freeN(dist);
521 }
522
523 static int interval_test(const int min, const int max, int p1, const int cycl)
524 {
525         if (cycl) {
526                 p1 = mod_i(p1 - min, (max - min + 1)) + min;
527         }
528         else {
529                 if      (p1 < min) p1 = min;
530                 else if (p1 > max) p1 = max;
531         }
532         return p1;
533 }
534
535
536 /* calculate the deformation implied by the curve path at a given parametric position,
537  * and returns whether this operation succeeded.
538  *
539  * note: ctime is normalized range <0-1>
540  *
541  * returns OK: 1/0
542  */
543 int where_on_path(Object *ob, float ctime, float vec[4], float dir[3], float quat[4], float *radius, float *weight)
544 {
545         Curve *cu;
546         Nurb *nu;
547         BevList *bl;
548         Path *path;
549         PathPoint *pp, *p0, *p1, *p2, *p3;
550         float fac;
551         float data[4];
552         int cycl = 0, s0, s1, s2, s3;
553         ListBase *nurbs;
554
555         if (ob == NULL || ob->type != OB_CURVE) return 0;
556         cu = ob->data;
557         if (ob->curve_cache == NULL || ob->curve_cache->path == NULL || ob->curve_cache->path->data == NULL) {
558                 printf("no path!\n");
559                 return 0;
560         }
561         path = ob->curve_cache->path;
562         pp = path->data;
563         
564         /* test for cyclic */
565         bl = ob->curve_cache->bev.first;
566         if (!bl) return 0;
567         if (!bl->nr) return 0;
568         if (bl->poly > -1) cycl = 1;
569
570         /* values below zero for non-cyclic curves give strange results */
571         BLI_assert(cycl || ctime >= 0.0f);
572
573         ctime *= (path->len - 1);
574         
575         s1 = (int)floor(ctime);
576         fac = (float)(s1 + 1) - ctime;
577
578         /* path->len is corected for cyclic */
579         s0 = interval_test(0, path->len - 1 - cycl, s1 - 1, cycl);
580         s1 = interval_test(0, path->len - 1 - cycl, s1, cycl);
581         s2 = interval_test(0, path->len - 1 - cycl, s1 + 1, cycl);
582         s3 = interval_test(0, path->len - 1 - cycl, s1 + 2, cycl);
583
584         p0 = pp + s0;
585         p1 = pp + s1;
586         p2 = pp + s2;
587         p3 = pp + s3;
588
589         /* NOTE: commented out for follow constraint
590          *
591          *       If it's ever be uncommented watch out for curve_deform_verts()
592          *       which used to temporary set CU_FOLLOW flag for the curve and no
593          *       longer does it (because of threading issues of such a thing.
594          */
595         //if (cu->flag & CU_FOLLOW) {
596
597         key_curve_tangent_weights(1.0f - fac, data, KEY_BSPLINE);
598
599         interp_v3_v3v3v3v3(dir, p0->vec, p1->vec, p2->vec, p3->vec, data);
600
601         /* make compatible with vectoquat */
602         negate_v3(dir);
603         //}
604
605         nurbs = BKE_curve_editNurbs_get(cu);
606         if (!nurbs)
607                 nurbs = &cu->nurb;
608         nu = nurbs->first;
609
610         /* make sure that first and last frame are included in the vectors here  */
611         if (nu->type == CU_POLY) key_curve_position_weights(1.0f - fac, data, KEY_LINEAR);
612         else if (nu->type == CU_BEZIER) key_curve_position_weights(1.0f - fac, data, KEY_LINEAR);
613         else if (s0 == s1 || p2 == p3) key_curve_position_weights(1.0f - fac, data, KEY_CARDINAL);
614         else key_curve_position_weights(1.0f - fac, data, KEY_BSPLINE);
615
616         vec[0] = data[0] * p0->vec[0] + data[1] * p1->vec[0] + data[2] * p2->vec[0] + data[3] * p3->vec[0]; /* X */
617         vec[1] = data[0] * p0->vec[1] + data[1] * p1->vec[1] + data[2] * p2->vec[1] + data[3] * p3->vec[1]; /* Y */
618         vec[2] = data[0] * p0->vec[2] + data[1] * p1->vec[2] + data[2] * p2->vec[2] + data[3] * p3->vec[2]; /* Z */
619         vec[3] = data[0] * p0->vec[3] + data[1] * p1->vec[3] + data[2] * p2->vec[3] + data[3] * p3->vec[3]; /* Tilt, should not be needed since we have quat still used */
620
621         if (quat) {
622                 float totfac, q1[4], q2[4];
623
624                 totfac = data[0] + data[3];
625                 if (totfac > FLT_EPSILON) interp_qt_qtqt(q1, p0->quat, p3->quat, data[3] / totfac);
626                 else copy_qt_qt(q1, p1->quat);
627
628                 totfac = data[1] + data[2];
629                 if (totfac > FLT_EPSILON) interp_qt_qtqt(q2, p1->quat, p2->quat, data[2] / totfac);
630                 else copy_qt_qt(q2, p3->quat);
631
632                 totfac = data[0] + data[1] + data[2] + data[3];
633                 if (totfac > FLT_EPSILON) interp_qt_qtqt(quat, q1, q2, (data[1] + data[2]) / totfac);
634                 else copy_qt_qt(quat, q2);
635         }
636
637         if (radius)
638                 *radius = data[0] * p0->radius + data[1] * p1->radius + data[2] * p2->radius + data[3] * p3->radius;
639
640         if (weight)
641                 *weight = data[0] * p0->weight + data[1] * p1->weight + data[2] * p2->weight + data[3] * p3->weight;
642
643         return 1;
644 }