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