Merge branch 'master' into blender2.8
[blender.git] / source / blender / blenkernel / intern / object_dupli.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/object_dupli.c
29  *  \ingroup bke
30  */
31
32 #include <limits.h>
33 #include <stdlib.h>
34 #include <stddef.h>
35
36 #include "MEM_guardedalloc.h"
37
38 #include "BLI_listbase.h"
39 #include "BLI_string_utf8.h"
40
41 #include "BLI_math.h"
42 #include "BLI_rand.h"
43
44 #include "DNA_anim_types.h"
45 #include "DNA_group_types.h"
46 #include "DNA_mesh_types.h"
47 #include "DNA_scene_types.h"
48 #include "DNA_vfont_types.h"
49
50 #include "BKE_animsys.h"
51 #include "BKE_collection.h"
52 #include "BKE_DerivedMesh.h"
53 #include "BKE_font.h"
54 #include "BKE_global.h"
55 #include "BKE_idprop.h"
56 #include "BKE_lattice.h"
57 #include "BKE_main.h"
58 #include "BKE_mesh.h"
59 #include "BKE_mesh_runtime.h"
60 #include "BKE_object.h"
61 #include "BKE_particle.h"
62 #include "BKE_scene.h"
63 #include "BKE_editmesh.h"
64 #include "BKE_anim.h"
65
66 #include "DEG_depsgraph.h"
67 #include "DEG_depsgraph_query.h"
68
69 #include "BLI_strict_flags.h"
70 #include "BLI_hash.h"
71
72 /* Dupli-Geometry */
73
74 typedef struct DupliContext {
75         Depsgraph *depsgraph;
76         Collection *collection; /* XXX child objects are selected from this group if set, could be nicer */
77         Object *obedit; /* Only to check if the object is in edit-mode. */
78
79         Scene *scene;
80         ViewLayer *view_layer;
81         Object *object;
82         float space_mat[4][4];
83
84         int persistent_id[MAX_DUPLI_RECUR];
85         int level;
86
87         const struct DupliGenerator *gen;
88
89         /* result containers */
90         ListBase *duplilist; /* legacy doubly-linked list */
91 } DupliContext;
92
93 typedef struct DupliGenerator {
94         short type;                             /* dupli type */
95         void (*make_duplis)(const DupliContext *ctx);
96 } DupliGenerator;
97
98 static const DupliGenerator *get_dupli_generator(const DupliContext *ctx);
99
100 /* create initial context for root object */
101 static void init_context(
102         DupliContext *r_ctx, Depsgraph *depsgraph,
103         Scene *scene, Object *ob, float space_mat[4][4])
104 {
105         r_ctx->depsgraph = depsgraph;
106         r_ctx->scene = scene;
107         r_ctx->view_layer = DEG_get_evaluated_view_layer(depsgraph);
108         r_ctx->collection = NULL;
109
110         r_ctx->object = ob;
111         r_ctx->obedit = OBEDIT_FROM_OBACT(ob);
112         if (space_mat)
113                 copy_m4_m4(r_ctx->space_mat, space_mat);
114         else
115                 unit_m4(r_ctx->space_mat);
116         r_ctx->level = 0;
117
118         r_ctx->gen = get_dupli_generator(r_ctx);
119
120         r_ctx->duplilist = NULL;
121 }
122
123 /* create sub-context for recursive duplis */
124 static void copy_dupli_context(DupliContext *r_ctx, const DupliContext *ctx, Object *ob, float mat[4][4], int index)
125 {
126         *r_ctx = *ctx;
127         
128         /* XXX annoying, previously was done by passing an ID* argument, this at least is more explicit */
129         if (ctx->gen->type == OB_DUPLICOLLECTION)
130                 r_ctx->collection = ctx->object->dup_group;
131
132         r_ctx->object = ob;
133         if (mat)
134                 mul_m4_m4m4(r_ctx->space_mat, (float (*)[4])ctx->space_mat, mat);
135         r_ctx->persistent_id[r_ctx->level] = index;
136         ++r_ctx->level;
137
138         r_ctx->gen = get_dupli_generator(r_ctx);
139 }
140
141 /* generate a dupli instance
142  * mat is transform of the object relative to current context (including object obmat)
143  */
144 static DupliObject *make_dupli(const DupliContext *ctx,
145                                Object *ob, float mat[4][4], int index)
146 {
147         DupliObject *dob;
148         int i;
149
150         /* add a DupliObject instance to the result container */
151         if (ctx->duplilist) {
152                 dob = MEM_callocN(sizeof(DupliObject), "dupli object");
153                 BLI_addtail(ctx->duplilist, dob);
154         }
155         else {
156                 return NULL;
157         }
158
159         dob->ob = ob;
160         mul_m4_m4m4(dob->mat, (float (*)[4])ctx->space_mat, mat);
161         dob->type = ctx->gen->type;
162
163         /* set persistent id, which is an array with a persistent index for each level
164          * (particle number, vertex number, ..). by comparing this we can find the same
165          * dupli object between frames, which is needed for motion blur. last level
166          * goes first in the array. */
167         dob->persistent_id[0] = index;
168         for (i = 1; i < ctx->level + 1; i++)
169                 dob->persistent_id[i] = ctx->persistent_id[ctx->level - i];
170         /* fill rest of values with INT_MAX which index will never have as value */
171         for (; i < MAX_DUPLI_RECUR; i++)
172                 dob->persistent_id[i] = INT_MAX;
173
174         /* metaballs never draw in duplis, they are instead merged into one by the basis
175          * mball outside of the group. this does mean that if that mball is not in the
176          * scene, they will not show up at all, limitation that should be solved once. */
177         if (ob->type == OB_MBALL)
178                 dob->no_draw = true;
179
180         /* random number */
181         /* the logic here is designed to match Cycles */
182         dob->random_id = BLI_hash_string(dob->ob->id.name + 2);
183
184         if (dob->persistent_id[0] != INT_MAX) {
185                 for (i = 0; i < MAX_DUPLI_RECUR * 2; i++) {
186                         dob->random_id = BLI_hash_int_2d(dob->random_id, (unsigned int)dob->persistent_id[i]);
187                 }
188         }
189         else {
190                 dob->random_id = BLI_hash_int_2d(dob->random_id, 0);
191         }
192
193         if (ctx->object != ob) {
194                 dob->random_id ^= BLI_hash_int(BLI_hash_string(ctx->object->id.name + 2));
195         }
196
197         return dob;
198 }
199
200 /* recursive dupli objects
201  * space_mat is the local dupli space (excluding dupli object obmat!)
202  */
203 static void make_recursive_duplis(const DupliContext *ctx, Object *ob, float space_mat[4][4], int index)
204 {
205         /* simple preventing of too deep nested collections with MAX_DUPLI_RECUR */
206         if (ctx->level < MAX_DUPLI_RECUR) {
207                 DupliContext rctx;
208                 copy_dupli_context(&rctx, ctx, ob, space_mat, index);
209                 if (rctx.gen) {
210                         rctx.gen->make_duplis(&rctx);
211                 }
212         }
213 }
214
215 /* ---- Child Duplis ---- */
216
217 typedef void (*MakeChildDuplisFunc)(const DupliContext *ctx, void *userdata, Object *child);
218
219 static bool is_child(const Object *ob, const Object *parent)
220 {
221         const Object *ob_parent = ob->parent;
222         while (ob_parent) {
223                 if (ob_parent == parent)
224                         return true;
225                 ob_parent = ob_parent->parent;
226         }
227         return false;
228 }
229
230 /* create duplis from every child in scene or collection */
231 static void make_child_duplis(const DupliContext *ctx, void *userdata, MakeChildDuplisFunc make_child_duplis_cb)
232 {
233         Object *parent = ctx->object;
234
235         if (ctx->collection) {
236                 int collectionid = 0;
237                 FOREACH_COLLECTION_BASE_RECURSIVE_BEGIN(ctx->collection, base)
238                 {
239                         Object *ob = base->object;
240                         if ((base->flag & BASE_VISIBLED) && ob != ctx->obedit && is_child(ob, parent)) {
241                                 DupliContext pctx;
242                                 copy_dupli_context(&pctx, ctx, ctx->object, NULL, collectionid);
243
244                                 /* mballs have a different dupli handling */
245                                 if (ob->type != OB_MBALL) {
246                                         ob->flag |= OB_DONE;  /* doesnt render */
247                                 }
248                                 make_child_duplis_cb(&pctx, userdata, ob);
249                         }
250                         collectionid++;
251                 }
252                 FOREACH_COLLECTION_BASE_RECURSIVE_END
253         }
254         else {
255                 int baseid = 0;
256                 ViewLayer *view_layer = ctx->view_layer;
257                 for (Base *base = view_layer->object_bases.first; base; base = base->next, baseid++) {
258                         Object *ob = base->object;
259                         if ((ob != ctx->obedit) && is_child(ob, parent)) {
260                                 DupliContext pctx;
261                                 copy_dupli_context(&pctx, ctx, ctx->object, NULL, baseid);
262
263                                 /* mballs have a different dupli handling */
264                                 if (ob->type != OB_MBALL)
265                                         ob->flag |= OB_DONE;  /* doesnt render */
266
267                                 make_child_duplis_cb(&pctx, userdata, ob);
268                         }
269                 }
270         }
271 }
272
273
274 /*---- Implementations ----*/
275
276 /* OB_DUPLICOLLECTION */
277 static void make_duplis_collection(const DupliContext *ctx)
278 {
279         Object *ob = ctx->object;
280         Collection *collection;
281         Base *base;
282         float collection_mat[4][4];
283         int id;
284
285         if (ob->dup_group == NULL) return;
286         collection = ob->dup_group;
287
288         /* combine collection offset and obmat */
289         unit_m4(collection_mat);
290         sub_v3_v3(collection_mat[3], collection->dupli_ofs);
291         mul_m4_m4m4(collection_mat, ob->obmat, collection_mat);
292         /* don't access 'ob->obmat' from now on. */
293
294         const ListBase dup_collection_objects = BKE_collection_object_cache_get(collection);
295         for (base = dup_collection_objects.first, id = 0; base; base = base->next, id++) {
296                 if (base->object != ob && (base->flag & BASE_VISIBLED)) {
297                         float mat[4][4];
298
299                         /* collection dupli offset, should apply after everything else */
300                         mul_m4_m4m4(mat, collection_mat, base->object->obmat);
301
302                         make_dupli(ctx, base->object, mat, id);
303
304                         /* recursion */
305                         make_recursive_duplis(ctx, base->object, collection_mat, id);
306                 }
307         }
308 }
309
310 static const DupliGenerator gen_dupli_collection = {
311     OB_DUPLICOLLECTION,             /* type */
312     make_duplis_collection          /* make_duplis */
313 };
314
315 /* OB_DUPLIFRAMES */
316 static void make_duplis_frames(const DupliContext *ctx)
317 {
318         Depsgraph *depsgraph = ctx->depsgraph;
319         Scene *scene = ctx->scene;
320         Object *ob = ctx->object;
321         extern int enable_cu_speed; /* object.c */
322         Object copyob;
323         int cfrao = scene->r.cfra;
324         int dupend = ob->dupend;
325
326         /* dupliframes not supported inside collections */
327         if (ctx->collection)
328                 return;
329         /* if we don't have any data/settings which will lead to object movement,
330          * don't waste time trying, as it will all look the same...
331          */
332         if (ob->parent == NULL && BLI_listbase_is_empty(&ob->constraints) && ob->adt == NULL)
333                 return;
334
335         /* make a copy of the object's original data (before any dupli-data overwrites it)
336          * as we'll need this to keep track of unkeyed data
337          *      - this doesn't take into account other data that can be reached from the object,
338          *        for example it's shapekeys or bones, hence the need for an update flush at the end
339          */
340         copyob = *ob;
341
342         /* duplicate over the required range */
343         if (ob->transflag & OB_DUPLINOSPEED) enable_cu_speed = 0;
344
345         for (scene->r.cfra = ob->dupsta; scene->r.cfra <= dupend; scene->r.cfra++) {
346                 int ok = 1;
347
348                 /* - dupoff = how often a frames within the range shouldn't be made into duplis
349                  * - dupon = the length of each "skipping" block in frames
350                  */
351                 if (ob->dupoff) {
352                         ok = scene->r.cfra - ob->dupsta;
353                         ok = ok % (ob->dupon + ob->dupoff);
354                         ok = (ok < ob->dupon);
355                 }
356
357                 if (ok) {
358                         /* WARNING: doing animation updates in this way is not terribly accurate, as the dependencies
359                          * and/or other objects which may affect this object's transforms are not updated either.
360                          * However, this has always been the way that this worked (i.e. pre 2.5), so I guess that it'll be fine!
361                          */
362                         /* ob-eval will do drivers, so we don't need to do them */
363                         BKE_animsys_evaluate_animdata(depsgraph, scene, &ob->id, ob->adt, (float)scene->r.cfra, ADT_RECALC_ANIM);
364                         BKE_object_where_is_calc_time(depsgraph, scene, ob, (float)scene->r.cfra);
365
366                         make_dupli(ctx, ob, ob->obmat, scene->r.cfra);
367                 }
368         }
369
370         enable_cu_speed = 1;
371
372         /* reset frame to original frame, then re-evaluate animation as above
373          * as 2.5 animation data may have far-reaching consequences
374          */
375         scene->r.cfra = cfrao;
376
377         /* ob-eval will do drivers, so we don't need to do them */
378         BKE_animsys_evaluate_animdata(depsgraph, scene, &ob->id, ob->adt, (float)scene->r.cfra, ADT_RECALC_ANIM);
379         BKE_object_where_is_calc_time(depsgraph, scene, ob, (float)scene->r.cfra);
380
381         /* but, to make sure unkeyed object transforms are still sane,
382          * let's copy object's original data back over
383          */
384         *ob = copyob;
385 }
386
387 static const DupliGenerator gen_dupli_frames = {
388     OB_DUPLIFRAMES,                 /* type */
389     make_duplis_frames              /* make_duplis */
390 };
391
392 /* OB_DUPLIVERTS */
393 typedef struct VertexDupliData {
394         DerivedMesh *dm;
395         BMEditMesh *edit_btmesh;
396         int totvert;
397         float (*orco)[3];
398         bool use_rotation;
399
400         const DupliContext *ctx;
401         Object *inst_ob; /* object to instantiate (argument for vertex map callback) */
402         float child_imat[4][4];
403 } VertexDupliData;
404
405 static void get_duplivert_transform(const float co[3], const float nor_f[3], const short nor_s[3],
406                                     bool use_rotation, short axis, short upflag, float mat[4][4])
407 {
408         float quat[4];
409         const float size[3] = {1.0f, 1.0f, 1.0f};
410
411         if (use_rotation) {
412                 float nor[3];
413                 /* construct rotation matrix from normals */
414                 if (nor_f) {
415                         nor[0] = -nor_f[0];
416                         nor[1] = -nor_f[1];
417                         nor[2] = -nor_f[2];
418                 }
419                 else if (nor_s) {
420                         nor[0] = (float)-nor_s[0];
421                         nor[1] = (float)-nor_s[1];
422                         nor[2] = (float)-nor_s[2];
423                 }
424                 vec_to_quat(quat, nor, axis, upflag);
425         }
426         else
427                 unit_qt(quat);
428
429         loc_quat_size_to_mat4(mat, co, quat, size);
430 }
431
432 static void vertex_dupli__mapFunc(void *userData, int index, const float co[3],
433                                   const float nor_f[3], const short nor_s[3])
434 {
435         const VertexDupliData *vdd = userData;
436         Object *inst_ob = vdd->inst_ob;
437         DupliObject *dob;
438         float obmat[4][4], space_mat[4][4];
439
440         /* obmat is transform to vertex */
441         get_duplivert_transform(co, nor_f, nor_s, vdd->use_rotation, inst_ob->trackflag, inst_ob->upflag, obmat);
442         /* make offset relative to inst_ob using relative child transform */
443         mul_mat3_m4_v3((float (*)[4])vdd->child_imat, obmat[3]);
444         /* apply obmat _after_ the local vertex transform */
445         mul_m4_m4m4(obmat, inst_ob->obmat, obmat);
446
447         /* space matrix is constructed by removing obmat transform,
448          * this yields the worldspace transform for recursive duplis
449          */
450         mul_m4_m4m4(space_mat, obmat, inst_ob->imat);
451
452         dob = make_dupli(vdd->ctx, vdd->inst_ob, obmat, index);
453
454         if (vdd->orco)
455                 copy_v3_v3(dob->orco, vdd->orco[index]);
456
457         /* recursion */
458         make_recursive_duplis(vdd->ctx, vdd->inst_ob, space_mat, index);
459 }
460
461 static void make_child_duplis_verts(const DupliContext *ctx, void *userdata, Object *child)
462 {
463         VertexDupliData *vdd = userdata;
464         DerivedMesh *dm = vdd->dm;
465
466         vdd->inst_ob = child;
467         invert_m4_m4(child->imat, child->obmat);
468         /* relative transform from parent to child space */
469         mul_m4_m4m4(vdd->child_imat, child->imat, ctx->object->obmat);
470
471         if (vdd->edit_btmesh) {
472                 dm->foreachMappedVert(dm, vertex_dupli__mapFunc, vdd,
473                                       vdd->use_rotation ? DM_FOREACH_USE_NORMAL : 0);
474         }
475         else {
476                 int a, totvert = vdd->totvert;
477                 float vec[3], no[3];
478
479                 if (vdd->use_rotation) {
480                         for (a = 0; a < totvert; a++) {
481                                 dm->getVertCo(dm, a, vec);
482                                 dm->getVertNo(dm, a, no);
483
484                                 vertex_dupli__mapFunc(vdd, a, vec, no, NULL);
485                         }
486                 }
487                 else {
488                         for (a = 0; a < totvert; a++) {
489                                 dm->getVertCo(dm, a, vec);
490
491                                 vertex_dupli__mapFunc(vdd, a, vec, NULL, NULL);
492                         }
493                 }
494         }
495 }
496
497 static void make_duplis_verts(const DupliContext *ctx)
498 {
499         Scene *scene = ctx->scene;
500         Object *parent = ctx->object;
501         bool use_texcoords = (DEG_get_mode(ctx->depsgraph) == DAG_EVAL_RENDER);
502         VertexDupliData vdd;
503
504         vdd.ctx = ctx;
505         vdd.use_rotation = parent->transflag & OB_DUPLIROT;
506
507         /* gather mesh info */
508         {
509                 Mesh *me = parent->data;
510                 BMEditMesh *em = BKE_editmesh_from_object(parent);
511                 CustomDataMask dm_mask = (use_texcoords ? CD_MASK_BAREMESH | CD_MASK_ORCO : CD_MASK_BAREMESH);
512
513                 if (DEG_get_mode(ctx->depsgraph) == DAG_EVAL_RENDER) {
514                         vdd.dm = mesh_create_derived_render(ctx->depsgraph, scene, parent, dm_mask);
515                 }
516                 else if (em) {
517                         vdd.dm = editbmesh_get_derived_cage(ctx->depsgraph, scene, parent, em, dm_mask);
518                 }
519                 else {
520                         vdd.dm = mesh_get_derived_final(ctx->depsgraph, scene, parent, dm_mask);
521                 }
522                 vdd.edit_btmesh = me->edit_btmesh;
523
524                 if (use_texcoords)
525                         vdd.orco = vdd.dm->getVertDataArray(vdd.dm, CD_ORCO);
526                 else
527                         vdd.orco = NULL;
528
529                 vdd.totvert = vdd.dm->getNumVerts(vdd.dm);
530         }
531
532         make_child_duplis(ctx, &vdd, make_child_duplis_verts);
533
534         vdd.dm->release(vdd.dm);
535 }
536
537 static const DupliGenerator gen_dupli_verts = {
538     OB_DUPLIVERTS,                  /* type */
539     make_duplis_verts               /* make_duplis */
540 };
541
542 /* OB_DUPLIVERTS - FONT */
543 static Object *find_family_object(Main *bmain, const char *family, size_t family_len, unsigned int ch, GHash *family_gh)
544 {
545         Object **ob_pt;
546         Object *ob;
547         void *ch_key = SET_UINT_IN_POINTER(ch);
548
549         if ((ob_pt = (Object **)BLI_ghash_lookup_p(family_gh, ch_key))) {
550                 ob = *ob_pt;
551         }
552         else {
553                 char ch_utf8[7];
554                 size_t ch_utf8_len;
555
556                 ch_utf8_len = BLI_str_utf8_from_unicode(ch, ch_utf8);
557                 ch_utf8[ch_utf8_len] = '\0';
558                 ch_utf8_len += 1;  /* compare with null terminator */
559
560                 for (ob = bmain->object.first; ob; ob = ob->id.next) {
561                         if (STREQLEN(ob->id.name + 2 + family_len, ch_utf8, ch_utf8_len)) {
562                                 if (STREQLEN(ob->id.name + 2, family, family_len)) {
563                                         break;
564                                 }
565                         }
566                 }
567
568                 /* inserted value can be NULL, just to save searches in future */
569                 BLI_ghash_insert(family_gh, ch_key, ob);
570         }
571
572         return ob;
573 }
574
575 static void make_duplis_font(const DupliContext *ctx)
576 {
577         Object *par = ctx->object;
578         GHash *family_gh;
579         Object *ob;
580         Curve *cu;
581         struct CharTrans *ct, *chartransdata = NULL;
582         float vec[3], obmat[4][4], pmat[4][4], fsize, xof, yof;
583         int text_len, a;
584         size_t family_len;
585         const wchar_t *text = NULL;
586         bool text_free = false;
587
588         /* font dupliverts not supported inside collections */
589         if (ctx->collection)
590                 return;
591
592         copy_m4_m4(pmat, par->obmat);
593
594         /* in par the family name is stored, use this to find the other objects */
595
596         BKE_vfont_to_curve_ex(par, par->data, FO_DUPLI, NULL,
597                               &text, &text_len, &text_free, &chartransdata);
598
599         if (text == NULL || chartransdata == NULL) {
600                 return;
601         }
602
603         cu = par->data;
604         fsize = cu->fsize;
605         xof = cu->xof;
606         yof = cu->yof;
607
608         ct = chartransdata;
609
610         /* cache result */
611         family_len = strlen(cu->family);
612         family_gh = BLI_ghash_int_new_ex(__func__, 256);
613
614         /* advance matching BLI_strncpy_wchar_from_utf8 */
615         for (a = 0; a < text_len; a++, ct++) {
616
617                 /* XXX That G.main is *really* ugly, but not sure what to do here...
618                  * Definitively don't think it would be safe to put back Main *bmain pointer in DupliContext as done in 2.7x? */
619                 ob = find_family_object(G.main, cu->family, family_len, (unsigned int)text[a], family_gh);
620                 if (ob) {
621                         vec[0] = fsize * (ct->xof - xof);
622                         vec[1] = fsize * (ct->yof - yof);
623                         vec[2] = 0.0;
624
625                         mul_m4_v3(pmat, vec);
626
627                         copy_m4_m4(obmat, par->obmat);
628
629                         if (UNLIKELY(ct->rot != 0.0f)) {
630                                 float rmat[4][4];
631
632                                 zero_v3(obmat[3]);
633                                 axis_angle_to_mat4_single(rmat, 'Z', -ct->rot);
634                                 mul_m4_m4m4(obmat, obmat, rmat);
635                         }
636
637                         copy_v3_v3(obmat[3], vec);
638
639                         make_dupli(ctx, ob, obmat, a);
640                 }
641         }
642
643         if (text_free) {
644                 MEM_freeN((void *)text);
645         }
646
647         BLI_ghash_free(family_gh, NULL, NULL);
648
649         MEM_freeN(chartransdata);
650 }
651
652 static const DupliGenerator gen_dupli_verts_font = {
653     OB_DUPLIVERTS,                  /* type */
654     make_duplis_font                /* make_duplis */
655 };
656
657 /* OB_DUPLIFACES */
658 typedef struct FaceDupliData {
659         DerivedMesh *dm;
660         int totface;
661         MPoly *mpoly;
662         MLoop *mloop;
663         MVert *mvert;
664         float (*orco)[3];
665         MLoopUV *mloopuv;
666         bool use_scale;
667 } FaceDupliData;
668
669 static void get_dupliface_transform(MPoly *mpoly, MLoop *mloop, MVert *mvert,
670                                     bool use_scale, float scale_fac, float mat[4][4])
671 {
672         float loc[3], quat[4], scale, size[3];
673         float f_no[3];
674
675         /* location */
676         BKE_mesh_calc_poly_center(mpoly, mloop, mvert, loc);
677         /* rotation */
678         {
679                 const float *v1, *v2, *v3;
680                 BKE_mesh_calc_poly_normal(mpoly, mloop, mvert, f_no);
681                 v1 = mvert[mloop[0].v].co;
682                 v2 = mvert[mloop[1].v].co;
683                 v3 = mvert[mloop[2].v].co;
684                 tri_to_quat_ex(quat, v1, v2, v3, f_no);
685         }
686         /* scale */
687         if (use_scale) {
688                 float area = BKE_mesh_calc_poly_area(mpoly, mloop, mvert);
689                 scale = sqrtf(area) * scale_fac;
690         }
691         else
692                 scale = 1.0f;
693         size[0] = size[1] = size[2] = scale;
694
695         loc_quat_size_to_mat4(mat, loc, quat, size);
696 }
697
698 static void make_child_duplis_faces(const DupliContext *ctx, void *userdata, Object *inst_ob)
699 {
700         FaceDupliData *fdd = userdata;
701         MPoly *mpoly = fdd->mpoly, *mp;
702         MLoop *mloop = fdd->mloop;
703         MVert *mvert = fdd->mvert;
704         float (*orco)[3] = fdd->orco;
705         MLoopUV *mloopuv = fdd->mloopuv;
706         int a, totface = fdd->totface;
707         bool use_texcoords = (DEG_get_mode(ctx->depsgraph) == DAG_EVAL_RENDER);
708         float child_imat[4][4];
709         DupliObject *dob;
710
711         invert_m4_m4(inst_ob->imat, inst_ob->obmat);
712         /* relative transform from parent to child space */
713         mul_m4_m4m4(child_imat, inst_ob->imat, ctx->object->obmat);
714
715         for (a = 0, mp = mpoly; a < totface; a++, mp++) {
716                 MLoop *loopstart = mloop + mp->loopstart;
717                 float space_mat[4][4], obmat[4][4];
718
719                 if (UNLIKELY(mp->totloop < 3))
720                         continue;
721
722                 /* obmat is transform to face */
723                 get_dupliface_transform(mp, loopstart, mvert, fdd->use_scale, ctx->object->dupfacesca, obmat);
724                 /* make offset relative to inst_ob using relative child transform */
725                 mul_mat3_m4_v3(child_imat, obmat[3]);
726
727                 /* XXX ugly hack to ensure same behavior as in master
728                  * this should not be needed, parentinv is not consistent
729                  * outside of parenting.
730                  */
731                 {
732                         float imat[3][3];
733                         copy_m3_m4(imat, inst_ob->parentinv);
734                         mul_m4_m3m4(obmat, imat, obmat);
735                 }
736
737                 /* apply obmat _after_ the local face transform */
738                 mul_m4_m4m4(obmat, inst_ob->obmat, obmat);
739
740                 /* space matrix is constructed by removing obmat transform,
741                  * this yields the worldspace transform for recursive duplis
742                  */
743                 mul_m4_m4m4(space_mat, obmat, inst_ob->imat);
744
745                 dob = make_dupli(ctx, inst_ob, obmat, a);
746                 if (use_texcoords) {
747                         float w = 1.0f / (float)mp->totloop;
748
749                         if (orco) {
750                                 int j;
751                                 for (j = 0; j < mp->totloop; j++) {
752                                         madd_v3_v3fl(dob->orco, orco[loopstart[j].v], w);
753                                 }
754                         }
755
756                         if (mloopuv) {
757                                 int j;
758                                 for (j = 0; j < mp->totloop; j++) {
759                                         madd_v2_v2fl(dob->uv, mloopuv[mp->loopstart + j].uv, w);
760                                 }
761                         }
762                 }
763
764                 /* recursion */
765                 make_recursive_duplis(ctx, inst_ob, space_mat, a);
766         }
767 }
768
769 static void make_duplis_faces(const DupliContext *ctx)
770 {
771         Scene *scene = ctx->scene;
772         Object *parent = ctx->object;
773         bool use_texcoords = (DEG_get_mode(ctx->depsgraph) == DAG_EVAL_RENDER);
774         FaceDupliData fdd;
775
776         fdd.use_scale = ((parent->transflag & OB_DUPLIFACES_SCALE) != 0);
777
778         /* gather mesh info */
779         {
780                 BMEditMesh *em = BKE_editmesh_from_object(parent);
781                 CustomDataMask dm_mask = (use_texcoords ? CD_MASK_BAREMESH | CD_MASK_ORCO | CD_MASK_MLOOPUV : CD_MASK_BAREMESH);
782
783                 if (DEG_get_mode(ctx->depsgraph) == DAG_EVAL_RENDER) {
784                         fdd.dm = mesh_create_derived_render(ctx->depsgraph, scene, parent, dm_mask);
785                 }
786                 else if (em) {
787                         fdd.dm = editbmesh_get_derived_cage(ctx->depsgraph, scene, parent, em, dm_mask);
788                 }
789                 else {
790                         fdd.dm = mesh_get_derived_final(ctx->depsgraph, scene, parent, dm_mask);
791                 }
792
793                 if (use_texcoords) {
794                         CustomData *ml_data = fdd.dm->getLoopDataLayout(fdd.dm);
795                         const int uv_idx = CustomData_get_render_layer(ml_data, CD_MLOOPUV);
796                         fdd.orco = fdd.dm->getVertDataArray(fdd.dm, CD_ORCO);
797                         fdd.mloopuv = CustomData_get_layer_n(ml_data, CD_MLOOPUV, uv_idx);
798                 }
799                 else {
800                         fdd.orco = NULL;
801                         fdd.mloopuv = NULL;
802                 }
803
804                 fdd.totface = fdd.dm->getNumPolys(fdd.dm);
805                 fdd.mpoly = fdd.dm->getPolyArray(fdd.dm);
806                 fdd.mloop = fdd.dm->getLoopArray(fdd.dm);
807                 fdd.mvert = fdd.dm->getVertArray(fdd.dm);
808         }
809
810         make_child_duplis(ctx, &fdd, make_child_duplis_faces);
811
812         fdd.dm->release(fdd.dm);
813 }
814
815 static const DupliGenerator gen_dupli_faces = {
816     OB_DUPLIFACES,                  /* type */
817     make_duplis_faces               /* make_duplis */
818 };
819
820 /* OB_DUPLIPARTS */
821 static void make_duplis_particle_system(const DupliContext *ctx, ParticleSystem *psys)
822 {
823         Scene *scene = ctx->scene;
824         Object *par = ctx->object;
825         bool for_render = DEG_get_mode(ctx->depsgraph) == DAG_EVAL_RENDER;
826         bool use_texcoords = for_render;
827
828         Object *ob = NULL, **oblist = NULL, obcopy, *obcopylist = NULL;
829         DupliObject *dob;
830         ParticleDupliWeight *dw;
831         ParticleSettings *part;
832         ParticleData *pa;
833         ChildParticle *cpa = NULL;
834         ParticleKey state;
835         ParticleCacheKey *cache;
836         float ctime, pa_time, scale = 1.0f;
837         float tmat[4][4], mat[4][4], pamat[4][4], vec[3], size = 0.0;
838         float (*obmat)[4];
839         int a, b, hair = 0;
840         int totpart, totchild, totcollection = 0 /*, pa_num */;
841         RNG *rng;
842
843         int no_draw_flag = PARS_UNEXIST;
844
845         if (psys == NULL) return;
846
847         part = psys->part;
848
849         if (part == NULL)
850                 return;
851
852         if (!psys_check_enabled(par, psys, (DEG_get_mode(ctx->depsgraph) == DAG_EVAL_RENDER)))
853                 return;
854
855         if (!for_render)
856                 no_draw_flag |= PARS_NO_DISP;
857
858         ctime = DEG_get_ctime(ctx->depsgraph); /* NOTE: in old animsys, used parent object's timeoffset... */
859
860         totpart = psys->totpart;
861         totchild = psys->totchild;
862
863         rng = BLI_rng_new_srandom(31415926u + (unsigned int)psys->seed);
864
865         if ((for_render || part->draw_as == PART_DRAW_REND) && ELEM(part->ren_as, PART_DRAW_OB, PART_DRAW_GR)) {
866                 ParticleSimulationData sim = {NULL};
867                 sim.depsgraph = ctx->depsgraph;
868                 sim.scene = scene;
869                 sim.ob = par;
870                 sim.psys = psys;
871                 sim.psmd = psys_get_modifier(par, psys);
872                 /* make sure emitter imat is in global coordinates instead of render view coordinates */
873                 invert_m4_m4(par->imat, par->obmat);
874
875                 /* first check for loops (particle system object used as dupli object) */
876                 if (part->ren_as == PART_DRAW_OB) {
877                         if (ELEM(part->dup_ob, NULL, par))
878                                 return;
879                 }
880                 else { /*PART_DRAW_GR */
881                         if (part->dup_group == NULL)
882                                 return;
883
884                         const ListBase dup_collection_objects = BKE_collection_object_cache_get(part->dup_group);
885                         if (BLI_listbase_is_empty(&dup_collection_objects))
886                                 return;
887
888                         if (BLI_findptr(&dup_collection_objects, par, offsetof(Base, object))) {
889                                 return;
890                         }
891                 }
892
893                 /* if we have a hair particle system, use the path cache */
894                 if (part->type == PART_HAIR) {
895                         if (psys->flag & PSYS_HAIR_DONE)
896                                 hair = (totchild == 0 || psys->childcache) && psys->pathcache;
897                         if (!hair)
898                                 return;
899
900                         /* we use cache, update totchild according to cached data */
901                         totchild = psys->totchildcache;
902                         totpart = psys->totcached;
903                 }
904
905                 psys_check_group_weights(part);
906
907                 psys->lattice_deform_data = psys_create_lattice_deform_data(&sim);
908
909                 /* gather list of objects or single object */
910                 if (part->ren_as == PART_DRAW_GR) {
911                         if (part->draw & PART_DRAW_COUNT_GR) {
912                                 for (dw = part->dupliweights.first; dw; dw = dw->next)
913                                         totcollection += dw->count;
914                         }
915                         else {
916                                 FOREACH_COLLECTION_OBJECT_RECURSIVE_BEGIN(part->dup_group, object)
917                                 {
918                                         (void) object;
919                                         totcollection++;
920                                 }
921                                 FOREACH_COLLECTION_OBJECT_RECURSIVE_END;
922                         }
923
924                         /* we also copy the actual objects to restore afterwards, since
925                          * BKE_object_where_is_calc_time will change the object which breaks transform */
926                         oblist = MEM_callocN((size_t)totcollection * sizeof(Object *), "dupcollection object list");
927                         obcopylist = MEM_callocN((size_t)totcollection * sizeof(Object), "dupcollection copy list");
928
929                         if (part->draw & PART_DRAW_COUNT_GR && totcollection) {
930                                 dw = part->dupliweights.first;
931
932                                 for (a = 0; a < totcollection; dw = dw->next) {
933                                         for (b = 0; b < dw->count; b++, a++) {
934                                                 oblist[a] = dw->ob;
935                                                 obcopylist[a] = *dw->ob;
936                                         }
937                                 }
938                         }
939                         else {
940                                 a = 0;
941                                 FOREACH_COLLECTION_OBJECT_RECURSIVE_BEGIN(part->dup_group, object)
942                                 {
943                                         oblist[a] = object;
944                                         obcopylist[a] = *object;
945                                         a++;
946
947                                         if (a >= totcollection) {
948                                                 continue;
949                                         }
950                                 }
951                                 FOREACH_COLLECTION_OBJECT_RECURSIVE_END;
952                         }
953                 }
954                 else {
955                         ob = part->dup_ob;
956                         obcopy = *ob;
957                 }
958
959                 if (totchild == 0 || part->draw & PART_DRAW_PARENT)
960                         a = 0;
961                 else
962                         a = totpart;
963
964                 for (pa = psys->particles; a < totpart + totchild; a++, pa++) {
965                         if (a < totpart) {
966                                 /* handle parent particle */
967                                 if (pa->flag & no_draw_flag)
968                                         continue;
969
970                                 /* pa_num = pa->num; */ /* UNUSED */
971                                 pa_time = pa->time;
972                                 size = pa->size;
973                         }
974                         else {
975                                 /* handle child particle */
976                                 cpa = &psys->child[a - totpart];
977
978                                 /* pa_num = a; */ /* UNUSED */
979                                 pa_time = psys->particles[cpa->parent].time;
980                                 size = psys_get_child_size(psys, cpa, ctime, NULL);
981                         }
982
983                         /* some hair paths might be non-existent so they can't be used for duplication */
984                         if (hair && psys->pathcache &&
985                             ((a < totpart && psys->pathcache[a]->segments < 0) ||
986                              (a >= totpart && psys->childcache[a - totpart]->segments < 0)))
987                         {
988                                 continue;
989                         }
990
991                         if (part->ren_as == PART_DRAW_GR) {
992                                 /* prevent divide by zero below [#28336] */
993                                 if (totcollection == 0)
994                                         continue;
995
996                                 /* for collections, pick the object based on settings */
997                                 if (part->draw & PART_DRAW_RAND_GR)
998                                         b = BLI_rng_get_int(rng) % totcollection;
999                                 else
1000                                         b = a % totcollection;
1001
1002                                 ob = oblist[b];
1003                                 obmat = oblist[b]->obmat;
1004                         }
1005                         else {
1006                                 obmat = ob->obmat;
1007                         }
1008
1009                         if (hair) {
1010                                 /* hair we handle separate and compute transform based on hair keys */
1011                                 if (a < totpart) {
1012                                         cache = psys->pathcache[a];
1013                                         psys_get_dupli_path_transform(&sim, pa, NULL, cache, pamat, &scale);
1014                                 }
1015                                 else {
1016                                         cache = psys->childcache[a - totpart];
1017                                         psys_get_dupli_path_transform(&sim, NULL, cpa, cache, pamat, &scale);
1018                                 }
1019
1020                                 copy_v3_v3(pamat[3], cache->co);
1021                                 pamat[3][3] = 1.0f;
1022
1023                         }
1024                         else {
1025                                 /* first key */
1026                                 state.time = ctime;
1027                                 if (psys_get_particle_state(&sim, a, &state, 0) == 0) {
1028                                         continue;
1029                                 }
1030                                 else {
1031                                         float tquat[4];
1032                                         normalize_qt_qt(tquat, state.rot);
1033                                         quat_to_mat4(pamat, tquat);
1034                                         copy_v3_v3(pamat[3], state.co);
1035                                         pamat[3][3] = 1.0f;
1036                                 }
1037                         }
1038
1039                         if (part->ren_as == PART_DRAW_GR && psys->part->draw & PART_DRAW_WHOLE_GR) {
1040                                 b = 0;
1041                                 FOREACH_COLLECTION_OBJECT_RECURSIVE_BEGIN(part->dup_group, object)
1042                                 {
1043                                         copy_m4_m4(tmat, oblist[b]->obmat);
1044
1045                                         /* apply particle scale */
1046                                         mul_mat3_m4_fl(tmat, size * scale);
1047                                         mul_v3_fl(tmat[3], size * scale);
1048
1049                                         /* collection dupli offset, should apply after everything else */
1050                                         if (!is_zero_v3(part->dup_group->dupli_ofs)) {
1051                                                 sub_v3_v3(tmat[3], part->dup_group->dupli_ofs);
1052                                         }
1053
1054                                         /* individual particle transform */
1055                                         mul_m4_m4m4(mat, pamat, tmat);
1056
1057                                         dob = make_dupli(ctx, object, mat, a);
1058                                         dob->particle_system = psys;
1059
1060                                         if (use_texcoords) {
1061                                                 psys_get_dupli_texture(psys, part, sim.psmd, pa, cpa, dob->uv, dob->orco);
1062                                         }
1063
1064                                         b++;
1065                                 }
1066                                 FOREACH_COLLECTION_OBJECT_RECURSIVE_END;
1067                         }
1068                         else {
1069                                 /* to give ipos in object correct offset */
1070                                 BKE_object_where_is_calc_time(ctx->depsgraph, scene, ob, ctime - pa_time);
1071
1072                                 copy_v3_v3(vec, obmat[3]);
1073                                 obmat[3][0] = obmat[3][1] = obmat[3][2] = 0.0f;
1074
1075                                 /* particle rotation uses x-axis as the aligned axis, so pre-rotate the object accordingly */
1076                                 if ((part->draw & PART_DRAW_ROTATE_OB) == 0) {
1077                                         float xvec[3], q[4], size_mat[4][4], original_size[3];
1078
1079                                         mat4_to_size(original_size, obmat);
1080                                         size_to_mat4(size_mat, original_size);
1081
1082                                         xvec[0] = -1.f;
1083                                         xvec[1] = xvec[2] = 0;
1084                                         vec_to_quat(q, xvec, ob->trackflag, ob->upflag);
1085                                         quat_to_mat4(obmat, q);
1086                                         obmat[3][3] = 1.0f;
1087
1088                                         /* add scaling if requested */
1089                                         if ((part->draw & PART_DRAW_NO_SCALE_OB) == 0)
1090                                                 mul_m4_m4m4(obmat, obmat, size_mat);
1091                                 }
1092                                 else if (part->draw & PART_DRAW_NO_SCALE_OB) {
1093                                         /* remove scaling */
1094                                         float size_mat[4][4], original_size[3];
1095
1096                                         mat4_to_size(original_size, obmat);
1097                                         size_to_mat4(size_mat, original_size);
1098                                         invert_m4(size_mat);
1099
1100                                         mul_m4_m4m4(obmat, obmat, size_mat);
1101                                 }
1102
1103                                 mul_m4_m4m4(tmat, pamat, obmat);
1104                                 mul_mat3_m4_fl(tmat, size * scale);
1105
1106                                 copy_m4_m4(mat, tmat);
1107
1108                                 if (part->draw & PART_DRAW_GLOBAL_OB)
1109                                         add_v3_v3v3(mat[3], mat[3], vec);
1110
1111                                 dob = make_dupli(ctx, ob, mat, a);
1112                                 dob->particle_system = psys;
1113                                 if (use_texcoords)
1114                                         psys_get_dupli_texture(psys, part, sim.psmd, pa, cpa, dob->uv, dob->orco);
1115                         }
1116                 }
1117
1118                 /* restore objects since they were changed in BKE_object_where_is_calc_time */
1119                 if (part->ren_as == PART_DRAW_GR) {
1120                         for (a = 0; a < totcollection; a++)
1121                                 *(oblist[a]) = obcopylist[a];
1122                 }
1123                 else
1124                         *ob = obcopy;
1125         }
1126
1127         /* clean up */
1128         if (oblist)
1129                 MEM_freeN(oblist);
1130         if (obcopylist)
1131                 MEM_freeN(obcopylist);
1132
1133         if (psys->lattice_deform_data) {
1134                 end_latt_deform(psys->lattice_deform_data);
1135                 psys->lattice_deform_data = NULL;
1136         }
1137
1138         BLI_rng_free(rng);
1139 }
1140
1141 static void make_duplis_particles(const DupliContext *ctx)
1142 {
1143         ParticleSystem *psys;
1144         int psysid;
1145
1146         /* particle system take up one level in id, the particles another */
1147         for (psys = ctx->object->particlesystem.first, psysid = 0; psys; psys = psys->next, psysid++) {
1148                 /* particles create one more level for persistent psys index */
1149                 DupliContext pctx;
1150                 copy_dupli_context(&pctx, ctx, ctx->object, NULL, psysid);
1151                 make_duplis_particle_system(&pctx, psys);
1152         }
1153 }
1154
1155 static const DupliGenerator gen_dupli_particles = {
1156     OB_DUPLIPARTS,                  /* type */
1157     make_duplis_particles           /* make_duplis */
1158 };
1159
1160 /* ------------- */
1161
1162 /* select dupli generator from given context */
1163 static const DupliGenerator *get_dupli_generator(const DupliContext *ctx)
1164 {
1165         int transflag = ctx->object->transflag;
1166         int restrictflag = ctx->object->restrictflag;
1167
1168         if ((transflag & OB_DUPLI) == 0)
1169                 return NULL;
1170
1171         /* Should the dupli's be generated for this object? - Respect restrict flags */
1172         if (DEG_get_mode(ctx->depsgraph) == DAG_EVAL_RENDER ? (restrictflag & OB_RESTRICT_RENDER) : (restrictflag & OB_RESTRICT_VIEW))
1173                 return NULL;
1174
1175         if (transflag & OB_DUPLIPARTS) {
1176                 return &gen_dupli_particles;
1177         }
1178         else if (transflag & OB_DUPLIVERTS) {
1179                 if (ctx->object->type == OB_MESH) {
1180                         return &gen_dupli_verts;
1181                 }
1182                 else if (ctx->object->type == OB_FONT) {
1183                         return &gen_dupli_verts_font;
1184                 }
1185         }
1186         else if (transflag & OB_DUPLIFACES) {
1187                 if (ctx->object->type == OB_MESH)
1188                         return &gen_dupli_faces;
1189         }
1190         else if (transflag & OB_DUPLIFRAMES) {
1191                 return &gen_dupli_frames;
1192         }
1193         else if (transflag & OB_DUPLICOLLECTION) {
1194                 return &gen_dupli_collection;
1195         }
1196
1197         return NULL;
1198 }
1199
1200
1201 /* ---- ListBase dupli container implementation ---- */
1202
1203 /* Returns a list of DupliObject */
1204 ListBase *object_duplilist(Depsgraph *depsgraph, Scene *sce, Object *ob)
1205 {
1206         ListBase *duplilist = MEM_callocN(sizeof(ListBase), "duplilist");
1207         DupliContext ctx;
1208         init_context(&ctx, depsgraph, sce, ob, NULL);
1209         if (ctx.gen) {
1210                 ctx.duplilist = duplilist;
1211                 ctx.gen->make_duplis(&ctx);
1212         }
1213
1214         return duplilist;
1215 }
1216
1217 void free_object_duplilist(ListBase *lb)
1218 {
1219         BLI_freelistN(lb);
1220         MEM_freeN(lb);
1221 }
1222
1223 int count_duplilist(Object *ob)
1224 {
1225         if (ob->transflag & OB_DUPLI) {
1226                 if (ob->transflag & OB_DUPLIVERTS) {
1227                         if (ob->type == OB_MESH) {
1228                                 if (ob->transflag & OB_DUPLIVERTS) {
1229                                         ParticleSystem *psys = ob->particlesystem.first;
1230                                         int pdup = 0;
1231
1232                                         for (; psys; psys = psys->next)
1233                                                 pdup += psys->totpart;
1234
1235                                         if (pdup == 0) {
1236                                                 Mesh *me = ob->data;
1237                                                 return me->totvert;
1238                                         }
1239                                         else
1240                                                 return pdup;
1241                                 }
1242                         }
1243                 }
1244                 else if (ob->transflag & OB_DUPLIFRAMES) {
1245                         int tot = ob->dupend - ob->dupsta;
1246                         tot /= (ob->dupon + ob->dupoff);
1247                         return tot * ob->dupon;
1248                 }
1249         }
1250         return 1;
1251 }
1252
1253 DupliApplyData *duplilist_apply(Depsgraph *depsgraph, Object *ob, Scene *scene, ListBase *duplilist)
1254 {
1255         DupliApplyData *apply_data = NULL;
1256         int num_objects = BLI_listbase_count(duplilist);
1257         
1258         if (num_objects > 0) {
1259                 DupliObject *dob;
1260                 int i;
1261                 apply_data = MEM_mallocN(sizeof(DupliApplyData), "DupliObject apply data");
1262                 apply_data->num_objects = num_objects;
1263                 apply_data->extra = MEM_mallocN(sizeof(DupliExtraData) * (size_t) num_objects,
1264                                                 "DupliObject apply extra data");
1265
1266                 for (dob = duplilist->first, i = 0; dob; dob = dob->next, ++i) {
1267                         /* make sure derivedmesh is calculated once, before drawing */
1268                         if (scene && !(dob->ob->transflag & OB_DUPLICALCDERIVED) && dob->ob->type == OB_MESH) {
1269                                 mesh_get_derived_final(depsgraph, scene, dob->ob, scene->customdata_mask);
1270                                 dob->ob->transflag |= OB_DUPLICALCDERIVED;
1271                         }
1272                 }
1273
1274                 for (dob = duplilist->first, i = 0; dob; dob = dob->next, ++i) {
1275                         /* copy obmat from duplis */
1276                         copy_m4_m4(apply_data->extra[i].obmat, dob->ob->obmat);
1277                         copy_m4_m4(dob->ob->obmat, dob->mat);
1278                         
1279                         /* copy layers from the main duplicator object */
1280                         apply_data->extra[i].lay = dob->ob->lay;
1281                         dob->ob->lay = ob->lay;
1282                 }
1283         }
1284         return apply_data;
1285 }
1286
1287 void duplilist_restore(ListBase *duplilist, DupliApplyData *apply_data)
1288 {
1289         DupliObject *dob;
1290         int i;
1291         /* Restore object matrices.
1292          * NOTE: this has to happen in reverse order, since nested
1293          * dupli objects can repeatedly override the obmat.
1294          */
1295         for (dob = duplilist->last, i = apply_data->num_objects - 1; dob; dob = dob->prev, --i) {
1296                 copy_m4_m4(dob->ob->obmat, apply_data->extra[i].obmat);
1297                 dob->ob->transflag &= ~OB_DUPLICALCDERIVED;
1298                 
1299                 dob->ob->lay = apply_data->extra[i].lay;
1300         }
1301 }
1302
1303 void duplilist_free_apply_data(DupliApplyData *apply_data)
1304 {
1305         MEM_freeN(apply_data->extra);
1306         MEM_freeN(apply_data);
1307 }