Fix #34040: Moving Normal Node with enabled Cycles Material Preview crashes
[blender.git] / source / blender / blenkernel / intern / mesh.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) 2001-2002 by NaN Holding BV.
19  * All rights reserved.
20  *
21  * Contributor(s): Blender Foundation
22  *
23  * ***** END GPL LICENSE BLOCK *****
24  */
25
26 /** \file blender/blenkernel/intern/mesh.c
27  *  \ingroup bke
28  */
29
30
31 #include <stdlib.h>
32 #include <string.h>
33 #include <stdio.h>
34 #include <math.h>
35
36 #include "MEM_guardedalloc.h"
37
38 #include "DNA_scene_types.h"
39 #include "DNA_material_types.h"
40 #include "DNA_object_types.h"
41 #include "DNA_key_types.h"
42 #include "DNA_mesh_types.h"
43 #include "DNA_meshdata_types.h"
44 #include "DNA_ipo_types.h"
45 #include "DNA_customdata_types.h"
46
47 #include "BLI_utildefines.h"
48 #include "BLI_blenlib.h"
49 #include "BLI_math.h"
50 #include "BLI_edgehash.h"
51 #include "BLI_scanfill.h"
52 #include "BLI_array.h"
53
54 #include "BKE_animsys.h"
55 #include "BKE_main.h"
56 #include "BKE_customdata.h"
57 #include "BKE_DerivedMesh.h"
58 #include "BKE_global.h"
59 #include "BKE_mesh.h"
60 #include "BKE_displist.h"
61 #include "BKE_library.h"
62 #include "BKE_material.h"
63 #include "BKE_modifier.h"
64 #include "BKE_multires.h"
65 #include "BKE_key.h"
66 /* these 2 are only used by conversion functions */
67 #include "BKE_curve.h"
68 /* -- */
69 #include "BKE_object.h"
70 #include "BKE_tessmesh.h"
71 #include "BLI_edgehash.h"
72
73 #include "bmesh.h"
74
75 enum {
76         MESHCMP_DVERT_WEIGHTMISMATCH = 1,
77         MESHCMP_DVERT_GROUPMISMATCH,
78         MESHCMP_DVERT_TOTGROUPMISMATCH,
79         MESHCMP_LOOPCOLMISMATCH,
80         MESHCMP_LOOPUVMISMATCH,
81         MESHCMP_LOOPMISMATCH,
82         MESHCMP_POLYVERTMISMATCH,
83         MESHCMP_POLYMISMATCH,
84         MESHCMP_EDGEUNKNOWN,
85         MESHCMP_VERTCOMISMATCH,
86         MESHCMP_CDLAYERS_MISMATCH
87 };
88
89 static const char *cmpcode_to_str(int code)
90 {
91         switch (code) {
92                 case MESHCMP_DVERT_WEIGHTMISMATCH:
93                         return "Vertex Weight Mismatch";
94                 case MESHCMP_DVERT_GROUPMISMATCH:
95                         return "Vertex Group Mismatch";
96                 case MESHCMP_DVERT_TOTGROUPMISMATCH:
97                         return "Vertex Doesn't Belong To Same Number Of Groups";
98                 case MESHCMP_LOOPCOLMISMATCH:
99                         return "Vertex Color Mismatch";
100                 case MESHCMP_LOOPUVMISMATCH:
101                         return "UV Mismatch";
102                 case MESHCMP_LOOPMISMATCH:
103                         return "Loop Mismatch";
104                 case MESHCMP_POLYVERTMISMATCH:
105                         return "Loop Vert Mismatch In Poly Test";
106                 case MESHCMP_POLYMISMATCH:
107                         return "Loop Vert Mismatch";
108                 case MESHCMP_EDGEUNKNOWN:
109                         return "Edge Mismatch";
110                 case MESHCMP_VERTCOMISMATCH:
111                         return "Vertex Coordinate Mismatch";
112                 case MESHCMP_CDLAYERS_MISMATCH:
113                         return "CustomData Layer Count Mismatch";
114                 default:
115                         return "Mesh Comparison Code Unknown";
116         }
117 }
118
119 /* thresh is threshold for comparing vertices, uvs, vertex colors,
120  * weights, etc.*/
121 static int customdata_compare(CustomData *c1, CustomData *c2, Mesh *m1, Mesh *m2, float thresh)
122 {
123         CustomDataLayer *l1, *l2;
124         int i, i1 = 0, i2 = 0, tot, j;
125         
126         for (i = 0; i < c1->totlayer; i++) {
127                 if (ELEM7(c1->layers[i].type, CD_MVERT, CD_MEDGE, CD_MPOLY,
128                           CD_MLOOPUV, CD_MLOOPCOL, CD_MTEXPOLY, CD_MDEFORMVERT))
129                 {
130                         i1++;
131                 }
132         }
133
134         for (i = 0; i < c2->totlayer; i++) {
135                 if (ELEM7(c2->layers[i].type, CD_MVERT, CD_MEDGE, CD_MPOLY,
136                           CD_MLOOPUV, CD_MLOOPCOL, CD_MTEXPOLY, CD_MDEFORMVERT))
137                 {
138                         i2++;
139                 }
140         }
141
142         if (i1 != i2)
143                 return MESHCMP_CDLAYERS_MISMATCH;
144         
145         l1 = c1->layers; l2 = c2->layers;
146         tot = i1;
147         i1 = 0; i2 = 0; 
148         for (i = 0; i < tot; i++) {
149                 while (i1 < c1->totlayer && !ELEM7(l1->type, CD_MVERT, CD_MEDGE, CD_MPOLY, 
150                                                    CD_MLOOPUV, CD_MLOOPCOL, CD_MTEXPOLY, CD_MDEFORMVERT))
151                 {
152                         i1++, l1++;
153                 }
154
155                 while (i2 < c2->totlayer && !ELEM7(l2->type, CD_MVERT, CD_MEDGE, CD_MPOLY,
156                                                    CD_MLOOPUV, CD_MLOOPCOL, CD_MTEXPOLY, CD_MDEFORMVERT))
157                 {
158                         i2++, l2++;
159                 }
160                 
161                 if (l1->type == CD_MVERT) {
162                         MVert *v1 = l1->data;
163                         MVert *v2 = l2->data;
164                         int vtot = m1->totvert;
165                         
166                         for (j = 0; j < vtot; j++, v1++, v2++) {
167                                 if (len_v3v3(v1->co, v2->co) > thresh)
168                                         return MESHCMP_VERTCOMISMATCH;
169                                 /* I don't care about normals, let's just do coodinates */
170                         }
171                 }
172                 
173                 /*we're order-agnostic for edges here*/
174                 if (l1->type == CD_MEDGE) {
175                         MEdge *e1 = l1->data;
176                         MEdge *e2 = l2->data;
177                         EdgeHash *eh = BLI_edgehash_new();
178                         int etot = m1->totedge;
179                 
180                         for (j = 0; j < etot; j++, e1++) {
181                                 BLI_edgehash_insert(eh, e1->v1, e1->v2, e1);
182                         }
183                         
184                         for (j = 0; j < etot; j++, e2++) {
185                                 if (!BLI_edgehash_lookup(eh, e2->v1, e2->v2))
186                                         return MESHCMP_EDGEUNKNOWN;
187                         }
188                         BLI_edgehash_free(eh, NULL);
189                 }
190                 
191                 if (l1->type == CD_MPOLY) {
192                         MPoly *p1 = l1->data;
193                         MPoly *p2 = l2->data;
194                         int ptot = m1->totpoly;
195                 
196                         for (j = 0; j < ptot; j++, p1++, p2++) {
197                                 MLoop *lp1, *lp2;
198                                 int k;
199                                 
200                                 if (p1->totloop != p2->totloop)
201                                         return MESHCMP_POLYMISMATCH;
202                                 
203                                 lp1 = m1->mloop + p1->loopstart;
204                                 lp2 = m2->mloop + p2->loopstart;
205                                 
206                                 for (k = 0; k < p1->totloop; k++, lp1++, lp2++) {
207                                         if (lp1->v != lp2->v)
208                                                 return MESHCMP_POLYVERTMISMATCH;
209                                 }
210                         }
211                 }
212                 if (l1->type == CD_MLOOP) {
213                         MLoop *lp1 = l1->data;
214                         MLoop *lp2 = l2->data;
215                         int ltot = m1->totloop;
216                 
217                         for (j = 0; j < ltot; j++, lp1++, lp2++) {
218                                 if (lp1->v != lp2->v)
219                                         return MESHCMP_LOOPMISMATCH;
220                         }
221                 }
222                 if (l1->type == CD_MLOOPUV) {
223                         MLoopUV *lp1 = l1->data;
224                         MLoopUV *lp2 = l2->data;
225                         int ltot = m1->totloop;
226                 
227                         for (j = 0; j < ltot; j++, lp1++, lp2++) {
228                                 if (len_v2v2(lp1->uv, lp2->uv) > thresh)
229                                         return MESHCMP_LOOPUVMISMATCH;
230                         }
231                 }
232                 
233                 if (l1->type == CD_MLOOPCOL) {
234                         MLoopCol *lp1 = l1->data;
235                         MLoopCol *lp2 = l2->data;
236                         int ltot = m1->totloop;
237                 
238                         for (j = 0; j < ltot; j++, lp1++, lp2++) {
239                                 if (ABS(lp1->r - lp2->r) > thresh || 
240                                     ABS(lp1->g - lp2->g) > thresh || 
241                                     ABS(lp1->b - lp2->b) > thresh || 
242                                     ABS(lp1->a - lp2->a) > thresh)
243                                 {
244                                         return MESHCMP_LOOPCOLMISMATCH;
245                                 }
246                         }
247                 }
248
249                 if (l1->type == CD_MDEFORMVERT) {
250                         MDeformVert *dv1 = l1->data;
251                         MDeformVert *dv2 = l2->data;
252                         int dvtot = m1->totvert;
253                 
254                         for (j = 0; j < dvtot; j++, dv1++, dv2++) {
255                                 int k;
256                                 MDeformWeight *dw1 = dv1->dw, *dw2 = dv2->dw;
257                                 
258                                 if (dv1->totweight != dv2->totweight)
259                                         return MESHCMP_DVERT_TOTGROUPMISMATCH;
260                                 
261                                 for (k = 0; k < dv1->totweight; k++, dw1++, dw2++) {
262                                         if (dw1->def_nr != dw2->def_nr)
263                                                 return MESHCMP_DVERT_GROUPMISMATCH;
264                                         if (ABS(dw1->weight - dw2->weight) > thresh)
265                                                 return MESHCMP_DVERT_WEIGHTMISMATCH;
266                                 }
267                         }
268                 }
269         }
270         
271         return 0;
272 }
273
274 /*used for testing.  returns an error string the two meshes don't match*/
275 const char *BKE_mesh_cmp(Mesh *me1, Mesh *me2, float thresh)
276 {
277         int c;
278         
279         if (!me1 || !me2)
280                 return "Requires two input meshes";
281         
282         if (me1->totvert != me2->totvert) 
283                 return "Number of verts don't match";
284         
285         if (me1->totedge != me2->totedge)
286                 return "Number of edges don't match";
287         
288         if (me1->totpoly != me2->totpoly)
289                 return "Number of faces don't match";
290                                 
291         if (me1->totloop != me2->totloop)
292                 return "Number of loops don't match";
293         
294         if ((c = customdata_compare(&me1->vdata, &me2->vdata, me1, me2, thresh)))
295                 return cmpcode_to_str(c);
296
297         if ((c = customdata_compare(&me1->edata, &me2->edata, me1, me2, thresh)))
298                 return cmpcode_to_str(c);
299
300         if ((c = customdata_compare(&me1->ldata, &me2->ldata, me1, me2, thresh)))
301                 return cmpcode_to_str(c);
302
303         if ((c = customdata_compare(&me1->pdata, &me2->pdata, me1, me2, thresh)))
304                 return cmpcode_to_str(c);
305         
306         return NULL;
307 }
308
309 static void mesh_ensure_tessellation_customdata(Mesh *me)
310 {
311         if (UNLIKELY((me->totface != 0) && (me->totpoly == 0))) {
312                 /* Pass, otherwise this function  clears 'mface' before
313                  * versioning 'mface -> mpoly' code kicks in [#30583]
314                  *
315                  * Callers could also check but safer to do here - campbell */
316         }
317         else {
318                 const int tottex_original = CustomData_number_of_layers(&me->pdata, CD_MTEXPOLY);
319                 const int totcol_original = CustomData_number_of_layers(&me->ldata, CD_MLOOPCOL);
320
321                 const int tottex_tessface = CustomData_number_of_layers(&me->fdata, CD_MTFACE);
322                 const int totcol_tessface = CustomData_number_of_layers(&me->fdata, CD_MCOL);
323
324                 if (tottex_tessface != tottex_original ||
325                     totcol_tessface != totcol_original)
326                 {
327                         BKE_mesh_tessface_clear(me);
328
329                         CustomData_from_bmeshpoly(&me->fdata, &me->pdata, &me->ldata, me->totface);
330
331                         /* TODO - add some --debug-mesh option */
332                         if (G.debug & G_DEBUG) {
333                                 /* note: this warning may be un-called for if we are initializing the mesh for the
334                                  * first time from bmesh, rather then giving a warning about this we could be smarter
335                                  * and check if there was any data to begin with, for now just print the warning with
336                                  * some info to help troubleshoot whats going on - campbell */
337                                 printf("%s: warning! Tessellation uvs or vcol data got out of sync, "
338                                        "had to reset!\n    CD_MTFACE: %d != CD_MTEXPOLY: %d || CD_MCOL: %d != CD_MLOOPCOL: %d\n",
339                                        __func__, tottex_tessface, tottex_original, totcol_tessface, totcol_original);
340                         }
341                 }
342         }
343 }
344
345 /* this ensures grouped customdata (e.g. mtexpoly and mloopuv and mtface, or
346  * mloopcol and mcol) have the same relative active/render/clone/mask indices.
347  *
348  * note that for undo mesh data we want to skip 'ensure_tess_cd' call since
349  * we don't want to store memory for tessface when its only used for older
350  * versions of the mesh. - campbell*/
351 static void mesh_update_linked_customdata(Mesh *me, const short do_ensure_tess_cd)
352 {
353         if (me->edit_btmesh)
354                 BMEdit_UpdateLinkedCustomData(me->edit_btmesh);
355
356         if (do_ensure_tess_cd) {
357                 mesh_ensure_tessellation_customdata(me);
358         }
359
360         CustomData_bmesh_update_active_layers(&me->fdata, &me->pdata, &me->ldata);
361 }
362
363 void mesh_update_customdata_pointers(Mesh *me, const short do_ensure_tess_cd)
364 {
365         mesh_update_linked_customdata(me, do_ensure_tess_cd);
366
367         me->mvert = CustomData_get_layer(&me->vdata, CD_MVERT);
368         me->dvert = CustomData_get_layer(&me->vdata, CD_MDEFORMVERT);
369
370         me->medge = CustomData_get_layer(&me->edata, CD_MEDGE);
371
372         me->mface = CustomData_get_layer(&me->fdata, CD_MFACE);
373         me->mcol = CustomData_get_layer(&me->fdata, CD_MCOL);
374         me->mtface = CustomData_get_layer(&me->fdata, CD_MTFACE);
375         
376         me->mpoly = CustomData_get_layer(&me->pdata, CD_MPOLY);
377         me->mloop = CustomData_get_layer(&me->ldata, CD_MLOOP);
378
379         me->mtpoly = CustomData_get_layer(&me->pdata, CD_MTEXPOLY);
380         me->mloopcol = CustomData_get_layer(&me->ldata, CD_MLOOPCOL);
381         me->mloopuv = CustomData_get_layer(&me->ldata, CD_MLOOPUV);
382 }
383
384 /* Note: unlinking is called when me->id.us is 0, question remains how
385  * much unlinking of Library data in Mesh should be done... probably
386  * we need a more generic method, like the expand() functions in
387  * readfile.c */
388
389 void BKE_mesh_unlink(Mesh *me)
390 {
391         int a;
392         
393         if (me == NULL) return;
394         
395         if (me->mat)
396         for (a = 0; a < me->totcol; a++) {
397                 if (me->mat[a]) me->mat[a]->id.us--;
398                 me->mat[a] = NULL;
399         }
400
401         if (me->key) {
402                 me->key->id.us--;
403         }
404         me->key = NULL;
405         
406         if (me->texcomesh) me->texcomesh = NULL;
407 }
408
409 /* do not free mesh itself */
410 void BKE_mesh_free(Mesh *me, int unlink)
411 {
412         if (unlink)
413                 BKE_mesh_unlink(me);
414
415         CustomData_free(&me->vdata, me->totvert);
416         CustomData_free(&me->edata, me->totedge);
417         CustomData_free(&me->fdata, me->totface);
418         CustomData_free(&me->ldata, me->totloop);
419         CustomData_free(&me->pdata, me->totpoly);
420
421         if (me->adt) {
422                 BKE_free_animdata(&me->id);
423                 me->adt = NULL;
424         }
425         
426         if (me->mat) MEM_freeN(me->mat);
427         
428         if (me->bb) MEM_freeN(me->bb);
429         if (me->mselect) MEM_freeN(me->mselect);
430         if (me->edit_btmesh) MEM_freeN(me->edit_btmesh);
431 }
432
433 static void mesh_tessface_clear_intern(Mesh *mesh, int free_customdata)
434 {
435         if (free_customdata) {
436                 CustomData_free(&mesh->fdata, mesh->totface);
437         }
438         else {
439                 CustomData_reset(&mesh->fdata);
440         }
441
442         mesh->mface = NULL;
443         mesh->mtface = NULL;
444         mesh->mcol = NULL;
445         mesh->totface = 0;
446 }
447
448 Mesh *BKE_mesh_add(Main *bmain, const char *name)
449 {
450         Mesh *me;
451         
452         me = BKE_libblock_alloc(&bmain->mesh, ID_ME, name);
453         
454         me->size[0] = me->size[1] = me->size[2] = 1.0;
455         me->smoothresh = 30;
456         me->texflag = ME_AUTOSPACE;
457         me->flag = ME_TWOSIDED;
458         me->drawflag = ME_DRAWEDGES | ME_DRAWFACES | ME_DRAWCREASES;
459
460         CustomData_reset(&me->vdata);
461         CustomData_reset(&me->edata);
462         CustomData_reset(&me->fdata);
463         CustomData_reset(&me->pdata);
464         CustomData_reset(&me->ldata);
465
466         return me;
467 }
468
469 Mesh *BKE_mesh_copy_ex(Main *bmain, Mesh *me)
470 {
471         Mesh *men;
472         MTFace *tface;
473         MTexPoly *txface;
474         int a, i;
475         const int do_tessface = ((me->totface != 0) && (me->totpoly == 0)); /* only do tessface if we have no polys */
476         
477         men = BKE_libblock_copy_ex(bmain, &me->id);
478         
479         men->mat = MEM_dupallocN(me->mat);
480         for (a = 0; a < men->totcol; a++) {
481                 id_us_plus((ID *)men->mat[a]);
482         }
483         id_us_plus((ID *)men->texcomesh);
484
485         CustomData_copy(&me->vdata, &men->vdata, CD_MASK_MESH, CD_DUPLICATE, men->totvert);
486         CustomData_copy(&me->edata, &men->edata, CD_MASK_MESH, CD_DUPLICATE, men->totedge);
487         CustomData_copy(&me->ldata, &men->ldata, CD_MASK_MESH, CD_DUPLICATE, men->totloop);
488         CustomData_copy(&me->pdata, &men->pdata, CD_MASK_MESH, CD_DUPLICATE, men->totpoly);
489         if (do_tessface) {
490                 CustomData_copy(&me->fdata, &men->fdata, CD_MASK_MESH, CD_DUPLICATE, men->totface);
491         }
492         else {
493                 mesh_tessface_clear_intern(men, FALSE);
494         }
495
496         mesh_update_customdata_pointers(men, do_tessface);
497
498         /* ensure indirect linked data becomes lib-extern */
499         for (i = 0; i < me->fdata.totlayer; i++) {
500                 if (me->fdata.layers[i].type == CD_MTFACE) {
501                         tface = (MTFace *)me->fdata.layers[i].data;
502
503                         for (a = 0; a < me->totface; a++, tface++)
504                                 if (tface->tpage)
505                                         id_lib_extern((ID *)tface->tpage);
506                 }
507         }
508         
509         for (i = 0; i < me->pdata.totlayer; i++) {
510                 if (me->pdata.layers[i].type == CD_MTEXPOLY) {
511                         txface = (MTexPoly *)me->pdata.layers[i].data;
512
513                         for (a = 0; a < me->totpoly; a++, txface++)
514                                 if (txface->tpage)
515                                         id_lib_extern((ID *)txface->tpage);
516                 }
517         }
518
519         men->mselect = NULL;
520         men->edit_btmesh = NULL;
521
522         men->bb = MEM_dupallocN(men->bb);
523         
524         men->key = BKE_key_copy(me->key);
525         if (men->key) men->key->from = (ID *)men;
526
527         return men;
528 }
529
530 Mesh *BKE_mesh_copy(Mesh *me)
531 {
532         return BKE_mesh_copy_ex(G.main, me);
533 }
534
535 BMesh *BKE_mesh_to_bmesh(Mesh *me, Object *ob)
536 {
537         BMesh *bm;
538
539         bm = BM_mesh_create(&bm_mesh_allocsize_default);
540
541         BM_mesh_bm_from_me(bm, me, true, ob->shapenr);
542
543         return bm;
544 }
545
546 static void expand_local_mesh(Mesh *me)
547 {
548         id_lib_extern((ID *)me->texcomesh);
549
550         if (me->mtface || me->mtpoly) {
551                 int a, i;
552
553                 for (i = 0; i < me->pdata.totlayer; i++) {
554                         if (me->pdata.layers[i].type == CD_MTEXPOLY) {
555                                 MTexPoly *txface = (MTexPoly *)me->pdata.layers[i].data;
556
557                                 for (a = 0; a < me->totpoly; a++, txface++) {
558                                         /* special case: ima always local immediately */
559                                         if (txface->tpage) {
560                                                 id_lib_extern((ID *)txface->tpage);
561                                         }
562                                 }
563                         }
564                 }
565
566                 for (i = 0; i < me->fdata.totlayer; i++) {
567                         if (me->fdata.layers[i].type == CD_MTFACE) {
568                                 MTFace *tface = (MTFace *)me->fdata.layers[i].data;
569
570                                 for (a = 0; a < me->totface; a++, tface++) {
571                                         /* special case: ima always local immediately */
572                                         if (tface->tpage) {
573                                                 id_lib_extern((ID *)tface->tpage);
574                                         }
575                                 }
576                         }
577                 }
578         }
579
580         if (me->mat) {
581                 extern_local_matarar(me->mat, me->totcol);
582         }
583 }
584
585 void BKE_mesh_make_local(Mesh *me)
586 {
587         Main *bmain = G.main;
588         Object *ob;
589         int is_local = FALSE, is_lib = FALSE;
590
591         /* - only lib users: do nothing
592          * - only local users: set flag
593          * - mixed: make copy
594          */
595
596         if (me->id.lib == NULL) return;
597         if (me->id.us == 1) {
598                 id_clear_lib_data(bmain, &me->id);
599                 expand_local_mesh(me);
600                 return;
601         }
602
603         for (ob = bmain->object.first; ob && ELEM(0, is_lib, is_local); ob = ob->id.next) {
604                 if (me == ob->data) {
605                         if (ob->id.lib) is_lib = TRUE;
606                         else is_local = TRUE;
607                 }
608         }
609
610         if (is_local && is_lib == FALSE) {
611                 id_clear_lib_data(bmain, &me->id);
612                 expand_local_mesh(me);
613         }
614         else if (is_local && is_lib) {
615                 Mesh *me_new = BKE_mesh_copy(me);
616                 me_new->id.us = 0;
617
618
619                 /* Remap paths of new ID using old library as base. */
620                 BKE_id_lib_local_paths(bmain, me->id.lib, &me_new->id);
621
622                 for (ob = bmain->object.first; ob; ob = ob->id.next) {
623                         if (me == ob->data) {
624                                 if (ob->id.lib == NULL) {
625                                         set_mesh(ob, me_new);
626                                 }
627                         }
628                 }
629         }
630 }
631
632 void BKE_mesh_boundbox_calc(Mesh *me, float r_loc[3], float r_size[3])
633 {
634         BoundBox *bb;
635         float min[3], max[3];
636         float mloc[3], msize[3];
637         
638         if (me->bb == NULL) me->bb = MEM_callocN(sizeof(BoundBox), "boundbox");
639         bb = me->bb;
640
641         if (!r_loc) r_loc = mloc;
642         if (!r_size) r_size = msize;
643         
644         INIT_MINMAX(min, max);
645         if (!BKE_mesh_minmax(me, min, max)) {
646                 min[0] = min[1] = min[2] = -1.0f;
647                 max[0] = max[1] = max[2] = 1.0f;
648         }
649
650         mid_v3_v3v3(r_loc, min, max);
651                 
652         r_size[0] = (max[0] - min[0]) / 2.0f;
653         r_size[1] = (max[1] - min[1]) / 2.0f;
654         r_size[2] = (max[2] - min[2]) / 2.0f;
655         
656         BKE_boundbox_init_from_minmax(bb, min, max);
657 }
658
659 void BKE_mesh_texspace_calc(Mesh *me)
660 {
661         float loc[3], size[3];
662         int a;
663
664         BKE_mesh_boundbox_calc(me, loc, size);
665
666         if (me->texflag & ME_AUTOSPACE) {
667                 for (a = 0; a < 3; a++) {
668                         if (size[a] == 0.0f) size[a] = 1.0f;
669                         else if (size[a] > 0.0f && size[a] < 0.00001f) size[a] = 0.00001f;
670                         else if (size[a] < 0.0f && size[a] > -0.00001f) size[a] = -0.00001f;
671                 }
672
673                 copy_v3_v3(me->loc, loc);
674                 copy_v3_v3(me->size, size);
675                 zero_v3(me->rot);
676         }
677 }
678
679 BoundBox *BKE_mesh_boundbox_get(Object *ob)
680 {
681         Mesh *me = ob->data;
682
683         if (ob->bb)
684                 return ob->bb;
685
686         if (!me->bb)
687                 BKE_mesh_texspace_calc(me);
688
689         return me->bb;
690 }
691
692 void BKE_mesh_texspace_get(Mesh *me, float r_loc[3], float r_rot[3], float r_size[3])
693 {
694         if (!me->bb) {
695                 BKE_mesh_texspace_calc(me);
696         }
697
698         if (r_loc) copy_v3_v3(r_loc,  me->loc);
699         if (r_rot) copy_v3_v3(r_rot,  me->rot);
700         if (r_size) copy_v3_v3(r_size, me->size);
701 }
702
703 float (*BKE_mesh_orco_verts_get(Object *ob))[3]
704 {
705         Mesh *me = ob->data;
706         MVert *mvert = NULL;
707         Mesh *tme = me->texcomesh ? me->texcomesh : me;
708         int a, totvert;
709         float (*vcos)[3] = NULL;
710
711         /* Get appropriate vertex coordinates */
712         vcos = MEM_callocN(sizeof(*vcos) * me->totvert, "orco mesh");
713         mvert = tme->mvert;
714         totvert = min_ii(tme->totvert, me->totvert);
715
716         for (a = 0; a < totvert; a++, mvert++) {
717                 copy_v3_v3(vcos[a], mvert->co);
718         }
719
720         return vcos;
721 }
722
723 void BKE_mesh_orco_verts_transform(Mesh *me, float (*orco)[3], int totvert, int invert)
724 {
725         float loc[3], size[3];
726         int a;
727
728         BKE_mesh_texspace_get(me->texcomesh ? me->texcomesh : me, loc, NULL, size);
729
730         if (invert) {
731                 for (a = 0; a < totvert; a++) {
732                         float *co = orco[a];
733                         madd_v3_v3v3v3(co, loc, co, size);
734                 }
735         }
736         else {
737                 for (a = 0; a < totvert; a++) {
738                         float *co = orco[a];
739                         co[0] = (co[0] - loc[0]) / size[0];
740                         co[1] = (co[1] - loc[1]) / size[1];
741                         co[2] = (co[2] - loc[2]) / size[2];
742                 }
743         }
744 }
745
746 /* rotates the vertices of a face in case v[2] or v[3] (vertex index) is = 0.
747  * this is necessary to make the if (mface->v4) check for quads work */
748 int test_index_face(MFace *mface, CustomData *fdata, int mfindex, int nr)
749 {
750         /* first test if the face is legal */
751         if ((mface->v3 || nr == 4) && mface->v3 == mface->v4) {
752                 mface->v4 = 0;
753                 nr--;
754         }
755         if ((mface->v2 || mface->v4) && mface->v2 == mface->v3) {
756                 mface->v3 = mface->v4;
757                 mface->v4 = 0;
758                 nr--;
759         }
760         if (mface->v1 == mface->v2) {
761                 mface->v2 = mface->v3;
762                 mface->v3 = mface->v4;
763                 mface->v4 = 0;
764                 nr--;
765         }
766
767         /* check corrupt cases, bow-tie geometry, cant handle these because edge data wont exist so just return 0 */
768         if (nr == 3) {
769                 if (
770                     /* real edges */
771                     mface->v1 == mface->v2 ||
772                     mface->v2 == mface->v3 ||
773                     mface->v3 == mface->v1)
774                 {
775                         return 0;
776                 }
777         }
778         else if (nr == 4) {
779                 if (
780                     /* real edges */
781                     mface->v1 == mface->v2 ||
782                     mface->v2 == mface->v3 ||
783                     mface->v3 == mface->v4 ||
784                     mface->v4 == mface->v1 ||
785                     /* across the face */
786                     mface->v1 == mface->v3 ||
787                     mface->v2 == mface->v4)
788                 {
789                         return 0;
790                 }
791         }
792
793         /* prevent a zero at wrong index location */
794         if (nr == 3) {
795                 if (mface->v3 == 0) {
796                         static int corner_indices[4] = {1, 2, 0, 3};
797
798                         SWAP(unsigned int, mface->v1, mface->v2);
799                         SWAP(unsigned int, mface->v2, mface->v3);
800
801                         if (fdata)
802                                 CustomData_swap(fdata, mfindex, corner_indices);
803                 }
804         }
805         else if (nr == 4) {
806                 if (mface->v3 == 0 || mface->v4 == 0) {
807                         static int corner_indices[4] = {2, 3, 0, 1};
808
809                         SWAP(unsigned int, mface->v1, mface->v3);
810                         SWAP(unsigned int, mface->v2, mface->v4);
811
812                         if (fdata)
813                                 CustomData_swap(fdata, mfindex, corner_indices);
814                 }
815         }
816
817         return nr;
818 }
819
820 Mesh *BKE_mesh_from_object(Object *ob)
821 {
822         
823         if (ob == NULL) return NULL;
824         if (ob->type == OB_MESH) return ob->data;
825         else return NULL;
826 }
827
828 void set_mesh(Object *ob, Mesh *me)
829 {
830         Mesh *old = NULL;
831
832         multires_force_update(ob);
833         
834         if (ob == NULL) return;
835         
836         if (ob->type == OB_MESH) {
837                 old = ob->data;
838                 if (old)
839                         old->id.us--;
840                 ob->data = me;
841                 id_us_plus((ID *)me);
842         }
843         
844         test_object_materials((ID *)me);
845
846         test_object_modifiers(ob);
847 }
848
849 /* ************** make edges in a Mesh, for outside of editmode */
850
851 struct edgesort {
852         unsigned int v1, v2;
853         short is_loose, is_draw;
854 };
855
856 /* edges have to be added with lowest index first for sorting */
857 static void to_edgesort(struct edgesort *ed,
858                         unsigned int v1, unsigned int v2,
859                         short is_loose, short is_draw)
860 {
861         if (v1 < v2) {
862                 ed->v1 = v1; ed->v2 = v2;
863         }
864         else {
865                 ed->v1 = v2; ed->v2 = v1;
866         }
867         ed->is_loose = is_loose;
868         ed->is_draw = is_draw;
869 }
870
871 static int vergedgesort(const void *v1, const void *v2)
872 {
873         const struct edgesort *x1 = v1, *x2 = v2;
874
875         if (x1->v1 > x2->v1) return 1;
876         else if (x1->v1 < x2->v1) return -1;
877         else if (x1->v2 > x2->v2) return 1;
878         else if (x1->v2 < x2->v2) return -1;
879         
880         return 0;
881 }
882
883
884 /* Create edges based on known verts and faces */
885 static void make_edges_mdata(MVert *UNUSED(allvert), MFace *allface, MLoop *allloop,
886                              MPoly *allpoly, int UNUSED(totvert), int totface, int UNUSED(totloop), int totpoly,
887                              int old, MEdge **alledge, int *_totedge)
888 {
889         MPoly *mpoly;
890         MLoop *mloop;
891         MFace *mface;
892         MEdge *medge;
893         EdgeHash *hash = BLI_edgehash_new();
894         struct edgesort *edsort, *ed;
895         int a, b, totedge = 0, final = 0;
896
897         /* we put all edges in array, sort them, and detect doubles that way */
898
899         for (a = totface, mface = allface; a > 0; a--, mface++) {
900                 if (mface->v4) totedge += 4;
901                 else if (mface->v3) totedge += 3;
902                 else totedge += 1;
903         }
904
905         if (totedge == 0) {
906                 /* flag that mesh has edges */
907                 (*alledge) = MEM_callocN(0, "make mesh edges");
908                 (*_totedge) = 0;
909                 return;
910         }
911
912         ed = edsort = MEM_mallocN(totedge * sizeof(struct edgesort), "edgesort");
913
914         for (a = totface, mface = allface; a > 0; a--, mface++) {
915                 to_edgesort(ed++, mface->v1, mface->v2, !mface->v3, mface->edcode & ME_V1V2);
916                 if (mface->v4) {
917                         to_edgesort(ed++, mface->v2, mface->v3, 0, mface->edcode & ME_V2V3);
918                         to_edgesort(ed++, mface->v3, mface->v4, 0, mface->edcode & ME_V3V4);
919                         to_edgesort(ed++, mface->v4, mface->v1, 0, mface->edcode & ME_V4V1);
920                 }
921                 else if (mface->v3) {
922                         to_edgesort(ed++, mface->v2, mface->v3, 0, mface->edcode & ME_V2V3);
923                         to_edgesort(ed++, mface->v3, mface->v1, 0, mface->edcode & ME_V3V1);
924                 }
925         }
926
927         qsort(edsort, totedge, sizeof(struct edgesort), vergedgesort);
928
929         /* count final amount */
930         for (a = totedge, ed = edsort; a > 1; a--, ed++) {
931                 /* edge is unique when it differs from next edge, or is last */
932                 if (ed->v1 != (ed + 1)->v1 || ed->v2 != (ed + 1)->v2) final++;
933         }
934         final++;
935
936         (*alledge) = medge = MEM_callocN(sizeof(MEdge) * final, "BKE_mesh_make_edges mdge");
937         (*_totedge) = final;
938
939         for (a = totedge, ed = edsort; a > 1; a--, ed++) {
940                 /* edge is unique when it differs from next edge, or is last */
941                 if (ed->v1 != (ed + 1)->v1 || ed->v2 != (ed + 1)->v2) {
942                         medge->v1 = ed->v1;
943                         medge->v2 = ed->v2;
944                         if (old == 0 || ed->is_draw) medge->flag = ME_EDGEDRAW | ME_EDGERENDER;
945                         if (ed->is_loose) medge->flag |= ME_LOOSEEDGE;
946
947                         /* order is swapped so extruding this edge as a surface wont flip face normals
948                          * with cyclic curves */
949                         if (ed->v1 + 1 != ed->v2) {
950                                 SWAP(unsigned int, medge->v1, medge->v2);
951                         }
952                         medge++;
953                 }
954                 else {
955                         /* equal edge, we merge the drawflag */
956                         (ed + 1)->is_draw |= ed->is_draw;
957                 }
958         }
959         /* last edge */
960         medge->v1 = ed->v1;
961         medge->v2 = ed->v2;
962         medge->flag = ME_EDGEDRAW;
963         if (ed->is_loose) medge->flag |= ME_LOOSEEDGE;
964         medge->flag |= ME_EDGERENDER;
965
966         MEM_freeN(edsort);
967         
968         /* set edge members of mloops */
969         medge = *alledge;
970         for (a = 0; a < *_totedge; a++, medge++) {
971                 BLI_edgehash_insert(hash, medge->v1, medge->v2, SET_INT_IN_POINTER(a));
972         }
973         
974         mpoly = allpoly;
975         for (a = 0; a < totpoly; a++, mpoly++) {
976                 mloop = allloop + mpoly->loopstart;
977                 for (b = 0; b < mpoly->totloop; b++) {
978                         int v1, v2;
979                         
980                         v1 = mloop[b].v;
981                         v2 = ME_POLY_LOOP_NEXT(mloop, mpoly, b)->v;
982                         mloop[b].e = GET_INT_FROM_POINTER(BLI_edgehash_lookup(hash, v1, v2));
983                 }
984         }
985         
986         BLI_edgehash_free(hash, NULL);
987 }
988
989 void BKE_mesh_make_edges(Mesh *me, int old)
990 {
991         MEdge *medge;
992         int totedge = 0;
993
994         make_edges_mdata(me->mvert, me->mface, me->mloop, me->mpoly, me->totvert, me->totface, me->totloop, me->totpoly, old, &medge, &totedge);
995         if (totedge == 0) {
996                 /* flag that mesh has edges */
997                 me->medge = medge;
998                 me->totedge = 0;
999                 return;
1000         }
1001
1002         medge = CustomData_add_layer(&me->edata, CD_MEDGE, CD_ASSIGN, medge, totedge);
1003         me->medge = medge;
1004         me->totedge = totedge;
1005
1006         BKE_mesh_strip_loose_faces(me);
1007 }
1008
1009 /* We need to keep this for edge creation (for now?), and some old readfile code... */
1010 void BKE_mesh_strip_loose_faces(Mesh *me)
1011 {
1012         MFace *f;
1013         int a, b;
1014
1015         for (a = b = 0, f = me->mface; a < me->totface; a++, f++) {
1016                 if (f->v3) {
1017                         if (a != b) {
1018                                 memcpy(&me->mface[b], f, sizeof(me->mface[b]));
1019                                 CustomData_copy_data(&me->fdata, &me->fdata, a, b, 1);
1020                         }
1021                         b++;
1022                 }
1023         }
1024         if (a != b) {
1025                 CustomData_free_elem(&me->fdata, b, a - b);
1026                 me->totface = b;
1027         }
1028 }
1029
1030 /* Works on both loops and polys! */
1031 /* Note: It won't try to guess which loops of an invalid poly to remove!
1032  *       this is the work of the caller, to mark those loops...
1033  *       See e.g. BKE_mesh_validate_arrays(). */
1034 void BKE_mesh_strip_loose_polysloops(Mesh *me)
1035 {
1036         MPoly *p;
1037         MLoop *l;
1038         int a, b;
1039         /* New loops idx! */
1040         int *new_idx = MEM_mallocN(sizeof(int) * me->totloop, __func__);
1041
1042         for (a = b = 0, p = me->mpoly; a < me->totpoly; a++, p++) {
1043                 int invalid = FALSE;
1044                 int i = p->loopstart;
1045                 int stop = i + p->totloop;
1046
1047                 if (stop > me->totloop || stop < i) {
1048                         invalid = TRUE;
1049                 }
1050                 else {
1051                         l = &me->mloop[i];
1052                         i = stop - i;
1053                         /* If one of the poly's loops is invalid, the whole poly is invalid! */
1054                         for (; i--; l++) {
1055                                 if (l->e == INVALID_LOOP_EDGE_MARKER) {
1056                                         invalid = TRUE;
1057                                         break;
1058                                 }
1059                         }
1060                 }
1061
1062                 if (p->totloop >= 3 && !invalid) {
1063                         if (a != b) {
1064                                 memcpy(&me->mpoly[b], p, sizeof(me->mpoly[b]));
1065                                 CustomData_copy_data(&me->pdata, &me->pdata, a, b, 1);
1066                         }
1067                         b++;
1068                 }
1069         }
1070         if (a != b) {
1071                 CustomData_free_elem(&me->pdata, b, a - b);
1072                 me->totpoly = b;
1073         }
1074
1075         /* And now, get rid of invalid loops. */
1076         for (a = b = 0, l = me->mloop; a < me->totloop; a++, l++) {
1077                 if (l->e != INVALID_LOOP_EDGE_MARKER) {
1078                         if (a != b) {
1079                                 memcpy(&me->mloop[b], l, sizeof(me->mloop[b]));
1080                                 CustomData_copy_data(&me->ldata, &me->ldata, a, b, 1);
1081                         }
1082                         new_idx[a] = b;
1083                         b++;
1084                 }
1085                 else {
1086                         /* XXX Theoretically, we should be able to not do this, as no remaining poly
1087                          *     should use any stripped loop. But for security's sake... */
1088                         new_idx[a] = -a;
1089                 }
1090         }
1091         if (a != b) {
1092                 CustomData_free_elem(&me->ldata, b, a - b);
1093                 me->totloop = b;
1094         }
1095
1096         /* And now, update polys' start loop index. */
1097         /* Note: At this point, there should never be any poly using a striped loop! */
1098         for (a = 0, p = me->mpoly; a < me->totpoly; a++, p++) {
1099                 p->loopstart = new_idx[p->loopstart];
1100         }
1101
1102         MEM_freeN(new_idx);
1103 }
1104
1105 void BKE_mesh_strip_loose_edges(Mesh *me)
1106 {
1107         MEdge *e;
1108         MLoop *l;
1109         int a, b;
1110         unsigned int *new_idx = MEM_mallocN(sizeof(int) * me->totedge, __func__);
1111
1112         for (a = b = 0, e = me->medge; a < me->totedge; a++, e++) {
1113                 if (e->v1 != e->v2) {
1114                         if (a != b) {
1115                                 memcpy(&me->medge[b], e, sizeof(me->medge[b]));
1116                                 CustomData_copy_data(&me->edata, &me->edata, a, b, 1);
1117                         }
1118                         new_idx[a] = b;
1119                         b++;
1120                 }
1121                 else {
1122                         new_idx[a] = INVALID_LOOP_EDGE_MARKER;
1123                 }
1124         }
1125         if (a != b) {
1126                 CustomData_free_elem(&me->edata, b, a - b);
1127                 me->totedge = b;
1128         }
1129
1130         /* And now, update loops' edge indices. */
1131         /* XXX We hope no loop was pointing to a striped edge!
1132          *     Else, its e will be set to INVALID_LOOP_EDGE_MARKER :/ */
1133         for (a = 0, l = me->mloop; a < me->totloop; a++, l++) {
1134                 l->e = new_idx[l->e];
1135         }
1136
1137         MEM_freeN(new_idx);
1138 }
1139
1140 void BKE_mesh_from_metaball(ListBase *lb, Mesh *me)
1141 {
1142         DispList *dl;
1143         MVert *mvert;
1144         MLoop *mloop, *allloop;
1145         MPoly *mpoly;
1146         float *nors, *verts;
1147         int a, *index;
1148         
1149         dl = lb->first;
1150         if (dl == NULL) return;
1151
1152         if (dl->type == DL_INDEX4) {
1153                 mvert = CustomData_add_layer(&me->vdata, CD_MVERT, CD_CALLOC, NULL, dl->nr);
1154                 allloop = mloop = CustomData_add_layer(&me->ldata, CD_MLOOP, CD_CALLOC, NULL, dl->parts * 4);
1155                 mpoly = CustomData_add_layer(&me->pdata, CD_MPOLY, CD_CALLOC, NULL, dl->parts);
1156                 me->mvert = mvert;
1157                 me->mloop = mloop;
1158                 me->mpoly = mpoly;
1159                 me->totvert = dl->nr;
1160                 me->totpoly = dl->parts;
1161
1162                 a = dl->nr;
1163                 nors = dl->nors;
1164                 verts = dl->verts;
1165                 while (a--) {
1166                         copy_v3_v3(mvert->co, verts);
1167                         normal_float_to_short_v3(mvert->no, nors);
1168                         mvert++;
1169                         nors += 3;
1170                         verts += 3;
1171                 }
1172                 
1173                 a = dl->parts;
1174                 index = dl->index;
1175                 while (a--) {
1176                         int count = index[2] != index[3] ? 4 : 3;
1177
1178                         mloop[0].v = index[0];
1179                         mloop[1].v = index[1];
1180                         mloop[2].v = index[2];
1181                         if (count == 4)
1182                                 mloop[3].v = index[3];
1183
1184                         mpoly->totloop = count;
1185                         mpoly->loopstart = (int)(mloop - allloop);
1186                         mpoly->flag = ME_SMOOTH;
1187
1188
1189                         mpoly++;
1190                         mloop += count;
1191                         me->totloop += count;
1192                         index += 4;
1193                 }
1194
1195                 mesh_update_customdata_pointers(me, TRUE);
1196
1197                 BKE_mesh_calc_normals(me->mvert, me->totvert, me->mloop, me->mpoly, me->totloop, me->totpoly, NULL);
1198
1199                 BKE_mesh_calc_edges(me, TRUE);
1200         }
1201 }
1202
1203 /* Initialize mverts, medges and, faces for converting nurbs to mesh and derived mesh */
1204 /* return non-zero on error */
1205 int BKE_mesh_nurbs_to_mdata(Object *ob, MVert **allvert, int *totvert,
1206                             MEdge **alledge, int *totedge, MLoop **allloop, MPoly **allpoly,
1207                             int *totloop, int *totpoly)
1208 {
1209         return BKE_mesh_nurbs_displist_to_mdata(ob, &ob->disp,
1210                                                 allvert, totvert,
1211                                                 alledge, totedge,
1212                                                 allloop, allpoly, NULL,
1213                                                 totloop, totpoly);
1214 }
1215
1216 /* BMESH: this doesn't calculate all edges from polygons,
1217  * only free standing edges are calculated */
1218
1219 /* Initialize mverts, medges and, faces for converting nurbs to mesh and derived mesh */
1220 /* use specified dispbase */
1221 int BKE_mesh_nurbs_displist_to_mdata(Object *ob, ListBase *dispbase,
1222                                      MVert **allvert, int *_totvert,
1223                                      MEdge **alledge, int *_totedge,
1224                                      MLoop **allloop, MPoly **allpoly,
1225                                      MLoopUV **alluv,
1226                                      int *_totloop, int *_totpoly)
1227 {
1228         DispList *dl;
1229         Curve *cu;
1230         MVert *mvert;
1231         MPoly *mpoly;
1232         MLoop *mloop;
1233         MLoopUV *mloopuv = NULL;
1234         MEdge *medge;
1235         float *data;
1236         int a, b, ofs, vertcount, startvert, totvert = 0, totedge = 0, totloop = 0, totvlak = 0;
1237         int p1, p2, p3, p4, *index;
1238         int conv_polys = 0;
1239
1240         cu = ob->data;
1241
1242         conv_polys |= cu->flag & CU_3D;      /* 2d polys are filled with DL_INDEX3 displists */
1243         conv_polys |= ob->type == OB_SURF;   /* surf polys are never filled */
1244
1245         /* count */
1246         dl = dispbase->first;
1247         while (dl) {
1248                 if (dl->type == DL_SEGM) {
1249                         totvert += dl->parts * dl->nr;
1250                         totedge += dl->parts * (dl->nr - 1);
1251                 }
1252                 else if (dl->type == DL_POLY) {
1253                         if (conv_polys) {
1254                                 totvert += dl->parts * dl->nr;
1255                                 totedge += dl->parts * dl->nr;
1256                         }
1257                 }
1258                 else if (dl->type == DL_SURF) {
1259                         int tot;
1260                         totvert += dl->parts * dl->nr;
1261                         tot = (dl->parts - 1 + ((dl->flag & DL_CYCL_V) == 2)) * (dl->nr - 1 + (dl->flag & DL_CYCL_U));
1262                         totvlak += tot;
1263                         totloop += tot * 4;
1264                 }
1265                 else if (dl->type == DL_INDEX3) {
1266                         int tot;
1267                         totvert += dl->nr;
1268                         tot = dl->parts;
1269                         totvlak += tot;
1270                         totloop += tot * 3;
1271                 }
1272                 dl = dl->next;
1273         }
1274
1275         if (totvert == 0) {
1276                 /* error("can't convert"); */
1277                 /* Make Sure you check ob->data is a curve */
1278                 return -1;
1279         }
1280
1281         *allvert = mvert = MEM_callocN(sizeof(MVert) * totvert, "nurbs_init mvert");
1282         *alledge = medge = MEM_callocN(sizeof(MEdge) * totedge, "nurbs_init medge");
1283         *allloop = mloop = MEM_callocN(sizeof(MLoop) * totvlak * 4, "nurbs_init mloop"); // totloop
1284         *allpoly = mpoly = MEM_callocN(sizeof(MPoly) * totvlak, "nurbs_init mloop");
1285
1286         if (alluv)
1287                 *alluv = mloopuv = MEM_callocN(sizeof(MLoopUV) * totvlak * 4, "nurbs_init mloopuv");
1288         
1289         /* verts and faces */
1290         vertcount = 0;
1291
1292         dl = dispbase->first;
1293         while (dl) {
1294                 int smooth = dl->rt & CU_SMOOTH ? 1 : 0;
1295
1296                 if (dl->type == DL_SEGM) {
1297                         startvert = vertcount;
1298                         a = dl->parts * dl->nr;
1299                         data = dl->verts;
1300                         while (a--) {
1301                                 copy_v3_v3(mvert->co, data);
1302                                 data += 3;
1303                                 vertcount++;
1304                                 mvert++;
1305                         }
1306
1307                         for (a = 0; a < dl->parts; a++) {
1308                                 ofs = a * dl->nr;
1309                                 for (b = 1; b < dl->nr; b++) {
1310                                         medge->v1 = startvert + ofs + b - 1;
1311                                         medge->v2 = startvert + ofs + b;
1312                                         medge->flag = ME_LOOSEEDGE | ME_EDGERENDER | ME_EDGEDRAW;
1313
1314                                         medge++;
1315                                 }
1316                         }
1317
1318                 }
1319                 else if (dl->type == DL_POLY) {
1320                         if (conv_polys) {
1321                                 startvert = vertcount;
1322                                 a = dl->parts * dl->nr;
1323                                 data = dl->verts;
1324                                 while (a--) {
1325                                         copy_v3_v3(mvert->co, data);
1326                                         data += 3;
1327                                         vertcount++;
1328                                         mvert++;
1329                                 }
1330
1331                                 for (a = 0; a < dl->parts; a++) {
1332                                         ofs = a * dl->nr;
1333                                         for (b = 0; b < dl->nr; b++) {
1334                                                 medge->v1 = startvert + ofs + b;
1335                                                 if (b == dl->nr - 1) medge->v2 = startvert + ofs;
1336                                                 else medge->v2 = startvert + ofs + b + 1;
1337                                                 medge->flag = ME_LOOSEEDGE | ME_EDGERENDER | ME_EDGEDRAW;
1338                                                 medge++;
1339                                         }
1340                                 }
1341                         }
1342                 }
1343                 else if (dl->type == DL_INDEX3) {
1344                         startvert = vertcount;
1345                         a = dl->nr;
1346                         data = dl->verts;
1347                         while (a--) {
1348                                 copy_v3_v3(mvert->co, data);
1349                                 data += 3;
1350                                 vertcount++;
1351                                 mvert++;
1352                         }
1353
1354                         a = dl->parts;
1355                         index = dl->index;
1356                         while (a--) {
1357                                 mloop[0].v = startvert + index[0];
1358                                 mloop[1].v = startvert + index[2];
1359                                 mloop[2].v = startvert + index[1];
1360                                 mpoly->loopstart = (int)(mloop - (*allloop));
1361                                 mpoly->totloop = 3;
1362                                 mpoly->mat_nr = dl->col;
1363
1364                                 if (mloopuv) {
1365                                         int i;
1366
1367                                         for (i = 0; i < 3; i++, mloopuv++) {
1368                                                 mloopuv->uv[0] = (mloop[i].v - startvert) / (float)(dl->nr - 1);
1369                                                 mloopuv->uv[1] = 0.0f;
1370                                         }
1371                                 }
1372
1373                                 if (smooth) mpoly->flag |= ME_SMOOTH;
1374                                 mpoly++;
1375                                 mloop += 3;
1376                                 index += 3;
1377                         }
1378                 }
1379                 else if (dl->type == DL_SURF) {
1380                         startvert = vertcount;
1381                         a = dl->parts * dl->nr;
1382                         data = dl->verts;
1383                         while (a--) {
1384                                 copy_v3_v3(mvert->co, data);
1385                                 data += 3;
1386                                 vertcount++;
1387                                 mvert++;
1388                         }
1389
1390                         for (a = 0; a < dl->parts; a++) {
1391
1392                                 if ( (dl->flag & DL_CYCL_V) == 0 && a == dl->parts - 1) break;
1393
1394                                 if (dl->flag & DL_CYCL_U) {         /* p2 -> p1 -> */
1395                                         p1 = startvert + dl->nr * a;    /* p4 -> p3 -> */
1396                                         p2 = p1 + dl->nr - 1;       /* -----> next row */
1397                                         p3 = p1 + dl->nr;
1398                                         p4 = p2 + dl->nr;
1399                                         b = 0;
1400                                 }
1401                                 else {
1402                                         p2 = startvert + dl->nr * a;
1403                                         p1 = p2 + 1;
1404                                         p4 = p2 + dl->nr;
1405                                         p3 = p1 + dl->nr;
1406                                         b = 1;
1407                                 }
1408                                 if ( (dl->flag & DL_CYCL_V) && a == dl->parts - 1) {
1409                                         p3 -= dl->parts * dl->nr;
1410                                         p4 -= dl->parts * dl->nr;
1411                                 }
1412
1413                                 for (; b < dl->nr; b++) {
1414                                         mloop[0].v = p1;
1415                                         mloop[1].v = p3;
1416                                         mloop[2].v = p4;
1417                                         mloop[3].v = p2;
1418                                         mpoly->loopstart = (int)(mloop - (*allloop));
1419                                         mpoly->totloop = 4;
1420                                         mpoly->mat_nr = dl->col;
1421
1422                                         if (mloopuv) {
1423                                                 int orco_sizeu = dl->nr - 1;
1424                                                 int orco_sizev = dl->parts - 1;
1425                                                 int i;
1426
1427                                                 /* exception as handled in convertblender.c too */
1428                                                 if (dl->flag & DL_CYCL_U) {
1429                                                         orco_sizeu++;
1430                                                         if (dl->flag & DL_CYCL_V)
1431                                                                 orco_sizev++;
1432                                                 }
1433
1434                                                 for (i = 0; i < 4; i++, mloopuv++) {
1435                                                         /* find uv based on vertex index into grid array */
1436                                                         int v = mloop[i].v - startvert;
1437
1438                                                         mloopuv->uv[0] = (v / dl->nr) / (float)orco_sizev;
1439                                                         mloopuv->uv[1] = (v % dl->nr) / (float)orco_sizeu;
1440
1441                                                         /* cyclic correction */
1442                                                         if ((i == 0 || i == 1) && mloopuv->uv[1] == 0.0f)
1443                                                                 mloopuv->uv[1] = 1.0f;
1444                                                 }
1445                                         }
1446
1447                                         if (smooth) mpoly->flag |= ME_SMOOTH;
1448                                         mpoly++;
1449                                         mloop += 4;
1450
1451                                         p4 = p3;
1452                                         p3++;
1453                                         p2 = p1;
1454                                         p1++;
1455                                 }
1456                         }
1457                 }
1458
1459                 dl = dl->next;
1460         }
1461         
1462         *_totpoly = totvlak;
1463         *_totloop = totloop;
1464         *_totedge = totedge;
1465         *_totvert = totvert;
1466
1467         /* not uded for bmesh */
1468 #if 0
1469         make_edges_mdata(*allvert, *allface, *allloop, *allpoly, totvert, totvlak, *_totloop, *_totpoly, 0, alledge, _totedge);
1470         mfaces_strip_loose(*allface, _totface);
1471 #endif
1472
1473         return 0;
1474 }
1475
1476
1477 /* this may fail replacing ob->data, be sure to check ob->type */
1478 void BKE_mesh_from_nurbs_displist(Object *ob, ListBase *dispbase, int use_orco_uv)
1479 {
1480         Main *bmain = G.main;
1481         Object *ob1;
1482         DerivedMesh *dm = ob->derivedFinal;
1483         Mesh *me;
1484         Curve *cu;
1485         MVert *allvert = NULL;
1486         MEdge *alledge = NULL;
1487         MLoop *allloop = NULL;
1488         MLoopUV *alluv = NULL;
1489         MPoly *allpoly = NULL;
1490         int totvert, totedge, totloop, totpoly;
1491
1492         cu = ob->data;
1493
1494         if (dm == NULL) {
1495                 if (BKE_mesh_nurbs_displist_to_mdata(ob, dispbase, &allvert, &totvert,
1496                                                      &alledge, &totedge, &allloop,
1497                                                      &allpoly, (use_orco_uv) ? &alluv : NULL,
1498                                                      &totloop, &totpoly) != 0)
1499                 {
1500                         /* Error initializing */
1501                         return;
1502                 }
1503
1504                 /* make mesh */
1505                 me = BKE_mesh_add(G.main, "Mesh");
1506                 me->totvert = totvert;
1507                 me->totedge = totedge;
1508                 me->totloop = totloop;
1509                 me->totpoly = totpoly;
1510
1511                 me->mvert = CustomData_add_layer(&me->vdata, CD_MVERT, CD_ASSIGN, allvert, me->totvert);
1512                 me->medge = CustomData_add_layer(&me->edata, CD_MEDGE, CD_ASSIGN, alledge, me->totedge);
1513                 me->mloop = CustomData_add_layer(&me->ldata, CD_MLOOP, CD_ASSIGN, allloop, me->totloop);
1514                 me->mpoly = CustomData_add_layer(&me->pdata, CD_MPOLY, CD_ASSIGN, allpoly, me->totpoly);
1515
1516                 if (alluv) {
1517                         const char *uvname = "Orco";
1518                         me->mtpoly = CustomData_add_layer_named(&me->pdata, CD_MTEXPOLY, CD_DEFAULT, NULL, me->totpoly, uvname);
1519                         me->mloopuv = CustomData_add_layer_named(&me->ldata, CD_MLOOPUV, CD_ASSIGN, alluv, me->totloop, uvname);
1520                 }
1521
1522                 BKE_mesh_calc_normals(me->mvert, me->totvert, me->mloop, me->mpoly, me->totloop, me->totpoly, NULL);
1523
1524                 BKE_mesh_calc_edges(me, TRUE);
1525         }
1526         else {
1527                 me = BKE_mesh_add(G.main, "Mesh");
1528                 DM_to_mesh(dm, me, ob);
1529         }
1530
1531         me->totcol = cu->totcol;
1532         me->mat = cu->mat;
1533
1534         BKE_mesh_texspace_calc(me);
1535
1536         cu->mat = NULL;
1537         cu->totcol = 0;
1538
1539         if (ob->data) {
1540                 BKE_libblock_free(&bmain->curve, ob->data);
1541         }
1542         ob->data = me;
1543         ob->type = OB_MESH;
1544
1545         /* other users */
1546         ob1 = bmain->object.first;
1547         while (ob1) {
1548                 if (ob1->data == cu) {
1549                         ob1->type = OB_MESH;
1550                 
1551                         ob1->data = ob->data;
1552                         id_us_plus((ID *)ob->data);
1553                 }
1554                 ob1 = ob1->id.next;
1555         }
1556 }
1557
1558 void BKE_mesh_from_nurbs(Object *ob)
1559 {
1560         BKE_mesh_from_nurbs_displist(ob, &ob->disp, false);
1561 }
1562
1563 typedef struct EdgeLink {
1564         Link *next, *prev;
1565         void *edge;
1566 } EdgeLink;
1567
1568 typedef struct VertLink {
1569         Link *next, *prev;
1570         unsigned int index;
1571 } VertLink;
1572
1573 static void prependPolyLineVert(ListBase *lb, unsigned int index)
1574 {
1575         VertLink *vl = MEM_callocN(sizeof(VertLink), "VertLink");
1576         vl->index = index;
1577         BLI_addhead(lb, vl);
1578 }
1579
1580 static void appendPolyLineVert(ListBase *lb, unsigned int index)
1581 {
1582         VertLink *vl = MEM_callocN(sizeof(VertLink), "VertLink");
1583         vl->index = index;
1584         BLI_addtail(lb, vl);
1585 }
1586
1587 void BKE_mesh_from_curve(Scene *scene, Object *ob)
1588 {
1589         /* make new mesh data from the original copy */
1590         DerivedMesh *dm = mesh_get_derived_final(scene, ob, CD_MASK_MESH);
1591
1592         MVert *mverts = dm->getVertArray(dm);
1593         MEdge *med, *medge = dm->getEdgeArray(dm);
1594         MFace *mf,  *mface = dm->getTessFaceArray(dm);
1595
1596         int totedge = dm->getNumEdges(dm);
1597         int totface = dm->getNumTessFaces(dm);
1598         int totedges = 0;
1599         int i, needsFree = 0;
1600
1601         /* only to detect edge polylines */
1602         EdgeHash *eh = BLI_edgehash_new();
1603         EdgeHash *eh_edge = BLI_edgehash_new();
1604
1605
1606         ListBase edges = {NULL, NULL};
1607
1608         /* create edges from all faces (so as to find edges not in any faces) */
1609         mf = mface;
1610         for (i = 0; i < totface; i++, mf++) {
1611                 if (!BLI_edgehash_haskey(eh, mf->v1, mf->v2))
1612                         BLI_edgehash_insert(eh, mf->v1, mf->v2, NULL);
1613                 if (!BLI_edgehash_haskey(eh, mf->v2, mf->v3))
1614                         BLI_edgehash_insert(eh, mf->v2, mf->v3, NULL);
1615
1616                 if (mf->v4) {
1617                         if (!BLI_edgehash_haskey(eh, mf->v3, mf->v4))
1618                                 BLI_edgehash_insert(eh, mf->v3, mf->v4, NULL);
1619                         if (!BLI_edgehash_haskey(eh, mf->v4, mf->v1))
1620                                 BLI_edgehash_insert(eh, mf->v4, mf->v1, NULL);
1621                 }
1622                 else {
1623                         if (!BLI_edgehash_haskey(eh, mf->v3, mf->v1))
1624                                 BLI_edgehash_insert(eh, mf->v3, mf->v1, NULL);
1625                 }
1626         }
1627
1628         med = medge;
1629         for (i = 0; i < totedge; i++, med++) {
1630                 if (!BLI_edgehash_haskey(eh, med->v1, med->v2)) {
1631                         EdgeLink *edl = MEM_callocN(sizeof(EdgeLink), "EdgeLink");
1632
1633                         BLI_edgehash_insert(eh_edge, med->v1, med->v2, NULL);
1634                         edl->edge = med;
1635
1636                         BLI_addtail(&edges, edl);   totedges++;
1637                 }
1638         }
1639         BLI_edgehash_free(eh_edge, NULL);
1640         BLI_edgehash_free(eh, NULL);
1641
1642         if (edges.first) {
1643                 Curve *cu = BKE_curve_add(G.main, ob->id.name + 2, OB_CURVE);
1644                 cu->flag |= CU_3D;
1645
1646                 while (edges.first) {
1647                         /* each iteration find a polyline and add this as a nurbs poly spline */
1648
1649                         ListBase polyline = {NULL, NULL}; /* store a list of VertLink's */
1650                         int closed = FALSE;
1651                         int totpoly = 0;
1652                         MEdge *med_current = ((EdgeLink *)edges.last)->edge;
1653                         unsigned int startVert = med_current->v1;
1654                         unsigned int endVert = med_current->v2;
1655                         int ok = TRUE;
1656
1657                         appendPolyLineVert(&polyline, startVert);   totpoly++;
1658                         appendPolyLineVert(&polyline, endVert);     totpoly++;
1659                         BLI_freelinkN(&edges, edges.last);          totedges--;
1660
1661                         while (ok) { /* while connected edges are found... */
1662                                 ok = FALSE;
1663                                 i = totedges;
1664                                 while (i) {
1665                                         EdgeLink *edl;
1666
1667                                         i -= 1;
1668                                         edl = BLI_findlink(&edges, i);
1669                                         med = edl->edge;
1670
1671                                         if (med->v1 == endVert) {
1672                                                 endVert = med->v2;
1673                                                 appendPolyLineVert(&polyline, med->v2); totpoly++;
1674                                                 BLI_freelinkN(&edges, edl);             totedges--;
1675                                                 ok = TRUE;
1676                                         }
1677                                         else if (med->v2 == endVert) {
1678                                                 endVert = med->v1;
1679                                                 appendPolyLineVert(&polyline, endVert); totpoly++;
1680                                                 BLI_freelinkN(&edges, edl);             totedges--;
1681                                                 ok = TRUE;
1682                                         }
1683                                         else if (med->v1 == startVert) {
1684                                                 startVert = med->v2;
1685                                                 prependPolyLineVert(&polyline, startVert);  totpoly++;
1686                                                 BLI_freelinkN(&edges, edl);                 totedges--;
1687                                                 ok = TRUE;
1688                                         }
1689                                         else if (med->v2 == startVert) {
1690                                                 startVert = med->v1;
1691                                                 prependPolyLineVert(&polyline, startVert);  totpoly++;
1692                                                 BLI_freelinkN(&edges, edl);                 totedges--;
1693                                                 ok = TRUE;
1694                                         }
1695                                 }
1696                         }
1697
1698                         /* Now we have a polyline, make into a curve */
1699                         if (startVert == endVert) {
1700                                 BLI_freelinkN(&polyline, polyline.last);
1701                                 totpoly--;
1702                                 closed = TRUE;
1703                         }
1704
1705                         /* --- nurbs --- */
1706                         {
1707                                 Nurb *nu;
1708                                 BPoint *bp;
1709                                 VertLink *vl;
1710
1711                                 /* create new 'nurb' within the curve */
1712                                 nu = (Nurb *)MEM_callocN(sizeof(Nurb), "MeshNurb");
1713
1714                                 nu->pntsu = totpoly;
1715                                 nu->pntsv = 1;
1716                                 nu->orderu = 4;
1717                                 nu->flagu = CU_NURB_ENDPOINT | (closed ? CU_NURB_CYCLIC : 0);  /* endpoint */
1718                                 nu->resolu = 12;
1719
1720                                 nu->bp = (BPoint *)MEM_callocN(sizeof(BPoint) * totpoly, "bpoints");
1721
1722                                 /* add points */
1723                                 vl = polyline.first;
1724                                 for (i = 0, bp = nu->bp; i < totpoly; i++, bp++, vl = (VertLink *)vl->next) {
1725                                         copy_v3_v3(bp->vec, mverts[vl->index].co);
1726                                         bp->f1 = SELECT;
1727                                         bp->radius = bp->weight = 1.0;
1728                                 }
1729                                 BLI_freelistN(&polyline);
1730
1731                                 /* add nurb to curve */
1732                                 BLI_addtail(&cu->nurb, nu);
1733                         }
1734                         /* --- done with nurbs --- */
1735                 }
1736
1737                 ((Mesh *)ob->data)->id.us--;
1738                 ob->data = cu;
1739                 ob->type = OB_CURVE;
1740
1741                 /* curve objects can't contain DM in usual cases, we could free memory */
1742                 needsFree = 1;
1743         }
1744
1745         dm->needsFree = needsFree;
1746         dm->release(dm);
1747
1748         if (needsFree) {
1749                 ob->derivedFinal = NULL;
1750
1751                 /* curve object could have got bounding box only in special cases */
1752                 if (ob->bb) {
1753                         MEM_freeN(ob->bb);
1754                         ob->bb = NULL;
1755                 }
1756         }
1757 }
1758
1759 void BKE_mesh_delete_material_index(Mesh *me, short index)
1760 {
1761         int i;
1762
1763         for (i = 0; i < me->totpoly; i++) {
1764                 MPoly *mp = &((MPoly *) me->mpoly)[i];
1765                 if (mp->mat_nr && mp->mat_nr >= index)
1766                         mp->mat_nr--;
1767         }
1768         
1769         for (i = 0; i < me->totface; i++) {
1770                 MFace *mf = &((MFace *) me->mface)[i];
1771                 if (mf->mat_nr && mf->mat_nr >= index)
1772                         mf->mat_nr--;
1773         }
1774 }
1775
1776 void BKE_mesh_smooth_flag_set(Object *meshOb, int enableSmooth) 
1777 {
1778         Mesh *me = meshOb->data;
1779         int i;
1780
1781         for (i = 0; i < me->totpoly; i++) {
1782                 MPoly *mp = &((MPoly *) me->mpoly)[i];
1783
1784                 if (enableSmooth) {
1785                         mp->flag |= ME_SMOOTH;
1786                 }
1787                 else {
1788                         mp->flag &= ~ME_SMOOTH;
1789                 }
1790         }
1791         
1792         for (i = 0; i < me->totface; i++) {
1793                 MFace *mf = &((MFace *) me->mface)[i];
1794
1795                 if (enableSmooth) {
1796                         mf->flag |= ME_SMOOTH;
1797                 }
1798                 else {
1799                         mf->flag &= ~ME_SMOOTH;
1800                 }
1801         }
1802 }
1803
1804 void BKE_mesh_calc_normals_mapping(MVert *mverts, int numVerts,
1805                                    MLoop *mloop, MPoly *mpolys, int numLoops, int numPolys, float (*polyNors_r)[3],
1806                                    MFace *mfaces, int numFaces, int *origIndexFace, float (*faceNors_r)[3])
1807 {
1808         BKE_mesh_calc_normals_mapping_ex(mverts, numVerts, mloop, mpolys,
1809                                          numLoops, numPolys, polyNors_r, mfaces, numFaces,
1810                                          origIndexFace, faceNors_r, FALSE);
1811 }
1812
1813 void BKE_mesh_calc_normals_mapping_ex(MVert *mverts, int numVerts,
1814                                       MLoop *mloop, MPoly *mpolys,
1815                                       int numLoops, int numPolys, float (*polyNors_r)[3],
1816                                       MFace *mfaces, int numFaces, int *origIndexFace, float (*faceNors_r)[3],
1817                                       const short only_face_normals)
1818 {
1819         float (*pnors)[3] = polyNors_r, (*fnors)[3] = faceNors_r;
1820         int i;
1821         MFace *mf;
1822         MPoly *mp;
1823
1824         if (numPolys == 0) {
1825                 return;
1826         }
1827
1828         /* if we are not calculating verts and no verts were passes then we have nothing to do */
1829         if ((only_face_normals == TRUE) && (polyNors_r == NULL) && (faceNors_r == NULL)) {
1830                 printf("%s: called with nothing to do\n", __func__);
1831                 return;
1832         }
1833
1834         if (!pnors) pnors = MEM_callocN(sizeof(float) * 3 * numPolys, "poly_nors mesh.c");
1835         /* if (!fnors) fnors = MEM_callocN(sizeof(float) * 3 * numFaces, "face nors mesh.c"); */ /* NO NEED TO ALLOC YET */
1836
1837
1838         if (only_face_normals == FALSE) {
1839                 /* vertex normals are optional, they require some extra calculations,
1840                  * so make them optional */
1841                 BKE_mesh_calc_normals(mverts, numVerts, mloop, mpolys, numLoops, numPolys, pnors);
1842         }
1843         else {
1844                 /* only calc poly normals */
1845                 mp = mpolys;
1846                 for (i = 0; i < numPolys; i++, mp++) {
1847                         BKE_mesh_calc_poly_normal(mp, mloop + mp->loopstart, mverts, pnors[i]);
1848                 }
1849         }
1850
1851         if (origIndexFace &&
1852             /* fnors == faceNors_r */ /* NO NEED TO ALLOC YET */
1853             fnors != NULL &&
1854             numFaces)
1855         {
1856                 mf = mfaces;
1857                 for (i = 0; i < numFaces; i++, mf++, origIndexFace++) {
1858                         if (*origIndexFace < numPolys) {
1859                                 copy_v3_v3(fnors[i], pnors[*origIndexFace]);
1860                         }
1861                         else {
1862                                 /* eek, we're not corresponding to polys */
1863                                 printf("error in BKE_mesh_calc_normals; tessellation face indices are incorrect.  normals may look bad.\n");
1864                         }
1865                 }
1866         }
1867
1868         if (pnors != polyNors_r) MEM_freeN(pnors);
1869         /* if (fnors != faceNors_r) MEM_freeN(fnors); */ /* NO NEED TO ALLOC YET */
1870
1871         fnors = pnors = NULL;
1872         
1873 }
1874
1875 void BKE_mesh_calc_normals(MVert *mverts, int numVerts, MLoop *mloop, MPoly *mpolys,
1876                            int UNUSED(numLoops), int numPolys, float (*polyNors_r)[3])
1877 {
1878         float (*pnors)[3] = polyNors_r;
1879
1880         float (*tnorms)[3], (*edgevecbuf)[3] = NULL;
1881         float **vertcos = NULL, **vertnos = NULL;
1882         BLI_array_declare(vertcos);
1883         BLI_array_declare(vertnos);
1884         BLI_array_declare(edgevecbuf);
1885
1886         int i, j;
1887         MPoly *mp;
1888         MLoop *ml;
1889
1890         if (!pnors) pnors = MEM_callocN(sizeof(float) * 3 * numPolys, "poly_nors mesh.c");
1891
1892         /* first go through and calculate normals for all the polys */
1893         tnorms = MEM_callocN(sizeof(float) * 3 * numVerts, "tnorms mesh.c");
1894
1895         mp = mpolys;
1896         for (i = 0; i < numPolys; i++, mp++) {
1897                 BKE_mesh_calc_poly_normal(mp, mloop + mp->loopstart, mverts, pnors[i]);
1898                 ml = mloop + mp->loopstart;
1899
1900                 BLI_array_empty(vertcos);
1901                 BLI_array_empty(vertnos);
1902                 BLI_array_grow_items(vertcos, mp->totloop);
1903                 BLI_array_grow_items(vertnos, mp->totloop);
1904
1905                 for (j = 0; j < mp->totloop; j++) {
1906                         int vindex = ml[j].v;
1907                         vertcos[j] = mverts[vindex].co;
1908                         vertnos[j] = tnorms[vindex];
1909                 }
1910
1911                 BLI_array_empty(edgevecbuf);
1912                 BLI_array_grow_items(edgevecbuf, mp->totloop);
1913
1914                 accumulate_vertex_normals_poly(vertnos, pnors[i], vertcos, edgevecbuf, mp->totloop);
1915         }
1916
1917         BLI_array_free(vertcos);
1918         BLI_array_free(vertnos);
1919         BLI_array_free(edgevecbuf);
1920
1921         /* following Mesh convention; we use vertex coordinate itself for normal in this case */
1922         for (i = 0; i < numVerts; i++) {
1923                 MVert *mv = &mverts[i];
1924                 float *no = tnorms[i];
1925
1926                 if (UNLIKELY(normalize_v3(no) == 0.0f)) {
1927                         normalize_v3_v3(no, mv->co);
1928                 }
1929
1930                 normal_float_to_short_v3(mv->no, no);
1931         }
1932
1933         MEM_freeN(tnorms);
1934
1935         if (pnors != polyNors_r) MEM_freeN(pnors);
1936 }
1937
1938 void BKE_mesh_calc_normals_tessface(MVert *mverts, int numVerts, MFace *mfaces, int numFaces, float (*faceNors_r)[3])
1939 {
1940         float (*tnorms)[3] = MEM_callocN(numVerts * sizeof(*tnorms), "tnorms");
1941         float (*fnors)[3] = (faceNors_r) ? faceNors_r : MEM_callocN(sizeof(*fnors) * numFaces, "meshnormals");
1942         int i;
1943
1944         for (i = 0; i < numFaces; i++) {
1945                 MFace *mf = &mfaces[i];
1946                 float *f_no = fnors[i];
1947                 float *n4 = (mf->v4) ? tnorms[mf->v4] : NULL;
1948                 float *c4 = (mf->v4) ? mverts[mf->v4].co : NULL;
1949
1950                 if (mf->v4)
1951                         normal_quad_v3(f_no, mverts[mf->v1].co, mverts[mf->v2].co, mverts[mf->v3].co, mverts[mf->v4].co);
1952                 else
1953                         normal_tri_v3(f_no, mverts[mf->v1].co, mverts[mf->v2].co, mverts[mf->v3].co);
1954
1955                 accumulate_vertex_normals(tnorms[mf->v1], tnorms[mf->v2], tnorms[mf->v3], n4,
1956                                           f_no, mverts[mf->v1].co, mverts[mf->v2].co, mverts[mf->v3].co, c4);
1957         }
1958
1959         /* following Mesh convention; we use vertex coordinate itself for normal in this case */
1960         for (i = 0; i < numVerts; i++) {
1961                 MVert *mv = &mverts[i];
1962                 float *no = tnorms[i];
1963                 
1964                 if (UNLIKELY(normalize_v3(no) == 0.0f)) {
1965                         normalize_v3_v3(no, mv->co);
1966                 }
1967
1968                 normal_float_to_short_v3(mv->no, no);
1969         }
1970         
1971         MEM_freeN(tnorms);
1972
1973         if (fnors != faceNors_r)
1974                 MEM_freeN(fnors);
1975 }
1976
1977 static void bm_corners_to_loops_ex(ID *id, CustomData *fdata, CustomData *ldata, CustomData *pdata,
1978                                    MFace *mface, int totloop, int findex, int loopstart, int numTex, int numCol)
1979 {
1980         MTFace *texface;
1981         MTexPoly *texpoly;
1982         MCol *mcol;
1983         MLoopCol *mloopcol;
1984         MLoopUV *mloopuv;
1985         MFace *mf;
1986         int i;
1987
1988         mf = mface + findex;
1989
1990         for (i = 0; i < numTex; i++) {
1991                 texface = CustomData_get_n(fdata, CD_MTFACE, findex, i);
1992                 texpoly = CustomData_get_n(pdata, CD_MTEXPOLY, findex, i);
1993
1994                 ME_MTEXFACE_CPY(texpoly, texface);
1995
1996                 mloopuv = CustomData_get_n(ldata, CD_MLOOPUV, loopstart, i);
1997                 copy_v2_v2(mloopuv->uv, texface->uv[0]); mloopuv++;
1998                 copy_v2_v2(mloopuv->uv, texface->uv[1]); mloopuv++;
1999                 copy_v2_v2(mloopuv->uv, texface->uv[2]); mloopuv++;
2000
2001                 if (mf->v4) {
2002                         copy_v2_v2(mloopuv->uv, texface->uv[3]); mloopuv++;
2003                 }
2004         }
2005
2006         for (i = 0; i < numCol; i++) {
2007                 mloopcol = CustomData_get_n(ldata, CD_MLOOPCOL, loopstart, i);
2008                 mcol = CustomData_get_n(fdata, CD_MCOL, findex, i);
2009
2010                 MESH_MLOOPCOL_FROM_MCOL(mloopcol, &mcol[0]); mloopcol++;
2011                 MESH_MLOOPCOL_FROM_MCOL(mloopcol, &mcol[1]); mloopcol++;
2012                 MESH_MLOOPCOL_FROM_MCOL(mloopcol, &mcol[2]); mloopcol++;
2013                 if (mf->v4) {
2014                         MESH_MLOOPCOL_FROM_MCOL(mloopcol, &mcol[3]); mloopcol++;
2015                 }
2016         }
2017
2018         if (CustomData_has_layer(fdata, CD_MDISPS)) {
2019                 MDisps *ld = CustomData_get(ldata, loopstart, CD_MDISPS);
2020                 MDisps *fd = CustomData_get(fdata, findex, CD_MDISPS);
2021                 float (*disps)[3] = fd->disps;
2022                 int tot = mf->v4 ? 4 : 3;
2023                 int side, corners;
2024
2025                 if (CustomData_external_test(fdata, CD_MDISPS)) {
2026                         if (id && fdata->external) {
2027                                 CustomData_external_add(ldata, id, CD_MDISPS,
2028                                                         totloop, fdata->external->filename);
2029                         }
2030                 }
2031
2032                 corners = multires_mdisp_corners(fd);
2033
2034                 if (corners == 0) {
2035                         /* Empty MDisp layers appear in at least one of the sintel.blend files.
2036                          * Not sure why this happens, but it seems fine to just ignore them here.
2037                          * If (corners == 0) for a non-empty layer though, something went wrong. */
2038                         BLI_assert(fd->totdisp == 0);
2039                 }
2040                 else {
2041                         side = sqrt(fd->totdisp / corners);
2042
2043                         for (i = 0; i < tot; i++, disps += side * side, ld++) {
2044                                 ld->totdisp = side * side;
2045                                 ld->level = (int)(logf(side - 1.0f) / (float)M_LN2) + 1;
2046
2047                                 if (ld->disps)
2048                                         MEM_freeN(ld->disps);
2049
2050                                 ld->disps = MEM_callocN(sizeof(float) * 3 * side * side, "converted loop mdisps");
2051                                 if (fd->disps) {
2052                                         memcpy(ld->disps, disps, sizeof(float) * 3 * side * side);
2053                                 }
2054                         }
2055                 }
2056         }
2057 }
2058
2059 void BKE_mesh_convert_mfaces_to_mpolys(Mesh *mesh)
2060 {
2061         BKE_mesh_convert_mfaces_to_mpolys_ex(&mesh->id, &mesh->fdata, &mesh->ldata, &mesh->pdata,
2062                                              mesh->totedge, mesh->totface, mesh->totloop, mesh->totpoly,
2063                                              mesh->medge, mesh->mface,
2064                                              &mesh->totloop, &mesh->totpoly, &mesh->mloop, &mesh->mpoly);
2065
2066         mesh_update_customdata_pointers(mesh, TRUE);
2067 }
2068
2069 /* the same as BKE_mesh_convert_mfaces_to_mpolys but oriented to be used in do_versions from readfile.c
2070  * the difference is how active/render/clone/stencil indices are handled here
2071  *
2072  * normally thay're being set from pdata which totally makes sense for meshes which are already
2073  * converted to bmesh structures, but when loading older files indices shall be updated in other
2074  * way around, so newly added pdata and ldata would have this indices set based on fdata layer
2075  *
2076  * this is normally only needed when reading older files, in all other cases BKE_mesh_convert_mfaces_to_mpolys
2077  * shall be always used
2078  */
2079 void BKE_mesh_do_versions_convert_mfaces_to_mpolys(Mesh *mesh)
2080 {
2081         BKE_mesh_convert_mfaces_to_mpolys_ex(&mesh->id, &mesh->fdata, &mesh->ldata, &mesh->pdata,
2082                                              mesh->totedge, mesh->totface, mesh->totloop, mesh->totpoly,
2083                                              mesh->medge, mesh->mface,
2084                                              &mesh->totloop, &mesh->totpoly, &mesh->mloop, &mesh->mpoly);
2085
2086         CustomData_bmesh_do_versions_update_active_layers(&mesh->fdata, &mesh->pdata, &mesh->ldata);
2087
2088         mesh_update_customdata_pointers(mesh, TRUE);
2089 }
2090
2091 void BKE_mesh_convert_mfaces_to_mpolys_ex(ID *id, CustomData *fdata, CustomData *ldata, CustomData *pdata,
2092                                           int totedge_i, int totface_i, int totloop_i, int totpoly_i,
2093                                           MEdge *medge, MFace *mface,
2094                                           int *totloop_r, int *totpoly_r,
2095                                           MLoop **mloop_r, MPoly **mpoly_r)
2096 {
2097         MFace *mf;
2098         MLoop *ml, *mloop;
2099         MPoly *mp, *mpoly;
2100         MEdge *me;
2101         EdgeHash *eh;
2102         int numTex, numCol;
2103         int i, j, totloop, totpoly, *polyindex;
2104
2105         /* just in case some of these layers are filled in (can happen with python created meshes) */
2106         CustomData_free(ldata, totloop_i);
2107         CustomData_free(pdata, totpoly_i);
2108
2109         totpoly = totface_i;
2110         mpoly = MEM_callocN(sizeof(MPoly) * totpoly, "mpoly converted");
2111         CustomData_add_layer(pdata, CD_MPOLY, CD_ASSIGN, mpoly, totpoly);
2112
2113         numTex = CustomData_number_of_layers(fdata, CD_MTFACE);
2114         numCol = CustomData_number_of_layers(fdata, CD_MCOL);
2115
2116         totloop = 0;
2117         mf = mface;
2118         for (i = 0; i < totface_i; i++, mf++) {
2119                 totloop += mf->v4 ? 4 : 3;
2120         }
2121
2122         mloop = MEM_callocN(sizeof(MLoop) * totloop, "mloop converted");
2123
2124         CustomData_add_layer(ldata, CD_MLOOP, CD_ASSIGN, mloop, totloop);
2125
2126         CustomData_to_bmeshpoly(fdata, pdata, ldata, totloop, totpoly);
2127
2128         if (id) {
2129                 /* ensure external data is transferred */
2130                 CustomData_external_read(fdata, id, CD_MASK_MDISPS, totface_i);
2131         }
2132
2133         eh = BLI_edgehash_new();
2134
2135         /* build edge hash */
2136         me = medge;
2137         for (i = 0; i < totedge_i; i++, me++) {
2138                 BLI_edgehash_insert(eh, me->v1, me->v2, SET_INT_IN_POINTER(i));
2139
2140                 /* unrelated but avoid having the FGON flag enabled, so we can reuse it later for something else */
2141                 me->flag &= ~ME_FGON;
2142         }
2143
2144         polyindex = CustomData_get_layer(fdata, CD_ORIGINDEX);
2145
2146         j = 0; /* current loop index */
2147         ml = mloop;
2148         mf = mface;
2149         mp = mpoly;
2150         for (i = 0; i < totface_i; i++, mf++, mp++) {
2151                 mp->loopstart = j;
2152
2153                 mp->totloop = mf->v4 ? 4 : 3;
2154
2155                 mp->mat_nr = mf->mat_nr;
2156                 mp->flag = mf->flag;
2157
2158 #       define ML(v1, v2) { \
2159                         ml->v = mf->v1; ml->e = GET_INT_FROM_POINTER(BLI_edgehash_lookup(eh, mf->v1, mf->v2)); ml++; j++; \
2160                 } (void)0
2161
2162                 ML(v1, v2);
2163                 ML(v2, v3);
2164                 if (mf->v4) {
2165                         ML(v3, v4);
2166                         ML(v4, v1);
2167                 }
2168                 else {
2169                         ML(v3, v1);
2170                 }
2171
2172 #       undef ML
2173
2174                 bm_corners_to_loops_ex(id, fdata, ldata, pdata, mface, totloop, i, mp->loopstart, numTex, numCol);
2175
2176                 if (polyindex) {
2177                         *polyindex = i;
2178                         polyindex++;
2179                 }
2180         }
2181
2182         /* note, we don't convert NGons at all, these are not even real ngons,
2183          * they have their own UV's, colors etc - its more an editing feature. */
2184
2185         BLI_edgehash_free(eh, NULL);
2186
2187         *totpoly_r = totpoly;
2188         *totloop_r = totloop;
2189         *mpoly_r = mpoly;
2190         *mloop_r = mloop;
2191 }
2192
2193 float (*mesh_getVertexCos(Mesh * me, int *numVerts_r))[3]
2194 {
2195         int i, numVerts = me->totvert;
2196         float (*cos)[3] = MEM_mallocN(sizeof(*cos) * numVerts, "vertexcos1");
2197
2198         if (numVerts_r) *numVerts_r = numVerts;
2199         for (i = 0; i < numVerts; i++)
2200                 copy_v3_v3(cos[i], me->mvert[i].co);
2201
2202         return cos;
2203 }
2204
2205
2206 /* ngon version wip, based on EDBM_uv_vert_map_create */
2207 /* this replaces the non bmesh function (in trunk) which takes MTFace's, if we ever need it back we could
2208  * but for now this replaces it because its unused. */
2209
2210 UvVertMap *BKE_mesh_uv_vert_map_make(struct MPoly *mpoly, struct MLoop *mloop, struct MLoopUV *mloopuv, unsigned int totpoly, unsigned int totvert, int selected, float *limit)
2211 {
2212         UvVertMap *vmap;
2213         UvMapVert *buf;
2214         MPoly *mp;
2215         unsigned int a;
2216         int i, totuv, nverts;
2217
2218         totuv = 0;
2219
2220         /* generate UvMapVert array */
2221         mp = mpoly;
2222         for (a = 0; a < totpoly; a++, mp++)
2223                 if (!selected || (!(mp->flag & ME_HIDE) && (mp->flag & ME_FACE_SEL)))
2224                         totuv += mp->totloop;
2225
2226         if (totuv == 0)
2227                 return NULL;
2228         
2229         vmap = (UvVertMap *)MEM_callocN(sizeof(*vmap), "UvVertMap");
2230         if (!vmap)
2231                 return NULL;
2232
2233         vmap->vert = (UvMapVert **)MEM_callocN(sizeof(*vmap->vert) * totvert, "UvMapVert*");
2234         buf = vmap->buf = (UvMapVert *)MEM_callocN(sizeof(*vmap->buf) * totuv, "UvMapVert");
2235
2236         if (!vmap->vert || !vmap->buf) {
2237                 BKE_mesh_uv_vert_map_free(vmap);
2238                 return NULL;
2239         }
2240
2241         mp = mpoly;
2242         for (a = 0; a < totpoly; a++, mp++) {
2243                 if (!selected || (!(mp->flag & ME_HIDE) && (mp->flag & ME_FACE_SEL))) {
2244                         nverts = mp->totloop;
2245
2246                         for (i = 0; i < nverts; i++) {
2247                                 buf->tfindex = i;
2248                                 buf->f = a;
2249                                 buf->separate = 0;
2250                                 buf->next = vmap->vert[mloop[mp->loopstart + i].v];
2251                                 vmap->vert[mloop[mp->loopstart + i].v] = buf;
2252                                 buf++;
2253                         }
2254                 }
2255         }
2256         
2257         /* sort individual uvs for each vert */
2258         for (a = 0; a < totvert; a++) {
2259                 UvMapVert *newvlist = NULL, *vlist = vmap->vert[a];
2260                 UvMapVert *iterv, *v, *lastv, *next;
2261                 float *uv, *uv2, uvdiff[2];
2262
2263                 while (vlist) {
2264                         v = vlist;
2265                         vlist = vlist->next;
2266                         v->next = newvlist;
2267                         newvlist = v;
2268
2269                         uv = mloopuv[mpoly[v->f].loopstart + v->tfindex].uv;
2270                         lastv = NULL;
2271                         iterv = vlist;
2272
2273                         while (iterv) {
2274                                 next = iterv->next;
2275
2276                                 uv2 = mloopuv[mpoly[iterv->f].loopstart + iterv->tfindex].uv;
2277                                 sub_v2_v2v2(uvdiff, uv2, uv);
2278
2279
2280                                 if (fabsf(uv[0] - uv2[0]) < limit[0] && fabsf(uv[1] - uv2[1]) < limit[1]) {
2281                                         if (lastv) lastv->next = next;
2282                                         else vlist = next;
2283                                         iterv->next = newvlist;
2284                                         newvlist = iterv;
2285                                 }
2286                                 else
2287                                         lastv = iterv;
2288
2289                                 iterv = next;
2290                         }
2291
2292                         newvlist->separate = 1;
2293                 }
2294
2295                 vmap->vert[a] = newvlist;
2296         }
2297         
2298         return vmap;
2299 }
2300
2301 UvMapVert *BKE_mesh_uv_vert_map_get_vert(UvVertMap *vmap, unsigned int v)
2302 {
2303         return vmap->vert[v];
2304 }
2305
2306 void BKE_mesh_uv_vert_map_free(UvVertMap *vmap)
2307 {
2308         if (vmap) {
2309                 if (vmap->vert) MEM_freeN(vmap->vert);
2310                 if (vmap->buf) MEM_freeN(vmap->buf);
2311                 MEM_freeN(vmap);
2312         }
2313 }
2314
2315 /* Generates a map where the key is the vertex and the value is a list
2316  * of polys that use that vertex as a corner. The lists are allocated
2317  * from one memory pool. */
2318 void create_vert_poly_map(MeshElemMap **map, int **mem,
2319                           const MPoly *mpoly, const MLoop *mloop,
2320                           int totvert, int totpoly, int totloop)
2321 {
2322         int i, j;
2323         int *indices;
2324
2325         (*map) = MEM_callocN(sizeof(MeshElemMap) * totvert, "vert poly map");
2326         (*mem) = MEM_mallocN(sizeof(int) * totloop, "vert poly map mem");
2327
2328         /* Count number of polys for each vertex */
2329         for (i = 0; i < totpoly; i++) {
2330                 const MPoly *p = &mpoly[i];
2331                 
2332                 for (j = 0; j < p->totloop; j++)
2333                         (*map)[mloop[p->loopstart + j].v].count++;
2334         }
2335
2336         /* Assign indices mem */
2337         indices = (*mem);
2338         for (i = 0; i < totvert; i++) {
2339                 (*map)[i].indices = indices;
2340                 indices += (*map)[i].count;
2341
2342                 /* Reset 'count' for use as index in last loop */
2343                 (*map)[i].count = 0;
2344         }
2345                 
2346         /* Find the users */
2347         for (i = 0; i < totpoly; i++) {
2348                 const MPoly *p = &mpoly[i];
2349                 
2350                 for (j = 0; j < p->totloop; j++) {
2351                         int v = mloop[p->loopstart + j].v;
2352                         
2353                         (*map)[v].indices[(*map)[v].count] = i;
2354                         (*map)[v].count++;
2355                 }
2356         }
2357 }
2358
2359 /* Generates a map where the key is the vertex and the value is a list
2360  * of edges that use that vertex as an endpoint. The lists are allocated
2361  * from one memory pool. */
2362 void create_vert_edge_map(MeshElemMap **map, int **mem,
2363                           const MEdge *medge, int totvert, int totedge)
2364 {
2365         int i, *indices;
2366
2367         (*map) = MEM_callocN(sizeof(MeshElemMap) * totvert, "vert-edge map");
2368         (*mem) = MEM_mallocN(sizeof(int) * totedge * 2, "vert-edge map mem");
2369
2370         /* Count number of edges for each vertex */
2371         for (i = 0; i < totedge; i++) {
2372                 (*map)[medge[i].v1].count++;
2373                 (*map)[medge[i].v2].count++;
2374         }
2375
2376         /* Assign indices mem */
2377         indices = (*mem);
2378         for (i = 0; i < totvert; i++) {
2379                 (*map)[i].indices = indices;
2380                 indices += (*map)[i].count;
2381
2382                 /* Reset 'count' for use as index in last loop */
2383                 (*map)[i].count = 0;
2384         }
2385                 
2386         /* Find the users */
2387         for (i = 0; i < totedge; i++) {
2388                 const int v[2] = {medge[i].v1, medge[i].v2};
2389
2390                 (*map)[v[0]].indices[(*map)[v[0]].count] = i;
2391                 (*map)[v[1]].indices[(*map)[v[1]].count] = i;
2392                 
2393                 (*map)[v[0]].count++;
2394                 (*map)[v[1]].count++;
2395         }
2396 }
2397
2398 void BKE_mesh_loops_to_mface_corners(CustomData *fdata, CustomData *ldata,
2399                                      CustomData *pdata, int lindex[4], int findex,
2400                                      const int polyindex,
2401                                      const int mf_len, /* 3 or 4 */
2402
2403                                      /* cache values to avoid lookups every time */
2404                                      const int numTex, /* CustomData_number_of_layers(pdata, CD_MTEXPOLY) */
2405                                      const int numCol, /* CustomData_number_of_layers(ldata, CD_MLOOPCOL) */
2406                                      const int hasPCol, /* CustomData_has_layer(ldata, CD_PREVIEW_MLOOPCOL) */
2407                                      const int hasOrigSpace /* CustomData_has_layer(ldata, CD_ORIGSPACE_MLOOP) */
2408                                      )
2409 {
2410         MTFace *texface;
2411         MTexPoly *texpoly;
2412         MCol *mcol;
2413         MLoopCol *mloopcol;
2414         MLoopUV *mloopuv;
2415         int i, j;
2416         
2417         for (i = 0; i < numTex; i++) {
2418                 texface = CustomData_get_n(fdata, CD_MTFACE, findex, i);
2419                 texpoly = CustomData_get_n(pdata, CD_MTEXPOLY, polyindex, i);
2420
2421                 ME_MTEXFACE_CPY(texface, texpoly);
2422
2423                 for (j = 0; j < mf_len; j++) {
2424                         mloopuv = CustomData_get_n(ldata, CD_MLOOPUV, lindex[j], i);
2425                         copy_v2_v2(texface->uv[j], mloopuv->uv);
2426                 }
2427         }
2428
2429         for (i = 0; i < numCol; i++) {
2430                 mcol = CustomData_get_n(fdata, CD_MCOL, findex, i);
2431
2432                 for (j = 0; j < mf_len; j++) {
2433                         mloopcol = CustomData_get_n(ldata, CD_MLOOPCOL, lindex[j], i);
2434                         MESH_MLOOPCOL_TO_MCOL(mloopcol, &mcol[j]);
2435                 }
2436         }
2437
2438         if (hasPCol) {
2439                 mcol = CustomData_get(fdata,  findex, CD_PREVIEW_MCOL);
2440
2441                 for (j = 0; j < mf_len; j++) {
2442                         mloopcol = CustomData_get(ldata, lindex[j], CD_PREVIEW_MLOOPCOL);
2443                         MESH_MLOOPCOL_TO_MCOL(mloopcol, &mcol[j]);
2444                 }
2445         }
2446
2447         if (hasOrigSpace) {
2448                 OrigSpaceFace *of = CustomData_get(fdata, findex, CD_ORIGSPACE);
2449                 OrigSpaceLoop *lof;
2450
2451                 for (j = 0; j < mf_len; j++) {
2452                         lof = CustomData_get(ldata, lindex[j], CD_ORIGSPACE_MLOOP);
2453                         copy_v2_v2(of->uv[j], lof->uv);
2454                 }
2455         }
2456 }
2457
2458 /*
2459  * this function recreates a tessellation.
2460  * returns number of tessellation faces.
2461  */
2462 int BKE_mesh_recalc_tessellation(CustomData *fdata,
2463                                  CustomData *ldata, CustomData *pdata,
2464                                  MVert *mvert, int totface, int totloop,
2465                                  int totpoly,
2466                                  /* when tessellating to recalculate normals after
2467                                   * we can skip copying here */
2468                                  const int do_face_nor_cpy)
2469 {
2470         /* use this to avoid locking pthread for _every_ polygon
2471          * and calling the fill function */
2472
2473 #define USE_TESSFACE_SPEEDUP
2474 #define USE_TESSFACE_QUADS // NEEDS FURTHER TESTING
2475
2476 #define TESSFACE_SCANFILL (1 << 0)
2477 #define TESSFACE_IS_QUAD  (1 << 1)
2478
2479         const int looptris_tot = poly_to_tri_count(totpoly, totloop);
2480
2481         MPoly *mp, *mpoly;
2482         MLoop *ml, *mloop;
2483         MFace *mface, *mf;
2484         ScanFillContext sf_ctx;
2485         ScanFillVert *sf_vert, *sf_vert_last, *sf_vert_first;
2486         ScanFillFace *sf_tri;
2487         int *mface_to_poly_map;
2488         int lindex[4]; /* only ever use 3 in this case */
2489         int poly_index, j, mface_index;
2490
2491         const int numTex = CustomData_number_of_layers(pdata, CD_MTEXPOLY);
2492         const int numCol = CustomData_number_of_layers(ldata, CD_MLOOPCOL);
2493         const int hasPCol = CustomData_has_layer(ldata, CD_PREVIEW_MLOOPCOL);
2494         const int hasOrigSpace = CustomData_has_layer(ldata, CD_ORIGSPACE_MLOOP);
2495
2496         mpoly = CustomData_get_layer(pdata, CD_MPOLY);
2497         mloop = CustomData_get_layer(ldata, CD_MLOOP);
2498
2499         /* allocate the length of totfaces, avoid many small reallocs,
2500          * if all faces are tri's it will be correct, quads == 2x allocs */
2501         /* take care. we are _not_ calloc'ing so be sure to initialize each field */
2502         mface_to_poly_map = MEM_mallocN(sizeof(*mface_to_poly_map) * looptris_tot, __func__);
2503         mface             = MEM_mallocN(sizeof(*mface) *             looptris_tot, __func__);
2504
2505         mface_index = 0;
2506         mp = mpoly;
2507         for (poly_index = 0; poly_index < totpoly; poly_index++, mp++) {
2508                 if (mp->totloop < 3) {
2509                         /* do nothing */
2510                 }
2511
2512 #ifdef USE_TESSFACE_SPEEDUP
2513
2514 #define ML_TO_MF(i1, i2, i3)                                                  \
2515                 mface_to_poly_map[mface_index] = poly_index;                          \
2516                 mf = &mface[mface_index];                                             \
2517                 /* set loop indices, transformed to vert indices later */             \
2518                 mf->v1 = mp->loopstart + i1;                                          \
2519                 mf->v2 = mp->loopstart + i2;                                          \
2520                 mf->v3 = mp->loopstart + i3;                                          \
2521                 mf->v4 = 0;                                                           \
2522                 mf->mat_nr = mp->mat_nr;                                              \
2523                 mf->flag = mp->flag;                                                  \
2524                 mf->edcode = 0;                                                       \
2525                 (void)0
2526
2527 /* ALMOST IDENTICAL TO DEFINE ABOVE (see EXCEPTION) */
2528 #define ML_TO_MF_QUAD()                                                       \
2529                 mface_to_poly_map[mface_index] = poly_index;                          \
2530                 mf = &mface[mface_index];                                             \
2531                 /* set loop indices, transformed to vert indices later */             \
2532                 mf->v1 = mp->loopstart + 0; /* EXCEPTION */                           \
2533                 mf->v2 = mp->loopstart + 1; /* EXCEPTION */                           \
2534                 mf->v3 = mp->loopstart + 2; /* EXCEPTION */                           \
2535                 mf->v4 = mp->loopstart + 3; /* EXCEPTION */                           \
2536                 mf->mat_nr = mp->mat_nr;                                              \
2537                 mf->flag = mp->flag;                                                  \
2538                 mf->edcode = TESSFACE_IS_QUAD; /* EXCEPTION */                        \
2539                 (void)0
2540
2541
2542                 else if (mp->totloop == 3) {
2543                         ML_TO_MF(0, 1, 2);
2544                         mface_index++;
2545                 }
2546                 else if (mp->totloop == 4) {
2547 #ifdef USE_TESSFACE_QUADS
2548                         ML_TO_MF_QUAD();
2549                         mface_index++;
2550 #else
2551                         ML_TO_MF(0, 1, 2);
2552                         mface_index++;
2553                         ML_TO_MF(0, 2, 3);
2554                         mface_index++;
2555 #endif
2556                 }
2557 #endif /* USE_TESSFACE_SPEEDUP */
2558                 else {
2559                         int totfilltri;
2560
2561                         ml = mloop + mp->loopstart;
2562                         
2563                         BLI_scanfill_begin(&sf_ctx);
2564                         sf_vert_first = NULL;
2565                         sf_vert_last = NULL;
2566                         for (j = 0; j < mp->totloop; j++, ml++) {
2567                                 sf_vert = BLI_scanfill_vert_add(&sf_ctx, mvert[ml->v].co);
2568         
2569                                 sf_vert->keyindex = mp->loopstart + j;
2570         
2571                                 if (sf_vert_last)
2572                                         BLI_scanfill_edge_add(&sf_ctx, sf_vert_last, sf_vert);
2573         
2574                                 if (!sf_vert_first)
2575                                         sf_vert_first = sf_vert;
2576                                 sf_vert_last = sf_vert;
2577                         }
2578                         BLI_scanfill_edge_add(&sf_ctx, sf_vert_last, sf_vert_first);
2579                         
2580                         totfilltri = BLI_scanfill_calc(&sf_ctx, 0);
2581                         BLI_assert(totfilltri <= mp->totloop - 2);
2582
2583                         for (sf_tri = sf_ctx.fillfacebase.first; sf_tri; sf_tri = sf_tri->next, mf++) {
2584                                 mface_to_poly_map[mface_index] = poly_index;
2585                                 mf = &mface[mface_index];
2586
2587                                 /* set loop indices, transformed to vert indices later */
2588                                 mf->v1 = sf_tri->v1->keyindex;
2589                                 mf->v2 = sf_tri->v2->keyindex;
2590                                 mf->v3 = sf_tri->v3->keyindex;
2591                                 mf->v4 = 0;
2592
2593                                 mf->mat_nr = mp->mat_nr;
2594                                 mf->flag = mp->flag;
2595
2596 #ifdef USE_TESSFACE_SPEEDUP
2597                                 mf->edcode = TESSFACE_SCANFILL; /* tag for sorting loop indices */
2598 #endif
2599
2600                                 mface_index++;
2601                         }
2602         
2603                         BLI_scanfill_end(&sf_ctx);
2604                 }
2605         }
2606
2607         CustomData_free(fdata, totface);
2608         totface = mface_index;
2609
2610         BLI_assert(totface <= looptris_tot);
2611
2612         /* not essential but without this we store over-alloc'd memory in the CustomData layers */
2613         if (LIKELY(looptris_tot != totface)) {
2614                 mface = MEM_reallocN(mface, sizeof(*mface) * totface);
2615                 mface_to_poly_map = MEM_reallocN(mface_to_poly_map, sizeof(*mface_to_poly_map) * totface);
2616         }
2617
2618         CustomData_add_layer(fdata, CD_MFACE, CD_ASSIGN, mface, totface);
2619
2620         /* CD_ORIGINDEX will contain an array of indices from tessfaces to the polygons
2621          * they are directly tessellated from */
2622         CustomData_add_layer(fdata, CD_ORIGINDEX, CD_ASSIGN, mface_to_poly_map, totface);
2623         CustomData_from_bmeshpoly(fdata, pdata, ldata, totface);
2624
2625         if (do_face_nor_cpy) {
2626                 /* If polys have a normals layer, copying that to faces can help
2627                  * avoid the need to recalculate normals later */
2628                 if (CustomData_has_layer(pdata, CD_NORMAL)) {
2629                         float (*pnors)[3] = CustomData_get_layer(pdata, CD_NORMAL);
2630                         float (*fnors)[3] = CustomData_add_layer(fdata, CD_NORMAL, CD_CALLOC, NULL, totface);
2631                         for (mface_index = 0; mface_index < totface; mface_index++) {
2632                                 copy_v3_v3(fnors[mface_index], pnors[mface_to_poly_map[mface_index]]);
2633                         }
2634                 }
2635         }
2636
2637         mf = mface;
2638         for (mface_index = 0; mface_index < totface; mface_index++, mf++) {
2639
2640 #ifdef USE_TESSFACE_QUADS
2641                 const int mf_len = mf->edcode & TESSFACE_IS_QUAD ? 4 : 3;
2642 #endif
2643
2644 #ifdef USE_TESSFACE_SPEEDUP
2645                 /* skip sorting when not using ngons */
2646                 if (UNLIKELY(mf->edcode & TESSFACE_SCANFILL))
2647 #endif
2648                 {
2649                         /* sort loop indices to ensure winding is correct */
2650                         if (mf->v1 > mf->v2) SWAP(unsigned int, mf->v1, mf->v2);
2651                         if (mf->v2 > mf->v3) SWAP(unsigned int, mf->v2, mf->v3);
2652                         if (mf->v1 > mf->v2) SWAP(unsigned int, mf->v1, mf->v2);
2653
2654                         if (mf->v1 > mf->v2) SWAP(unsigned int, mf->v1, mf->v2);
2655                         if (mf->v2 > mf->v3) SWAP(unsigned int, mf->v2, mf->v3);
2656                         if (mf->v1 > mf->v2) SWAP(unsigned int, mf->v1, mf->v2);
2657                 }
2658
2659                 /* end abusing the edcode */
2660 #if defined(USE_TESSFACE_QUADS) || defined(USE_TESSFACE_SPEEDUP)
2661                 mf->edcode = 0;
2662 #endif
2663
2664
2665                 lindex[0] = mf->v1;
2666                 lindex[1] = mf->v2;
2667                 lindex[2] = mf->v3;
2668 #ifdef USE_TESSFACE_QUADS
2669                 if (mf_len == 4) lindex[3] = mf->v4;
2670 #endif
2671
2672                 /*transform loop indices to vert indices*/
2673                 mf->v1 = mloop[mf->v1].v;
2674                 mf->v2 = mloop[mf->v2].v;
2675                 mf->v3 = mloop[mf->v3].v;
2676 #ifdef USE_TESSFACE_QUADS
2677                 if (mf_len == 4) mf->v4 = mloop[mf->v4].v;
2678 #endif
2679
2680                 BKE_mesh_loops_to_mface_corners(fdata, ldata, pdata,
2681                                                 lindex, mface_index, mface_to_poly_map[mface_index],
2682 #ifdef USE_TESSFACE_QUADS
2683                                                 mf_len,
2684 #else
2685                                                 3,
2686 #endif
2687                                                 numTex, numCol, hasPCol, hasOrigSpace);
2688
2689
2690 #ifdef USE_TESSFACE_QUADS
2691                 test_index_face(mf, fdata, mface_index, mf_len);
2692 #endif
2693
2694         }
2695
2696         return totface;
2697
2698 #undef USE_TESSFACE_SPEEDUP
2699
2700 }
2701
2702
2703 #ifdef USE_BMESH_SAVE_AS_COMPAT
2704
2705 /*
2706  * this function recreates a tessellation.
2707  * returns number of tessellation faces.
2708  */
2709 int BKE_mesh_mpoly_to_mface(struct CustomData *fdata, struct CustomData *ldata,
2710                             struct CustomData *pdata, int totface, int UNUSED(totloop), int totpoly)
2711 {
2712         MLoop *mloop;
2713
2714         int lindex[4];
2715         int i;
2716         int k;
2717
2718         MPoly *mp, *mpoly;
2719         MFace *mface = NULL, *mf;
2720         BLI_array_declare(mface);
2721
2722         const int numTex = CustomData_number_of_layers(pdata, CD_MTEXPOLY);
2723         const int numCol = CustomData_number_of_layers(ldata, CD_MLOOPCOL);
2724         const int hasPCol = CustomData_has_layer(ldata, CD_PREVIEW_MLOOPCOL);
2725         const int hasOrigSpace = CustomData_has_layer(ldata, CD_ORIGSPACE_MLOOP);
2726
2727         mpoly = CustomData_get_layer(pdata, CD_MPOLY);
2728         mloop = CustomData_get_layer(ldata, CD_MLOOP);
2729
2730         mp = mpoly;
2731         k = 0;
2732         for (i = 0; i < totpoly; i++, mp++) {
2733                 if (ELEM(mp->totloop, 3, 4)) {
2734                         BLI_array_grow_one(mface);
2735                         mf = &mface[k];
2736
2737                         mf->mat_nr = mp->mat_nr;
2738                         mf->flag = mp->flag;
2739
2740                         mf->v1 = mp->loopstart + 0;
2741                         mf->v2 = mp->loopstart + 1;
2742                         mf->v3 = mp->loopstart + 2;
2743                         mf->v4 = (mp->totloop == 4) ? (mp->loopstart + 3) : 0;
2744
2745                         /* abuse edcode for temp storage and clear next loop */
2746                         mf->edcode = (char)mp->totloop; /* only ever 3 or 4 */
2747
2748                         k++;
2749                 }
2750         }
2751
2752         CustomData_free(fdata, totface);
2753
2754         totface = k;
2755
2756         CustomData_add_layer(fdata, CD_MFACE, CD_ASSIGN, mface, totface);
2757
2758         CustomData_from_bmeshpoly(fdata, pdata, ldata, totface);
2759
2760         mp = mpoly;
2761         k = 0;
2762         for (i = 0; i < totpoly; i++, mp++) {
2763                 if (ELEM(mp->totloop, 3, 4)) {
2764                         mf = &mface[k];
2765
2766                         if (mf->edcode == 3) {
2767                                 /* sort loop indices to ensure winding is correct */
2768                                 /* NO SORT - looks like we can skip this */
2769
2770                                 lindex[0] = mf->v1;
2771                                 lindex[1] = mf->v2;
2772                                 lindex[2] = mf->v3;
2773                                 lindex[3] = 0; /* unused */
2774
2775                                 /* transform loop indices to vert indices */
2776                                 mf->v1 = mloop[mf->v1].v;
2777                                 mf->v2 = mloop[mf->v2].v;
2778                                 mf->v3 = mloop[mf->v3].v;
2779
2780                                 BKE_mesh_loops_to_mface_corners(fdata, ldata, pdata,
2781                                                                 lindex, k, i, 3,
2782                                                                 numTex, numCol, hasPCol, hasOrigSpace);
2783                                 test_index_face(mf, fdata, k, 3);
2784                         }
2785                         else {
2786                                 /* sort loop indices to ensure winding is correct */
2787                                 /* NO SORT - looks like we can skip this */
2788
2789                                 lindex[0] = mf->v1;
2790                                 lindex[1] = mf->v2;
2791                                 lindex[2] = mf->v3;
2792                                 lindex[3] = mf->v4;
2793
2794                                 /* transform loop indices to vert indices */
2795                                 mf->v1 = mloop[mf->v1].v;
2796                                 mf->v2 = mloop[mf->v2].v;
2797                                 mf->v3 = mloop[mf->v3].v;
2798                                 mf->v4 = mloop[mf->v4].v;
2799
2800                                 BKE_mesh_loops_to_mface_corners(fdata, ldata, pdata,
2801                                                                 lindex, k, i, 4,
2802                                                                 numTex, numCol, hasPCol, hasOrigSpace);
2803                                 test_index_face(mf, fdata, k, 4);
2804                         }
2805
2806                         mf->edcode = 0;
2807
2808                         k++;
2809                 }
2810         }
2811
2812         return k;
2813 }
2814 #endif /* USE_BMESH_SAVE_AS_COMPAT */
2815
2816 /*
2817  * COMPUTE POLY NORMAL
2818  *
2819  * Computes the normal of a planar 
2820  * polygon See Graphics Gems for 
2821  * computing newell normal.
2822  *
2823  */
2824 static void mesh_calc_ngon_normal(MPoly *mpoly, MLoop *loopstart, 
2825                                   MVert *mvert, float normal[3])
2826 {
2827         const int nverts = mpoly->totloop;
2828         float const *v_prev = mvert[loopstart[nverts - 1].v].co;
2829         float const *v_curr;
2830         int i;
2831
2832         zero_v3(normal);
2833
2834         /* Newell's Method */
2835         for (i = 0; i < nverts; i++) {
2836                 v_curr = mvert[loopstart[i].v].co;
2837                 add_newell_cross_v3_v3v3(normal, v_prev, v_curr);
2838                 v_prev = v_curr;
2839         }
2840
2841         if (UNLIKELY(normalize_v3(normal) == 0.0f)) {
2842                 normal[2] = 1.0f; /* other axis set to 0.0 */
2843         }
2844 }
2845
2846 void BKE_mesh_calc_poly_normal(MPoly *mpoly, MLoop *loopstart,
2847                                MVert *mvarray, float no[3])
2848 {
2849         if (mpoly->totloop > 4) {
2850                 mesh_calc_ngon_normal(mpoly, loopstart, mvarray, no);
2851         }
2852         else if (mpoly->totloop == 3) {
2853                 normal_tri_v3(no,
2854                               mvarray[loopstart[0].v].co,
2855                               mvarray[loopstart[1].v].co,
2856                               mvarray[loopstart[2].v].co
2857                               );
2858         }
2859         else if (mpoly->totloop == 4) {
2860                 normal_quad_v3(no,
2861                                mvarray[loopstart[0].v].co,
2862                                mvarray[loopstart[1].v].co,
2863                                mvarray[loopstart[2].v].co,
2864                                mvarray[loopstart[3].v].co
2865                                );
2866         }
2867         else { /* horrible, two sided face! */
2868                 no[0] = 0.0;
2869                 no[1] = 0.0;
2870                 no[2] = 1.0;
2871         }
2872 }
2873 /* duplicate of function above _but_ takes coords rather then mverts */
2874 static void mesh_calc_ngon_normal_coords(MPoly *mpoly, MLoop *loopstart,
2875                                          const float (*vertex_coords)[3], float normal[3])
2876 {
2877         const int nverts = mpoly->totloop;
2878         float const *v_prev = vertex_coords[loopstart[nverts - 1].v];
2879         float const *v_curr;
2880         int i;
2881
2882         zero_v3(normal);
2883
2884         /* Newell's Method */
2885         for (i = 0; i < nverts; i++) {
2886                 v_curr = vertex_coords[loopstart[i].v];
2887                 add_newell_cross_v3_v3v3(normal, v_prev, v_curr);
2888                 v_prev = v_curr;
2889         }
2890
2891         if (UNLIKELY(normalize_v3(normal) == 0.0f)) {
2892                 normal[2] = 1.0f; /* other axis set to 0.0 */
2893         }
2894 }
2895
2896 void BKE_mesh_calc_poly_normal_coords(MPoly *mpoly, MLoop *loopstart,
2897                                       const float (*vertex_coords)[3], float no[3])
2898 {
2899         if (mpoly->totloop > 4) {
2900                 mesh_calc_ngon_normal_coords(mpoly, loopstart, vertex_coords, no);
2901         }
2902         else if (mpoly->totloop == 3) {
2903                 normal_tri_v3(no,
2904                               vertex_coords[loopstart[0].v],
2905                               vertex_coords[loopstart[1].v],
2906                               vertex_coords[loopstart[2].v]
2907                               );
2908         }
2909         else if (mpoly->totloop == 4) {
2910                 normal_quad_v3(no,
2911                                vertex_coords[loopstart[0].v],
2912                                vertex_coords[loopstart[1].v],
2913                                vertex_coords[loopstart[2].v],
2914                                vertex_coords[loopstart[3].v]
2915                                );
2916         }
2917         else { /* horrible, two sided face! */
2918                 no[0] = 0.0;
2919                 no[1] = 0.0;
2920                 no[2] = 1.0;
2921         }
2922 }
2923
2924 static void mesh_calc_ngon_center(MPoly *mpoly, MLoop *loopstart,
2925                                   MVert *mvert, float cent[3])
2926 {
2927         const float w = 1.0f / (float)mpoly->totloop;
2928         int i;
2929
2930         zero_v3(cent);
2931
2932         for (i = 0; i < mpoly->totloop; i++) {
2933                 madd_v3_v3fl(cent, mvert[(loopstart++)->v].co, w);
2934         }
2935 }
2936
2937 void BKE_mesh_calc_poly_center(MPoly *mpoly, MLoop *loopstart,
2938                                MVert *mvarray, float cent[3])
2939 {
2940         if (mpoly->totloop == 3) {
2941                 cent_tri_v3(cent,
2942                             mvarray[loopstart[0].v].co,
2943                             mvarray[loopstart[1].v].co,
2944                             mvarray[loopstart[2].v].co
2945                             );
2946         }
2947         else if (mpoly->totloop == 4) {
2948                 cent_quad_v3(cent,
2949                              mvarray[loopstart[0].v].co,
2950                              mvarray[loopstart[1].v].co,
2951                              mvarray[loopstart[2].v].co,
2952                              mvarray[loopstart[3].v].co
2953                              );
2954         }
2955         else {
2956                 mesh_calc_ngon_center(mpoly, loopstart, mvarray, cent);
2957         }
2958 }
2959
2960 /* note, passing polynormal is only a speedup so we can skip calculating it */
2961 float BKE_mesh_calc_poly_area(MPoly *mpoly, MLoop *loopstart,
2962                               MVert *mvarray, const float polynormal[3])
2963 {
2964         if (mpoly->totloop == 3) {
2965                 return area_tri_v3(mvarray[loopstart[0].v].co,
2966                                    mvarray[loopstart[1].v].co,
2967                                    mvarray[loopstart[2].v].co
2968                                    );
2969         }
2970         else if (mpoly->totloop == 4) {
2971                 return area_quad_v3(mvarray[loopstart[0].v].co,
2972                                     mvarray[loopstart[1].v].co,
2973                                     mvarray[loopstart[2].v].co,
2974                                     mvarray[loopstart[3].v].co
2975                                     );
2976         }
2977         else {
2978                 int i;
2979                 MLoop *l_iter = loopstart;
2980                 float area, polynorm_local[3];
2981                 float (*vertexcos)[3] = BLI_array_alloca(vertexcos, mpoly->totloop);
2982                 const float *no = polynormal ? polynormal : polynorm_local;
2983
2984                 /* pack vertex cos into an array for area_poly_v3 */
2985                 for (i = 0; i < mpoly->totloop; i++, l_iter++) {
2986                         copy_v3_v3(vertexcos[i], mvarray[l_iter->v].co);
2987                 }
2988
2989                 /* need normal for area_poly_v3 as well */
2990                 if (polynormal == NULL) {
2991                         BKE_mesh_calc_poly_normal(mpoly, loopstart, mvarray, polynorm_local);
2992                 }
2993
2994                 /* finally calculate the area */
2995                 area = area_poly_v3(mpoly->totloop, vertexcos, no);
2996
2997                 return area;
2998         }
2999 }
3000
3001 /**
3002  * This function takes the difference between 2 vertex-coord-arrays
3003  * (\a vert_cos_src, \a vert_cos_dst),
3004  * and applies the difference to \a vert_cos_new relative to \a vert_cos_org.
3005  *
3006  * \param vert_cos_src reference deform source.
3007  * \param vert_cos_dst reference deform destination.
3008  *
3009  * \param vert_cos_org reference for the output location.
3010  * \param vert_cos_new resulting coords.
3011  */
3012 void BKE_mesh_calc_relative_deform(
3013         const MPoly *mpoly, const int totpoly,
3014         const MLoop *mloop, const int totvert,
3015
3016         const float (*vert_cos_src)[3],
3017         const float (*vert_cos_dst)[3],
3018
3019         const float (*vert_cos_org)[3],
3020               float (*vert_cos_new)[3])
3021 {
3022         const MPoly *mp;
3023         int i;
3024
3025         int *vert_accum = MEM_callocN(sizeof(*vert_accum) * totvert, __func__);
3026
3027         memset(vert_cos_new, '\0', sizeof(*vert_cos_new) * totvert);
3028
3029         for (i = 0, mp = mpoly; i < totpoly; i++, mp++) {
3030                 const MLoop *loopstart = mloop + mp->loopstart;
3031                 int j;
3032
3033                 for (j = 0; j < mp->totloop; j++) {
3034                         int v_prev = (loopstart + ((mp->totloop + (j - 1)) % mp->totloop))->v;
3035                         int v_curr = (loopstart + j)->v;
3036                         int v_next = (loopstart + ((j + 1) % mp->totloop))->v;
3037
3038                         float tvec[3];
3039
3040                         barycentric_transform(
3041                                     tvec, vert_cos_dst[v_curr],
3042                                     vert_cos_org[v_prev], vert_cos_org[v_curr], vert_cos_org[v_next],
3043                                     vert_cos_src[v_prev], vert_cos_src[v_curr], vert_cos_src[v_next]
3044                                     );
3045
3046                         add_v3_v3(vert_cos_new[v_curr], tvec);
3047                         vert_accum[v_curr] += 1;
3048                 }
3049         }
3050
3051         for (i = 0; i < totvert; i++) {
3052                 if (vert_accum[i]) {
3053                         mul_v3_fl(vert_cos_new[i], 1.0f / (float)vert_accum[i]);
3054                 }
3055                 else {
3056                         copy_v3_v3(vert_cos_new[i], vert_cos_org[i]);
3057                 }
3058         }
3059
3060         MEM_freeN(vert_accum);
3061 }
3062
3063 /* Find the index of the loop in 'poly' which references vertex,
3064  * returns -1 if not found */
3065 int poly_find_loop_from_vert(const MPoly *poly, const MLoop *loopstart,
3066                              unsigned vert)
3067 {
3068         int j;
3069         for (j = 0; j < poly->totloop; j++, loopstart++) {
3070                 if (loopstart->v == vert)
3071                         return j;
3072         }
3073         
3074         return -1;
3075 }
3076
3077 /* Fill 'adj_r' with the loop indices in 'poly' adjacent to the
3078  * vertex. Returns the index of the loop matching vertex, or -1 if the
3079  * vertex is not in 'poly' */
3080 int poly_get_adj_loops_from_vert(unsigned adj_r[3], const MPoly *poly,
3081                                  const MLoop *mloop, unsigned vert)
3082 {
3083         int corner = poly_find_loop_from_vert(poly,
3084                                               &mloop[poly->loopstart],
3085                                               vert);
3086                 
3087         if (corner != -1) {
3088                 const MLoop *ml = &mloop[poly->loopstart + corner];
3089
3090                 /* vertex was found */
3091                 adj_r[0] = ME_POLY_LOOP_PREV(mloop, poly, corner)->v;
3092                 adj_r[1] = ml->v;
3093                 adj_r[2] = ME_POLY_LOOP_NEXT(mloop, poly, corner)->v;
3094         }
3095
3096         return corner;
3097 }
3098
3099 /* Return the index of the edge vert that is not equal to 'v'. If
3100  * neither edge vertex is equal to 'v', returns -1. */
3101 int BKE_mesh_edge_other_vert(const MEdge *e, int v)
3102 {
3103         if (e->v1 == v)
3104                 return e->v2;
3105         else if (e->v2 == v)
3106                 return e->v1;
3107         else
3108                 return -1;
3109 }
3110
3111 /* update the hide flag for edges and faces from the corresponding
3112  * flag in verts */
3113 void BKE_mesh_flush_hidden_from_verts(const MVert *mvert,
3114                             &nbs