Fix modifier freeing code re. ID refcounting.
[blender.git] / source / blender / modifiers / intern / MOD_weightvgproximity.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software  Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2011 by Bastien Montagne.
19  * All rights reserved.
20  *
21  * Contributor(s): None yet.
22  *
23  * ***** END GPL LICENSE BLOCK *****
24  *
25  */
26
27 /** \file blender/modifiers/intern/MOD_weightvgproximity.c
28  *  \ingroup modifiers
29  */
30
31 #include "BLI_utildefines.h"
32 #include "BLI_ghash.h"
33 #include "BLI_listbase.h"
34 #include "BLI_math.h"
35 #include "BLI_rand.h"
36 #include "BLI_task.h"
37
38 #include "DNA_mesh_types.h"
39 #include "DNA_meshdata_types.h"
40 #include "DNA_modifier_types.h"
41 #include "DNA_object_types.h"
42
43 #include "BKE_cdderivedmesh.h"
44 #include "BKE_deform.h"
45 #include "BKE_library.h"
46 #include "BKE_library_query.h"
47 #include "BKE_modifier.h"
48 #include "BKE_texture.h"          /* Texture masking. */
49
50 #include "depsgraph_private.h"
51 #include "DEG_depsgraph_build.h"
52
53 #include "MEM_guardedalloc.h"
54
55 #include "MOD_weightvg_util.h"
56 #include "MOD_modifiertypes.h"
57
58 //#define USE_TIMEIT
59
60 #ifdef USE_TIMEIT
61 #  include "PIL_time.h"
62 #  include "PIL_time_utildefines.h"
63 #endif
64
65 /**************************************
66  * Util functions.                    *
67  **************************************/
68
69 /* Util macro. */
70 #define OUT_OF_MEMORY() ((void)printf("WeightVGProximity: Out of memory.\n"))
71
72 typedef struct Vert2GeomData {
73         /* Read-only data */
74         float (*v_cos)[3];
75
76         const SpaceTransform *loc2trgt;
77
78         BVHTreeFromMesh *treeData[3];
79
80         /* Write data, but not needing locking (two different threads will never write same index). */
81         float *dist[3];
82 } Vert2GeomData;
83
84 /* Data which is localized to each computed chunk (i.e. thread-safe, and with continous subset of index range). */
85 typedef struct Vert2GeomDataChunk {
86         /* Read-only data */
87         float last_hit_co[3][3];
88         bool is_init[3];
89 } Vert2GeomDataChunk;
90
91 /**
92  * Callback used by BLI_task 'for loop' helper.
93  */
94 static void vert2geom_task_cb_ex(
95         void *__restrict userdata,
96         const int iter,
97         const ParallelRangeTLS *__restrict tls)
98 {
99         Vert2GeomData *data = userdata;
100         Vert2GeomDataChunk *data_chunk = tls->userdata_chunk;
101
102         float tmp_co[3];
103         int i;
104
105         /* Convert the vertex to tree coordinates. */
106         copy_v3_v3(tmp_co, data->v_cos[iter]);
107         BLI_space_transform_apply(data->loc2trgt, tmp_co);
108
109         for (i = 0; i < ARRAY_SIZE(data->dist); i++) {
110                 if (data->dist[i]) {
111                         BVHTreeNearest nearest = {0};
112
113                         /* Note that we use local proximity heuristics (to reduce the nearest search).
114                          *
115                          * If we already had an hit before in same chunk of tasks (i.e. previous vertex by index),
116                          * we assume this vertex is going to have a close hit to that other vertex, so we can initiate
117                          * the "nearest.dist" with the expected value to that last hit.
118                          * This will lead in pruning of the search tree.
119                          */
120                         nearest.dist_sq = data_chunk->is_init[i] ? len_squared_v3v3(tmp_co, data_chunk->last_hit_co[i]) : FLT_MAX;
121                         nearest.index = -1;
122
123                         /* Compute and store result. If invalid (-1 idx), keep FLT_MAX dist. */
124                         BLI_bvhtree_find_nearest(data->treeData[i]->tree, tmp_co, &nearest,
125                                                  data->treeData[i]->nearest_callback, data->treeData[i]);
126                         data->dist[i][iter] = sqrtf(nearest.dist_sq);
127
128                         if (nearest.index != -1) {
129                                 copy_v3_v3(data_chunk->last_hit_co[i], nearest.co);
130                                 data_chunk->is_init[i] = true;
131                         }
132                 }
133         }
134 }
135
136 /**
137  * Find nearest vertex and/or edge and/or face, for each vertex (adapted from shrinkwrap.c).
138  */
139 static void get_vert2geom_distance(int numVerts, float (*v_cos)[3],
140                                    float *dist_v, float *dist_e, float *dist_f,
141                                    DerivedMesh *target, const SpaceTransform *loc2trgt)
142 {
143         Vert2GeomData data = {0};
144         Vert2GeomDataChunk data_chunk = {{{0}}};
145
146         BVHTreeFromMesh treeData_v = {NULL};
147         BVHTreeFromMesh treeData_e = {NULL};
148         BVHTreeFromMesh treeData_f = {NULL};
149
150         if (dist_v) {
151                 /* Create a bvh-tree of the given target's verts. */
152                 bvhtree_from_mesh_verts(&treeData_v, target, 0.0, 2, 6);
153                 if (treeData_v.tree == NULL) {
154                         OUT_OF_MEMORY();
155                         return;
156                 }
157         }
158         if (dist_e) {
159                 /* Create a bvh-tree of the given target's edges. */
160                 bvhtree_from_mesh_edges(&treeData_e, target, 0.0, 2, 6);
161                 if (treeData_e.tree == NULL) {
162                         OUT_OF_MEMORY();
163                         return;
164                 }
165         }
166         if (dist_f) {
167                 /* Create a bvh-tree of the given target's faces. */
168                 bvhtree_from_mesh_looptri(&treeData_f, target, 0.0, 2, 6);
169                 if (treeData_f.tree == NULL) {
170                         OUT_OF_MEMORY();
171                         return;
172                 }
173         }
174
175         data.v_cos = v_cos;
176         data.loc2trgt = loc2trgt;
177         data.treeData[0] = &treeData_v;
178         data.treeData[1] = &treeData_e;
179         data.treeData[2] = &treeData_f;
180         data.dist[0] = dist_v;
181         data.dist[1] = dist_e;
182         data.dist[2] = dist_f;
183
184         ParallelRangeSettings settings;
185         BLI_parallel_range_settings_defaults(&settings);
186         settings.use_threading = (numVerts > 10000);
187         settings.userdata_chunk = &data_chunk;
188         settings.userdata_chunk_size = sizeof(data_chunk);
189         BLI_task_parallel_range(
190                 0, numVerts,
191                 &data,
192                 vert2geom_task_cb_ex,
193                 &settings);
194
195         if (dist_v)
196                 free_bvhtree_from_mesh(&treeData_v);
197         if (dist_e)
198                 free_bvhtree_from_mesh(&treeData_e);
199         if (dist_f)
200                 free_bvhtree_from_mesh(&treeData_f);
201 }
202
203 /**
204  * Returns the real distance between a vertex and another reference object.
205  * Note that it works in final world space (i.e. with constraints etc. applied).
206  */
207 static void get_vert2ob_distance(int numVerts, float (*v_cos)[3], float *dist,
208                                  Object *ob, Object *obr)
209 {
210         /* Vertex and ref object coordinates. */
211         float v_wco[3];
212         unsigned int i = numVerts;
213
214         while (i-- > 0) {
215                 /* Get world-coordinates of the vertex (constraints and anim included). */
216                 mul_v3_m4v3(v_wco, ob->obmat, v_cos[i]);
217                 /* Return distance between both coordinates. */
218                 dist[i] = len_v3v3(v_wco, obr->obmat[3]);
219         }
220 }
221
222 /**
223  * Returns the real distance between an object and another reference object.
224  * Note that it works in final world space (i.e. with constraints etc. applied).
225  */
226 static float get_ob2ob_distance(const Object *ob, const Object *obr)
227 {
228         return len_v3v3(ob->obmat[3], obr->obmat[3]); 
229 }
230
231 /**
232  * Maps distances to weights, with an optional "smoothing" mapping.
233  */
234 static void do_map(Object *ob, float *weights, const int nidx, const float min_d, const float max_d, short mode)
235 {
236         const float range_inv = 1.0f / (max_d - min_d); /* invert since multiplication is faster */
237         unsigned int i = nidx;
238         if (max_d == min_d) {
239                 while (i-- > 0) {
240                         weights[i] = (weights[i] >= max_d) ? 1.0f : 0.0f; /* "Step" behavior... */
241                 }
242         }
243         else if (max_d > min_d) {
244                 while (i-- > 0) {
245                         if     (weights[i] >= max_d) weights[i] = 1.0f;  /* most likely case first */
246                         else if (weights[i] <= min_d) weights[i] = 0.0f;
247                         else weights[i] = (weights[i] - min_d) * range_inv;
248                 }
249         }
250         else {
251                 while (i-- > 0) {
252                         if     (weights[i] <= max_d) weights[i] = 1.0f;  /* most likely case first */
253                         else if (weights[i] >= min_d) weights[i] = 0.0f;
254                         else weights[i] = (weights[i] - min_d) * range_inv;
255                 }
256         }
257
258         if (!ELEM(mode, MOD_WVG_MAPPING_NONE, MOD_WVG_MAPPING_CURVE)) {
259                 RNG *rng = NULL;
260
261                 if (mode == MOD_WVG_MAPPING_RANDOM)
262                         rng = BLI_rng_new_srandom(BLI_ghashutil_strhash(ob->id.name + 2));
263
264                 weightvg_do_map(nidx, weights, mode, NULL, rng);
265
266                 if (rng)
267                         BLI_rng_free(rng);
268         }
269 }
270
271 /**************************************
272  * Modifiers functions.               *
273  **************************************/
274 static void initData(ModifierData *md)
275 {
276         WeightVGProximityModifierData *wmd = (WeightVGProximityModifierData *) md;
277
278         wmd->proximity_mode       = MOD_WVG_PROXIMITY_OBJECT;
279         wmd->proximity_flags      = MOD_WVG_PROXIMITY_GEOM_VERTS;
280
281         wmd->falloff_type         = MOD_WVG_MAPPING_NONE;
282
283         wmd->mask_constant        = 1.0f;
284         wmd->mask_tex_use_channel = MOD_WVG_MASK_TEX_USE_INT; /* Use intensity by default. */
285         wmd->mask_tex_mapping     = MOD_DISP_MAP_LOCAL;
286         wmd->max_dist             = 1.0f; /* vert arbitrary distance, but don't use 0 */
287 }
288
289 static void copyData(ModifierData *md, ModifierData *target)
290 {
291 #if 0
292         WeightVGProximityModifierData *wmd  = (WeightVGProximityModifierData *) md;
293         WeightVGProximityModifierData *twmd = (WeightVGProximityModifierData *) target;
294 #endif
295
296         modifier_copyData_generic(md, target);
297 }
298
299 static CustomDataMask requiredDataMask(Object *UNUSED(ob), ModifierData *md)
300 {
301         WeightVGProximityModifierData *wmd = (WeightVGProximityModifierData *) md;
302         CustomDataMask dataMask = 0;
303
304         /* We need vertex groups! */
305         dataMask |= CD_MASK_MDEFORMVERT;
306
307         /* Ask for UV coordinates if we need them. */
308         if (wmd->mask_tex_mapping == MOD_DISP_MAP_UV)
309                 dataMask |= CD_MASK_MTFACE;
310
311         /* No need to ask for CD_PREVIEW_MLOOPCOL... */
312
313         return dataMask;
314 }
315
316 static bool dependsOnTime(ModifierData *md)
317 {
318         WeightVGProximityModifierData *wmd = (WeightVGProximityModifierData *) md;
319
320         if (wmd->mask_texture)
321                 return BKE_texture_dependsOnTime(wmd->mask_texture);
322         return 0;
323 }
324
325 static void foreachObjectLink(ModifierData *md, Object *ob, ObjectWalkFunc walk, void *userData)
326 {
327         WeightVGProximityModifierData *wmd = (WeightVGProximityModifierData *) md;
328         walk(userData, ob, &wmd->proximity_ob_target, IDWALK_CB_NOP);
329         walk(userData, ob, &wmd->mask_tex_map_obj, IDWALK_CB_NOP);
330 }
331
332 static void foreachIDLink(ModifierData *md, Object *ob, IDWalkFunc walk, void *userData)
333 {
334         WeightVGProximityModifierData *wmd = (WeightVGProximityModifierData *) md;
335
336         walk(userData, ob, (ID **)&wmd->mask_texture, IDWALK_CB_USER);
337
338         foreachObjectLink(md, ob, (ObjectWalkFunc)walk, userData);
339 }
340
341 static void foreachTexLink(ModifierData *md, Object *ob, TexWalkFunc walk, void *userData)
342 {
343         walk(userData, ob, md, "mask_texture");
344 }
345
346 static void updateDepgraph(ModifierData *md, const ModifierUpdateDepsgraphContext *ctx)
347 {
348         WeightVGProximityModifierData *wmd = (WeightVGProximityModifierData *) md;
349         DagNode *curNode;
350
351         if (wmd->proximity_ob_target) {
352                 curNode = dag_get_node(ctx->forest, wmd->proximity_ob_target);
353                 dag_add_relation(ctx->forest, curNode, ctx->obNode, DAG_RL_DATA_DATA | DAG_RL_OB_DATA,
354                                  "WeightVGProximity Modifier");
355         }
356
357         if (wmd->mask_tex_map_obj && wmd->mask_tex_mapping == MOD_DISP_MAP_OBJECT) {
358                 curNode = dag_get_node(ctx->forest, wmd->mask_tex_map_obj);
359
360                 dag_add_relation(ctx->forest, curNode, ctx->obNode, DAG_RL_DATA_DATA | DAG_RL_OB_DATA,
361                                  "WeightVGProximity Modifier");
362         }
363
364         if (wmd->mask_tex_mapping == MOD_DISP_MAP_GLOBAL)
365                 dag_add_relation(ctx->forest, ctx->obNode, ctx->obNode, DAG_RL_DATA_DATA | DAG_RL_OB_DATA,
366                                  "WeightVGProximity Modifier");
367 }
368
369 static void updateDepsgraph(ModifierData *md, const ModifierUpdateDepsgraphContext *ctx)
370 {
371         WeightVGProximityModifierData *wmd = (WeightVGProximityModifierData *)md;
372         if (wmd->proximity_ob_target != NULL) {
373                 DEG_add_object_relation(ctx->node, wmd->proximity_ob_target, DEG_OB_COMP_TRANSFORM, "WeightVGProximity Modifier");
374                 DEG_add_object_relation(ctx->node, wmd->proximity_ob_target, DEG_OB_COMP_GEOMETRY, "WeightVGProximity Modifier");
375         }
376         if (wmd->mask_tex_map_obj != NULL && wmd->mask_tex_mapping == MOD_DISP_MAP_OBJECT) {
377                 DEG_add_object_relation(ctx->node, wmd->mask_tex_map_obj, DEG_OB_COMP_TRANSFORM, "WeightVGProximity Modifier");
378                 DEG_add_object_relation(ctx->node, wmd->mask_tex_map_obj, DEG_OB_COMP_GEOMETRY, "WeightVGProximity Modifier");
379         }
380         if (wmd->mask_tex_mapping == MOD_DISP_MAP_GLOBAL) {
381                 DEG_add_object_relation(ctx->node, ctx->object, DEG_OB_COMP_TRANSFORM, "WeightVGProximity Modifier");
382                 DEG_add_object_relation(ctx->node, ctx->object, DEG_OB_COMP_GEOMETRY, "WeightVGProximity Modifier");
383         }
384 }
385
386 static bool isDisabled(ModifierData *md, int UNUSED(useRenderParams))
387 {
388         WeightVGProximityModifierData *wmd = (WeightVGProximityModifierData *) md;
389         /* If no vertex group, bypass. */
390         if (wmd->defgrp_name[0] == '\0') return 1;
391         /* If no target object, bypass. */
392         return (wmd->proximity_ob_target == NULL);
393 }
394
395 static DerivedMesh *applyModifier(ModifierData *md, Object *ob, DerivedMesh *derivedData,
396                                   ModifierApplyFlag UNUSED(flag))
397 {
398         WeightVGProximityModifierData *wmd = (WeightVGProximityModifierData *) md;
399         DerivedMesh *dm = derivedData;
400         MDeformVert *dvert = NULL;
401         MDeformWeight **dw, **tdw;
402         int numVerts;
403         float (*v_cos)[3] = NULL; /* The vertices coordinates. */
404         Object *obr = NULL; /* Our target object. */
405         int defgrp_index;
406         float *tw = NULL;
407         float *org_w = NULL;
408         float *new_w = NULL;
409         int *tidx, *indices = NULL;
410         int numIdx = 0;
411         int i;
412         /* Flags. */
413 #if 0
414         const bool do_prev = (wmd->modifier.mode & eModifierMode_DoWeightPreview) != 0;
415 #endif
416
417 #ifdef USE_TIMEIT
418         TIMEIT_START(perf);
419 #endif
420
421         /* Get number of verts. */
422         numVerts = dm->getNumVerts(dm);
423
424         /* Check if we can just return the original mesh.
425          * Must have verts and therefore verts assigned to vgroups to do anything useful!
426          */
427         if ((numVerts == 0) || BLI_listbase_is_empty(&ob->defbase))
428                 return dm;
429
430         /* Get our target object. */
431         obr = wmd->proximity_ob_target;
432         if (obr == NULL)
433                 return dm;
434
435         /* Get vgroup idx from its name. */
436         defgrp_index = defgroup_name_index(ob, wmd->defgrp_name);
437         if (defgrp_index == -1)
438                 return dm;
439
440         dvert = CustomData_duplicate_referenced_layer(&dm->vertData, CD_MDEFORMVERT, numVerts);
441         /* If no vertices were ever added to an object's vgroup, dvert might be NULL.
442          * As this modifier never add vertices to vgroup, just return. */
443         if (!dvert)
444                 return dm;
445
446         /* Find out which vertices to work on (all vertices in vgroup), and get their relevant weight.
447          */
448         tidx = MEM_malloc_arrayN(numVerts, sizeof(int), "WeightVGProximity Modifier, tidx");
449         tw = MEM_malloc_arrayN(numVerts, sizeof(float), "WeightVGProximity Modifier, tw");
450         tdw = MEM_malloc_arrayN(numVerts, sizeof(MDeformWeight *), "WeightVGProximity Modifier, tdw");
451         for (i = 0; i < numVerts; i++) {
452                 MDeformWeight *_dw = defvert_find_index(&dvert[i], defgrp_index);
453                 if (_dw) {
454                         tidx[numIdx] = i;
455                         tw[numIdx] = _dw->weight;
456                         tdw[numIdx++] = _dw;
457                 }
458         }
459         /* If no vertices found, return org data! */
460         if (numIdx == 0) {
461                 MEM_freeN(tidx);
462                 MEM_freeN(tw);
463                 MEM_freeN(tdw);
464                 return dm;
465         }
466         if (numIdx != numVerts) {
467                 indices = MEM_malloc_arrayN(numIdx, sizeof(int), "WeightVGProximity Modifier, indices");
468                 memcpy(indices, tidx, sizeof(int) * numIdx);
469                 org_w = MEM_malloc_arrayN(numIdx, sizeof(float), "WeightVGProximity Modifier, org_w");
470                 memcpy(org_w, tw, sizeof(float) * numIdx);
471                 dw = MEM_malloc_arrayN(numIdx, sizeof(MDeformWeight *), "WeightVGProximity Modifier, dw");
472                 memcpy(dw, tdw, sizeof(MDeformWeight *) * numIdx);
473                 MEM_freeN(tw);
474                 MEM_freeN(tdw);
475         }
476         else {
477                 org_w = tw;
478                 dw = tdw;
479         }
480         new_w = MEM_malloc_arrayN(numIdx, sizeof(float), "WeightVGProximity Modifier, new_w");
481         MEM_freeN(tidx);
482
483         /* Get our vertex coordinates. */
484         v_cos = MEM_malloc_arrayN(numIdx, sizeof(float[3]), "WeightVGProximity Modifier, v_cos");
485         if (numIdx != numVerts) {
486                 /* XXX In some situations, this code can be up to about 50 times more performant
487                  *     than simply using getVertCo for each affected vertex...
488                  */
489                 float (*tv_cos)[3] = MEM_malloc_arrayN(numVerts, sizeof(float[3]), "WeightVGProximity Modifier, tv_cos");
490                 dm->getVertCos(dm, tv_cos);
491                 for (i = 0; i < numIdx; i++)
492                         copy_v3_v3(v_cos[i], tv_cos[indices[i]]);
493                 MEM_freeN(tv_cos);
494         }
495         else
496                 dm->getVertCos(dm, v_cos);
497
498         /* Compute wanted distances. */
499         if (wmd->proximity_mode == MOD_WVG_PROXIMITY_OBJECT) {
500                 const float dist = get_ob2ob_distance(ob, obr);
501                 for (i = 0; i < numIdx; i++)
502                         new_w[i] = dist;
503         }
504         else if (wmd->proximity_mode == MOD_WVG_PROXIMITY_GEOMETRY) {
505                 const bool use_trgt_verts = (wmd->proximity_flags & MOD_WVG_PROXIMITY_GEOM_VERTS) != 0;
506                 const bool use_trgt_edges = (wmd->proximity_flags & MOD_WVG_PROXIMITY_GEOM_EDGES) != 0;
507                 const bool use_trgt_faces = (wmd->proximity_flags & MOD_WVG_PROXIMITY_GEOM_FACES) != 0;
508
509                 if (use_trgt_verts || use_trgt_edges || use_trgt_faces) {
510                         DerivedMesh *target_dm = obr->derivedFinal;
511                         bool free_target_dm = false;
512                         if (!target_dm) {
513                                 if (ELEM(obr->type, OB_CURVE, OB_SURF, OB_FONT))
514                                         target_dm = CDDM_from_curve(obr);
515                                 else if (obr->type == OB_MESH) {
516                                         Mesh *me = (Mesh *)obr->data;
517                                         if (me->edit_btmesh)
518                                                 target_dm = CDDM_from_editbmesh(me->edit_btmesh, false, false);
519                                         else
520                                                 target_dm = CDDM_from_mesh(me);
521                                 }
522                                 free_target_dm = true;
523                         }
524
525                         /* We must check that we do have a valid target_dm! */
526                         if (target_dm) {
527                                 SpaceTransform loc2trgt;
528                                 float *dists_v = use_trgt_verts ? MEM_malloc_arrayN(numIdx, sizeof(float), "dists_v") : NULL;
529                                 float *dists_e = use_trgt_edges ? MEM_malloc_arrayN(numIdx, sizeof(float), "dists_e") : NULL;
530                                 float *dists_f = use_trgt_faces ? MEM_malloc_arrayN(numIdx, sizeof(float), "dists_f") : NULL;
531
532                                 BLI_SPACE_TRANSFORM_SETUP(&loc2trgt, ob, obr);
533                                 get_vert2geom_distance(numIdx, v_cos, dists_v, dists_e, dists_f,
534                                                        target_dm, &loc2trgt);
535                                 for (i = 0; i < numIdx; i++) {
536                                         new_w[i] = dists_v ? dists_v[i] : FLT_MAX;
537                                         if (dists_e)
538                                                 new_w[i] = min_ff(dists_e[i], new_w[i]);
539                                         if (dists_f)
540                                                 new_w[i] = min_ff(dists_f[i], new_w[i]);
541                                 }
542                                 if (free_target_dm) target_dm->release(target_dm);
543                                 if (dists_v) MEM_freeN(dists_v);
544                                 if (dists_e) MEM_freeN(dists_e);
545                                 if (dists_f) MEM_freeN(dists_f);
546                         }
547                         /* Else, fall back to default obj2vert behavior. */
548                         else {
549                                 get_vert2ob_distance(numIdx, v_cos, new_w, ob, obr);
550                         }
551                 }
552                 else {
553                         get_vert2ob_distance(numIdx, v_cos, new_w, ob, obr);
554                 }
555         }
556
557         /* Map distances to weights. */
558         do_map(ob, new_w, numIdx, wmd->min_dist, wmd->max_dist, wmd->falloff_type);
559
560         /* Do masking. */
561         weightvg_do_mask(numIdx, indices, org_w, new_w, ob, dm, wmd->mask_constant,
562                          wmd->mask_defgrp_name, wmd->modifier.scene, wmd->mask_texture,
563                          wmd->mask_tex_use_channel, wmd->mask_tex_mapping,
564                          wmd->mask_tex_map_obj, wmd->mask_tex_uvlayer_name);
565
566         /* Update vgroup. Note we never add nor remove vertices from vgroup here. */
567         weightvg_update_vg(dvert, defgrp_index, dw, numIdx, indices, org_w, false, 0.0f, false, 0.0f);
568
569         /* If weight preview enabled... */
570 #if 0 /* XXX Currently done in mod stack :/ */
571         if (do_prev)
572                 DM_update_weight_mcol(ob, dm, 0, org_w, numIdx, indices);
573 #endif
574
575         /* Freeing stuff. */
576         MEM_freeN(org_w);
577         MEM_freeN(new_w);
578         MEM_freeN(dw);
579         if (indices)
580                 MEM_freeN(indices);
581         MEM_freeN(v_cos);
582
583 #ifdef USE_TIMEIT
584         TIMEIT_END(perf);
585 #endif
586
587         /* Return the vgroup-modified mesh. */
588         return dm;
589 }
590
591
592 ModifierTypeInfo modifierType_WeightVGProximity = {
593         /* name */              "VertexWeightProximity",
594         /* structName */        "WeightVGProximityModifierData",
595         /* structSize */        sizeof(WeightVGProximityModifierData),
596         /* type */              eModifierTypeType_NonGeometrical,
597         /* flags */             eModifierTypeFlag_AcceptsMesh |
598                                 eModifierTypeFlag_SupportsMapping |
599                                 eModifierTypeFlag_SupportsEditmode |
600                                 eModifierTypeFlag_UsesPreview,
601
602         /* copyData */          copyData,
603         /* deformVerts */       NULL,
604         /* deformMatrices */    NULL,
605         /* deformVertsEM */     NULL,
606         /* deformMatricesEM */  NULL,
607         /* applyModifier */     applyModifier,
608         /* applyModifierEM */   NULL,
609         /* initData */          initData,
610         /* requiredDataMask */  requiredDataMask,
611         /* freeData */          NULL,
612         /* isDisabled */        isDisabled,
613         /* updateDepgraph */    updateDepgraph,
614         /* updateDepsgraph */   updateDepsgraph,
615         /* dependsOnTime */     dependsOnTime,
616         /* dependsOnNormals */  NULL,
617         /* foreachObjectLink */ foreachObjectLink,
618         /* foreachIDLink */     foreachIDLink,
619         /* foreachTexLink */    foreachTexLink,
620 };