Merge branch 'master' into blender2.8
[blender.git] / source / blender / modifiers / intern / MOD_util.c
index a271d0baa9186d86b4958358acb0ddf885c43369..6de0cd753cd1467820f6767497f9a401252e48ea 100644 (file)
@@ -34,6 +34,7 @@
 
 #include "DNA_image_types.h"
 #include "DNA_meshdata_types.h"
+#include "DNA_mesh_types.h"
 #include "DNA_modifier_types.h"
 #include "DNA_object_types.h"
 #include "DNA_scene_types.h"
 
 #include "BKE_cdderivedmesh.h"
 #include "BKE_deform.h"
+#include "BKE_editmesh.h"
 #include "BKE_image.h"
 #include "BKE_lattice.h"
+#include "BKE_library.h"
 #include "BKE_mesh.h"
 
 #include "BKE_modifier.h"
@@ -55,6 +58,8 @@
 
 #include "MEM_guardedalloc.h"
 
+#include "bmesh.h"
+
 void modifier_init_texture(const Scene *scene, Tex *tex)
 {
        if (!tex)
@@ -65,6 +70,88 @@ void modifier_init_texture(const Scene *scene, Tex *tex)
        }
 }
 
+/* TODO to be renamed to get_texture_coords once we are done with moving modifiers to Mesh. */
+/** \param cos may be NULL, in which case we use directly mesh vertices' coordinates. */
+void get_texture_coords_mesh(
+        MappingInfoModifierData *dmd,
+        Object *ob,
+        Mesh *mesh,
+        float (*cos)[3],
+        float (*r_texco)[3])
+{
+       const int numVerts = mesh->totvert;
+       int i;
+       int texmapping = dmd->texmapping;
+       float mapob_imat[4][4];
+
+       if (texmapping == MOD_DISP_MAP_OBJECT) {
+               if (dmd->map_object)
+                       invert_m4_m4(mapob_imat, dmd->map_object->obmat);
+               else /* if there is no map object, default to local */
+                       texmapping = MOD_DISP_MAP_LOCAL;
+       }
+
+       /* UVs need special handling, since they come from faces */
+       if (texmapping == MOD_DISP_MAP_UV) {
+               if (CustomData_has_layer(&mesh->ldata, CD_MLOOPUV)) {
+                       MPoly *mpoly = mesh->mpoly;
+                       MPoly *mp;
+                       MLoop *mloop = mesh->mloop;
+                       BLI_bitmap *done = BLI_BITMAP_NEW(numVerts, __func__);
+                       const int numPolys = mesh->totpoly;
+                       char uvname[MAX_CUSTOMDATA_LAYER_NAME];
+                       MLoopUV *mloop_uv;
+
+                       CustomData_validate_layer_name(&mesh->ldata, CD_MLOOPUV, dmd->uvlayer_name, uvname);
+                       mloop_uv = CustomData_get_layer_named(&mesh->ldata, CD_MLOOPUV, uvname);
+
+                       /* verts are given the UV from the first face that uses them */
+                       for (i = 0, mp = mpoly; i < numPolys; ++i, ++mp) {
+                               unsigned int fidx = mp->totloop - 1;
+
+                               do {
+                                       unsigned int lidx = mp->loopstart + fidx;
+                                       unsigned int vidx = mloop[lidx].v;
+
+                                       if (!BLI_BITMAP_TEST(done, vidx)) {
+                                               /* remap UVs from [0, 1] to [-1, 1] */
+                                               r_texco[vidx][0] = (mloop_uv[lidx].uv[0] * 2.0f) - 1.0f;
+                                               r_texco[vidx][1] = (mloop_uv[lidx].uv[1] * 2.0f) - 1.0f;
+                                               BLI_BITMAP_ENABLE(done, vidx);
+                                       }
+
+                               } while (fidx--);
+                       }
+
+                       MEM_freeN(done);
+                       return;
+               }
+               else {
+                       /* if there are no UVs, default to local */
+                       texmapping = MOD_DISP_MAP_LOCAL;
+               }
+       }
+
+       MVert *mv = mesh->mvert;
+       for (i = 0; i < numVerts; ++i, ++mv, ++r_texco) {
+               switch (texmapping) {
+                       case MOD_DISP_MAP_LOCAL:
+                               copy_v3_v3(*r_texco, cos != NULL ? *cos : mv->co);
+                               break;
+                       case MOD_DISP_MAP_GLOBAL:
+                               mul_v3_m4v3(*r_texco, ob->obmat, cos != NULL ? *cos : mv->co);
+                               break;
+                       case MOD_DISP_MAP_OBJECT:
+                               mul_v3_m4v3(*r_texco, ob->obmat, cos != NULL ? *cos : mv->co);
+                               mul_m4_v3(mapob_imat, *r_texco);
+                               break;
+               }
+               if (cos != NULL) {
+                       cos++;
+               }
+       }
+}
+
 void get_texture_coords(
         MappingInfoModifierData *dmd, Object *ob,
         DerivedMesh *dm,
@@ -204,6 +291,53 @@ DerivedMesh *get_dm(
        return dm;
 }
 
+/* returns a mesh if mesh == NULL, for deforming modifiers that need it */
+Mesh *get_mesh(
+        Object *ob, struct BMEditMesh *em, Mesh *mesh,
+        float (*vertexCos)[3], bool use_normals, bool use_orco)
+{
+       if (mesh) {
+               /* pass */
+       }
+       else if (ob->type == OB_MESH) {
+               struct BMeshToMeshParams bmtmp = {0};
+               if (em) mesh = BKE_bmesh_to_mesh_nomain(em->bm, &bmtmp);
+               else {
+                       /* TODO(sybren): after modifier conversion of DM to Mesh is done, check whether
+                        * we really need a copy here. Maybe the CoW ob->data can be directly used. */
+                       BKE_id_copy_ex(
+                               NULL, ob->data, (ID **)&mesh,
+                               LIB_ID_CREATE_NO_MAIN |
+                               LIB_ID_CREATE_NO_USER_REFCOUNT |
+                               LIB_ID_CREATE_NO_DEG_TAG,
+                               false);
+               }
+
+               /* TODO(sybren): after modifier conversion of DM to Mesh is done, check whether
+                * we really need vertexCos here. */
+               if (vertexCos) {
+                       BKE_mesh_apply_vert_coords(mesh, vertexCos);
+                       mesh->runtime.cd_dirty_vert |= CD_MASK_NORMAL;
+               }
+
+               if (use_orco) {
+                       CustomData_add_layer(&mesh->vdata, CD_ORCO, CD_ASSIGN, BKE_mesh_orco_verts_get(ob), mesh->totvert);
+               }
+       }
+       else if (ELEM(ob->type, OB_FONT, OB_CURVE, OB_SURF)) {
+               /* TODO(sybren): get evaluated mesh from depsgraph once that's properly generated for curves. */
+               mesh = BKE_mesh_new_nomain_from_curve(ob);
+       }
+
+       if (use_normals) {
+               if (LIKELY(mesh)) {
+                       BKE_mesh_ensure_normals(mesh);
+               }
+       }
+
+       return mesh;
+}
+
 /* Get derived mesh for other object, which is used as an operand for the modifier,
  * i.e. second operand for boolean modifier.
  */
@@ -231,6 +365,20 @@ void modifier_get_vgroup(Object *ob, DerivedMesh *dm, const char *name, MDeformV
        }
 }
 
+/* TODO(sybren): replace the above function with this one, once we got rid of DerivedMesh for modifiers. */
+void modifier_get_vgroup_mesh(Object *ob, struct Mesh *mesh, const char *name, MDeformVert **dvert, int *defgrp_index)
+{
+       *defgrp_index = defgroup_name_index(ob, name);
+       *dvert = NULL;
+
+       if (*defgrp_index != -1) {
+               if (ob->type == OB_LATTICE)
+                       *dvert = BKE_lattice_deform_verts_get(ob);
+               else if (mesh)
+                       *dvert = mesh->dvert;
+       }
+}
+
 
 /* only called by BKE_modifier.h/modifier.c */
 void modifier_type_init(ModifierTypeInfo *types[])