Fixed memory access in mesh_calc_normals(). Now it uses much less memory.
[blender.git] / source / blender / blenkernel / intern / mesh.c
1
2 /*  mesh.c
3  *
4  *  
5  * 
6  * $Id$
7  *
8  * ***** BEGIN GPL LICENSE BLOCK *****
9  *
10  * This program is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU General Public License
12  * as published by the Free Software Foundation; either version 2
13  * of the License, or (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software Foundation,
22  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
23  *
24  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
25  * All rights reserved.
26  *
27  * Contributor(s): Blender Foundation
28  *
29  * ***** END GPL LICENSE BLOCK *****
30  */
31
32 /** \file blender/blenkernel/intern/mesh.c
33  *  \ingroup bke
34  */
35
36
37 #include <stdlib.h>
38 #include <string.h>
39 #include <stdio.h>
40 #include <math.h>
41
42 #include "MEM_guardedalloc.h"
43
44 #include "DNA_scene_types.h"
45 #include "DNA_material_types.h"
46 #include "DNA_object_types.h"
47 #include "DNA_key_types.h"
48 #include "DNA_meshdata_types.h"
49 #include "DNA_ipo_types.h"
50
51 #include "BLI_blenlib.h"
52 #include "BLI_editVert.h"
53 #include "BLI_math.h"
54 #include "BLI_edgehash.h"
55 #include "BLI_utildefines.h"
56
57 #include "BKE_animsys.h"
58 #include "BKE_main.h"
59 #include "BKE_DerivedMesh.h"
60 #include "BKE_global.h"
61 #include "BKE_mesh.h"
62 #include "BKE_displist.h"
63 #include "BKE_library.h"
64 #include "BKE_material.h"
65 #include "BKE_modifier.h"
66 #include "BKE_multires.h"
67 #include "BKE_key.h"
68 /* these 2 are only used by conversion functions */
69 #include "BKE_curve.h"
70 /* -- */
71 #include "BKE_object.h"
72
73
74 EditMesh *BKE_mesh_get_editmesh(Mesh *me)
75 {
76         return me->edit_mesh;
77 }
78
79 void BKE_mesh_end_editmesh(Mesh *UNUSED(me), EditMesh *UNUSED(em))
80 {
81 }
82
83
84 void mesh_update_customdata_pointers(Mesh *me)
85 {
86         me->mvert = CustomData_get_layer(&me->vdata, CD_MVERT);
87         me->dvert = CustomData_get_layer(&me->vdata, CD_MDEFORMVERT);
88         me->msticky = CustomData_get_layer(&me->vdata, CD_MSTICKY);
89
90         me->medge = CustomData_get_layer(&me->edata, CD_MEDGE);
91
92         me->mface = CustomData_get_layer(&me->fdata, CD_MFACE);
93         me->mcol = CustomData_get_layer(&me->fdata, CD_MCOL);
94         me->mtface = CustomData_get_layer(&me->fdata, CD_MTFACE);
95 }
96
97 /* Note: unlinking is called when me->id.us is 0, question remains how
98  * much unlinking of Library data in Mesh should be done... probably
99  * we need a more generic method, like the expand() functions in
100  * readfile.c */
101
102 void unlink_mesh(Mesh *me)
103 {
104         int a;
105         
106         if(me==NULL) return;
107         
108         for(a=0; a<me->totcol; a++) {
109                 if(me->mat[a]) me->mat[a]->id.us--;
110                 me->mat[a]= NULL;
111         }
112
113         if(me->key) {
114                    me->key->id.us--;
115                 if (me->key->id.us == 0 && me->key->ipo )
116                         me->key->ipo->id.us--;
117         }
118         me->key= NULL;
119         
120         if(me->texcomesh) me->texcomesh= NULL;
121 }
122
123
124 /* do not free mesh itself */
125 void free_mesh(Mesh *me)
126 {
127         unlink_mesh(me);
128
129         if(me->pv) {
130                 if(me->pv->vert_map) MEM_freeN(me->pv->vert_map);
131                 if(me->pv->edge_map) MEM_freeN(me->pv->edge_map);
132                 if(me->pv->old_faces) MEM_freeN(me->pv->old_faces);
133                 if(me->pv->old_edges) MEM_freeN(me->pv->old_edges);
134                 me->totvert= me->pv->totvert;
135                 me->totedge= me->pv->totedge;
136                 me->totface= me->pv->totface;
137                 MEM_freeN(me->pv);
138         }
139
140         CustomData_free(&me->vdata, me->totvert);
141         CustomData_free(&me->edata, me->totedge);
142         CustomData_free(&me->fdata, me->totface);
143         
144         if(me->adt) {
145                 BKE_free_animdata(&me->id);
146                 me->adt= NULL;
147         }
148         
149         if(me->mat) MEM_freeN(me->mat);
150         
151         if(me->bb) MEM_freeN(me->bb);
152         if(me->mselect) MEM_freeN(me->mselect);
153         if(me->edit_mesh) MEM_freeN(me->edit_mesh);
154 }
155
156 void copy_dverts(MDeformVert *dst, MDeformVert *src, int copycount)
157 {
158         /* Assumes dst is already set up */
159         int i;
160
161         if (!src || !dst)
162                 return;
163
164         memcpy (dst, src, copycount * sizeof(MDeformVert));
165         
166         for (i=0; i<copycount; i++){
167                 if (src[i].dw){
168                         dst[i].dw = MEM_callocN (sizeof(MDeformWeight)*src[i].totweight, "copy_deformWeight");
169                         memcpy (dst[i].dw, src[i].dw, sizeof (MDeformWeight)*src[i].totweight);
170                 }
171         }
172
173 }
174
175 void free_dverts(MDeformVert *dvert, int totvert)
176 {
177         /* Instead of freeing the verts directly,
178         call this function to delete any special
179         vert data */
180         int     i;
181
182         if (!dvert)
183                 return;
184
185         /* Free any special data from the verts */
186         for (i=0; i<totvert; i++){
187                 if (dvert[i].dw) MEM_freeN (dvert[i].dw);
188         }
189         MEM_freeN (dvert);
190 }
191
192 Mesh *add_mesh(const char *name)
193 {
194         Mesh *me;
195         
196         me= alloc_libblock(&G.main->mesh, ID_ME, name);
197         
198         me->size[0]= me->size[1]= me->size[2]= 1.0;
199         me->smoothresh= 30;
200         me->texflag= AUTOSPACE;
201         me->flag= ME_TWOSIDED;
202         me->bb= unit_boundbox();
203         me->drawflag= ME_DRAWEDGES|ME_DRAWFACES|ME_DRAWCREASES;
204         
205         return me;
206 }
207
208 Mesh *copy_mesh(Mesh *me)
209 {
210         Mesh *men;
211         MTFace *tface;
212         int a, i;
213         
214         men= copy_libblock(me);
215         
216         men->mat= MEM_dupallocN(me->mat);
217         for(a=0; a<men->totcol; a++) {
218                 id_us_plus((ID *)men->mat[a]);
219         }
220         id_us_plus((ID *)men->texcomesh);
221
222         CustomData_copy(&me->vdata, &men->vdata, CD_MASK_MESH, CD_DUPLICATE, men->totvert);
223         CustomData_copy(&me->edata, &men->edata, CD_MASK_MESH, CD_DUPLICATE, men->totedge);
224         CustomData_copy(&me->fdata, &men->fdata, CD_MASK_MESH, CD_DUPLICATE, men->totface);
225         mesh_update_customdata_pointers(men);
226
227         /* ensure indirect linked data becomes lib-extern */
228         for(i=0; i<me->fdata.totlayer; i++) {
229                 if(me->fdata.layers[i].type == CD_MTFACE) {
230                         tface= (MTFace*)me->fdata.layers[i].data;
231
232                         for(a=0; a<me->totface; a++, tface++)
233                                 if(tface->tpage)
234                                         id_lib_extern((ID*)tface->tpage);
235                 }
236         }
237         
238         men->mselect= NULL;
239
240         men->bb= MEM_dupallocN(men->bb);
241         
242         men->key= copy_key(me->key);
243         if(men->key) men->key->from= (ID *)men;
244
245         return men;
246 }
247
248 void make_local_tface(Mesh *me)
249 {
250         MTFace *tface;
251         Image *ima;
252         int a, i;
253         
254         for(i=0; i<me->fdata.totlayer; i++) {
255                 if(me->fdata.layers[i].type == CD_MTFACE) {
256                         tface= (MTFace*)me->fdata.layers[i].data;
257                         
258                         for(a=0; a<me->totface; a++, tface++) {
259                                 /* special case: ima always local immediately */
260                                 if(tface->tpage) {
261                                         ima= tface->tpage;
262                                         if(ima->id.lib) {
263                                                 ima->id.lib= NULL;
264                                                 ima->id.flag= LIB_LOCAL;
265                                                 new_id(NULL, (ID *)ima, NULL);
266                                         }
267                                 }
268                         }
269                 }
270         }
271 }
272
273 void make_local_mesh(Mesh *me)
274 {
275         Main *bmain= G.main;
276         Object *ob;
277         Mesh *men;
278         int local=0, lib=0;
279
280         /* - only lib users: do nothing
281                 * - only local users: set flag
282                 * - mixed: make copy
283                 */
284         
285         if(me->id.lib==NULL) return;
286         if(me->id.us==1) {
287                 me->id.lib= NULL;
288                 me->id.flag= LIB_LOCAL;
289                 new_id(NULL, (ID *)me, NULL);
290                 
291                 if(me->mtface) make_local_tface(me);
292                 
293                 return;
294         }
295         
296         ob= bmain->object.first;
297         while(ob) {
298                 if( me==get_mesh(ob) ) {
299                         if(ob->id.lib) lib= 1;
300                         else local= 1;
301                 }
302                 ob= ob->id.next;
303         }
304         
305         if(local && lib==0) {
306                 me->id.lib= NULL;
307                 me->id.flag= LIB_LOCAL;
308                 new_id(NULL, (ID *)me, NULL);
309                 
310                 if(me->mtface) make_local_tface(me);
311                 
312         }
313         else if(local && lib) {
314                 men= copy_mesh(me);
315                 men->id.us= 0;
316                 
317                 ob= bmain->object.first;
318                 while(ob) {
319                         if( me==get_mesh(ob) ) {                                
320                                 if(ob->id.lib==NULL) {
321                                         set_mesh(ob, men);
322                                 }
323                         }
324                         ob= ob->id.next;
325                 }
326         }
327 }
328
329 void boundbox_mesh(Mesh *me, float *loc, float *size)
330 {
331         BoundBox *bb;
332         float min[3], max[3];
333         float mloc[3], msize[3];
334         
335         if(me->bb==NULL) me->bb= MEM_callocN(sizeof(BoundBox), "boundbox");
336         bb= me->bb;
337
338         if (!loc) loc= mloc;
339         if (!size) size= msize;
340         
341         INIT_MINMAX(min, max);
342         if(!minmax_mesh(me, min, max)) {
343                 min[0] = min[1] = min[2] = -1.0f;
344                 max[0] = max[1] = max[2] = 1.0f;
345         }
346
347         mid_v3_v3v3(loc, min, max);
348                 
349         size[0]= (max[0]-min[0])/2.0f;
350         size[1]= (max[1]-min[1])/2.0f;
351         size[2]= (max[2]-min[2])/2.0f;
352         
353         boundbox_set_from_min_max(bb, min, max);
354 }
355
356 void tex_space_mesh(Mesh *me)
357 {
358         float loc[3], size[3];
359         int a;
360
361         boundbox_mesh(me, loc, size);
362
363         if(me->texflag & AUTOSPACE) {
364                 for (a=0; a<3; a++) {
365                         if(size[a]==0.0) size[a]= 1.0;
366                         else if(size[a]>0.0 && size[a]<0.00001) size[a]= 0.00001;
367                         else if(size[a]<0.0 && size[a]> -0.00001) size[a]= -0.00001;
368                 }
369
370                 copy_v3_v3(me->loc, loc);
371                 copy_v3_v3(me->size, size);
372                 zero_v3(me->rot);
373         }
374 }
375
376 BoundBox *mesh_get_bb(Object *ob)
377 {
378         Mesh *me= ob->data;
379
380         if(ob->bb)
381                 return ob->bb;
382
383         if (!me->bb)
384                 tex_space_mesh(me);
385
386         return me->bb;
387 }
388
389 void mesh_get_texspace(Mesh *me, float *loc_r, float *rot_r, float *size_r)
390 {
391         if (!me->bb) {
392                 tex_space_mesh(me);
393         }
394
395         if (loc_r) VECCOPY(loc_r, me->loc);
396         if (rot_r) VECCOPY(rot_r, me->rot);
397         if (size_r) VECCOPY(size_r, me->size);
398 }
399
400 float *get_mesh_orco_verts(Object *ob)
401 {
402         Mesh *me = ob->data;
403         MVert *mvert = NULL;
404         Mesh *tme = me->texcomesh?me->texcomesh:me;
405         int a, totvert;
406         float (*vcos)[3] = NULL;
407
408         /* Get appropriate vertex coordinates */
409         vcos = MEM_callocN(sizeof(*vcos)*me->totvert, "orco mesh");
410         mvert = tme->mvert;
411         totvert = MIN2(tme->totvert, me->totvert);
412
413         for(a=0; a<totvert; a++, mvert++) {
414                 copy_v3_v3(vcos[a], mvert->co);
415         }
416
417         return (float*)vcos;
418 }
419
420 void transform_mesh_orco_verts(Mesh *me, float (*orco)[3], int totvert, int invert)
421 {
422         float loc[3], size[3];
423         int a;
424
425         mesh_get_texspace(me->texcomesh?me->texcomesh:me, loc, NULL, size);
426
427         if(invert) {
428                 for(a=0; a<totvert; a++) {
429                         float *co = orco[a];
430                         madd_v3_v3v3v3(co, loc, co, size);
431                 }
432         }
433         else {
434                 for(a=0; a<totvert; a++) {
435                         float *co = orco[a];
436                         co[0] = (co[0]-loc[0])/size[0];
437                         co[1] = (co[1]-loc[1])/size[1];
438                         co[2] = (co[2]-loc[2])/size[2];
439                 }
440         }
441 }
442
443 /* rotates the vertices of a face in case v[2] or v[3] (vertex index) is = 0.
444    this is necessary to make the if(mface->v4) check for quads work */
445 int test_index_face(MFace *mface, CustomData *fdata, int mfindex, int nr)
446 {
447         /* first test if the face is legal */
448         if((mface->v3 || nr==4) && mface->v3==mface->v4) {
449                 mface->v4= 0;
450                 nr--;
451         }
452         if((mface->v2 || mface->v4) && mface->v2==mface->v3) {
453                 mface->v3= mface->v4;
454                 mface->v4= 0;
455                 nr--;
456         }
457         if(mface->v1==mface->v2) {
458                 mface->v2= mface->v3;
459                 mface->v3= mface->v4;
460                 mface->v4= 0;
461                 nr--;
462         }
463
464         /* check corrupt cases, bowtie geometry, cant handle these because edge data wont exist so just return 0 */
465         if(nr==3) {
466                 if(
467                 /* real edges */
468                         mface->v1==mface->v2 ||
469                         mface->v2==mface->v3 ||
470                         mface->v3==mface->v1
471                 ) {
472                         return 0;
473                 }
474         }
475         else if(nr==4) {
476                 if(
477                 /* real edges */
478                         mface->v1==mface->v2 ||
479                         mface->v2==mface->v3 ||
480                         mface->v3==mface->v4 ||
481                         mface->v4==mface->v1 ||
482                 /* across the face */
483                         mface->v1==mface->v3 ||
484                         mface->v2==mface->v4
485                 ) {
486                         return 0;
487                 }
488         }
489
490         /* prevent a zero at wrong index location */
491         if(nr==3) {
492                 if(mface->v3==0) {
493                         static int corner_indices[4] = {1, 2, 0, 3};
494
495                         SWAP(int, mface->v1, mface->v2);
496                         SWAP(int, mface->v2, mface->v3);
497
498                         if(fdata)
499                                 CustomData_swap(fdata, mfindex, corner_indices);
500                 }
501         }
502         else if(nr==4) {
503                 if(mface->v3==0 || mface->v4==0) {
504                         static int corner_indices[4] = {2, 3, 0, 1};
505
506                         SWAP(int, mface->v1, mface->v3);
507                         SWAP(int, mface->v2, mface->v4);
508
509                         if(fdata)
510                                 CustomData_swap(fdata, mfindex, corner_indices);
511                 }
512         }
513
514         return nr;
515 }
516
517 Mesh *get_mesh(Object *ob)
518 {
519         
520         if(ob==NULL) return NULL;
521         if(ob->type==OB_MESH) return ob->data;
522         else return NULL;
523 }
524
525 void set_mesh(Object *ob, Mesh *me)
526 {
527         Mesh *old=NULL;
528
529         multires_force_update(ob);
530         
531         if(ob==NULL) return;
532         
533         if(ob->type==OB_MESH) {
534                 old= ob->data;
535                 if (old)
536                         old->id.us--;
537                 ob->data= me;
538                 id_us_plus((ID *)me);
539         }
540         
541         test_object_materials((ID *)me);
542
543         test_object_modifiers(ob);
544 }
545
546 /* ************** make edges in a Mesh, for outside of editmode */
547
548 struct edgesort {
549         int v1, v2;
550         short is_loose, is_draw;
551 };
552
553 /* edges have to be added with lowest index first for sorting */
554 static void to_edgesort(struct edgesort *ed, int v1, int v2, short is_loose, short is_draw)
555 {
556         if(v1<v2) {
557                 ed->v1= v1; ed->v2= v2;
558         }
559         else {
560                 ed->v1= v2; ed->v2= v1;
561         }
562         ed->is_loose= is_loose;
563         ed->is_draw= is_draw;
564 }
565
566 static int vergedgesort(const void *v1, const void *v2)
567 {
568         const struct edgesort *x1=v1, *x2=v2;
569
570         if( x1->v1 > x2->v1) return 1;
571         else if( x1->v1 < x2->v1) return -1;
572         else if( x1->v2 > x2->v2) return 1;
573         else if( x1->v2 < x2->v2) return -1;
574         
575         return 0;
576 }
577
578 static void mfaces_strip_loose(MFace *mface, int *totface)
579 {
580         int a,b;
581
582         for (a=b=0; a<*totface; a++) {
583                 if (mface[a].v3) {
584                         if (a!=b) {
585                                 memcpy(&mface[b],&mface[a],sizeof(mface[b]));
586                         }
587                         b++;
588                 }
589         }
590
591         *totface= b;
592 }
593
594 /* Create edges based on known verts and faces */
595 static void make_edges_mdata(MVert *UNUSED(allvert), MFace *allface, int UNUSED(totvert), int totface,
596         int old, MEdge **alledge, int *_totedge)
597 {
598         MFace *mface;
599         MEdge *medge;
600         struct edgesort *edsort, *ed;
601         int a, totedge=0, final=0;
602
603         /* we put all edges in array, sort them, and detect doubles that way */
604
605         for(a= totface, mface= allface; a>0; a--, mface++) {
606                 if(mface->v4) totedge+=4;
607                 else if(mface->v3) totedge+=3;
608                 else totedge+=1;
609         }
610
611         if(totedge==0) {
612                 /* flag that mesh has edges */
613                 (*alledge)= MEM_callocN(0, "make mesh edges");
614                 (*_totedge) = 0;
615                 return;
616         }
617
618         ed= edsort= MEM_mallocN(totedge*sizeof(struct edgesort), "edgesort");
619
620         for(a= totface, mface= allface; a>0; a--, mface++) {
621                 to_edgesort(ed++, mface->v1, mface->v2, !mface->v3, mface->edcode & ME_V1V2);
622                 if(mface->v4) {
623                         to_edgesort(ed++, mface->v2, mface->v3, 0, mface->edcode & ME_V2V3);
624                         to_edgesort(ed++, mface->v3, mface->v4, 0, mface->edcode & ME_V3V4);
625                         to_edgesort(ed++, mface->v4, mface->v1, 0, mface->edcode & ME_V4V1);
626                 }
627                 else if(mface->v3) {
628                         to_edgesort(ed++, mface->v2, mface->v3, 0, mface->edcode & ME_V2V3);
629                         to_edgesort(ed++, mface->v3, mface->v1, 0, mface->edcode & ME_V3V1);
630                 }
631         }
632
633         qsort(edsort, totedge, sizeof(struct edgesort), vergedgesort);
634
635         /* count final amount */
636         for(a=totedge, ed=edsort; a>1; a--, ed++) {
637                 /* edge is unique when it differs from next edge, or is last */
638                 if(ed->v1 != (ed+1)->v1 || ed->v2 != (ed+1)->v2) final++;
639         }
640         final++;
641
642         (*alledge)= medge= MEM_callocN(sizeof (MEdge) * final, "make_edges mdge");
643         (*_totedge)= final;
644
645         for(a=totedge, ed=edsort; a>1; a--, ed++) {
646                 /* edge is unique when it differs from next edge, or is last */
647                 if(ed->v1 != (ed+1)->v1 || ed->v2 != (ed+1)->v2) {
648                         medge->v1= ed->v1;
649                         medge->v2= ed->v2;
650                         if(old==0 || ed->is_draw) medge->flag= ME_EDGEDRAW|ME_EDGERENDER;
651                         if(ed->is_loose) medge->flag|= ME_LOOSEEDGE;
652
653                         /* order is swapped so extruding this edge as a surface wont flip face normals
654                          * with cyclic curves */
655                         if(ed->v1+1 != ed->v2) {
656                                 SWAP(int, medge->v1, medge->v2);
657                         }
658                         medge++;
659                 }
660                 else {
661                         /* equal edge, we merge the drawflag */
662                         (ed+1)->is_draw |= ed->is_draw;
663                 }
664         }
665         /* last edge */
666         medge->v1= ed->v1;
667         medge->v2= ed->v2;
668         medge->flag= ME_EDGEDRAW;
669         if(ed->is_loose) medge->flag|= ME_LOOSEEDGE;
670         medge->flag |= ME_EDGERENDER;
671
672         MEM_freeN(edsort);
673 }
674
675 void make_edges(Mesh *me, int old)
676 {
677         MEdge *medge;
678         int totedge=0;
679
680         make_edges_mdata(me->mvert, me->mface, me->totvert, me->totface, old, &medge, &totedge);
681         if(totedge==0) {
682                 /* flag that mesh has edges */
683                 me->medge = medge;
684                 me->totedge = 0;
685                 return;
686         }
687
688         medge= CustomData_add_layer(&me->edata, CD_MEDGE, CD_ASSIGN, medge, totedge);
689         me->medge= medge;
690         me->totedge= totedge;
691
692         mesh_strip_loose_faces(me);
693 }
694
695 void mesh_strip_loose_faces(Mesh *me)
696 {
697         int a,b;
698
699         for (a=b=0; a<me->totface; a++) {
700                 if (me->mface[a].v3) {
701                         if (a!=b) {
702                                 memcpy(&me->mface[b],&me->mface[a],sizeof(me->mface[b]));
703                                 CustomData_copy_data(&me->fdata, &me->fdata, a, b, 1);
704                                 CustomData_free_elem(&me->fdata, a, 1);
705                         }
706                         b++;
707                 }
708         }
709         me->totface = b;
710 }
711
712 void mesh_strip_loose_edges(Mesh *me)
713 {
714         int a,b;
715
716         for (a=b=0; a<me->totedge; a++) {
717                 if (me->medge[a].v1!=me->medge[a].v2) {
718                         if (a!=b) {
719                                 memcpy(&me->medge[b],&me->medge[a],sizeof(me->medge[b]));
720                                 CustomData_copy_data(&me->edata, &me->edata, a, b, 1);
721                                 CustomData_free_elem(&me->edata, a, 1);
722                         }
723                         b++;
724                 }
725         }
726         me->totedge = b;
727 }
728
729 void mball_to_mesh(ListBase *lb, Mesh *me)
730 {
731         DispList *dl;
732         MVert *mvert;
733         MFace *mface;
734         float *nors, *verts;
735         int a, *index;
736         
737         dl= lb->first;
738         if(dl==NULL) return;
739
740         if(dl->type==DL_INDEX4) {
741                 me->totvert= dl->nr;
742                 me->totface= dl->parts;
743                 
744                 mvert= CustomData_add_layer(&me->vdata, CD_MVERT, CD_CALLOC, NULL, dl->nr);
745                 mface= CustomData_add_layer(&me->fdata, CD_MFACE, CD_CALLOC, NULL, dl->parts);
746                 me->mvert= mvert;
747                 me->mface= mface;
748
749                 a= dl->nr;
750                 nors= dl->nors;
751                 verts= dl->verts;
752                 while(a--) {
753                         VECCOPY(mvert->co, verts);
754                         mvert->no[0]= (short int)(nors[0]*32767.0);
755                         mvert->no[1]= (short int)(nors[1]*32767.0);
756                         mvert->no[2]= (short int)(nors[2]*32767.0);
757                         mvert++;
758                         nors+= 3;
759                         verts+= 3;
760                 }
761                 
762                 a= dl->parts;
763                 index= dl->index;
764                 while(a--) {
765                         mface->v1= index[0];
766                         mface->v2= index[1];
767                         mface->v3= index[2];
768                         mface->v4= index[3];
769                         mface->flag= ME_SMOOTH;
770
771                         test_index_face(mface, NULL, 0, (mface->v3==mface->v4)? 3: 4);
772
773                         mface++;
774                         index+= 4;
775                 }
776
777                 make_edges(me, 0);      // all edges
778         }       
779 }
780
781 /* Initialize mverts, medges and, faces for converting nurbs to mesh and derived mesh */
782 /* return non-zero on error */
783 int nurbs_to_mdata(Object *ob, MVert **allvert, int *totvert,
784         MEdge **alledge, int *totedge, MFace **allface, int *totface)
785 {
786         return nurbs_to_mdata_customdb(ob, &ob->disp,
787                 allvert, totvert, alledge, totedge, allface, totface);
788 }
789
790 /* Initialize mverts, medges and, faces for converting nurbs to mesh and derived mesh */
791 /* use specified dispbase  */
792 int nurbs_to_mdata_customdb(Object *ob, ListBase *dispbase, MVert **allvert, int *_totvert,
793         MEdge **alledge, int *_totedge, MFace **allface, int *_totface)
794 {
795         DispList *dl;
796         Curve *cu;
797         MVert *mvert;
798         MFace *mface;
799         float *data;
800         int a, b, ofs, vertcount, startvert, totvert=0, totvlak=0;
801         int p1, p2, p3, p4, *index;
802         int conv_polys= 0;
803
804         cu= ob->data;
805
806         conv_polys|= cu->flag & CU_3D;          /* 2d polys are filled with DL_INDEX3 displists */
807         conv_polys|= ob->type == OB_SURF;       /* surf polys are never filled */
808
809         /* count */
810         dl= dispbase->first;
811         while(dl) {
812                 if(dl->type==DL_SEGM) {
813                         totvert+= dl->parts*dl->nr;
814                         totvlak+= dl->parts*(dl->nr-1);
815                 }
816                 else if(dl->type==DL_POLY) {
817                         if(conv_polys) {
818                                 totvert+= dl->parts*dl->nr;
819                                 totvlak+= dl->parts*dl->nr;
820                         }
821                 }
822                 else if(dl->type==DL_SURF) {
823                         totvert+= dl->parts*dl->nr;
824                         totvlak+= (dl->parts-1+((dl->flag & DL_CYCL_V)==2))*(dl->nr-1+(dl->flag & DL_CYCL_U));
825                 }
826                 else if(dl->type==DL_INDEX3) {
827                         totvert+= dl->nr;
828                         totvlak+= dl->parts;
829                 }
830                 dl= dl->next;
831         }
832
833         if(totvert==0) {
834                 /* error("can't convert"); */
835                 /* Make Sure you check ob->data is a curve */
836                 return -1;
837         }
838
839         *allvert= mvert= MEM_callocN(sizeof (MVert) * totvert, "nurbs_init mvert");
840         *allface= mface= MEM_callocN(sizeof (MVert) * totvlak, "nurbs_init mface");
841
842         /* verts and faces */
843         vertcount= 0;
844
845         dl= dispbase->first;
846         while(dl) {
847                 int smooth= dl->rt & CU_SMOOTH ? 1 : 0;
848
849                 if(dl->type==DL_SEGM) {
850                         startvert= vertcount;
851                         a= dl->parts*dl->nr;
852                         data= dl->verts;
853                         while(a--) {
854                                 VECCOPY(mvert->co, data);
855                                 data+=3;
856                                 vertcount++;
857                                 mvert++;
858                         }
859
860                         for(a=0; a<dl->parts; a++) {
861                                 ofs= a*dl->nr;
862                                 for(b=1; b<dl->nr; b++) {
863                                         mface->v1= startvert+ofs+b-1;
864                                         mface->v2= startvert+ofs+b;
865                                         if(smooth) mface->flag |= ME_SMOOTH;
866                                         mface++;
867                                 }
868                         }
869
870                 }
871                 else if(dl->type==DL_POLY) {
872                         if(conv_polys) {
873                                 startvert= vertcount;
874                                 a= dl->parts*dl->nr;
875                                 data= dl->verts;
876                                 while(a--) {
877                                         VECCOPY(mvert->co, data);
878                                         data+=3;
879                                         vertcount++;
880                                         mvert++;
881                                 }
882
883                                 for(a=0; a<dl->parts; a++) {
884                                         ofs= a*dl->nr;
885                                         for(b=0; b<dl->nr; b++) {
886                                                 mface->v1= startvert+ofs+b;
887                                                 if(b==dl->nr-1) mface->v2= startvert+ofs;
888                                                 else mface->v2= startvert+ofs+b+1;
889                                                 if(smooth) mface->flag |= ME_SMOOTH;
890                                                 mface++;
891                                         }
892                                 }
893                         }
894                 }
895                 else if(dl->type==DL_INDEX3) {
896                         startvert= vertcount;
897                         a= dl->nr;
898                         data= dl->verts;
899                         while(a--) {
900                                 VECCOPY(mvert->co, data);
901                                 data+=3;
902                                 vertcount++;
903                                 mvert++;
904                         }
905
906                         a= dl->parts;
907                         index= dl->index;
908                         while(a--) {
909                                 mface->v1= startvert+index[0];
910                                 mface->v2= startvert+index[2];
911                                 mface->v3= startvert+index[1];
912                                 mface->v4= 0;
913                                 mface->mat_nr= (unsigned char)dl->col;
914                                 test_index_face(mface, NULL, 0, 3);
915
916                                 if(smooth) mface->flag |= ME_SMOOTH;
917                                 mface++;
918                                 index+= 3;
919                         }
920
921
922                 }
923                 else if(dl->type==DL_SURF) {
924                         startvert= vertcount;
925                         a= dl->parts*dl->nr;
926                         data= dl->verts;
927                         while(a--) {
928                                 VECCOPY(mvert->co, data);
929                                 data+=3;
930                                 vertcount++;
931                                 mvert++;
932                         }
933
934                         for(a=0; a<dl->parts; a++) {
935
936                                 if( (dl->flag & DL_CYCL_V)==0 && a==dl->parts-1) break;
937
938                                 if(dl->flag & DL_CYCL_U) {                      /* p2 -> p1 -> */
939                                         p1= startvert+ dl->nr*a;        /* p4 -> p3 -> */
940                                         p2= p1+ dl->nr-1;               /* -----> next row */
941                                         p3= p1+ dl->nr;
942                                         p4= p2+ dl->nr;
943                                         b= 0;
944                                 }
945                                 else {
946                                         p2= startvert+ dl->nr*a;
947                                         p1= p2+1;
948                                         p4= p2+ dl->nr;
949                                         p3= p1+ dl->nr;
950                                         b= 1;
951                                 }
952                                 if( (dl->flag & DL_CYCL_V) && a==dl->parts-1) {
953                                         p3-= dl->parts*dl->nr;
954                                         p4-= dl->parts*dl->nr;
955                                 }
956
957                                 for(; b<dl->nr; b++) {
958                                         mface->v1= p1;
959                                         mface->v2= p3;
960                                         mface->v3= p4;
961                                         mface->v4= p2;
962                                         mface->mat_nr= (unsigned char)dl->col;
963                                         test_index_face(mface, NULL, 0, 4);
964
965                                         if(smooth) mface->flag |= ME_SMOOTH;
966                                         mface++;
967
968                                         p4= p3;
969                                         p3++;
970                                         p2= p1;
971                                         p1++;
972                                 }
973                         }
974
975                 }
976
977                 dl= dl->next;
978         }
979
980         *_totvert= totvert;
981         *_totface= totvlak;
982
983         make_edges_mdata(*allvert, *allface, totvert, totvlak, 0, alledge, _totedge);
984         mfaces_strip_loose(*allface, _totface);
985
986         return 0;
987 }
988
989 /* this may fail replacing ob->data, be sure to check ob->type */
990 void nurbs_to_mesh(Object *ob)
991 {
992         Main *bmain= G.main;
993         Object *ob1;
994         DerivedMesh *dm= ob->derivedFinal;
995         Mesh *me;
996         Curve *cu;
997         MVert *allvert= NULL;
998         MEdge *alledge= NULL;
999         MFace *allface= NULL;
1000         int totvert, totedge, totface;
1001
1002         cu= ob->data;
1003
1004         if (dm == NULL) {
1005                 if (nurbs_to_mdata (ob, &allvert, &totvert, &alledge, &totedge, &allface, &totface) != 0) {
1006                         /* Error initializing */
1007                         return;
1008                 }
1009
1010                 /* make mesh */
1011                 me= add_mesh("Mesh");
1012                 me->totvert= totvert;
1013                 me->totface= totface;
1014                 me->totedge= totedge;
1015
1016                 me->mvert= CustomData_add_layer(&me->vdata, CD_MVERT, CD_ASSIGN, allvert, me->totvert);
1017                 me->mface= CustomData_add_layer(&me->fdata, CD_MFACE, CD_ASSIGN, allface, me->totface);
1018                 me->medge= CustomData_add_layer(&me->edata, CD_MEDGE, CD_ASSIGN, alledge, me->totedge);
1019
1020                 mesh_calc_normals(me->mvert, me->totvert, me->mface, me->totface, NULL);
1021         } else {
1022                 me= add_mesh("Mesh");
1023                 DM_to_mesh(dm, me);
1024         }
1025
1026         me->totcol= cu->totcol;
1027         me->mat= cu->mat;
1028
1029         tex_space_mesh(me);
1030
1031         cu->mat= NULL;
1032         cu->totcol= 0;
1033
1034         if(ob->data) {
1035                 free_libblock(&bmain->curve, ob->data);
1036         }
1037         ob->data= me;
1038         ob->type= OB_MESH;
1039
1040         /* other users */
1041         ob1= bmain->object.first;
1042         while(ob1) {
1043                 if(ob1->data==cu) {
1044                         ob1->type= OB_MESH;
1045                 
1046                         ob1->data= ob->data;
1047                         id_us_plus((ID *)ob->data);
1048                 }
1049                 ob1= ob1->id.next;
1050         }
1051 }
1052
1053 typedef struct EdgeLink {
1054         Link *next, *prev;
1055         void *edge;
1056 } EdgeLink;
1057
1058 typedef struct VertLink {
1059         Link *next, *prev;
1060         int index;
1061 } VertLink;
1062
1063 static void prependPolyLineVert(ListBase *lb, int index)
1064 {
1065         VertLink *vl= MEM_callocN(sizeof(VertLink), "VertLink");
1066         vl->index = index;
1067         BLI_addhead(lb, vl);
1068 }
1069
1070 static void appendPolyLineVert(ListBase *lb, int index)
1071 {
1072         VertLink *vl= MEM_callocN(sizeof(VertLink), "VertLink");
1073         vl->index = index;
1074         BLI_addtail(lb, vl);
1075 }
1076
1077 void mesh_to_curve(Scene *scene, Object *ob)
1078 {
1079         /* make new mesh data from the original copy */
1080         DerivedMesh *dm= mesh_get_derived_final(scene, ob, CD_MASK_MESH);
1081
1082         MVert *mverts= dm->getVertArray(dm);
1083         MEdge *med, *medge= dm->getEdgeArray(dm);
1084         MFace *mf,  *mface= dm->getFaceArray(dm);
1085
1086         int totedge = dm->getNumEdges(dm);
1087         int totface = dm->getNumFaces(dm);
1088         int totedges = 0;
1089         int i, needsFree = 0;
1090
1091         /* only to detect edge polylines */
1092         EdgeHash *eh = BLI_edgehash_new();
1093         EdgeHash *eh_edge = BLI_edgehash_new();
1094
1095
1096         ListBase edges = {NULL, NULL};
1097
1098         /* create edges from all faces (so as to find edges not in any faces) */
1099         mf= mface;
1100         for (i = 0; i < totface; i++, mf++) {
1101                 if (!BLI_edgehash_haskey(eh, mf->v1, mf->v2))
1102                         BLI_edgehash_insert(eh, mf->v1, mf->v2, NULL);
1103                 if (!BLI_edgehash_haskey(eh, mf->v2, mf->v3))
1104                         BLI_edgehash_insert(eh, mf->v2, mf->v3, NULL);
1105
1106                 if (mf->v4) {
1107                         if (!BLI_edgehash_haskey(eh, mf->v3, mf->v4))
1108                                 BLI_edgehash_insert(eh, mf->v3, mf->v4, NULL);
1109                         if (!BLI_edgehash_haskey(eh, mf->v4, mf->v1))
1110                                 BLI_edgehash_insert(eh, mf->v4, mf->v1, NULL);
1111                 } else {
1112                         if (!BLI_edgehash_haskey(eh, mf->v3, mf->v1))
1113                                 BLI_edgehash_insert(eh, mf->v3, mf->v1, NULL);
1114                 }
1115         }
1116
1117         med= medge;
1118         for(i=0; i<totedge; i++, med++) {
1119                 if (!BLI_edgehash_haskey(eh, med->v1, med->v2)) {
1120                         EdgeLink *edl= MEM_callocN(sizeof(EdgeLink), "EdgeLink");
1121
1122                         BLI_edgehash_insert(eh_edge, med->v1, med->v2, NULL);
1123                         edl->edge= med;
1124
1125                         BLI_addtail(&edges, edl);       totedges++;
1126                 }
1127         }
1128         BLI_edgehash_free(eh_edge, NULL);
1129         BLI_edgehash_free(eh, NULL);
1130
1131         if(edges.first) {
1132                 Curve *cu = add_curve(ob->id.name+2, OB_CURVE);
1133                 cu->flag |= CU_3D;
1134
1135                 while(edges.first) {
1136                         /* each iteration find a polyline and add this as a nurbs poly spline */
1137
1138                         ListBase polyline = {NULL, NULL}; /* store a list of VertLink's */
1139                         int closed = FALSE;
1140                         int totpoly= 0;
1141                         MEdge *med_current= ((EdgeLink *)edges.last)->edge;
1142                         int startVert= med_current->v1;
1143                         int endVert= med_current->v2;
1144                         int ok= TRUE;
1145
1146                         appendPolyLineVert(&polyline, startVert);       totpoly++;
1147                         appendPolyLineVert(&polyline, endVert);         totpoly++;
1148                         BLI_freelinkN(&edges, edges.last);                      totedges--;
1149
1150                         while(ok) { /* while connected edges are found... */
1151                                 ok = FALSE;
1152                                 i= totedges;
1153                                 while(i) {
1154                                         EdgeLink *edl;
1155
1156                                         i-=1;
1157                                         edl= BLI_findlink(&edges, i);
1158                                         med= edl->edge;
1159
1160                                         if(med->v1==endVert) {
1161                                                 endVert = med->v2;
1162                                                 appendPolyLineVert(&polyline, med->v2); totpoly++;
1163                                                 BLI_freelinkN(&edges, edl);                             totedges--;
1164                                                 ok= TRUE;
1165                                         }
1166                                         else if(med->v2==endVert) {
1167                                                 endVert = med->v1;
1168                                                 appendPolyLineVert(&polyline, endVert); totpoly++;
1169                                                 BLI_freelinkN(&edges, edl);                             totedges--;
1170                                                 ok= TRUE;
1171                                         }
1172                                         else if(med->v1==startVert) {
1173                                                 startVert = med->v2;
1174                                                 prependPolyLineVert(&polyline, startVert);      totpoly++;
1175                                                 BLI_freelinkN(&edges, edl);                                     totedges--;
1176                                                 ok= TRUE;
1177                                         }
1178                                         else if(med->v2==startVert) {
1179                                                 startVert = med->v1;
1180                                                 prependPolyLineVert(&polyline, startVert);      totpoly++;
1181                                                 BLI_freelinkN(&edges, edl);                                     totedges--;
1182                                                 ok= TRUE;
1183                                         }
1184                                 }
1185                         }
1186
1187                         /* Now we have a polyline, make into a curve */
1188                         if(startVert==endVert) {
1189                                 BLI_freelinkN(&polyline, polyline.last);
1190                                 totpoly--;
1191                                 closed = TRUE;
1192                         }
1193
1194                         /* --- nurbs --- */
1195                         {
1196                                 Nurb *nu;
1197                                 BPoint *bp;
1198                                 VertLink *vl;
1199
1200                                 /* create new 'nurb' within the curve */
1201                                 nu = (Nurb *)MEM_callocN(sizeof(Nurb), "MeshNurb");
1202
1203                                 nu->pntsu= totpoly;
1204                                 nu->pntsv= 1;
1205                                 nu->orderu= 4;
1206                                 nu->flagu= CU_NURB_ENDPOINT | (closed ? CU_NURB_CYCLIC:0);      /* endpoint */
1207                                 nu->resolu= 12;
1208
1209                                 nu->bp= (BPoint *)MEM_callocN(sizeof(BPoint)*totpoly, "bpoints");
1210
1211                                 /* add points */
1212                                 vl= polyline.first;
1213                                 for (i=0, bp=nu->bp; i < totpoly; i++, bp++, vl=(VertLink *)vl->next) {
1214                                         copy_v3_v3(bp->vec, mverts[vl->index].co);
1215                                         bp->f1= SELECT;
1216                                         bp->radius = bp->weight = 1.0;
1217                                 }
1218                                 BLI_freelistN(&polyline);
1219
1220                                 /* add nurb to curve */
1221                                 BLI_addtail(&cu->nurb, nu);
1222                         }
1223                         /* --- done with nurbs --- */
1224                 }
1225
1226                 ((Mesh *)ob->data)->id.us--;
1227                 ob->data= cu;
1228                 ob->type= OB_CURVE;
1229
1230                 /* curve objects can't contain DM in usual cases, we could free memory */
1231                 needsFree= 1;
1232         }
1233
1234         dm->needsFree = needsFree;
1235         dm->release(dm);
1236
1237         if (needsFree) {
1238                 ob->derivedFinal = NULL;
1239
1240                 /* curve object could have got bounding box only in special cases */
1241                 if(ob->bb) {
1242                         MEM_freeN(ob->bb);
1243                         ob->bb= NULL;
1244                 }
1245         }
1246 }
1247
1248 void mesh_delete_material_index(Mesh *me, int index)
1249 {
1250         int i;
1251
1252         for (i=0; i<me->totface; i++) {
1253                 MFace *mf = &((MFace*) me->mface)[i];
1254                 if (mf->mat_nr && mf->mat_nr>=index) 
1255                         mf->mat_nr--;
1256         }
1257 }
1258
1259 void mesh_set_smooth_flag(Object *meshOb, int enableSmooth) 
1260 {
1261         Mesh *me = meshOb->data;
1262         int i;
1263
1264         for (i=0; i<me->totface; i++) {
1265                 MFace *mf = &((MFace*) me->mface)[i];
1266
1267                 if (enableSmooth) {
1268                         mf->flag |= ME_SMOOTH;
1269                 } else {
1270                         mf->flag &= ~ME_SMOOTH;
1271                 }
1272         }
1273
1274         mesh_calc_normals(me->mvert, me->totvert, me->mface, me->totface, NULL);
1275 }
1276
1277 void mesh_calc_normals(MVert *mverts, int numVerts, MFace *mfaces, int numFaces, float (*faceNors_r)[3]) 
1278 {
1279         float (*tnorms)[3]= MEM_callocN(numVerts*sizeof(*tnorms), "tnorms");
1280         float (*fnors)[3]= (faceNors_r)? faceNors_r: MEM_callocN(sizeof(*fnors)*numFaces, "meshnormals");
1281         int i;
1282         int found_flat=0;
1283
1284         for(i=0; i<numFaces; i++) {
1285                 MFace *mf= &mfaces[i];
1286                 float *f_no= fnors[i];
1287
1288                 if(mf->v4)
1289                         normal_quad_v3(f_no, mverts[mf->v1].co, mverts[mf->v2].co, mverts[mf->v3].co, mverts[mf->v4].co);
1290                 else
1291                         normal_tri_v3(f_no, mverts[mf->v1].co, mverts[mf->v2].co, mverts[mf->v3].co);
1292
1293                 if(mf->flag & ME_SMOOTH) {
1294                         float *n4 = (mf->v4)? tnorms[mf->v4]: NULL;
1295                         float *c4 = (mf->v4)? mverts[mf->v4].co: NULL;
1296
1297                         accumulate_vertex_normals(tnorms[mf->v1], tnorms[mf->v2], tnorms[mf->v3], n4,
1298                                 f_no, mverts[mf->v1].co, mverts[mf->v2].co, mverts[mf->v3].co, c4);
1299                 }
1300                 else {
1301                         found_flat=1;
1302                 }
1303         }
1304
1305         /* build smooth normals for uninitialized normals at faces set to flat */
1306         if(found_flat!=0) {
1307                 const int nr_bits= sizeof(int)*8;
1308                 const int nr_words= (numVerts+(nr_bits-1))/nr_bits;
1309                 int *bit_array= (int*)MEM_callocN(sizeof(int)*MAX2(nr_words, 1), "temp buffer");
1310
1311                 for(i=0; i<numFaces; i++) {
1312                         MFace *mf= &mfaces[i];
1313
1314                         if(!(mf->flag & ME_SMOOTH)) {
1315                                 if(is_zero_v3(tnorms[mf->v1])) bit_array[mf->v1/nr_bits]|=(1<<(mf->v1&(nr_bits-1)));
1316                                 if(is_zero_v3(tnorms[mf->v2])) bit_array[mf->v2/nr_bits]|=(1<<(mf->v2&(nr_bits-1)));
1317                                 if(is_zero_v3(tnorms[mf->v3])) bit_array[mf->v3/nr_bits]|=(1<<(mf->v3&(nr_bits-1)));
1318                                 if(mf->v4 && is_zero_v3(tnorms[mf->v4])) bit_array[mf->v4/nr_bits]|=(1<<(mf->v4&(nr_bits-1)));
1319                         }
1320                 }
1321
1322                 for(i=0; i<numFaces; i++) {
1323                         MFace *mf= &mfaces[i];
1324
1325                         if((mf->flag&ME_SMOOTH)==0) {
1326                                 float *f_no= fnors[i];
1327
1328                                 if(bit_array[mf->v1/nr_bits]&(1<<(mf->v1&(nr_bits-1)))) add_v3_v3(tnorms[mf->v1], f_no);
1329                                 if(bit_array[mf->v2/nr_bits]&(1<<(mf->v2&(nr_bits-1)))) add_v3_v3(tnorms[mf->v2], f_no);
1330                                 if(bit_array[mf->v3/nr_bits]&(1<<(mf->v3&(nr_bits-1)))) add_v3_v3(tnorms[mf->v3], f_no);
1331                                 if(mf->v4 && bit_array[mf->v4/nr_bits]&(1<<(mf->v4&(nr_bits-1)))) add_v3_v3(tnorms[mf->v4], f_no);
1332                         }
1333                 }
1334
1335                 MEM_freeN(bit_array);
1336         }
1337
1338         /* following Mesh convention; we use vertex coordinate itself for normal in this case */
1339         for(i=0; i<numVerts; i++) {
1340                 MVert *mv= &mverts[i];
1341                 float *no= tnorms[i];
1342                 
1343                 if(normalize_v3(no) == 0.0f)
1344                         normalize_v3_v3(no, mv->co);
1345
1346                 normal_float_to_short_v3(mv->no, no);
1347         }
1348         
1349         MEM_freeN(tnorms);
1350
1351         if(fnors != faceNors_r)
1352                 MEM_freeN(fnors);
1353 }
1354
1355 float (*mesh_getVertexCos(Mesh *me, int *numVerts_r))[3]
1356 {
1357         int i, numVerts = me->totvert;
1358         float (*cos)[3] = MEM_mallocN(sizeof(*cos)*numVerts, "vertexcos1");
1359         
1360         if (numVerts_r) *numVerts_r = numVerts;
1361         for (i=0; i<numVerts; i++)
1362                 VECCOPY(cos[i], me->mvert[i].co);
1363         
1364         return cos;
1365 }
1366
1367 UvVertMap *make_uv_vert_map(struct MFace *mface, struct MTFace *tface, unsigned int totface, unsigned int totvert, int selected, float *limit)
1368 {
1369         UvVertMap *vmap;
1370         UvMapVert *buf;
1371         MFace *mf;
1372         MTFace *tf;
1373         unsigned int a;
1374         int     i, totuv, nverts;
1375
1376         totuv = 0;
1377
1378         /* generate UvMapVert array */
1379         mf= mface;
1380         tf= tface;
1381         for(a=0; a<totface; a++, mf++, tf++)
1382                 if(!selected || (!(mf->flag & ME_HIDE) && (mf->flag & ME_FACE_SEL)))
1383                         totuv += (mf->v4)? 4: 3;
1384                 
1385         if(totuv==0)
1386                 return NULL;
1387         
1388         vmap= (UvVertMap*)MEM_callocN(sizeof(*vmap), "UvVertMap");
1389         if (!vmap)
1390                 return NULL;
1391
1392         vmap->vert= (UvMapVert**)MEM_callocN(sizeof(*vmap->vert)*totvert, "UvMapVert*");
1393         buf= vmap->buf= (UvMapVert*)MEM_callocN(sizeof(*vmap->buf)*totuv, "UvMapVert");
1394
1395         if (!vmap->vert || !vmap->buf) {
1396                 free_uv_vert_map(vmap);
1397                 return NULL;
1398         }
1399
1400         mf= mface;
1401         tf= tface;
1402         for(a=0; a<totface; a++, mf++, tf++) {
1403                 if(!selected || (!(mf->flag & ME_HIDE) && (mf->flag & ME_FACE_SEL))) {
1404                         nverts= (mf->v4)? 4: 3;
1405
1406                         for(i=0; i<nverts; i++) {
1407                                 buf->tfindex= i;
1408                                 buf->f= a;
1409                                 buf->separate = 0;
1410                                 buf->next= vmap->vert[*(&mf->v1 + i)];
1411                                 vmap->vert[*(&mf->v1 + i)]= buf;
1412                                 buf++;
1413                         }
1414                 }
1415         }
1416         
1417         /* sort individual uvs for each vert */
1418         tf= tface;
1419         for(a=0; a<totvert; a++) {
1420                 UvMapVert *newvlist= NULL, *vlist=vmap->vert[a];
1421                 UvMapVert *iterv, *v, *lastv, *next;
1422                 float *uv, *uv2, uvdiff[2];
1423
1424                 while(vlist) {
1425                         v= vlist;
1426                         vlist= vlist->next;
1427                         v->next= newvlist;
1428                         newvlist= v;
1429
1430                         uv= (tf+v->f)->uv[v->tfindex];
1431                         lastv= NULL;
1432                         iterv= vlist;
1433
1434                         while(iterv) {
1435                                 next= iterv->next;
1436
1437                                 uv2= (tf+iterv->f)->uv[iterv->tfindex];
1438                                 sub_v2_v2v2(uvdiff, uv2, uv);
1439
1440
1441                                 if(fabs(uv[0]-uv2[0]) < limit[0] && fabs(uv[1]-uv2[1]) < limit[1]) {
1442                                         if(lastv) lastv->next= next;
1443                                         else vlist= next;
1444                                         iterv->next= newvlist;
1445                                         newvlist= iterv;
1446                                 }
1447                                 else
1448                                         lastv=iterv;
1449
1450                                 iterv= next;
1451                         }
1452
1453                         newvlist->separate = 1;
1454                 }
1455
1456                 vmap->vert[a]= newvlist;
1457         }
1458         
1459         return vmap;
1460 }
1461
1462 UvMapVert *get_uv_map_vert(UvVertMap *vmap, unsigned int v)
1463 {
1464         return vmap->vert[v];
1465 }
1466
1467 void free_uv_vert_map(UvVertMap *vmap)
1468 {
1469         if (vmap) {
1470                 if (vmap->vert) MEM_freeN(vmap->vert);
1471                 if (vmap->buf) MEM_freeN(vmap->buf);
1472                 MEM_freeN(vmap);
1473         }
1474 }
1475
1476 /* Generates a map where the key is the vertex and the value is a list
1477    of faces that use that vertex as a corner. The lists are allocated
1478    from one memory pool. */
1479 void create_vert_face_map(ListBase **map, IndexNode **mem, const MFace *mface, const int totvert, const int totface)
1480 {
1481         int i,j;
1482         IndexNode *node = NULL;
1483         
1484         (*map) = MEM_callocN(sizeof(ListBase) * totvert, "vert face map");
1485         (*mem) = MEM_callocN(sizeof(IndexNode) * totface*4, "vert face map mem");
1486         node = *mem;
1487         
1488         /* Find the users */
1489         for(i = 0; i < totface; ++i){
1490                 for(j = 0; j < (mface[i].v4?4:3); ++j, ++node) {
1491                         node->index = i;
1492                         BLI_addtail(&(*map)[((unsigned int*)(&mface[i]))[j]], node);
1493                 }
1494         }
1495 }
1496
1497 /* Generates a map where the key is the vertex and the value is a list
1498    of edges that use that vertex as an endpoint. The lists are allocated
1499    from one memory pool. */
1500 void create_vert_edge_map(ListBase **map, IndexNode **mem, const MEdge *medge, const int totvert, const int totedge)
1501 {
1502         int i, j;
1503         IndexNode *node = NULL;
1504  
1505         (*map) = MEM_callocN(sizeof(ListBase) * totvert, "vert edge map");
1506         (*mem) = MEM_callocN(sizeof(IndexNode) * totedge * 2, "vert edge map mem");
1507         node = *mem;
1508        
1509         /* Find the users */
1510         for(i = 0; i < totedge; ++i){
1511                 for(j = 0; j < 2; ++j, ++node) {
1512                         node->index = i;
1513                         BLI_addtail(&(*map)[((unsigned int*)(&medge[i].v1))[j]], node);
1514                 }
1515         }
1516 }
1517
1518 /* Partial Mesh Visibility */
1519 PartialVisibility *mesh_pmv_copy(PartialVisibility *pmv)
1520 {
1521         PartialVisibility *n= MEM_dupallocN(pmv);
1522         n->vert_map= MEM_dupallocN(pmv->vert_map);
1523         n->edge_map= MEM_dupallocN(pmv->edge_map);
1524         n->old_edges= MEM_dupallocN(pmv->old_edges);
1525         n->old_faces= MEM_dupallocN(pmv->old_faces);
1526         return n;
1527 }
1528
1529 void mesh_pmv_free(PartialVisibility *pv)
1530 {
1531         MEM_freeN(pv->vert_map);
1532         MEM_freeN(pv->edge_map);
1533         MEM_freeN(pv->old_faces);
1534         MEM_freeN(pv->old_edges);
1535         MEM_freeN(pv);
1536 }
1537
1538 void mesh_pmv_revert(Mesh *me)
1539 {
1540         if(me->pv) {
1541                 unsigned i;
1542                 MVert *nve, *old_verts;
1543                 
1544                 /* Reorder vertices */
1545                 nve= me->mvert;
1546                 old_verts = MEM_mallocN(sizeof(MVert)*me->pv->totvert,"PMV revert verts");
1547                 for(i=0; i<me->pv->totvert; ++i)
1548                         old_verts[i]= nve[me->pv->vert_map[i]];
1549
1550                 /* Restore verts, edges and faces */
1551                 CustomData_free_layer_active(&me->vdata, CD_MVERT, me->totvert);
1552                 CustomData_free_layer_active(&me->edata, CD_MEDGE, me->totedge);
1553                 CustomData_free_layer_active(&me->fdata, CD_MFACE, me->totface);
1554
1555                 CustomData_add_layer(&me->vdata, CD_MVERT, CD_ASSIGN, old_verts, me->pv->totvert);
1556                 CustomData_add_layer(&me->edata, CD_MEDGE, CD_ASSIGN, me->pv->old_edges, me->pv->totedge);
1557                 CustomData_add_layer(&me->fdata, CD_MFACE, CD_ASSIGN, me->pv->old_faces, me->pv->totface);
1558                 mesh_update_customdata_pointers(me);
1559
1560                 me->totvert= me->pv->totvert;
1561                 me->totedge= me->pv->totedge;
1562                 me->totface= me->pv->totface;
1563
1564                 me->pv->old_edges= NULL;
1565                 me->pv->old_faces= NULL;
1566
1567                 /* Free maps */
1568                 MEM_freeN(me->pv->edge_map);
1569                 me->pv->edge_map= NULL;
1570                 MEM_freeN(me->pv->vert_map);
1571                 me->pv->vert_map= NULL;
1572         }
1573 }
1574
1575 void mesh_pmv_off(Mesh *me)
1576 {
1577         if(me->pv) {
1578                 mesh_pmv_revert(me);
1579                 MEM_freeN(me->pv);
1580                 me->pv= NULL;
1581         }
1582 }
1583
1584 /* basic vertex data functions */
1585 int minmax_mesh(Mesh *me, float min[3], float max[3])
1586 {
1587         int i= me->totvert;
1588         MVert *mvert;
1589         for(mvert= me->mvert; i--; mvert++) {
1590                 DO_MINMAX(mvert->co, min, max);
1591         }
1592         
1593         return (me->totvert != 0);
1594 }
1595
1596 int mesh_center_median(Mesh *me, float cent[3])
1597 {
1598         int i= me->totvert;
1599         MVert *mvert;
1600         zero_v3(cent);
1601         for(mvert= me->mvert; i--; mvert++) {
1602                 add_v3_v3(cent, mvert->co);
1603         }
1604         /* otherwise we get NAN for 0 verts */
1605         if(me->totvert) {
1606                 mul_v3_fl(cent, 1.0f/(float)me->totvert);
1607         }
1608
1609         return (me->totvert != 0);
1610 }
1611
1612 int mesh_center_bounds(Mesh *me, float cent[3])
1613 {
1614         float min[3], max[3];
1615         INIT_MINMAX(min, max);
1616         if(minmax_mesh(me, min, max)) {
1617                 mid_v3_v3v3(cent, min, max);
1618                 return 1;
1619         }
1620
1621         return 0;
1622 }
1623
1624 void mesh_translate(Mesh *me, float offset[3], int do_keys)
1625 {
1626         int i= me->totvert;
1627         MVert *mvert;
1628         for(mvert= me->mvert; i--; mvert++) {
1629                 add_v3_v3(mvert->co, offset);
1630         }
1631         
1632         if (do_keys && me->key) {
1633                 KeyBlock *kb;
1634                 for (kb=me->key->block.first; kb; kb=kb->next) {
1635                         float *fp= kb->data;
1636                         for (i= kb->totelem; i--; fp+=3) {
1637                                 add_v3_v3(fp, offset);
1638                         }
1639                 }
1640         }
1641 }