1624a02e773127fc28fefeb636bfd0097c3da685
[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 <stdio.h>
33 #include <math.h>
34 #include <string.h>
35
36 #include "MEM_guardedalloc.h"
37
38 #include "BLI_blenlib.h"
39 #include "BLI_math.h"
40 #include "BLI_rand.h"
41 #include "BLI_utildefines.h"
42
43 #include "BLF_translation.h"
44
45 #include "DNA_anim_types.h"
46 #include "DNA_armature_types.h"
47 #include "DNA_group_types.h"
48 #include "DNA_key_types.h"
49 #include "DNA_mesh_types.h"
50 #include "DNA_meshdata_types.h"
51 #include "DNA_scene_types.h"
52 #include "DNA_vfont_types.h"
53
54 #include "BKE_animsys.h"
55 #include "BKE_curve.h"
56 #include "BKE_DerivedMesh.h"
57 #include "BKE_depsgraph.h"
58 #include "BKE_font.h"
59 #include "BKE_group.h"
60 #include "BKE_global.h"
61 #include "BKE_key.h"
62 #include "BKE_lattice.h"
63 #include "BKE_main.h"
64 #include "BKE_mesh.h"
65 #include "BKE_object.h"
66 #include "BKE_particle.h"
67 #include "BKE_scene.h"
68 #include "BKE_editmesh.h"
69 #include "BKE_depsgraph.h"
70 #include "BKE_anim.h"
71 #include "BKE_report.h"
72
73
74 // XXX bad level call...
75
76 /* --------------------- */
77 /* forward declarations */
78
79 static void object_duplilist_recursive(ID *id, Scene *scene, Object *ob, ListBase *duplilist, float par_space_mat[4][4],
80                                        int persistent_id[MAX_DUPLI_RECUR], int level, int index, short flag);
81
82 /* ******************************************************************** */
83 /* Animation Visualization */
84
85 /* Initialize the default settings for animation visualization */
86 void animviz_settings_init(bAnimVizSettings *avs)
87 {
88         /* sanity check */
89         if (avs == NULL)
90                 return;
91
92         /* ghosting settings */
93         avs->ghost_bc = avs->ghost_ac = 10;
94
95         avs->ghost_sf = 1; /* xxx - take from scene instead? */
96         avs->ghost_ef = 250; /* xxx - take from scene instead? */
97
98         avs->ghost_step = 1;
99
100
101         /* path settings */
102         avs->path_bc = avs->path_ac = 10;
103
104         avs->path_sf = 1; /* xxx - take from scene instead? */
105         avs->path_ef = 250; /* xxx - take from scene instead? */
106
107         avs->path_viewflag = (MOTIONPATH_VIEW_KFRAS | MOTIONPATH_VIEW_KFNOS);
108
109         avs->path_step = 1;
110 }
111
112 /* ------------------- */
113
114 /* Free the given motion path's cache */
115 void animviz_free_motionpath_cache(bMotionPath *mpath) 
116 {
117         /* sanity check */
118         if (mpath == NULL) 
119                 return;
120                 
121         /* free the path if necessary */
122         if (mpath->points)
123                 MEM_freeN(mpath->points);
124         
125         /* reset the relevant parameters */
126         mpath->points = NULL;
127         mpath->length = 0;
128 }
129
130 /* Free the given motion path instance and its data 
131  * NOTE: this frees the motion path given!
132  */
133 void animviz_free_motionpath(bMotionPath *mpath)
134 {
135         /* sanity check */
136         if (mpath == NULL)
137                 return;
138         
139         /* free the cache first */
140         animviz_free_motionpath_cache(mpath);
141         
142         /* now the instance itself */
143         MEM_freeN(mpath);
144 }
145
146 /* ------------------- */
147
148 /* Setup motion paths for the given data
149  * - Only used when explicitly calculating paths on bones which may/may not be consider already
150  *
151  * < scene: current scene (for frame ranges, etc.)
152  * < ob: object to add paths for (must be provided)
153  * < pchan: posechannel to add paths for (optional; if not provided, object-paths are assumed)
154  */
155 bMotionPath *animviz_verify_motionpaths(ReportList *reports, Scene *scene, Object *ob, bPoseChannel *pchan)
156 {
157         bAnimVizSettings *avs;
158         bMotionPath *mpath, **dst;
159         
160         /* sanity checks */
161         if (ELEM(NULL, scene, ob))
162                 return NULL;
163                 
164         /* get destination data */
165         if (pchan) {
166                 /* paths for posechannel - assume that posechannel belongs to the object */
167                 avs = &ob->pose->avs;
168                 dst = &pchan->mpath;
169         }
170         else {
171                 /* paths for object */
172                 avs = &ob->avs;
173                 dst = &ob->mpath;
174         }
175
176         /* avoid 0 size allocs */
177         if (avs->path_sf >= avs->path_ef) {
178                 BKE_reportf(reports, RPT_ERROR,
179                             "Motion path frame extents invalid for %s (%d to %d)%s",
180                             (pchan) ? pchan->name : ob->id.name,
181                             avs->path_sf, avs->path_ef,
182                             (avs->path_sf == avs->path_ef) ? TIP_(", cannot have single-frame paths") : "");
183                 return NULL;
184         }
185
186         /* if there is already a motionpath, just return that,
187          * provided it's settings are ok (saves extra free+alloc)
188          */
189         if (*dst != NULL) {
190                 int expected_length = avs->path_ef - avs->path_sf;
191                 
192                 mpath = *dst;
193                 
194                 /* path is "valid" if length is valid, but must also be of the same length as is being requested */
195                 if ((mpath->start_frame != mpath->end_frame) && (mpath->length > 0)) {
196                         /* outer check ensures that we have some curve data for this path */
197                         if (mpath->length == expected_length) {
198                                 /* return/use this as it is already valid length */
199                                 return mpath;
200                         }
201                         else {
202                                 /* clear the existing path (as the range has changed), and reallocate below */
203                                 animviz_free_motionpath_cache(mpath);
204                         }
205                 }
206         }
207         else {
208                 /* create a new motionpath, and assign it */
209                 mpath = MEM_callocN(sizeof(bMotionPath), "bMotionPath");
210                 *dst = mpath;
211         }
212         
213         /* set settings from the viz settings */
214         mpath->start_frame = avs->path_sf;
215         mpath->end_frame = avs->path_ef;
216         
217         mpath->length = mpath->end_frame - mpath->start_frame;
218         
219         if (avs->path_bakeflag & MOTIONPATH_BAKE_HEADS)
220                 mpath->flag |= MOTIONPATH_FLAG_BHEAD;
221         else
222                 mpath->flag &= ~MOTIONPATH_FLAG_BHEAD;
223         
224         /* allocate a cache */
225         mpath->points = MEM_callocN(sizeof(bMotionPathVert) * mpath->length, "bMotionPathVerts");
226         
227         /* tag viz settings as currently having some path(s) which use it */
228         avs->path_bakeflag |= MOTIONPATH_BAKE_HAS_PATHS;
229         
230         /* return it */
231         return mpath;
232 }
233
234 /* ------------------- */
235
236 /* Motion path needing to be baked (mpt) */
237 typedef struct MPathTarget {
238         struct MPathTarget *next, *prev;
239         
240         bMotionPath *mpath;         /* motion path in question */
241         
242         Object *ob;                 /* source object */
243         bPoseChannel *pchan;        /* source posechannel (if applicable) */
244 } MPathTarget;
245
246 /* ........ */
247
248 /* get list of motion paths to be baked for the given object
249  *  - assumes the given list is ready to be used
250  */
251 /* TODO: it would be nice in future to be able to update objects dependent on these bones too? */
252 void animviz_get_object_motionpaths(Object *ob, ListBase *targets)
253 {
254         MPathTarget *mpt;
255         
256         /* object itself first */
257         if ((ob->avs.recalc & ANIMVIZ_RECALC_PATHS) && (ob->mpath)) {
258                 /* new target for object */
259                 mpt = MEM_callocN(sizeof(MPathTarget), "MPathTarget Ob");
260                 BLI_addtail(targets, mpt);
261                 
262                 mpt->mpath = ob->mpath;
263                 mpt->ob = ob;
264         }
265         
266         /* bones */
267         if ((ob->pose) && (ob->pose->avs.recalc & ANIMVIZ_RECALC_PATHS)) {
268                 bArmature *arm = ob->data;
269                 bPoseChannel *pchan;
270                 
271                 for (pchan = ob->pose->chanbase.first; pchan; pchan = pchan->next) {
272                         if ((pchan->bone) && (arm->layer & pchan->bone->layer) && (pchan->mpath)) {
273                                 /* new target for bone */
274                                 mpt = MEM_callocN(sizeof(MPathTarget), "MPathTarget PoseBone");
275                                 BLI_addtail(targets, mpt);
276                                 
277                                 mpt->mpath = pchan->mpath;
278                                 mpt->ob = ob;
279                                 mpt->pchan = pchan;
280                         }
281                 }
282         }
283 }
284
285 /* ........ */
286
287 /* Note on evaluation optimizations:
288  * Optimization's currently used here play tricks with the depsgraph in order to try and
289  * evaluate as few objects as strictly necessary to get nicer performance under standard
290  * production conditions. For those people who really need the accurate version, 
291  * disable the ifdef (i.e. 1 -> 0) and comment out the call to motionpaths_calc_optimise_depsgraph()
292  */
293
294 /* tweak the object ordering to trick depsgraph into making MotionPath calculations run faster */
295 static void motionpaths_calc_optimise_depsgraph(Scene *scene, ListBase *targets)
296 {
297         Base *base, *baseNext;
298         MPathTarget *mpt;
299         
300         /* make sure our temp-tag isn't already in use */
301         for (base = scene->base.first; base; base = base->next)
302                 base->object->flag &= ~BA_TEMP_TAG;
303         
304         /* for each target, dump its object to the start of the list if it wasn't moved already */
305         for (mpt = targets->first; mpt; mpt = mpt->next) {
306                 for (base = scene->base.first; base; base = baseNext) {
307                         baseNext = base->next;
308                         
309                         if ((base->object == mpt->ob) && !(mpt->ob->flag & BA_TEMP_TAG)) {
310                                 BLI_remlink(&scene->base, base);
311                                 BLI_addhead(&scene->base, base);
312                                 
313                                 mpt->ob->flag |= BA_TEMP_TAG;
314                                 
315                                 /* we really don't need to continue anymore once this happens, but this line might really 'break' */
316                                 break;
317                         }
318                 }
319         }
320         
321         /* "brew me a list that's sorted a bit faster now depsy" */
322         DAG_scene_relations_rebuild(G.main, scene);
323 }
324
325 /* update scene for current frame */
326 static void motionpaths_calc_update_scene(Scene *scene)
327 {
328 #if 1 // 'production' optimizations always on
329
330         /* rigid body simulation needs complete update to work correctly for now */
331         /* RB_TODO investigate if we could avoid updating everything */
332         if (BKE_scene_check_rigidbody_active(scene)) {
333                 BKE_scene_update_for_newframe(G.main, scene, scene->lay);
334         }
335         else { /* otherwise we can optimize by restricting updates */
336                 Base *base, *last = NULL;
337                 
338                 /* only stuff that moves or needs display still */
339                 DAG_scene_update_flags(G.main, scene, scene->lay, TRUE);
340                 
341                 /* find the last object with the tag 
342                  * - all those afterwards are assumed to not be relevant for our calculations
343                  */
344                 /* optimize further by moving out... */
345                 for (base = scene->base.first; base; base = base->next) {
346                         if (base->object->flag & BA_TEMP_TAG)
347                                 last = base;
348                 }
349                 
350                 /* perform updates for tagged objects */
351                 /* XXX: this will break if rigs depend on scene or other data that
352                  * is animated but not attached to/updatable from objects */
353                 for (base = scene->base.first; base; base = base->next) {
354                         /* update this object */
355                         BKE_object_handle_update(scene, base->object);
356                         
357                         /* if this is the last one we need to update, let's stop to save some time */
358                         if (base == last)
359                                 break;
360                 }
361         }
362 #else // original, 'always correct' version
363           /* do all updates
364            *  - if this is too slow, resort to using a more efficient way
365            *    that doesn't force complete update, but for now, this is the
366            *    most accurate way!
367            */
368         BKE_scene_update_for_newframe(G.main, scene, scene->lay); /* XXX this is the best way we can get anything moving */
369 #endif
370 }
371
372 /* ........ */
373
374 /* perform baking for the targets on the current frame */
375 static void motionpaths_calc_bake_targets(Scene *scene, ListBase *targets)
376 {
377         MPathTarget *mpt;
378         
379         /* for each target, check if it can be baked on the current frame */
380         for (mpt = targets->first; mpt; mpt = mpt->next) {
381                 bMotionPath *mpath = mpt->mpath;
382                 bMotionPathVert *mpv;
383                 
384                 /* current frame must be within the range the cache works for 
385                  *      - is inclusive of the first frame, but not the last otherwise we get buffer overruns
386                  */
387                 if ((CFRA < mpath->start_frame) || (CFRA >= mpath->end_frame))
388                         continue;
389                 
390                 /* get the relevant cache vert to write to */
391                 mpv = mpath->points + (CFRA - mpath->start_frame);
392                 
393                 /* pose-channel or object path baking? */
394                 if (mpt->pchan) {
395                         /* heads or tails */
396                         if (mpath->flag & MOTIONPATH_FLAG_BHEAD) {
397                                 copy_v3_v3(mpv->co, mpt->pchan->pose_head);
398                         }
399                         else {
400                                 copy_v3_v3(mpv->co, mpt->pchan->pose_tail);
401                         }
402                         
403                         /* result must be in worldspace */
404                         mul_m4_v3(mpt->ob->obmat, mpv->co);
405                 }
406                 else {
407                         /* worldspace object location */
408                         copy_v3_v3(mpv->co, mpt->ob->obmat[3]);
409                 }
410         }
411 }
412
413 /* Perform baking of the given object's and/or its bones' transforms to motion paths 
414  *      - scene: current scene
415  *      - ob: object whose flagged motionpaths should get calculated
416  *      - recalc: whether we need to
417  */
418 /* TODO: include reports pointer? */
419 void animviz_calc_motionpaths(Scene *scene, ListBase *targets)
420 {
421         MPathTarget *mpt;
422         int sfra, efra;
423         int cfra;
424         
425         /* sanity check */
426         if (ELEM(NULL, targets, targets->first))
427                 return;
428         
429         /* set frame values */
430         cfra = CFRA;
431         sfra = efra = cfra;
432         
433         /* TODO: this method could be improved...
434          * 1) max range for standard baking
435          * 2) minimum range for recalc baking (i.e. between keyframes, but how?) */
436         for (mpt = targets->first; mpt; mpt = mpt->next) {
437                 /* try to increase area to do (only as much as needed) */
438                 sfra = MIN2(sfra, mpt->mpath->start_frame);
439                 efra = MAX2(efra, mpt->mpath->end_frame);
440         }
441         if (efra <= sfra) return;
442         
443         /* optimize the depsgraph for faster updates */
444         /* TODO: whether this is used should depend on some setting for the level of optimizations used */
445         motionpaths_calc_optimise_depsgraph(scene, targets);
446         
447         /* calculate path over requested range */
448         for (CFRA = sfra; CFRA <= efra; CFRA++) {
449                 /* update relevant data for new frame */
450                 motionpaths_calc_update_scene(scene);
451                 
452                 /* perform baking for targets */
453                 motionpaths_calc_bake_targets(scene, targets);
454         }
455         
456         /* reset original environment */
457         CFRA = cfra;
458         motionpaths_calc_update_scene(scene);
459         
460         /* clear recalc flags from targets */
461         for (mpt = targets->first; mpt; mpt = mpt->next) {
462                 bAnimVizSettings *avs;
463                 
464                 /* get pointer to animviz settings for each target */
465                 if (mpt->pchan)
466                         avs = &mpt->ob->pose->avs;
467                 else
468                         avs = &mpt->ob->avs;
469                 
470                 /* clear the flag requesting recalculation of targets */
471                 avs->recalc &= ~ANIMVIZ_RECALC_PATHS;
472         }
473 }
474
475 /* ******************************************************************** */
476 /* Curve Paths - for curve deforms and/or curve following */
477
478 /* free curve path data 
479  * NOTE: frees the path itself!
480  * NOTE: this is increasingly inaccurate with non-uniform BevPoint subdivisions [#24633]
481  */
482 void free_path(Path *path)
483 {
484         if (path->data) MEM_freeN(path->data);
485         MEM_freeN(path);
486 }
487
488 /* calculate a curve-deform path for a curve 
489  *  - only called from displist.c -> do_makeDispListCurveTypes
490  */
491 void calc_curvepath(Object *ob, ListBase *nurbs)
492 {
493         BevList *bl;
494         BevPoint *bevp, *bevpn, *bevpfirst, *bevplast;
495         PathPoint *pp;
496         Nurb *nu;
497         Path *path;
498         float *fp, *dist, *maxdist, xyz[3];
499         float fac, d = 0, fac1, fac2;
500         int a, tot, cycl = 0;
501         
502         /* in a path vertices are with equal differences: path->len = number of verts */
503         /* NOW WITH BEVELCURVE!!! */
504         
505         if (ob == NULL || ob->type != OB_CURVE) {
506                 return;
507         }
508
509         if (ob->curve_cache->path) free_path(ob->curve_cache->path);
510         ob->curve_cache->path = NULL;
511         
512         /* weak! can only use first curve */
513         bl = ob->curve_cache->bev.first;
514         if (bl == NULL || !bl->nr) {
515                 return;
516         }
517
518         nu = nurbs->first;
519
520         ob->curve_cache->path = path = MEM_callocN(sizeof(Path), "calc_curvepath");
521         
522         /* if POLY: last vertice != first vertice */
523         cycl = (bl->poly != -1);
524         
525         tot = cycl ? bl->nr : bl->nr - 1;
526         
527         path->len = tot + 1;
528         /* exception: vector handle paths and polygon paths should be subdivided at least a factor resolu */
529         if (path->len < nu->resolu * SEGMENTSU(nu)) {
530                 path->len = nu->resolu * SEGMENTSU(nu);
531         }
532         
533         dist = (float *)MEM_mallocN(sizeof(float) * (tot + 1), "calcpathdist");
534
535         /* all lengths in *dist */
536         bevp = bevpfirst = (BevPoint *)(bl + 1);
537         fp = dist;
538         *fp = 0.0f;
539         for (a = 0; a < tot; a++) {
540                 fp++;
541                 if (cycl && a == tot - 1)
542                         sub_v3_v3v3(xyz, bevpfirst->vec, bevp->vec);
543                 else
544                         sub_v3_v3v3(xyz, (bevp + 1)->vec, bevp->vec);
545                 
546                 *fp = *(fp - 1) + len_v3(xyz);
547                 bevp++;
548         }
549         
550         path->totdist = *fp;
551         
552         /* the path verts  in path->data */
553         /* now also with TILT value */
554         pp = path->data = (PathPoint *)MEM_callocN(sizeof(PathPoint) * path->len, "pathdata");
555
556         bevp = bevpfirst;
557         bevpn = bevp + 1;
558         bevplast = bevpfirst + (bl->nr - 1);
559         fp = dist + 1;
560         maxdist = dist + tot;
561         fac = 1.0f / ((float)path->len - 1.0f);
562         fac = fac * path->totdist;
563         
564         for (a = 0; a < path->len; a++) {
565                 
566                 d = ((float)a) * fac;
567                 
568                 /* we're looking for location (distance) 'd' in the array */
569                 while ((d >= *fp) && fp < maxdist) {
570                         fp++;
571                         if (bevp < bevplast) bevp++;
572                         bevpn = bevp + 1;
573                         if (UNLIKELY(bevpn > bevplast)) {
574                                 bevpn = cycl ? bevpfirst : bevplast;
575                         }
576                 }
577                 
578                 fac1 = (*(fp) - d) / (*(fp) - *(fp - 1));
579                 fac2 = 1.0f - fac1;
580
581                 interp_v3_v3v3(pp->vec, bevp->vec, bevpn->vec, fac2);
582                 pp->vec[3] = fac1 * bevp->alfa   + fac2 * bevpn->alfa;
583                 pp->radius = fac1 * bevp->radius + fac2 * bevpn->radius;
584                 pp->weight = fac1 * bevp->weight + fac2 * bevpn->weight;
585                 interp_qt_qtqt(pp->quat, bevp->quat, bevpn->quat, fac2);
586                 normalize_qt(pp->quat);
587                 
588                 pp++;
589         }
590         
591         MEM_freeN(dist);
592 }
593
594 static int interval_test(const int min, const int max, int p1, const int cycl)
595 {
596         if (cycl) {
597                 p1 = mod_i(p1 - min, (max - min + 1)) + min;
598         }
599         else {
600                 if      (p1 < min) p1 = min;
601                 else if (p1 > max) p1 = max;
602         }
603         return p1;
604 }
605
606
607 /* calculate the deformation implied by the curve path at a given parametric position,
608  * and returns whether this operation succeeded.
609  *
610  * note: ctime is normalized range <0-1>
611  *
612  * returns OK: 1/0
613  */
614 int where_on_path(Object *ob, float ctime, float vec[4], float dir[3], float quat[4], float *radius, float *weight)
615 {
616         Curve *cu;
617         Nurb *nu;
618         BevList *bl;
619         Path *path;
620         PathPoint *pp, *p0, *p1, *p2, *p3;
621         float fac;
622         float data[4];
623         int cycl = 0, s0, s1, s2, s3;
624         ListBase *nurbs;
625
626         if (ob == NULL || ob->type != OB_CURVE) return 0;
627         cu = ob->data;
628         if (ob->curve_cache == NULL || ob->curve_cache->path == NULL || ob->curve_cache->path->data == NULL) {
629                 printf("no path!\n");
630                 return 0;
631         }
632         path = ob->curve_cache->path;
633         pp = path->data;
634         
635         /* test for cyclic */
636         bl = ob->curve_cache->bev.first;
637         if (!bl) return 0;
638         if (!bl->nr) return 0;
639         if (bl->poly > -1) cycl = 1;
640
641         ctime *= (path->len - 1);
642         
643         s1 = (int)floor(ctime);
644         fac = (float)(s1 + 1) - ctime;
645
646         /* path->len is corected for cyclic */
647         s0 = interval_test(0, path->len - 1 - cycl, s1 - 1, cycl);
648         s1 = interval_test(0, path->len - 1 - cycl, s1, cycl);
649         s2 = interval_test(0, path->len - 1 - cycl, s1 + 1, cycl);
650         s3 = interval_test(0, path->len - 1 - cycl, s1 + 2, cycl);
651
652         p0 = pp + s0;
653         p1 = pp + s1;
654         p2 = pp + s2;
655         p3 = pp + s3;
656
657         /* note, commented out for follow constraint */
658         //if (cu->flag & CU_FOLLOW) {
659
660         key_curve_tangent_weights(1.0f - fac, data, KEY_BSPLINE);
661
662         interp_v3_v3v3v3v3(dir, p0->vec, p1->vec, p2->vec, p3->vec, data);
663
664         /* make compatible with vectoquat */
665         negate_v3(dir);
666         //}
667
668         nurbs = BKE_curve_editNurbs_get(cu);
669         if (!nurbs)
670                 nurbs = &cu->nurb;
671         nu = nurbs->first;
672
673         /* make sure that first and last frame are included in the vectors here  */
674         if (nu->type == CU_POLY) key_curve_position_weights(1.0f - fac, data, KEY_LINEAR);
675         else if (nu->type == CU_BEZIER) key_curve_position_weights(1.0f - fac, data, KEY_LINEAR);
676         else if (s0 == s1 || p2 == p3) key_curve_position_weights(1.0f - fac, data, KEY_CARDINAL);
677         else key_curve_position_weights(1.0f - fac, data, KEY_BSPLINE);
678
679         vec[0] = data[0] * p0->vec[0] + data[1] * p1->vec[0] + data[2] * p2->vec[0] + data[3] * p3->vec[0]; /* X */
680         vec[1] = data[0] * p0->vec[1] + data[1] * p1->vec[1] + data[2] * p2->vec[1] + data[3] * p3->vec[1]; /* Y */
681         vec[2] = data[0] * p0->vec[2] + data[1] * p1->vec[2] + data[2] * p2->vec[2] + data[3] * p3->vec[2]; /* Z */
682         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 */
683
684         if (quat) {
685                 float totfac, q1[4], q2[4];
686
687                 totfac = data[0] + data[3];
688                 if (totfac > FLT_EPSILON) interp_qt_qtqt(q1, p0->quat, p3->quat, data[3] / totfac);
689                 else copy_qt_qt(q1, p1->quat);
690
691                 totfac = data[1] + data[2];
692                 if (totfac > FLT_EPSILON) interp_qt_qtqt(q2, p1->quat, p2->quat, data[2] / totfac);
693                 else copy_qt_qt(q2, p3->quat);
694
695                 totfac = data[0] + data[1] + data[2] + data[3];
696                 if (totfac > FLT_EPSILON) interp_qt_qtqt(quat, q1, q2, (data[1] + data[2]) / totfac);
697                 else copy_qt_qt(quat, q2);
698         }
699
700         if (radius)
701                 *radius = data[0] * p0->radius + data[1] * p1->radius + data[2] * p2->radius + data[3] * p3->radius;
702
703         if (weight)
704                 *weight = data[0] * p0->weight + data[1] * p1->weight + data[2] * p2->weight + data[3] * p3->weight;
705
706         return 1;
707 }
708
709 /* ******************************************************************** */
710 /* Dupli-Geometry */
711
712 #define DUPLILIST_DO_UPDATE             1
713 #define DUPLILIST_FOR_RENDER    2
714 #define DUPLILIST_ANIMATED              4
715
716 static DupliObject *new_dupli_object(ListBase *lb, Object *ob, float mat[4][4], int lay,
717                                      int persistent_id[MAX_DUPLI_RECUR], int level, int index, int type, short flag)
718 {
719         DupliObject *dob = MEM_callocN(sizeof(DupliObject), "dupliobject");
720         int i;
721
722         BLI_addtail(lb, dob);
723         dob->ob = ob;
724         copy_m4_m4(dob->mat, mat);
725         copy_m4_m4(dob->omat, ob->obmat);
726         dob->origlay = ob->lay;
727         dob->type = type;
728         dob->animated = (type == OB_DUPLIGROUP) && (flag & DUPLILIST_ANIMATED);
729         ob->lay = lay;
730
731         /* set persistent id, which is an array with a persistent index for each level
732          * (particle number, vertex number, ..). by comparing this we can find the same
733          * dupli object between frames, which is needed for motion blur. last level
734          * goes first in the array. */
735         dob->persistent_id[0] = index;
736         for (i = 1; i < level; i++)
737                 dob->persistent_id[i] = persistent_id[level - 1 - i];
738         
739         /* metaballs never draw in duplis, they are instead merged into one by the basis
740          * mball outside of the group. this does mean that if that mball is not in the
741          * scene, they will not show up at all, limitation that should be solved once. */
742         if (ob->type == OB_MBALL)
743                 dob->no_draw = TRUE;
744         
745         return dob;
746 }
747
748 static void group_duplilist(ListBase *lb, Scene *scene, Object *ob, int persistent_id[MAX_DUPLI_RECUR],
749                             int level, short flag)
750 {
751         DupliObject *dob;
752         Group *group;
753         GroupObject *go;
754         float mat[4][4], ob_obmat_ofs[4][4], id;
755
756         if (ob->dup_group == NULL) return;
757         group = ob->dup_group;
758         
759         /* simple preventing of too deep nested groups */
760         if (level > MAX_DUPLI_RECUR) return;
761         
762         /* don't access 'ob->obmat' from now on. */
763         copy_m4_m4(ob_obmat_ofs, ob->obmat);
764
765         if (!is_zero_v3(group->dupli_ofs)) {
766                 float tvec[3];
767                 copy_v3_v3(tvec, group->dupli_ofs);
768                 mul_mat3_m4_v3(ob_obmat_ofs, tvec);
769                 sub_v3_v3(ob_obmat_ofs[3], tvec);
770         }
771
772         /* handles animated groups, and */
773
774         /* we need to check update for objects that are not in scene... */
775         if (flag & DUPLILIST_DO_UPDATE) {
776                 /* note: update is optional because we don't always need object
777                  * transformations to be correct. Also fixes bug [#29616]. */
778                 BKE_group_handle_recalc_and_update(scene, ob, group);
779         }
780
781         if (BKE_group_is_animated(group, ob))
782                 flag |= DUPLILIST_ANIMATED;
783         
784         for (go = group->gobject.first, id = 0; go; go = go->next, id++) {
785                 /* note, if you check on layer here, render goes wrong... it still deforms verts and uses parent imat */
786                 if (go->ob != ob) {
787                         
788                         /* group dupli offset, should apply after everything else */
789                         mul_m4_m4m4(mat, ob_obmat_ofs, go->ob->obmat);
790                         
791                         dob = new_dupli_object(lb, go->ob, mat, ob->lay, persistent_id, level, id, OB_DUPLIGROUP, flag);
792
793                         /* check the group instance and object layers match, also that the object visible flags are ok. */
794                         if ((dob->origlay & group->layer) == 0 ||
795                             ((G.is_rendering == FALSE) && dob->ob->restrictflag & OB_RESTRICT_VIEW) ||
796                             ((G.is_rendering == TRUE)  && dob->ob->restrictflag & OB_RESTRICT_RENDER))
797                         {
798                                 dob->no_draw = TRUE;
799                         }
800
801                         if (go->ob->transflag & OB_DUPLI) {
802                                 copy_m4_m4(dob->ob->obmat, dob->mat);
803                                 object_duplilist_recursive(&group->id, scene, go->ob, lb, ob_obmat_ofs, persistent_id, level + 1, id, flag);
804                                 copy_m4_m4(dob->ob->obmat, dob->omat);
805                         }
806                 }
807         }
808 }
809
810 static void frames_duplilist(ListBase *lb, Scene *scene, Object *ob, int persistent_id[MAX_DUPLI_RECUR], int level, short flag)
811 {
812         extern int enable_cu_speed; /* object.c */
813         Object copyob;
814         int cfrao = scene->r.cfra;
815         int dupend = ob->dupend;
816         
817         /* simple prevention of too deep nested groups */
818         if (level > MAX_DUPLI_RECUR) return;
819         
820         /* if we don't have any data/settings which will lead to object movement,
821          * don't waste time trying, as it will all look the same...
822          */
823         if (ob->parent == NULL && ob->constraints.first == NULL && ob->adt == NULL)
824                 return;
825         
826         /* make a copy of the object's original data (before any dupli-data overwrites it) 
827          * as we'll need this to keep track of unkeyed data
828          *      - this doesn't take into account other data that can be reached from the object,
829          *        for example it's shapekeys or bones, hence the need for an update flush at the end
830          */
831         copyob = *ob;
832         
833         /* duplicate over the required range */
834         if (ob->transflag & OB_DUPLINOSPEED) enable_cu_speed = 0;
835         
836         for (scene->r.cfra = ob->dupsta; scene->r.cfra <= dupend; scene->r.cfra++) {
837                 short ok = 1;
838                 
839                 /* - dupoff = how often a frames within the range shouldn't be made into duplis
840                  * - dupon = the length of each "skipping" block in frames
841                  */
842                 if (ob->dupoff) {
843                         ok = scene->r.cfra - ob->dupsta;
844                         ok = ok % (ob->dupon + ob->dupoff);
845                         ok = (ok < ob->dupon);
846                 }
847                 
848                 if (ok) {
849                         DupliObject *dob;
850                         
851                         /* WARNING: doing animation updates in this way is not terribly accurate, as the dependencies
852                          * and/or other objects which may affect this object's transforms are not updated either.
853                          * However, this has always been the way that this worked (i.e. pre 2.5), so I guess that it'll be fine!
854                          */
855                         BKE_animsys_evaluate_animdata(scene, &ob->id, ob->adt, (float)scene->r.cfra, ADT_RECALC_ANIM); /* ob-eval will do drivers, so we don't need to do them */
856                         BKE_object_where_is_calc_time(scene, ob, (float)scene->r.cfra);
857                         
858                         dob = new_dupli_object(lb, ob, ob->obmat, ob->lay, persistent_id, level, scene->r.cfra, OB_DUPLIFRAMES, flag);
859                         copy_m4_m4(dob->omat, copyob.obmat);
860                 }
861         }
862
863         enable_cu_speed = 1;
864         
865         /* reset frame to original frame, then re-evaluate animation as above 
866          * as 2.5 animation data may have far-reaching consequences
867          */
868         scene->r.cfra = cfrao;
869         
870         BKE_animsys_evaluate_animdata(scene, &ob->id, ob->adt, (float)scene->r.cfra, ADT_RECALC_ANIM); /* ob-eval will do drivers, so we don't need to do them */
871         BKE_object_where_is_calc_time(scene, ob, (float)scene->r.cfra);
872         
873         /* but, to make sure unkeyed object transforms are still sane, 
874          * let's copy object's original data back over
875          */
876         *ob = copyob;
877 }
878
879 typedef struct VertexDupliData {
880         ID *id; /* scene or group, for recursive loops */
881         int level;
882         short flag;
883         ListBase *lb;
884         float pmat[4][4];
885         float obmat[4][4]; /* Only used for dupliverts inside dupligroups, where the ob->obmat is modified */
886         Scene *scene;
887         Object *ob, *par;
888         float (*orco)[3];
889         int *persistent_id;
890 } VertexDupliData;
891
892 /* ------------- */
893
894 static void vertex_dupli__mapFunc(void *userData, int index, const float co[3],
895                                   const float no_f[3], const short no_s[3])
896 {
897         DupliObject *dob;
898         VertexDupliData *vdd = userData;
899         float vec[3], q2[4], mat[3][3], tmat[4][4], obmat[4][4];
900         int origlay;
901         
902         mul_v3_m4v3(vec, vdd->pmat, co);
903         sub_v3_v3(vec, vdd->pmat[3]);
904         add_v3_v3(vec, vdd->obmat[3]);
905         
906         copy_m4_m4(obmat, vdd->obmat);
907         copy_v3_v3(obmat[3], vec);
908         
909         if (vdd->par->transflag & OB_DUPLIROT) {
910                 if (no_f) {
911                         vec[0] = -no_f[0]; vec[1] = -no_f[1]; vec[2] = -no_f[2];
912                 }
913                 else if (no_s) {
914                         vec[0] = -no_s[0]; vec[1] = -no_s[1]; vec[2] = -no_s[2];
915                 }
916                 
917                 vec_to_quat(q2, vec, vdd->ob->trackflag, vdd->ob->upflag);
918                 
919                 quat_to_mat3(mat, q2);
920                 copy_m4_m4(tmat, obmat);
921                 mul_m4_m4m3(obmat, tmat, mat);
922         }
923
924         origlay = vdd->ob->lay;
925         
926         dob = new_dupli_object(vdd->lb, vdd->ob, obmat, vdd->par->lay, vdd->persistent_id, vdd->level, index, OB_DUPLIVERTS, vdd->flag);
927
928         /* restore the original layer so that each dupli will have proper dob->origlay */
929         vdd->ob->lay = origlay;
930
931         if (vdd->orco)
932                 copy_v3_v3(dob->orco, vdd->orco[index]);
933         
934         if (vdd->ob->transflag & OB_DUPLI) {
935                 float tmpmat[4][4];
936                 copy_m4_m4(tmpmat, vdd->ob->obmat);
937                 copy_m4_m4(vdd->ob->obmat, obmat); /* pretend we are really this mat */
938                 object_duplilist_recursive((ID *)vdd->id, vdd->scene, vdd->ob, vdd->lb, obmat, vdd->persistent_id, vdd->level + 1, index, vdd->flag);
939                 copy_m4_m4(vdd->ob->obmat, tmpmat);
940         }
941 }
942
943 static void vertex_duplilist(ListBase *lb, ID *id, Scene *scene, Object *par, float par_space_mat[4][4], int persistent_id[MAX_DUPLI_RECUR],
944                              int level, short flag)
945 {
946         const bool use_rotation = (par->transflag & OB_DUPLIROT) != 0;
947         Object *ob, *ob_iter;
948         Mesh *me = par->data;
949         Base *base = NULL;
950         DerivedMesh *dm;
951         VertexDupliData vdd;
952         Scene *sce = NULL;
953         Group *group = NULL;
954         GroupObject *go = NULL;
955         BMEditMesh *em;
956         float vec[3], no[3], pmat[4][4];
957         int totvert, a, oblay;
958         unsigned int lay;
959         CustomDataMask dm_mask;
960         
961         copy_m4_m4(pmat, par->obmat);
962         
963         /* simple preventing of too deep nested groups */
964         if (level > MAX_DUPLI_RECUR) return;
965         
966         em = BKE_editmesh_from_object(par);
967         
968         /* get derived mesh */
969         dm_mask = CD_MASK_BAREMESH;
970         if (flag & DUPLILIST_FOR_RENDER)
971                 dm_mask |= CD_MASK_ORCO;
972         
973         if (em)
974                 dm = editbmesh_get_derived_cage(scene, par, em, dm_mask);
975         else
976                 dm = mesh_get_derived_final(scene, par, dm_mask);
977         
978         if (flag & DUPLILIST_FOR_RENDER)
979                 vdd.orco = dm->getVertDataArray(dm, CD_ORCO);
980         else
981                 vdd.orco = NULL;
982         
983         totvert = dm->getNumVerts(dm);
984
985         /* having to loop on scene OR group objects is NOT FUN */
986         if (GS(id->name) == ID_SCE) {
987                 sce = (Scene *)id;
988                 lay = sce->lay;
989                 base = sce->base.first;
990         }
991         else {
992                 group = (Group *)id;
993                 lay = group->layer;
994                 go = group->gobject.first;
995         }
996         
997         /* Start looping on Scene OR Group objects */
998         while (base || go) {
999                 if (sce) {
1000                         ob_iter = base->object;
1001                         oblay = base->lay;
1002                 }
1003                 else {
1004                         ob_iter = go->ob;
1005                         oblay = ob_iter->lay;
1006                 }
1007                 
1008                 if (lay & oblay && scene->obedit != ob_iter) {
1009                         ob = ob_iter->parent;
1010                         while (ob) {
1011                                 if (ob == par) {
1012                                         ob = ob_iter;
1013                                         /* End Scene/Group object loop, below is generic */
1014                                         
1015                                         
1016                                         /* par_space_mat - only used for groups so we can modify the space dupli's are in
1017                                          * when par_space_mat is NULL ob->obmat can be used instead of ob__obmat
1018                                          */
1019                                         if (par_space_mat)
1020                                                 mul_m4_m4m4(vdd.obmat, par_space_mat, ob->obmat);
1021                                         else
1022                                                 copy_m4_m4(vdd.obmat, ob->obmat);
1023
1024                                         vdd.id = id;
1025                                         vdd.level = level;
1026                                         vdd.flag = flag;
1027                                         vdd.lb = lb;
1028                                         vdd.ob = ob;
1029                                         vdd.scene = scene;
1030                                         vdd.par = par;
1031                                         copy_m4_m4(vdd.pmat, pmat);
1032                                         vdd.persistent_id = persistent_id;
1033                                         
1034                                         /* mballs have a different dupli handling */
1035                                         if (ob->type != OB_MBALL) ob->flag |= OB_DONE;  /* doesnt render */
1036
1037                                         if (me->edit_btmesh) {
1038                                                 dm->foreachMappedVert(dm, vertex_dupli__mapFunc, (void *) &vdd,
1039                                                                       use_rotation ? DM_FOREACH_USE_NORMAL : 0);
1040                                         }
1041                                         else {
1042                                                 const float *no_pt = use_rotation ? no : NULL;
1043                                                 for (a = 0; a < totvert; a++) {
1044                                                         dm->getVertCo(dm, a, vec);
1045
1046                                                         if (use_rotation) {
1047                                                                 dm->getVertNo(dm, a, no);
1048                                                         }
1049                                                         
1050                                                         vertex_dupli__mapFunc(&vdd, a, vec, no_pt, NULL);
1051                                                 }
1052                                         }
1053                                         if (sce) {
1054                                                 /* Set proper layer in case of scene looping,
1055                                                  * in case of groups the object layer will be
1056                                                  * changed when it's duplicated due to the
1057                                                  * group duplication.
1058                                                  */
1059                                                 ob->lay = vdd.par->lay;
1060                                         }
1061                                         
1062                                         break;
1063                                 }
1064                                 ob = ob->parent;
1065                         }
1066                 }
1067                 if (sce) base = base->next;     /* scene loop */
1068                 else go = go->next;             /* group loop */
1069         }
1070
1071         dm->release(dm);
1072 }
1073
1074 static void face_duplilist(ListBase *lb, ID *id, Scene *scene, Object *par, float par_space_mat[4][4], int persistent_id[MAX_DUPLI_RECUR],
1075                            int level, short flag)
1076 {
1077         Object *ob, *ob_iter;
1078         Base *base = NULL;
1079         DupliObject *dob;
1080         DerivedMesh *dm;
1081         MLoopUV *mloopuv;
1082         MPoly *mpoly, *mp;
1083         MLoop *mloop;
1084         MVert *mvert;
1085         float pmat[4][4], imat[3][3], (*orco)[3] = NULL, w;
1086         int lay, oblay, totface, a;
1087         Scene *sce = NULL;
1088         Group *group = NULL;
1089         GroupObject *go = NULL;
1090         BMEditMesh *em;
1091         float ob__obmat[4][4]; /* needed for groups where the object matrix needs to be modified */
1092         CustomDataMask dm_mask;
1093         
1094         /* simple preventing of too deep nested groups */
1095         if (level > MAX_DUPLI_RECUR) return;
1096         
1097         copy_m4_m4(pmat, par->obmat);
1098         em = BKE_editmesh_from_object(par);
1099
1100         /* get derived mesh */
1101         dm_mask = CD_MASK_BAREMESH;
1102         if (flag & DUPLILIST_FOR_RENDER)
1103                 dm_mask |= CD_MASK_ORCO | CD_MASK_MLOOPUV;
1104
1105         if (em) {
1106                 dm = editbmesh_get_derived_cage(scene, par, em, dm_mask);
1107         }
1108         else {
1109                 dm = mesh_get_derived_final(scene, par, dm_mask);
1110         }
1111
1112         totface = dm->getNumPolys(dm);
1113         mpoly = dm->getPolyArray(dm);
1114         mloop = dm->getLoopArray(dm);
1115         mvert = dm->getVertArray(dm);
1116
1117         if (flag & DUPLILIST_FOR_RENDER) {
1118                 orco = dm->getVertDataArray(dm, CD_ORCO);
1119                 mloopuv = dm->getLoopDataArray(dm, CD_MLOOPUV);
1120         }
1121         else {
1122                 orco = NULL;
1123                 mloopuv = NULL;
1124         }
1125         
1126         /* having to loop on scene OR group objects is NOT FUN */
1127         if (GS(id->name) == ID_SCE) {
1128                 sce = (Scene *)id;
1129                 lay = sce->lay;
1130                 base = sce->base.first;
1131         }
1132         else {
1133                 group = (Group *)id;
1134                 lay = group->layer;
1135                 go = group->gobject.first;
1136         }
1137         
1138         /* Start looping on Scene OR Group objects */
1139         while (base || go) {
1140                 if (sce) {
1141                         ob_iter = base->object;
1142                         oblay = base->lay;
1143                 }
1144                 else {
1145                         ob_iter = go->ob;
1146                         oblay = ob_iter->lay;
1147                 }
1148                 
1149                 if (lay & oblay && scene->obedit != ob_iter) {
1150                         ob = ob_iter->parent;
1151                         while (ob) {
1152                                 if (ob == par) {
1153                                         ob = ob_iter;
1154                                         /* End Scene/Group object loop, below is generic */
1155                                         
1156                                         /* par_space_mat - only used for groups so we can modify the space dupli's are in
1157                                          * when par_space_mat is NULL ob->obmat can be used instead of ob__obmat
1158                                          */
1159                                         if (par_space_mat)
1160                                                 mul_m4_m4m4(ob__obmat, par_space_mat, ob->obmat);
1161                                         else
1162                                                 copy_m4_m4(ob__obmat, ob->obmat);
1163                                         
1164                                         copy_m3_m4(imat, ob->parentinv);
1165                                                 
1166                                         /* mballs have a different dupli handling */
1167                                         if (ob->type != OB_MBALL) ob->flag |= OB_DONE;  /* doesnt render */
1168
1169                                         for (a = 0, mp = mpoly; a < totface; a++, mp++) {
1170                                                 float *v1;
1171                                                 float *v2;
1172                                                 float *v3;
1173                                                 /* float *v4; */ /* UNUSED */
1174                                                 float cent[3], quat[4], mat[3][3], mat3[3][3], tmat[4][4], obmat[4][4];
1175                                                 float f_no[3];
1176                                                 MLoop *loopstart = mloop + mp->loopstart;
1177
1178                                                 if (UNLIKELY(mp->totloop < 3)) {
1179                                                         continue;
1180                                                 }
1181                                                 else {
1182                                                         BKE_mesh_calc_poly_normal(mp, mloop + mp->loopstart, mvert, f_no);
1183                                                         v1 = mvert[loopstart[0].v].co;
1184                                                         v2 = mvert[loopstart[1].v].co;
1185                                                         v3 = mvert[loopstart[2].v].co;
1186                                                 }
1187
1188                                                 /* translation */
1189                                                 BKE_mesh_calc_poly_center(mp, loopstart, mvert, cent);
1190
1191                                                 mul_m4_v3(pmat, cent);
1192                                                 
1193                                                 sub_v3_v3v3(cent, cent, pmat[3]);
1194                                                 add_v3_v3(cent, ob__obmat[3]);
1195                                                 
1196                                                 copy_m4_m4(obmat, ob__obmat);
1197                                                 
1198                                                 copy_v3_v3(obmat[3], cent);
1199                                                 
1200                                                 /* rotation */
1201                                                 tri_to_quat_ex(quat, v1, v2, v3, f_no);
1202                                                 quat_to_mat3(mat, quat);
1203                                                 
1204                                                 /* scale */
1205                                                 if (par->transflag & OB_DUPLIFACES_SCALE) {
1206                                                         float size = BKE_mesh_calc_poly_area(mp, loopstart, mvert, f_no);
1207                                                         size = sqrtf(size) * par->dupfacesca;
1208                                                         mul_m3_fl(mat, size);
1209                                                 }
1210                                                 
1211                                                 copy_m3_m3(mat3, mat);
1212                                                 mul_m3_m3m3(mat, imat, mat3);
1213                                                 
1214                                                 copy_m4_m4(tmat, obmat);
1215                                                 mul_m4_m4m3(obmat, tmat, mat);
1216                                                 
1217                                                 dob = new_dupli_object(lb, ob, obmat, par->lay, persistent_id, level, a, OB_DUPLIFACES, (flag & DUPLILIST_ANIMATED));
1218                                                 if (flag & DUPLILIST_FOR_RENDER) {
1219                                                         w = 1.0f / (float)mp->totloop;
1220
1221                                                         if (orco) {
1222                                                                 int j;
1223                                                                 for (j = 0; j < mpoly->totloop; j++) {
1224                                                                         madd_v3_v3fl(dob->orco, orco[loopstart[j].v], w);
1225                                                                 }
1226                                                         }
1227
1228                                                         if (mloopuv) {
1229                                                                 int j;
1230                                                                 for (j = 0; j < mpoly->totloop; j++) {
1231                                                                         madd_v2_v2fl(dob->uv, mloopuv[mp->loopstart + j].uv, w);
1232                                                                 }
1233                                                         }
1234                                                 }
1235                                                 
1236                                                 if (ob->transflag & OB_DUPLI) {
1237                                                         float tmpmat[4][4];
1238                                                         copy_m4_m4(tmpmat, ob->obmat);
1239                                                         copy_m4_m4(ob->obmat, obmat); /* pretend we are really this mat */
1240                                                         object_duplilist_recursive((ID *)id, scene, ob, lb, ob->obmat, persistent_id, level + 1, a, flag);
1241                                                         copy_m4_m4(ob->obmat, tmpmat);
1242                                                 }
1243                                         }
1244                                         
1245                                         break;
1246                                 }
1247                                 ob = ob->parent;
1248                         }
1249                 }
1250                 if (sce) base = base->next;     /* scene loop */
1251                 else go = go->next;             /* group loop */
1252         }
1253
1254         dm->release(dm);
1255 }
1256
1257 static void new_particle_duplilist(ListBase *lb, ID *id, Scene *scene, Object *par, float par_space_mat[4][4],
1258                                    int persistent_id[MAX_DUPLI_RECUR], ParticleSystem *psys,
1259                                    int level, short flag)
1260 {
1261         GroupObject *go;
1262         Object *ob = NULL, **oblist = NULL, obcopy, *obcopylist = NULL;
1263         DupliObject *dob;
1264         ParticleDupliWeight *dw;
1265         ParticleSettings *part;
1266         ParticleData *pa;
1267         ChildParticle *cpa = NULL;
1268         ParticleKey state;
1269         ParticleCacheKey *cache;
1270         float ctime, pa_time, scale = 1.0f;
1271         float tmat[4][4], mat[4][4], pamat[4][4], vec[3], size = 0.0;
1272         float (*obmat)[4], (*oldobmat)[4];
1273         int a, b, hair = 0;
1274         int totpart, totchild, totgroup = 0 /*, pa_num */;
1275         int dupli_type_hack = !BKE_scene_use_new_shading_nodes(scene);
1276
1277         int no_draw_flag = PARS_UNEXIST;
1278
1279         if (psys == NULL) return;
1280         
1281         /* simple preventing of too deep nested groups */
1282         if (level > MAX_DUPLI_RECUR) return;
1283         
1284         part = psys->part;
1285
1286         if (part == NULL)
1287                 return;
1288
1289         if (!psys_check_enabled(par, psys))
1290                 return;
1291
1292         if (G.is_rendering == FALSE)
1293                 no_draw_flag |= PARS_NO_DISP;
1294         
1295         ctime = BKE_scene_frame_get(scene); /* NOTE: in old animsys, used parent object's timeoffset... */
1296
1297         totpart = psys->totpart;
1298         totchild = psys->totchild;
1299
1300         BLI_srandom(31415926 + psys->seed);
1301
1302         if ((psys->renderdata || part->draw_as == PART_DRAW_REND) && ELEM(part->ren_as, PART_DRAW_OB, PART_DRAW_GR)) {
1303                 ParticleSimulationData sim = {NULL};
1304                 sim.scene = scene;
1305                 sim.ob = par;
1306                 sim.psys = psys;
1307                 sim.psmd = psys_get_modifier(par, psys);
1308                 /* make sure emitter imat is in global coordinates instead of render view coordinates */
1309                 invert_m4_m4(par->imat, par->obmat);
1310
1311                 /* first check for loops (particle system object used as dupli object) */
1312                 if (part->ren_as == PART_DRAW_OB) {
1313                         if (ELEM(part->dup_ob, NULL, par))
1314                                 return;
1315                 }
1316                 else { /*PART_DRAW_GR */
1317                         if (part->dup_group == NULL || part->dup_group->gobject.first == NULL)
1318                                 return;
1319
1320                         if (BLI_findptr(&part->dup_group->gobject, par, offsetof(GroupObject, ob))) {
1321                                 return;
1322                         }
1323                 }
1324
1325                 /* if we have a hair particle system, use the path cache */
1326                 if (part->type == PART_HAIR) {
1327                         if (psys->flag & PSYS_HAIR_DONE)
1328                                 hair = (totchild == 0 || psys->childcache) && psys->pathcache;
1329                         if (!hair)
1330                                 return;
1331                         
1332                         /* we use cache, update totchild according to cached data */
1333                         totchild = psys->totchildcache;
1334                         totpart = psys->totcached;
1335                 }
1336
1337                 psys_check_group_weights(part);
1338
1339                 psys->lattice_deform_data = psys_create_lattice_deform_data(&sim);
1340
1341                 /* gather list of objects or single object */
1342                 if (part->ren_as == PART_DRAW_GR) {
1343                         if (flag & DUPLILIST_DO_UPDATE) {
1344                                 BKE_group_handle_recalc_and_update(scene, par, part->dup_group);
1345                         }
1346
1347                         if (part->draw & PART_DRAW_COUNT_GR) {
1348                                 for (dw = part->dupliweights.first; dw; dw = dw->next)
1349                                         totgroup += dw->count;
1350                         }
1351                         else {
1352                                 for (go = part->dup_group->gobject.first; go; go = go->next)
1353                                         totgroup++;
1354                         }
1355
1356                         /* we also copy the actual objects to restore afterwards, since
1357                          * BKE_object_where_is_calc_time will change the object which breaks transform */
1358                         oblist = MEM_callocN(totgroup * sizeof(Object *), "dupgroup object list");
1359                         obcopylist = MEM_callocN(totgroup * sizeof(Object), "dupgroup copy list");
1360
1361                         
1362                         if (part->draw & PART_DRAW_COUNT_GR && totgroup) {
1363                                 dw = part->dupliweights.first;
1364
1365                                 for (a = 0; a < totgroup; dw = dw->next) {
1366                                         for (b = 0; b < dw->count; b++, a++) {
1367                                                 oblist[a] = dw->ob;
1368                                                 obcopylist[a] = *dw->ob;
1369                                         }
1370                                 }
1371                         }
1372                         else {
1373                                 go = part->dup_group->gobject.first;
1374                                 for (a = 0; a < totgroup; a++, go = go->next) {
1375                                         oblist[a] = go->ob;
1376                                         obcopylist[a] = *go->ob;
1377                                 }
1378                         }
1379                 }
1380                 else {
1381                         ob = part->dup_ob;
1382                         obcopy = *ob;
1383                 }
1384
1385                 if (totchild == 0 || part->draw & PART_DRAW_PARENT)
1386                         a = 0;
1387                 else
1388                         a = totpart;
1389
1390                 for (pa = psys->particles; a < totpart + totchild; a++, pa++) {
1391                         if (a < totpart) {
1392                                 /* handle parent particle */
1393                                 if (pa->flag & no_draw_flag)
1394                                         continue;
1395
1396                                 /* pa_num = pa->num; */ /* UNUSED */
1397                                 pa_time = pa->time;
1398                                 size = pa->size;
1399                         }
1400                         else {
1401                                 /* handle child particle */
1402                                 cpa = &psys->child[a - totpart];
1403
1404                                 /* pa_num = a; */ /* UNUSED */
1405                                 pa_time = psys->particles[cpa->parent].time;
1406                                 size = psys_get_child_size(psys, cpa, ctime, NULL);
1407                         }
1408
1409                         /* some hair paths might be non-existent so they can't be used for duplication */
1410                         if (hair &&
1411                             ((a < totpart && psys->pathcache[a]->steps < 0) ||
1412                              (a >= totpart && psys->childcache[a - totpart]->steps < 0)))
1413                         {
1414                                 continue;
1415                         }
1416
1417                         if (part->ren_as == PART_DRAW_GR) {
1418                                 /* prevent divide by zero below [#28336] */
1419                                 if (totgroup == 0)
1420                                         continue;
1421
1422                                 /* for groups, pick the object based on settings */
1423                                 if (part->draw & PART_DRAW_RAND_GR)
1424                                         b = BLI_rand() % totgroup;
1425                                 else
1426                                         b = a % totgroup;
1427
1428                                 ob = oblist[b];
1429                                 obmat = oblist[b]->obmat;
1430                                 oldobmat = obcopylist[b].obmat;
1431                         }
1432                         else {
1433                                 obmat = ob->obmat;
1434                                 oldobmat = obcopy.obmat;
1435                         }
1436
1437                         if (hair) {
1438                                 /* hair we handle separate and compute transform based on hair keys */
1439                                 if (a < totpart) {
1440                                         cache = psys->pathcache[a];
1441                                         psys_get_dupli_path_transform(&sim, pa, NULL, cache, pamat, &scale);
1442                                 }
1443                                 else {
1444                                         cache = psys->childcache[a - totpart];
1445                                         psys_get_dupli_path_transform(&sim, NULL, cpa, cache, pamat, &scale);
1446                                 }
1447
1448                                 copy_v3_v3(pamat[3], cache->co);
1449                                 pamat[3][3] = 1.0f;
1450                                 
1451                         }
1452                         else {
1453                                 /* first key */
1454                                 state.time = ctime;
1455                                 if (psys_get_particle_state(&sim, a, &state, 0) == 0) {
1456                                         continue;
1457                                 }
1458                                 else {
1459                                         float tquat[4];
1460                                         normalize_qt_qt(tquat, state.rot);
1461                                         quat_to_mat4(pamat, tquat);
1462                                         copy_v3_v3(pamat[3], state.co);
1463                                         pamat[3][3] = 1.0f;
1464                                 }
1465                         }
1466
1467                         if (part->ren_as == PART_DRAW_GR && psys->part->draw & PART_DRAW_WHOLE_GR) {
1468                                 for (go = part->dup_group->gobject.first, b = 0; go; go = go->next, b++) {
1469
1470                                         copy_m4_m4(tmat, oblist[b]->obmat);
1471                                         /* apply particle scale */
1472                                         mul_mat3_m4_fl(tmat, size * scale);
1473                                         mul_v3_fl(tmat[3], size * scale);
1474                                         /* group dupli offset, should apply after everything else */
1475                                         if (!is_zero_v3(part->dup_group->dupli_ofs))
1476                                                 sub_v3_v3v3(tmat[3], tmat[3], part->dup_group->dupli_ofs);
1477                                         /* individual particle transform */
1478                                         mul_m4_m4m4(tmat, pamat, tmat);
1479
1480                                         if (par_space_mat)
1481                                                 mul_m4_m4m4(mat, par_space_mat, tmat);
1482                                         else
1483                                                 copy_m4_m4(mat, tmat);
1484
1485                                         dob = new_dupli_object(lb, go->ob, mat, par->lay, persistent_id, level, a, OB_DUPLIPARTS, (flag & DUPLILIST_ANIMATED));
1486                                         dob->particle_system = psys;
1487                                         copy_m4_m4(dob->omat, obcopylist[b].obmat);
1488                                         if (flag & DUPLILIST_FOR_RENDER)
1489                                                 psys_get_dupli_texture(psys, part, sim.psmd, pa, cpa, dob->uv, dob->orco);
1490                                 }
1491                         }
1492                         else {
1493                                 int dupli_type = OB_DUPLIPARTS;
1494
1495                                 /* blender internal needs this to be set to dupligroup to render
1496                                  * groups correctly, but we don't want this hack for cycles */
1497                                 if (dupli_type_hack && GS(id->name) == ID_GR)
1498                                         dupli_type = OB_DUPLIGROUP;
1499
1500                                 /* to give ipos in object correct offset */
1501                                 BKE_object_where_is_calc_time(scene, ob, ctime - pa_time);
1502
1503                                 copy_v3_v3(vec, obmat[3]);
1504                                 obmat[3][0] = obmat[3][1] = obmat[3][2] = 0.0f;
1505
1506                                 /* particle rotation uses x-axis as the aligned axis, so pre-rotate the object accordingly */
1507                                 if ((part->draw & PART_DRAW_ROTATE_OB) == 0) {
1508                                         float xvec[3], q[4], size_mat[4][4], original_size[3];
1509
1510                                         mat4_to_size(original_size, obmat);
1511                                         size_to_mat4(size_mat, original_size);
1512
1513                                         xvec[0] = -1.f;
1514                                         xvec[1] = xvec[2] = 0;
1515                                         vec_to_quat(q, xvec, ob->trackflag, ob->upflag);
1516                                         quat_to_mat4(obmat, q);
1517                                         obmat[3][3] = 1.0f;
1518                                         
1519                                         /* add scaling if requested */
1520                                         if ((part->draw & PART_DRAW_NO_SCALE_OB) == 0)
1521                                                 mul_m4_m4m4(obmat, obmat, size_mat);
1522                                 }
1523                                 else if (part->draw & PART_DRAW_NO_SCALE_OB) {
1524                                         /* remove scaling */
1525                                         float size_mat[4][4], original_size[3];
1526
1527                                         mat4_to_size(original_size, obmat);
1528                                         size_to_mat4(size_mat, original_size);
1529                                         invert_m4(size_mat);
1530
1531                                         mul_m4_m4m4(obmat, obmat, size_mat);
1532                                 }
1533                                 
1534                                 /* Normal particles and cached hair live in global space so we need to
1535                                  * remove the real emitter's transformation before 2nd order duplication.
1536                                  */
1537                                 if (par_space_mat && GS(id->name) != ID_GR)
1538                                         mul_m4_m4m4(mat, psys->imat, pamat);
1539                                 else
1540                                         copy_m4_m4(mat, pamat);
1541
1542                                 mul_m4_m4m4(tmat, mat, obmat);
1543                                 mul_mat3_m4_fl(tmat, size * scale);
1544
1545                                 if (par_space_mat)
1546                                         mul_m4_m4m4(mat, par_space_mat, tmat);
1547                                 else
1548                                         copy_m4_m4(mat, tmat);
1549
1550                                 if (part->draw & PART_DRAW_GLOBAL_OB)
1551                                         add_v3_v3v3(mat[3], mat[3], vec);
1552
1553                                 dob = new_dupli_object(lb, ob, mat, ob->lay, persistent_id, level, a, dupli_type, (flag & DUPLILIST_ANIMATED));
1554                                 dob->particle_system = psys;
1555                                 copy_m4_m4(dob->omat, oldobmat);
1556                                 if (flag & DUPLILIST_FOR_RENDER)
1557                                         psys_get_dupli_texture(psys, part, sim.psmd, pa, cpa, dob->uv, dob->orco);
1558                         }
1559                 }
1560
1561                 /* restore objects since they were changed in BKE_object_where_is_calc_time */
1562                 if (part->ren_as == PART_DRAW_GR) {
1563                         for (a = 0; a < totgroup; a++)
1564                                 *(oblist[a]) = obcopylist[a];
1565                 }
1566                 else
1567                         *ob = obcopy;
1568         }
1569
1570         /* clean up */
1571         if (oblist)
1572                 MEM_freeN(oblist);
1573         if (obcopylist)
1574                 MEM_freeN(obcopylist);
1575
1576         if (psys->lattice_deform_data) {
1577                 end_latt_deform(psys->lattice_deform_data);
1578                 psys->lattice_deform_data = NULL;
1579         }
1580 }
1581
1582 static Object *find_family_object(Object **obar, char *family, char ch)
1583 {
1584         Object *ob;
1585         int flen;
1586         
1587         if (obar[(int)ch]) return obar[(int)ch];
1588         
1589         flen = strlen(family);
1590         
1591         ob = G.main->object.first;
1592         while (ob) {
1593                 if (ob->id.name[flen + 2] == ch) {
1594                         if (strncmp(ob->id.name + 2, family, flen) == 0) break;
1595                 }
1596                 ob = ob->id.next;
1597         }
1598         
1599         obar[(int)ch] = ob;
1600         
1601         return ob;
1602 }
1603
1604
1605 static void font_duplilist(ListBase *lb, Scene *scene, Object *par, int persistent_id[MAX_DUPLI_RECUR], int level, short flag)
1606 {
1607         Object *ob, *obar[256] = {NULL};
1608         Curve *cu;
1609         struct CharTrans *ct, *chartransdata;
1610         float vec[3], obmat[4][4], pmat[4][4], fsize, xof, yof;
1611         int slen, a;
1612         
1613         /* simple preventing of too deep nested groups */
1614         if (level > MAX_DUPLI_RECUR) return;
1615         
1616         copy_m4_m4(pmat, par->obmat);
1617         
1618         /* in par the family name is stored, use this to find the other objects */
1619         
1620         chartransdata = BKE_vfont_to_curve(G.main, scene, par, FO_DUPLI);
1621         if (chartransdata == NULL) return;
1622
1623         cu = par->data;
1624         slen = strlen(cu->str);
1625         fsize = cu->fsize;
1626         xof = cu->xof;
1627         yof = cu->yof;
1628         
1629         ct = chartransdata;
1630         
1631         for (a = 0; a < slen; a++, ct++) {
1632                 
1633                 ob = find_family_object(obar, cu->family, cu->str[a]);
1634                 if (ob) {
1635                         vec[0] = fsize * (ct->xof - xof);
1636                         vec[1] = fsize * (ct->yof - yof);
1637                         vec[2] = 0.0;
1638                         
1639                         mul_m4_v3(pmat, vec);
1640                         
1641                         copy_m4_m4(obmat, par->obmat);
1642                         copy_v3_v3(obmat[3], vec);
1643                         
1644                         new_dupli_object(lb, ob, obmat, par->lay, persistent_id, level, a, OB_DUPLIVERTS, flag);
1645                 }
1646         }
1647         
1648         MEM_freeN(chartransdata);
1649 }
1650
1651 /* ------------- */
1652
1653 static void object_duplilist_recursive(ID *id, Scene *scene, Object *ob, ListBase *duplilist, float par_space_mat[4][4],
1654                                        int persistent_id[MAX_DUPLI_RECUR], int level, int index, short flag)
1655 {       
1656         if ((ob->transflag & OB_DUPLI) == 0)
1657                 return;
1658         
1659         /* Should the dupli's be generated for this object? - Respect restrict flags */
1660         if (G.is_rendering) {
1661                 if (ob->restrictflag & OB_RESTRICT_RENDER) {
1662                         return;
1663                 }
1664         }
1665         else {
1666                 if (ob->restrictflag & OB_RESTRICT_VIEW) {
1667                         return;
1668                 }
1669         }
1670
1671         /* keep track of persistent id */
1672         if (level > 0)
1673                 persistent_id[level - 1] = index;
1674
1675         if (ob->transflag & OB_DUPLIPARTS) {
1676                 ParticleSystem *psys = ob->particlesystem.first;
1677                 int psysid = 0;
1678
1679                 /* particle system take up one level in id, the particles another */
1680                 for (; psys; psys = psys->next, psysid++) {
1681                         persistent_id[level] = psysid;
1682                         new_particle_duplilist(duplilist, id, scene, ob, par_space_mat, persistent_id, psys, level + 2, flag);
1683                 }
1684
1685                 persistent_id[level] = 0;
1686         }
1687         else if (ob->transflag & OB_DUPLIVERTS) {
1688                 if (ob->type == OB_MESH) {
1689                         vertex_duplilist(duplilist, id, scene, ob, par_space_mat, persistent_id, level + 1, flag);
1690                 }
1691                 else if (ob->type == OB_FONT) {
1692                         if (GS(id->name) == ID_SCE) { /* TODO - support dupligroups */
1693                                 font_duplilist(duplilist, scene, ob, persistent_id, level + 1, flag);
1694                         }
1695                 }
1696         }
1697         else if (ob->transflag & OB_DUPLIFACES) {
1698                 if (ob->type == OB_MESH)
1699                         face_duplilist(duplilist, id, scene, ob, par_space_mat, persistent_id, level + 1, flag);
1700         }
1701         else if (ob->transflag & OB_DUPLIFRAMES) {
1702                 if (GS(id->name) == ID_SCE) { /* TODO - support dupligroups */
1703                         frames_duplilist(duplilist, scene, ob, persistent_id, level + 1, flag);
1704                 }
1705         }
1706         else if (ob->transflag & OB_DUPLIGROUP) {
1707                 DupliObject *dob;
1708                 
1709                 group_duplilist(duplilist, scene, ob, persistent_id, level + 1, flag); /* now recursive */
1710
1711                 if (level == 0) {
1712                         for (dob = duplilist->first; dob; dob = dob->next)
1713                                 if (dob->type == OB_DUPLIGROUP)
1714                                         copy_m4_m4(dob->ob->obmat, dob->mat);
1715                 }
1716         }
1717
1718         /* clear persistent id */
1719         if (level > 0)
1720                 persistent_id[level - 1] = 0;
1721 }
1722
1723 /* Returns a list of DupliObject
1724  * note; group dupli's already set transform matrix. see note in group_duplilist() */
1725 ListBase *object_duplilist_ex(Scene *sce, Object *ob, bool update, bool for_render)
1726 {
1727         ListBase *duplilist = MEM_mallocN(sizeof(ListBase), "duplilist");
1728         int persistent_id[MAX_DUPLI_RECUR] = {0};
1729         int flag = 0;
1730
1731         /* don't allow BKE_object_handle_update for viewport during render, can crash */
1732         if (update && !(G.is_rendering && !for_render))
1733                 flag |= DUPLILIST_DO_UPDATE;
1734         if (for_render)
1735                 flag |= DUPLILIST_FOR_RENDER;
1736
1737         duplilist->first = duplilist->last = NULL;
1738         object_duplilist_recursive((ID *)sce, sce, ob, duplilist, NULL, persistent_id, 0, 0, flag);
1739         return duplilist;
1740 }
1741
1742 /* note: previously updating was always done, this is why it defaults to be on
1743  * but there are likely places it can be called without updating */
1744 ListBase *object_duplilist(Scene *sce, Object *ob, bool for_render)
1745 {
1746         return object_duplilist_ex(sce, ob, true, for_render);
1747 }
1748
1749
1750 void free_object_duplilist(ListBase *lb)
1751 {
1752         DupliObject *dob;
1753         
1754         /* loop in reverse order, if object is instanced multiple times
1755          * the original layer may not really be original otherwise, proper
1756          * solution is more complicated */
1757         for (dob = lb->last; dob; dob = dob->prev) {
1758                 dob->ob->lay = dob->origlay;
1759                 copy_m4_m4(dob->ob->obmat, dob->omat);
1760         }
1761         
1762         BLI_freelistN(lb);
1763         MEM_freeN(lb);
1764 }
1765
1766 int count_duplilist(Object *ob)
1767 {
1768         if (ob->transflag & OB_DUPLI) {
1769                 if (ob->transflag & OB_DUPLIVERTS) {
1770                         if (ob->type == OB_MESH) {
1771                                 if (ob->transflag & OB_DUPLIVERTS) {
1772                                         ParticleSystem *psys = ob->particlesystem.first;
1773                                         int pdup = 0;
1774
1775                                         for (; psys; psys = psys->next)
1776                                                 pdup += psys->totpart;
1777
1778                                         if (pdup == 0) {
1779                                                 Mesh *me = ob->data;
1780                                                 return me->totvert;
1781                                         }
1782                                         else
1783                                                 return pdup;
1784                                 }
1785                         }
1786                 }
1787                 else if (ob->transflag & OB_DUPLIFRAMES) {
1788                         int tot = ob->dupend - ob->dupsta;
1789                         tot /= (ob->dupon + ob->dupoff);
1790                         return tot * ob->dupon;
1791                 }
1792         }
1793         return 1;
1794 }