From 3ca1cf2b5197c0c2d5afa44ac28e7b63d401b489 Mon Sep 17 00:00:00 2001 From: Hans Goudey Date: Thu, 1 Apr 2021 15:54:48 -0500 Subject: Cleanup: Move object_dupli.cc to C++ This will hopefully allow more refactoring in the future to improve the instancing system, especially how it relates to geometry nodes instances. --- source/blender/blenkernel/CMakeLists.txt | 2 +- source/blender/blenkernel/intern/object_dupli.c | 1613 ---------------------- source/blender/blenkernel/intern/object_dupli.cc | 1602 +++++++++++++++++++++ 3 files changed, 1603 insertions(+), 1614 deletions(-) delete mode 100644 source/blender/blenkernel/intern/object_dupli.c create mode 100644 source/blender/blenkernel/intern/object_dupli.cc (limited to 'source/blender') diff --git a/source/blender/blenkernel/CMakeLists.txt b/source/blender/blenkernel/CMakeLists.txt index 8fbf49b31d4..f1f08b3ad3a 100644 --- a/source/blender/blenkernel/CMakeLists.txt +++ b/source/blender/blenkernel/CMakeLists.txt @@ -212,7 +212,7 @@ set(SRC intern/node_ui_storage.cc intern/object.c intern/object_deform.c - intern/object_dupli.c + intern/object_dupli.cc intern/object_facemap.c intern/object_update.c intern/ocean.c diff --git a/source/blender/blenkernel/intern/object_dupli.c b/source/blender/blenkernel/intern/object_dupli.c deleted file mode 100644 index 632e7519f05..00000000000 --- a/source/blender/blenkernel/intern/object_dupli.c +++ /dev/null @@ -1,1613 +0,0 @@ -/* - * This program is free software; you can redistribute it and/or - * modify it under the terms of the GNU General Public License - * as published by the Free Software Foundation; either version 2 - * of the License, or (at your option) any later version. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with this program; if not, write to the Free Software Foundation, - * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. - * - * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV. - * All rights reserved. - */ - -/** \file - * \ingroup bke - */ - -#include -#include -#include - -#include "MEM_guardedalloc.h" - -#include "BLI_listbase.h" -#include "BLI_string_utf8.h" - -#include "BLI_alloca.h" -#include "BLI_math.h" -#include "BLI_rand.h" - -#include "DNA_anim_types.h" -#include "DNA_collection_types.h" -#include "DNA_mesh_types.h" -#include "DNA_meshdata_types.h" -#include "DNA_pointcloud_types.h" -#include "DNA_scene_types.h" -#include "DNA_vfont_types.h" - -#include "BKE_collection.h" -#include "BKE_duplilist.h" -#include "BKE_editmesh.h" -#include "BKE_editmesh_cache.h" -#include "BKE_font.h" -#include "BKE_geometry_set.h" -#include "BKE_global.h" -#include "BKE_idprop.h" -#include "BKE_lattice.h" -#include "BKE_main.h" -#include "BKE_mesh.h" -#include "BKE_mesh_iterators.h" -#include "BKE_mesh_runtime.h" -#include "BKE_object.h" -#include "BKE_particle.h" -#include "BKE_scene.h" - -#include "DEG_depsgraph.h" -#include "DEG_depsgraph_query.h" - -#include "BLI_hash.h" -#include "BLI_strict_flags.h" - -/* -------------------------------------------------------------------- */ -/** \name Internal Duplicate Context - * \{ */ - -typedef struct DupliContext { - Depsgraph *depsgraph; - /** XXX child objects are selected from this group if set, could be nicer. */ - Collection *collection; - /** Only to check if the object is in edit-mode. */ - Object *obedit; - - Scene *scene; - ViewLayer *view_layer; - Object *object; - float space_mat[4][4]; - - int persistent_id[MAX_DUPLI_RECUR]; - int level; - - const struct DupliGenerator *gen; - - /** Result containers. */ - ListBase *duplilist; /* Legacy doubly-linked list. */ -} DupliContext; - -typedef struct DupliGenerator { - short type; /* Dupli Type, see members of #OB_DUPLI. */ - void (*make_duplis)(const DupliContext *ctx); -} DupliGenerator; - -static const DupliGenerator *get_dupli_generator(const DupliContext *ctx); - -/** - * Create initial context for root object. - */ -static void init_context(DupliContext *r_ctx, - Depsgraph *depsgraph, - Scene *scene, - Object *ob, - const float space_mat[4][4]) -{ - r_ctx->depsgraph = depsgraph; - r_ctx->scene = scene; - r_ctx->view_layer = DEG_get_evaluated_view_layer(depsgraph); - r_ctx->collection = NULL; - - r_ctx->object = ob; - r_ctx->obedit = OBEDIT_FROM_OBACT(ob); - if (space_mat) { - copy_m4_m4(r_ctx->space_mat, space_mat); - } - else { - unit_m4(r_ctx->space_mat); - } - r_ctx->level = 0; - - r_ctx->gen = get_dupli_generator(r_ctx); - - r_ctx->duplilist = NULL; -} - -/** - * Create sub-context for recursive duplis. - */ -static void copy_dupli_context( - DupliContext *r_ctx, const DupliContext *ctx, Object *ob, const float mat[4][4], int index) -{ - *r_ctx = *ctx; - - /* XXX annoying, previously was done by passing an ID* argument, - * this at least is more explicit. */ - if (ctx->gen->type == OB_DUPLICOLLECTION) { - r_ctx->collection = ctx->object->instance_collection; - } - - r_ctx->object = ob; - if (mat) { - mul_m4_m4m4(r_ctx->space_mat, (float(*)[4])ctx->space_mat, mat); - } - r_ctx->persistent_id[r_ctx->level] = index; - ++r_ctx->level; - - r_ctx->gen = get_dupli_generator(r_ctx); -} - -/** - * Generate a dupli instance. - * - * \param mat: is transform of the object relative to current context (including #Object.obmat). - */ -static DupliObject *make_dupli(const DupliContext *ctx, - Object *ob, - const float mat[4][4], - int index) -{ - DupliObject *dob; - int i; - - /* Add a #DupliObject instance to the result container. */ - if (ctx->duplilist) { - dob = MEM_callocN(sizeof(DupliObject), "dupli object"); - BLI_addtail(ctx->duplilist, dob); - } - else { - return NULL; - } - - dob->ob = ob; - mul_m4_m4m4(dob->mat, (float(*)[4])ctx->space_mat, mat); - dob->type = ctx->gen->type; - - /* Set persistent id, which is an array with a persistent index for each level - * (particle number, vertex number, ..). by comparing this we can find the same - * dupli-object between frames, which is needed for motion blur. - * The last level is ordered first in the array. */ - dob->persistent_id[0] = index; - for (i = 1; i < ctx->level + 1; i++) { - dob->persistent_id[i] = ctx->persistent_id[ctx->level - i]; - } - /* Fill rest of values with #INT_MAX which index will never have as value. */ - for (; i < MAX_DUPLI_RECUR; i++) { - dob->persistent_id[i] = INT_MAX; - } - - /* Meta-balls never draw in duplis, they are instead merged into one by the basis - * meta-ball outside of the group. this does mean that if that meta-ball is not in the - * scene, they will not show up at all, limitation that should be solved once. */ - if (ob->type == OB_MBALL) { - dob->no_draw = true; - } - - /* Random number. - * The logic here is designed to match Cycles. */ - dob->random_id = BLI_hash_string(dob->ob->id.name + 2); - - if (dob->persistent_id[0] != INT_MAX) { - for (i = 0; i < MAX_DUPLI_RECUR; i++) { - dob->random_id = BLI_hash_int_2d(dob->random_id, (unsigned int)dob->persistent_id[i]); - } - } - else { - dob->random_id = BLI_hash_int_2d(dob->random_id, 0); - } - - if (ctx->object != ob) { - dob->random_id ^= BLI_hash_int(BLI_hash_string(ctx->object->id.name + 2)); - } - - return dob; -} - -/** - * Recursive dupli-objects. - * - * \param space_mat: is the local dupli-space (excluding dupli #Object.obmat). - */ -static void make_recursive_duplis(const DupliContext *ctx, - Object *ob, - const float space_mat[4][4], - int index) -{ - /* Simple preventing of too deep nested collections with #MAX_DUPLI_RECUR. */ - if (ctx->level < MAX_DUPLI_RECUR) { - DupliContext rctx; - copy_dupli_context(&rctx, ctx, ob, space_mat, index); - if (rctx.gen) { - rctx.gen->make_duplis(&rctx); - } - } -} - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Internal Child Duplicates (Used by Other Functions) - * \{ */ - -typedef void (*MakeChildDuplisFunc)(const DupliContext *ctx, void *userdata, Object *child); - -static bool is_child(const Object *ob, const Object *parent) -{ - const Object *ob_parent = ob->parent; - while (ob_parent) { - if (ob_parent == parent) { - return true; - } - ob_parent = ob_parent->parent; - } - return false; -} - -/** - * Create duplis from every child in scene or collection. - */ -static void make_child_duplis(const DupliContext *ctx, - void *userdata, - MakeChildDuplisFunc make_child_duplis_cb) -{ - Object *parent = ctx->object; - - if (ctx->collection) { - eEvaluationMode mode = DEG_get_mode(ctx->depsgraph); - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN (ctx->collection, ob, mode) { - if ((ob != ctx->obedit) && is_child(ob, parent)) { - DupliContext pctx; - copy_dupli_context(&pctx, ctx, ctx->object, NULL, _base_id); - - /* Meta-balls have a different dupli handling. */ - if (ob->type != OB_MBALL) { - ob->flag |= OB_DONE; /* Doesn't render. */ - } - make_child_duplis_cb(&pctx, userdata, ob); - } - } - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; - } - else { - int baseid = 0; - ViewLayer *view_layer = ctx->view_layer; - for (Base *base = view_layer->object_bases.first; base; base = base->next, baseid++) { - Object *ob = base->object; - if ((ob != ctx->obedit) && is_child(ob, parent)) { - DupliContext pctx; - copy_dupli_context(&pctx, ctx, ctx->object, NULL, baseid); - - /* Meta-balls have a different dupli-handling. */ - if (ob->type != OB_MBALL) { - ob->flag |= OB_DONE; /* Doesn't render. */ - } - - make_child_duplis_cb(&pctx, userdata, ob); - } - } - } -} - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Internal Data Access Utilities - * \{ */ - -static Mesh *mesh_data_from_duplicator_object(Object *ob, - BMEditMesh **r_em, - const float (**r_vert_coords)[3], - const float (**r_vert_normals)[3]) -{ - /* Gather mesh info. */ - BMEditMesh *em = BKE_editmesh_from_object(ob); - Mesh *me_eval; - - *r_em = NULL; - *r_vert_coords = NULL; - if (r_vert_normals != NULL) { - *r_vert_normals = NULL; - } - - /* We do not need any render-specific handling anymore, depsgraph takes care of that. */ - /* NOTE: Do direct access to the evaluated mesh: this function is used - * during meta balls evaluation. But even without those all the objects - * which are needed for correct instancing are already evaluated. */ - if (em != NULL) { - /* Note that this will only show deformation if #eModifierMode_OnCage is enabled. - * We could change this but it matches 2.7x behavior. */ - me_eval = em->mesh_eval_cage; - if ((me_eval == NULL) || (me_eval->runtime.wrapper_type == ME_WRAPPER_TYPE_BMESH)) { - EditMeshData *emd = me_eval ? me_eval->runtime.edit_data : NULL; - - /* Only assign edit-mesh in the case we can't use `me_eval`. */ - *r_em = em; - me_eval = NULL; - - if ((emd != NULL) && (emd->vertexCos != NULL)) { - *r_vert_coords = emd->vertexCos; - if (r_vert_normals != NULL) { - BKE_editmesh_cache_ensure_vert_normals(em, emd); - *r_vert_normals = emd->vertexNos; - } - } - } - } - else { - me_eval = BKE_object_get_evaluated_mesh(ob); - } - return me_eval; -} - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Dupli-Collection Implementation (#OB_DUPLICOLLECTION) - * \{ */ - -static void make_duplis_collection(const DupliContext *ctx) -{ - Object *ob = ctx->object; - Collection *collection; - float collection_mat[4][4]; - - if (ob->instance_collection == NULL) { - return; - } - collection = ob->instance_collection; - - /* Combine collection offset and `obmat`. */ - unit_m4(collection_mat); - sub_v3_v3(collection_mat[3], collection->instance_offset); - mul_m4_m4m4(collection_mat, ob->obmat, collection_mat); - /* Don't access 'ob->obmat' from now on. */ - - eEvaluationMode mode = DEG_get_mode(ctx->depsgraph); - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN (collection, cob, mode) { - if (cob != ob) { - float mat[4][4]; - - /* Collection dupli-offset, should apply after everything else. */ - mul_m4_m4m4(mat, collection_mat, cob->obmat); - - make_dupli(ctx, cob, mat, _base_id); - - /* Recursion. */ - make_recursive_duplis(ctx, cob, collection_mat, _base_id); - } - } - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; -} - -static const DupliGenerator gen_dupli_collection = { - OB_DUPLICOLLECTION, /* type */ - make_duplis_collection /* make_duplis */ -}; - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Dupli-Vertices Implementation (#OB_DUPLIVERTS for Geometry) - * \{ */ - -/** Values shared between different mesh types. */ -typedef struct VertexDupliData_Params { - /** - * It's important we use this context instead of the `ctx` passed into #make_child_duplis - * since these won't match in the case of recursion. - */ - const DupliContext *ctx; - - bool use_rotation; -} VertexDupliData_Params; - -typedef struct VertexDupliData_Mesh { - VertexDupliData_Params params; - - int totvert; - const MVert *mvert; - - const float (*orco)[3]; -} VertexDupliData_Mesh; - -typedef struct VertexDupliData_EditMesh { - VertexDupliData_Params params; - - BMEditMesh *em; - - /* Can be NULL. */ - const float (*vert_coords)[3]; - const float (*vert_normals)[3]; - - /** - * \note The edit-mesh may assign #DupliObject.orco in cases when a regular mesh wouldn't. - * For edit-meshes we only check for deformation, for regular meshes we check if #CD_ORCO exists. - * - * At the moment this isn't a meaningful difference since requesting #CD_ORCO causes the - * edit-mesh to be converted into a mesh. - */ - bool has_orco; -} VertexDupliData_EditMesh; - -/** - * \param no: The direction, - * currently this is copied from a `short[3]` normal without division. - * Can be null when \a use_rotation is false. - */ -static void get_duplivert_transform(const float co[3], - const float no[3], - const bool use_rotation, - const short axis, - const short upflag, - float r_mat[4][4]) -{ - float quat[4]; - const float size[3] = {1.0f, 1.0f, 1.0f}; - - if (use_rotation) { - /* Construct rotation matrix from normals. */ - float no_flip[3]; - negate_v3_v3(no_flip, no); - vec_to_quat(quat, no_flip, axis, upflag); - } - else { - unit_qt(quat); - } - - loc_quat_size_to_mat4(r_mat, co, quat, size); -} - -static DupliObject *vertex_dupli(const DupliContext *ctx, - Object *inst_ob, - const float child_imat[4][4], - int index, - const float co[3], - const float no[3], - const bool use_rotation) -{ - /* `obmat` is transform to vertex. */ - float obmat[4][4]; - get_duplivert_transform(co, no, use_rotation, inst_ob->trackflag, inst_ob->upflag, obmat); - - float space_mat[4][4]; - - /* Make offset relative to inst_ob using relative child transform. */ - mul_mat3_m4_v3(child_imat, obmat[3]); - /* Apply `obmat` _after_ the local vertex transform. */ - mul_m4_m4m4(obmat, inst_ob->obmat, obmat); - - /* Space matrix is constructed by removing `obmat` transform, - * this yields the world-space transform for recursive duplis. */ - mul_m4_m4m4(space_mat, obmat, inst_ob->imat); - - DupliObject *dob = make_dupli(ctx, inst_ob, obmat, index); - - /* Recursion. */ - make_recursive_duplis(ctx, inst_ob, space_mat, index); - - return dob; -} - -static void make_child_duplis_verts_from_mesh(const DupliContext *ctx, - void *userdata, - Object *inst_ob) -{ - VertexDupliData_Mesh *vdd = userdata; - const bool use_rotation = vdd->params.use_rotation; - - const MVert *mvert = vdd->mvert; - const int totvert = vdd->totvert; - - invert_m4_m4(inst_ob->imat, inst_ob->obmat); - /* Relative transform from parent to child space. */ - float child_imat[4][4]; - mul_m4_m4m4(child_imat, inst_ob->imat, ctx->object->obmat); - - const MVert *mv = mvert; - for (int i = 0; i < totvert; i++, mv++) { - const float *co = mv->co; - const float no[3] = {UNPACK3(mv->no)}; - DupliObject *dob = vertex_dupli(vdd->params.ctx, inst_ob, child_imat, i, co, no, use_rotation); - if (vdd->orco) { - copy_v3_v3(dob->orco, vdd->orco[i]); - } - } -} - -static void make_child_duplis_verts_from_editmesh(const DupliContext *ctx, - void *userdata, - Object *inst_ob) -{ - VertexDupliData_EditMesh *vdd = userdata; - BMEditMesh *em = vdd->em; - const bool use_rotation = vdd->params.use_rotation; - - invert_m4_m4(inst_ob->imat, inst_ob->obmat); - /* Relative transform from parent to child space. */ - float child_imat[4][4]; - mul_m4_m4m4(child_imat, inst_ob->imat, ctx->object->obmat); - - BMVert *v; - BMIter iter; - int i; - - const float(*vert_coords)[3] = vdd->vert_coords; - const float(*vert_normals)[3] = vdd->vert_normals; - - BM_ITER_MESH_INDEX (v, &iter, em->bm, BM_VERTS_OF_MESH, i) { - const float *co, *no; - if (vert_coords != NULL) { - co = vert_coords[i]; - no = vert_normals ? vert_normals[i] : NULL; - } - else { - co = v->co; - no = v->no; - } - - DupliObject *dob = vertex_dupli(vdd->params.ctx, inst_ob, child_imat, i, co, no, use_rotation); - if (vdd->has_orco) { - copy_v3_v3(dob->orco, v->co); - } - } -} - -static void make_duplis_verts(const DupliContext *ctx) -{ - Object *parent = ctx->object; - const bool use_rotation = parent->transflag & OB_DUPLIROT; - - /* Gather mesh info. */ - BMEditMesh *em = NULL; - const float(*vert_coords)[3] = NULL; - const float(*vert_normals)[3] = NULL; - Mesh *me_eval = mesh_data_from_duplicator_object( - parent, &em, &vert_coords, use_rotation ? &vert_normals : NULL); - if (em == NULL && me_eval == NULL) { - return; - } - - VertexDupliData_Params vdd_params = { - .ctx = ctx, - .use_rotation = use_rotation, - }; - - if (em != NULL) { - VertexDupliData_EditMesh vdd = { - .params = vdd_params, - .em = em, - .vert_coords = vert_coords, - .vert_normals = vert_normals, - .has_orco = (vert_coords != NULL), - }; - make_child_duplis(ctx, &vdd, make_child_duplis_verts_from_editmesh); - } - else { - VertexDupliData_Mesh vdd = { - .params = vdd_params, - .totvert = me_eval->totvert, - .mvert = me_eval->mvert, - .orco = CustomData_get_layer(&me_eval->vdata, CD_ORCO), - }; - make_child_duplis(ctx, &vdd, make_child_duplis_verts_from_mesh); - } -} - -static const DupliGenerator gen_dupli_verts = { - OB_DUPLIVERTS, /* type */ - make_duplis_verts /* make_duplis */ -}; - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Dupli-Vertices Implementation (#OB_DUPLIVERTS for 3D Text) - * \{ */ - -static Object *find_family_object( - Main *bmain, const char *family, size_t family_len, unsigned int ch, GHash *family_gh) -{ - Object **ob_pt; - Object *ob; - void *ch_key = POINTER_FROM_UINT(ch); - - if ((ob_pt = (Object **)BLI_ghash_lookup_p(family_gh, ch_key))) { - ob = *ob_pt; - } - else { - char ch_utf8[7]; - size_t ch_utf8_len; - - ch_utf8_len = BLI_str_utf8_from_unicode(ch, ch_utf8); - ch_utf8[ch_utf8_len] = '\0'; - ch_utf8_len += 1; /* Compare with null terminator. */ - - for (ob = bmain->objects.first; ob; ob = ob->id.next) { - if (STREQLEN(ob->id.name + 2 + family_len, ch_utf8, ch_utf8_len)) { - if (STREQLEN(ob->id.name + 2, family, family_len)) { - break; - } - } - } - - /* Inserted value can be NULL, just to save searches in future. */ - BLI_ghash_insert(family_gh, ch_key, ob); - } - - return ob; -} - -static void make_duplis_font(const DupliContext *ctx) -{ - Object *par = ctx->object; - GHash *family_gh; - Object *ob; - Curve *cu; - struct CharTrans *ct, *chartransdata = NULL; - float vec[3], obmat[4][4], pmat[4][4], fsize, xof, yof; - int text_len, a; - size_t family_len; - const char32_t *text = NULL; - bool text_free = false; - - /* Font dupli-verts not supported inside collections. */ - if (ctx->collection) { - return; - } - - copy_m4_m4(pmat, par->obmat); - - /* In `par` the family name is stored, use this to find the other objects. */ - - BKE_vfont_to_curve_ex( - par, par->data, FO_DUPLI, NULL, &text, &text_len, &text_free, &chartransdata); - - if (text == NULL || chartransdata == NULL) { - return; - } - - cu = par->data; - fsize = cu->fsize; - xof = cu->xof; - yof = cu->yof; - - ct = chartransdata; - - /* Cache result. */ - family_len = strlen(cu->family); - family_gh = BLI_ghash_int_new_ex(__func__, 256); - - /* Safety check even if it might fail badly when called for original object. */ - const bool is_eval_curve = DEG_is_evaluated_id(&cu->id); - - /* Advance matching BLI_str_utf8_as_utf32. */ - for (a = 0; a < text_len; a++, ct++) { - - /* XXX That G.main is *really* ugly, but not sure what to do here. - * Definitively don't think it would be safe to put back `Main *bmain` pointer - * in #DupliContext as done in 2.7x? */ - ob = find_family_object(G.main, cu->family, family_len, (unsigned int)text[a], family_gh); - - if (is_eval_curve) { - /* Workaround for the above hack. */ - ob = DEG_get_evaluated_object(ctx->depsgraph, ob); - } - - if (ob) { - vec[0] = fsize * (ct->xof - xof); - vec[1] = fsize * (ct->yof - yof); - vec[2] = 0.0; - - mul_m4_v3(pmat, vec); - - copy_m4_m4(obmat, par->obmat); - - if (UNLIKELY(ct->rot != 0.0f)) { - float rmat[4][4]; - - zero_v3(obmat[3]); - axis_angle_to_mat4_single(rmat, 'Z', -ct->rot); - mul_m4_m4m4(obmat, obmat, rmat); - } - - copy_v3_v3(obmat[3], vec); - - make_dupli(ctx, ob, obmat, a); - } - } - - if (text_free) { - MEM_freeN((void *)text); - } - - BLI_ghash_free(family_gh, NULL, NULL); - - MEM_freeN(chartransdata); -} - -static const DupliGenerator gen_dupli_verts_font = { - OB_DUPLIVERTS, /* type */ - make_duplis_font /* make_duplis */ -}; - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Dupli-Vertices Implementation (#OB_DUPLIVERTS for #PointCloud) - * \{ */ - -static void make_child_duplis_pointcloud(const DupliContext *ctx, - void *UNUSED(userdata), - Object *child) -{ - const Object *parent = ctx->object; - const PointCloud *pointcloud = parent->data; - const float(*co)[3] = pointcloud->co; - const float *radius = pointcloud->radius; - const float(*rotation)[4] = NULL; /* TODO: add optional rotation attribute. */ - const float(*orco)[3] = NULL; /* TODO: add optional texture coordinate attribute. */ - - /* Relative transform from parent to child space. */ - float child_imat[4][4]; - mul_m4_m4m4(child_imat, child->imat, parent->obmat); - - for (int i = 0; i < pointcloud->totpoint; i++) { - /* Transform matrix from point position, radius and rotation. */ - float quat[4] = {1.0f, 0.0f, 0.0f, 0.0f}; - float size[3] = {1.0f, 1.0f, 1.0f}; - if (radius) { - copy_v3_fl(size, radius[i]); - } - if (rotation) { - copy_v4_v4(quat, rotation[i]); - } - - float space_mat[4][4]; - loc_quat_size_to_mat4(space_mat, co[i], quat, size); - - /* Make offset relative to child object using relative child transform, - * and apply object matrix after local vertex transform. */ - mul_mat3_m4_v3(child_imat, space_mat[3]); - - /* Create dupli object. */ - float obmat[4][4]; - mul_m4_m4m4(obmat, child->obmat, space_mat); - DupliObject *dob = make_dupli(ctx, child, obmat, i); - if (orco) { - copy_v3_v3(dob->orco, orco[i]); - } - - /* Recursion. */ - make_recursive_duplis(ctx, child, space_mat, i); - } -} - -static void make_duplis_pointcloud(const DupliContext *ctx) -{ - make_child_duplis(ctx, NULL, make_child_duplis_pointcloud); -} - -static const DupliGenerator gen_dupli_verts_pointcloud = { - OB_DUPLIVERTS, /* type */ - make_duplis_pointcloud /* make_duplis */ -}; - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Instances Geometry Component Implementation - * \{ */ - -static void make_duplis_instances_component(const DupliContext *ctx) -{ - float(*instance_offset_matrices)[4][4]; - InstancedData *instanced_data; - const int *almost_unique_ids; - const int amount = BKE_geometry_set_instances(ctx->object->runtime.geometry_set_eval, - &instance_offset_matrices, - &almost_unique_ids, - &instanced_data); - - for (int i = 0; i < amount; i++) { - InstancedData *data = &instanced_data[i]; - - const int id = almost_unique_ids[i]; - - if (data->type == INSTANCE_DATA_TYPE_OBJECT) { - Object *object = data->data.object; - if (object != NULL) { - float matrix[4][4]; - mul_m4_m4m4(matrix, ctx->object->obmat, instance_offset_matrices[i]); - make_dupli(ctx, object, matrix, id); - - float space_matrix[4][4]; - mul_m4_m4m4(space_matrix, instance_offset_matrices[i], object->imat); - mul_m4_m4_pre(space_matrix, ctx->object->obmat); - make_recursive_duplis(ctx, object, space_matrix, id); - } - } - else if (data->type == INSTANCE_DATA_TYPE_COLLECTION) { - Collection *collection = data->data.collection; - if (collection != NULL) { - float collection_matrix[4][4]; - unit_m4(collection_matrix); - sub_v3_v3(collection_matrix[3], collection->instance_offset); - mul_m4_m4_pre(collection_matrix, instance_offset_matrices[i]); - mul_m4_m4_pre(collection_matrix, ctx->object->obmat); - - eEvaluationMode mode = DEG_get_mode(ctx->depsgraph); - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN (collection, object, mode) { - if (object == ctx->object) { - continue; - } - - float instance_matrix[4][4]; - mul_m4_m4m4(instance_matrix, collection_matrix, object->obmat); - - make_dupli(ctx, object, instance_matrix, id); - make_recursive_duplis(ctx, object, collection_matrix, id); - } - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; - } - } - } -} - -static const DupliGenerator gen_dupli_instances_component = { - 0, - make_duplis_instances_component, -}; - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Dupli-Faces Implementation (#OB_DUPLIFACES) - * \{ */ - -/** Values shared between different mesh types. */ -typedef struct FaceDupliData_Params { - /** - * It's important we use this context instead of the `ctx` passed into #make_child_duplis - * since these won't match in the case of recursion. - */ - const DupliContext *ctx; - - bool use_scale; -} FaceDupliData_Params; - -typedef struct FaceDupliData_Mesh { - FaceDupliData_Params params; - - int totface; - const MPoly *mpoly; - const MLoop *mloop; - const MVert *mvert; - const float (*orco)[3]; - const MLoopUV *mloopuv; -} FaceDupliData_Mesh; - -typedef struct FaceDupliData_EditMesh { - FaceDupliData_Params params; - - BMEditMesh *em; - - bool has_orco, has_uvs; - int cd_loop_uv_offset; - /* Can be NULL. */ - const float (*vert_coords)[3]; -} FaceDupliData_EditMesh; - -static void get_dupliface_transform_from_coords(const float coords[][3], - const int coords_len, - const bool use_scale, - const float scale_fac, - float r_mat[4][4]) -{ - float loc[3], quat[4], scale, size[3]; - - /* Location. */ - { - const float w = 1.0f / (float)coords_len; - zero_v3(loc); - for (int i = 0; i < coords_len; i++) { - madd_v3_v3fl(loc, coords[i], w); - } - } - /* Rotation. */ - { - float f_no[3]; - cross_poly_v3(f_no, coords, (uint)coords_len); - normalize_v3(f_no); - tri_to_quat_ex(quat, coords[0], coords[1], coords[2], f_no); - } - /* Scale. */ - if (use_scale) { - const float area = area_poly_v3(coords, (uint)coords_len); - scale = sqrtf(area) * scale_fac; - } - else { - scale = 1.0f; - } - size[0] = size[1] = size[2] = scale; - - loc_quat_size_to_mat4(r_mat, loc, quat, size); -} - -static DupliObject *face_dupli(const DupliContext *ctx, - Object *inst_ob, - const float child_imat[4][4], - const int index, - const bool use_scale, - const float scale_fac, - const float (*coords)[3], - const int coords_len) -{ - float obmat[4][4]; - float space_mat[4][4]; - - /* `obmat` is transform to face. */ - get_dupliface_transform_from_coords(coords, coords_len, use_scale, scale_fac, obmat); - - /* Make offset relative to inst_ob using relative child transform. */ - mul_mat3_m4_v3(child_imat, obmat[3]); - - /* XXX ugly hack to ensure same behavior as in master. - * This should not be needed, #Object.parentinv is not consistent outside of parenting. */ - { - float imat[3][3]; - copy_m3_m4(imat, inst_ob->parentinv); - mul_m4_m3m4(obmat, imat, obmat); - } - - /* Apply `obmat` _after_ the local face transform. */ - mul_m4_m4m4(obmat, inst_ob->obmat, obmat); - - /* Space matrix is constructed by removing `obmat` transform, - * this yields the world-space transform for recursive duplis. */ - mul_m4_m4m4(space_mat, obmat, inst_ob->imat); - - DupliObject *dob = make_dupli(ctx, inst_ob, obmat, index); - - /* Recursion. */ - make_recursive_duplis(ctx, inst_ob, space_mat, index); - - return dob; -} - -/** Wrap #face_dupli, needed since we can't #alloca in a loop. */ -static DupliObject *face_dupli_from_mesh(const DupliContext *ctx, - Object *inst_ob, - const float child_imat[4][4], - const int index, - const bool use_scale, - const float scale_fac, - - /* Mesh variables. */ - const MPoly *mpoly, - const MLoop *mloopstart, - const MVert *mvert) -{ - const int coords_len = mpoly->totloop; - float(*coords)[3] = BLI_array_alloca(coords, (size_t)coords_len); - - const MLoop *ml = mloopstart; - for (int i = 0; i < coords_len; i++, ml++) { - copy_v3_v3(coords[i], mvert[ml->v].co); - } - - return face_dupli(ctx, inst_ob, child_imat, index, use_scale, scale_fac, coords, coords_len); -} - -/** Wrap #face_dupli, needed since we can't #alloca in a loop. */ -static DupliObject *face_dupli_from_editmesh(const DupliContext *ctx, - Object *inst_ob, - const float child_imat[4][4], - const int index, - const bool use_scale, - const float scale_fac, - - /* Mesh variables. */ - BMFace *f, - const float (*vert_coords)[3]) -{ - const int coords_len = f->len; - float(*coords)[3] = BLI_array_alloca(coords, (size_t)coords_len); - - BMLoop *l_first, *l_iter; - int i = 0; - l_iter = l_first = BM_FACE_FIRST_LOOP(f); - if (vert_coords != NULL) { - do { - copy_v3_v3(coords[i++], vert_coords[BM_elem_index_get(l_iter->v)]); - } while ((l_iter = l_iter->next) != l_first); - } - else { - do { - copy_v3_v3(coords[i++], l_iter->v->co); - } while ((l_iter = l_iter->next) != l_first); - } - - return face_dupli(ctx, inst_ob, child_imat, index, use_scale, scale_fac, coords, coords_len); -} - -static void make_child_duplis_faces_from_mesh(const DupliContext *ctx, - void *userdata, - Object *inst_ob) -{ - FaceDupliData_Mesh *fdd = userdata; - const MPoly *mpoly = fdd->mpoly, *mp; - const MLoop *mloop = fdd->mloop; - const MVert *mvert = fdd->mvert; - const float(*orco)[3] = fdd->orco; - const MLoopUV *mloopuv = fdd->mloopuv; - const int totface = fdd->totface; - const bool use_scale = fdd->params.use_scale; - int a; - - float child_imat[4][4]; - - invert_m4_m4(inst_ob->imat, inst_ob->obmat); - /* Relative transform from parent to child space. */ - mul_m4_m4m4(child_imat, inst_ob->imat, ctx->object->obmat); - const float scale_fac = ctx->object->instance_faces_scale; - - for (a = 0, mp = mpoly; a < totface; a++, mp++) { - const MLoop *loopstart = mloop + mp->loopstart; - DupliObject *dob = face_dupli_from_mesh( - fdd->params.ctx, inst_ob, child_imat, a, use_scale, scale_fac, mp, loopstart, mvert); - - const float w = 1.0f / (float)mp->totloop; - if (orco) { - for (int j = 0; j < mp->totloop; j++) { - madd_v3_v3fl(dob->orco, orco[loopstart[j].v], w); - } - } - if (mloopuv) { - for (int j = 0; j < mp->totloop; j++) { - madd_v2_v2fl(dob->uv, mloopuv[mp->loopstart + j].uv, w); - } - } - } -} - -static void make_child_duplis_faces_from_editmesh(const DupliContext *ctx, - void *userdata, - Object *inst_ob) -{ - FaceDupliData_EditMesh *fdd = userdata; - BMEditMesh *em = fdd->em; - float child_imat[4][4]; - int a; - BMFace *f; - BMIter iter; - const bool use_scale = fdd->params.use_scale; - - const float(*vert_coords)[3] = fdd->vert_coords; - - BLI_assert((vert_coords == NULL) || (em->bm->elem_index_dirty & BM_VERT) == 0); - - invert_m4_m4(inst_ob->imat, inst_ob->obmat); - /* Relative transform from parent to child space. */ - mul_m4_m4m4(child_imat, inst_ob->imat, ctx->object->obmat); - const float scale_fac = ctx->object->instance_faces_scale; - - BM_ITER_MESH_INDEX (f, &iter, em->bm, BM_FACES_OF_MESH, a) { - DupliObject *dob = face_dupli_from_editmesh( - fdd->params.ctx, inst_ob, child_imat, a, use_scale, scale_fac, f, vert_coords); - - if (fdd->has_orco) { - const float w = 1.0f / (float)f->len; - BMLoop *l_first, *l_iter; - l_iter = l_first = BM_FACE_FIRST_LOOP(f); - do { - madd_v3_v3fl(dob->orco, l_iter->v->co, w); - } while ((l_iter = l_iter->next) != l_first); - } - if (fdd->has_uvs) { - BM_face_uv_calc_center_median(f, fdd->cd_loop_uv_offset, dob->uv); - } - } -} - -static void make_duplis_faces(const DupliContext *ctx) -{ - Object *parent = ctx->object; - - /* Gather mesh info. */ - BMEditMesh *em = NULL; - const float(*vert_coords)[3] = NULL; - Mesh *me_eval = mesh_data_from_duplicator_object(parent, &em, &vert_coords, NULL); - if (em == NULL && me_eval == NULL) { - return; - } - - FaceDupliData_Params fdd_params = { - .ctx = ctx, - .use_scale = parent->transflag & OB_DUPLIFACES_SCALE, - }; - - if (em != NULL) { - const int uv_idx = CustomData_get_render_layer(&em->bm->ldata, CD_MLOOPUV); - FaceDupliData_EditMesh fdd = { - .params = fdd_params, - .em = em, - .vert_coords = vert_coords, - .has_orco = (vert_coords != NULL), - .has_uvs = (uv_idx != -1), - .cd_loop_uv_offset = (uv_idx != -1) ? - CustomData_get_n_offset(&em->bm->ldata, CD_MLOOPUV, uv_idx) : - -1, - }; - make_child_duplis(ctx, &fdd, make_child_duplis_faces_from_editmesh); - } - else { - const int uv_idx = CustomData_get_render_layer(&me_eval->ldata, CD_MLOOPUV); - FaceDupliData_Mesh fdd = { - .params = fdd_params, - .totface = me_eval->totpoly, - .mpoly = me_eval->mpoly, - .mloop = me_eval->mloop, - .mvert = me_eval->mvert, - .mloopuv = (uv_idx != -1) ? CustomData_get_layer_n(&me_eval->ldata, CD_MLOOPUV, uv_idx) : - NULL, - .orco = CustomData_get_layer(&me_eval->vdata, CD_ORCO), - }; - make_child_duplis(ctx, &fdd, make_child_duplis_faces_from_mesh); - } -} - -static const DupliGenerator gen_dupli_faces = { - OB_DUPLIFACES, /* type */ - make_duplis_faces /* make_duplis */ -}; - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Dupli-Particles Implementation (#OB_DUPLIPARTS) - * \{ */ - -static void make_duplis_particle_system(const DupliContext *ctx, ParticleSystem *psys) -{ - Scene *scene = ctx->scene; - Object *par = ctx->object; - eEvaluationMode mode = DEG_get_mode(ctx->depsgraph); - bool for_render = mode == DAG_EVAL_RENDER; - - Object *ob = NULL, **oblist = NULL; - DupliObject *dob; - ParticleDupliWeight *dw; - ParticleSettings *part; - ParticleData *pa; - ChildParticle *cpa = NULL; - ParticleKey state; - ParticleCacheKey *cache; - float ctime, scale = 1.0f; - float tmat[4][4], mat[4][4], pamat[4][4], size = 0.0; - int a, b, hair = 0; - int totpart, totchild; - - int no_draw_flag = PARS_UNEXIST; - - if (psys == NULL) { - return; - } - - part = psys->part; - - if (part == NULL) { - return; - } - - if (!psys_check_enabled(par, psys, for_render)) { - return; - } - - if (!for_render) { - no_draw_flag |= PARS_NO_DISP; - } - - /* NOTE: in old animation system, used parent object's time-offset. */ - ctime = DEG_get_ctime(ctx->depsgraph); - - totpart = psys->totpart; - totchild = psys->totchild; - - if ((for_render || part->draw_as == PART_DRAW_REND) && - ELEM(part->ren_as, PART_DRAW_OB, PART_DRAW_GR)) { - ParticleSimulationData sim = {NULL}; - sim.depsgraph = ctx->depsgraph; - sim.scene = scene; - sim.ob = par; - sim.psys = psys; - sim.psmd = psys_get_modifier(par, psys); - /* Make sure emitter `imat` is in global coordinates instead of render view coordinates. */ - invert_m4_m4(par->imat, par->obmat); - - /* First check for loops (particle system object used as dupli-object). */ - if (part->ren_as == PART_DRAW_OB) { - if (ELEM(part->instance_object, NULL, par)) { - return; - } - } - else { /* #PART_DRAW_GR. */ - if (part->instance_collection == NULL) { - return; - } - - const ListBase dup_collection_objects = BKE_collection_object_cache_get( - part->instance_collection); - if (BLI_listbase_is_empty(&dup_collection_objects)) { - return; - } - - if (BLI_findptr(&dup_collection_objects, par, offsetof(Base, object))) { - return; - } - } - - /* If we have a hair particle system, use the path cache. */ - if (part->type == PART_HAIR) { - if (psys->flag & PSYS_HAIR_DONE) { - hair = (totchild == 0 || psys->childcache) && psys->pathcache; - } - if (!hair) { - return; - } - - /* We use cache, update `totchild` according to cached data. */ - totchild = psys->totchildcache; - totpart = psys->totcached; - } - - RNG *rng = BLI_rng_new_srandom(31415926u + (unsigned int)psys->seed); - - psys->lattice_deform_data = psys_create_lattice_deform_data(&sim); - - /* Gather list of objects or single object. */ - int totcollection = 0; - - const bool use_whole_collection = part->draw & PART_DRAW_WHOLE_GR; - const bool use_collection_count = part->draw & PART_DRAW_COUNT_GR && !use_whole_collection; - if (part->ren_as == PART_DRAW_GR) { - if (use_collection_count) { - psys_find_group_weights(part); - - for (dw = part->instance_weights.first; dw; dw = dw->next) { - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN ( - part->instance_collection, object, mode) { - if (dw->ob == object) { - totcollection += dw->count; - break; - } - } - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; - } - } - else { - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN ( - part->instance_collection, object, mode) { - (void)object; - totcollection++; - } - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; - } - - oblist = MEM_callocN((size_t)totcollection * sizeof(Object *), "dupcollection object list"); - - if (use_collection_count) { - a = 0; - for (dw = part->instance_weights.first; dw; dw = dw->next) { - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN ( - part->instance_collection, object, mode) { - if (dw->ob == object) { - for (b = 0; b < dw->count; b++, a++) { - oblist[a] = dw->ob; - } - break; - } - } - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; - } - } - else { - a = 0; - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN ( - part->instance_collection, object, mode) { - oblist[a] = object; - a++; - } - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; - } - } - else { - ob = part->instance_object; - } - - if (totchild == 0 || part->draw & PART_DRAW_PARENT) { - a = 0; - } - else { - a = totpart; - } - - for (pa = psys->particles; a < totpart + totchild; a++, pa++) { - if (a < totpart) { - /* Handle parent particle. */ - if (pa->flag & no_draw_flag) { - continue; - } - -#if 0 /* UNUSED */ - pa_num = pa->num; -#endif - size = pa->size; - } - else { - /* Handle child particle. */ - cpa = &psys->child[a - totpart]; - -#if 0 /* UNUSED */ - pa_num = a; -#endif - size = psys_get_child_size(psys, cpa, ctime, NULL); - } - - /* Some hair paths might be non-existent so they can't be used for duplication. */ - if (hair && psys->pathcache && - ((a < totpart && psys->pathcache[a]->segments < 0) || - (a >= totpart && psys->childcache[a - totpart]->segments < 0))) { - continue; - } - - if (part->ren_as == PART_DRAW_GR) { - /* Prevent divide by zero below T28336. */ - if (totcollection == 0) { - continue; - } - - /* For collections, pick the object based on settings. */ - if (part->draw & PART_DRAW_RAND_GR && !use_whole_collection) { - b = BLI_rng_get_int(rng) % totcollection; - } - else { - b = a % totcollection; - } - - ob = oblist[b]; - } - - if (hair) { - /* Hair we handle separate and compute transform based on hair keys. */ - if (a < totpart) { - cache = psys->pathcache[a]; - psys_get_dupli_path_transform(&sim, pa, NULL, cache, pamat, &scale); - } - else { - cache = psys->childcache[a - totpart]; - psys_get_dupli_path_transform(&sim, NULL, cpa, cache, pamat, &scale); - } - - copy_v3_v3(pamat[3], cache->co); - pamat[3][3] = 1.0f; - } - else { - /* First key. */ - state.time = ctime; - if (psys_get_particle_state(&sim, a, &state, 0) == 0) { - continue; - } - - float tquat[4]; - normalize_qt_qt(tquat, state.rot); - quat_to_mat4(pamat, tquat); - copy_v3_v3(pamat[3], state.co); - pamat[3][3] = 1.0f; - } - - if (part->ren_as == PART_DRAW_GR && psys->part->draw & PART_DRAW_WHOLE_GR) { - b = 0; - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN ( - part->instance_collection, object, mode) { - copy_m4_m4(tmat, oblist[b]->obmat); - - /* Apply collection instance offset. */ - sub_v3_v3(tmat[3], part->instance_collection->instance_offset); - - /* Apply particle scale. */ - mul_mat3_m4_fl(tmat, size * scale); - mul_v3_fl(tmat[3], size * scale); - - /* Individual particle transform. */ - mul_m4_m4m4(mat, pamat, tmat); - - dob = make_dupli(ctx, object, mat, a); - dob->particle_system = psys; - - psys_get_dupli_texture(psys, part, sim.psmd, pa, cpa, dob->uv, dob->orco); - - b++; - } - FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; - } - else { - float obmat[4][4]; - copy_m4_m4(obmat, ob->obmat); - - float vec[3]; - copy_v3_v3(vec, obmat[3]); - zero_v3(obmat[3]); - - /* Particle rotation uses x-axis as the aligned axis, - * so pre-rotate the object accordingly. */ - if ((part->draw & PART_DRAW_ROTATE_OB) == 0) { - float xvec[3], q[4], size_mat[4][4], original_size[3]; - - mat4_to_size(original_size, obmat); - size_to_mat4(size_mat, original_size); - - xvec[0] = -1.0f; - xvec[1] = xvec[2] = 0; - vec_to_quat(q, xvec, ob->trackflag, ob->upflag); - quat_to_mat4(obmat, q); - obmat[3][3] = 1.0f; - - /* Add scaling if requested. */ - if ((part->draw & PART_DRAW_NO_SCALE_OB) == 0) { - mul_m4_m4m4(obmat, obmat, size_mat); - } - } - else if (part->draw & PART_DRAW_NO_SCALE_OB) { - /* Remove scaling. */ - float size_mat[4][4], original_size[3]; - - mat4_to_size(original_size, obmat); - size_to_mat4(size_mat, original_size); - invert_m4(size_mat); - - mul_m4_m4m4(obmat, obmat, size_mat); - } - - mul_m4_m4m4(tmat, pamat, obmat); - mul_mat3_m4_fl(tmat, size * scale); - - copy_m4_m4(mat, tmat); - - if (part->draw & PART_DRAW_GLOBAL_OB) { - add_v3_v3v3(mat[3], mat[3], vec); - } - - dob = make_dupli(ctx, ob, mat, a); - dob->particle_system = psys; - psys_get_dupli_texture(psys, part, sim.psmd, pa, cpa, dob->uv, dob->orco); - } - } - - BLI_rng_free(rng); - } - - /* Clean up. */ - if (oblist) { - MEM_freeN(oblist); - } - - if (psys->lattice_deform_data) { - BKE_lattice_deform_data_destroy(psys->lattice_deform_data); - psys->lattice_deform_data = NULL; - } -} - -static void make_duplis_particles(const DupliContext *ctx) -{ - ParticleSystem *psys; - int psysid; - - /* Particle system take up one level in id, the particles another. */ - for (psys = ctx->object->particlesystem.first, psysid = 0; psys; psys = psys->next, psysid++) { - /* Particles create one more level for persistent `psys` index. */ - DupliContext pctx; - copy_dupli_context(&pctx, ctx, ctx->object, NULL, psysid); - make_duplis_particle_system(&pctx, psys); - } -} - -static const DupliGenerator gen_dupli_particles = { - OB_DUPLIPARTS, /* type */ - make_duplis_particles /* make_duplis */ -}; - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Dupli-Generator Selector For The Given Context - * \{ */ - -static const DupliGenerator *get_dupli_generator(const DupliContext *ctx) -{ - int transflag = ctx->object->transflag; - int restrictflag = ctx->object->restrictflag; - - if ((transflag & OB_DUPLI) == 0 && ctx->object->runtime.geometry_set_eval == NULL) { - return NULL; - } - - /* Should the dupli's be generated for this object? - Respect restrict flags. */ - if (DEG_get_mode(ctx->depsgraph) == DAG_EVAL_RENDER ? (restrictflag & OB_RESTRICT_RENDER) : - (restrictflag & OB_RESTRICT_VIEWPORT)) { - return NULL; - } - - if (ctx->object->runtime.geometry_set_eval != NULL) { - if (BKE_geometry_set_has_instances(ctx->object->runtime.geometry_set_eval)) { - return &gen_dupli_instances_component; - } - } - - if (transflag & OB_DUPLIPARTS) { - return &gen_dupli_particles; - } - if (transflag & OB_DUPLIVERTS) { - if (ctx->object->type == OB_MESH) { - return &gen_dupli_verts; - } - if (ctx->object->type == OB_FONT) { - return &gen_dupli_verts_font; - } - if (ctx->object->type == OB_POINTCLOUD) { - return &gen_dupli_verts_pointcloud; - } - } - else if (transflag & OB_DUPLIFACES) { - if (ctx->object->type == OB_MESH) { - return &gen_dupli_faces; - } - } - else if (transflag & OB_DUPLICOLLECTION) { - return &gen_dupli_collection; - } - - return NULL; -} - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Dupli-Container Implementation - * \{ */ - -/** - * \return a #ListBase of #DupliObject. - */ -ListBase *object_duplilist(Depsgraph *depsgraph, Scene *sce, Object *ob) -{ - ListBase *duplilist = MEM_callocN(sizeof(ListBase), "duplilist"); - DupliContext ctx; - init_context(&ctx, depsgraph, sce, ob, NULL); - if (ctx.gen) { - ctx.duplilist = duplilist; - ctx.gen->make_duplis(&ctx); - } - - return duplilist; -} - -void free_object_duplilist(ListBase *lb) -{ - BLI_freelistN(lb); - MEM_freeN(lb); -} - -/** \} */ diff --git a/source/blender/blenkernel/intern/object_dupli.cc b/source/blender/blenkernel/intern/object_dupli.cc new file mode 100644 index 00000000000..1830870b854 --- /dev/null +++ b/source/blender/blenkernel/intern/object_dupli.cc @@ -0,0 +1,1602 @@ +/* + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software Foundation, + * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. + * + * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV. + * All rights reserved. + */ + +/** \file + * \ingroup bke + */ + +#include +#include +#include + +#include "MEM_guardedalloc.h" + +#include "BLI_listbase.h" +#include "BLI_string_utf8.h" + +#include "BLI_alloca.h" +#include "BLI_math.h" +#include "BLI_rand.h" + +#include "DNA_anim_types.h" +#include "DNA_collection_types.h" +#include "DNA_mesh_types.h" +#include "DNA_meshdata_types.h" +#include "DNA_pointcloud_types.h" +#include "DNA_scene_types.h" +#include "DNA_vfont_types.h" + +#include "BKE_collection.h" +#include "BKE_duplilist.h" +#include "BKE_editmesh.h" +#include "BKE_editmesh_cache.h" +#include "BKE_font.h" +#include "BKE_geometry_set.h" +#include "BKE_global.h" +#include "BKE_idprop.h" +#include "BKE_lattice.h" +#include "BKE_main.h" +#include "BKE_mesh.h" +#include "BKE_mesh_iterators.h" +#include "BKE_mesh_runtime.h" +#include "BKE_object.h" +#include "BKE_particle.h" +#include "BKE_scene.h" + +#include "DEG_depsgraph.h" +#include "DEG_depsgraph_query.h" + +#include "BLI_hash.h" +#include "BLI_strict_flags.h" + +/* -------------------------------------------------------------------- */ +/** \name Internal Duplicate Context + * \{ */ + +struct DupliContext { + Depsgraph *depsgraph; + /** XXX child objects are selected from this group if set, could be nicer. */ + Collection *collection; + /** Only to check if the object is in edit-mode. */ + Object *obedit; + + Scene *scene; + ViewLayer *view_layer; + Object *object; + float space_mat[4][4]; + + int persistent_id[MAX_DUPLI_RECUR]; + int level; + + const struct DupliGenerator *gen; + + /** Result containers. */ + ListBase *duplilist; /* Legacy doubly-linked list. */ +}; + +struct DupliGenerator { + short type; /* Dupli Type, see members of #OB_DUPLI. */ + void (*make_duplis)(const DupliContext *ctx); +}; + +static const DupliGenerator *get_dupli_generator(const DupliContext *ctx); + +/** + * Create initial context for root object. + */ +static void init_context(DupliContext *r_ctx, + Depsgraph *depsgraph, + Scene *scene, + Object *ob, + const float space_mat[4][4]) +{ + r_ctx->depsgraph = depsgraph; + r_ctx->scene = scene; + r_ctx->view_layer = DEG_get_evaluated_view_layer(depsgraph); + r_ctx->collection = nullptr; + + r_ctx->object = ob; + r_ctx->obedit = OBEDIT_FROM_OBACT(ob); + if (space_mat) { + copy_m4_m4(r_ctx->space_mat, space_mat); + } + else { + unit_m4(r_ctx->space_mat); + } + r_ctx->level = 0; + + r_ctx->gen = get_dupli_generator(r_ctx); + + r_ctx->duplilist = nullptr; +} + +/** + * Create sub-context for recursive duplis. + */ +static void copy_dupli_context( + DupliContext *r_ctx, const DupliContext *ctx, Object *ob, const float mat[4][4], int index) +{ + *r_ctx = *ctx; + + /* XXX annoying, previously was done by passing an ID* argument, + * this at least is more explicit. */ + if (ctx->gen->type == OB_DUPLICOLLECTION) { + r_ctx->collection = ctx->object->instance_collection; + } + + r_ctx->object = ob; + if (mat) { + mul_m4_m4m4(r_ctx->space_mat, (float(*)[4])ctx->space_mat, mat); + } + r_ctx->persistent_id[r_ctx->level] = index; + ++r_ctx->level; + + r_ctx->gen = get_dupli_generator(r_ctx); +} + +/** + * Generate a dupli instance. + * + * \param mat: is transform of the object relative to current context (including #Object.obmat). + */ +static DupliObject *make_dupli(const DupliContext *ctx, + Object *ob, + const float mat[4][4], + int index) +{ + DupliObject *dob; + int i; + + /* Add a #DupliObject instance to the result container. */ + if (ctx->duplilist) { + dob = (DupliObject *)MEM_callocN(sizeof(DupliObject), "dupli object"); + BLI_addtail(ctx->duplilist, dob); + } + else { + return nullptr; + } + + dob->ob = ob; + mul_m4_m4m4(dob->mat, (float(*)[4])ctx->space_mat, mat); + dob->type = ctx->gen->type; + + /* Set persistent id, which is an array with a persistent index for each level + * (particle number, vertex number, ..). by comparing this we can find the same + * dupli-object between frames, which is needed for motion blur. + * The last level is ordered first in the array. */ + dob->persistent_id[0] = index; + for (i = 1; i < ctx->level + 1; i++) { + dob->persistent_id[i] = ctx->persistent_id[ctx->level - i]; + } + /* Fill rest of values with #INT_MAX which index will never have as value. */ + for (; i < MAX_DUPLI_RECUR; i++) { + dob->persistent_id[i] = INT_MAX; + } + + /* Meta-balls never draw in duplis, they are instead merged into one by the basis + * meta-ball outside of the group. this does mean that if that meta-ball is not in the + * scene, they will not show up at all, limitation that should be solved once. */ + if (ob->type == OB_MBALL) { + dob->no_draw = true; + } + + /* Random number. + * The logic here is designed to match Cycles. */ + dob->random_id = BLI_hash_string(dob->ob->id.name + 2); + + if (dob->persistent_id[0] != INT_MAX) { + for (i = 0; i < MAX_DUPLI_RECUR; i++) { + dob->random_id = BLI_hash_int_2d(dob->random_id, (unsigned int)dob->persistent_id[i]); + } + } + else { + dob->random_id = BLI_hash_int_2d(dob->random_id, 0); + } + + if (ctx->object != ob) { + dob->random_id ^= BLI_hash_int(BLI_hash_string(ctx->object->id.name + 2)); + } + + return dob; +} + +/** + * Recursive dupli-objects. + * + * \param space_mat: is the local dupli-space (excluding dupli #Object.obmat). + */ +static void make_recursive_duplis(const DupliContext *ctx, + Object *ob, + const float space_mat[4][4], + int index) +{ + /* Simple preventing of too deep nested collections with #MAX_DUPLI_RECUR. */ + if (ctx->level < MAX_DUPLI_RECUR) { + DupliContext rctx; + copy_dupli_context(&rctx, ctx, ob, space_mat, index); + if (rctx.gen) { + rctx.gen->make_duplis(&rctx); + } + } +} + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Internal Child Duplicates (Used by Other Functions) + * \{ */ + +using MakeChildDuplisFunc = void (*)(const DupliContext *ctx, void *userdata, Object *child); + +static bool is_child(const Object *ob, const Object *parent) +{ + const Object *ob_parent = ob->parent; + while (ob_parent) { + if (ob_parent == parent) { + return true; + } + ob_parent = ob_parent->parent; + } + return false; +} + +/** + * Create duplis from every child in scene or collection. + */ +static void make_child_duplis(const DupliContext *ctx, + void *userdata, + MakeChildDuplisFunc make_child_duplis_cb) +{ + Object *parent = ctx->object; + + if (ctx->collection) { + eEvaluationMode mode = DEG_get_mode(ctx->depsgraph); + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN (ctx->collection, ob, mode) { + if ((ob != ctx->obedit) && is_child(ob, parent)) { + DupliContext pctx; + copy_dupli_context(&pctx, ctx, ctx->object, nullptr, _base_id); + + /* Meta-balls have a different dupli handling. */ + if (ob->type != OB_MBALL) { + ob->flag |= OB_DONE; /* Doesn't render. */ + } + make_child_duplis_cb(&pctx, userdata, ob); + } + } + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; + } + else { + int baseid; + ViewLayer *view_layer = ctx->view_layer; + LISTBASE_FOREACH_INDEX (Base *, base, &view_layer->object_bases, baseid) { + Object *ob = base->object; + if ((ob != ctx->obedit) && is_child(ob, parent)) { + DupliContext pctx; + copy_dupli_context(&pctx, ctx, ctx->object, nullptr, baseid); + + /* Meta-balls have a different dupli-handling. */ + if (ob->type != OB_MBALL) { + ob->flag |= OB_DONE; /* Doesn't render. */ + } + + make_child_duplis_cb(&pctx, userdata, ob); + } + } + } +} + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Internal Data Access Utilities + * \{ */ + +static Mesh *mesh_data_from_duplicator_object(Object *ob, + BMEditMesh **r_em, + const float (**r_vert_coords)[3], + const float (**r_vert_normals)[3]) +{ + /* Gather mesh info. */ + BMEditMesh *em = BKE_editmesh_from_object(ob); + Mesh *me_eval; + + *r_em = nullptr; + *r_vert_coords = nullptr; + if (r_vert_normals != nullptr) { + *r_vert_normals = nullptr; + } + + /* We do not need any render-specific handling anymore, depsgraph takes care of that. */ + /* NOTE: Do direct access to the evaluated mesh: this function is used + * during meta balls evaluation. But even without those all the objects + * which are needed for correct instancing are already evaluated. */ + if (em != nullptr) { + /* Note that this will only show deformation if #eModifierMode_OnCage is enabled. + * We could change this but it matches 2.7x behavior. */ + me_eval = em->mesh_eval_cage; + if ((me_eval == nullptr) || (me_eval->runtime.wrapper_type == ME_WRAPPER_TYPE_BMESH)) { + EditMeshData *emd = me_eval ? me_eval->runtime.edit_data : nullptr; + + /* Only assign edit-mesh in the case we can't use `me_eval`. */ + *r_em = em; + me_eval = nullptr; + + if ((emd != nullptr) && (emd->vertexCos != nullptr)) { + *r_vert_coords = emd->vertexCos; + if (r_vert_normals != nullptr) { + BKE_editmesh_cache_ensure_vert_normals(em, emd); + *r_vert_normals = emd->vertexNos; + } + } + } + } + else { + me_eval = BKE_object_get_evaluated_mesh(ob); + } + return me_eval; +} + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Dupli-Collection Implementation (#OB_DUPLICOLLECTION) + * \{ */ + +static void make_duplis_collection(const DupliContext *ctx) +{ + Object *ob = ctx->object; + Collection *collection; + float collection_mat[4][4]; + + if (ob->instance_collection == nullptr) { + return; + } + collection = ob->instance_collection; + + /* Combine collection offset and `obmat`. */ + unit_m4(collection_mat); + sub_v3_v3(collection_mat[3], collection->instance_offset); + mul_m4_m4m4(collection_mat, ob->obmat, collection_mat); + /* Don't access 'ob->obmat' from now on. */ + + eEvaluationMode mode = DEG_get_mode(ctx->depsgraph); + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN (collection, cob, mode) { + if (cob != ob) { + float mat[4][4]; + + /* Collection dupli-offset, should apply after everything else. */ + mul_m4_m4m4(mat, collection_mat, cob->obmat); + + make_dupli(ctx, cob, mat, _base_id); + + /* Recursion. */ + make_recursive_duplis(ctx, cob, collection_mat, _base_id); + } + } + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; +} + +static const DupliGenerator gen_dupli_collection = { + OB_DUPLICOLLECTION, /* type */ + make_duplis_collection /* make_duplis */ +}; + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Dupli-Vertices Implementation (#OB_DUPLIVERTS for Geometry) + * \{ */ + +/** Values shared between different mesh types. */ +struct VertexDupliData_Params { + /** + * It's important we use this context instead of the `ctx` passed into #make_child_duplis + * since these won't match in the case of recursion. + */ + const DupliContext *ctx; + + bool use_rotation; +}; + +struct VertexDupliData_Mesh { + VertexDupliData_Params params; + + int totvert; + const MVert *mvert; + + const float (*orco)[3]; +}; + +struct VertexDupliData_EditMesh { + VertexDupliData_Params params; + + BMEditMesh *em; + + /* Can be nullptr. */ + const float (*vert_coords)[3]; + const float (*vert_normals)[3]; + + /** + * \note The edit-mesh may assign #DupliObject.orco in cases when a regular mesh wouldn't. + * For edit-meshes we only check for deformation, for regular meshes we check if #CD_ORCO exists. + * + * At the moment this isn't a meaningful difference since requesting #CD_ORCO causes the + * edit-mesh to be converted into a mesh. + */ + bool has_orco; +}; + +/** + * \param no: The direction, + * currently this is copied from a `short[3]` normal without division. + * Can be null when \a use_rotation is false. + */ +static void get_duplivert_transform(const float co[3], + const float no[3], + const bool use_rotation, + const short axis, + const short upflag, + float r_mat[4][4]) +{ + float quat[4]; + const float size[3] = {1.0f, 1.0f, 1.0f}; + + if (use_rotation) { + /* Construct rotation matrix from normals. */ + float no_flip[3]; + negate_v3_v3(no_flip, no); + vec_to_quat(quat, no_flip, axis, upflag); + } + else { + unit_qt(quat); + } + + loc_quat_size_to_mat4(r_mat, co, quat, size); +} + +static DupliObject *vertex_dupli(const DupliContext *ctx, + Object *inst_ob, + const float child_imat[4][4], + int index, + const float co[3], + const float no[3], + const bool use_rotation) +{ + /* `obmat` is transform to vertex. */ + float obmat[4][4]; + get_duplivert_transform(co, no, use_rotation, inst_ob->trackflag, inst_ob->upflag, obmat); + + float space_mat[4][4]; + + /* Make offset relative to inst_ob using relative child transform. */ + mul_mat3_m4_v3(child_imat, obmat[3]); + /* Apply `obmat` _after_ the local vertex transform. */ + mul_m4_m4m4(obmat, inst_ob->obmat, obmat); + + /* Space matrix is constructed by removing `obmat` transform, + * this yields the world-space transform for recursive duplis. */ + mul_m4_m4m4(space_mat, obmat, inst_ob->imat); + + DupliObject *dob = make_dupli(ctx, inst_ob, obmat, index); + + /* Recursion. */ + make_recursive_duplis(ctx, inst_ob, space_mat, index); + + return dob; +} + +static void make_child_duplis_verts_from_mesh(const DupliContext *ctx, + void *userdata, + Object *inst_ob) +{ + VertexDupliData_Mesh *vdd = (VertexDupliData_Mesh *)userdata; + const bool use_rotation = vdd->params.use_rotation; + + const MVert *mvert = vdd->mvert; + const int totvert = vdd->totvert; + + invert_m4_m4(inst_ob->imat, inst_ob->obmat); + /* Relative transform from parent to child space. */ + float child_imat[4][4]; + mul_m4_m4m4(child_imat, inst_ob->imat, ctx->object->obmat); + + const MVert *mv = mvert; + for (int i = 0; i < totvert; i++, mv++) { + const float *co = mv->co; + float no[3]; + normal_short_to_float_v3(no, mv->no); + DupliObject *dob = vertex_dupli(vdd->params.ctx, inst_ob, child_imat, i, co, no, use_rotation); + if (vdd->orco) { + copy_v3_v3(dob->orco, vdd->orco[i]); + } + } +} + +static void make_child_duplis_verts_from_editmesh(const DupliContext *ctx, + void *userdata, + Object *inst_ob) +{ + VertexDupliData_EditMesh *vdd = (VertexDupliData_EditMesh *)userdata; + BMEditMesh *em = vdd->em; + const bool use_rotation = vdd->params.use_rotation; + + invert_m4_m4(inst_ob->imat, inst_ob->obmat); + /* Relative transform from parent to child space. */ + float child_imat[4][4]; + mul_m4_m4m4(child_imat, inst_ob->imat, ctx->object->obmat); + + BMVert *v; + BMIter iter; + int i; + + const float(*vert_coords)[3] = vdd->vert_coords; + const float(*vert_normals)[3] = vdd->vert_normals; + + BM_ITER_MESH_INDEX (v, &iter, em->bm, BM_VERTS_OF_MESH, i) { + const float *co, *no; + if (vert_coords != nullptr) { + co = vert_coords[i]; + no = vert_normals ? vert_normals[i] : nullptr; + } + else { + co = v->co; + no = v->no; + } + + DupliObject *dob = vertex_dupli(vdd->params.ctx, inst_ob, child_imat, i, co, no, use_rotation); + if (vdd->has_orco) { + copy_v3_v3(dob->orco, v->co); + } + } +} + +static void make_duplis_verts(const DupliContext *ctx) +{ + Object *parent = ctx->object; + const bool use_rotation = parent->transflag & OB_DUPLIROT; + + /* Gather mesh info. */ + BMEditMesh *em = nullptr; + const float(*vert_coords)[3] = nullptr; + const float(*vert_normals)[3] = nullptr; + Mesh *me_eval = mesh_data_from_duplicator_object( + parent, &em, &vert_coords, use_rotation ? &vert_normals : nullptr); + if (em == nullptr && me_eval == nullptr) { + return; + } + + VertexDupliData_Params vdd_params{ctx, use_rotation}; + + if (em != nullptr) { + VertexDupliData_EditMesh vdd{}; + vdd.params = vdd_params; + vdd.em = em; + vdd.vert_coords = vert_coords; + vdd.vert_normals = vert_normals; + vdd.has_orco = (vert_coords != nullptr); + + make_child_duplis(ctx, &vdd, make_child_duplis_verts_from_editmesh); + } + else { + VertexDupliData_Mesh vdd{}; + vdd.params = vdd_params; + vdd.totvert = me_eval->totvert; + vdd.mvert = me_eval->mvert; + vdd.orco = (const float(*)[3])CustomData_get_layer(&me_eval->vdata, CD_ORCO); + + make_child_duplis(ctx, &vdd, make_child_duplis_verts_from_mesh); + } +} + +static const DupliGenerator gen_dupli_verts = { + OB_DUPLIVERTS, /* type */ + make_duplis_verts /* make_duplis */ +}; + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Dupli-Vertices Implementation (#OB_DUPLIVERTS for 3D Text) + * \{ */ + +static Object *find_family_object( + Main *bmain, const char *family, size_t family_len, unsigned int ch, GHash *family_gh) +{ + void *ch_key = POINTER_FROM_UINT(ch); + + Object **ob_pt; + if ((ob_pt = (Object **)BLI_ghash_lookup_p(family_gh, ch_key))) { + return *ob_pt; + } + + char ch_utf8[7]; + size_t ch_utf8_len; + + ch_utf8_len = BLI_str_utf8_from_unicode(ch, ch_utf8); + ch_utf8[ch_utf8_len] = '\0'; + ch_utf8_len += 1; /* Compare with null terminator. */ + + LISTBASE_FOREACH (Object *, ob, &bmain->objects) { + if (STREQLEN(ob->id.name + 2 + family_len, ch_utf8, ch_utf8_len)) { + if (STREQLEN(ob->id.name + 2, family, family_len)) { + /* Inserted value can be nullptr, just to save searches in future. */ + BLI_ghash_insert(family_gh, ch_key, ob); + return ob; + } + } + } + + return nullptr; +} + +static void make_duplis_font(const DupliContext *ctx) +{ + Object *par = ctx->object; + GHash *family_gh; + Object *ob; + Curve *cu; + struct CharTrans *ct, *chartransdata = nullptr; + float vec[3], obmat[4][4], pmat[4][4], fsize, xof, yof; + int text_len, a; + size_t family_len; + const char32_t *text = nullptr; + bool text_free = false; + + /* Font dupli-verts not supported inside collections. */ + if (ctx->collection) { + return; + } + + copy_m4_m4(pmat, par->obmat); + + /* In `par` the family name is stored, use this to find the other objects. */ + + BKE_vfont_to_curve_ex( + par, (Curve *)par->data, FO_DUPLI, nullptr, &text, &text_len, &text_free, &chartransdata); + + if (text == nullptr || chartransdata == nullptr) { + return; + } + + cu = (Curve *)par->data; + fsize = cu->fsize; + xof = cu->xof; + yof = cu->yof; + + ct = chartransdata; + + /* Cache result. */ + family_len = strlen(cu->family); + family_gh = BLI_ghash_int_new_ex(__func__, 256); + + /* Safety check even if it might fail badly when called for original object. */ + const bool is_eval_curve = DEG_is_evaluated_id(&cu->id); + + /* Advance matching BLI_str_utf8_as_utf32. */ + for (a = 0; a < text_len; a++, ct++) { + + /* XXX That G.main is *really* ugly, but not sure what to do here. + * Definitively don't think it would be safe to put back `Main *bmain` pointer + * in #DupliContext as done in 2.7x? */ + ob = find_family_object(G.main, cu->family, family_len, (unsigned int)text[a], family_gh); + + if (is_eval_curve) { + /* Workaround for the above hack. */ + ob = DEG_get_evaluated_object(ctx->depsgraph, ob); + } + + if (ob) { + vec[0] = fsize * (ct->xof - xof); + vec[1] = fsize * (ct->yof - yof); + vec[2] = 0.0; + + mul_m4_v3(pmat, vec); + + copy_m4_m4(obmat, par->obmat); + + if (UNLIKELY(ct->rot != 0.0f)) { + float rmat[4][4]; + + zero_v3(obmat[3]); + axis_angle_to_mat4_single(rmat, 'Z', -ct->rot); + mul_m4_m4m4(obmat, obmat, rmat); + } + + copy_v3_v3(obmat[3], vec); + + make_dupli(ctx, ob, obmat, a); + } + } + + if (text_free) { + MEM_freeN((void *)text); + } + + BLI_ghash_free(family_gh, nullptr, nullptr); + + MEM_freeN(chartransdata); +} + +static const DupliGenerator gen_dupli_verts_font = { + OB_DUPLIVERTS, /* type */ + make_duplis_font /* make_duplis */ +}; + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Dupli-Vertices Implementation (#OB_DUPLIVERTS for #PointCloud) + * \{ */ + +static void make_child_duplis_pointcloud(const DupliContext *ctx, + void *UNUSED(userdata), + Object *child) +{ + const Object *parent = ctx->object; + const PointCloud *pointcloud = (PointCloud *)parent->data; + const float(*co)[3] = pointcloud->co; + const float *radius = pointcloud->radius; + const float(*rotation)[4] = nullptr; /* TODO: add optional rotation attribute. */ + const float(*orco)[3] = nullptr; /* TODO: add optional texture coordinate attribute. */ + + /* Relative transform from parent to child space. */ + float child_imat[4][4]; + mul_m4_m4m4(child_imat, child->imat, parent->obmat); + + for (int i = 0; i < pointcloud->totpoint; i++) { + /* Transform matrix from point position, radius and rotation. */ + float quat[4] = {1.0f, 0.0f, 0.0f, 0.0f}; + float size[3] = {1.0f, 1.0f, 1.0f}; + if (radius) { + copy_v3_fl(size, radius[i]); + } + if (rotation) { + copy_v4_v4(quat, rotation[i]); + } + + float space_mat[4][4]; + loc_quat_size_to_mat4(space_mat, co[i], quat, size); + + /* Make offset relative to child object using relative child transform, + * and apply object matrix after local vertex transform. */ + mul_mat3_m4_v3(child_imat, space_mat[3]); + + /* Create dupli object. */ + float obmat[4][4]; + mul_m4_m4m4(obmat, child->obmat, space_mat); + DupliObject *dob = make_dupli(ctx, child, obmat, i); + if (orco) { + copy_v3_v3(dob->orco, orco[i]); + } + + /* Recursion. */ + make_recursive_duplis(ctx, child, space_mat, i); + } +} + +static void make_duplis_pointcloud(const DupliContext *ctx) +{ + make_child_duplis(ctx, nullptr, make_child_duplis_pointcloud); +} + +static const DupliGenerator gen_dupli_verts_pointcloud = { + OB_DUPLIVERTS, /* type */ + make_duplis_pointcloud /* make_duplis */ +}; + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Instances Geometry Component Implementation + * \{ */ + +static void make_duplis_instances_component(const DupliContext *ctx) +{ + float(*instance_offset_matrices)[4][4]; + InstancedData *instanced_data; + const int *almost_unique_ids; + const int amount = BKE_geometry_set_instances(ctx->object->runtime.geometry_set_eval, + &instance_offset_matrices, + &almost_unique_ids, + &instanced_data); + + for (int i = 0; i < amount; i++) { + InstancedData *data = &instanced_data[i]; + + const int id = almost_unique_ids[i]; + + if (data->type == INSTANCE_DATA_TYPE_OBJECT) { + Object *object = data->data.object; + if (object != nullptr) { + float matrix[4][4]; + mul_m4_m4m4(matrix, ctx->object->obmat, instance_offset_matrices[i]); + make_dupli(ctx, object, matrix, id); + + float space_matrix[4][4]; + mul_m4_m4m4(space_matrix, instance_offset_matrices[i], object->imat); + mul_m4_m4_pre(space_matrix, ctx->object->obmat); + make_recursive_duplis(ctx, object, space_matrix, id); + } + } + else if (data->type == INSTANCE_DATA_TYPE_COLLECTION) { + Collection *collection = data->data.collection; + if (collection != nullptr) { + float collection_matrix[4][4]; + unit_m4(collection_matrix); + sub_v3_v3(collection_matrix[3], collection->instance_offset); + mul_m4_m4_pre(collection_matrix, instance_offset_matrices[i]); + mul_m4_m4_pre(collection_matrix, ctx->object->obmat); + + eEvaluationMode mode = DEG_get_mode(ctx->depsgraph); + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN (collection, object, mode) { + if (object == ctx->object) { + continue; + } + + float instance_matrix[4][4]; + mul_m4_m4m4(instance_matrix, collection_matrix, object->obmat); + + make_dupli(ctx, object, instance_matrix, id); + make_recursive_duplis(ctx, object, collection_matrix, id); + } + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; + } + } + } +} + +static const DupliGenerator gen_dupli_instances_component = { + 0, + make_duplis_instances_component, +}; + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Dupli-Faces Implementation (#OB_DUPLIFACES) + * \{ */ + +/** Values shared between different mesh types. */ +struct FaceDupliData_Params { + /** + * It's important we use this context instead of the `ctx` passed into #make_child_duplis + * since these won't match in the case of recursion. + */ + const DupliContext *ctx; + + bool use_scale; +}; + +struct FaceDupliData_Mesh { + FaceDupliData_Params params; + + int totface; + const MPoly *mpoly; + const MLoop *mloop; + const MVert *mvert; + const float (*orco)[3]; + const MLoopUV *mloopuv; +}; + +struct FaceDupliData_EditMesh { + FaceDupliData_Params params; + + BMEditMesh *em; + + bool has_orco, has_uvs; + int cd_loop_uv_offset; + /* Can be nullptr. */ + const float (*vert_coords)[3]; +}; + +static void get_dupliface_transform_from_coords(const float coords[][3], + const int coords_len, + const bool use_scale, + const float scale_fac, + float r_mat[4][4]) +{ + float loc[3], quat[4], scale, size[3]; + + /* Location. */ + { + const float w = 1.0f / (float)coords_len; + zero_v3(loc); + for (int i = 0; i < coords_len; i++) { + madd_v3_v3fl(loc, coords[i], w); + } + } + /* Rotation. */ + { + float f_no[3]; + cross_poly_v3(f_no, coords, (uint)coords_len); + normalize_v3(f_no); + tri_to_quat_ex(quat, coords[0], coords[1], coords[2], f_no); + } + /* Scale. */ + if (use_scale) { + const float area = area_poly_v3(coords, (uint)coords_len); + scale = sqrtf(area) * scale_fac; + } + else { + scale = 1.0f; + } + size[0] = size[1] = size[2] = scale; + + loc_quat_size_to_mat4(r_mat, loc, quat, size); +} + +static DupliObject *face_dupli(const DupliContext *ctx, + Object *inst_ob, + const float child_imat[4][4], + const int index, + const bool use_scale, + const float scale_fac, + const float (*coords)[3], + const int coords_len) +{ + float obmat[4][4]; + float space_mat[4][4]; + + /* `obmat` is transform to face. */ + get_dupliface_transform_from_coords(coords, coords_len, use_scale, scale_fac, obmat); + + /* Make offset relative to inst_ob using relative child transform. */ + mul_mat3_m4_v3(child_imat, obmat[3]); + + /* XXX ugly hack to ensure same behavior as in master. + * This should not be needed, #Object.parentinv is not consistent outside of parenting. */ + { + float imat[3][3]; + copy_m3_m4(imat, inst_ob->parentinv); + mul_m4_m3m4(obmat, imat, obmat); + } + + /* Apply `obmat` _after_ the local face transform. */ + mul_m4_m4m4(obmat, inst_ob->obmat, obmat); + + /* Space matrix is constructed by removing `obmat` transform, + * this yields the world-space transform for recursive duplis. */ + mul_m4_m4m4(space_mat, obmat, inst_ob->imat); + + DupliObject *dob = make_dupli(ctx, inst_ob, obmat, index); + + /* Recursion. */ + make_recursive_duplis(ctx, inst_ob, space_mat, index); + + return dob; +} + +/** Wrap #face_dupli, needed since we can't #alloca in a loop. */ +static DupliObject *face_dupli_from_mesh(const DupliContext *ctx, + Object *inst_ob, + const float child_imat[4][4], + const int index, + const bool use_scale, + const float scale_fac, + + /* Mesh variables. */ + const MPoly *mpoly, + const MLoop *mloopstart, + const MVert *mvert) +{ + const int coords_len = mpoly->totloop; + float(*coords)[3] = BLI_array_alloca(coords, (size_t)coords_len); + + const MLoop *ml = mloopstart; + for (int i = 0; i < coords_len; i++, ml++) { + copy_v3_v3(coords[i], mvert[ml->v].co); + } + + return face_dupli(ctx, inst_ob, child_imat, index, use_scale, scale_fac, coords, coords_len); +} + +/** Wrap #face_dupli, needed since we can't #alloca in a loop. */ +static DupliObject *face_dupli_from_editmesh(const DupliContext *ctx, + Object *inst_ob, + const float child_imat[4][4], + const int index, + const bool use_scale, + const float scale_fac, + + /* Mesh variables. */ + BMFace *f, + const float (*vert_coords)[3]) +{ + const int coords_len = f->len; + float(*coords)[3] = BLI_array_alloca(coords, (size_t)coords_len); + + BMLoop *l_first, *l_iter; + int i = 0; + l_iter = l_first = BM_FACE_FIRST_LOOP(f); + if (vert_coords != nullptr) { + do { + copy_v3_v3(coords[i++], vert_coords[BM_elem_index_get(l_iter->v)]); + } while ((l_iter = l_iter->next) != l_first); + } + else { + do { + copy_v3_v3(coords[i++], l_iter->v->co); + } while ((l_iter = l_iter->next) != l_first); + } + + return face_dupli(ctx, inst_ob, child_imat, index, use_scale, scale_fac, coords, coords_len); +} + +static void make_child_duplis_faces_from_mesh(const DupliContext *ctx, + void *userdata, + Object *inst_ob) +{ + FaceDupliData_Mesh *fdd = (FaceDupliData_Mesh *)userdata; + const MPoly *mpoly = fdd->mpoly, *mp; + const MLoop *mloop = fdd->mloop; + const MVert *mvert = fdd->mvert; + const float(*orco)[3] = fdd->orco; + const MLoopUV *mloopuv = fdd->mloopuv; + const int totface = fdd->totface; + const bool use_scale = fdd->params.use_scale; + int a; + + float child_imat[4][4]; + + invert_m4_m4(inst_ob->imat, inst_ob->obmat); + /* Relative transform from parent to child space. */ + mul_m4_m4m4(child_imat, inst_ob->imat, ctx->object->obmat); + const float scale_fac = ctx->object->instance_faces_scale; + + for (a = 0, mp = mpoly; a < totface; a++, mp++) { + const MLoop *loopstart = mloop + mp->loopstart; + DupliObject *dob = face_dupli_from_mesh( + fdd->params.ctx, inst_ob, child_imat, a, use_scale, scale_fac, mp, loopstart, mvert); + + const float w = 1.0f / (float)mp->totloop; + if (orco) { + for (int j = 0; j < mp->totloop; j++) { + madd_v3_v3fl(dob->orco, orco[loopstart[j].v], w); + } + } + if (mloopuv) { + for (int j = 0; j < mp->totloop; j++) { + madd_v2_v2fl(dob->uv, mloopuv[mp->loopstart + j].uv, w); + } + } + } +} + +static void make_child_duplis_faces_from_editmesh(const DupliContext *ctx, + void *userdata, + Object *inst_ob) +{ + FaceDupliData_EditMesh *fdd = (FaceDupliData_EditMesh *)userdata; + BMEditMesh *em = fdd->em; + float child_imat[4][4]; + int a; + BMFace *f; + BMIter iter; + const bool use_scale = fdd->params.use_scale; + + const float(*vert_coords)[3] = fdd->vert_coords; + + BLI_assert((vert_coords == nullptr) || (em->bm->elem_index_dirty & BM_VERT) == 0); + + invert_m4_m4(inst_ob->imat, inst_ob->obmat); + /* Relative transform from parent to child space. */ + mul_m4_m4m4(child_imat, inst_ob->imat, ctx->object->obmat); + const float scale_fac = ctx->object->instance_faces_scale; + + BM_ITER_MESH_INDEX (f, &iter, em->bm, BM_FACES_OF_MESH, a) { + DupliObject *dob = face_dupli_from_editmesh( + fdd->params.ctx, inst_ob, child_imat, a, use_scale, scale_fac, f, vert_coords); + + if (fdd->has_orco) { + const float w = 1.0f / (float)f->len; + BMLoop *l_first, *l_iter; + l_iter = l_first = BM_FACE_FIRST_LOOP(f); + do { + madd_v3_v3fl(dob->orco, l_iter->v->co, w); + } while ((l_iter = l_iter->next) != l_first); + } + if (fdd->has_uvs) { + BM_face_uv_calc_center_median(f, fdd->cd_loop_uv_offset, dob->uv); + } + } +} + +static void make_duplis_faces(const DupliContext *ctx) +{ + Object *parent = ctx->object; + + /* Gather mesh info. */ + BMEditMesh *em = nullptr; + const float(*vert_coords)[3] = nullptr; + Mesh *me_eval = mesh_data_from_duplicator_object(parent, &em, &vert_coords, nullptr); + if (em == nullptr && me_eval == nullptr) { + return; + } + + FaceDupliData_Params fdd_params = {ctx, (parent->transflag & OB_DUPLIFACES_SCALE) != 0}; + + if (em != nullptr) { + const int uv_idx = CustomData_get_render_layer(&em->bm->ldata, CD_MLOOPUV); + FaceDupliData_EditMesh fdd{}; + fdd.params = fdd_params; + fdd.em = em; + fdd.vert_coords = vert_coords; + fdd.has_orco = (vert_coords != nullptr); + fdd.has_uvs = (uv_idx != -1); + fdd.cd_loop_uv_offset = (uv_idx != -1) ? + CustomData_get_n_offset(&em->bm->ldata, CD_MLOOPUV, uv_idx) : + -1; + make_child_duplis(ctx, &fdd, make_child_duplis_faces_from_editmesh); + } + else { + const int uv_idx = CustomData_get_render_layer(&me_eval->ldata, CD_MLOOPUV); + FaceDupliData_Mesh fdd{}; + fdd.params = fdd_params; + fdd.totface = me_eval->totpoly; + fdd.mpoly = me_eval->mpoly; + fdd.mloop = me_eval->mloop; + fdd.mvert = me_eval->mvert; + fdd.mloopuv = (uv_idx != -1) ? (const MLoopUV *)CustomData_get_layer_n( + &me_eval->ldata, CD_MLOOPUV, uv_idx) : + nullptr; + fdd.orco = (const float(*)[3])CustomData_get_layer(&me_eval->vdata, CD_ORCO); + + make_child_duplis(ctx, &fdd, make_child_duplis_faces_from_mesh); + } +} + +static const DupliGenerator gen_dupli_faces = { + OB_DUPLIFACES, /* type */ + make_duplis_faces /* make_duplis */ +}; + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Dupli-Particles Implementation (#OB_DUPLIPARTS) + * \{ */ + +static void make_duplis_particle_system(const DupliContext *ctx, ParticleSystem *psys) +{ + Scene *scene = ctx->scene; + Object *par = ctx->object; + eEvaluationMode mode = DEG_get_mode(ctx->depsgraph); + bool for_render = mode == DAG_EVAL_RENDER; + + Object *ob = nullptr, **oblist = nullptr; + DupliObject *dob; + ParticleSettings *part; + ParticleData *pa; + ChildParticle *cpa = nullptr; + ParticleKey state; + ParticleCacheKey *cache; + float ctime, scale = 1.0f; + float tmat[4][4], mat[4][4], pamat[4][4], size = 0.0; + int a, b, hair = 0; + int totpart, totchild; + + int no_draw_flag = PARS_UNEXIST; + + if (psys == nullptr) { + return; + } + + part = psys->part; + + if (part == nullptr) { + return; + } + + if (!psys_check_enabled(par, psys, for_render)) { + return; + } + + if (!for_render) { + no_draw_flag |= PARS_NO_DISP; + } + + /* NOTE: in old animation system, used parent object's time-offset. */ + ctime = DEG_get_ctime(ctx->depsgraph); + + totpart = psys->totpart; + totchild = psys->totchild; + + if ((for_render || part->draw_as == PART_DRAW_REND) && + ELEM(part->ren_as, PART_DRAW_OB, PART_DRAW_GR)) { + ParticleSimulationData sim = {nullptr}; + sim.depsgraph = ctx->depsgraph; + sim.scene = scene; + sim.ob = par; + sim.psys = psys; + sim.psmd = psys_get_modifier(par, psys); + /* Make sure emitter `imat` is in global coordinates instead of render view coordinates. */ + invert_m4_m4(par->imat, par->obmat); + + /* First check for loops (particle system object used as dupli-object). */ + if (part->ren_as == PART_DRAW_OB) { + if (ELEM(part->instance_object, nullptr, par)) { + return; + } + } + else { /* #PART_DRAW_GR. */ + if (part->instance_collection == nullptr) { + return; + } + + const ListBase dup_collection_objects = BKE_collection_object_cache_get( + part->instance_collection); + if (BLI_listbase_is_empty(&dup_collection_objects)) { + return; + } + + if (BLI_findptr(&dup_collection_objects, par, offsetof(Base, object))) { + return; + } + } + + /* If we have a hair particle system, use the path cache. */ + if (part->type == PART_HAIR) { + if (psys->flag & PSYS_HAIR_DONE) { + hair = (totchild == 0 || psys->childcache) && psys->pathcache; + } + if (!hair) { + return; + } + + /* We use cache, update `totchild` according to cached data. */ + totchild = psys->totchildcache; + totpart = psys->totcached; + } + + RNG *rng = BLI_rng_new_srandom(31415926u + (unsigned int)psys->seed); + + psys->lattice_deform_data = psys_create_lattice_deform_data(&sim); + + /* Gather list of objects or single object. */ + int totcollection = 0; + + const bool use_whole_collection = part->draw & PART_DRAW_WHOLE_GR; + const bool use_collection_count = part->draw & PART_DRAW_COUNT_GR && !use_whole_collection; + if (part->ren_as == PART_DRAW_GR) { + if (use_collection_count) { + psys_find_group_weights(part); + LISTBASE_FOREACH (ParticleDupliWeight *, dw, &part->instance_weights) { + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN ( + part->instance_collection, object, mode) { + if (dw->ob == object) { + totcollection += dw->count; + break; + } + } + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; + } + } + else { + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN ( + part->instance_collection, object, mode) { + (void)object; + totcollection++; + } + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; + } + + oblist = (Object **)MEM_callocN((size_t)totcollection * sizeof(Object *), + "dupcollection object list"); + + if (use_collection_count) { + a = 0; + LISTBASE_FOREACH (ParticleDupliWeight *, dw, &part->instance_weights) { + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN ( + part->instance_collection, object, mode) { + if (dw->ob == object) { + for (b = 0; b < dw->count; b++, a++) { + oblist[a] = dw->ob; + } + break; + } + } + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; + } + } + else { + a = 0; + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN ( + part->instance_collection, object, mode) { + oblist[a] = object; + a++; + } + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; + } + } + else { + ob = part->instance_object; + } + + if (totchild == 0 || part->draw & PART_DRAW_PARENT) { + a = 0; + } + else { + a = totpart; + } + + for (pa = psys->particles; a < totpart + totchild; a++, pa++) { + if (a < totpart) { + /* Handle parent particle. */ + if (pa->flag & no_draw_flag) { + continue; + } + +#if 0 /* UNUSED */ + pa_num = pa->num; +#endif + size = pa->size; + } + else { + /* Handle child particle. */ + cpa = &psys->child[a - totpart]; + +#if 0 /* UNUSED */ + pa_num = a; +#endif + size = psys_get_child_size(psys, cpa, ctime, nullptr); + } + + /* Some hair paths might be non-existent so they can't be used for duplication. */ + if (hair && psys->pathcache && + ((a < totpart && psys->pathcache[a]->segments < 0) || + (a >= totpart && psys->childcache[a - totpart]->segments < 0))) { + continue; + } + + if (part->ren_as == PART_DRAW_GR) { + /* Prevent divide by zero below T28336. */ + if (totcollection == 0) { + continue; + } + + /* For collections, pick the object based on settings. */ + if (part->draw & PART_DRAW_RAND_GR && !use_whole_collection) { + b = BLI_rng_get_int(rng) % totcollection; + } + else { + b = a % totcollection; + } + + ob = oblist[b]; + } + + if (hair) { + /* Hair we handle separate and compute transform based on hair keys. */ + if (a < totpart) { + cache = psys->pathcache[a]; + psys_get_dupli_path_transform(&sim, pa, nullptr, cache, pamat, &scale); + } + else { + cache = psys->childcache[a - totpart]; + psys_get_dupli_path_transform(&sim, nullptr, cpa, cache, pamat, &scale); + } + + copy_v3_v3(pamat[3], cache->co); + pamat[3][3] = 1.0f; + } + else { + /* First key. */ + state.time = ctime; + if (psys_get_particle_state(&sim, a, &state, 0) == 0) { + continue; + } + + float tquat[4]; + normalize_qt_qt(tquat, state.rot); + quat_to_mat4(pamat, tquat); + copy_v3_v3(pamat[3], state.co); + pamat[3][3] = 1.0f; + } + + if (part->ren_as == PART_DRAW_GR && psys->part->draw & PART_DRAW_WHOLE_GR) { + b = 0; + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_BEGIN ( + part->instance_collection, object, mode) { + copy_m4_m4(tmat, oblist[b]->obmat); + + /* Apply collection instance offset. */ + sub_v3_v3(tmat[3], part->instance_collection->instance_offset); + + /* Apply particle scale. */ + mul_mat3_m4_fl(tmat, size * scale); + mul_v3_fl(tmat[3], size * scale); + + /* Individual particle transform. */ + mul_m4_m4m4(mat, pamat, tmat); + + dob = make_dupli(ctx, object, mat, a); + dob->particle_system = psys; + + psys_get_dupli_texture(psys, part, sim.psmd, pa, cpa, dob->uv, dob->orco); + + b++; + } + FOREACH_COLLECTION_VISIBLE_OBJECT_RECURSIVE_END; + } + else { + float obmat[4][4]; + copy_m4_m4(obmat, ob->obmat); + + float vec[3]; + copy_v3_v3(vec, obmat[3]); + zero_v3(obmat[3]); + + /* Particle rotation uses x-axis as the aligned axis, + * so pre-rotate the object accordingly. */ + if ((part->draw & PART_DRAW_ROTATE_OB) == 0) { + float xvec[3], q[4], size_mat[4][4], original_size[3]; + + mat4_to_size(original_size, obmat); + size_to_mat4(size_mat, original_size); + + xvec[0] = -1.0f; + xvec[1] = xvec[2] = 0; + vec_to_quat(q, xvec, ob->trackflag, ob->upflag); + quat_to_mat4(obmat, q); + obmat[3][3] = 1.0f; + + /* Add scaling if requested. */ + if ((part->draw & PART_DRAW_NO_SCALE_OB) == 0) { + mul_m4_m4m4(obmat, obmat, size_mat); + } + } + else if (part->draw & PART_DRAW_NO_SCALE_OB) { + /* Remove scaling. */ + float size_mat[4][4], original_size[3]; + + mat4_to_size(original_size, obmat); + size_to_mat4(size_mat, original_size); + invert_m4(size_mat); + + mul_m4_m4m4(obmat, obmat, size_mat); + } + + mul_m4_m4m4(tmat, pamat, obmat); + mul_mat3_m4_fl(tmat, size * scale); + + copy_m4_m4(mat, tmat); + + if (part->draw & PART_DRAW_GLOBAL_OB) { + add_v3_v3v3(mat[3], mat[3], vec); + } + + dob = make_dupli(ctx, ob, mat, a); + dob->particle_system = psys; + psys_get_dupli_texture(psys, part, sim.psmd, pa, cpa, dob->uv, dob->orco); + } + } + + BLI_rng_free(rng); + } + + /* Clean up. */ + if (oblist) { + MEM_freeN(oblist); + } + + if (psys->lattice_deform_data) { + BKE_lattice_deform_data_destroy(psys->lattice_deform_data); + psys->lattice_deform_data = nullptr; + } +} + +static void make_duplis_particles(const DupliContext *ctx) +{ + /* Particle system take up one level in id, the particles another. */ + int psysid; + LISTBASE_FOREACH_INDEX (ParticleSystem *, psys, &ctx->object->particlesystem, psysid) { + /* Particles create one more level for persistent `psys` index. */ + DupliContext pctx; + copy_dupli_context(&pctx, ctx, ctx->object, nullptr, psysid); + make_duplis_particle_system(&pctx, psys); + } +} + +static const DupliGenerator gen_dupli_particles = { + OB_DUPLIPARTS, /* type */ + make_duplis_particles /* make_duplis */ +}; + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Dupli-Generator Selector For The Given Context + * \{ */ + +static const DupliGenerator *get_dupli_generator(const DupliContext *ctx) +{ + int transflag = ctx->object->transflag; + int restrictflag = ctx->object->restrictflag; + + if ((transflag & OB_DUPLI) == 0 && ctx->object->runtime.geometry_set_eval == nullptr) { + return nullptr; + } + + /* Should the dupli's be generated for this object? - Respect restrict flags. */ + if (DEG_get_mode(ctx->depsgraph) == DAG_EVAL_RENDER ? (restrictflag & OB_RESTRICT_RENDER) : + (restrictflag & OB_RESTRICT_VIEWPORT)) { + return nullptr; + } + + if (ctx->object->runtime.geometry_set_eval != nullptr) { + if (BKE_geometry_set_has_instances(ctx->object->runtime.geometry_set_eval)) { + return &gen_dupli_instances_component; + } + } + + if (transflag & OB_DUPLIPARTS) { + return &gen_dupli_particles; + } + if (transflag & OB_DUPLIVERTS) { + if (ctx->object->type == OB_MESH) { + return &gen_dupli_verts; + } + if (ctx->object->type == OB_FONT) { + return &gen_dupli_verts_font; + } + if (ctx->object->type == OB_POINTCLOUD) { + return &gen_dupli_verts_pointcloud; + } + } + else if (transflag & OB_DUPLIFACES) { + if (ctx->object->type == OB_MESH) { + return &gen_dupli_faces; + } + } + else if (transflag & OB_DUPLICOLLECTION) { + return &gen_dupli_collection; + } + + return nullptr; +} + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Dupli-Container Implementation + * \{ */ + +/** + * \return a #ListBase of #DupliObject. + */ +ListBase *object_duplilist(Depsgraph *depsgraph, Scene *sce, Object *ob) +{ + ListBase *duplilist = (ListBase *)MEM_callocN(sizeof(ListBase), "duplilist"); + DupliContext ctx; + init_context(&ctx, depsgraph, sce, ob, nullptr); + if (ctx.gen) { + ctx.duplilist = duplilist; + ctx.gen->make_duplis(&ctx); + } + + return duplilist; +} + +void free_object_duplilist(ListBase *lb) +{ + BLI_freelistN(lb); + MEM_freeN(lb); +} + +/** \} */ -- cgit v1.2.3