minor edits to previous commit & set VKey to toggle vertex select in weight paint...
[blender.git] / source / blender / blenkernel / intern / DerivedMesh.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) 2005 Blender Foundation.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): none yet.
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/blenkernel/intern/DerivedMesh.c
29  *  \ingroup bke
30  */
31
32
33 #include <string.h>
34
35
36 #include "MEM_guardedalloc.h"
37
38 #include "DNA_cloth_types.h"
39 #include "DNA_key_types.h"
40 #include "DNA_meshdata_types.h"
41 #include "DNA_armature_types.h"
42 #include "DNA_object_types.h"
43 #include "DNA_scene_types.h" // N_T
44
45 #include "BLI_blenlib.h"
46 #include "BLI_editVert.h"
47 #include "BLI_math.h"
48 #include "BLI_memarena.h"
49 #include "BLI_pbvh.h"
50 #include "BLI_utildefines.h"
51 #include "BLI_linklist.h"
52
53 #include "BKE_cdderivedmesh.h"
54 #include "BKE_displist.h"
55 #include "BKE_key.h"
56 #include "BKE_modifier.h"
57 #include "BKE_mesh.h"
58 #include "BKE_object.h"
59 #include "BKE_paint.h"
60 #include "BKE_texture.h"
61 #include "BKE_multires.h"
62 #include "BKE_armature.h"
63 #include "BKE_deform.h"
64
65 #ifdef WITH_GAMEENGINE
66 #include "BKE_navmesh_conversion.h"
67 static DerivedMesh *navmesh_dm_createNavMeshForVisualization(DerivedMesh *dm);
68 #endif
69
70 #include "BLO_sys_types.h" // for intptr_t support
71
72 #include "GL/glew.h"
73
74 #include "GPU_buffers.h"
75 #include "GPU_draw.h"
76 #include "GPU_extensions.h"
77 #include "GPU_material.h"
78
79 extern GLubyte stipple_quarttone[128]; /* glutil.c, bad level data */
80
81 ///////////////////////////////////
82 ///////////////////////////////////
83
84 static MVert *dm_getVertArray(DerivedMesh *dm)
85 {
86         MVert *mvert = CustomData_get_layer(&dm->vertData, CD_MVERT);
87
88         if (!mvert) {
89                 mvert = CustomData_add_layer(&dm->vertData, CD_MVERT, CD_CALLOC, NULL,
90                         dm->getNumVerts(dm));
91                 CustomData_set_layer_flag(&dm->vertData, CD_MVERT, CD_FLAG_TEMPORARY);
92                 dm->copyVertArray(dm, mvert);
93         }
94
95         return mvert;
96 }
97
98 static MEdge *dm_getEdgeArray(DerivedMesh *dm)
99 {
100         MEdge *medge = CustomData_get_layer(&dm->edgeData, CD_MEDGE);
101
102         if (!medge) {
103                 medge = CustomData_add_layer(&dm->edgeData, CD_MEDGE, CD_CALLOC, NULL,
104                         dm->getNumEdges(dm));
105                 CustomData_set_layer_flag(&dm->edgeData, CD_MEDGE, CD_FLAG_TEMPORARY);
106                 dm->copyEdgeArray(dm, medge);
107         }
108
109         return medge;
110 }
111
112 static MFace *dm_getFaceArray(DerivedMesh *dm)
113 {
114         MFace *mface = CustomData_get_layer(&dm->faceData, CD_MFACE);
115
116         if (!mface) {
117                 mface = CustomData_add_layer(&dm->faceData, CD_MFACE, CD_CALLOC, NULL,
118                         dm->getNumFaces(dm));
119                 CustomData_set_layer_flag(&dm->faceData, CD_MFACE, CD_FLAG_TEMPORARY);
120                 dm->copyFaceArray(dm, mface);
121         }
122
123         return mface;
124 }
125
126 static MVert *dm_dupVertArray(DerivedMesh *dm)
127 {
128         MVert *tmp = MEM_callocN(sizeof(*tmp) * dm->getNumVerts(dm),
129                                                          "dm_dupVertArray tmp");
130
131         if(tmp) dm->copyVertArray(dm, tmp);
132
133         return tmp;
134 }
135
136 static MEdge *dm_dupEdgeArray(DerivedMesh *dm)
137 {
138         MEdge *tmp = MEM_callocN(sizeof(*tmp) * dm->getNumEdges(dm),
139                                                          "dm_dupEdgeArray tmp");
140
141         if(tmp) dm->copyEdgeArray(dm, tmp);
142
143         return tmp;
144 }
145
146 static MFace *dm_dupFaceArray(DerivedMesh *dm)
147 {
148         MFace *tmp = MEM_callocN(sizeof(*tmp) * dm->getNumFaces(dm),
149                                                          "dm_dupFaceArray tmp");
150
151         if(tmp) dm->copyFaceArray(dm, tmp);
152
153         return tmp;
154 }
155
156 void DM_init_funcs(DerivedMesh *dm)
157 {
158         /* default function implementations */
159         dm->getVertArray = dm_getVertArray;
160         dm->getEdgeArray = dm_getEdgeArray;
161         dm->getFaceArray = dm_getFaceArray;
162         dm->dupVertArray = dm_dupVertArray;
163         dm->dupEdgeArray = dm_dupEdgeArray;
164         dm->dupFaceArray = dm_dupFaceArray;
165
166         dm->getVertData = DM_get_vert_data;
167         dm->getEdgeData = DM_get_edge_data;
168         dm->getFaceData = DM_get_face_data;
169         dm->getVertDataArray = DM_get_vert_data_layer;
170         dm->getEdgeDataArray = DM_get_edge_data_layer;
171         dm->getFaceDataArray = DM_get_face_data_layer;
172
173         bvhcache_init(&dm->bvhCache);
174 }
175
176 void DM_init(DerivedMesh *dm, DerivedMeshType type,
177                          int numVerts, int numEdges, int numFaces)
178 {
179         dm->type = type;
180         dm->numVertData = numVerts;
181         dm->numEdgeData = numEdges;
182         dm->numFaceData = numFaces;
183
184         DM_init_funcs(dm);
185         
186         dm->needsFree = 1;
187         dm->auto_bump_scale = -1.0f;
188 }
189
190 void DM_from_template(DerivedMesh *dm, DerivedMesh *source, DerivedMeshType type,
191                                           int numVerts, int numEdges, int numFaces)
192 {
193         CustomData_copy(&source->vertData, &dm->vertData, CD_MASK_DERIVEDMESH,
194                                         CD_CALLOC, numVerts);
195         CustomData_copy(&source->edgeData, &dm->edgeData, CD_MASK_DERIVEDMESH,
196                                         CD_CALLOC, numEdges);
197         CustomData_copy(&source->faceData, &dm->faceData, CD_MASK_DERIVEDMESH,
198                                         CD_CALLOC, numFaces);
199
200         dm->type = type;
201         dm->numVertData = numVerts;
202         dm->numEdgeData = numEdges;
203         dm->numFaceData = numFaces;
204
205         DM_init_funcs(dm);
206
207         dm->needsFree = 1;
208 }
209
210 int DM_release(DerivedMesh *dm)
211 {
212         if (dm->needsFree) {
213                 bvhcache_free(&dm->bvhCache);
214                 GPU_drawobject_free( dm );
215                 CustomData_free(&dm->vertData, dm->numVertData);
216                 CustomData_free(&dm->edgeData, dm->numEdgeData);
217                 CustomData_free(&dm->faceData, dm->numFaceData);
218
219                 return 1;
220         }
221         else {
222                 CustomData_free_temporary(&dm->vertData, dm->numVertData);
223                 CustomData_free_temporary(&dm->edgeData, dm->numEdgeData);
224                 CustomData_free_temporary(&dm->faceData, dm->numFaceData);
225
226                 return 0;
227         }
228 }
229
230 void DM_to_mesh(DerivedMesh *dm, Mesh *me)
231 {
232         /* dm might depend on me, so we need to do everything with a local copy */
233         Mesh tmp = *me;
234         int totvert, totedge, totface;
235
236         memset(&tmp.vdata, 0, sizeof(tmp.vdata));
237         memset(&tmp.edata, 0, sizeof(tmp.edata));
238         memset(&tmp.fdata, 0, sizeof(tmp.fdata));
239
240         totvert = tmp.totvert = dm->getNumVerts(dm);
241         totedge = tmp.totedge = dm->getNumEdges(dm);
242         totface = tmp.totface = dm->getNumFaces(dm);
243
244         CustomData_copy(&dm->vertData, &tmp.vdata, CD_MASK_MESH, CD_DUPLICATE, totvert);
245         CustomData_copy(&dm->edgeData, &tmp.edata, CD_MASK_MESH, CD_DUPLICATE, totedge);
246         CustomData_copy(&dm->faceData, &tmp.fdata, CD_MASK_MESH, CD_DUPLICATE, totface);
247
248         /* not all DerivedMeshes store their verts/edges/faces in CustomData, so
249            we set them here in case they are missing */
250         if(!CustomData_has_layer(&tmp.vdata, CD_MVERT))
251                 CustomData_add_layer(&tmp.vdata, CD_MVERT, CD_ASSIGN, dm->dupVertArray(dm), totvert);
252         if(!CustomData_has_layer(&tmp.edata, CD_MEDGE))
253                 CustomData_add_layer(&tmp.edata, CD_MEDGE, CD_ASSIGN, dm->dupEdgeArray(dm), totedge);
254         if(!CustomData_has_layer(&tmp.fdata, CD_MFACE))
255                 CustomData_add_layer(&tmp.fdata, CD_MFACE, CD_ASSIGN, dm->dupFaceArray(dm), totface);
256
257         /* object had got displacement layer, should copy this layer to save sculpted data */
258         /* NOTE: maybe some other layers should be copied? nazgul */
259         if(CustomData_has_layer(&me->fdata, CD_MDISPS)) {
260                 if (totface == me->totface) {
261                         MDisps *mdisps = CustomData_get_layer(&me->fdata, CD_MDISPS);
262                         CustomData_add_layer(&tmp.fdata, CD_MDISPS, CD_DUPLICATE, mdisps, totface);
263                 }
264         }
265
266         mesh_update_customdata_pointers(&tmp);
267
268         CustomData_free(&me->vdata, me->totvert);
269         CustomData_free(&me->edata, me->totedge);
270         CustomData_free(&me->fdata, me->totface);
271
272         /* if the number of verts has changed, remove invalid data */
273         if(tmp.totvert != me->totvert) {
274                 if(tmp.key) tmp.key->id.us--;
275                 tmp.key = NULL;
276         }
277
278         *me = tmp;
279 }
280
281 void DM_to_meshkey(DerivedMesh *dm, Mesh *me, KeyBlock *kb)
282 {
283         int a, totvert = dm->getNumVerts(dm);
284         float *fp;
285         MVert *mvert;
286         
287         if(totvert==0 || me->totvert==0 || me->totvert!=totvert) return;
288         
289         if(kb->data) MEM_freeN(kb->data);
290         kb->data= MEM_callocN(me->key->elemsize*me->totvert, "kb->data");
291         kb->totelem= totvert;
292         
293         fp= kb->data;
294         mvert=dm->getVertDataArray(dm, CD_MVERT);
295         
296         for(a=0; a<kb->totelem; a++, fp+=3, mvert++) {
297                 copy_v3_v3(fp, mvert->co);
298         }
299 }
300
301 void DM_set_only_copy(DerivedMesh *dm, CustomDataMask mask)
302 {
303         CustomData_set_only_copy(&dm->vertData, mask);
304         CustomData_set_only_copy(&dm->edgeData, mask);
305         CustomData_set_only_copy(&dm->faceData, mask);
306 }
307
308 void DM_add_vert_layer(DerivedMesh *dm, int type, int alloctype, void *layer)
309 {
310         CustomData_add_layer(&dm->vertData, type, alloctype, layer, dm->numVertData);
311 }
312
313 void DM_add_edge_layer(DerivedMesh *dm, int type, int alloctype, void *layer)
314 {
315         CustomData_add_layer(&dm->edgeData, type, alloctype, layer, dm->numEdgeData);
316 }
317
318 void DM_add_face_layer(DerivedMesh *dm, int type, int alloctype, void *layer)
319 {
320         CustomData_add_layer(&dm->faceData, type, alloctype, layer, dm->numFaceData);
321 }
322
323 void *DM_get_vert_data(DerivedMesh *dm, int index, int type)
324 {
325         return CustomData_get(&dm->vertData, index, type);
326 }
327
328 void *DM_get_edge_data(DerivedMesh *dm, int index, int type)
329 {
330         return CustomData_get(&dm->edgeData, index, type);
331 }
332
333 void *DM_get_face_data(DerivedMesh *dm, int index, int type)
334 {
335         return CustomData_get(&dm->faceData, index, type);
336 }
337
338 void *DM_get_vert_data_layer(DerivedMesh *dm, int type)
339 {
340         if(type == CD_MVERT)
341                 return dm->getVertArray(dm);
342
343         return CustomData_get_layer(&dm->vertData, type);
344 }
345
346 void *DM_get_edge_data_layer(DerivedMesh *dm, int type)
347 {
348         if(type == CD_MEDGE)
349                 return dm->getEdgeArray(dm);
350
351         return CustomData_get_layer(&dm->edgeData, type);
352 }
353
354 void *DM_get_face_data_layer(DerivedMesh *dm, int type)
355 {
356         if(type == CD_MFACE)
357                 return dm->getFaceArray(dm);
358
359         return CustomData_get_layer(&dm->faceData, type);
360 }
361
362 void DM_set_vert_data(DerivedMesh *dm, int index, int type, void *data)
363 {
364         CustomData_set(&dm->vertData, index, type, data);
365 }
366
367 void DM_set_edge_data(DerivedMesh *dm, int index, int type, void *data)
368 {
369         CustomData_set(&dm->edgeData, index, type, data);
370 }
371
372 void DM_set_face_data(DerivedMesh *dm, int index, int type, void *data)
373 {
374         CustomData_set(&dm->faceData, index, type, data);
375 }
376
377 void DM_copy_vert_data(DerivedMesh *source, DerivedMesh *dest,
378                                            int source_index, int dest_index, int count)
379 {
380         CustomData_copy_data(&source->vertData, &dest->vertData,
381                                                  source_index, dest_index, count);
382 }
383
384 void DM_copy_edge_data(DerivedMesh *source, DerivedMesh *dest,
385                                            int source_index, int dest_index, int count)
386 {
387         CustomData_copy_data(&source->edgeData, &dest->edgeData,
388                                                  source_index, dest_index, count);
389 }
390
391 void DM_copy_face_data(DerivedMesh *source, DerivedMesh *dest,
392                                            int source_index, int dest_index, int count)
393 {
394         CustomData_copy_data(&source->faceData, &dest->faceData,
395                                                  source_index, dest_index, count);
396 }
397
398 void DM_free_vert_data(struct DerivedMesh *dm, int index, int count)
399 {
400         CustomData_free_elem(&dm->vertData, index, count);
401 }
402
403 void DM_free_edge_data(struct DerivedMesh *dm, int index, int count)
404 {
405         CustomData_free_elem(&dm->edgeData, index, count);
406 }
407
408 void DM_free_face_data(struct DerivedMesh *dm, int index, int count)
409 {
410         CustomData_free_elem(&dm->faceData, index, count);
411 }
412
413 void DM_interp_vert_data(DerivedMesh *source, DerivedMesh *dest,
414                                                  int *src_indices, float *weights,
415                                                  int count, int dest_index)
416 {
417         CustomData_interp(&source->vertData, &dest->vertData, src_indices,
418                                           weights, NULL, count, dest_index);
419 }
420
421 void DM_interp_edge_data(DerivedMesh *source, DerivedMesh *dest,
422                                                  int *src_indices,
423                                                  float *weights, EdgeVertWeight *vert_weights,
424                                                  int count, int dest_index)
425 {
426         CustomData_interp(&source->edgeData, &dest->edgeData, src_indices,
427                                           weights, (float*)vert_weights, count, dest_index);
428 }
429
430 void DM_interp_face_data(DerivedMesh *source, DerivedMesh *dest,
431                                                  int *src_indices,
432                                                  float *weights, FaceVertWeight *vert_weights,
433                                                  int count, int dest_index)
434 {
435         CustomData_interp(&source->faceData, &dest->faceData, src_indices,
436                                           weights, (float*)vert_weights, count, dest_index);
437 }
438
439 void DM_swap_face_data(DerivedMesh *dm, int index, const int *corner_indices)
440 {
441         CustomData_swap(&dm->faceData, index, corner_indices);
442 }
443
444 ///
445
446 DerivedMesh *mesh_create_derived(Mesh *me, Object *ob, float (*vertCos)[3])
447 {
448         DerivedMesh *dm = CDDM_from_mesh(me, ob);
449         
450         if(!dm)
451                 return NULL;
452         
453         if (vertCos)
454                 CDDM_apply_vert_coords(dm, vertCos);
455
456         CDDM_calc_normals(dm);
457
458         return dm;
459 }
460
461 DerivedMesh *mesh_create_derived_for_modifier(Scene *scene, Object *ob, ModifierData *md)
462 {
463         Mesh *me = ob->data;
464         ModifierTypeInfo *mti = modifierType_getInfo(md->type);
465         DerivedMesh *dm;
466
467         md->scene= scene;
468         
469         if (!(md->mode&eModifierMode_Realtime)) return NULL;
470         if (mti->isDisabled && mti->isDisabled(md, 0)) return NULL;
471
472         if (mti->type==eModifierTypeType_OnlyDeform) {
473                 int numVerts;
474                 float (*deformedVerts)[3] = mesh_getVertexCos(me, &numVerts);
475
476                 mti->deformVerts(md, ob, NULL, deformedVerts, numVerts, 0, 0);
477                 dm = mesh_create_derived(me, ob, deformedVerts);
478
479                 MEM_freeN(deformedVerts);
480         } else {
481                 DerivedMesh *tdm = mesh_create_derived(me, ob, NULL);
482                 dm = mti->applyModifier(md, ob, tdm, 0, 0);
483
484                 if(tdm != dm) tdm->release(tdm);
485         }
486
487         return dm;
488 }
489
490 static float *get_editmesh_orco_verts(EditMesh *em)
491 {
492         EditVert *eve;
493         float *orco;
494         int a, totvert;
495
496         /* these may not really be the orco's, but it's only for preview.
497          * could be solver better once, but isn't simple */
498
499         totvert= 0;
500         for(eve=em->verts.first; eve; eve=eve->next)
501                 totvert++;
502         
503         orco = MEM_mallocN(sizeof(float)*3*totvert, "EditMesh Orco");
504
505         for(a=0, eve=em->verts.first; eve; eve=eve->next, a+=3) {
506                 copy_v3_v3(orco+a, eve->co);
507         }
508         
509         return orco;
510 }
511
512 /* orco custom data layer */
513
514 static void *get_orco_coords_dm(Object *ob, EditMesh *em, int layer, int *free)
515 {
516         *free= 0;
517
518         if(layer == CD_ORCO) {
519                 /* get original coordinates */
520                 *free= 1;
521
522                 if(em)
523                         return (float(*)[3])get_editmesh_orco_verts(em);
524                 else
525                         return (float(*)[3])get_mesh_orco_verts(ob);
526         }
527         else if(layer == CD_CLOTH_ORCO) {
528                 /* apply shape key for cloth, this should really be solved
529                    by a more flexible customdata system, but not simple */
530                 if(!em) {
531                         ClothModifierData *clmd = (ClothModifierData *)modifiers_findByType(ob, eModifierType_Cloth);
532                         KeyBlock *kb= key_get_keyblock(ob_get_key(ob), clmd->sim_parms->shapekey_rest);
533
534                         if(kb->data)
535                                 return kb->data;
536                 }
537
538                 return NULL;
539         }
540
541         return NULL;
542 }
543
544 static DerivedMesh *create_orco_dm(Object *ob, Mesh *me, EditMesh *em, int layer)
545 {
546         DerivedMesh *dm;
547         float (*orco)[3];
548         int free;
549
550         if(em) dm= CDDM_from_editmesh(em, me);
551         else dm= CDDM_from_mesh(me, ob);
552
553         orco= get_orco_coords_dm(ob, em, layer, &free);
554
555         if(orco) {
556                 CDDM_apply_vert_coords(dm, orco);
557                 if(free) MEM_freeN(orco);
558         }
559
560         CDDM_calc_normals(dm);
561
562         return dm;
563 }
564
565 static void add_orco_dm(Object *ob, EditMesh *em, DerivedMesh *dm, DerivedMesh *orcodm, int layer)
566 {
567         float (*orco)[3], (*layerorco)[3];
568         int totvert, free;
569
570         totvert= dm->getNumVerts(dm);
571
572         if(orcodm) {
573                 orco= MEM_callocN(sizeof(float)*3*totvert, "dm orco");
574                 free= 1;
575
576                 if(orcodm->getNumVerts(orcodm) == totvert)
577                         orcodm->getVertCos(orcodm, orco);
578                 else
579                         dm->getVertCos(dm, orco);
580         }
581         else
582                 orco= get_orco_coords_dm(ob, em, layer, &free);
583
584         if(orco) {
585                 if(layer == CD_ORCO)
586                         transform_mesh_orco_verts(ob->data, orco, totvert, 0);
587
588                 if(!(layerorco = DM_get_vert_data_layer(dm, layer))) {
589                         DM_add_vert_layer(dm, layer, CD_CALLOC, NULL);
590                         layerorco = DM_get_vert_data_layer(dm, layer);
591                 }
592
593                 memcpy(layerorco, orco, sizeof(float)*3*totvert);
594                 if(free) MEM_freeN(orco);
595         }
596 }
597
598 /* weight paint colors */
599
600 /* Something of a hack, at the moment deal with weightpaint
601  * by tucking into colors during modifier eval, only in
602  * wpaint mode. Works ok but need to make sure recalc
603  * happens on enter/exit wpaint.
604  */
605
606 void weight_to_rgb(float r_rgb[3], const float weight)
607 {
608         const float blend= ((weight/2.0f)+0.5f);
609
610         if (weight<=0.25f){     // blue->cyan
611                 r_rgb[0]= 0.0f;
612                 r_rgb[1]= blend*weight*4.0f;
613                 r_rgb[2]= blend;
614         }
615         else if (weight<=0.50f){        // cyan->green
616                 r_rgb[0]= 0.0f;
617                 r_rgb[1]= blend;
618                 r_rgb[2]= blend*(1.0f-((weight-0.25f)*4.0f));
619         }
620         else if (weight <= 0.75f){      // green->yellow
621                 r_rgb[0]= blend * ((weight-0.50f)*4.0f);
622                 r_rgb[1]= blend;
623                 r_rgb[2]= 0.0f;
624         }
625         else if (weight <= 1.0f){ // yellow->red
626                 r_rgb[0]= blend;
627                 r_rgb[1]= blend * (1.0f-((weight-0.75f)*4.0f));
628                 r_rgb[2]= 0.0f;
629         }
630         else {
631                 /* exceptional value, unclamped or nan,
632                  * avoid uninitialized memory use */
633                 r_rgb[0]= 1.0f;
634                 r_rgb[1]= 0.0f;
635                 r_rgb[2]= 1.0f;
636         }
637 }
638
639 /* draw_flag's for calc_weightpaint_vert_color */
640 enum {
641         CALC_WP_MULTIPAINT= (1<<0),
642         CALC_WP_AUTO_NORMALIZE= (1<<1)
643 };
644
645 static void weightpaint_color(unsigned char r_col[4], ColorBand *coba, const float input)
646 {
647         float colf[4];
648
649         if(coba) do_colorband(coba, input, colf);
650         else     weight_to_rgb(colf, input);
651
652         r_col[3] = (unsigned char)(colf[0] * 255.0f);
653         r_col[2] = (unsigned char)(colf[1] * 255.0f);
654         r_col[1] = (unsigned char)(colf[2] * 255.0f);
655         r_col[0] = 255;
656 }
657
658
659 static void calc_weightpaint_vert_color(
660         unsigned char r_col[4],
661         MDeformVert *dv, ColorBand *coba,
662         const int defbase_tot, const int defbase_act,
663         const char *dg_flags,
664         const int selected, const int draw_flag)
665 {
666         float input = 0.0f;
667         
668         int make_black= FALSE;
669
670         if ((selected > 1) && (draw_flag & CALC_WP_MULTIPAINT)) {
671                 int was_a_nonzero= FALSE;
672                 unsigned int i;
673
674                 MDeformWeight *dw= dv->dw;
675                 for (i = dv->totweight; i != 0; i--, dw++) {
676                         /* in multipaint, get the average if auto normalize is inactive
677                          * get the sum if it is active */
678                         if (dw->def_nr < defbase_tot) {
679                                 if (dg_flags[dw->def_nr]) {
680                                         if (dw->weight) {
681                                                 input += dw->weight;
682                                                 was_a_nonzero= TRUE;
683                                         }
684                                 }
685                         }
686                 }
687
688                 /* make it black if the selected groups have no weight on a vertex */
689                 if (was_a_nonzero == FALSE) {
690                         make_black = TRUE;
691                 }
692                 else if ((draw_flag & CALC_WP_AUTO_NORMALIZE) == FALSE) {
693                         input /= selected; /* get the average */
694                 }
695         }
696         else {
697                 /* default, non tricky behavior */
698                 input= defvert_find_weight(dv, defbase_act);
699         }
700
701         if (make_black) { /* TODO, theme color */
702                 r_col[3] = 0;
703                 r_col[2] = 0;
704                 r_col[1] = 0;
705                 r_col[0] = 255;
706         }
707         else {
708                 CLAMP(input, 0.0f, 1.0f);
709                 weightpaint_color(r_col, coba, input);
710         }
711 }
712
713 static ColorBand *stored_cb= NULL;
714
715 void vDM_ColorBand_store(ColorBand *coba)
716 {
717         stored_cb= coba;
718 }
719
720 /* return an array of vertex weight colors, caller must free.
721  *
722  * note that we could save some memory and allocate RGB only but then we'd need to
723  * re-arrange the colors when copying to the face since MCol has odd ordering,
724  * so leave this as is - campbell */
725 static unsigned char *calc_weightpaint_vert_array(Object *ob, int const draw_flag, ColorBand *coba)
726 {
727         Mesh *me = ob->data;
728         unsigned char *wtcol_v = MEM_mallocN (sizeof(unsigned char) * me->totvert * 4, "weightmap_v");
729
730         if (me->dvert) {
731                 unsigned char *wc = wtcol_v;
732                 MDeformVert *dv= me->dvert;
733                 unsigned int i;
734
735                 /* varisbles for multipaint */
736                 const int defbase_tot = BLI_countlist(&ob->defbase);
737                 const int defbase_act = ob->actdef-1;
738                 char *dg_flags = MEM_mallocN(defbase_tot * sizeof(char), __func__);
739                 const int selected = get_selected_defgroups(ob, dg_flags, defbase_tot);
740                 /* const int unselected = defbase_tot - selected; */ /* UNUSED */
741
742                 for (i = me->totvert; i != 0; i--, wc += 4, dv++) {
743                         calc_weightpaint_vert_color(wc, dv, coba, defbase_tot, defbase_act, dg_flags, selected, draw_flag);
744                 }
745
746                 MEM_freeN(dg_flags);
747         }
748         else {
749                 int col_i;
750                 weightpaint_color((unsigned char *)&col_i, coba, 0.0f);
751                 fill_vn_i((int *)wtcol_v, me->totvert, col_i);
752         }
753
754         return wtcol_v;
755 }
756
757 static void add_weight_mcol_dm(Object *ob, DerivedMesh *dm, int const draw_flag)
758 {
759         ColorBand *coba= stored_cb;     /* warning, not a local var */
760
761         Mesh *me = ob->data;
762         unsigned char *wtcol_v = calc_weightpaint_vert_array(ob, draw_flag, coba);
763         unsigned char *wtcol_f = MEM_mallocN (sizeof(unsigned char) * me->totface*4*4, "weightmap_f");
764
765         MFace *mf = me->mface;
766         int i;
767
768         for (i=0; i<me->totface; i++, mf++) {
769                 unsigned int fidx= mf->v4 ? 3:2;
770                 do {
771                         copy_v4_v4_char((char *)&wtcol_f[(4 * i + fidx) * 4],
772                                         (char *)&wtcol_v[4 * (*(&mf->v1 + fidx))]);
773                 } while (fidx--);
774         }
775
776         MEM_freeN(wtcol_v);
777
778         CustomData_add_layer(&dm->faceData, CD_WEIGHT_MCOL, CD_ASSIGN, wtcol_f, dm->numFaceData);
779 }
780
781 /* new value for useDeform -1  (hack for the gameengine):
782  * - apply only the modifier stack of the object, skipping the virtual modifiers,
783  * - don't apply the key
784  * - apply deform modifiers and input vertexco
785  */
786 static void mesh_calc_modifiers(Scene *scene, Object *ob, float (*inputVertexCos)[3],
787                                                                 DerivedMesh **deform_r, DerivedMesh **final_r,
788                                                                 int useRenderParams, int useDeform,
789                                                                 int needMapping, CustomDataMask dataMask, int index, int useCache)
790 {
791         Mesh *me = ob->data;
792         ModifierData *firstmd, *md;
793         LinkNode *datamasks, *curr;
794         CustomDataMask mask, nextmask, append_mask = 0;
795         float (*deformedVerts)[3] = NULL;
796         DerivedMesh *dm, *orcodm, *clothorcodm, *finaldm;
797         int numVerts = me->totvert;
798         int required_mode;
799         int isPrevDeform= FALSE;
800         int skipVirtualArmature = (useDeform < 0);
801         MultiresModifierData *mmd= get_multires_modifier(scene, ob, 0);
802         int has_multires = mmd != NULL, multires_applied = 0;
803         int sculpt_mode = ob->mode & OB_MODE_SCULPT && ob->sculpt;
804
805         int draw_flag= ((scene->toolsettings->multipaint ? CALC_WP_MULTIPAINT : 0) |
806                         (scene->toolsettings->auto_normalize ? CALC_WP_AUTO_NORMALIZE : 0));
807
808         if(mmd && !mmd->sculptlvl)
809                 has_multires = 0;
810
811         if(!skipVirtualArmature) {
812                 firstmd = modifiers_getVirtualModifierList(ob);
813         }
814         else {
815                 /* game engine exception */
816                 firstmd = ob->modifiers.first;
817                 if(firstmd && firstmd->type == eModifierType_Armature)
818                         firstmd = firstmd->next;
819         }
820
821         md = firstmd;
822
823         modifiers_clearErrors(ob);
824
825         if(useRenderParams) required_mode = eModifierMode_Render;
826         else required_mode = eModifierMode_Realtime;
827
828         datamasks = modifiers_calcDataMasks(scene, ob, md, dataMask, required_mode);
829         curr = datamasks;
830
831         if(deform_r) *deform_r = NULL;
832         *final_r = NULL;
833
834         if(useDeform) {
835                 if(inputVertexCos)
836                         deformedVerts = inputVertexCos;
837                 
838                 /* Apply all leading deforming modifiers */
839                 for(;md; md = md->next, curr = curr->next) {
840                         ModifierTypeInfo *mti = modifierType_getInfo(md->type);
841
842                         md->scene= scene;
843                         
844                         if(!modifier_isEnabled(scene, md, required_mode)) continue;
845                         if(useDeform < 0 && mti->dependsOnTime && mti->dependsOnTime(md)) continue;
846
847                         if(mti->type == eModifierTypeType_OnlyDeform) {
848                                 if(!deformedVerts)
849                                         deformedVerts = mesh_getVertexCos(me, &numVerts);
850
851                                 mti->deformVerts(md, ob, NULL, deformedVerts, numVerts, useRenderParams, useDeform);
852                         } else {
853                                 break;
854                         }
855                         
856                         /* grab modifiers until index i */
857                         if((index >= 0) && (modifiers_indexInObject(ob, md) >= index))
858                                 break;
859                 }
860
861                 /* Result of all leading deforming modifiers is cached for
862                  * places that wish to use the original mesh but with deformed
863                  * coordinates (vpaint, etc.)
864                  */
865                 if (deform_r) {
866                         *deform_r = CDDM_from_mesh(me, ob);
867
868                         if(deformedVerts) {
869                                 CDDM_apply_vert_coords(*deform_r, deformedVerts);
870                                 CDDM_calc_normals(*deform_r);
871                         }
872                 }
873         } else {
874                 /* default behaviour for meshes */
875                 if(inputVertexCos)
876                         deformedVerts = inputVertexCos;
877                 else
878                         deformedVerts = mesh_getVertexCos(me, &numVerts);
879         }
880
881
882         /* Now apply all remaining modifiers. If useDeform is off then skip
883          * OnlyDeform ones. 
884          */
885         dm = NULL;
886         orcodm = NULL;
887         clothorcodm = NULL;
888
889         for(;md; md = md->next, curr = curr->next) {
890                 ModifierTypeInfo *mti = modifierType_getInfo(md->type);
891
892                 md->scene= scene;
893
894                 if(!modifier_isEnabled(scene, md, required_mode)) continue;
895                 if(mti->type == eModifierTypeType_OnlyDeform && !useDeform) continue;
896                 if((mti->flags & eModifierTypeFlag_RequiresOriginalData) && dm) {
897                         modifier_setError(md, "Modifier requires original data, bad stack position.");
898                         continue;
899                 }
900                 if(sculpt_mode && (!has_multires || multires_applied)) {
901                         int unsupported= 0;
902
903                         if(scene->toolsettings->sculpt->flags & SCULPT_ONLY_DEFORM)
904                                 unsupported|= mti->type != eModifierTypeType_OnlyDeform;
905
906                         unsupported|= md->type == eModifierType_Multires && ((MultiresModifierData*)md)->sculptlvl==0;
907                         unsupported|= multires_applied;
908
909                         if(unsupported) {
910                                 modifier_setError(md, "Not supported in sculpt mode.");
911                                 continue;
912                         }
913                 }
914                 if(needMapping && !modifier_supportsMapping(md)) continue;
915                 if(useDeform < 0 && mti->dependsOnTime && mti->dependsOnTime(md)) continue;
916
917                 /* add an orco layer if needed by this modifier */
918                 if(mti->requiredDataMask)
919                         mask = mti->requiredDataMask(ob, md);
920                 else
921                         mask = 0;
922
923                 if(dm && (mask & CD_MASK_ORCO))
924                         add_orco_dm(ob, NULL, dm, orcodm, CD_ORCO);
925
926                 /* How to apply modifier depends on (a) what we already have as
927                  * a result of previous modifiers (could be a DerivedMesh or just
928                  * deformed vertices) and (b) what type the modifier is.
929                  */
930
931                 if(mti->type == eModifierTypeType_OnlyDeform) {
932                         /* No existing verts to deform, need to build them. */
933                         if(!deformedVerts) {
934                                 if(dm) {
935                                         /* Deforming a derived mesh, read the vertex locations
936                                          * out of the mesh and deform them. Once done with this
937                                          * run of deformers verts will be written back.
938                                          */
939                                         numVerts = dm->getNumVerts(dm);
940                                         deformedVerts =
941                                                 MEM_mallocN(sizeof(*deformedVerts) * numVerts, "dfmv");
942                                         dm->getVertCos(dm, deformedVerts);
943                                 } else {
944                                         deformedVerts = mesh_getVertexCos(me, &numVerts);
945                                 }
946                         }
947
948                         /* if this is not the last modifier in the stack then recalculate the normals
949                          * to avoid giving bogus normals to the next modifier see: [#23673] */
950                         if(isPrevDeform &&  mti->dependsOnNormals && mti->dependsOnNormals(md)) {
951                                 /* XXX, this covers bug #23673, but we may need normal calc for other types */
952                                 if(dm && dm->type == DM_TYPE_CDDM) {
953                                         CDDM_apply_vert_coords(dm, deformedVerts);
954                                         CDDM_calc_normals(dm);
955                                 }
956                         }
957
958                         mti->deformVerts(md, ob, dm, deformedVerts, numVerts, useRenderParams, useDeform);
959                 } else {
960                         DerivedMesh *ndm;
961
962                         /* determine which data layers are needed by following modifiers */
963                         if(curr->next)
964                                 nextmask= (CustomDataMask)GET_INT_FROM_POINTER(curr->next->link);
965                         else
966                                 nextmask= dataMask;
967
968                         /* apply vertex coordinates or build a DerivedMesh as necessary */
969                         if(dm) {
970                                 if(deformedVerts) {
971                                         DerivedMesh *tdm = CDDM_copy(dm);
972                                         dm->release(dm);
973                                         dm = tdm;
974
975                                         CDDM_apply_vert_coords(dm, deformedVerts);
976                                         CDDM_calc_normals(dm);
977                                 }
978                         } else {
979                                 dm = CDDM_from_mesh(me, ob);
980
981                                 if(deformedVerts) {
982                                         CDDM_apply_vert_coords(dm, deformedVerts);
983                                         CDDM_calc_normals(dm);
984                                 }
985
986                                 if((dataMask & CD_MASK_WEIGHT_MCOL) && (ob->mode & OB_MODE_WEIGHT_PAINT))
987                                         add_weight_mcol_dm(ob, dm, draw_flag); 
988
989                                 /* Constructive modifiers need to have an origindex
990                                  * otherwise they wont have anywhere to copy the data from.
991                                  *
992                                  * Also create ORIGINDEX data if any of the following modifiers
993                                  * requests it, this way Mirror, Solidify etc will keep ORIGINDEX
994                                  * data by using generic DM_copy_vert_data() functions.
995                                  */
996                                 if(needMapping || (nextmask & CD_MASK_ORIGINDEX)) {
997                                         /* calc */
998                                         DM_add_vert_layer(dm, CD_ORIGINDEX, CD_CALLOC, NULL);
999                                         DM_add_edge_layer(dm, CD_ORIGINDEX, CD_CALLOC, NULL);
1000                                         DM_add_face_layer(dm, CD_ORIGINDEX, CD_CALLOC, NULL);
1001
1002                                         range_vn_i(DM_get_vert_data_layer(dm, CD_ORIGINDEX), dm->numVertData, 0);
1003                                         range_vn_i(DM_get_edge_data_layer(dm, CD_ORIGINDEX), dm->numEdgeData, 0);
1004                                         range_vn_i(DM_get_face_data_layer(dm, CD_ORIGINDEX), dm->numFaceData, 0);
1005                                 }
1006                         }
1007
1008                         
1009                         /* set the DerivedMesh to only copy needed data */
1010                         mask= (CustomDataMask)GET_INT_FROM_POINTER(curr->link);
1011                         /* needMapping check here fixes bug [#28112], otherwise its
1012                          * possible that it wont be copied */
1013                         mask |= append_mask;
1014                         DM_set_only_copy(dm, mask | (needMapping ? CD_MASK_ORIGINDEX : 0));
1015                         
1016                         /* add cloth rest shape key if need */
1017                         if(mask & CD_MASK_CLOTH_ORCO)
1018                                 add_orco_dm(ob, NULL, dm, clothorcodm, CD_CLOTH_ORCO);
1019
1020                         /* add an origspace layer if needed */
1021                         if(((CustomDataMask)GET_INT_FROM_POINTER(curr->link)) & CD_MASK_ORIGSPACE)
1022                                 if(!CustomData_has_layer(&dm->faceData, CD_ORIGSPACE))
1023                                         DM_add_face_layer(dm, CD_ORIGSPACE, CD_DEFAULT, NULL);
1024
1025                         ndm = mti->applyModifier(md, ob, dm, useRenderParams, useCache);
1026
1027                         if(ndm) {
1028                                 /* if the modifier returned a new dm, release the old one */
1029                                 if(dm && dm != ndm) dm->release(dm);
1030
1031                                 dm = ndm;
1032
1033                                 if(deformedVerts) {
1034                                         if(deformedVerts != inputVertexCos)
1035                                                 MEM_freeN(deformedVerts);
1036
1037                                         deformedVerts = NULL;
1038                                 }
1039                         } 
1040
1041                         /* create an orco derivedmesh in parallel */
1042                         if(nextmask & CD_MASK_ORCO) {
1043                                 if(!orcodm)
1044                                         orcodm= create_orco_dm(ob, me, NULL, CD_ORCO);
1045
1046                                 nextmask &= ~CD_MASK_ORCO;
1047                                 DM_set_only_copy(orcodm, nextmask | CD_MASK_ORIGINDEX);
1048                                 ndm = mti->applyModifier(md, ob, orcodm, useRenderParams, 0);
1049
1050                                 if(ndm) {
1051                                         /* if the modifier returned a new dm, release the old one */
1052                                         if(orcodm && orcodm != ndm) orcodm->release(orcodm);
1053                                         orcodm = ndm;
1054                                 }
1055                         }
1056
1057                         /* create cloth orco derivedmesh in parallel */
1058                         if(nextmask & CD_MASK_CLOTH_ORCO) {
1059                                 if(!clothorcodm)
1060                                         clothorcodm= create_orco_dm(ob, me, NULL, CD_CLOTH_ORCO);
1061
1062                                 nextmask &= ~CD_MASK_CLOTH_ORCO;
1063                                 DM_set_only_copy(clothorcodm, nextmask | CD_MASK_ORIGINDEX);
1064                                 ndm = mti->applyModifier(md, ob, clothorcodm, useRenderParams, 0);
1065
1066                                 if(ndm) {
1067                                         /* if the modifier returned a new dm, release the old one */
1068                                         if(clothorcodm && clothorcodm != ndm) clothorcodm->release(clothorcodm);
1069                                         clothorcodm = ndm;
1070                                 }
1071                         }
1072
1073                         /* in case of dynamic paint, make sure preview mask remains for following modifiers */
1074                         if (md->type == eModifierType_DynamicPaint)
1075                                 append_mask |= CD_MASK_WEIGHT_MCOL;
1076                 }
1077
1078                 isPrevDeform= (mti->type == eModifierTypeType_OnlyDeform);
1079
1080                 /* grab modifiers until index i */
1081                 if((index >= 0) && (modifiers_indexInObject(ob, md) >= index))
1082                         break;
1083
1084                 if(sculpt_mode && md->type == eModifierType_Multires)
1085                         multires_applied = 1;
1086         }
1087
1088         for(md=firstmd; md; md=md->next)
1089                 modifier_freeTemporaryData(md);
1090
1091         /* Yay, we are done. If we have a DerivedMesh and deformed vertices
1092          * need to apply these back onto the DerivedMesh. If we have no
1093          * DerivedMesh then we need to build one.
1094          */
1095         if(dm && deformedVerts) {
1096                 finaldm = CDDM_copy(dm);
1097
1098                 dm->release(dm);
1099
1100                 CDDM_apply_vert_coords(finaldm, deformedVerts);
1101                 CDDM_calc_normals(finaldm);
1102
1103                 if((dataMask & CD_MASK_WEIGHT_MCOL) && (ob->mode & OB_MODE_WEIGHT_PAINT))
1104                         add_weight_mcol_dm(ob, finaldm, draw_flag);
1105         } else if(dm) {
1106                 finaldm = dm;
1107         } else {
1108                 finaldm = CDDM_from_mesh(me, ob);
1109
1110                 if(deformedVerts) {
1111                         CDDM_apply_vert_coords(finaldm, deformedVerts);
1112                         CDDM_calc_normals(finaldm);
1113                 }
1114
1115                 if((dataMask & CD_MASK_WEIGHT_MCOL) && (ob->mode & OB_MODE_WEIGHT_PAINT))
1116                         add_weight_mcol_dm(ob, finaldm, draw_flag);
1117         }
1118
1119         /* add an orco layer if needed */
1120         if(dataMask & CD_MASK_ORCO) {
1121                 add_orco_dm(ob, NULL, finaldm, orcodm, CD_ORCO);
1122
1123                 if(deform_r && *deform_r)
1124                         add_orco_dm(ob, NULL, *deform_r, NULL, CD_ORCO);
1125         }
1126
1127 #ifdef WITH_GAMEENGINE
1128         /* NavMesh - this is a hack but saves having a NavMesh modifier */
1129         if ((ob->gameflag & OB_NAVMESH) && (finaldm->type == DM_TYPE_CDDM)) {
1130                 DerivedMesh *tdm;
1131                 tdm= navmesh_dm_createNavMeshForVisualization(finaldm);
1132                 if (finaldm != tdm) {
1133                         finaldm->release(finaldm);
1134                         finaldm= tdm;
1135                 }
1136         }
1137 #endif /* WITH_GAMEENGINE */
1138
1139         *final_r = finaldm;
1140
1141         if(orcodm)
1142                 orcodm->release(orcodm);
1143         if(clothorcodm)
1144                 clothorcodm->release(clothorcodm);
1145
1146         if(deformedVerts && deformedVerts != inputVertexCos)
1147                 MEM_freeN(deformedVerts);
1148
1149         BLI_linklist_free(datamasks, NULL);
1150 }
1151
1152 float (*editmesh_get_vertex_cos(EditMesh *em, int *numVerts_r))[3]
1153 {
1154         int i, numVerts = *numVerts_r = BLI_countlist(&em->verts);
1155         float (*cos)[3];
1156         EditVert *eve;
1157
1158         cos = MEM_mallocN(sizeof(*cos)*numVerts, "vertexcos");
1159         for (i=0,eve=em->verts.first; i<numVerts; i++,eve=eve->next) {
1160                 copy_v3_v3(cos[i], eve->co);
1161         }
1162
1163         return cos;
1164 }
1165
1166 int editmesh_modifier_is_enabled(Scene *scene, ModifierData *md, DerivedMesh *dm)
1167 {
1168         ModifierTypeInfo *mti = modifierType_getInfo(md->type);
1169         int required_mode = eModifierMode_Realtime | eModifierMode_Editmode;
1170
1171         if(!modifier_isEnabled(scene, md, required_mode)) return 0;
1172         if((mti->flags & eModifierTypeFlag_RequiresOriginalData) && dm) {
1173                 modifier_setError(md, "Modifier requires original data, bad stack position.");
1174                 return 0;
1175         }
1176         
1177         return 1;
1178 }
1179
1180 static void editmesh_calc_modifiers(Scene *scene, Object *ob, EditMesh *em, DerivedMesh **cage_r,
1181                                                                         DerivedMesh **final_r,
1182                                                                         CustomDataMask dataMask)
1183 {
1184         ModifierData *md;
1185         float (*deformedVerts)[3] = NULL;
1186         CustomDataMask mask;
1187         DerivedMesh *dm, *orcodm = NULL;
1188         int i, numVerts = 0, cageIndex = modifiers_getCageIndex(scene, ob, NULL, 1);
1189         LinkNode *datamasks, *curr;
1190         int required_mode = eModifierMode_Realtime | eModifierMode_Editmode;
1191
1192         modifiers_clearErrors(ob);
1193
1194         if(cage_r && cageIndex == -1) {
1195                 *cage_r = editmesh_get_derived(em, NULL);
1196         }
1197
1198         dm = NULL;
1199         md = modifiers_getVirtualModifierList(ob);
1200
1201         datamasks = modifiers_calcDataMasks(scene, ob, md, dataMask, required_mode);
1202
1203         curr = datamasks;
1204         for(i = 0; md; i++, md = md->next, curr = curr->next) {
1205                 ModifierTypeInfo *mti = modifierType_getInfo(md->type);
1206
1207                 md->scene= scene;
1208                 
1209                 if(!editmesh_modifier_is_enabled(scene, md, dm))
1210                         continue;
1211
1212                 /* add an orco layer if needed by this modifier */
1213                 if(dm && mti->requiredDataMask) {
1214                         mask = mti->requiredDataMask(ob, md);
1215                         if(mask & CD_MASK_ORCO)
1216                                 add_orco_dm(ob, em, dm, orcodm, CD_ORCO);
1217                 }
1218
1219                 /* How to apply modifier depends on (a) what we already have as
1220                  * a result of previous modifiers (could be a DerivedMesh or just
1221                  * deformed vertices) and (b) what type the modifier is.
1222                  */
1223
1224                 if(mti->type == eModifierTypeType_OnlyDeform) {
1225                         /* No existing verts to deform, need to build them. */
1226                         if(!deformedVerts) {
1227                                 if(dm) {
1228                                         /* Deforming a derived mesh, read the vertex locations
1229                                          * out of the mesh and deform them. Once done with this
1230                                          * run of deformers verts will be written back.
1231                                          */
1232                                         numVerts = dm->getNumVerts(dm);
1233                                         deformedVerts =
1234                                                 MEM_mallocN(sizeof(*deformedVerts) * numVerts, "dfmv");
1235                                         dm->getVertCos(dm, deformedVerts);
1236                                 } else {
1237                                         deformedVerts = editmesh_get_vertex_cos(em, &numVerts);
1238                                 }
1239                         }
1240
1241                         if (mti->deformVertsEM)
1242                                 mti->deformVertsEM(md, ob, em, dm, deformedVerts, numVerts);
1243                         else mti->deformVerts(md, ob, dm, deformedVerts, numVerts, 0, 0);
1244                 } else {
1245                         DerivedMesh *ndm;
1246
1247                         /* apply vertex coordinates or build a DerivedMesh as necessary */
1248                         if(dm) {
1249                                 if(deformedVerts) {
1250                                         DerivedMesh *tdm = CDDM_copy(dm);
1251                                         if(!(cage_r && dm == *cage_r)) dm->release(dm);
1252                                         dm = tdm;
1253
1254                                         CDDM_apply_vert_coords(dm, deformedVerts);
1255                                         CDDM_calc_normals(dm);
1256                                 } else if(cage_r && dm == *cage_r) {
1257                                         /* dm may be changed by this modifier, so we need to copy it
1258                                          */
1259                                         dm = CDDM_copy(dm);
1260                                 }
1261
1262                         } else {
1263                                 dm = CDDM_from_editmesh(em, ob->data);
1264
1265                                 if(deformedVerts) {
1266                                         CDDM_apply_vert_coords(dm, deformedVerts);
1267                                         CDDM_calc_normals(dm);
1268                                 }
1269                         }
1270
1271                         /* create an orco derivedmesh in parallel */
1272                         mask= (CustomDataMask)GET_INT_FROM_POINTER(curr->link);
1273                         if(mask & CD_MASK_ORCO) {
1274                                 if(!orcodm)
1275                                         orcodm= create_orco_dm(ob, ob->data, em, CD_ORCO);
1276
1277                                 mask &= ~CD_MASK_ORCO;
1278                                 DM_set_only_copy(orcodm, mask | CD_MASK_ORIGINDEX);
1279
1280                                 if (mti->applyModifierEM)
1281                                         ndm = mti->applyModifierEM(md, ob, em, orcodm);
1282                                 else
1283                                         ndm = mti->applyModifier(md, ob, orcodm, 0, 0);
1284
1285                                 if(ndm) {
1286                                         /* if the modifier returned a new dm, release the old one */
1287                                         if(orcodm && orcodm != ndm) orcodm->release(orcodm);
1288                                         orcodm = ndm;
1289                                 }
1290                         }
1291
1292                         /* set the DerivedMesh to only copy needed data */
1293                         mask= (CustomDataMask)GET_INT_FROM_POINTER(curr->link); /* CD_MASK_ORCO may have been cleared above */
1294
1295                         DM_set_only_copy(dm, mask | CD_MASK_ORIGINDEX);
1296
1297                         if(mask & CD_MASK_ORIGSPACE)
1298                                 if(!CustomData_has_layer(&dm->faceData, CD_ORIGSPACE))
1299                                         DM_add_face_layer(dm, CD_ORIGSPACE, CD_DEFAULT, NULL);
1300                         
1301                         if (mti->applyModifierEM)
1302                                 ndm = mti->applyModifierEM(md, ob, em, dm);
1303                         else
1304                                 ndm = mti->applyModifier(md, ob, dm, 0, 0);
1305
1306                         if (ndm) {
1307                                 if(dm && dm != ndm)
1308                                         dm->release(dm);
1309
1310                                 dm = ndm;
1311
1312                                 if (deformedVerts) {
1313                                         MEM_freeN(deformedVerts);
1314                                         deformedVerts = NULL;
1315                                 }
1316                         }
1317                 }
1318
1319                 if(cage_r && i == cageIndex) {
1320                         if(dm && deformedVerts) {
1321                                 *cage_r = CDDM_copy(dm);
1322                                 CDDM_apply_vert_coords(*cage_r, deformedVerts);
1323                         } else if(dm) {
1324                                 *cage_r = dm;
1325                         } else {
1326                                 *cage_r =
1327                                         editmesh_get_derived(em,
1328                                                 deformedVerts ? MEM_dupallocN(deformedVerts) : NULL);
1329                         }
1330                 }
1331         }
1332
1333         BLI_linklist_free(datamasks, NULL);
1334
1335         /* Yay, we are done. If we have a DerivedMesh and deformed vertices need
1336          * to apply these back onto the DerivedMesh. If we have no DerivedMesh
1337          * then we need to build one.
1338          */
1339         if(dm && deformedVerts) {
1340                 *final_r = CDDM_copy(dm);
1341
1342                 if(!(cage_r && dm == *cage_r)) dm->release(dm);
1343
1344                 CDDM_apply_vert_coords(*final_r, deformedVerts);
1345                 CDDM_calc_normals(*final_r);
1346         } else if (dm) {
1347                 *final_r = dm;
1348         } else if (!deformedVerts && cage_r && *cage_r) {
1349                 *final_r = *cage_r;
1350         } else {
1351                 *final_r = editmesh_get_derived(em, deformedVerts);
1352                 deformedVerts = NULL;
1353         }
1354
1355         /* add an orco layer if needed */
1356         if(dataMask & CD_MASK_ORCO)
1357                 add_orco_dm(ob, em, *final_r, orcodm, CD_ORCO);
1358
1359         if(orcodm)
1360                 orcodm->release(orcodm);
1361
1362         if(deformedVerts)
1363                 MEM_freeN(deformedVerts);
1364 }
1365
1366 static void clear_mesh_caches(Object *ob)
1367 {
1368         Mesh *me= ob->data;
1369
1370                 /* also serves as signal to remake texspace */
1371         if (ob->bb) {
1372                 MEM_freeN(ob->bb);
1373                 ob->bb = NULL;
1374         }
1375         if (me->bb) {
1376                 MEM_freeN(me->bb);
1377                 me->bb = NULL;
1378         }
1379
1380         freedisplist(&ob->disp);
1381
1382         if (ob->derivedFinal) {
1383                 ob->derivedFinal->needsFree = 1;
1384                 ob->derivedFinal->release(ob->derivedFinal);
1385                 ob->derivedFinal= NULL;
1386         }
1387         if (ob->derivedDeform) {
1388                 ob->derivedDeform->needsFree = 1;
1389                 ob->derivedDeform->release(ob->derivedDeform);
1390                 ob->derivedDeform= NULL;
1391         }
1392
1393         if(ob->sculpt) {
1394                 object_sculpt_modifiers_changed(ob);
1395         }
1396 }
1397
1398 static void mesh_build_data(Scene *scene, Object *ob, CustomDataMask dataMask)
1399 {
1400         Object *obact = scene->basact?scene->basact->object:NULL;
1401         int editing = paint_facesel_test(ob) || paint_vertsel_test(ob);/* paint_vertsel_test */
1402         /* weight paint and face select need original indices because of selection buffer drawing */
1403         int needMapping = (ob==obact) && (editing || (ob->mode & (OB_MODE_WEIGHT_PAINT|OB_MODE_VERTEX_PAINT)));
1404
1405         clear_mesh_caches(ob);
1406
1407         mesh_calc_modifiers(scene, ob, NULL, &ob->derivedDeform,
1408                                                 &ob->derivedFinal, 0, 1,
1409                                                 needMapping, dataMask, -1, 1);
1410
1411         DM_set_object_boundbox (ob, ob->derivedFinal);
1412
1413         ob->derivedFinal->needsFree = 0;
1414         ob->derivedDeform->needsFree = 0;
1415         ob->lastDataMask = dataMask;
1416 }
1417
1418 static void editmesh_build_data(Scene *scene, Object *obedit, EditMesh *em, CustomDataMask dataMask)
1419 {
1420         clear_mesh_caches(obedit);
1421
1422         if (em->derivedFinal) {
1423                 if (em->derivedFinal!=em->derivedCage) {
1424                         em->derivedFinal->needsFree = 1;
1425                         em->derivedFinal->release(em->derivedFinal);
1426                 }
1427                 em->derivedFinal = NULL;
1428         }
1429         if (em->derivedCage) {
1430                 em->derivedCage->needsFree = 1;
1431                 em->derivedCage->release(em->derivedCage);
1432                 em->derivedCage = NULL;
1433         }
1434
1435         editmesh_calc_modifiers(scene, obedit, em, &em->derivedCage, &em->derivedFinal, dataMask);
1436         DM_set_object_boundbox (obedit, em->derivedFinal);
1437
1438         em->lastDataMask = dataMask;
1439         em->derivedFinal->needsFree = 0;
1440         em->derivedCage->needsFree = 0;
1441 }
1442
1443 void makeDerivedMesh(Scene *scene, Object *ob, EditMesh *em, CustomDataMask dataMask)
1444 {
1445         if (em) {
1446                 editmesh_build_data(scene, ob, em, dataMask);
1447         } else {
1448                 mesh_build_data(scene, ob, dataMask);
1449         }
1450 }
1451
1452 /***/
1453
1454 DerivedMesh *mesh_get_derived_final(Scene *scene, Object *ob, CustomDataMask dataMask)
1455 {
1456         /* if there's no derived mesh or the last data mask used doesn't include
1457          * the data we need, rebuild the derived mesh
1458          */
1459         if(!ob->derivedFinal || (dataMask & ob->lastDataMask) != dataMask)
1460                 mesh_build_data(scene, ob, dataMask);
1461
1462         return ob->derivedFinal;
1463 }
1464
1465 DerivedMesh *mesh_get_derived_deform(Scene *scene, Object *ob, CustomDataMask dataMask)
1466 {
1467         /* if there's no derived mesh or the last data mask used doesn't include
1468          * the data we need, rebuild the derived mesh
1469          */
1470         if(!ob->derivedDeform || (dataMask & ob->lastDataMask) != dataMask)
1471                 mesh_build_data(scene, ob, dataMask);
1472
1473         return ob->derivedDeform;
1474 }
1475
1476 DerivedMesh *mesh_create_derived_render(Scene *scene, Object *ob, CustomDataMask dataMask)
1477 {
1478         DerivedMesh *final;
1479         
1480         mesh_calc_modifiers(scene, ob, NULL, NULL, &final, 1, 1, 0, dataMask, -1, 0);
1481
1482         return final;
1483 }
1484
1485 DerivedMesh *mesh_create_derived_index_render(Scene *scene, Object *ob, CustomDataMask dataMask, int index)
1486 {
1487         DerivedMesh *final;
1488         
1489         mesh_calc_modifiers(scene, ob, NULL, NULL, &final, 1, 1, 0, dataMask, index, 0);
1490
1491         return final;
1492 }
1493
1494 DerivedMesh *mesh_create_derived_view(Scene *scene, Object *ob, CustomDataMask dataMask)
1495 {
1496         DerivedMesh *final;
1497
1498         mesh_calc_modifiers(scene, ob, NULL, NULL, &final, 0, 1, 0, dataMask, -1, 0);
1499
1500         return final;
1501 }
1502
1503 DerivedMesh *mesh_create_derived_no_deform(Scene *scene, Object *ob, float (*vertCos)[3],
1504                                                                                    CustomDataMask dataMask)
1505 {
1506         DerivedMesh *final;
1507         
1508         mesh_calc_modifiers(scene, ob, vertCos, NULL, &final, 0, 0, 0, dataMask, -1, 0);
1509
1510         return final;
1511 }
1512
1513 DerivedMesh *mesh_create_derived_no_virtual(Scene *scene, Object *ob, float (*vertCos)[3],
1514                                                                                         CustomDataMask dataMask)
1515 {
1516         DerivedMesh *final;
1517         
1518         mesh_calc_modifiers(scene, ob, vertCos, NULL, &final, 0, -1, 0, dataMask, -1, 0);
1519
1520         return final;
1521 }
1522
1523 DerivedMesh *mesh_create_derived_physics(Scene *scene, Object *ob, float (*vertCos)[3],
1524                                                                                         CustomDataMask dataMask)
1525 {
1526         DerivedMesh *final;
1527         
1528         mesh_calc_modifiers(scene, ob, vertCos, NULL, &final, 0, -1, 1, dataMask, -1, 0);
1529
1530         return final;
1531 }
1532
1533 DerivedMesh *mesh_create_derived_no_deform_render(Scene *scene, Object *ob,
1534                                                                                                   float (*vertCos)[3],
1535                                                                                                   CustomDataMask dataMask)
1536 {
1537         DerivedMesh *final;
1538
1539         mesh_calc_modifiers(scene, ob, vertCos, NULL, &final, 1, 0, 0, dataMask, -1, 0);
1540
1541         return final;
1542 }
1543
1544 /***/
1545
1546 DerivedMesh *editmesh_get_derived_cage_and_final(Scene *scene, Object *obedit, EditMesh *em, DerivedMesh **final_r,
1547                                                                                                  CustomDataMask dataMask)
1548 {
1549         /* if there's no derived mesh or the last data mask used doesn't include
1550          * the data we need, rebuild the derived mesh
1551          */
1552         if(!em->derivedCage ||
1553            (em->lastDataMask & dataMask) != dataMask)
1554                 editmesh_build_data(scene, obedit, em, dataMask);
1555
1556         *final_r = em->derivedFinal;
1557         return em->derivedCage;
1558 }
1559
1560 DerivedMesh *editmesh_get_derived_cage(Scene *scene, Object *obedit, EditMesh *em, CustomDataMask dataMask)
1561 {
1562         /* if there's no derived mesh or the last data mask used doesn't include
1563          * the data we need, rebuild the derived mesh
1564          */
1565         if(!em->derivedCage ||
1566            (em->lastDataMask & dataMask) != dataMask)
1567                 editmesh_build_data(scene, obedit, em, dataMask);
1568
1569         return em->derivedCage;
1570 }
1571
1572 DerivedMesh *editmesh_get_derived_base(Object *UNUSED(obedit), EditMesh *em)
1573 {
1574         return editmesh_get_derived(em, NULL);
1575 }
1576
1577
1578 /* ********* For those who don't grasp derived stuff! (ton) :) *************** */
1579
1580 static void make_vertexcosnos__mapFunc(void *userData, int index, float *co, float *no_f, short *no_s)
1581 {
1582         float *vec = userData;
1583         
1584         vec+= 6*index;
1585
1586         /* check if we've been here before (normal should not be 0) */
1587         if(vec[3] || vec[4] || vec[5]) return;
1588
1589         copy_v3_v3(vec, co);
1590         vec+= 3;
1591         if(no_f) {
1592                 copy_v3_v3(vec, no_f);
1593         }
1594         else {
1595                 normal_short_to_float_v3(vec, no_s);
1596         }
1597 }
1598
1599 /* always returns original amount me->totvert of vertices and normals, but fully deformed and subsurfered */
1600 /* this is needed for all code using vertexgroups (no subsurf support) */
1601 /* it stores the normals as floats, but they can still be scaled as shorts (32767 = unit) */
1602 /* in use now by vertex/weight paint and particle generating */
1603
1604 float *mesh_get_mapped_verts_nors(Scene *scene, Object *ob)
1605 {
1606         Mesh *me= ob->data;
1607         DerivedMesh *dm;
1608         float *vertexcosnos;
1609         
1610         /* lets prevent crashing... */
1611         if(ob->type!=OB_MESH || me->totvert==0)
1612                 return NULL;
1613         
1614         dm= mesh_get_derived_final(scene, ob, CD_MASK_BAREMESH);
1615         vertexcosnos= MEM_callocN(6*sizeof(float)*me->totvert, "vertexcosnos map");
1616         
1617         if(dm->foreachMappedVert) {
1618                 dm->foreachMappedVert(dm, make_vertexcosnos__mapFunc, vertexcosnos);
1619         }
1620         else {
1621                 float *fp= vertexcosnos;
1622                 int a;
1623                 
1624                 for(a=0; a< me->totvert; a++, fp+=6) {
1625                         dm->getVertCo(dm, a, fp);
1626                         dm->getVertNo(dm, a, fp+3);
1627                 }
1628         }
1629         
1630         dm->release(dm);
1631         return vertexcosnos;
1632 }
1633
1634 /* ******************* GLSL ******************** */
1635
1636 typedef struct
1637 {
1638         float * precomputedFaceNormals;
1639         MTFace * mtface;        // texture coordinates
1640         MFace * mface;          // indices
1641         MVert * mvert;          // vertices & normals
1642         float (*orco)[3];
1643         float (*tangent)[4];    // destination
1644         int numFaces;
1645
1646 } SGLSLMeshToTangent;
1647
1648 // interface
1649 #include "mikktspace.h"
1650
1651 static int GetNumFaces(const SMikkTSpaceContext * pContext)
1652 {
1653         SGLSLMeshToTangent * pMesh = (SGLSLMeshToTangent *) pContext->m_pUserData;
1654         return pMesh->numFaces;
1655 }
1656
1657 static int GetNumVertsOfFace(const SMikkTSpaceContext * pContext, const int face_num)
1658 {
1659         SGLSLMeshToTangent * pMesh = (SGLSLMeshToTangent *) pContext->m_pUserData;
1660         return pMesh->mface[face_num].v4!=0 ? 4 : 3;
1661 }
1662
1663 static void GetPosition(const SMikkTSpaceContext * pContext, float fPos[], const int face_num, const int vert_index)
1664 {
1665         //assert(vert_index>=0 && vert_index<4);
1666         SGLSLMeshToTangent * pMesh = (SGLSLMeshToTangent *) pContext->m_pUserData;
1667         const float *co= pMesh->mvert[(&pMesh->mface[face_num].v1)[vert_index]].co;
1668         copy_v3_v3(fPos, co);
1669 }
1670
1671 static void GetTextureCoordinate(const SMikkTSpaceContext * pContext, float fUV[], const int face_num, const int vert_index)
1672 {
1673         //assert(vert_index>=0 && vert_index<4);
1674         SGLSLMeshToTangent * pMesh = (SGLSLMeshToTangent *) pContext->m_pUserData;
1675
1676         if(pMesh->mtface!=NULL) {
1677                 float * uv = pMesh->mtface[face_num].uv[vert_index];
1678                 fUV[0]=uv[0]; fUV[1]=uv[1];
1679         }
1680         else {
1681                 const float *orco= pMesh->orco[(&pMesh->mface[face_num].v1)[vert_index]];
1682                 map_to_sphere( &fUV[0], &fUV[1], orco[0], orco[1], orco[2]);
1683         }
1684 }
1685
1686 static void GetNormal(const SMikkTSpaceContext * pContext, float fNorm[], const int face_num, const int vert_index)
1687 {
1688         //assert(vert_index>=0 && vert_index<4);
1689         SGLSLMeshToTangent * pMesh = (SGLSLMeshToTangent *) pContext->m_pUserData;
1690
1691         const int smoothnormal = (pMesh->mface[face_num].flag & ME_SMOOTH);
1692         if(!smoothnormal) {     // flat
1693                 if(pMesh->precomputedFaceNormals) {
1694                         copy_v3_v3(fNorm, &pMesh->precomputedFaceNormals[3*face_num]);
1695                 }
1696                 else {
1697                         MFace *mf= &pMesh->mface[face_num];
1698                         float *p0= pMesh->mvert[mf->v1].co;
1699                         float *p1= pMesh->mvert[mf->v2].co;
1700                         float *p2= pMesh->mvert[mf->v3].co;
1701
1702                         if(mf->v4) {
1703                                 float *p3 = pMesh->mvert[mf->v4].co;
1704                                 normal_quad_v3(fNorm, p0, p1, p2, p3);
1705                         }
1706                         else {
1707                                 normal_tri_v3(fNorm, p0, p1, p2);
1708                         }
1709                 }
1710         }
1711         else {
1712                 const short *no= pMesh->mvert[(&pMesh->mface[face_num].v1)[vert_index]].no;
1713                 normal_short_to_float_v3(fNorm, no);
1714         }
1715 }
1716 static void SetTSpace(const SMikkTSpaceContext * pContext, const float fvTangent[], const float fSign, const int face_num, const int iVert)
1717 {
1718         //assert(vert_index>=0 && vert_index<4);
1719         SGLSLMeshToTangent * pMesh = (SGLSLMeshToTangent *) pContext->m_pUserData;
1720         float * pRes = pMesh->tangent[4*face_num+iVert];
1721         copy_v3_v3(pRes, fvTangent);
1722         pRes[3]=fSign;
1723 }
1724
1725
1726 void DM_add_tangent_layer(DerivedMesh *dm)
1727 {
1728         /* mesh vars */
1729         MTFace *mtface, *tf;
1730         MFace *mface, *mf;
1731         MVert *mvert, *v1, *v2, *v3, *v4;
1732         MemArena *arena= NULL;
1733         VertexTangent **vtangents= NULL;
1734         float (*orco)[3]= NULL, (*tangent)[4];
1735         float *uv1, *uv2, *uv3, *uv4, *vtang;
1736         float fno[3], tang[3], uv[4][2];
1737         int i, j, len, mf_vi[4], totvert, totface, iCalcNewMethod;
1738         float *nors;
1739
1740         if(CustomData_get_layer_index(&dm->faceData, CD_TANGENT) != -1)
1741                 return;
1742
1743         nors = dm->getFaceDataArray(dm, CD_NORMAL);
1744
1745         /* check we have all the needed layers */
1746         totvert= dm->getNumVerts(dm);
1747         totface= dm->getNumFaces(dm);
1748
1749         mvert= dm->getVertArray(dm);
1750         mface= dm->getFaceArray(dm);
1751         mtface= dm->getFaceDataArray(dm, CD_MTFACE);
1752
1753         if(!mtface) {
1754                 orco= dm->getVertDataArray(dm, CD_ORCO);
1755                 if(!orco)
1756                         return;
1757         }
1758         
1759         /* create tangent layer */
1760         DM_add_face_layer(dm, CD_TANGENT, CD_CALLOC, NULL);
1761         tangent= DM_get_face_data_layer(dm, CD_TANGENT);
1762         
1763         /* allocate some space */
1764         arena= BLI_memarena_new(BLI_MEMARENA_STD_BUFSIZE, "tangent layer arena");
1765         BLI_memarena_use_calloc(arena);
1766         vtangents= MEM_callocN(sizeof(VertexTangent*)*totvert, "VertexTangent");
1767
1768         // new computation method
1769         iCalcNewMethod = 1;
1770         if(iCalcNewMethod != 0) {
1771                 SGLSLMeshToTangent mesh2tangent= {0};
1772                 SMikkTSpaceContext sContext= {0};
1773                 SMikkTSpaceInterface sInterface= {0};
1774
1775                 mesh2tangent.precomputedFaceNormals = nors;
1776                 mesh2tangent.mtface = mtface;
1777                 mesh2tangent.mface = mface;
1778                 mesh2tangent.mvert = mvert;
1779                 mesh2tangent.orco = orco;
1780                 mesh2tangent.tangent = tangent;
1781                 mesh2tangent.numFaces = totface;
1782
1783                 sContext.m_pUserData = &mesh2tangent;
1784                 sContext.m_pInterface = &sInterface;
1785                 sInterface.m_getNumFaces = GetNumFaces;
1786                 sInterface.m_getNumVerticesOfFace = GetNumVertsOfFace;
1787                 sInterface.m_getPosition = GetPosition;
1788                 sInterface.m_getTexCoord = GetTextureCoordinate;
1789                 sInterface.m_getNormal = GetNormal;
1790                 sInterface.m_setTSpaceBasic = SetTSpace;
1791
1792                 // 0 if failed
1793                 iCalcNewMethod = genTangSpaceDefault(&sContext);
1794         }
1795
1796         if(!iCalcNewMethod) {
1797                 /* sum tangents at connected vertices */
1798                 for(i=0, tf=mtface, mf=mface; i < totface; mf++, tf++, i++) {
1799                         v1= &mvert[mf->v1];
1800                         v2= &mvert[mf->v2];
1801                         v3= &mvert[mf->v3];
1802
1803                         if (mf->v4) {
1804                                 v4= &mvert[mf->v4];
1805                                 normal_quad_v3( fno,v4->co, v3->co, v2->co, v1->co);
1806                         }
1807                         else {
1808                                 v4= NULL;
1809                                 normal_tri_v3( fno,v3->co, v2->co, v1->co);
1810                         }
1811                 
1812                         if(mtface) {
1813                                 uv1= tf->uv[0];
1814                                 uv2= tf->uv[1];
1815                                 uv3= tf->uv[2];
1816                                 uv4= tf->uv[3];
1817                         }
1818                         else {
1819                                 uv1= uv[0]; uv2= uv[1]; uv3= uv[2]; uv4= uv[3];
1820                                 map_to_sphere( &uv[0][0], &uv[0][1],orco[mf->v1][0], orco[mf->v1][1], orco[mf->v1][2]);
1821                                 map_to_sphere( &uv[1][0], &uv[1][1],orco[mf->v2][0], orco[mf->v2][1], orco[mf->v2][2]);
1822                                 map_to_sphere( &uv[2][0], &uv[2][1],orco[mf->v3][0], orco[mf->v3][1], orco[mf->v3][2]);
1823                                 if(v4)
1824                                         map_to_sphere( &uv[3][0], &uv[3][1],orco[mf->v4][0], orco[mf->v4][1], orco[mf->v4][2]);
1825                         }
1826                 
1827                         tangent_from_uv(uv1, uv2, uv3, v1->co, v2->co, v3->co, fno, tang);
1828                         sum_or_add_vertex_tangent(arena, &vtangents[mf->v1], tang, uv1);
1829                         sum_or_add_vertex_tangent(arena, &vtangents[mf->v2], tang, uv2);
1830                         sum_or_add_vertex_tangent(arena, &vtangents[mf->v3], tang, uv3);
1831                 
1832                         if(mf->v4) {
1833                                 v4= &mvert[mf->v4];
1834                         
1835                                 tangent_from_uv(uv1, uv3, uv4, v1->co, v3->co, v4->co, fno, tang);
1836                                 sum_or_add_vertex_tangent(arena, &vtangents[mf->v1], tang, uv1);
1837                                 sum_or_add_vertex_tangent(arena, &vtangents[mf->v3], tang, uv3);
1838                                 sum_or_add_vertex_tangent(arena, &vtangents[mf->v4], tang, uv4);
1839                         }
1840                 }
1841         
1842                 /* write tangent to layer */
1843                 for(i=0, tf=mtface, mf=mface; i < totface; mf++, tf++, i++, tangent+=4) {
1844                         len= (mf->v4)? 4 : 3; 
1845
1846                         if(mtface == NULL) {
1847                                 map_to_sphere( &uv[0][0], &uv[0][1],orco[mf->v1][0], orco[mf->v1][1], orco[mf->v1][2]);
1848                                 map_to_sphere( &uv[1][0], &uv[1][1],orco[mf->v2][0], orco[mf->v2][1], orco[mf->v2][2]);
1849                                 map_to_sphere( &uv[2][0], &uv[2][1],orco[mf->v3][0], orco[mf->v3][1], orco[mf->v3][2]);
1850                                 if(len==4)
1851                                         map_to_sphere( &uv[3][0], &uv[3][1],orco[mf->v4][0], orco[mf->v4][1], orco[mf->v4][2]);
1852                         }
1853                 
1854                         mf_vi[0]= mf->v1;
1855                         mf_vi[1]= mf->v2;
1856                         mf_vi[2]= mf->v3;
1857                         mf_vi[3]= mf->v4;
1858                 
1859                         for(j=0; j<len; j++) {
1860                                 vtang= find_vertex_tangent(vtangents[mf_vi[j]], mtface ? tf->uv[j] : uv[j]);
1861                                 normalize_v3_v3(tangent[j], vtang);
1862                                 ((float *) tangent[j])[3]=1.0f;
1863                         }
1864                 }
1865         }
1866         
1867         BLI_memarena_free(arena);
1868         MEM_freeN(vtangents);
1869 }
1870
1871 void DM_calc_auto_bump_scale(DerivedMesh *dm)
1872 {
1873         /* int totvert= dm->getNumVerts(dm); */ /* UNUSED */
1874         int totface= dm->getNumFaces(dm);
1875
1876         MVert * mvert = dm->getVertArray(dm);
1877         MFace * mface = dm->getFaceArray(dm);
1878         MTFace * mtface = dm->getFaceDataArray(dm, CD_MTFACE);
1879
1880         if(mtface)
1881         {
1882                 double dsum = 0.0;
1883                 int nr_accumulated = 0;
1884                 int f;
1885
1886                 for ( f=0; f<totface; f++ )
1887                 {
1888                         {
1889                                 float * verts[4], * tex_coords[4];
1890                                 const int nr_verts = mface[f].v4!=0 ? 4 : 3;
1891                                 int i, is_degenerate;
1892
1893                                 verts[0]=mvert[mface[f].v1].co; verts[1]=mvert[mface[f].v2].co; verts[2]=mvert[mface[f].v3].co;
1894                                 tex_coords[0]=mtface[f].uv[0]; tex_coords[1]=mtface[f].uv[1]; tex_coords[2]=mtface[f].uv[2];
1895                                 if(nr_verts==4)
1896                                 {
1897                                         verts[3]=mvert[mface[f].v4].co;
1898                                         tex_coords[3]=mtface[f].uv[3];
1899                                 }
1900
1901                                 // discard degenerate faces
1902                                 is_degenerate = 0;
1903                                 if(     equals_v3v3(verts[0], verts[1]) || equals_v3v3(verts[0], verts[2]) || equals_v3v3(verts[1], verts[2]) ||
1904                                         equals_v2v2(tex_coords[0], tex_coords[1]) || equals_v2v2(tex_coords[0], tex_coords[2]) || equals_v2v2(tex_coords[1], tex_coords[2]) )
1905                                 {
1906                                         is_degenerate = 1;
1907                                 }
1908
1909                                 // verify last vertex as well if this is a quad
1910                                 if ( is_degenerate==0 && nr_verts==4 )
1911                                 {
1912                                         if(     equals_v3v3(verts[3], verts[0]) || equals_v3v3(verts[3], verts[1]) || equals_v3v3(verts[3], verts[2]) ||
1913                                                 equals_v2v2(tex_coords[3], tex_coords[0]) || equals_v2v2(tex_coords[3], tex_coords[1]) || equals_v2v2(tex_coords[3], tex_coords[2]) )
1914                                         {
1915                                                 is_degenerate = 1;
1916                                         }
1917
1918                                         // verify the winding is consistent
1919                                         if ( is_degenerate==0 )
1920                                         {
1921                                                 float prev_edge[2];
1922                                                 int is_signed = 0;
1923                                                 sub_v2_v2v2(prev_edge, tex_coords[0], tex_coords[3]);
1924
1925                                                 i = 0;
1926                                                 while ( is_degenerate==0 && i<4 )
1927                                                 {
1928                                                         float cur_edge[2], signed_area;
1929                                                         sub_v2_v2v2(cur_edge, tex_coords[(i+1)&0x3], tex_coords[i]);
1930                                                         signed_area = prev_edge[0]*cur_edge[1] - prev_edge[1]*cur_edge[0];
1931                                                         if ( i==0 ) is_signed = signed_area<0.0f ? 1 : 0;
1932                                                         else if((is_signed!=0)!=(signed_area<0.0f)) is_degenerate=1;
1933
1934                                                         if ( is_degenerate==0 )
1935                                                         {
1936                                                                 copy_v2_v2(prev_edge, cur_edge);
1937                                                                 ++i;
1938                                                         }
1939                                                 }
1940                                         }
1941                                 }
1942
1943                                 // proceed if not a degenerate face
1944                                 if ( is_degenerate==0 )
1945                                 {
1946                                         int nr_tris_to_pile=0;
1947                                         // quads split at shortest diagonal
1948                                         int offs = 0;           // initial triangulation is 0,1,2 and 0, 2, 3
1949                                         if ( nr_verts==4 )
1950                                         {
1951                                                 float pos_len_diag0, pos_len_diag1;
1952                                                 float vtmp[3];
1953                                                 sub_v3_v3v3(vtmp, verts[2], verts[0]);
1954                                                 pos_len_diag0 = dot_v3v3(vtmp, vtmp);
1955                                                 sub_v3_v3v3(vtmp, verts[3], verts[1]);
1956                                                 pos_len_diag1 = dot_v3v3(vtmp, vtmp);
1957
1958                                                 if(pos_len_diag1<pos_len_diag0)
1959                                                         offs=1;         // alter split
1960                                                 else if(pos_len_diag0==pos_len_diag1)           // do UV check instead
1961                                                 {
1962                                                         float tex_len_diag0, tex_len_diag1;
1963
1964                                                         sub_v2_v2v2(vtmp, tex_coords[2], tex_coords[0]);
1965                                                         tex_len_diag0 = dot_v2v2(vtmp, vtmp);
1966                                                         sub_v2_v2v2(vtmp, tex_coords[3], tex_coords[1]);
1967                                                         tex_len_diag1 = dot_v2v2(vtmp, vtmp);
1968
1969                                                         if(tex_len_diag1<tex_len_diag0)
1970                                                         {
1971                                                                 offs=1;         // alter split
1972                                                         }
1973                                                 }
1974                                         }
1975                                         nr_tris_to_pile = nr_verts-2 ;
1976                                         if ( nr_tris_to_pile==1 || nr_tris_to_pile==2 )
1977                                         {
1978                                                 const int indices[] = {offs+0, offs+1, offs+2, offs+0, offs+2, (offs+3)&0x3 };
1979                                                 int t;
1980                                                 for ( t=0; t<nr_tris_to_pile; t++ )
1981                                                 {
1982                                                         float f2x_area_uv;
1983                                                         float * p0 = verts[indices[t*3+0]];
1984                                                         float * p1 = verts[indices[t*3+1]];
1985                                                         float * p2 = verts[indices[t*3+2]];
1986
1987                                                         float edge_t0[2], edge_t1[2];
1988                                                         sub_v2_v2v2(edge_t0, tex_coords[indices[t*3+1]], tex_coords[indices[t*3+0]]);
1989                                                         sub_v2_v2v2(edge_t1, tex_coords[indices[t*3+2]], tex_coords[indices[t*3+0]]);
1990
1991                                                         f2x_area_uv = fabsf(edge_t0[0]*edge_t1[1] - edge_t0[1]*edge_t1[0]);
1992                                                         if ( f2x_area_uv>FLT_EPSILON )
1993                                                         {
1994                                                                 float norm[3], v0[3], v1[3], f2x_surf_area, fsurf_ratio;
1995                                                                 sub_v3_v3v3(v0, p1, p0);
1996                                                                 sub_v3_v3v3(v1, p2, p0);
1997                                                                 cross_v3_v3v3(norm, v0, v1);
1998
1999                                                                 f2x_surf_area = len_v3(norm);
2000                                                                 fsurf_ratio = f2x_surf_area/f2x_area_uv;        // tri area divided by texture area
2001
2002                                                                 ++nr_accumulated;
2003                                                                 dsum += (double)(fsurf_ratio);
2004                                                         }
2005                                                 }
2006                                         }
2007                                 }
2008                         }
2009                 }
2010
2011                 // finalize
2012                 {
2013                         const float avg_area_ratio = (nr_accumulated>0) ? ((float)(dsum / nr_accumulated)) : 1.0f;
2014                         const float use_as_render_bump_scale = sqrtf(avg_area_ratio);           // use width of average surface ratio as your bump scale
2015                         dm->auto_bump_scale = use_as_render_bump_scale;
2016                 }
2017         }
2018         else
2019         {
2020                 dm->auto_bump_scale = 1.0f;
2021         }
2022 }
2023
2024 void DM_vertex_attributes_from_gpu(DerivedMesh *dm, GPUVertexAttribs *gattribs, DMVertexAttribs *attribs)
2025 {
2026         CustomData *vdata, *fdata, *tfdata = NULL;
2027         int a, b, layer;
2028
2029         /* From the layers requested by the GLSL shader, figure out which ones are
2030          * actually available for this derivedmesh, and retrieve the pointers */
2031
2032         memset(attribs, 0, sizeof(DMVertexAttribs));
2033
2034         vdata = &dm->vertData;
2035         fdata = &dm->faceData;
2036
2037         /* ugly hack, editmesh derivedmesh doesn't copy face data, this way we
2038          * can use offsets instead */
2039         if(dm->type == DM_TYPE_EDITMESH)
2040                 tfdata = &((EditMeshDerivedMesh*)dm)->em->fdata;
2041         else
2042                 tfdata = fdata;
2043
2044         /* calc auto bump scale if necessary */
2045         if(dm->auto_bump_scale<=0.0f)
2046                 DM_calc_auto_bump_scale(dm);
2047
2048         /* add a tangent layer if necessary */
2049         for(b = 0; b < gattribs->totlayer; b++)
2050                 if(gattribs->layer[b].type == CD_TANGENT)
2051                         if(CustomData_get_layer_index(fdata, CD_TANGENT) == -1)
2052                                 DM_add_tangent_layer(dm);
2053
2054         for(b = 0; b < gattribs->totlayer; b++) {
2055                 if(gattribs->layer[b].type == CD_MTFACE) {
2056                         /* uv coordinates */
2057                         if(gattribs->layer[b].name[0])
2058                                 layer = CustomData_get_named_layer_index(tfdata, CD_MTFACE,
2059                                         gattribs->layer[b].name);
2060                         else
2061                                 layer = CustomData_get_active_layer_index(tfdata, CD_MTFACE);
2062
2063                         if(layer != -1) {
2064                                 a = attribs->tottface++;
2065
2066                                 attribs->tface[a].array = tfdata->layers[layer].data;
2067                                 attribs->tface[a].emOffset = tfdata->layers[layer].offset;
2068                                 attribs->tface[a].glIndex = gattribs->layer[b].glindex;
2069                                 attribs->tface[a].glTexco = gattribs->layer[b].gltexco;
2070                         }
2071                 }
2072                 else if(gattribs->layer[b].type == CD_MCOL) {
2073                         /* vertex colors */
2074                         if(gattribs->layer[b].name[0])
2075                                 layer = CustomData_get_named_layer_index(tfdata, CD_MCOL,
2076                                         gattribs->layer[b].name);
2077                         else
2078                                 layer = CustomData_get_active_layer_index(tfdata, CD_MCOL);
2079
2080                         if(layer != -1) {
2081                                 a = attribs->totmcol++;
2082
2083                                 attribs->mcol[a].array = tfdata->layers[layer].data;
2084                                 attribs->mcol[a].emOffset = tfdata->layers[layer].offset;
2085                                 attribs->mcol[a].glIndex = gattribs->layer[b].glindex;
2086                         }
2087                 }
2088                 else if(gattribs->layer[b].type == CD_TANGENT) {
2089                         /* tangents */
2090                         layer = CustomData_get_layer_index(fdata, CD_TANGENT);
2091
2092                         if(layer != -1) {
2093                                 attribs->tottang = 1;
2094
2095                                 attribs->tang.array = fdata->layers[layer].data;
2096                                 attribs->tang.emOffset = fdata->layers[layer].offset;
2097                                 attribs->tang.glIndex = gattribs->layer[b].glindex;
2098                         }
2099                 }
2100                 else if(gattribs->layer[b].type == CD_ORCO) {
2101                         /* original coordinates */
2102                         layer = CustomData_get_layer_index(vdata, CD_ORCO);
2103
2104                         if(layer != -1) {
2105                                 attribs->totorco = 1;
2106
2107                                 attribs->orco.array = vdata->layers[layer].data;
2108                                 attribs->orco.emOffset = vdata->layers[layer].offset;
2109                                 attribs->orco.glIndex = gattribs->layer[b].glindex;
2110                                 attribs->orco.glTexco = gattribs->layer[b].gltexco;
2111                         }
2112                 }
2113         }
2114 }
2115
2116 /* Set object's bounding box based on DerivedMesh min/max data */
2117 void DM_set_object_boundbox(Object *ob, DerivedMesh *dm)
2118 {
2119         float min[3], max[3];
2120
2121         INIT_MINMAX(min, max);
2122
2123         dm->getMinMax(dm, min, max);
2124
2125         if(!ob->bb)
2126                 ob->bb= MEM_callocN(sizeof(BoundBox), "DM-BoundBox");
2127
2128         boundbox_set_from_min_max(ob->bb, min, max);
2129 }
2130
2131 /* --- NAVMESH (begin) --- */
2132 #ifdef WITH_GAMEENGINE
2133
2134 BM_INLINE int navmesh_bit(int a, int b)
2135 {
2136         return (a & (1 << b)) >> b;
2137 }
2138
2139 BM_INLINE void navmesh_intToCol(int i, float col[3])
2140 {
2141         int     r = navmesh_bit(i, 0) + navmesh_bit(i, 3) * 2 + 1;
2142         int     g = navmesh_bit(i, 1) + navmesh_bit(i, 4) * 2 + 1;
2143         int     b = navmesh_bit(i, 2) + navmesh_bit(i, 5) * 2 + 1;
2144         col[0] = 1 - r*63.0f/255.0f;
2145         col[1] = 1 - g*63.0f/255.0f;
2146         col[2] = 1 - b*63.0f/255.0f;
2147 }
2148
2149 static void navmesh_drawColored(DerivedMesh *dm)
2150 {
2151         int a, glmode;
2152         MVert *mvert = (MVert *)CustomData_get_layer(&dm->vertData, CD_MVERT);
2153         MFace *mface = (MFace *)CustomData_get_layer(&dm->faceData, CD_MFACE);
2154         int *polygonIdx = (int *)CustomData_get_layer(&dm->faceData, CD_RECAST);
2155         float col[3];
2156
2157         if (!polygonIdx)
2158                 return;
2159
2160         /*
2161         //UI_ThemeColor(TH_WIRE);
2162         glDisable(GL_LIGHTING);
2163         glLineWidth(2.0);
2164         dm->drawEdges(dm, 0, 1);
2165         glLineWidth(1.0);
2166         glEnable(GL_LIGHTING);*/
2167
2168         glDisable(GL_LIGHTING);
2169         /*  if(GPU_buffer_legacy(dm) ) */ { /* TODO - VBO draw code, not high priority - campbell */
2170                 DEBUG_VBO( "Using legacy code. drawNavMeshColored\n" );
2171                 //glShadeModel(GL_SMOOTH);
2172                 glBegin(glmode = GL_QUADS);
2173                 for(a = 0; a < dm->numFaceData; a++, mface++) {
2174                         int new_glmode = mface->v4?GL_QUADS:GL_TRIANGLES;
2175                         int pi = polygonIdx[a];
2176                         if (pi <= 0) {
2177                                 zero_v3(col);
2178                         }
2179                         else {
2180                                 navmesh_intToCol(pi, col);
2181                         }
2182
2183                         if(new_glmode != glmode) {
2184                                 glEnd();
2185                                 glBegin(glmode = new_glmode);
2186                         }
2187                         glColor3fv(col);
2188                         glVertex3fv(mvert[mface->v1].co);
2189                         glVertex3fv(mvert[mface->v2].co);
2190                         glVertex3fv(mvert[mface->v3].co);
2191                         if(mface->v4) {
2192                                 glVertex3fv(mvert[mface->v4].co);
2193                         }
2194                 }
2195                 glEnd();
2196         }
2197         glEnable(GL_LIGHTING);
2198 }
2199
2200 static void navmesh_DM_drawFacesTex(DerivedMesh *dm,
2201                         int (*setDrawOptions)(MTFace *tface, int has_mcol, int matnr),
2202                         int (*compareDrawOptions)(void *userData, int cur_index, int next_index),
2203                         void *userData)
2204 {
2205         (void) setDrawOptions;
2206         (void) compareDrawOptions;
2207         (void) userData;
2208
2209         navmesh_drawColored(dm);
2210 }
2211
2212 static void navmesh_DM_drawFacesSolid(DerivedMesh *dm,
2213                                       float (*partial_redraw_planes)[4],
2214                                       int UNUSED(fast), int (*setMaterial)(int, void *attribs))
2215 {
2216         (void) partial_redraw_planes;
2217         (void) setMaterial;
2218
2219         //drawFacesSolid_original(dm, partial_redraw_planes, fast, setMaterial);
2220         navmesh_drawColored(dm);
2221 }
2222
2223 static DerivedMesh *navmesh_dm_createNavMeshForVisualization(DerivedMesh *dm)
2224 {
2225         DerivedMesh *result;
2226         int maxFaces = dm->getNumFaces(dm);
2227         int *recastData;
2228         int vertsPerPoly=0, nverts=0, ndtris=0, npolys=0;
2229         float* verts=NULL;
2230         unsigned short *dtris=NULL, *dmeshes=NULL, *polys=NULL;
2231         int *dtrisToPolysMap=NULL, *dtrisToTrisMap=NULL, *trisToFacesMap=NULL;
2232         int res;
2233
2234         result = CDDM_copy(dm);
2235         if (!CustomData_has_layer(&result->faceData, CD_RECAST)) {
2236                 int *sourceRecastData = (int*)CustomData_get_layer(&dm->faceData, CD_RECAST);
2237                 if (sourceRecastData) {
2238                         CustomData_add_layer_named(&result->faceData, CD_RECAST, CD_DUPLICATE,
2239                                                    sourceRecastData, maxFaces, "recastData");
2240                 }
2241         }
2242         recastData = (int*)CustomData_get_layer(&result->faceData, CD_RECAST);
2243
2244         /* note: This is not good design! - really should not be doing this */
2245         result->drawFacesTex =  navmesh_DM_drawFacesTex;
2246         result->drawFacesSolid = navmesh_DM_drawFacesSolid;
2247
2248
2249         /* process mesh */
2250         res  = buildNavMeshDataByDerivedMesh(dm, &vertsPerPoly, &nverts, &verts, &ndtris, &dtris,
2251                                              &npolys, &dmeshes, &polys, &dtrisToPolysMap, &dtrisToTrisMap,
2252                                              &trisToFacesMap);
2253         if (res) {
2254                 size_t polyIdx;
2255
2256                 /* invalidate concave polygon */
2257                 for (polyIdx=0; polyIdx<(size_t)npolys; polyIdx++) {
2258                         unsigned short* poly = &polys[polyIdx*2*vertsPerPoly];
2259                         if (!polyIsConvex(poly, vertsPerPoly, verts)) {
2260                                 /* set negative polygon idx to all faces */
2261                                 unsigned short *dmesh = &dmeshes[4*polyIdx];
2262                                 unsigned short tbase = dmesh[2];
2263                                 unsigned short tnum = dmesh[3];
2264                                 unsigned short ti;
2265
2266                                 for (ti=0; ti<tnum; ti++) {
2267                                         unsigned short triidx = dtrisToTrisMap[tbase+ti];
2268                                         unsigned short faceidx = trisToFacesMap[triidx];
2269                                         if (recastData[faceidx] > 0) {
2270                                                 recastData[faceidx] = -recastData[faceidx];
2271                                         }
2272                                 }
2273                         }
2274                 }
2275         }
2276         else {
2277                 printf("Error during creation polygon infos\n");
2278         }
2279
2280         /* clean up */
2281         if (verts!=NULL)
2282                 MEM_freeN(verts);
2283         if (dtris!=NULL)
2284                 MEM_freeN(dtris);
2285         if (dmeshes!=NULL)
2286                 MEM_freeN(dmeshes);
2287         if (polys!=NULL)
2288                 MEM_freeN(polys);
2289         if (dtrisToPolysMap!=NULL)
2290                 MEM_freeN(dtrisToPolysMap);
2291         if (dtrisToTrisMap!=NULL)
2292                 MEM_freeN(dtrisToTrisMap);
2293         if (trisToFacesMap!=NULL)
2294                 MEM_freeN(trisToFacesMap);
2295
2296         return result;
2297 }
2298
2299 #endif /* WITH_GAMEENGINE */
2300
2301 /* --- NAVMESH (end) --- */