/* * ***** BEGIN GPL LICENSE BLOCK ***** * * 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) 2005 Blender Foundation. * All rights reserved. * * The Original Code is: all of this file. * * Contributor(s): none yet. * * ***** END GPL LICENSE BLOCK ***** */ /** \file blender/blenkernel/intern/DerivedMesh.c * \ingroup bke */ #include #include #include "MEM_guardedalloc.h" #include "DNA_cloth_types.h" #include "DNA_key_types.h" #include "DNA_material_types.h" #include "DNA_mesh_types.h" #include "DNA_meshdata_types.h" #include "DNA_object_types.h" #include "DNA_scene_types.h" #include "BLI_array.h" #include "BLI_blenlib.h" #include "BLI_bitmap.h" #include "BLI_math.h" #include "BLI_utildefines.h" #include "BLI_linklist.h" #include "BLI_task.h" #include "BKE_cdderivedmesh.h" #include "BKE_editmesh.h" #include "BKE_key.h" #include "BKE_library.h" #include "BKE_material.h" #include "BKE_modifier.h" #include "BKE_mesh.h" #include "BKE_mesh_mapping.h" #include "BKE_object.h" #include "BKE_object_deform.h" #include "BKE_paint.h" #include "BKE_texture.h" #include "BKE_multires.h" #include "BKE_bvhutils.h" #include "BKE_deform.h" #include "BKE_global.h" /* For debug flag, DM_update_tessface_data() func. */ #ifdef WITH_GAMEENGINE #include "BKE_navmesh_conversion.h" static DerivedMesh *navmesh_dm_createNavMeshForVisualization(DerivedMesh *dm); #endif #include "BLI_sys_types.h" /* for intptr_t support */ #include "GPU_buffers.h" #include "GPU_glew.h" #include "GPU_shader.h" #ifdef WITH_OPENSUBDIV # include "BKE_depsgraph.h" # include "DNA_userdef_types.h" #endif /* very slow! enable for testing only! */ //#define USE_MODIFIER_VALIDATE #ifdef USE_MODIFIER_VALIDATE # define ASSERT_IS_VALID_DM(dm) (BLI_assert((dm == NULL) || (DM_is_valid(dm) == true))) #else # define ASSERT_IS_VALID_DM(dm) #endif static ThreadMutex loops_cache_lock = BLI_MUTEX_INITIALIZER; static void add_shapekey_layers(DerivedMesh *dm, Mesh *me, Object *ob); static void shapekey_layers_to_keyblocks(DerivedMesh *dm, Mesh *me, int actshape_uid); /* -------------------------------------------------------------------- */ static MVert *dm_getVertArray(DerivedMesh *dm) { MVert *mvert = CustomData_get_layer(&dm->vertData, CD_MVERT); if (!mvert) { mvert = CustomData_add_layer(&dm->vertData, CD_MVERT, CD_CALLOC, NULL, dm->getNumVerts(dm)); CustomData_set_layer_flag(&dm->vertData, CD_MVERT, CD_FLAG_TEMPORARY); dm->copyVertArray(dm, mvert); } return mvert; } static MEdge *dm_getEdgeArray(DerivedMesh *dm) { MEdge *medge = CustomData_get_layer(&dm->edgeData, CD_MEDGE); if (!medge) { medge = CustomData_add_layer(&dm->edgeData, CD_MEDGE, CD_CALLOC, NULL, dm->getNumEdges(dm)); CustomData_set_layer_flag(&dm->edgeData, CD_MEDGE, CD_FLAG_TEMPORARY); dm->copyEdgeArray(dm, medge); } return medge; } static MFace *dm_getTessFaceArray(DerivedMesh *dm) { MFace *mface = CustomData_get_layer(&dm->faceData, CD_MFACE); if (!mface) { int numTessFaces = dm->getNumTessFaces(dm); if (!numTessFaces) { /* Do not add layer if there's no elements in it, this leads to issues later when * this layer is needed with non-zero size, but currently CD stuff does not check * for requested layer size on creation and just returns layer which was previously * added (sergey) */ return NULL; } mface = CustomData_add_layer(&dm->faceData, CD_MFACE, CD_CALLOC, NULL, numTessFaces); CustomData_set_layer_flag(&dm->faceData, CD_MFACE, CD_FLAG_TEMPORARY); dm->copyTessFaceArray(dm, mface); } return mface; } static MLoop *dm_getLoopArray(DerivedMesh *dm) { MLoop *mloop = CustomData_get_layer(&dm->loopData, CD_MLOOP); if (!mloop) { mloop = CustomData_add_layer(&dm->loopData, CD_MLOOP, CD_CALLOC, NULL, dm->getNumLoops(dm)); CustomData_set_layer_flag(&dm->loopData, CD_MLOOP, CD_FLAG_TEMPORARY); dm->copyLoopArray(dm, mloop); } return mloop; } static MPoly *dm_getPolyArray(DerivedMesh *dm) { MPoly *mpoly = CustomData_get_layer(&dm->polyData, CD_MPOLY); if (!mpoly) { mpoly = CustomData_add_layer(&dm->polyData, CD_MPOLY, CD_CALLOC, NULL, dm->getNumPolys(dm)); CustomData_set_layer_flag(&dm->polyData, CD_MPOLY, CD_FLAG_TEMPORARY); dm->copyPolyArray(dm, mpoly); } return mpoly; } static MVert *dm_dupVertArray(DerivedMesh *dm) { MVert *tmp = MEM_mallocN(sizeof(*tmp) * dm->getNumVerts(dm), "dm_dupVertArray tmp"); if (tmp) dm->copyVertArray(dm, tmp); return tmp; } static MEdge *dm_dupEdgeArray(DerivedMesh *dm) { MEdge *tmp = MEM_mallocN(sizeof(*tmp) * dm->getNumEdges(dm), "dm_dupEdgeArray tmp"); if (tmp) dm->copyEdgeArray(dm, tmp); return tmp; } static MFace *dm_dupFaceArray(DerivedMesh *dm) { MFace *tmp = MEM_mallocN(sizeof(*tmp) * dm->getNumTessFaces(dm), "dm_dupFaceArray tmp"); if (tmp) dm->copyTessFaceArray(dm, tmp); return tmp; } static MLoop *dm_dupLoopArray(DerivedMesh *dm) { MLoop *tmp = MEM_mallocN(sizeof(*tmp) * dm->getNumLoops(dm), "dm_dupLoopArray tmp"); if (tmp) dm->copyLoopArray(dm, tmp); return tmp; } static MPoly *dm_dupPolyArray(DerivedMesh *dm) { MPoly *tmp = MEM_mallocN(sizeof(*tmp) * dm->getNumPolys(dm), "dm_dupPolyArray tmp"); if (tmp) dm->copyPolyArray(dm, tmp); return tmp; } static int dm_getNumLoopTri(DerivedMesh *dm) { const int numlooptris = poly_to_tri_count(dm->getNumPolys(dm), dm->getNumLoops(dm)); BLI_assert(ELEM(dm->looptris.num, 0, numlooptris)); return numlooptris; } static const MLoopTri *dm_getLoopTriArray(DerivedMesh *dm) { if (dm->looptris.array) { BLI_assert(dm->getNumLoopTri(dm) == dm->looptris.num); } else { BLI_mutex_lock(&loops_cache_lock); /* We need to ensure array is still NULL inside mutex-protected code, some other thread might have already * recomputed those looptris. */ if (dm->looptris.array == NULL) { dm->recalcLoopTri(dm); } BLI_mutex_unlock(&loops_cache_lock); } return dm->looptris.array; } static CustomData *dm_getVertCData(DerivedMesh *dm) { return &dm->vertData; } static CustomData *dm_getEdgeCData(DerivedMesh *dm) { return &dm->edgeData; } static CustomData *dm_getTessFaceCData(DerivedMesh *dm) { return &dm->faceData; } static CustomData *dm_getLoopCData(DerivedMesh *dm) { return &dm->loopData; } static CustomData *dm_getPolyCData(DerivedMesh *dm) { return &dm->polyData; } /** * Utility function to initialize a DerivedMesh's function pointers to * the default implementation (for those functions which have a default) */ void DM_init_funcs(DerivedMesh *dm) { /* default function implementations */ dm->getVertArray = dm_getVertArray; dm->getEdgeArray = dm_getEdgeArray; dm->getTessFaceArray = dm_getTessFaceArray; dm->getLoopArray = dm_getLoopArray; dm->getPolyArray = dm_getPolyArray; dm->dupVertArray = dm_dupVertArray; dm->dupEdgeArray = dm_dupEdgeArray; dm->dupTessFaceArray = dm_dupFaceArray; dm->dupLoopArray = dm_dupLoopArray; dm->dupPolyArray = dm_dupPolyArray; dm->getLoopTriArray = dm_getLoopTriArray; /* subtypes handle getting actual data */ dm->getNumLoopTri = dm_getNumLoopTri; dm->getVertDataLayout = dm_getVertCData; dm->getEdgeDataLayout = dm_getEdgeCData; dm->getTessFaceDataLayout = dm_getTessFaceCData; dm->getLoopDataLayout = dm_getLoopCData; dm->getPolyDataLayout = dm_getPolyCData; dm->getVertData = DM_get_vert_data; dm->getEdgeData = DM_get_edge_data; dm->getTessFaceData = DM_get_tessface_data; dm->getPolyData = DM_get_poly_data; dm->getVertDataArray = DM_get_vert_data_layer; dm->getEdgeDataArray = DM_get_edge_data_layer; dm->getTessFaceDataArray = DM_get_tessface_data_layer; dm->getPolyDataArray = DM_get_poly_data_layer; dm->getLoopDataArray = DM_get_loop_data_layer; bvhcache_init(&dm->bvhCache); } /** * Utility function to initialize a DerivedMesh for the desired number * of vertices, edges and faces (doesn't allocate memory for them, just * sets up the custom data layers) */ void DM_init( DerivedMesh *dm, DerivedMeshType type, int numVerts, int numEdges, int numTessFaces, int numLoops, int numPolys) { dm->type = type; dm->numVertData = numVerts; dm->numEdgeData = numEdges; dm->numTessFaceData = numTessFaces; dm->numLoopData = numLoops; dm->numPolyData = numPolys; DM_init_funcs(dm); dm->needsFree = 1; dm->auto_bump_scale = -1.0f; dm->dirty = 0; /* don't use CustomData_reset(...); because we dont want to touch customdata */ copy_vn_i(dm->vertData.typemap, CD_NUMTYPES, -1); copy_vn_i(dm->edgeData.typemap, CD_NUMTYPES, -1); copy_vn_i(dm->faceData.typemap, CD_NUMTYPES, -1); copy_vn_i(dm->loopData.typemap, CD_NUMTYPES, -1); copy_vn_i(dm->polyData.typemap, CD_NUMTYPES, -1); } /** * Utility function to initialize a DerivedMesh for the desired number * of vertices, edges and faces, with a layer setup copied from source */ void DM_from_template_ex( DerivedMesh *dm, DerivedMesh *source, DerivedMeshType type, int numVerts, int numEdges, int numTessFaces, int numLoops, int numPolys, CustomDataMask mask) { CustomData_copy(&source->vertData, &dm->vertData, mask, CD_CALLOC, numVerts); CustomData_copy(&source->edgeData, &dm->edgeData, mask, CD_CALLOC, numEdges); CustomData_copy(&source->faceData, &dm->faceData, mask, CD_CALLOC, numTessFaces); CustomData_copy(&source->loopData, &dm->loopData, mask, CD_CALLOC, numLoops); CustomData_copy(&source->polyData, &dm->polyData, mask, CD_CALLOC, numPolys); dm->cd_flag = source->cd_flag; dm->type = type; dm->numVertData = numVerts; dm->numEdgeData = numEdges; dm->numTessFaceData = numTessFaces; dm->numLoopData = numLoops; dm->numPolyData = numPolys; DM_init_funcs(dm); dm->needsFree = 1; dm->dirty = 0; } void DM_from_template( DerivedMesh *dm, DerivedMesh *source, DerivedMeshType type, int numVerts, int numEdges, int numTessFaces, int numLoops, int numPolys) { DM_from_template_ex( dm, source, type, numVerts, numEdges, numTessFaces, numLoops, numPolys, CD_MASK_DERIVEDMESH); } int DM_release(DerivedMesh *dm) { if (dm->needsFree) { bvhcache_free(&dm->bvhCache); GPU_drawobject_free(dm); CustomData_free(&dm->vertData, dm->numVertData); CustomData_free(&dm->edgeData, dm->numEdgeData); CustomData_free(&dm->faceData, dm->numTessFaceData); CustomData_free(&dm->loopData, dm->numLoopData); CustomData_free(&dm->polyData, dm->numPolyData); if (dm->mat) { MEM_freeN(dm->mat); dm->mat = NULL; dm->totmat = 0; } MEM_SAFE_FREE(dm->looptris.array); dm->looptris.num = 0; dm->looptris.num_alloc = 0; return 1; } else { CustomData_free_temporary(&dm->vertData, dm->numVertData); CustomData_free_temporary(&dm->edgeData, dm->numEdgeData); CustomData_free_temporary(&dm->faceData, dm->numTessFaceData); CustomData_free_temporary(&dm->loopData, dm->numLoopData); CustomData_free_temporary(&dm->polyData, dm->numPolyData); return 0; } } void DM_DupPolys(DerivedMesh *source, DerivedMesh *target) { CustomData_free(&target->loopData, source->numLoopData); CustomData_free(&target->polyData, source->numPolyData); CustomData_copy(&source->loopData, &target->loopData, CD_MASK_DERIVEDMESH, CD_DUPLICATE, source->numLoopData); CustomData_copy(&source->polyData, &target->polyData, CD_MASK_DERIVEDMESH, CD_DUPLICATE, source->numPolyData); target->numLoopData = source->numLoopData; target->numPolyData = source->numPolyData; if (!CustomData_has_layer(&target->polyData, CD_MPOLY)) { MPoly *mpoly; MLoop *mloop; mloop = source->dupLoopArray(source); mpoly = source->dupPolyArray(source); CustomData_add_layer(&target->loopData, CD_MLOOP, CD_ASSIGN, mloop, source->numLoopData); CustomData_add_layer(&target->polyData, CD_MPOLY, CD_ASSIGN, mpoly, source->numPolyData); } } void DM_ensure_normals(DerivedMesh *dm) { if (dm->dirty & DM_DIRTY_NORMALS) { dm->calcNormals(dm); } BLI_assert((dm->dirty & DM_DIRTY_NORMALS) == 0); } static void DM_calc_loop_normals(DerivedMesh *dm, const bool use_split_normals, float split_angle) { dm->calcLoopNormals(dm, use_split_normals, split_angle); dm->dirty |= DM_DIRTY_TESS_CDLAYERS; } /* note: until all modifiers can take MPoly's as input, * use this at the start of modifiers */ void DM_ensure_tessface(DerivedMesh *dm) { const int numTessFaces = dm->getNumTessFaces(dm); const int numPolys = dm->getNumPolys(dm); if ((numTessFaces == 0) && (numPolys != 0)) { dm->recalcTessellation(dm); if (dm->getNumTessFaces(dm) != 0) { /* printf("info %s: polys -> ngons calculated\n", __func__); */ } else { printf("warning %s: could not create tessfaces from %d polygons, dm->type=%u\n", __func__, numPolys, dm->type); } } else if (dm->dirty & DM_DIRTY_TESS_CDLAYERS) { BLI_assert(CustomData_has_layer(&dm->faceData, CD_ORIGINDEX) || numTessFaces == 0); DM_update_tessface_data(dm); } dm->dirty &= ~DM_DIRTY_TESS_CDLAYERS; } /** * Ensure the array is large enough */ void DM_ensure_looptri_data(DerivedMesh *dm) { const unsigned int totpoly = dm->numPolyData; const unsigned int totloop = dm->numLoopData; const int looptris_num = poly_to_tri_count(totpoly, totloop); if ((looptris_num > dm->looptris.num_alloc) || (looptris_num < dm->looptris.num_alloc * 2) || (totpoly == 0)) { MEM_SAFE_FREE(dm->looptris.array); dm->looptris.num_alloc = 0; dm->looptris.num = 0; } if (totpoly) { if (dm->looptris.array == NULL) { dm->looptris.array = MEM_mallocN(sizeof(*dm->looptris.array) * looptris_num, __func__); dm->looptris.num_alloc = looptris_num; } dm->looptris.num = looptris_num; } } void DM_verttri_from_looptri(MVertTri *verttri, const MLoop *mloop, const MLoopTri *looptri, int looptri_num) { int i; for (i = 0; i < looptri_num; i++) { verttri[i].tri[0] = mloop[looptri[i].tri[0]].v; verttri[i].tri[1] = mloop[looptri[i].tri[1]].v; verttri[i].tri[2] = mloop[looptri[i].tri[2]].v; } } /* Update tessface CD data from loop/poly ones. Needed when not retessellating after modstack evaluation. */ /* NOTE: Assumes dm has valid tessellated data! */ void DM_update_tessface_data(DerivedMesh *dm) { MFace *mf, *mface = dm->getTessFaceArray(dm); MPoly *mp = dm->getPolyArray(dm); MLoop *ml = dm->getLoopArray(dm); CustomData *fdata = dm->getTessFaceDataLayout(dm); CustomData *pdata = dm->getPolyDataLayout(dm); CustomData *ldata = dm->getLoopDataLayout(dm); const int totface = dm->getNumTessFaces(dm); int mf_idx; int *polyindex = CustomData_get_layer(fdata, CD_ORIGINDEX); unsigned int (*loopindex)[4]; /* Should never occure, but better abort than segfault! */ if (!polyindex) return; CustomData_from_bmeshpoly(fdata, pdata, ldata, totface); if (CustomData_has_layer(fdata, CD_MTFACE) || CustomData_has_layer(fdata, CD_MCOL) || CustomData_has_layer(fdata, CD_PREVIEW_MCOL) || CustomData_has_layer(fdata, CD_ORIGSPACE) || CustomData_has_layer(fdata, CD_TESSLOOPNORMAL) || CustomData_has_layer(fdata, CD_TANGENT)) { loopindex = MEM_mallocN(sizeof(*loopindex) * totface, __func__); for (mf_idx = 0, mf = mface; mf_idx < totface; mf_idx++, mf++) { const int mf_len = mf->v4 ? 4 : 3; unsigned int *ml_idx = loopindex[mf_idx]; int i, not_done; /* Find out loop indices. */ /* NOTE: This assumes tessface are valid and in sync with loop/poly... Else, most likely, segfault! */ for (i = mp[polyindex[mf_idx]].loopstart, not_done = mf_len; not_done; i++) { const int tf_v = BKE_MESH_TESSFACE_VINDEX_ORDER(mf, ml[i].v); if (tf_v != -1) { ml_idx[tf_v] = i; not_done--; } } } /* NOTE: quad detection issue - fourth vertidx vs fourth loopidx: * Here, our tfaces' fourth vertex index is never 0 for a quad. However, we know our fourth loop index may be * 0 for quads (because our quads may have been rotated compared to their org poly, see tessellation code). * So we pass the MFace's, and BKE_mesh_loops_to_tessdata will use MFace->v4 index as quad test. */ BKE_mesh_loops_to_tessdata(fdata, ldata, pdata, mface, polyindex, loopindex, totface); MEM_freeN(loopindex); } if (G.debug & G_DEBUG) printf("%s: Updated tessellated customdata of dm %p\n", __func__, dm); dm->dirty &= ~DM_DIRTY_TESS_CDLAYERS; } void DM_generate_tangent_tessface_data(DerivedMesh *dm, bool generate) { MFace *mf, *mface = dm->getTessFaceArray(dm); MPoly *mp = dm->getPolyArray(dm); MLoop *ml = dm->getLoopArray(dm); CustomData *fdata = dm->getTessFaceDataLayout(dm); CustomData *pdata = dm->getPolyDataLayout(dm); CustomData *ldata = dm->getLoopDataLayout(dm); const int totface = dm->getNumTessFaces(dm); int mf_idx; int *polyindex = CustomData_get_layer(fdata, CD_ORIGINDEX); unsigned int (*loopindex)[4] = NULL; /* Should never occure, but better abort than segfault! */ if (!polyindex) return; if (generate) { for (int j = 0; j < ldata->totlayer; j++) { if (ldata->layers[j].type == CD_TANGENT) { CustomData_add_layer_named(fdata, CD_TANGENT, CD_CALLOC, NULL, totface, ldata->layers[j].name); CustomData_bmesh_update_active_layers(fdata, pdata, ldata); if (!loopindex) { loopindex = MEM_mallocN(sizeof(*loopindex) * totface, __func__); for (mf_idx = 0, mf = mface; mf_idx < totface; mf_idx++, mf++) { const int mf_len = mf->v4 ? 4 : 3; unsigned int *ml_idx = loopindex[mf_idx]; /* Find out loop indices. */ /* NOTE: This assumes tessface are valid and in sync with loop/poly... Else, most likely, segfault! */ for (int i = mp[polyindex[mf_idx]].loopstart, not_done = mf_len; not_done; i++) { const int tf_v = BKE_MESH_TESSFACE_VINDEX_ORDER(mf, ml[i].v); if (tf_v != -1) { ml_idx[tf_v] = i; not_done--; } } } } /* NOTE: quad detection issue - fourth vertidx vs fourth loopidx: * Here, our tfaces' fourth vertex index is never 0 for a quad. However, we know our fourth loop index may be * 0 for quads (because our quads may have been rotated compared to their org poly, see tessellation code). * So we pass the MFace's, and BKE_mesh_loops_to_tessdata will use MFace->v4 index as quad test. */ BKE_mesh_tangent_loops_to_tessdata(fdata, ldata, mface, polyindex, loopindex, totface, ldata->layers[j].name); } } if (loopindex) MEM_freeN(loopindex); BLI_assert(CustomData_from_bmeshpoly_test(fdata, pdata, ldata, true)); } if (G.debug & G_DEBUG) printf("%s: Updated tessellated tangents of dm %p\n", __func__, dm); } void DM_update_materials(DerivedMesh *dm, Object *ob) { int i, totmat = ob->totcol + 1; /* materials start from 1, default material is 0 */ if (dm->totmat != totmat) { dm->totmat = totmat; /* invalidate old materials */ if (dm->mat) MEM_freeN(dm->mat); dm->mat = MEM_mallocN(totmat * sizeof(*dm->mat), "DerivedMesh.mat"); } /* we leave last material as empty - rationale here is being able to index * the materials by using the mf->mat_nr directly and leaving the last * material as NULL in case no materials exist on mesh, so indexing will not fail */ for (i = 0; i < totmat - 1; i++) { dm->mat[i] = give_current_material(ob, i + 1); } dm->mat[i] = NULL; } MLoopUV *DM_paint_uvlayer_active_get(DerivedMesh *dm, int mat_nr) { MLoopUV *uv_base; BLI_assert(mat_nr < dm->totmat); if (dm->mat[mat_nr] && dm->mat[mat_nr]->texpaintslot && dm->mat[mat_nr]->texpaintslot[dm->mat[mat_nr]->paint_active_slot].uvname) { uv_base = CustomData_get_layer_named(&dm->loopData, CD_MLOOPUV, dm->mat[mat_nr]->texpaintslot[dm->mat[mat_nr]->paint_active_slot].uvname); /* This can fail if we have changed the name in the UV layer list and have assigned the old name in the material * texture slot.*/ if (!uv_base) uv_base = CustomData_get_layer(&dm->loopData, CD_MLOOPUV); } else { uv_base = CustomData_get_layer(&dm->loopData, CD_MLOOPUV); } return uv_base; } void DM_to_mesh(DerivedMesh *dm, Mesh *me, Object *ob, CustomDataMask mask, bool take_ownership) { /* dm might depend on me, so we need to do everything with a local copy */ Mesh tmp = *me; int totvert, totedge /*, totface */ /* UNUSED */, totloop, totpoly; int did_shapekeys = 0; int alloctype = CD_DUPLICATE; if (take_ownership && dm->type == DM_TYPE_CDDM && dm->needsFree) { bool has_any_referenced_layers = CustomData_has_referenced(&dm->vertData) || CustomData_has_referenced(&dm->edgeData) || CustomData_has_referenced(&dm->loopData) || CustomData_has_referenced(&dm->faceData) || CustomData_has_referenced(&dm->polyData); if (!has_any_referenced_layers) { alloctype = CD_ASSIGN; } } CustomData_reset(&tmp.vdata); CustomData_reset(&tmp.edata); CustomData_reset(&tmp.fdata); CustomData_reset(&tmp.ldata); CustomData_reset(&tmp.pdata); DM_ensure_normals(dm); totvert = tmp.totvert = dm->getNumVerts(dm); totedge = tmp.totedge = dm->getNumEdges(dm); totloop = tmp.totloop = dm->getNumLoops(dm); totpoly = tmp.totpoly = dm->getNumPolys(dm); tmp.totface = 0; CustomData_copy(&dm->vertData, &tmp.vdata, mask, alloctype, totvert); CustomData_copy(&dm->edgeData, &tmp.edata, mask, alloctype, totedge); CustomData_copy(&dm->loopData, &tmp.ldata, mask, alloctype, totloop); CustomData_copy(&dm->polyData, &tmp.pdata, mask, alloctype, totpoly); tmp.cd_flag = dm->cd_flag; if (CustomData_has_layer(&dm->vertData, CD_SHAPEKEY)) { KeyBlock *kb; int uid; if (ob) { kb = BLI_findlink(&me->key->block, ob->shapenr - 1); if (kb) { uid = kb->uid; } else { printf("%s: error - could not find active shapekey %d!\n", __func__, ob->shapenr - 1); uid = INT_MAX; } } else { /* if no object, set to INT_MAX so we don't mess up any shapekey layers */ uid = INT_MAX; } shapekey_layers_to_keyblocks(dm, me, uid); did_shapekeys = 1; } /* copy texture space */ if (ob) { BKE_mesh_texspace_copy_from_object(&tmp, ob); } /* not all DerivedMeshes store their verts/edges/faces in CustomData, so * we set them here in case they are missing */ if (!CustomData_has_layer(&tmp.vdata, CD_MVERT)) { CustomData_add_layer(&tmp.vdata, CD_MVERT, CD_ASSIGN, (alloctype == CD_ASSIGN) ? dm->getVertArray(dm) : dm->dupVertArray(dm), totvert); } if (!CustomData_has_layer(&tmp.edata, CD_MEDGE)) { CustomData_add_layer(&tmp.edata, CD_MEDGE, CD_ASSIGN, (alloctype == CD_ASSIGN) ? dm->getEdgeArray(dm) : dm->dupEdgeArray(dm), totedge); } if (!CustomData_has_layer(&tmp.pdata, CD_MPOLY)) { tmp.mloop = (alloctype == CD_ASSIGN) ? dm->getLoopArray(dm) : dm->dupLoopArray(dm); tmp.mpoly = (alloctype == CD_ASSIGN) ? dm->getPolyArray(dm) : dm->dupPolyArray(dm); CustomData_add_layer(&tmp.ldata, CD_MLOOP, CD_ASSIGN, tmp.mloop, tmp.totloop); CustomData_add_layer(&tmp.pdata, CD_MPOLY, CD_ASSIGN, tmp.mpoly, tmp.totpoly); } /* object had got displacement layer, should copy this layer to save sculpted data */ /* NOTE: maybe some other layers should be copied? nazgul */ if (CustomData_has_layer(&me->ldata, CD_MDISPS)) { if (totloop == me->totloop) { MDisps *mdisps = CustomData_get_layer(&me->ldata, CD_MDISPS); CustomData_add_layer(&tmp.ldata, CD_MDISPS, alloctype, mdisps, totloop); } } /* yes, must be before _and_ after tessellate */ BKE_mesh_update_customdata_pointers(&tmp, false); /* since 2.65 caller must do! */ // BKE_mesh_tessface_calc(&tmp); CustomData_free(&me->vdata, me->totvert); CustomData_free(&me->edata, me->totedge); CustomData_free(&me->fdata, me->totface); CustomData_free(&me->ldata, me->totloop); CustomData_free(&me->pdata, me->totpoly); /* ok, this should now use new CD shapekey data, * which should be fed through the modifier * stack */ if (tmp.totvert != me->totvert && !did_shapekeys && me->key) { printf("%s: YEEK! this should be recoded! Shape key loss!: ID '%s'\n", __func__, tmp.id.name); if (tmp.key) id_us_min(&tmp.key->id); tmp.key = NULL; } /* Clear selection history */ MEM_SAFE_FREE(tmp.mselect); tmp.totselect = 0; BLI_assert(ELEM(tmp.bb, NULL, me->bb)); if (me->bb) { MEM_freeN(me->bb); tmp.bb = NULL; } /* skip the listbase */ MEMCPY_STRUCT_OFS(me, &tmp, id.prev); if (take_ownership) { if (alloctype == CD_ASSIGN) { CustomData_free_typemask(&dm->vertData, dm->numVertData, ~mask); CustomData_free_typemask(&dm->edgeData, dm->numEdgeData, ~mask); CustomData_free_typemask(&dm->loopData, dm->numLoopData, ~mask); CustomData_free_typemask(&dm->polyData, dm->numPolyData, ~mask); } dm->release(dm); } } void DM_to_meshkey(DerivedMesh *dm, Mesh *me, KeyBlock *kb) { int a, totvert = dm->getNumVerts(dm); float *fp; MVert *mvert; if (totvert == 0 || me->totvert == 0 || me->totvert != totvert) { return; } if (kb->data) MEM_freeN(kb->data); kb->data = MEM_mallocN(me->key->elemsize * me->totvert, "kb->data"); kb->totelem = totvert; fp = kb->data; mvert = dm->getVertDataArray(dm, CD_MVERT); for (a = 0; a < kb->totelem; a++, fp += 3, mvert++) { copy_v3_v3(fp, mvert->co); } } /** * set the CD_FLAG_NOCOPY flag in custom data layers where the mask is * zero for the layer type, so only layer types specified by the mask * will be copied */ void DM_set_only_copy(DerivedMesh *dm, CustomDataMask mask) { CustomData_set_only_copy(&dm->vertData, mask); CustomData_set_only_copy(&dm->edgeData, mask); CustomData_set_only_copy(&dm->faceData, mask); /* this wasn't in 2.63 and is disabled for 2.64 because it gives problems with * weight paint mode when there are modifiers applied, needs further investigation, * see replies to r50969, Campbell */ #if 0 CustomData_set_only_copy(&dm->loopData, mask); CustomData_set_only_copy(&dm->polyData, mask); #endif } void DM_add_vert_layer(DerivedMesh *dm, int type, int alloctype, void *layer) { CustomData_add_layer(&dm->vertData, type, alloctype, layer, dm->numVertData); } void DM_add_edge_layer(DerivedMesh *dm, int type, int alloctype, void *layer) { CustomData_add_layer(&dm->edgeData, type, alloctype, layer, dm->numEdgeData); } void DM_add_tessface_layer(DerivedMesh *dm, int type, int alloctype, void *layer) { CustomData_add_layer(&dm->faceData, type, alloctype, layer, dm->numTessFaceData); } void DM_add_loop_layer(DerivedMesh *dm, int type, int alloctype, void *layer) { CustomData_add_layer(&dm->loopData, type, alloctype, layer, dm->numLoopData); } void DM_add_poly_layer(DerivedMesh *dm, int type, int alloctype, void *layer) { CustomData_add_layer(&dm->polyData, type, alloctype, layer, dm->numPolyData); } void *DM_get_vert_data(DerivedMesh *dm, int index, int type) { BLI_assert(index >= 0 && index < dm->getNumVerts(dm)); return CustomData_get(&dm->vertData, index, type); } void *DM_get_edge_data(DerivedMesh *dm, int index, int type) { BLI_assert(index >= 0 && index < dm->getNumEdges(dm)); return CustomData_get(&dm->edgeData, index, type); } void *DM_get_tessface_data(DerivedMesh *dm, int index, int type) { BLI_assert(index >= 0 && index < dm->getNumTessFaces(dm)); return CustomData_get(&dm->faceData, index, type); } void *DM_get_poly_data(DerivedMesh *dm, int index, int type) { BLI_assert(index >= 0 && index < dm->getNumPolys(dm)); return CustomData_get(&dm->polyData, index, type); } void *DM_get_vert_data_layer(DerivedMesh *dm, int type) { if (type == CD_MVERT) return dm->getVertArray(dm); return CustomData_get_layer(&dm->vertData, type); } void *DM_get_edge_data_layer(DerivedMesh *dm, int type) { if (type == CD_MEDGE) return dm->getEdgeArray(dm); return CustomData_get_layer(&dm->edgeData, type); } void *DM_get_tessface_data_layer(DerivedMesh *dm, int type) { if (type == CD_MFACE) return dm->getTessFaceArray(dm); return CustomData_get_layer(&dm->faceData, type); } void *DM_get_poly_data_layer(DerivedMesh *dm, int type) { return CustomData_get_layer(&dm->polyData, type); } void *DM_get_loop_data_layer(DerivedMesh *dm, int type) { return CustomData_get_layer(&dm->loopData, type); } void DM_set_vert_data(DerivedMesh *dm, int index, int type, void *data) { CustomData_set(&dm->vertData, index, type, data); } void DM_set_edge_data(DerivedMesh *dm, int index, int type, void *data) { CustomData_set(&dm->edgeData, index, type, data); } void DM_set_tessface_data(DerivedMesh *dm, int index, int type, void *data) { CustomData_set(&dm->faceData, index, type, data); } void DM_copy_vert_data(DerivedMesh *source, DerivedMesh *dest, int source_index, int dest_index, int count) { CustomData_copy_data(&source->vertData, &dest->vertData, source_index, dest_index, count); } void DM_copy_edge_data(DerivedMesh *source, DerivedMesh *dest, int source_index, int dest_index, int count) { CustomData_copy_data(&source->edgeData, &dest->edgeData, source_index, dest_index, count); } void DM_copy_tessface_data(DerivedMesh *source, DerivedMesh *dest, int source_index, int dest_index, int count) { CustomData_copy_data(&source->faceData, &dest->faceData, source_index, dest_index, count); } void DM_copy_loop_data(DerivedMesh *source, DerivedMesh *dest, int source_index, int dest_index, int count) { CustomData_copy_data(&source->loopData, &dest->loopData, source_index, dest_index, count); } void DM_copy_poly_data(DerivedMesh *source, DerivedMesh *dest, int source_index, int dest_index, int count) { CustomData_copy_data(&source->polyData, &dest->polyData, source_index, dest_index, count); } void DM_free_vert_data(struct DerivedMesh *dm, int index, int count) { CustomData_free_elem(&dm->vertData, index, count); } void DM_free_edge_data(struct DerivedMesh *dm, int index, int count) { CustomData_free_elem(&dm->edgeData, index, count); } void DM_free_tessface_data(struct DerivedMesh *dm, int index, int count) { CustomData_free_elem(&dm->faceData, index, count); } void DM_free_loop_data(struct DerivedMesh *dm, int index, int count) { CustomData_free_elem(&dm->loopData, index, count); } void DM_free_poly_data(struct DerivedMesh *dm, int index, int count) { CustomData_free_elem(&dm->polyData, index, count); } /** * interpolates vertex data from the vertices indexed by src_indices in the * source mesh using the given weights and stores the result in the vertex * indexed by dest_index in the dest mesh */ void DM_interp_vert_data( DerivedMesh *source, DerivedMesh *dest, int *src_indices, float *weights, int count, int dest_index) { CustomData_interp(&source->vertData, &dest->vertData, src_indices, weights, NULL, count, dest_index); } /** * interpolates edge data from the edges indexed by src_indices in the * source mesh using the given weights and stores the result in the edge indexed * by dest_index in the dest mesh. * if weights is NULL, all weights default to 1. * if vert_weights is non-NULL, any per-vertex edge data is interpolated using * vert_weights[i] multiplied by weights[i]. */ void DM_interp_edge_data( DerivedMesh *source, DerivedMesh *dest, int *src_indices, float *weights, EdgeVertWeight *vert_weights, int count, int dest_index) { CustomData_interp(&source->edgeData, &dest->edgeData, src_indices, weights, (float *)vert_weights, count, dest_index); } /** * interpolates face data from the faces indexed by src_indices in the * source mesh using the given weights and stores the result in the face indexed * by dest_index in the dest mesh. * if weights is NULL, all weights default to 1. * if vert_weights is non-NULL, any per-vertex face data is interpolated using * vert_weights[i] multiplied by weights[i]. */ void DM_interp_tessface_data( DerivedMesh *source, DerivedMesh *dest, int *src_indices, float *weights, FaceVertWeight *vert_weights, int count, int dest_index) { CustomData_interp(&source->faceData, &dest->faceData, src_indices, weights, (float *)vert_weights, count, dest_index); } void DM_swap_tessface_data(DerivedMesh *dm, int index, const int *corner_indices) { CustomData_swap_corners(&dm->faceData, index, corner_indices); } void DM_interp_loop_data( DerivedMesh *source, DerivedMesh *dest, int *src_indices, float *weights, int count, int dest_index) { CustomData_interp(&source->loopData, &dest->loopData, src_indices, weights, NULL, count, dest_index); } void DM_interp_poly_data( DerivedMesh *source, DerivedMesh *dest, int *src_indices, float *weights, int count, int dest_index) { CustomData_interp(&source->polyData, &dest->polyData, src_indices, weights, NULL, count, dest_index); } DerivedMesh *mesh_create_derived(Mesh *me, float (*vertCos)[3]) { DerivedMesh *dm = CDDM_from_mesh(me); if (!dm) return NULL; if (vertCos) { CDDM_apply_vert_coords(dm, vertCos); } return dm; } DerivedMesh *mesh_create_derived_for_modifier( Scene *scene, Object *ob, ModifierData *md, int build_shapekey_layers) { Mesh *me = ob->data; const ModifierTypeInfo *mti = modifierType_getInfo(md->type); DerivedMesh *dm; KeyBlock *kb; md->scene = scene; if (!(md->mode & eModifierMode_Realtime)) { return NULL; } if (mti->isDisabled && mti->isDisabled(md, 0)) { return NULL; } if (build_shapekey_layers && me->key && (kb = BLI_findlink(&me->key->block, ob->shapenr - 1))) { BKE_keyblock_convert_to_mesh(kb, me); } if (mti->type == eModifierTypeType_OnlyDeform) { int numVerts; float (*deformedVerts)[3] = BKE_mesh_vertexCos_get(me, &numVerts); modwrap_deformVerts(md, ob, NULL, deformedVerts, numVerts, 0); dm = mesh_create_derived(me, deformedVerts); if (build_shapekey_layers) add_shapekey_layers(dm, me, ob); MEM_freeN(deformedVerts); } else { DerivedMesh *tdm = mesh_create_derived(me, NULL); if (build_shapekey_layers) add_shapekey_layers(tdm, me, ob); dm = modwrap_applyModifier(md, ob, tdm, 0); ASSERT_IS_VALID_DM(dm); if (tdm != dm) tdm->release(tdm); } return dm; } static float (*get_editbmesh_orco_verts(BMEditMesh *em))[3] { BMIter iter; BMVert *eve; float (*orco)[3]; int i; /* these may not really be the orco's, but it's only for preview. * could be solver better once, but isn't simple */ orco = MEM_mallocN(sizeof(float) * 3 * em->bm->totvert, "BMEditMesh Orco"); BM_ITER_MESH_INDEX (eve, &iter, em->bm, BM_VERTS_OF_MESH, i) { copy_v3_v3(orco[i], eve->co); } return orco; } /* orco custom data layer */ static float (*get_orco_coords_dm(Object *ob, BMEditMesh *em, int layer, int *free))[3] { *free = 0; if (layer == CD_ORCO) { /* get original coordinates */ *free = 1; if (em) return get_editbmesh_orco_verts(em); else return BKE_mesh_orco_verts_get(ob); } else if (layer == CD_CLOTH_ORCO) { /* apply shape key for cloth, this should really be solved * by a more flexible customdata system, but not simple */ if (!em) { ClothModifierData *clmd = (ClothModifierData *)modifiers_findByType(ob, eModifierType_Cloth); KeyBlock *kb = BKE_keyblock_from_key(BKE_key_from_object(ob), clmd->sim_parms->shapekey_rest); if (kb && kb->data) { return kb->data; } } return NULL; } return NULL; } static DerivedMesh *create_orco_dm(Object *ob, Mesh *me, BMEditMesh *em, int layer) { DerivedMesh *dm; float (*orco)[3]; int free; if (em) { dm = CDDM_from_editbmesh(em, false, false); } else { dm = CDDM_from_mesh(me); } orco = get_orco_coords_dm(ob, em, layer, &free); if (orco) { CDDM_apply_vert_coords(dm, orco); if (free) MEM_freeN(orco); } return dm; } static void add_orco_dm( Object *ob, BMEditMesh *em, DerivedMesh *dm, DerivedMesh *orcodm, int layer) { float (*orco)[3], (*layerorco)[3]; int totvert, free; totvert = dm->getNumVerts(dm); if (orcodm) { orco = MEM_callocN(sizeof(float[3]) * totvert, "dm orco"); free = 1; if (orcodm->getNumVerts(orcodm) == totvert) orcodm->getVertCos(orcodm, orco); else dm->getVertCos(dm, orco); } else orco = get_orco_coords_dm(ob, em, layer, &free); if (orco) { if (layer == CD_ORCO) BKE_mesh_orco_verts_transform(ob->data, orco, totvert, 0); if (!(layerorco = DM_get_vert_data_layer(dm, layer))) { DM_add_vert_layer(dm, layer, CD_CALLOC, NULL); layerorco = DM_get_vert_data_layer(dm, layer); } memcpy(layerorco, orco, sizeof(float) * 3 * totvert); if (free) MEM_freeN(orco); } } /* weight paint colors */ /* Something of a hack, at the moment deal with weightpaint * by tucking into colors during modifier eval, only in * wpaint mode. Works ok but need to make sure recalc * happens on enter/exit wpaint. */ void weight_to_rgb(float r_rgb[3], const float weight) { const float blend = ((weight / 2.0f) + 0.5f); if (weight <= 0.25f) { /* blue->cyan */ r_rgb[0] = 0.0f; r_rgb[1] = blend * weight * 4.0f; r_rgb[2] = blend; } else if (weight <= 0.50f) { /* cyan->green */ r_rgb[0] = 0.0f; r_rgb[1] = blend; r_rgb[2] = blend * (1.0f - ((weight - 0.25f) * 4.0f)); } else if (weight <= 0.75f) { /* green->yellow */ r_rgb[0] = blend * ((weight - 0.50f) * 4.0f); r_rgb[1] = blend; r_rgb[2] = 0.0f; } else if (weight <= 1.0f) { /* yellow->red */ r_rgb[0] = blend; r_rgb[1] = blend * (1.0f - ((weight - 0.75f) * 4.0f)); r_rgb[2] = 0.0f; } else { /* exceptional value, unclamped or nan, * avoid uninitialized memory use */ r_rgb[0] = 1.0f; r_rgb[1] = 0.0f; r_rgb[2] = 1.0f; } } /* draw_flag's for calc_weightpaint_vert_color */ enum { /* only one of these should be set, keep first (for easy bit-shifting) */ CALC_WP_GROUP_USER_ACTIVE = (1 << 1), CALC_WP_GROUP_USER_ALL = (1 << 2), CALC_WP_MULTIPAINT = (1 << 3), CALC_WP_AUTO_NORMALIZE = (1 << 4), CALC_WP_MIRROR_X = (1 << 5), }; typedef struct DMWeightColorInfo { const ColorBand *coba; const char *alert_color; } DMWeightColorInfo; static int dm_drawflag_calc(const ToolSettings *ts, const Mesh *me) { return ((ts->multipaint ? CALC_WP_MULTIPAINT : 0) | /* CALC_WP_GROUP_USER_ACTIVE or CALC_WP_GROUP_USER_ALL */ (1 << ts->weightuser) | (ts->auto_normalize ? CALC_WP_AUTO_NORMALIZE : 0) | ((me->editflag & ME_EDIT_MIRROR_X) ? CALC_WP_MIRROR_X : 0)); } static void weightpaint_color(unsigned char r_col[4], DMWeightColorInfo *dm_wcinfo, const float input) { float colf[4]; if (dm_wcinfo && dm_wcinfo->coba) { do_colorband(dm_wcinfo->coba, input, colf); } else { weight_to_rgb(colf, input); } /* don't use rgb_float_to_uchar() here because * the resulting float doesn't need 0-1 clamp check */ r_col[0] = (unsigned char)(colf[0] * 255.0f); r_col[1] = (unsigned char)(colf[1] * 255.0f); r_col[2] = (unsigned char)(colf[2] * 255.0f); r_col[3] = 255; } static void calc_weightpaint_vert_color( unsigned char r_col[4], const MDeformVert *dv, DMWeightColorInfo *dm_wcinfo, const int defbase_tot, const int defbase_act, const bool *defbase_sel, const int defbase_sel_tot, const int draw_flag) { float input = 0.0f; bool show_alert_color = false; if ((defbase_sel_tot > 1) && (draw_flag & CALC_WP_MULTIPAINT)) { /* Multi-Paint feature */ input = BKE_defvert_multipaint_collective_weight( dv, defbase_tot, defbase_sel, defbase_sel_tot, (draw_flag & CALC_WP_AUTO_NORMALIZE) != 0); /* make it black if the selected groups have no weight on a vertex */ if (input == 0.0f) { show_alert_color = true; } } else { /* default, non tricky behavior */ input = defvert_find_weight(dv, defbase_act); if (draw_flag & CALC_WP_GROUP_USER_ACTIVE) { if (input == 0.0f) { show_alert_color = true; } } else if (draw_flag & CALC_WP_GROUP_USER_ALL) { if (input == 0.0f) { show_alert_color = defvert_is_weight_zero(dv, defbase_tot); } } } if (show_alert_color == false) { CLAMP(input, 0.0f, 1.0f); weightpaint_color(r_col, dm_wcinfo, input); } else { copy_v3_v3_char((char *)r_col, dm_wcinfo->alert_color); r_col[3] = 255; } } static DMWeightColorInfo G_dm_wcinfo; void vDM_ColorBand_store(const ColorBand *coba, const char alert_color[4]) { G_dm_wcinfo.coba = coba; G_dm_wcinfo.alert_color = alert_color; } /** * return an array of vertex weight colors, caller must free. * * \note that we could save some memory and allocate RGB only but then we'd need to * re-arrange the colors when copying to the face since MCol has odd ordering, * so leave this as is - campbell */ static void calc_weightpaint_vert_array( Object *ob, DerivedMesh *dm, int const draw_flag, DMWeightColorInfo *dm_wcinfo, unsigned char (*r_wtcol_v)[4]) { BMEditMesh *em = (dm->type == DM_TYPE_EDITBMESH) ? BKE_editmesh_from_object(ob) : NULL; const int numVerts = dm->getNumVerts(dm); if ((ob->actdef != 0) && (CustomData_has_layer(em ? &em->bm->vdata : &dm->vertData, CD_MDEFORMVERT))) { unsigned char (*wc)[4] = r_wtcol_v; unsigned int i; /* variables for multipaint */ const int defbase_tot = BLI_listbase_count(&ob->defbase); const int defbase_act = ob->actdef - 1; int defbase_sel_tot = 0; bool *defbase_sel = NULL; if (draw_flag & CALC_WP_MULTIPAINT) { defbase_sel = BKE_object_defgroup_selected_get(ob, defbase_tot, &defbase_sel_tot); if (defbase_sel_tot > 1 && (draw_flag & CALC_WP_MIRROR_X)) { BKE_object_defgroup_mirror_selection(ob, defbase_tot, defbase_sel, defbase_sel, &defbase_sel_tot); } } /* editmesh won't have deform verts unless modifiers require it, * avoid having to create an array of deform-verts only for drawing * by reading from the bmesh directly. */ if (em) { BMIter iter; BMVert *eve; const int cd_dvert_offset = CustomData_get_offset(&em->bm->vdata, CD_MDEFORMVERT); BLI_assert(cd_dvert_offset != -1); BM_ITER_MESH_INDEX (eve, &iter, em->bm, BM_VERTS_OF_MESH, i) { const MDeformVert *dv = BM_ELEM_CD_GET_VOID_P(eve, cd_dvert_offset); calc_weightpaint_vert_color( (unsigned char *)wc, dv, dm_wcinfo, defbase_tot, defbase_act, defbase_sel, defbase_sel_tot, draw_flag); wc++; } } else { const MDeformVert *dv = DM_get_vert_data_layer(dm, CD_MDEFORMVERT); for (i = numVerts; i != 0; i--, wc++, dv++) { calc_weightpaint_vert_color( (unsigned char *)wc, dv, dm_wcinfo, defbase_tot, defbase_act, defbase_sel, defbase_sel_tot, draw_flag); } } if (defbase_sel) { MEM_freeN(defbase_sel); } } else { unsigned char col[4]; if ((ob->actdef == 0) && !BLI_listbase_is_empty(&ob->defbase)) { /* color-code for missing data (full brightness isn't easy on the eye). */ ARRAY_SET_ITEMS(col, 0xa0, 0, 0xa0, 0xff); } else if (draw_flag & (CALC_WP_GROUP_USER_ACTIVE | CALC_WP_GROUP_USER_ALL)) { copy_v3_v3_char((char *)col, dm_wcinfo->alert_color); col[3] = 255; } else { weightpaint_color(col, dm_wcinfo, 0.0f); } copy_vn_i((int *)r_wtcol_v, numVerts, *((int *)col)); } } /** return an array of vertex weight colors from given weights, caller must free. * * \note that we could save some memory and allocate RGB only but then we'd need to * re-arrange the colors when copying to the face since MCol has odd ordering, * so leave this as is - campbell */ static void calc_colors_from_weights_array( const int num, const float *weights, unsigned char (*r_wtcol_v)[4]) { unsigned char (*wc)[4] = r_wtcol_v; int i; for (i = 0; i < num; i++, wc++, weights++) { weightpaint_color((unsigned char *)wc, NULL, *weights); } } void DM_update_weight_mcol( Object *ob, DerivedMesh *dm, int const draw_flag, float *weights, int num, const int *indices) { BMEditMesh *em = (dm->type == DM_TYPE_EDITBMESH) ? BKE_editmesh_from_object(ob) : NULL; unsigned char (*wtcol_v)[4]; int numVerts = dm->getNumVerts(dm); int i; if (em) { BKE_editmesh_color_ensure(em, BM_VERT); wtcol_v = em->derivedVertColor; } else { wtcol_v = MEM_mallocN(sizeof(*wtcol_v) * numVerts, __func__); } /* Weights are given by caller. */ if (weights) { float *w = weights; /* If indices is not NULL, it means we do not have weights for all vertices, * so we must create them (and set them to zero)... */ if (indices) { w = MEM_callocN(sizeof(float) * numVerts, "Temp weight array DM_update_weight_mcol"); i = num; while (i--) w[indices[i]] = weights[i]; } /* Convert float weights to colors. */ calc_colors_from_weights_array(numVerts, w, wtcol_v); if (indices) MEM_freeN(w); } else { /* No weights given, take them from active vgroup(s). */ calc_weightpaint_vert_array(ob, dm, draw_flag, &G_dm_wcinfo, wtcol_v); } if (dm->type == DM_TYPE_EDITBMESH) { /* editmesh draw function checks specifically for this */ } else { const int dm_totpoly = dm->getNumPolys(dm); const int dm_totloop = dm->getNumLoops(dm); unsigned char(*wtcol_l)[4] = CustomData_get_layer(dm->getLoopDataLayout(dm), CD_PREVIEW_MLOOPCOL); MLoop *mloop = dm->getLoopArray(dm), *ml; MPoly *mp = dm->getPolyArray(dm); int l_index; int j; /* now add to loops, so the data can be passed through the modifier stack * If no CD_PREVIEW_MLOOPCOL existed yet, we have to add a new one! */ if (!wtcol_l) { wtcol_l = MEM_mallocN(sizeof(*wtcol_l) * dm_totloop, __func__); CustomData_add_layer(&dm->loopData, CD_PREVIEW_MLOOPCOL, CD_ASSIGN, wtcol_l, dm_totloop); } l_index = 0; for (i = 0; i < dm_totpoly; i++, mp++) { ml = mloop + mp->loopstart; for (j = 0; j < mp->totloop; j++, ml++, l_index++) { copy_v4_v4_uchar(&wtcol_l[l_index][0], &wtcol_v[ml->v][0]); } } MEM_freeN(wtcol_v); dm->dirty |= DM_DIRTY_TESS_CDLAYERS; } } static void DM_update_statvis_color(const Scene *scene, Object *ob, DerivedMesh *dm) { BMEditMesh *em = BKE_editmesh_from_object(ob); BKE_editmesh_statvis_calc(em, dm, &scene->toolsettings->statvis); } static void shapekey_layers_to_keyblocks(DerivedMesh *dm, Mesh *me, int actshape_uid) { KeyBlock *kb; int i, j, tot; if (!me->key) return; tot = CustomData_number_of_layers(&dm->vertData, CD_SHAPEKEY); for (i = 0; i < tot; i++) { CustomDataLayer *layer = &dm->vertData.layers[CustomData_get_layer_index_n(&dm->vertData, CD_SHAPEKEY, i)]; float (*cos)[3], (*kbcos)[3]; for (kb = me->key->block.first; kb; kb = kb->next) { if (kb->uid == layer->uid) break; } if (!kb) { kb = BKE_keyblock_add(me->key, layer->name); kb->uid = layer->uid; } if (kb->data) MEM_freeN(kb->data); cos = CustomData_get_layer_n(&dm->vertData, CD_SHAPEKEY, i); kb->totelem = dm->numVertData; kb->data = kbcos = MEM_mallocN(sizeof(float) * 3 * kb->totelem, "kbcos DerivedMesh.c"); if (kb->uid == actshape_uid) { MVert *mvert = dm->getVertArray(dm); for (j = 0; j < dm->numVertData; j++, kbcos++, mvert++) { copy_v3_v3(*kbcos, mvert->co); } } else { for (j = 0; j < kb->totelem; j++, cos++, kbcos++) { copy_v3_v3(*kbcos, *cos); } } } for (kb = me->key->block.first; kb; kb = kb->next) { if (kb->totelem != dm->numVertData) { if (kb->data) MEM_freeN(kb->data); kb->totelem = dm->numVertData; kb->data = MEM_callocN(sizeof(float) * 3 * kb->totelem, "kb->data derivedmesh.c"); fprintf(stderr, "%s: lost a shapekey layer: '%s'! (bmesh internal error)\n", __func__, kb->name); } } } static void add_shapekey_layers(DerivedMesh *dm, Mesh *me, Object *UNUSED(ob)) { KeyBlock *kb; Key *key = me->key; int i; const size_t shape_alloc_len = sizeof(float) * 3 * me->totvert; if (!me->key) return; /* ensure we can use mesh vertex count for derived mesh custom data */ if (me->totvert != dm->getNumVerts(dm)) { fprintf(stderr, "%s: vertex size mismatch (mesh/dm) '%s' (%d != %d)\n", __func__, me->id.name + 2, me->totvert, dm->getNumVerts(dm)); return; } for (i = 0, kb = key->block.first; kb; kb = kb->next, i++) { int ci; float *array; if (me->totvert != kb->totelem) { fprintf(stderr, "%s: vertex size mismatch (Mesh '%s':%d != KeyBlock '%s':%d)\n", __func__, me->id.name + 2, me->totvert, kb->name, kb->totelem); array = MEM_callocN(shape_alloc_len, __func__); } else { array = MEM_mallocN(shape_alloc_len, __func__); memcpy(array, kb->data, shape_alloc_len); } CustomData_add_layer_named(&dm->vertData, CD_SHAPEKEY, CD_ASSIGN, array, dm->numVertData, kb->name); ci = CustomData_get_layer_index_n(&dm->vertData, CD_SHAPEKEY, i); dm->vertData.layers[ci].uid = kb->uid; } } /** * Called after calculating all modifiers. * * \note tessfaces should already be calculated. */ static void dm_ensure_display_normals(DerivedMesh *dm) { /* Note: dm *may* have a poly CD_NORMAL layer (generated by a modifier needing poly normals e.g.). * We do not use it here, though. And it should be tagged as temp! */ /* BLI_assert((CustomData_has_layer(&dm->polyData, CD_NORMAL) == false)); */ if ((dm->type == DM_TYPE_CDDM) && ((dm->dirty & DM_DIRTY_NORMALS) || CustomData_has_layer(&dm->polyData, CD_NORMAL) == false)) { /* if normals are dirty we want to calculate vertex normals too */ CDDM_calc_normals_mapping_ex(dm, (dm->dirty & DM_DIRTY_NORMALS) ? false : true); } } /** * new value for useDeform -1 (hack for the gameengine): * * - apply only the modifier stack of the object, skipping the virtual modifiers, * - don't apply the key * - apply deform modifiers and input vertexco */ static void mesh_calc_modifiers( Scene *scene, Object *ob, float (*inputVertexCos)[3], const bool useRenderParams, int useDeform, const bool need_mapping, CustomDataMask dataMask, const int index, const bool useCache, const bool build_shapekey_layers, const bool allow_gpu, /* return args */ DerivedMesh **r_deform, DerivedMesh **r_final) { Mesh *me = ob->data; ModifierData *firstmd, *md, *previewmd = NULL; CDMaskLink *datamasks, *curr; /* XXX Always copying POLYINDEX, else tessellated data are no more valid! */ CustomDataMask mask, nextmask, previewmask = 0, append_mask = CD_MASK_ORIGINDEX; float (*deformedVerts)[3] = NULL; DerivedMesh *dm = NULL, *orcodm, *clothorcodm, *finaldm; int numVerts = me->totvert; const int required_mode = useRenderParams ? eModifierMode_Render : eModifierMode_Realtime; bool isPrevDeform = false; const bool skipVirtualArmature = (useDeform < 0); MultiresModifierData *mmd = get_multires_modifier(scene, ob, 0); const bool has_multires = (mmd && mmd->sculptlvl != 0); bool multires_applied = false; const bool sculpt_mode = ob->mode & OB_MODE_SCULPT && ob->sculpt && !useRenderParams; const bool sculpt_dyntopo = (sculpt_mode && ob->sculpt->bm) && !useRenderParams; const int draw_flag = dm_drawflag_calc(scene->toolsettings, me); /* Generic preview only in object mode! */ const bool do_mod_mcol = (ob->mode == OB_MODE_OBJECT); #if 0 /* XXX Will re-enable this when we have global mod stack options. */ const bool do_final_wmcol = (scene->toolsettings->weights_preview == WP_WPREVIEW_FINAL) && do_wmcol; #endif const bool do_final_wmcol = false; const bool do_init_wmcol = ((dataMask & CD_MASK_PREVIEW_MLOOPCOL) && (ob->mode & OB_MODE_WEIGHT_PAINT) && !do_final_wmcol); /* XXX Same as above... For now, only weights preview in WPaint mode. */ const bool do_mod_wmcol = do_init_wmcol; const bool do_loop_normals = (me->flag & ME_AUTOSMOOTH) != 0; const float loop_normals_split_angle = me->smoothresh; VirtualModifierData virtualModifierData; ModifierApplyFlag app_flags = useRenderParams ? MOD_APPLY_RENDER : 0; ModifierApplyFlag deform_app_flags = app_flags; if (useCache) app_flags |= MOD_APPLY_USECACHE; if (allow_gpu) app_flags |= MOD_APPLY_ALLOW_GPU; if (useDeform) deform_app_flags |= MOD_APPLY_USECACHE; if (!skipVirtualArmature) { firstmd = modifiers_getVirtualModifierList(ob, &virtualModifierData); } else { /* game engine exception */ firstmd = ob->modifiers.first; if (firstmd && firstmd->type == eModifierType_Armature) firstmd = firstmd->next; } md = firstmd; modifiers_clearErrors(ob); if (do_mod_wmcol || do_mod_mcol) { /* Find the last active modifier generating a preview, or NULL if none. */ /* XXX Currently, DPaint modifier just ignores this. * Needs a stupid hack... * The whole "modifier preview" thing has to be (re?)designed, anyway! */ previewmd = modifiers_getLastPreview(scene, md, required_mode); /* even if the modifier doesn't need the data, to make a preview it may */ if (previewmd) { if (do_mod_wmcol) { previewmask = CD_MASK_MDEFORMVERT; } } } datamasks = modifiers_calcDataMasks(scene, ob, md, dataMask, required_mode, previewmd, previewmask); curr = datamasks; if (r_deform) { *r_deform = NULL; } *r_final = NULL; if (useDeform) { if (inputVertexCos) deformedVerts = inputVertexCos; /* Apply all leading deforming modifiers */ for (; md; md = md->next, curr = curr->next) { const ModifierTypeInfo *mti = modifierType_getInfo(md->type); md->scene = scene; if (!modifier_isEnabled(scene, md, required_mode)) { continue; } if (useDeform < 0 && mti->dependsOnTime && mti->dependsOnTime(md)) { continue; } if (mti->type == eModifierTypeType_OnlyDeform && !sculpt_dyntopo) { if (!deformedVerts) deformedVerts = BKE_mesh_vertexCos_get(me, &numVerts); modwrap_deformVerts(md, ob, NULL, deformedVerts, numVerts, deform_app_flags); } else { break; } /* grab modifiers until index i */ if ((index != -1) && (BLI_findindex(&ob->modifiers, md) >= index)) break; } /* Result of all leading deforming modifiers is cached for * places that wish to use the original mesh but with deformed * coordinates (vpaint, etc.) */ if (r_deform) { *r_deform = CDDM_from_mesh(me); if (build_shapekey_layers) add_shapekey_layers(dm, me, ob); if (deformedVerts) { CDDM_apply_vert_coords(*r_deform, deformedVerts); } } } else { /* default behavior for meshes */ if (inputVertexCos) deformedVerts = inputVertexCos; else deformedVerts = BKE_mesh_vertexCos_get(me, &numVerts); } /* Now apply all remaining modifiers. If useDeform is off then skip * OnlyDeform ones. */ dm = NULL; orcodm = NULL; clothorcodm = NULL; for (; md; md = md->next, curr = curr->next) { const ModifierTypeInfo *mti = modifierType_getInfo(md->type); md->scene = scene; if (!modifier_isEnabled(scene, md, required_mode)) { continue; } if (mti->type == eModifierTypeType_OnlyDeform && !useDeform) { continue; } if ((mti->flags & eModifierTypeFlag_RequiresOriginalData) && dm) { modifier_setError(md, "Modifier requires original data, bad stack position"); continue; } if (sculpt_mode && (!has_multires || multires_applied || sculpt_dyntopo)) { bool unsupported = false; if (md->type == eModifierType_Multires && ((MultiresModifierData *)md)->sculptlvl == 0) { /* If multires is on level 0 skip it silently without warning message. */ if (!sculpt_dyntopo) { continue; } } if (sculpt_dyntopo && !useRenderParams) unsupported = true; if (scene->toolsettings->sculpt->flags & SCULPT_ONLY_DEFORM) unsupported |= (mti->type != eModifierTypeType_OnlyDeform); unsupported |= multires_applied; if (unsupported) { if (sculpt_dyntopo) modifier_setError(md, "Not supported in dyntopo"); else modifier_setError(md, "Not supported in sculpt mode"); continue; } else { modifier_setError(md, "Hide, Mask and optimized display disabled"); } } if (need_mapping && !modifier_supportsMapping(md)) { continue; } if (useDeform < 0 && mti->dependsOnTime && mti->dependsOnTime(md)) { continue; } /* add an orco layer if needed by this modifier */ if (mti->requiredDataMask) mask = mti->requiredDataMask(ob, md); else mask = 0; if (dm && (mask & CD_MASK_ORCO)) add_orco_dm(ob, NULL, dm, orcodm, CD_ORCO); /* How to apply modifier depends on (a) what we already have as * a result of previous modifiers (could be a DerivedMesh or just * deformed vertices) and (b) what type the modifier is. */ if (mti->type == eModifierTypeType_OnlyDeform) { /* No existing verts to deform, need to build them. */ if (!deformedVerts) { if (dm) { /* Deforming a derived mesh, read the vertex locations * out of the mesh and deform them. Once done with this * run of deformers verts will be written back. */ numVerts = dm->getNumVerts(dm); deformedVerts = MEM_mallocN(sizeof(*deformedVerts) * numVerts, "dfmv"); dm->getVertCos(dm, deformedVerts); } else { deformedVerts = BKE_mesh_vertexCos_get(me, &numVerts); } } /* if this is not the last modifier in the stack then recalculate the normals * to avoid giving bogus normals to the next modifier see: [#23673] */ if (isPrevDeform && mti->dependsOnNormals && mti->dependsOnNormals(md)) { /* XXX, this covers bug #23673, but we may need normal calc for other types */ if (dm && dm->type == DM_TYPE_CDDM) { CDDM_apply_vert_coords(dm, deformedVerts); } } modwrap_deformVerts(md, ob, dm, deformedVerts, numVerts, deform_app_flags); } else { DerivedMesh *ndm; /* determine which data layers are needed by following modifiers */ if (curr->next) nextmask = curr->next->mask; else nextmask = dataMask; /* apply vertex coordinates or build a DerivedMesh as necessary */ if (dm) { if (deformedVerts) { DerivedMesh *tdm = CDDM_copy(dm); dm->release(dm); dm = tdm; CDDM_apply_vert_coords(dm, deformedVerts); } } else { dm = CDDM_from_mesh(me); ASSERT_IS_VALID_DM(dm); if (build_shapekey_layers) add_shapekey_layers(dm, me, ob); if (deformedVerts) { CDDM_apply_vert_coords(dm, deformedVerts); } if (do_init_wmcol) DM_update_weight_mcol(ob, dm, draw_flag, NULL, 0, NULL); /* Constructive modifiers need to have an origindex * otherwise they wont have anywhere to copy the data from. * * Also create ORIGINDEX data if any of the following modifiers * requests it, this way Mirror, Solidify etc will keep ORIGINDEX * data by using generic DM_copy_vert_data() functions. */ if (need_mapping || (nextmask & CD_MASK_ORIGINDEX)) { /* calc */ DM_add_vert_layer(dm, CD_ORIGINDEX, CD_CALLOC, NULL); DM_add_edge_layer(dm, CD_ORIGINDEX, CD_CALLOC, NULL); DM_add_poly_layer(dm, CD_ORIGINDEX, CD_CALLOC, NULL); /* Not worth parallelizing this, gives less than 0.1% overall speedup in best of best cases... */ range_vn_i(DM_get_vert_data_layer(dm, CD_ORIGINDEX), dm->numVertData, 0); range_vn_i(DM_get_edge_data_layer(dm, CD_ORIGINDEX), dm->numEdgeData, 0); range_vn_i(DM_get_poly_data_layer(dm, CD_ORIGINDEX), dm->numPolyData, 0); } } /* set the DerivedMesh to only copy needed data */ mask = curr->mask; /* needMapping check here fixes bug [#28112], otherwise it's * possible that it won't be copied */ mask |= append_mask; DM_set_only_copy(dm, mask | (need_mapping ? CD_MASK_ORIGINDEX : 0)); /* add cloth rest shape key if needed */ if (mask & CD_MASK_CLOTH_ORCO) add_orco_dm(ob, NULL, dm, clothorcodm, CD_CLOTH_ORCO); /* add an origspace layer if needed */ if ((curr->mask) & CD_MASK_ORIGSPACE_MLOOP) { if (!CustomData_has_layer(&dm->loopData, CD_ORIGSPACE_MLOOP)) { DM_add_loop_layer(dm, CD_ORIGSPACE_MLOOP, CD_CALLOC, NULL); DM_init_origspace(dm); } } ndm = modwrap_applyModifier(md, ob, dm, app_flags); ASSERT_IS_VALID_DM(ndm); if (ndm) { /* if the modifier returned a new dm, release the old one */ if (dm && dm != ndm) dm->release(dm); dm = ndm; if (deformedVerts) { if (deformedVerts != inputVertexCos) MEM_freeN(deformedVerts); deformedVerts = NULL; } } /* create an orco derivedmesh in parallel */ if (nextmask & CD_MASK_ORCO) { if (!orcodm) orcodm = create_orco_dm(ob, me, NULL, CD_ORCO); nextmask &= ~CD_MASK_ORCO; DM_set_only_copy(orcodm, nextmask | CD_MASK_ORIGINDEX | (mti->requiredDataMask ? mti->requiredDataMask(ob, md) : 0)); ndm = modwrap_applyModifier(md, ob, orcodm, (app_flags & ~MOD_APPLY_USECACHE) | MOD_APPLY_ORCO); ASSERT_IS_VALID_DM(ndm); if (ndm) { /* if the modifier returned a new dm, release the old one */ if (orcodm && orcodm != ndm) orcodm->release(orcodm); orcodm = ndm; } } /* create cloth orco derivedmesh in parallel */ if (nextmask & CD_MASK_CLOTH_ORCO) { if (!clothorcodm) clothorcodm = create_orco_dm(ob, me, NULL, CD_CLOTH_ORCO); nextmask &= ~CD_MASK_CLOTH_ORCO; DM_set_only_copy(clothorcodm, nextmask | CD_MASK_ORIGINDEX); ndm = modwrap_applyModifier(md, ob, clothorcodm, (app_flags & ~MOD_APPLY_USECACHE) | MOD_APPLY_ORCO); ASSERT_IS_VALID_DM(ndm); if (ndm) { /* if the modifier returned a new dm, release the old one */ if (clothorcodm && clothorcodm != ndm) { clothorcodm->release(clothorcodm); } clothorcodm = ndm; } } /* in case of dynamic paint, make sure preview mask remains for following modifiers */ /* XXX Temp and hackish solution! */ if (md->type == eModifierType_DynamicPaint) append_mask |= CD_MASK_PREVIEW_MLOOPCOL; /* In case of active preview modifier, make sure preview mask remains for following modifiers. */ else if ((md == previewmd) && (do_mod_wmcol)) { DM_update_weight_mcol(ob, dm, draw_flag, NULL, 0, NULL); append_mask |= CD_MASK_PREVIEW_MLOOPCOL; } dm->deformedOnly = false; } isPrevDeform = (mti->type == eModifierTypeType_OnlyDeform); /* grab modifiers until index i */ if ((index != -1) && (BLI_findindex(&ob->modifiers, md) >= index)) break; if (sculpt_mode && md->type == eModifierType_Multires) { multires_applied = true; } } for (md = firstmd; md; md = md->next) modifier_freeTemporaryData(md); /* Yay, we are done. If we have a DerivedMesh and deformed vertices * need to apply these back onto the DerivedMesh. If we have no * DerivedMesh then we need to build one. */ if (dm && deformedVerts) { finaldm = CDDM_copy(dm); dm->release(dm); CDDM_apply_vert_coords(finaldm, deformedVerts); #if 0 /* For later nice mod preview! */ /* In case we need modified weights in CD_PREVIEW_MCOL, we have to re-compute it. */ if (do_final_wmcol) DM_update_weight_mcol(ob, finaldm, draw_flag, NULL, 0, NULL); #endif } else if (dm) { finaldm = dm; #if 0 /* For later nice mod preview! */ /* In case we need modified weights in CD_PREVIEW_MCOL, we have to re-compute it. */ if (do_final_wmcol) DM_update_weight_mcol(ob, finaldm, draw_flag, NULL, 0, NULL); #endif } else { finaldm = CDDM_from_mesh(me); if (build_shapekey_layers) { add_shapekey_layers(finaldm, me, ob); } if (deformedVerts) { CDDM_apply_vert_coords(finaldm, deformedVerts); } /* In this case, we should never have weight-modifying modifiers in stack... */ if (do_init_wmcol) DM_update_weight_mcol(ob, finaldm, draw_flag, NULL, 0, NULL); } /* add an orco layer if needed */ if (dataMask & CD_MASK_ORCO) { add_orco_dm(ob, NULL, finaldm, orcodm, CD_ORCO); if (r_deform && *r_deform) add_orco_dm(ob, NULL, *r_deform, NULL, CD_ORCO); } if (do_loop_normals) { /* Compute loop normals (note: will compute poly and vert normals as well, if needed!) */ DM_calc_loop_normals(finaldm, do_loop_normals, loop_normals_split_angle); } if (sculpt_dyntopo == false) { /* watch this! after 2.75a we move to from tessface to looptri (by default) */ if (dataMask & CD_MASK_MFACE) { DM_ensure_tessface(finaldm); } /* without this, drawing ngon tri's faces will show ugly tessellated face * normals and will also have to calculate normals on the fly, try avoid * this where possible since calculating polygon normals isn't fast, * note that this isn't a problem for subsurf (only quads) or editmode * which deals with drawing differently. * * Only calc vertex normals if they are flagged as dirty. * If using loop normals, poly nors have already been computed. */ if (!do_loop_normals) { dm_ensure_display_normals(finaldm); } } /* Some modifiers, like datatransfer, may generate those data as temp layer, we do not want to keep them, * as they are used by display code when available (i.e. even if autosmooth is disabled). */ if (!do_loop_normals && CustomData_has_layer(&finaldm->loopData, CD_NORMAL)) { CustomData_free_layers(&finaldm->loopData, CD_NORMAL, finaldm->numLoopData); } #ifdef WITH_GAMEENGINE /* NavMesh - this is a hack but saves having a NavMesh modifier */ if ((ob->gameflag & OB_NAVMESH) && (finaldm->type == DM_TYPE_CDDM)) { DerivedMesh *tdm; tdm = navmesh_dm_createNavMeshForVisualization(finaldm); if (finaldm != tdm) { finaldm->release(finaldm); finaldm = tdm; } DM_ensure_tessface(finaldm); } #endif /* WITH_GAMEENGINE */ *r_final = finaldm; if (orcodm) orcodm->release(orcodm); if (clothorcodm) clothorcodm->release(clothorcodm); if (deformedVerts && deformedVerts != inputVertexCos) MEM_freeN(deformedVerts); BLI_linklist_free((LinkNode *)datamasks, NULL); } float (*editbmesh_get_vertex_cos(BMEditMesh *em, int *r_numVerts))[3] { BMIter iter; BMVert *eve; float (*cos)[3]; int i; *r_numVerts = em->bm->totvert; cos = MEM_mallocN(sizeof(float) * 3 * em->bm->totvert, "vertexcos"); BM_ITER_MESH_INDEX (eve, &iter, em->bm, BM_VERTS_OF_MESH, i) { copy_v3_v3(cos[i], eve->co); } return cos; } bool editbmesh_modifier_is_enabled(Scene *scene, ModifierData *md, DerivedMesh *dm) { const ModifierTypeInfo *mti = modifierType_getInfo(md->type); const int required_mode = eModifierMode_Realtime | eModifierMode_Editmode; if (!modifier_isEnabled(scene, md, required_mode)) { return false; } if ((mti->flags & eModifierTypeFlag_RequiresOriginalData) && dm) { modifier_setError(md, "Modifier requires original data, bad stack position"); return false; } return true; } static void editbmesh_calc_modifiers( Scene *scene, Object *ob, BMEditMesh *em, CustomDataMask dataMask, /* return args */ DerivedMesh **r_cage, DerivedMesh **r_final) { ModifierData *md, *previewmd = NULL; float (*deformedVerts)[3] = NULL; CustomDataMask mask = 0, previewmask = 0, append_mask = 0; DerivedMesh *dm = NULL, *orcodm = NULL; int i, numVerts = 0, cageIndex = modifiers_getCageIndex(scene, ob, NULL, 1); CDMaskLink *datamasks, *curr; const int required_mode = eModifierMode_Realtime | eModifierMode_Editmode; int draw_flag = dm_drawflag_calc(scene->toolsettings, ob->data); // const bool do_mod_mcol = true; // (ob->mode == OB_MODE_OBJECT); #if 0 /* XXX Will re-enable this when we have global mod stack options. */ const bool do_final_wmcol = (scene->toolsettings->weights_preview == WP_WPREVIEW_FINAL) && do_wmcol; #endif const bool do_final_wmcol = false; const bool do_init_wmcol = ((((Mesh *)ob->data)->drawflag & ME_DRAWEIGHT) && !do_final_wmcol); const bool do_init_statvis = ((((Mesh *)ob->data)->drawflag & ME_DRAW_STATVIS) && !do_init_wmcol); const bool do_mod_wmcol = do_init_wmcol; VirtualModifierData virtualModifierData; const bool do_loop_normals = (((Mesh *)(ob->data))->flag & ME_AUTOSMOOTH) != 0; const float loop_normals_split_angle = ((Mesh *)(ob->data))->smoothresh; modifiers_clearErrors(ob); if (r_cage && cageIndex == -1) { *r_cage = getEditDerivedBMesh(em, ob, dataMask, NULL); } md = modifiers_getVirtualModifierList(ob, &virtualModifierData); /* copied from mesh_calc_modifiers */ if (do_mod_wmcol) { previewmd = modifiers_getLastPreview(scene, md, required_mode); /* even if the modifier doesn't need the data, to make a preview it may */ if (previewmd) { previewmask = CD_MASK_MDEFORMVERT; } } datamasks = modifiers_calcDataMasks(scene, ob, md, dataMask, required_mode, previewmd, previewmask); curr = datamasks; for (i = 0; md; i++, md = md->next, curr = curr->next) { const ModifierTypeInfo *mti = modifierType_getInfo(md->type); md->scene = scene; if (!editbmesh_modifier_is_enabled(scene, md, dm)) { continue; } /* add an orco layer if needed by this modifier */ if (dm && mti->requiredDataMask) { mask = mti->requiredDataMask(ob, md); if (mask & CD_MASK_ORCO) add_orco_dm(ob, em, dm, orcodm, CD_ORCO); } /* How to apply modifier depends on (a) what we already have as * a result of previous modifiers (could be a DerivedMesh or just * deformed vertices) and (b) what type the modifier is. */ if (mti->type == eModifierTypeType_OnlyDeform) { /* No existing verts to deform, need to build them. */ if (!deformedVerts) { if (dm) { /* Deforming a derived mesh, read the vertex locations * out of the mesh and deform them. Once done with this * run of deformers verts will be written back. */ numVerts = dm->getNumVerts(dm); deformedVerts = MEM_mallocN(sizeof(*deformedVerts) * numVerts, "dfmv"); dm->getVertCos(dm, deformedVerts); } else { deformedVerts = editbmesh_get_vertex_cos(em, &numVerts); } } if (mti->deformVertsEM) modwrap_deformVertsEM(md, ob, em, dm, deformedVerts, numVerts); else modwrap_deformVerts(md, ob, dm, deformedVerts, numVerts, 0); } else { DerivedMesh *ndm; /* apply vertex coordinates or build a DerivedMesh as necessary */ if (dm) { if (deformedVerts) { DerivedMesh *tdm = CDDM_copy(dm); if (!(r_cage && dm == *r_cage)) { dm->release(dm); } dm = tdm; CDDM_apply_vert_coords(dm, deformedVerts); } else if (r_cage && dm == *r_cage) { /* dm may be changed by this modifier, so we need to copy it */ dm = CDDM_copy(dm); } } else { dm = CDDM_from_editbmesh(em, false, false); ASSERT_IS_VALID_DM(dm); if (deformedVerts) { CDDM_apply_vert_coords(dm, deformedVerts); } if (do_init_wmcol) { DM_update_weight_mcol(ob, dm, draw_flag, NULL, 0, NULL); } } /* create an orco derivedmesh in parallel */ mask = curr->mask; if (mask & CD_MASK_ORCO) { if (!orcodm) orcodm = create_orco_dm(ob, ob->data, em, CD_ORCO); mask &= ~CD_MASK_ORCO; DM_set_only_copy(orcodm, mask | CD_MASK_ORIGINDEX); if (mti->applyModifierEM) { ndm = modwrap_applyModifierEM(md, ob, em, orcodm, MOD_APPLY_ORCO); } else { ndm = modwrap_applyModifier(md, ob, orcodm, MOD_APPLY_ORCO); } ASSERT_IS_VALID_DM(ndm); if (ndm) { /* if the modifier returned a new dm, release the old one */ if (orcodm && orcodm != ndm) orcodm->release(orcodm); orcodm = ndm; } } /* set the DerivedMesh to only copy needed data */ mask |= append_mask; mask = curr->mask; /* CD_MASK_ORCO may have been cleared above */ DM_set_only_copy(dm, mask | CD_MASK_ORIGINDEX); if (mask & CD_MASK_ORIGSPACE_MLOOP) { if (!CustomData_has_layer(&dm->loopData, CD_ORIGSPACE_MLOOP)) { DM_add_loop_layer(dm, CD_ORIGSPACE_MLOOP, CD_CALLOC, NULL); DM_init_origspace(dm); } } if (mti->applyModifierEM) ndm = modwrap_applyModifierEM(md, ob, em, dm, MOD_APPLY_USECACHE | MOD_APPLY_ALLOW_GPU); else ndm = modwrap_applyModifier(md, ob, dm, MOD_APPLY_USECACHE | MOD_APPLY_ALLOW_GPU); ASSERT_IS_VALID_DM(ndm); if (ndm) { if (dm && dm != ndm) dm->release(dm); dm = ndm; if (deformedVerts) { MEM_freeN(deformedVerts); deformedVerts = NULL; } } dm->deformedOnly = false; } /* In case of active preview modifier, make sure preview mask remains for following modifiers. */ if ((md == previewmd) && (do_mod_wmcol)) { DM_update_weight_mcol(ob, dm, draw_flag, NULL, 0, NULL); append_mask |= CD_MASK_PREVIEW_MLOOPCOL; } if (r_cage && i == cageIndex) { if (dm && deformedVerts) { *r_cage = CDDM_copy(dm); CDDM_apply_vert_coords(*r_cage, deformedVerts); } else if (dm) { *r_cage = dm; } else { *r_cage = getEditDerivedBMesh( em, ob, mask, deformedVerts ? MEM_dupallocN(deformedVerts) : NULL); } } } BLI_linklist_free((LinkNode *)datamasks, NULL); /* Yay, we are done. If we have a DerivedMesh and deformed vertices need * to apply these back onto the DerivedMesh. If we have no DerivedMesh * then we need to build one. */ if (dm && deformedVerts) { *r_final = CDDM_copy(dm); if (!(r_cage && dm == *r_cage)) { dm->release(dm); } CDDM_apply_vert_coords(*r_final, deformedVerts); } else if (dm) { *r_final = dm; } else if (!deformedVerts && r_cage && *r_cage) { /* cage should already have up to date normals */ *r_final = *r_cage; /* In this case, we should never have weight-modifying modifiers in stack... */ if (do_init_wmcol) DM_update_weight_mcol(ob, *r_final, draw_flag, NULL, 0, NULL); if (do_init_statvis) DM_update_statvis_color(scene, ob, *r_final); } else { /* this is just a copy of the editmesh, no need to calc normals */ *r_final = getEditDerivedBMesh(em, ob, dataMask, deformedVerts); deformedVerts = NULL; /* In this case, we should never have weight-modifying modifiers in stack... */ if (do_init_wmcol) DM_update_weight_mcol(ob, *r_final, draw_flag, NULL, 0, NULL); if (do_init_statvis) DM_update_statvis_color(scene, ob, *r_final); } if (do_loop_normals) { /* Compute loop normals */ DM_calc_loop_normals(*r_final, do_loop_normals, loop_normals_split_angle); if (r_cage && *r_cage && (*r_cage != *r_final)) { DM_calc_loop_normals(*r_cage, do_loop_normals, loop_normals_split_angle); } } /* BMESH_ONLY, ensure tessface's used for drawing, * but don't recalculate if the last modifier in the stack gives us tessfaces * check if the derived meshes are DM_TYPE_EDITBMESH before calling, this isn't essential * but quiets annoying error messages since tessfaces wont be created. */ if (dataMask & CD_MASK_MFACE) { if ((*r_final)->type != DM_TYPE_EDITBMESH) { DM_ensure_tessface(*r_final); } if (r_cage && *r_cage) { if ((*r_cage)->type != DM_TYPE_EDITBMESH) { if (*r_cage != *r_final) { DM_ensure_tessface(*r_cage); } } } } /* --- */ /* same as mesh_calc_modifiers (if using loop normals, poly nors have already been computed). */ if (!do_loop_normals) { dm_ensure_display_normals(*r_final); /* Some modifiers, like datatransfer, may generate those data, we do not want to keep them, * as they are used by display code when available (i.e. even if autosmooth is disabled). */ if (CustomData_has_layer(&(*r_final)->loopData, CD_NORMAL)) { CustomData_free_layers(&(*r_final)->loopData, CD_NORMAL, (*r_final)->numLoopData); } if (r_cage && CustomData_has_layer(&(*r_cage)->loopData, CD_NORMAL)) { CustomData_free_layers(&(*r_cage)->loopData, CD_NORMAL, (*r_cage)->numLoopData); } } /* add an orco layer if needed */ if (dataMask & CD_MASK_ORCO) add_orco_dm(ob, em, *r_final, orcodm, CD_ORCO); if (orcodm) orcodm->release(orcodm); if (deformedVerts) MEM_freeN(deformedVerts); } #ifdef WITH_OPENSUBDIV /* The idea is to skip CPU-side ORCO calculation when * we'll be using GPU backend of OpenSubdiv. This is so * playback performance is kept as high as possible. */ static bool calc_modifiers_skip_orco(Scene *scene, Object *ob, bool use_render_params) { ModifierData *last_md = ob->modifiers.last; const int required_mode = use_render_params ? eModifierMode_Render : eModifierMode_Realtime; if (last_md != NULL && last_md->type == eModifierType_Subsurf && modifier_isEnabled(scene, last_md, required_mode)) { if (U.opensubdiv_compute_type == USER_OPENSUBDIV_COMPUTE_NONE) { return false; } else if ((ob->mode & (OB_MODE_VERTEX_PAINT | OB_MODE_WEIGHT_PAINT | OB_MODE_TEXTURE_PAINT)) != 0) { return false; } else if ((DAG_get_eval_flags_for_object(scene, ob) & DAG_EVAL_NEED_CPU) != 0) { return false; } SubsurfModifierData *smd = (SubsurfModifierData *)last_md; /* TODO(sergey): Deduplicate this with checks from subsurf_ccg.c. */ return smd->use_opensubdiv != 0; } return false; } #endif static void mesh_build_data( Scene *scene, Object *ob, CustomDataMask dataMask, const bool build_shapekey_layers, const bool need_mapping) { BLI_assert(ob->type == OB_MESH); BKE_object_free_derived_caches(ob); BKE_object_sculpt_modifiers_changed(ob); #ifdef WITH_OPENSUBDIV if (calc_modifiers_skip_orco(scene, ob, false)) { dataMask &= ~(CD_MASK_ORCO | CD_MASK_PREVIEW_MCOL); } #endif mesh_calc_modifiers( scene, ob, NULL, false, 1, need_mapping, dataMask, -1, true, build_shapekey_layers, true, &ob->derivedDeform, &ob->derivedFinal); DM_set_object_boundbox(ob, ob->derivedFinal); ob->derivedFinal->needsFree = 0; ob->derivedDeform->needsFree = 0; ob->lastDataMask = dataMask; ob->lastNeedMapping = need_mapping; if ((ob->mode & OB_MODE_ALL_SCULPT) && ob->sculpt) { /* create PBVH immediately (would be created on the fly too, * but this avoids waiting on first stroke) */ BKE_sculpt_update_mesh_elements(scene, scene->toolsettings->sculpt, ob, false, false); } BLI_assert(!(ob->derivedFinal->dirty & DM_DIRTY_NORMALS)); } static void editbmesh_build_data(Scene *scene, Object *obedit, BMEditMesh *em, CustomDataMask dataMask) { BKE_object_free_derived_caches(obedit); BKE_object_sculpt_modifiers_changed(obedit); BKE_editmesh_free_derivedmesh(em); #ifdef WITH_OPENSUBDIV if (calc_modifiers_skip_orco(scene, obedit, false)) { dataMask &= ~(CD_MASK_ORCO | CD_MASK_PREVIEW_MCOL); } #endif editbmesh_calc_modifiers( scene, obedit, em, dataMask, &em->derivedCage, &em->derivedFinal); DM_set_object_boundbox(obedit, em->derivedFinal); em->lastDataMask = dataMask; em->derivedFinal->needsFree = 0; em->derivedCage->needsFree = 0; BLI_assert(!(em->derivedFinal->dirty & DM_DIRTY_NORMALS)); } static CustomDataMask object_get_datamask(const Scene *scene, Object *ob, bool *r_need_mapping) { Object *actob = scene->basact ? scene->basact->object : NULL; CustomDataMask mask = ob->customdata_mask; if (r_need_mapping) { *r_need_mapping = false; } if (ob == actob) { bool editing = BKE_paint_select_face_test(ob); /* weight paint and face select need original indices because of selection buffer drawing */ if (r_need_mapping) { *r_need_mapping = (editing || (ob->mode & (OB_MODE_WEIGHT_PAINT | OB_MODE_VERTEX_PAINT))); } /* check if we need tfaces & mcols due to face select or texture paint */ if ((ob->mode & OB_MODE_TEXTURE_PAINT) || editing) { mask |= CD_MASK_MLOOPUV | CD_MASK_MLOOPCOL; } /* check if we need mcols due to vertex paint or weightpaint */ if (ob->mode & OB_MODE_VERTEX_PAINT) { mask |= CD_MASK_MLOOPCOL; } if (ob->mode & OB_MODE_WEIGHT_PAINT) { mask |= CD_MASK_PREVIEW_MLOOPCOL; } if (ob->mode & OB_MODE_EDIT) mask |= CD_MASK_MVERT_SKIN; } return mask; } void makeDerivedMesh( Scene *scene, Object *ob, BMEditMesh *em, CustomDataMask dataMask, const bool build_shapekey_layers) { bool need_mapping; dataMask |= object_get_datamask(scene, ob, &need_mapping); if (em) { editbmesh_build_data(scene, ob, em, dataMask); } else { mesh_build_data(scene, ob, dataMask, build_shapekey_layers, need_mapping); } } /***/ DerivedMesh *mesh_get_derived_final(Scene *scene, Object *ob, CustomDataMask dataMask) { /* if there's no derived mesh or the last data mask used doesn't include * the data we need, rebuild the derived mesh */ bool need_mapping; dataMask |= object_get_datamask(scene, ob, &need_mapping); if (!ob->derivedFinal || ((dataMask & ob->lastDataMask) != dataMask) || (need_mapping != ob->lastNeedMapping)) { mesh_build_data(scene, ob, dataMask, false, need_mapping); } if (ob->derivedFinal) { BLI_assert(!(ob->derivedFinal->dirty & DM_DIRTY_NORMALS)); } return ob->derivedFinal; } DerivedMesh *mesh_get_derived_deform(Scene *scene, Object *ob, CustomDataMask dataMask) { /* if there's no derived mesh or the last data mask used doesn't include * the data we need, rebuild the derived mesh */ bool need_mapping; dataMask |= object_get_datamask(scene, ob, &need_mapping); if (!ob->derivedDeform || ((dataMask & ob->lastDataMask) != dataMask) || (need_mapping != ob->lastNeedMapping)) { mesh_build_data(scene, ob, dataMask, false, need_mapping); } return ob->derivedDeform; } DerivedMesh *mesh_create_derived_render(Scene *scene, Object *ob, CustomDataMask dataMask) { DerivedMesh *final; mesh_calc_modifiers( scene, ob, NULL, true, 1, false, dataMask, -1, false, false, false, NULL, &final); return final; } DerivedMesh *mesh_create_derived_index_render(Scene *scene, Object *ob, CustomDataMask dataMask, int index) { DerivedMesh *final; mesh_calc_modifiers( scene, ob, NULL, true, 1, false, dataMask, index, false, false, false, NULL, &final); return final; } DerivedMesh *mesh_create_derived_view( Scene *scene, Object *ob, CustomDataMask dataMask) { DerivedMesh *final; /* XXX hack * psys modifier updates particle state when called during dupli-list generation, * which can lead to wrong transforms. This disables particle system modifier execution. */ ob->transflag |= OB_NO_PSYS_UPDATE; mesh_calc_modifiers( scene, ob, NULL, false, 1, false, dataMask, -1, false, false, false, NULL, &final); ob->transflag &= ~OB_NO_PSYS_UPDATE; return final; } DerivedMesh *mesh_create_derived_no_deform( Scene *scene, Object *ob, float (*vertCos)[3], CustomDataMask dataMask) { DerivedMesh *final; mesh_calc_modifiers( scene, ob, vertCos, false, 0, false, dataMask, -1, false, false, false, NULL, &final); return final; } DerivedMesh *mesh_create_derived_no_virtual( Scene *scene, Object *ob, float (*vertCos)[3], CustomDataMask dataMask) { DerivedMesh *final; mesh_calc_modifiers( scene, ob, vertCos, false, -1, false, dataMask, -1, false, false, false, NULL, &final); return final; } DerivedMesh *mesh_create_derived_physics( Scene *scene, Object *ob, float (*vertCos)[3], CustomDataMask dataMask) { DerivedMesh *final; mesh_calc_modifiers( scene, ob, vertCos, false, -1, true, dataMask, -1, false, false, false, NULL, &final); return final; } DerivedMesh *mesh_create_derived_no_deform_render( Scene *scene, Object *ob, float (*vertCos)[3], CustomDataMask dataMask) { DerivedMesh *final; mesh_calc_modifiers( scene, ob, vertCos, true, 0, false, dataMask, -1, false, false, false, NULL, &final); return final; } /***/ DerivedMesh *editbmesh_get_derived_cage_and_final( Scene *scene, Object *obedit, BMEditMesh *em, CustomDataMask dataMask, /* return args */ DerivedMesh **r_final) { /* if there's no derived mesh or the last data mask used doesn't include * the data we need, rebuild the derived mesh */ dataMask |= object_get_datamask(scene, obedit, NULL); if (!em->derivedCage || (em->lastDataMask & dataMask) != dataMask) { editbmesh_build_data(scene, obedit, em, dataMask); } *r_final = em->derivedFinal; if (em->derivedFinal) { BLI_assert(!(em->derivedFinal->dirty & DM_DIRTY_NORMALS)); } return em->derivedCage; } DerivedMesh *editbmesh_get_derived_cage(Scene *scene, Object *obedit, BMEditMesh *em, CustomDataMask dataMask) { /* if there's no derived mesh or the last data mask used doesn't include * the data we need, rebuild the derived mesh */ dataMask |= object_get_datamask(scene, obedit, NULL); if (!em->derivedCage || (em->lastDataMask & dataMask) != dataMask) { editbmesh_build_data(scene, obedit, em, dataMask); } return em->derivedCage; } DerivedMesh *editbmesh_get_derived_base(Object *obedit, BMEditMesh *em, CustomDataMask data_mask) { return getEditDerivedBMesh(em, obedit, data_mask, NULL); } /***/ /* get derived mesh from an object, using editbmesh if available. */ DerivedMesh *object_get_derived_final(Object *ob, const bool for_render) { if (for_render) { /* TODO(sergey): use proper derived render here in the future. */ return ob->derivedFinal; } /* only return the editmesh if its from this object because * we don't a mesh from another object's modifier stack: T43122 */ if (ob->type == OB_MESH) { Mesh *me = ob->data; BMEditMesh *em = me->edit_btmesh; if (em && (em->ob == ob)) { DerivedMesh *dm = em->derivedFinal; return dm; } } return ob->derivedFinal; } /* UNUSED */ #if 0 /* ********* For those who don't grasp derived stuff! (ton) :) *************** */ static void make_vertexcosnos__mapFunc(void *userData, int index, const float co[3], const float no_f[3], const short no_s[3]) { DMCoNo *co_no = &((DMCoNo *)userData)[index]; /* check if we've been here before (normal should not be 0) */ if (!is_zero_v3(co_no->no)) { return; } copy_v3_v3(co_no->co, co); if (no_f) { copy_v3_v3(co_no->no, no_f); } else { normal_short_to_float_v3(co_no->no, no_s); } } /* always returns original amount me->totvert of vertices and normals, but fully deformed and subsurfered */ /* this is needed for all code using vertexgroups (no subsurf support) */ /* it stores the normals as floats, but they can still be scaled as shorts (32767 = unit) */ /* in use now by vertex/weight paint and particle generating */ DMCoNo *mesh_get_mapped_verts_nors(Scene *scene, Object *ob) { Mesh *me = ob->data; DerivedMesh *dm; DMCoNo *vertexcosnos; /* lets prevent crashing... */ if (ob->type != OB_MESH || me->totvert == 0) return NULL; dm = mesh_get_derived_final(scene, ob, CD_MASK_BAREMESH | CD_MASK_ORIGINDEX); if (dm->foreachMappedVert) { vertexcosnos = MEM_callocN(sizeof(DMCoNo) * me->totvert, "vertexcosnos map"); dm->foreachMappedVert(dm, make_vertexcosnos__mapFunc, vertexcosnos); } else { DMCoNo *v_co_no = vertexcosnos = MEM_mallocN(sizeof(DMCoNo) * me->totvert, "vertexcosnos map"); int a; for (a = 0; a < me->totvert; a++, v_co_no++) { dm->getVertCo(dm, a, v_co_no->co); dm->getVertNo(dm, a, v_co_no->no); } } dm->release(dm); return vertexcosnos; } #endif /* same as above but for vert coords */ typedef struct { float (*vertexcos)[3]; BLI_bitmap *vertex_visit; } MappedUserData; static void make_vertexcos__mapFunc( void *userData, int index, const float co[3], const float UNUSED(no_f[3]), const short UNUSED(no_s[3])) { MappedUserData *mappedData = (MappedUserData *)userData; if (BLI_BITMAP_TEST(mappedData->vertex_visit, index) == 0) { /* we need coord from prototype vertex, not from copies, * assume they stored in the beginning of vertex array stored in DM * (mirror modifier for eg does this) */ copy_v3_v3(mappedData->vertexcos[index], co); BLI_BITMAP_ENABLE(mappedData->vertex_visit, index); } } void mesh_get_mapped_verts_coords(DerivedMesh *dm, float (*r_cos)[3], const int totcos) { if (dm->foreachMappedVert) { MappedUserData userData; memset(r_cos, 0, sizeof(*r_cos) * totcos); userData.vertexcos = r_cos; userData.vertex_visit = BLI_BITMAP_NEW(totcos, "vertexcos flags"); dm->foreachMappedVert(dm, make_vertexcos__mapFunc, &userData, DM_FOREACH_NOP); MEM_freeN(userData.vertex_visit); } else { int i; for (i = 0; i < totcos; i++) { dm->getVertCo(dm, i, r_cos[i]); } } } /* ******************* GLSL ******************** */ /** \name Tangent Space Calculation * \{ */ /* Necessary complexity to handle looptri's as quads for correct tangents */ #define USE_LOOPTRI_DETECT_QUADS typedef struct { float (*precomputedFaceNormals)[3]; float (*precomputedLoopNormals)[3]; const MLoopTri *looptri; MLoopUV *mloopuv; /* texture coordinates */ MPoly *mpoly; /* indices */ MLoop *mloop; /* indices */ MVert *mvert; /* vertices & normals */ float (*orco)[3]; float (*tangent)[4]; /* destination */ int numTessFaces; #ifdef USE_LOOPTRI_DETECT_QUADS /* map from 'fake' face index to looptri, * quads will point to the first looptri of the quad */ const int *face_as_quad_map; int num_face_as_quad_map; #endif } SGLSLMeshToTangent; /* interface */ #include "mikktspace.h" static int dm_ts_GetNumFaces(const SMikkTSpaceContext *pContext) { SGLSLMeshToTangent *pMesh = pContext->m_pUserData; #ifdef USE_LOOPTRI_DETECT_QUADS return pMesh->num_face_as_quad_map; #else return pMesh->numTessFaces; #endif } static int dm_ts_GetNumVertsOfFace(const SMikkTSpaceContext *pContext, const int face_num) { #ifdef USE_LOOPTRI_DETECT_QUADS SGLSLMeshToTangent *pMesh = pContext->m_pUserData; if (pMesh->face_as_quad_map) { const MLoopTri *lt = &pMesh->looptri[pMesh->face_as_quad_map[face_num]]; const MPoly *mp = &pMesh->mpoly[lt->poly]; if (mp->totloop == 4) { return 4; } } return 3; #else UNUSED_VARS(pContext, face_num); return 3; #endif } static void dm_ts_GetPosition( const SMikkTSpaceContext *pContext, float r_co[3], const int face_num, const int vert_index) { //assert(vert_index >= 0 && vert_index < 4); SGLSLMeshToTangent *pMesh = pContext->m_pUserData; const MLoopTri *lt; int loop_index; const float *co; #ifdef USE_LOOPTRI_DETECT_QUADS if (pMesh->face_as_quad_map) { lt = &pMesh->looptri[pMesh->face_as_quad_map[face_num]]; const MPoly *mp = &pMesh->mpoly[lt->poly]; if (mp->totloop == 4) { loop_index = mp->loopstart + vert_index; goto finally; } /* fall through to regular triangle */ } else { lt = &pMesh->looptri[face_num]; } #else lt = &pMesh->looptri[face_num]; #endif loop_index = lt->tri[vert_index]; finally: co = pMesh->mvert[pMesh->mloop[loop_index].v].co; copy_v3_v3(r_co, co); } static void dm_ts_GetTextureCoordinate( const SMikkTSpaceContext *pContext, float r_uv[2], const int face_num, const int vert_index) { //assert(vert_index >= 0 && vert_index < 4); SGLSLMeshToTangent *pMesh = pContext->m_pUserData; const MLoopTri *lt; int loop_index; #ifdef USE_LOOPTRI_DETECT_QUADS if (pMesh->face_as_quad_map) { lt = &pMesh->looptri[pMesh->face_as_quad_map[face_num]]; const MPoly *mp = &pMesh->mpoly[lt->poly]; if (mp->totloop == 4) { loop_index = mp->loopstart + vert_index; goto finally; } /* fall through to regular triangle */ } else { lt = &pMesh->looptri[face_num]; } #else lt = &pMesh->looptri[face_num]; #endif loop_index = lt->tri[vert_index]; finally: if (pMesh->mloopuv != NULL) { const float *uv = pMesh->mloopuv[loop_index].uv; copy_v2_v2(r_uv, uv); } else { const float *orco = pMesh->orco[pMesh->mloop[loop_index].v]; map_to_sphere(&r_uv[0], &r_uv[1], orco[0], orco[1], orco[2]); } } static void dm_ts_GetNormal( const SMikkTSpaceContext *pContext, float r_no[3], const int face_num, const int vert_index) { //assert(vert_index >= 0 && vert_index < 4); SGLSLMeshToTangent *pMesh = (SGLSLMeshToTangent *) pContext->m_pUserData; const MLoopTri *lt; int loop_index; #ifdef USE_LOOPTRI_DETECT_QUADS if (pMesh->face_as_quad_map) { lt = &pMesh->looptri[pMesh->face_as_quad_map[face_num]]; const MPoly *mp = &pMesh->mpoly[lt->poly]; if (mp->totloop == 4) { loop_index = mp->loopstart + vert_index; goto finally; } /* fall through to regular triangle */ } else { lt = &pMesh->looptri[face_num]; } #else lt = &pMesh->looptri[face_num]; #endif loop_index = lt->tri[vert_index]; finally: if (pMesh->precomputedLoopNormals) { copy_v3_v3(r_no, pMesh->precomputedLoopNormals[loop_index]); } else if ((pMesh->mpoly[lt->poly].flag & ME_SMOOTH) == 0) { /* flat */ if (pMesh->precomputedFaceNormals) { copy_v3_v3(r_no, pMesh->precomputedFaceNormals[lt->poly]); } else { #ifdef USE_LOOPTRI_DETECT_QUADS const MPoly *mp = &pMesh->mpoly[lt->poly]; if (mp->totloop == 4) { normal_quad_v3( r_no, pMesh->mvert[pMesh->mloop[mp->loopstart + 0].v].co, pMesh->mvert[pMesh->mloop[mp->loopstart + 1].v].co, pMesh->mvert[pMesh->mloop[mp->loopstart + 2].v].co, pMesh->mvert[pMesh->mloop[mp->loopstart + 3].v].co); } else #endif { normal_tri_v3( r_no, pMesh->mvert[pMesh->mloop[lt->tri[0]].v].co, pMesh->mvert[pMesh->mloop[lt->tri[1]].v].co, pMesh->mvert[pMesh->mloop[lt->tri[2]].v].co); } } } else { const short *no = pMesh->mvert[pMesh->mloop[loop_index].v].no; normal_short_to_float_v3(r_no, no); } } static void dm_ts_SetTSpace( const SMikkTSpaceContext *pContext, const float fvTangent[3], const float fSign, const int face_num, const int vert_index) { //assert(vert_index >= 0 && vert_index < 4); SGLSLMeshToTangent *pMesh = (SGLSLMeshToTangent *) pContext->m_pUserData; const MLoopTri *lt; int loop_index; #ifdef USE_LOOPTRI_DETECT_QUADS if (pMesh->face_as_quad_map) { lt = &pMesh->looptri[pMesh->face_as_quad_map[face_num]]; const MPoly *mp = &pMesh->mpoly[lt->poly]; if (mp->totloop == 4) { loop_index = mp->loopstart + vert_index; goto finally; } /* fall through to regular triangle */ } else { lt = &pMesh->looptri[face_num]; } #else lt = &pMesh->looptri[face_num]; #endif loop_index = lt->tri[vert_index]; float *pRes; finally: pRes = pMesh->tangent[loop_index]; copy_v3_v3(pRes, fvTangent); pRes[3] = fSign; } void DM_calc_tangents_names_from_gpu( const GPUVertexAttribs *gattribs, char (*tangent_names)[MAX_NAME], int *r_tangent_names_count) { int count = 0; for (int b = 0; b < gattribs->totlayer; b++) { if (gattribs->layer[b].type == CD_TANGENT) { strcpy(tangent_names[count++], gattribs->layer[b].name); } } *r_tangent_names_count = count; } static void DM_calc_loop_tangents_thread(TaskPool * __restrict UNUSED(pool), void *taskdata, int UNUSED(threadid)) { struct SGLSLMeshToTangent *mesh2tangent = taskdata; /* new computation method */ { SMikkTSpaceContext sContext = {NULL}; SMikkTSpaceInterface sInterface = {NULL}; sContext.m_pUserData = mesh2tangent; sContext.m_pInterface = &sInterface; sInterface.m_getNumFaces = dm_ts_GetNumFaces; sInterface.m_getNumVerticesOfFace = dm_ts_GetNumVertsOfFace; sInterface.m_getPosition = dm_ts_GetPosition; sInterface.m_getTexCoord = dm_ts_GetTextureCoordinate; sInterface.m_getNormal = dm_ts_GetNormal; sInterface.m_setTSpaceBasic = dm_ts_SetTSpace; /* 0 if failed */ genTangSpaceDefault(&sContext); } } void DM_add_named_tangent_layer_for_uv( CustomData *uv_data, CustomData *tan_data, int numLoopData, const char *layer_name) { if (CustomData_get_named_layer_index(tan_data, CD_TANGENT, layer_name) == -1 && CustomData_get_named_layer_index(uv_data, CD_MLOOPUV, layer_name) != -1) { CustomData_add_layer_named( tan_data, CD_TANGENT, CD_CALLOC, NULL, numLoopData, layer_name); } } /** * Here we get some useful information such as active uv layer name and search if it is already in tangent_names. * Also, we calculate tangent_mask that works as a descriptor of tangents state. * If tangent_mask has changed, then recalculate tangents. */ void DM_calc_loop_tangents_step_0( const CustomData *loopData, bool calc_active_tangent, const char (*tangent_names)[MAX_NAME], int tangent_names_count, bool *rcalc_act, bool *rcalc_ren, int *ract_uv_n, int *rren_uv_n, char *ract_uv_name, char *rren_uv_name, short *rtangent_mask) { /* Active uv in viewport */ int layer_index = CustomData_get_layer_index(loopData, CD_MLOOPUV); *ract_uv_n = CustomData_get_active_layer(loopData, CD_MLOOPUV); ract_uv_name[0] = 0; if (*ract_uv_n != -1) { strcpy(ract_uv_name, loopData->layers[*ract_uv_n + layer_index].name); } /* Active tangent in render */ *rren_uv_n = CustomData_get_render_layer(loopData, CD_MLOOPUV); rren_uv_name[0] = 0; if (*rren_uv_n != -1) { strcpy(rren_uv_name, loopData->layers[*rren_uv_n + layer_index].name); } /* If active tangent not in tangent_names we take it into account */ *rcalc_act = false; *rcalc_ren = false; for (int i = 0; i < tangent_names_count; i++) { if (tangent_names[i][0] == 0) { calc_active_tangent = true; } } if (calc_active_tangent) { *rcalc_act = true; *rcalc_ren = true; for (int i = 0; i < tangent_names_count; i++) { if (STREQ(ract_uv_name, tangent_names[i])) *rcalc_act = false; if (STREQ(rren_uv_name, tangent_names[i])) *rcalc_ren = false; } } *rtangent_mask = 0; const int uv_layer_num = CustomData_number_of_layers(loopData, CD_MLOOPUV); for (int n = 0; n < uv_layer_num; n++) { const char *name = CustomData_get_layer_name(loopData, CD_MLOOPUV, n); bool add = false; for (int i = 0; i < tangent_names_count; i++) { if (tangent_names[i][0] && STREQ(tangent_names[i], name)) { add = true; break; } } if ((*rcalc_act && ract_uv_name[0] && STREQ(ract_uv_name, name)) || (*rcalc_ren && rren_uv_name[0] && STREQ(rren_uv_name, name))) { add = true; } if (add) *rtangent_mask |= 1 << n; } if (uv_layer_num == 0) *rtangent_mask |= DM_TANGENT_MASK_ORCO; } void DM_calc_loop_tangents( DerivedMesh *dm, bool calc_active_tangent, const char (*tangent_names)[MAX_NAME], int tangent_names_count) { int act_uv_n = -1; int ren_uv_n = -1; bool calc_act = false; bool calc_ren = false; char act_uv_name[MAX_NAME]; char ren_uv_name[MAX_NAME]; short tangent_mask = 0; DM_calc_loop_tangents_step_0( &dm->loopData, calc_active_tangent, tangent_names, tangent_names_count, &calc_act, &calc_ren, &act_uv_n, &ren_uv_n, act_uv_name, ren_uv_name, &tangent_mask); if ((dm->tangent_mask | tangent_mask) != dm->tangent_mask) { /* Check we have all the needed layers */ MPoly *mpoly = dm->getPolyArray(dm); const MLoopTri *looptri = dm->getLoopTriArray(dm); int totface = dm->getNumLoopTri(dm); /* Allocate needed tangent layers */ for (int i = 0; i < tangent_names_count; i++) if (tangent_names[i][0]) DM_add_named_tangent_layer_for_uv(&dm->loopData, &dm->loopData, dm->numLoopData, tangent_names[i]); if ((tangent_mask & DM_TANGENT_MASK_ORCO) && CustomData_get_named_layer_index(&dm->loopData, CD_TANGENT, "") == -1) CustomData_add_layer_named(&dm->loopData, CD_TANGENT, CD_CALLOC, NULL, dm->numLoopData, ""); if (calc_act && act_uv_name[0]) DM_add_named_tangent_layer_for_uv(&dm->loopData, &dm->loopData, dm->numLoopData, act_uv_name); if (calc_ren && ren_uv_name[0]) DM_add_named_tangent_layer_for_uv(&dm->loopData, &dm->loopData, dm->numLoopData, ren_uv_name); #ifdef USE_LOOPTRI_DETECT_QUADS int num_face_as_quad_map; int *face_as_quad_map = NULL; /* map faces to quads */ if (totface != dm->getNumPolys(dm)) { /* over alloc, since we dont know how many ngon or quads we have */ /* map fake face index to looptri */ face_as_quad_map = MEM_mallocN(sizeof(int) * totface, __func__); int k, j; for (k = 0, j = 0; j < totface; k++, j++) { face_as_quad_map[k] = j; /* step over all quads */ if (mpoly[looptri[j].poly].totloop == 4) { j++; /* skips the nest looptri */ } } num_face_as_quad_map = k; } else { num_face_as_quad_map = totface; } #endif /* Calculation */ { TaskScheduler *scheduler = BLI_task_scheduler_get(); TaskPool *task_pool; task_pool = BLI_task_pool_create(scheduler, NULL); dm->tangent_mask = 0; /* Calculate tangent layers */ SGLSLMeshToTangent data_array[MAX_MTFACE]; const int tangent_layer_num = CustomData_number_of_layers(&dm->loopData, CD_TANGENT); for (int n = 0; n < tangent_layer_num; n++) { int index = CustomData_get_layer_index_n(&dm->loopData, CD_TANGENT, n); BLI_assert(n < MAX_MTFACE); SGLSLMeshToTangent *mesh2tangent = &data_array[n]; mesh2tangent->numTessFaces = totface; #ifdef USE_LOOPTRI_DETECT_QUADS mesh2tangent->face_as_quad_map = face_as_quad_map; mesh2tangent->num_face_as_quad_map = num_face_as_quad_map; #endif mesh2tangent->mvert = dm->getVertArray(dm); mesh2tangent->mpoly = dm->getPolyArray(dm); mesh2tangent->mloop = dm->getLoopArray(dm); mesh2tangent->looptri = dm->getLoopTriArray(dm); /* Note, we assume we do have tessellated loop normals at this point (in case it is object-enabled), * have to check this is valid... */ mesh2tangent->precomputedLoopNormals = dm->getLoopDataArray(dm, CD_NORMAL); mesh2tangent->precomputedFaceNormals = CustomData_get_layer(&dm->polyData, CD_NORMAL); mesh2tangent->orco = NULL; mesh2tangent->mloopuv = CustomData_get_layer_named(&dm->loopData, CD_MLOOPUV, dm->loopData.layers[index].name); /* Fill the resulting tangent_mask */ if (!mesh2tangent->mloopuv) { mesh2tangent->orco = dm->getVertDataArray(dm, CD_ORCO); if (!mesh2tangent->orco) continue; dm->tangent_mask |= DM_TANGENT_MASK_ORCO; } else { int uv_ind = CustomData_get_named_layer_index(&dm->loopData, CD_MLOOPUV, dm->loopData.layers[index].name); int uv_start = CustomData_get_layer_index(&dm->loopData, CD_MLOOPUV); BLI_assert(uv_ind != -1 && uv_start != -1); BLI_assert(uv_ind - uv_start < MAX_MTFACE); dm->tangent_mask |= 1 << (uv_ind - uv_start); } mesh2tangent->tangent = dm->loopData.layers[index].data; BLI_task_pool_push(task_pool, DM_calc_loop_tangents_thread, mesh2tangent, false, TASK_PRIORITY_LOW); } BLI_assert(dm->tangent_mask == tangent_mask); BLI_task_pool_work_and_wait(task_pool); BLI_task_pool_free(task_pool); } #ifdef USE_LOOPTRI_DETECT_QUADS if (face_as_quad_map) { MEM_freeN(face_as_quad_map); } #undef USE_LOOPTRI_DETECT_QUADS #endif /* Update active layer index */ int act_uv_index = CustomData_get_layer_index_n(&dm->loopData, CD_MLOOPUV, act_uv_n); if (act_uv_index != -1) { int tan_index = CustomData_get_named_layer_index(&dm->loopData, CD_TANGENT, dm->loopData.layers[act_uv_index].name); CustomData_set_layer_active_index(&dm->loopData, CD_TANGENT, tan_index); } /* else tangent has been built from orco */ /* Update render layer index */ int ren_uv_index = CustomData_get_layer_index_n(&dm->loopData, CD_MLOOPUV, ren_uv_n); if (ren_uv_index != -1) { int tan_index = CustomData_get_named_layer_index(&dm->loopData, CD_TANGENT, dm->loopData.layers[ren_uv_index].name); CustomData_set_layer_render_index(&dm->loopData, CD_TANGENT, tan_index); } /* else tangent has been built from orco */ } } /** \} */ void DM_calc_auto_bump_scale(DerivedMesh *dm) { /* int totvert = dm->getNumVerts(dm); */ /* UNUSED */ int totface = dm->getNumTessFaces(dm); MVert *mvert = dm->getVertArray(dm); MFace *mface = dm->getTessFaceArray(dm); MTFace *mtface = dm->getTessFaceDataArray(dm, CD_MTFACE); if (mtface) { double dsum = 0.0; int nr_accumulated = 0; int f; for (f = 0; f < totface; f++) { { float *verts[4], *tex_coords[4]; const int nr_verts = mface[f].v4 != 0 ? 4 : 3; bool is_degenerate; int i; verts[0] = mvert[mface[f].v1].co; verts[1] = mvert[mface[f].v2].co; verts[2] = mvert[mface[f].v3].co; tex_coords[0] = mtface[f].uv[0]; tex_coords[1] = mtface[f].uv[1]; tex_coords[2] = mtface[f].uv[2]; if (nr_verts == 4) { verts[3] = mvert[mface[f].v4].co; tex_coords[3] = mtface[f].uv[3]; } /* discard degenerate faces */ is_degenerate = 0; if (equals_v3v3(verts[0], verts[1]) || equals_v3v3(verts[0], verts[2]) || equals_v3v3(verts[1], verts[2]) || equals_v2v2(tex_coords[0], tex_coords[1]) || equals_v2v2(tex_coords[0], tex_coords[2]) || equals_v2v2(tex_coords[1], tex_coords[2])) { is_degenerate = 1; } /* verify last vertex as well if this is a quad */ if (is_degenerate == 0 && nr_verts == 4) { if (equals_v3v3(verts[3], verts[0]) || equals_v3v3(verts[3], verts[1]) || equals_v3v3(verts[3], verts[2]) || equals_v2v2(tex_coords[3], tex_coords[0]) || equals_v2v2(tex_coords[3], tex_coords[1]) || equals_v2v2(tex_coords[3], tex_coords[2])) { is_degenerate = 1; } /* verify the winding is consistent */ if (is_degenerate == 0) { float prev_edge[2]; bool is_signed = 0; sub_v2_v2v2(prev_edge, tex_coords[0], tex_coords[3]); i = 0; while (is_degenerate == 0 && i < 4) { float cur_edge[2], signed_area; sub_v2_v2v2(cur_edge, tex_coords[(i + 1) & 0x3], tex_coords[i]); signed_area = cross_v2v2(prev_edge, cur_edge); if (i == 0) { is_signed = (signed_area < 0.0f) ? 1 : 0; } else if ((is_signed != 0) != (signed_area < 0.0f)) { is_degenerate = 1; } if (is_degenerate == 0) { copy_v2_v2(prev_edge, cur_edge); i++; } } } } /* proceed if not a degenerate face */ if (is_degenerate == 0) { int nr_tris_to_pile = 0; /* quads split at shortest diagonal */ int offs = 0; /* initial triangulation is 0,1,2 and 0, 2, 3 */ if (nr_verts == 4) { float pos_len_diag0, pos_len_diag1; pos_len_diag0 = len_squared_v3v3(verts[2], verts[0]); pos_len_diag1 = len_squared_v3v3(verts[3], verts[1]); if (pos_len_diag1 < pos_len_diag0) { offs = 1; // alter split } else if (pos_len_diag0 == pos_len_diag1) { /* do UV check instead */ float tex_len_diag0, tex_len_diag1; tex_len_diag0 = len_squared_v2v2(tex_coords[2], tex_coords[0]); tex_len_diag1 = len_squared_v2v2(tex_coords[3], tex_coords[1]); if (tex_len_diag1 < tex_len_diag0) { offs = 1; /* alter split */ } } } nr_tris_to_pile = nr_verts - 2; if (nr_tris_to_pile == 1 || nr_tris_to_pile == 2) { const int indices[6] = {offs + 0, offs + 1, offs + 2, offs + 0, offs + 2, (offs + 3) & 0x3 }; int t; for (t = 0; t < nr_tris_to_pile; t++) { float f2x_area_uv; const float *p0 = verts[indices[t * 3 + 0]]; const float *p1 = verts[indices[t * 3 + 1]]; const float *p2 = verts[indices[t * 3 + 2]]; float edge_t0[2], edge_t1[2]; sub_v2_v2v2(edge_t0, tex_coords[indices[t * 3 + 1]], tex_coords[indices[t * 3 + 0]]); sub_v2_v2v2(edge_t1, tex_coords[indices[t * 3 + 2]], tex_coords[indices[t * 3 + 0]]); f2x_area_uv = fabsf(cross_v2v2(edge_t0, edge_t1)); if (f2x_area_uv > FLT_EPSILON) { float norm[3], v0[3], v1[3], f2x_surf_area, fsurf_ratio; sub_v3_v3v3(v0, p1, p0); sub_v3_v3v3(v1, p2, p0); cross_v3_v3v3(norm, v0, v1); f2x_surf_area = len_v3(norm); fsurf_ratio = f2x_surf_area / f2x_area_uv; /* tri area divided by texture area */ nr_accumulated++; dsum += (double)(fsurf_ratio); } } } } } } /* finalize */ { const float avg_area_ratio = (nr_accumulated > 0) ? ((float)(dsum / nr_accumulated)) : 1.0f; const float use_as_render_bump_scale = sqrtf(avg_area_ratio); // use width of average surface ratio as your bump scale dm->auto_bump_scale = use_as_render_bump_scale; } } else { dm->auto_bump_scale = 1.0f; } } void DM_vertex_attributes_from_gpu(DerivedMesh *dm, GPUVertexAttribs *gattribs, DMVertexAttribs *attribs) { CustomData *vdata, *ldata; int a, b, layer; const bool is_editmesh = (dm->type == DM_TYPE_EDITBMESH); /* From the layers requested by the GLSL shader, figure out which ones are * actually available for this derivedmesh, and retrieve the pointers */ memset(attribs, 0, sizeof(DMVertexAttribs)); vdata = &dm->vertData; ldata = dm->getLoopDataLayout(dm); /* calc auto bump scale if necessary */ if (dm->auto_bump_scale <= 0.0f) DM_calc_auto_bump_scale(dm); char tangent_names[MAX_MTFACE][MAX_NAME]; int tangent_names_count; /* Add a tangent layer/layers. */ DM_calc_tangents_names_from_gpu(gattribs, tangent_names, &tangent_names_count); if (tangent_names_count) dm->calcLoopTangents(dm, false, (const char (*)[MAX_NAME])tangent_names, tangent_names_count); for (b = 0; b < gattribs->totlayer; b++) { int type = gattribs->layer[b].type; layer = -1; if (type == CD_AUTO_FROM_NAME) { /* We need to deduct what exact layer is used. * * We do it based on the specified name. */ if (gattribs->layer[b].name[0]) { layer = CustomData_get_named_layer_index(ldata, CD_MLOOPUV, gattribs->layer[b].name); type = CD_MTFACE; if (layer == -1) { layer = CustomData_get_named_layer_index(ldata, CD_MLOOPCOL, gattribs->layer[b].name); type = CD_MCOL; } if (layer == -1) { layer = CustomData_get_named_layer_index(&dm->loopData, CD_TANGENT, gattribs->layer[b].name); type = CD_TANGENT; } if (layer == -1) { continue; } } else { /* Fall back to the UV layer, which matches old behavior. */ type = CD_MTFACE; } } if (type == CD_MTFACE) { /* uv coordinates */ if (layer == -1) { if (gattribs->layer[b].name[0]) layer = CustomData_get_named_layer_index(ldata, CD_MLOOPUV, gattribs->layer[b].name); else layer = CustomData_get_active_layer_index(ldata, CD_MLOOPUV); } a = attribs->tottface++; if (layer != -1) { attribs->tface[a].array = is_editmesh ? NULL : ldata->layers[layer].data; attribs->tface[a].em_offset = ldata->layers[layer].offset; } else { attribs->tface[a].array = NULL; attribs->tface[a].em_offset = -1; } attribs->tface[a].gl_index = gattribs->layer[b].glindex; attribs->tface[a].gl_info_index = gattribs->layer[b].glinfoindoex; attribs->tface[a].gl_texco = gattribs->layer[b].gltexco; } else if (type == CD_MCOL) { if (layer == -1) { if (gattribs->layer[b].name[0]) layer = CustomData_get_named_layer_index(ldata, CD_MLOOPCOL, gattribs->layer[b].name); else layer = CustomData_get_active_layer_index(ldata, CD_MLOOPCOL); } a = attribs->totmcol++; if (layer != -1) { attribs->mcol[a].array = is_editmesh ? NULL : ldata->layers[layer].data; /* odd, store the offset for a different layer type here, but editmode draw code expects it */ attribs->mcol[a].em_offset = ldata->layers[layer].offset; } else { attribs->mcol[a].array = NULL; attribs->mcol[a].em_offset = -1; } attribs->mcol[a].gl_index = gattribs->layer[b].glindex; attribs->mcol[a].gl_info_index = gattribs->layer[b].glinfoindoex; } else if (type == CD_TANGENT) { /* note, even with 'is_editmesh' this uses the derived-meshes loop data */ if (layer == -1) { if (gattribs->layer[b].name[0]) layer = CustomData_get_named_layer_index(&dm->loopData, CD_TANGENT, gattribs->layer[b].name); else layer = CustomData_get_active_layer_index(&dm->loopData, CD_TANGENT); } a = attribs->tottang++; if (layer != -1) { attribs->tang[a].array = dm->loopData.layers[layer].data; attribs->tang[a].em_offset = dm->loopData.layers[layer].offset; } else { attribs->tang[a].array = NULL; attribs->tang[a].em_offset = -1; } attribs->tang[a].gl_index = gattribs->layer[b].glindex; attribs->tang[a].gl_info_index = gattribs->layer[b].glinfoindoex; } else if (type == CD_ORCO) { /* original coordinates */ if (layer == -1) { layer = CustomData_get_layer_index(vdata, CD_ORCO); } attribs->totorco = 1; if (layer != -1) { attribs->orco.array = vdata->layers[layer].data; attribs->orco.em_offset = vdata->layers[layer].offset; } else { attribs->orco.array = NULL; attribs->orco.em_offset = -1; } attribs->orco.gl_index = gattribs->layer[b].glindex; attribs->orco.gl_texco = gattribs->layer[b].gltexco; attribs->orco.gl_info_index = gattribs->layer[b].glinfoindoex; } } } /** * Set vertex shader attribute inputs for a particular tessface vert * * \param a: tessface index * \param index: vertex index * \param vert: corner index (0, 1, 2, 3) * \param loop: absolute loop corner index */ void DM_draw_attrib_vertex(DMVertexAttribs *attribs, int a, int index, int vert, int loop) { const float zero[4] = {0.0f, 0.0f, 0.0f, 0.0f}; int b; UNUSED_VARS(a, vert); /* orco texture coordinates */ if (attribs->totorco) { /*const*/ float (*array)[3] = attribs->orco.array; const float *orco = (array) ? array[index] : zero; if (attribs->orco.gl_texco) glTexCoord3fv(orco); else glVertexAttrib3fv(attribs->orco.gl_index, orco); } /* uv texture coordinates */ for (b = 0; b < attribs->tottface; b++) { const float *uv; if (attribs->tface[b].array) { const MLoopUV *mloopuv = &attribs->tface[b].array[loop]; uv = mloopuv->uv; } else { uv = zero; } if (attribs->tface[b].gl_texco) glTexCoord2fv(uv); else glVertexAttrib2fv(attribs->tface[b].gl_index, uv); } /* vertex colors */ for (b = 0; b < attribs->totmcol; b++) { GLfloat col[4]; if (attribs->mcol[b].array) { const MLoopCol *cp = &attribs->mcol[b].array[loop]; rgba_uchar_to_float(col, &cp->r); } else { zero_v4(col); } glVertexAttrib4fv(attribs->mcol[b].gl_index, col); } /* tangent for normal mapping */ for (b = 0; b < attribs->tottang; b++) { if (attribs->tang[b].array) { /*const*/ float (*array)[4] = attribs->tang[b].array; const float *tang = (array) ? array[loop] : zero; glVertexAttrib4fv(attribs->tang[b].gl_index, tang); } } } void DM_draw_attrib_vertex_uniforms(const DMVertexAttribs *attribs) { int i; if (attribs->totorco) { glUniform1i(attribs->orco.gl_info_index, 0); } for (i = 0; i < attribs->tottface; i++) { glUniform1i(attribs->tface[i].gl_info_index, 0); } for (i = 0; i < attribs->totmcol; i++) { glUniform1i(attribs->mcol[i].gl_info_index, GPU_ATTR_INFO_SRGB); } for (i = 0; i < attribs->tottang; i++) { glUniform1i(attribs->tang[i].gl_info_index, 0); } } /* Set object's bounding box based on DerivedMesh min/max data */ void DM_set_object_boundbox(Object *ob, DerivedMesh *dm) { float min[3], max[3]; INIT_MINMAX(min, max); dm->getMinMax(dm, min, max); if (!ob->bb) ob->bb = MEM_callocN(sizeof(BoundBox), "DM-BoundBox"); BKE_boundbox_init_from_minmax(ob->bb, min, max); ob->bb->flag &= ~BOUNDBOX_DIRTY; } /* --- NAVMESH (begin) --- */ #ifdef WITH_GAMEENGINE /* BMESH_TODO, navmesh is not working right currently * All tools set this as MPoly data, but derived mesh currently draws from MFace (tessface) * * Proposed solution, rather then copy CD_RECAST into the MFace array, * use ORIGINDEX to get the original poly index and then get the CD_RECAST * data from the original me->mpoly layer. - campbell */ BLI_INLINE int navmesh_bit(int a, int b) { return (a & (1 << b)) >> b; } BLI_INLINE void navmesh_intToCol(int i, float col[3]) { int r = navmesh_bit(i, 0) + navmesh_bit(i, 3) * 2 + 1; int g = navmesh_bit(i, 1) + navmesh_bit(i, 4) * 2 + 1; int b = navmesh_bit(i, 2) + navmesh_bit(i, 5) * 2 + 1; col[0] = 1 - r * 63.0f / 255.0f; col[1] = 1 - g * 63.0f / 255.0f; col[2] = 1 - b * 63.0f / 255.0f; } static void navmesh_drawColored(DerivedMesh *dm) { int a, glmode; MVert *mvert = (MVert *)CustomData_get_layer(&dm->vertData, CD_MVERT); MFace *mface = (MFace *)CustomData_get_layer(&dm->faceData, CD_MFACE); int *polygonIdx = (int *)CustomData_get_layer(&dm->polyData, CD_RECAST); float col[3]; if (!polygonIdx) return; #if 0 //UI_ThemeColor(TH_WIRE); glLineWidth(2.0); dm->drawEdges(dm, 0, 1); #endif /* if (GPU_buffer_legacy(dm) ) */ /* TODO - VBO draw code, not high priority - campbell */ { DEBUG_VBO("Using legacy code. drawNavMeshColored\n"); glBegin(glmode = GL_QUADS); for (a = 0; a < dm->numTessFaceData; a++, mface++) { int new_glmode = mface->v4 ? GL_QUADS : GL_TRIANGLES; int pi = polygonIdx[a]; if (pi <= 0) { zero_v3(col); } else { navmesh_intToCol(pi, col); } if (new_glmode != glmode) { glEnd(); glBegin(glmode = new_glmode); } glColor3fv(col); glVertex3fv(mvert[mface->v1].co); glVertex3fv(mvert[mface->v2].co); glVertex3fv(mvert[mface->v3].co); if (mface->v4) { glVertex3fv(mvert[mface->v4].co); } } glEnd(); } } static void navmesh_DM_drawFacesTex( DerivedMesh *dm, DMSetDrawOptionsTex UNUSED(setDrawOptions), DMCompareDrawOptions UNUSED(compareDrawOptions), void *UNUSED(userData), DMDrawFlag UNUSED(flag)) { navmesh_drawColored(dm); } static void navmesh_DM_drawFacesSolid( DerivedMesh *dm, float (*partial_redraw_planes)[4], bool UNUSED(fast), DMSetMaterial UNUSED(setMaterial)) { UNUSED_VARS(partial_redraw_planes); //drawFacesSolid_original(dm, partial_redraw_planes, fast, setMaterial); navmesh_drawColored(dm); } static DerivedMesh *navmesh_dm_createNavMeshForVisualization(DerivedMesh *dm) { DerivedMesh *result; int maxFaces = dm->getNumPolys(dm); int *recastData; int vertsPerPoly = 0, nverts = 0, ndtris = 0, npolys = 0; float *verts = NULL; unsigned short *dtris = NULL, *dmeshes = NULL, *polys = NULL; int *dtrisToPolysMap = NULL, *dtrisToTrisMap = NULL, *trisToFacesMap = NULL; int res; result = CDDM_copy(dm); if (!CustomData_has_layer(&result->polyData, CD_RECAST)) { int *sourceRecastData = (int *)CustomData_get_layer(&dm->polyData, CD_RECAST); if (sourceRecastData) { CustomData_add_layer_named(&result->polyData, CD_RECAST, CD_DUPLICATE, sourceRecastData, maxFaces, "recastData"); } } recastData = (int *)CustomData_get_layer(&result->polyData, CD_RECAST); /* note: This is not good design! - really should not be doing this */ result->drawFacesTex = navmesh_DM_drawFacesTex; result->drawFacesSolid = navmesh_DM_drawFacesSolid; /* process mesh */ res = buildNavMeshDataByDerivedMesh(dm, &vertsPerPoly, &nverts, &verts, &ndtris, &dtris, &npolys, &dmeshes, &polys, &dtrisToPolysMap, &dtrisToTrisMap, &trisToFacesMap); if (res) { size_t polyIdx; /* invalidate concave polygon */ for (polyIdx = 0; polyIdx < (size_t)npolys; polyIdx++) { unsigned short *poly = &polys[polyIdx * 2 * vertsPerPoly]; if (!polyIsConvex(poly, vertsPerPoly, verts)) { /* set negative polygon idx to all faces */ unsigned short *dmesh = &dmeshes[4 * polyIdx]; unsigned short tbase = dmesh[2]; unsigned short tnum = dmesh[3]; unsigned short ti; for (ti = 0; ti < tnum; ti++) { unsigned short triidx = dtrisToTrisMap[tbase + ti]; unsigned short faceidx = trisToFacesMap[triidx]; if (recastData[faceidx] > 0) { recastData[faceidx] = -recastData[faceidx]; } } } } } else { printf("Navmesh: Unable to generate valid Navmesh"); } /* clean up */ if (verts != NULL) MEM_freeN(verts); if (dtris != NULL) MEM_freeN(dtris); if (dmeshes != NULL) MEM_freeN(dmeshes); if (polys != NULL) MEM_freeN(polys); if (dtrisToPolysMap != NULL) MEM_freeN(dtrisToPolysMap); if (dtrisToTrisMap != NULL) MEM_freeN(dtrisToTrisMap); if (trisToFacesMap != NULL) MEM_freeN(trisToFacesMap); return result; } #endif /* WITH_GAMEENGINE */ /* --- NAVMESH (end) --- */ void DM_init_origspace(DerivedMesh *dm) { const float default_osf[4][2] = {{0, 0}, {1, 0}, {1, 1}, {0, 1}}; OrigSpaceLoop *lof_array = CustomData_get_layer(&dm->loopData, CD_ORIGSPACE_MLOOP); const int numpoly = dm->getNumPolys(dm); // const int numloop = dm->getNumLoops(dm); MVert *mv = dm->getVertArray(dm); MLoop *ml = dm->getLoopArray(dm); MPoly *mp = dm->getPolyArray(dm); int i, j, k; float (*vcos_2d)[2] = NULL; BLI_array_staticdeclare(vcos_2d, 64); for (i = 0; i < numpoly; i++, mp++) { OrigSpaceLoop *lof = lof_array + mp->loopstart; if (mp->totloop == 3 || mp->totloop == 4) { for (j = 0; j < mp->totloop; j++, lof++) { copy_v2_v2(lof->uv, default_osf[j]); } } else { MLoop *l = &ml[mp->loopstart]; float p_nor[3], co[3]; float mat[3][3]; float min[2] = {FLT_MAX, FLT_MAX}, max[2] = {-FLT_MAX, -FLT_MAX}; float translate[2], scale[2]; BKE_mesh_calc_poly_normal(mp, l, mv, p_nor); axis_dominant_v3_to_m3(mat, p_nor); BLI_array_empty(vcos_2d); BLI_array_reserve(vcos_2d, mp->totloop); for (j = 0; j < mp->totloop; j++, l++) { mul_v3_m3v3(co, mat, mv[l->v].co); copy_v2_v2(vcos_2d[j], co); for (k = 0; k < 2; k++) { if (co[k] > max[k]) max[k] = co[k]; else if (co[k] < min[k]) min[k] = co[k]; } } /* Brings min to (0, 0). */ negate_v2_v2(translate, min); /* Scale will bring max to (1, 1). */ sub_v2_v2v2(scale, max, min); if (scale[0] == 0.0f) scale[0] = 1e-9f; if (scale[1] == 0.0f) scale[1] = 1e-9f; invert_v2(scale); /* Finally, transform all vcos_2d into ((0, 0), (1, 1)) square and assing them as origspace. */ for (j = 0; j < mp->totloop; j++, lof++) { add_v2_v2v2(lof->uv, vcos_2d[j], translate); mul_v2_v2(lof->uv, scale); } } } dm->dirty |= DM_DIRTY_TESS_CDLAYERS; BLI_array_free(vcos_2d); } /* derivedmesh info printing function, * to help track down differences DM output */ #ifndef NDEBUG #include "BLI_dynstr.h" static void dm_debug_info_layers( DynStr *dynstr, DerivedMesh *dm, CustomData *cd, void *(*getElemDataArray)(DerivedMesh *, int)) { int type; for (type = 0; type < CD_NUMTYPES; type++) { if (CustomData_has_layer(cd, type)) { /* note: doesnt account for multiple layers */ const char *name = CustomData_layertype_name(type); const int size = CustomData_sizeof(type); const void *pt = getElemDataArray(dm, type); const int pt_size = pt ? (int)(MEM_allocN_len(pt) / size) : 0; const char *structname; int structnum; CustomData_file_write_info(type, &structname, &structnum); BLI_dynstr_appendf(dynstr, " dict(name='%s', struct='%s', type=%d, ptr='%p', elem=%d, length=%d),\n", name, structname, type, (const void *)pt, size, pt_size); } } } char *DM_debug_info(DerivedMesh *dm) { DynStr *dynstr = BLI_dynstr_new(); char *ret; const char *tstr; BLI_dynstr_appendf(dynstr, "{\n"); BLI_dynstr_appendf(dynstr, " 'ptr': '%p',\n", (void *)dm); switch (dm->type) { case DM_TYPE_CDDM: tstr = "DM_TYPE_CDDM"; break; case DM_TYPE_EDITBMESH: tstr = "DM_TYPE_EDITMESH"; break; case DM_TYPE_CCGDM: tstr = "DM_TYPE_CCGDM"; break; default: tstr = "UNKNOWN"; break; } BLI_dynstr_appendf(dynstr, " 'type': '%s',\n", tstr); BLI_dynstr_appendf(dynstr, " 'numVertData': %d,\n", dm->numVertData); BLI_dynstr_appendf(dynstr, " 'numEdgeData': %d,\n", dm->numEdgeData); BLI_dynstr_appendf(dynstr, " 'numTessFaceData': %d,\n", dm->numTessFaceData); BLI_dynstr_appendf(dynstr, " 'numPolyData': %d,\n", dm->numPolyData); BLI_dynstr_appendf(dynstr, " 'deformedOnly': %d,\n", dm->deformedOnly); BLI_dynstr_appendf(dynstr, " 'vertexLayers': (\n"); dm_debug_info_layers(dynstr, dm, &dm->vertData, dm->getVertDataArray); BLI_dynstr_appendf(dynstr, " ),\n"); BLI_dynstr_appendf(dynstr, " 'edgeLayers': (\n"); dm_debug_info_layers(dynstr, dm, &dm->edgeData, dm->getEdgeDataArray); BLI_dynstr_appendf(dynstr, " ),\n"); BLI_dynstr_appendf(dynstr, " 'loopLayers': (\n"); dm_debug_info_layers(dynstr, dm, &dm->loopData, dm->getLoopDataArray); BLI_dynstr_appendf(dynstr, " ),\n"); BLI_dynstr_appendf(dynstr, " 'polyLayers': (\n"); dm_debug_info_layers(dynstr, dm, &dm->polyData, dm->getPolyDataArray); BLI_dynstr_appendf(dynstr, " ),\n"); BLI_dynstr_appendf(dynstr, " 'tessFaceLayers': (\n"); dm_debug_info_layers(dynstr, dm, &dm->faceData, dm->getTessFaceDataArray); BLI_dynstr_appendf(dynstr, " ),\n"); BLI_dynstr_appendf(dynstr, "}\n"); ret = BLI_dynstr_get_cstring(dynstr); BLI_dynstr_free(dynstr); return ret; } void DM_debug_print(DerivedMesh *dm) { char *str = DM_debug_info(dm); puts(str); fflush(stdout); MEM_freeN(str); } void DM_debug_print_cdlayers(CustomData *data) { int i; const CustomDataLayer *layer; printf("{\n"); for (i = 0, layer = data->layers; i < data->totlayer; i++, layer++) { const char *name = CustomData_layertype_name(layer->type); const int size = CustomData_sizeof(layer->type); const char *structname; int structnum; CustomData_file_write_info(layer->type, &structname, &structnum); printf(" dict(name='%s', struct='%s', type=%d, ptr='%p', elem=%d, length=%d),\n", name, structname, layer->type, (const void *)layer->data, size, (int)(MEM_allocN_len(layer->data) / size)); } printf("}\n"); } bool DM_is_valid(DerivedMesh *dm) { const bool do_verbose = true; const bool do_fixes = false; bool is_valid = true; bool changed = true; is_valid &= BKE_mesh_validate_all_customdata( dm->getVertDataLayout(dm), dm->getEdgeDataLayout(dm), dm->getLoopDataLayout(dm), dm->getPolyDataLayout(dm), false, /* setting mask here isn't useful, gives false positives */ do_verbose, do_fixes, &changed); is_valid &= BKE_mesh_validate_arrays( NULL, dm->getVertArray(dm), dm->getNumVerts(dm), dm->getEdgeArray(dm), dm->getNumEdges(dm), dm->getTessFaceArray(dm), dm->getNumTessFaces(dm), dm->getLoopArray(dm), dm->getNumLoops(dm), dm->getPolyArray(dm), dm->getNumPolys(dm), dm->getVertDataArray(dm, CD_MDEFORMVERT), do_verbose, do_fixes, &changed); BLI_assert(changed == false); return is_valid; } #endif /* NDEBUG */ /* -------------------------------------------------------------------- */ MVert *DM_get_vert_array(DerivedMesh *dm, bool *allocated) { CustomData *vert_data = dm->getVertDataLayout(dm); MVert *mvert = CustomData_get_layer(vert_data, CD_MVERT); *allocated = false; if (mvert == NULL) { mvert = MEM_mallocN(sizeof(MVert) * dm->getNumVerts(dm), "dmvh vert data array"); dm->copyVertArray(dm, mvert); *allocated = true; } return mvert; } MEdge *DM_get_edge_array(DerivedMesh *dm, bool *allocated) { CustomData *edge_data = dm->getEdgeDataLayout(dm); MEdge *medge = CustomData_get_layer(edge_data, CD_MEDGE); *allocated = false; if (medge == NULL) { medge = MEM_mallocN(sizeof(MEdge) * dm->getNumEdges(dm), "dm medge data array"); dm->copyEdgeArray(dm, medge); *allocated = true; } return medge; } MLoop *DM_get_loop_array(DerivedMesh *dm, bool *r_allocated) { CustomData *loop_data = dm->getLoopDataLayout(dm); MLoop *mloop = CustomData_get_layer(loop_data, CD_MLOOP); *r_allocated = false; if (mloop == NULL) { mloop = MEM_mallocN(sizeof(MLoop) * dm->getNumLoops(dm), "dm loop data array"); dm->copyLoopArray(dm, mloop); *r_allocated = true; } return mloop; } MPoly *DM_get_poly_array(DerivedMesh *dm, bool *r_allocated) { CustomData *poly_data = dm->getPolyDataLayout(dm); MPoly *mpoly = CustomData_get_layer(poly_data, CD_MPOLY); *r_allocated = false; if (mpoly == NULL) { mpoly = MEM_mallocN(sizeof(MPoly) * dm->getNumPolys(dm), "dm poly data array"); dm->copyPolyArray(dm, mpoly); *r_allocated = true; } return mpoly; } MFace *DM_get_tessface_array(DerivedMesh *dm, bool *r_allocated) { CustomData *tessface_data = dm->getTessFaceDataLayout(dm); MFace *mface = CustomData_get_layer(tessface_data, CD_MFACE); *r_allocated = false; if (mface == NULL) { int numTessFaces = dm->getNumTessFaces(dm); if (numTessFaces > 0) { mface = MEM_mallocN(sizeof(MFace) * numTessFaces, "bvh mface data array"); dm->copyTessFaceArray(dm, mface); *r_allocated = true; } } return mface; }