svn merge https://svn.blender.org/svnroot/bf-blender/trunk/blender -r22668:22701
[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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, 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 #ifdef HAVE_CONFIG_H
33 #include <config.h>
34 #endif
35
36 #include <stdlib.h>
37 #include <string.h>
38 #include <stdio.h>
39 #include <math.h>
40
41 #include "MEM_guardedalloc.h"
42
43 #include "DNA_ID.h"
44 #include "DNA_curve_types.h"
45 #include "DNA_material_types.h"
46 #include "DNA_object_types.h"
47 #include "DNA_image_types.h"
48 #include "DNA_key_types.h"
49 #include "DNA_mesh_types.h"
50 #include "DNA_meshdata_types.h"
51 #include "DNA_ipo_types.h"
52
53 #include "BKE_customdata.h"
54 #include "BKE_depsgraph.h"
55 #include "BKE_main.h"
56 #include "BKE_DerivedMesh.h"
57 #include "BKE_global.h"
58 #include "BKE_mesh.h"
59 #include "BKE_subsurf.h"
60 #include "BKE_displist.h"
61 #include "BKE_library.h"
62 #include "BKE_material.h"
63 #include "BKE_key.h"
64 /* these 2 are only used by conversion functions */
65 #include "BKE_curve.h"
66 /* -- */
67 #include "BKE_object.h"
68 #include "BKE_utildefines.h"
69
70 #include "BLI_blenlib.h"
71 #include "BLI_editVert.h"
72 #include "BLI_arithb.h"
73
74
75 EditMesh *BKE_mesh_get_editmesh(Mesh *me)
76 {
77         return me->edit_mesh;
78 }
79
80 void BKE_mesh_end_editmesh(Mesh *me, EditMesh *em)
81 {
82 }
83
84
85 void mesh_update_customdata_pointers(Mesh *me)
86 {
87         me->mvert = CustomData_get_layer(&me->vdata, CD_MVERT);
88         me->dvert = CustomData_get_layer(&me->vdata, CD_MDEFORMVERT);
89         me->msticky = CustomData_get_layer(&me->vdata, CD_MSTICKY);
90
91         me->medge = CustomData_get_layer(&me->edata, CD_MEDGE);
92
93         me->mface = CustomData_get_layer(&me->fdata, CD_MFACE);
94         me->mcol = CustomData_get_layer(&me->fdata, CD_MCOL);
95         me->mtface = CustomData_get_layer(&me->fdata, CD_MTFACE);
96 }
97
98 /* Note: unlinking is called when me->id.us is 0, question remains how
99  * much unlinking of Library data in Mesh should be done... probably
100  * we need a more generic method, like the expand() functions in
101  * readfile.c */
102
103 void unlink_mesh(Mesh *me)
104 {
105         int a;
106         
107         if(me==0) return;
108         
109         for(a=0; a<me->totcol; a++) {
110                 if(me->mat[a]) me->mat[a]->id.us--;
111                 me->mat[a]= 0;
112         }
113
114         if(me->key) {
115                 me->key->id.us--;
116                 if (me->key->id.us == 0 && me->key->ipo )
117                         me->key->ipo->id.us--;
118         }
119         me->key= 0;
120         
121         if(me->texcomesh) me->texcomesh= 0;
122 }
123
124
125 /* do not free mesh itself */
126 void free_mesh(Mesh *me)
127 {
128         unlink_mesh(me);
129
130         if(me->pv) {
131                 if(me->pv->vert_map) MEM_freeN(me->pv->vert_map);
132                 if(me->pv->edge_map) MEM_freeN(me->pv->edge_map);
133                 if(me->pv->old_faces) MEM_freeN(me->pv->old_faces);
134                 if(me->pv->old_edges) MEM_freeN(me->pv->old_edges);
135                 me->totvert= me->pv->totvert;
136                 me->totedge= me->pv->totedge;
137                 me->totface= me->pv->totface;
138                 MEM_freeN(me->pv);
139         }
140
141         CustomData_free(&me->vdata, me->totvert);
142         CustomData_free(&me->edata, me->totedge);
143         CustomData_free(&me->fdata, me->totface);
144
145         if(me->mat) MEM_freeN(me->mat);
146         
147         if(me->bb) MEM_freeN(me->bb);
148         if(me->mselect) MEM_freeN(me->mselect);
149         if(me->edit_mesh) MEM_freeN(me->edit_mesh);
150 }
151
152 void copy_dverts(MDeformVert *dst, MDeformVert *src, int copycount)
153 {
154         /* Assumes dst is already set up */
155         int i;
156
157         if (!src || !dst)
158                 return;
159
160         memcpy (dst, src, copycount * sizeof(MDeformVert));
161         
162         for (i=0; i<copycount; i++){
163                 if (src[i].dw){
164                         dst[i].dw = MEM_callocN (sizeof(MDeformWeight)*src[i].totweight, "copy_deformWeight");
165                         memcpy (dst[i].dw, src[i].dw, sizeof (MDeformWeight)*src[i].totweight);
166                 }
167         }
168
169 }
170
171 void free_dverts(MDeformVert *dvert, int totvert)
172 {
173         /* Instead of freeing the verts directly,
174         call this function to delete any special
175         vert data */
176         int     i;
177
178         if (!dvert)
179                 return;
180
181         /* Free any special data from the verts */
182         for (i=0; i<totvert; i++){
183                 if (dvert[i].dw) MEM_freeN (dvert[i].dw);
184         }
185         MEM_freeN (dvert);
186 }
187
188 Mesh *add_mesh(char *name)
189 {
190         Mesh *me;
191         
192         me= alloc_libblock(&G.main->mesh, ID_ME, name);
193         
194         me->size[0]= me->size[1]= me->size[2]= 1.0;
195         me->smoothresh= 30;
196         me->texflag= AUTOSPACE;
197         me->flag= ME_TWOSIDED;
198         me->bb= unit_boundbox();
199         me->drawflag= ME_DRAWEDGES|ME_DRAWFACES|ME_DRAWCREASES;
200         
201         return me;
202 }
203
204 Mesh *copy_mesh(Mesh *me)
205 {
206         Mesh *men;
207         MTFace *tface;
208         int a, i;
209         
210         men= copy_libblock(me);
211         
212         men->mat= MEM_dupallocN(me->mat);
213         for(a=0; a<men->totcol; a++) {
214                 id_us_plus((ID *)men->mat[a]);
215         }
216         id_us_plus((ID *)men->texcomesh);
217
218         CustomData_copy(&me->vdata, &men->vdata, CD_MASK_MESH, CD_DUPLICATE, men->totvert);
219         CustomData_copy(&me->edata, &men->edata, CD_MASK_MESH, CD_DUPLICATE, men->totedge);
220         CustomData_copy(&me->fdata, &men->fdata, CD_MASK_MESH, CD_DUPLICATE, men->totface);
221         mesh_update_customdata_pointers(men);
222
223         /* ensure indirect linked data becomes lib-extern */
224         for(i=0; i<me->fdata.totlayer; i++) {
225                 if(me->fdata.layers[i].type == CD_MTFACE) {
226                         tface= (MTFace*)me->fdata.layers[i].data;
227
228                         for(a=0; a<me->totface; a++, tface++)
229                                 if(tface->tpage)
230                                         id_lib_extern((ID*)tface->tpage);
231                 }
232         }
233         
234         men->mselect= NULL;
235
236         men->bb= MEM_dupallocN(men->bb);
237         
238         men->key= copy_key(me->key);
239         if(men->key) men->key->from= (ID *)men;
240
241         return men;
242 }
243
244 void make_local_tface(Mesh *me)
245 {
246         MTFace *tface;
247         Image *ima;
248         int a, i;
249         
250         for(i=0; i<me->fdata.totlayer; i++) {
251                 if(me->fdata.layers[i].type == CD_MTFACE) {
252                         tface= (MTFace*)me->fdata.layers[i].data;
253                         
254                         for(a=0; a<me->totface; a++, tface++) {
255                                 /* special case: ima always local immediately */
256                                 if(tface->tpage) {
257                                         ima= tface->tpage;
258                                         if(ima->id.lib) {
259                                                 ima->id.lib= 0;
260                                                 ima->id.flag= LIB_LOCAL;
261                                                 new_id(0, (ID *)ima, 0);
262                                         }
263                                 }
264                         }
265                 }
266         }
267 }
268
269 void make_local_mesh(Mesh *me)
270 {
271         Object *ob;
272         Mesh *men;
273         int local=0, lib=0;
274
275         /* - only lib users: do nothing
276             * - only local users: set flag
277             * - mixed: make copy
278             */
279         
280         if(me->id.lib==0) return;
281         if(me->id.us==1) {
282                 me->id.lib= 0;
283                 me->id.flag= LIB_LOCAL;
284                 new_id(0, (ID *)me, 0);
285                 
286                 if(me->mtface) make_local_tface(me);
287                 
288                 return;
289         }
290         
291         ob= G.main->object.first;
292         while(ob) {
293                 if( me==get_mesh(ob) ) {
294                         if(ob->id.lib) lib= 1;
295                         else local= 1;
296                 }
297                 ob= ob->id.next;
298         }
299         
300         if(local && lib==0) {
301                 me->id.lib= 0;
302                 me->id.flag= LIB_LOCAL;
303                 new_id(0, (ID *)me, 0);
304                 
305                 if(me->mtface) make_local_tface(me);
306                 
307         }
308         else if(local && lib) {
309                 men= copy_mesh(me);
310                 men->id.us= 0;
311                 
312                 ob= G.main->object.first;
313                 while(ob) {
314                         if( me==get_mesh(ob) ) {                                
315                                 if(ob->id.lib==0) {
316                                         set_mesh(ob, men);
317                                 }
318                         }
319                         ob= ob->id.next;
320                 }
321         }
322 }
323
324 void boundbox_mesh(Mesh *me, float *loc, float *size)
325 {
326         MVert *mvert;
327         BoundBox *bb;
328         float min[3], max[3];
329         float mloc[3], msize[3];
330         int a;
331         
332         if(me->bb==0) me->bb= MEM_callocN(sizeof(BoundBox), "boundbox");
333         bb= me->bb;
334         
335         INIT_MINMAX(min, max);
336
337         if (!loc) loc= mloc;
338         if (!size) size= msize;
339         
340         mvert= me->mvert;
341         for(a=0; a<me->totvert; a++, mvert++) {
342                 DO_MINMAX(mvert->co, min, max);
343         }
344
345         if(!me->totvert) {
346                 min[0] = min[1] = min[2] = -1.0f;
347                 max[0] = max[1] = max[2] = 1.0f;
348         }
349
350         loc[0]= (min[0]+max[0])/2.0f;
351         loc[1]= (min[1]+max[1])/2.0f;
352         loc[2]= (min[2]+max[2])/2.0f;
353                 
354         size[0]= (max[0]-min[0])/2.0f;
355         size[1]= (max[1]-min[1])/2.0f;
356         size[2]= (max[2]-min[2])/2.0f;
357         
358         boundbox_set_from_min_max(bb, min, max);
359 }
360
361 void tex_space_mesh(Mesh *me)
362 {
363         KeyBlock *kb;
364         float *fp, loc[3], size[3], min[3], max[3];
365         int a;
366
367         boundbox_mesh(me, loc, size);
368
369         if(me->texflag & AUTOSPACE) {
370                 if(me->key) {
371                         kb= me->key->refkey;
372                         if (kb) {
373                                 
374                                 INIT_MINMAX(min, max);
375                                 
376                                 fp= kb->data;
377                                 for(a=0; a<kb->totelem; a++, fp+=3) {   
378                                         DO_MINMAX(fp, min, max);
379                                 }
380                                 if(kb->totelem) {
381                                         loc[0]= (min[0]+max[0])/2.0f; loc[1]= (min[1]+max[1])/2.0f; loc[2]= (min[2]+max[2])/2.0f;
382                                         size[0]= (max[0]-min[0])/2.0f; size[1]= (max[1]-min[1])/2.0f; size[2]= (max[2]-min[2])/2.0f;
383                                 }
384                                 else {
385                                         loc[0]= loc[1]= loc[2]= 0.0;
386                                         size[0]= size[1]= size[2]= 0.0;
387                                 }
388                                 
389                         }
390                 }
391
392                 for (a=0; a<3; a++) {
393                         if(size[a]==0.0) size[a]= 1.0;
394                         else if(size[a]>0.0 && size[a]<0.00001) size[a]= 0.00001;
395                         else if(size[a]<0.0 && size[a]> -0.00001) size[a]= -0.00001;
396                 }
397
398                 VECCOPY(me->loc, loc);
399                 VECCOPY(me->size, size);
400                 me->rot[0]= me->rot[1]= me->rot[2]= 0.0;
401         }
402 }
403
404 BoundBox *mesh_get_bb(Object *ob)
405 {
406         Mesh *me= ob->data;
407
408         if(ob->bb)
409                 return ob->bb;
410
411         if (!me->bb)
412                 tex_space_mesh(me);
413
414         return me->bb;
415 }
416
417 void mesh_get_texspace(Mesh *me, float *loc_r, float *rot_r, float *size_r)
418 {
419         if (!me->bb) {
420                 tex_space_mesh(me);
421         }
422
423         if (loc_r) VECCOPY(loc_r, me->loc);
424         if (rot_r) VECCOPY(rot_r, me->rot);
425         if (size_r) VECCOPY(size_r, me->size);
426 }
427
428 float *get_mesh_orco_verts(Object *ob)
429 {
430         Mesh *me = ob->data;
431         int a, totvert;
432         float (*vcos)[3] = NULL;
433
434         /* Get appropriate vertex coordinates */
435         if(me->key && me->texcomesh==0 && me->key->refkey) {
436                 vcos= mesh_getRefKeyCos(me, &totvert);
437         }
438         else {
439                 MVert *mvert = NULL;            
440                 Mesh *tme = me->texcomesh?me->texcomesh:me;
441
442                 vcos = MEM_callocN(sizeof(*vcos)*me->totvert, "orco mesh");
443                 mvert = tme->mvert;
444                 totvert = MIN2(tme->totvert, me->totvert);
445
446                 for(a=0; a<totvert; a++, mvert++) {
447                         vcos[a][0]= mvert->co[0];
448                         vcos[a][1]= mvert->co[1];
449                         vcos[a][2]= mvert->co[2];
450                 }
451         }
452
453         return (float*)vcos;
454 }
455
456 void transform_mesh_orco_verts(Mesh *me, float (*orco)[3], int totvert, int invert)
457 {
458         float loc[3], size[3];
459         int a;
460
461         mesh_get_texspace(me->texcomesh?me->texcomesh:me, loc, NULL, size);
462
463         if(invert) {
464                 for(a=0; a<totvert; a++) {
465                         float *co = orco[a];
466                         co[0] = co[0]*size[0] + loc[0];
467                         co[1] = co[1]*size[1] + loc[1];
468                         co[2] = co[2]*size[2] + loc[2];
469                 }
470         }
471         else {
472                 for(a=0; a<totvert; a++) {
473                         float *co = orco[a];
474                         co[0] = (co[0]-loc[0])/size[0];
475                         co[1] = (co[1]-loc[1])/size[1];
476                         co[2] = (co[2]-loc[2])/size[2];
477                 }
478         }
479 }
480
481 /* rotates the vertices of a face in case v[2] or v[3] (vertex index) is = 0.
482    this is necessary to make the if(mface->v4) check for quads work */
483 int test_index_face(MFace *mface, CustomData *fdata, int mfindex, int nr)
484 {
485         /* first test if the face is legal */
486         if(mface->v3 && mface->v3==mface->v4) {
487                 mface->v4= 0;
488                 nr--;
489         }
490         if(mface->v2 && mface->v2==mface->v3) {
491                 mface->v3= mface->v4;
492                 mface->v4= 0;
493                 nr--;
494         }
495         if(mface->v1==mface->v2) {
496                 mface->v2= mface->v3;
497                 mface->v3= mface->v4;
498                 mface->v4= 0;
499                 nr--;
500         }
501
502         /* prevent a zero at wrong index location */
503         if(nr==3) {
504                 if(mface->v3==0) {
505                         static int corner_indices[4] = {1, 2, 0, 3};
506
507                         SWAP(int, mface->v1, mface->v2);
508                         SWAP(int, mface->v2, mface->v3);
509
510                         if(fdata)
511                                 CustomData_swap(fdata, mfindex, corner_indices);
512                 }
513         }
514         else if(nr==4) {
515                 if(mface->v3==0 || mface->v4==0) {
516                         static int corner_indices[4] = {2, 3, 0, 1};
517
518                         SWAP(int, mface->v1, mface->v3);
519                         SWAP(int, mface->v2, mface->v4);
520
521                         if(fdata)
522                                 CustomData_swap(fdata, mfindex, corner_indices);
523                 }
524         }
525
526         return nr;
527 }
528
529 Mesh *get_mesh(Object *ob)
530 {
531         
532         if(ob==0) return 0;
533         if(ob->type==OB_MESH) return ob->data;
534         else return 0;
535 }
536
537 void set_mesh(Object *ob, Mesh *me)
538 {
539         Mesh *old=0;
540         
541         if(ob==0) return;
542         
543         if(ob->type==OB_MESH) {
544                 old= ob->data;
545                 old->id.us--;
546                 ob->data= me;
547                 id_us_plus((ID *)me);
548         }
549         
550         test_object_materials((ID *)me);
551 }
552
553 /* ************** make edges in a Mesh, for outside of editmode */
554
555 struct edgesort {
556         int v1, v2;
557         short is_loose, is_draw;
558 };
559
560 /* edges have to be added with lowest index first for sorting */
561 static void to_edgesort(struct edgesort *ed, int v1, int v2, short is_loose, short is_draw)
562 {
563         if(v1<v2) {
564                 ed->v1= v1; ed->v2= v2;
565         }
566         else {
567                 ed->v1= v2; ed->v2= v1;
568         }
569         ed->is_loose= is_loose;
570         ed->is_draw= is_draw;
571 }
572
573 static int vergedgesort(const void *v1, const void *v2)
574 {
575         const struct edgesort *x1=v1, *x2=v2;
576
577         if( x1->v1 > x2->v1) return 1;
578         else if( x1->v1 < x2->v1) return -1;
579         else if( x1->v2 > x2->v2) return 1;
580         else if( x1->v2 < x2->v2) return -1;
581         
582         return 0;
583 }
584
585 void make_edges(Mesh *me, int old)
586 {
587         MFace *mface;
588         MEdge *medge;
589         struct edgesort *edsort, *ed;
590         int a, totedge=0, final=0;
591         
592         /* we put all edges in array, sort them, and detect doubles that way */
593         
594         for(a= me->totface, mface= me->mface; a>0; a--, mface++) {
595                 if(mface->v4) totedge+=4;
596                 else if(mface->v3) totedge+=3;
597                 else totedge+=1;
598         }
599         
600         if(totedge==0) {
601                 /* flag that mesh has edges */
602                 me->medge = MEM_callocN(0, "make mesh edges");
603                 me->totedge = 0;
604                 return;
605         }
606         
607         ed= edsort= MEM_mallocN(totedge*sizeof(struct edgesort), "edgesort");
608         
609         for(a= me->totface, mface= me->mface; a>0; a--, mface++) {
610                 to_edgesort(ed++, mface->v1, mface->v2, !mface->v3, mface->edcode & ME_V1V2);
611                 if(mface->v4) {
612                         to_edgesort(ed++, mface->v2, mface->v3, 0, mface->edcode & ME_V2V3);
613                         to_edgesort(ed++, mface->v3, mface->v4, 0, mface->edcode & ME_V3V4);
614                         to_edgesort(ed++, mface->v4, mface->v1, 0, mface->edcode & ME_V4V1);
615                 }
616                 else if(mface->v3) {
617                         to_edgesort(ed++, mface->v2, mface->v3, 0, mface->edcode & ME_V2V3);
618                         to_edgesort(ed++, mface->v3, mface->v1, 0, mface->edcode & ME_V3V1);
619                 }
620         }
621         
622         qsort(edsort, totedge, sizeof(struct edgesort), vergedgesort);
623         
624         /* count final amount */
625         for(a=totedge, ed=edsort; a>1; a--, ed++) {
626                 /* edge is unique when it differs from next edge, or is last */
627                 if(ed->v1 != (ed+1)->v1 || ed->v2 != (ed+1)->v2) final++;
628         }
629         final++;
630         
631
632         medge= CustomData_add_layer(&me->edata, CD_MEDGE, CD_CALLOC, NULL, final);
633         me->medge= medge;
634         me->totedge= final;
635         
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) {
639                         medge->v1= ed->v1;
640                         medge->v2= ed->v2;
641                         if(old==0 || ed->is_draw) medge->flag= ME_EDGEDRAW|ME_EDGERENDER;
642                         if(ed->is_loose) medge->flag|= ME_LOOSEEDGE;
643                         medge++;
644                 }
645                 else {
646                         /* equal edge, we merge the drawflag */
647                         (ed+1)->is_draw |= ed->is_draw;
648                 }
649         }
650         /* last edge */
651         medge->v1= ed->v1;
652         medge->v2= ed->v2;
653         medge->flag= ME_EDGEDRAW;
654         if(ed->is_loose) medge->flag|= ME_LOOSEEDGE;
655         medge->flag |= ME_EDGERENDER;
656
657         MEM_freeN(edsort);
658
659         mesh_strip_loose_faces(me);
660 }
661
662 void mesh_strip_loose_faces(Mesh *me)
663 {
664         int a,b;
665
666         for (a=b=0; a<me->totface; a++) {
667                 if (me->mface[a].v3) {
668                         if (a!=b) {
669                                 memcpy(&me->mface[b],&me->mface[a],sizeof(me->mface[b]));
670                                 CustomData_copy_data(&me->fdata, &me->fdata, a, b, 1);
671                                 CustomData_free_elem(&me->fdata, a, 1);
672                         }
673                         b++;
674                 }
675         }
676         me->totface = b;
677 }
678
679
680 void mball_to_mesh(ListBase *lb, Mesh *me)
681 {
682         DispList *dl;
683         MVert *mvert;
684         MFace *mface;
685         float *nors, *verts;
686         int a, *index;
687         
688         dl= lb->first;
689         if(dl==0) return;
690
691         if(dl->type==DL_INDEX4) {
692                 me->flag= ME_NOPUNOFLIP;
693                 me->totvert= dl->nr;
694                 me->totface= dl->parts;
695                 
696                 mvert= CustomData_add_layer(&me->vdata, CD_MVERT, CD_CALLOC, NULL, dl->nr);
697                 mface= CustomData_add_layer(&me->fdata, CD_MFACE, CD_CALLOC, NULL, dl->parts);
698                 me->mvert= mvert;
699                 me->mface= mface;
700
701                 a= dl->nr;
702                 nors= dl->nors;
703                 verts= dl->verts;
704                 while(a--) {
705                         VECCOPY(mvert->co, verts);
706                         mvert->no[0]= (short int)(nors[0]*32767.0);
707                         mvert->no[1]= (short int)(nors[1]*32767.0);
708                         mvert->no[2]= (short int)(nors[2]*32767.0);
709                         mvert++;
710                         nors+= 3;
711                         verts+= 3;
712                 }
713                 
714                 a= dl->parts;
715                 index= dl->index;
716                 while(a--) {
717                         mface->v1= index[0];
718                         mface->v2= index[1];
719                         mface->v3= index[2];
720                         mface->v4= index[3];
721                         mface->flag= ME_SMOOTH;
722
723                         test_index_face(mface, NULL, 0, (mface->v3==mface->v4)? 3: 4);
724
725                         mface++;
726                         index+= 4;
727                 }
728
729                 make_edges(me, 0);      // all edges
730         }       
731 }
732
733 /* this may fail replacing ob->data, be sure to check ob->type */
734 void nurbs_to_mesh(Object *ob)
735 {
736         Object *ob1;
737         DispList *dl;
738         Mesh *me;
739         Curve *cu;
740         MVert *mvert;
741         MFace *mface;
742         float *data;
743         int a, b, ofs, vertcount, startvert, totvert=0, totvlak=0;
744         int p1, p2, p3, p4, *index;
745
746         cu= ob->data;
747
748         /* count */
749         dl= cu->disp.first;
750         while(dl) {
751                 if(dl->type==DL_SEGM) {
752                         totvert+= dl->parts*dl->nr;
753                         totvlak+= dl->parts*(dl->nr-1);
754                 }
755                 else if(dl->type==DL_POLY) {
756                         /* cyclic polys are filled. except when 3D */
757                         if(cu->flag & CU_3D) {
758                                 totvert+= dl->parts*dl->nr;
759                                 totvlak+= dl->parts*dl->nr;
760                         }
761                 }
762                 else if(dl->type==DL_SURF) {
763                         totvert+= dl->parts*dl->nr;
764                         totvlak+= (dl->parts-1+((dl->flag & DL_CYCL_V)==2))*(dl->nr-1+(dl->flag & DL_CYCL_U));
765                 }
766                 else if(dl->type==DL_INDEX3) {
767                         totvert+= dl->nr;
768                         totvlak+= dl->parts;
769                 }
770                 dl= dl->next;
771         }
772         if(totvert==0) {
773                 /* error("can't convert"); */
774                 /* Make Sure you check ob->data is a curve */
775                 return;
776         }
777
778         /* make mesh */
779         me= add_mesh("Mesh");
780         me->totvert= totvert;
781         me->totface= totvlak;
782
783         me->totcol= cu->totcol;
784         me->mat= cu->mat;
785         cu->mat= 0;
786         cu->totcol= 0;
787
788         mvert= CustomData_add_layer(&me->vdata, CD_MVERT, CD_CALLOC, NULL, me->totvert);
789         mface= CustomData_add_layer(&me->fdata, CD_MFACE, CD_CALLOC, NULL, me->totface);
790         me->mvert= mvert;
791         me->mface= mface;
792
793         /* verts and faces */
794         vertcount= 0;
795
796         dl= cu->disp.first;
797         while(dl) {
798                 int smooth= dl->rt & CU_SMOOTH ? 1 : 0;
799                 
800                 if(dl->type==DL_SEGM) {
801                         startvert= vertcount;
802                         a= dl->parts*dl->nr;
803                         data= dl->verts;
804                         while(a--) {
805                                 VECCOPY(mvert->co, data);
806                                 data+=3;
807                                 vertcount++;
808                                 mvert++;
809                         }
810
811                         for(a=0; a<dl->parts; a++) {
812                                 ofs= a*dl->nr;
813                                 for(b=1; b<dl->nr; b++) {
814                                         mface->v1= startvert+ofs+b-1;
815                                         mface->v2= startvert+ofs+b;
816                                         if(smooth) mface->flag |= ME_SMOOTH;
817                                         mface++;
818                                 }
819                         }
820
821                 }
822                 else if(dl->type==DL_POLY) {
823                         /* 3d polys are not filled */
824                         if(cu->flag & CU_3D) {
825                                 startvert= vertcount;
826                                 a= dl->parts*dl->nr;
827                                 data= dl->verts;
828                                 while(a--) {
829                                         VECCOPY(mvert->co, data);
830                                         data+=3;
831                                         vertcount++;
832                                         mvert++;
833                                 }
834         
835                                 for(a=0; a<dl->parts; a++) {
836                                         ofs= a*dl->nr;
837                                         for(b=0; b<dl->nr; b++) {
838                                                 mface->v1= startvert+ofs+b;
839                                                 if(b==dl->nr-1) mface->v2= startvert+ofs;
840                                                 else mface->v2= startvert+ofs+b+1;
841                                                 if(smooth) mface->flag |= ME_SMOOTH;
842                                                 mface++;
843                                         }
844                                 }
845                         }
846                 }
847                 else if(dl->type==DL_INDEX3) {
848                         startvert= vertcount;
849                         a= dl->nr;
850                         data= dl->verts;
851                         while(a--) {
852                                 VECCOPY(mvert->co, data);
853                                 data+=3;
854                                 vertcount++;
855                                 mvert++;
856                         }
857
858                         a= dl->parts;
859                         index= dl->index;
860                         while(a--) {
861                                 mface->v1= startvert+index[0];
862                                 mface->v2= startvert+index[2];
863                                 mface->v3= startvert+index[1];
864                                 mface->v4= 0;
865                                 test_index_face(mface, NULL, 0, 3);
866                                 
867                                 if(smooth) mface->flag |= ME_SMOOTH;
868                                 mface++;
869                                 index+= 3;
870                         }
871         
872         
873                 }
874                 else if(dl->type==DL_SURF) {
875                         startvert= vertcount;
876                         a= dl->parts*dl->nr;
877                         data= dl->verts;
878                         while(a--) {
879                                 VECCOPY(mvert->co, data);
880                                 data+=3;
881                                 vertcount++;
882                                 mvert++;
883                         }
884
885                         for(a=0; a<dl->parts; a++) {
886
887                                 if( (dl->flag & DL_CYCL_V)==0 && a==dl->parts-1) break;
888
889                                 if(dl->flag & DL_CYCL_U) {                      /* p2 -> p1 -> */
890                                         p1= startvert+ dl->nr*a;        /* p4 -> p3 -> */
891                                         p2= p1+ dl->nr-1;               /* -----> next row */
892                                         p3= p1+ dl->nr;
893                                         p4= p2+ dl->nr;
894                                         b= 0;
895                                 }
896                                 else {
897                                         p2= startvert+ dl->nr*a;
898                                         p1= p2+1;
899                                         p4= p2+ dl->nr;
900                                         p3= p1+ dl->nr;
901                                         b= 1;
902                                 }
903                                 if( (dl->flag & DL_CYCL_V) && a==dl->parts-1) {
904                                         p3-= dl->parts*dl->nr;
905                                         p4-= dl->parts*dl->nr;
906                                 }
907
908                                 for(; b<dl->nr; b++) {
909                                         mface->v1= p1;
910                                         mface->v2= p3;
911                                         mface->v3= p4;
912                                         mface->v4= p2;
913                                         mface->mat_nr= (unsigned char)dl->col;
914                                         test_index_face(mface, NULL, 0, 4);
915                                         
916                                         if(smooth) mface->flag |= ME_SMOOTH;
917                                         mface++;
918
919                                         p4= p3; 
920                                         p3++;
921                                         p2= p1; 
922                                         p1++;
923                                 }
924                         }
925
926                 }
927
928                 dl= dl->next;
929         }
930
931         make_edges(me, 0);      // all edges
932         mesh_calc_normals(me->mvert, me->totvert, me->mface, me->totface, NULL);
933
934         if(ob->data) {
935                 free_libblock(&G.main->curve, ob->data);
936         }
937         ob->data= me;
938         ob->type= OB_MESH;
939         
940         /* other users */
941         ob1= G.main->object.first;
942         while(ob1) {
943                 if(ob1->data==cu) {
944                         ob1->type= OB_MESH;
945                 
946                         ob1->data= ob->data;
947                         id_us_plus((ID *)ob->data);
948                 }
949                 ob1= ob1->id.next;
950         }
951
952 }
953
954 void mesh_delete_material_index(Mesh *me, int index)
955 {
956         int i;
957
958         for (i=0; i<me->totface; i++) {
959                 MFace *mf = &((MFace*) me->mface)[i];
960                 if (mf->mat_nr && mf->mat_nr>=index) 
961                         mf->mat_nr--;
962         }
963 }
964
965 void mesh_set_smooth_flag(Object *meshOb, int enableSmooth) 
966 {
967         Mesh *me = meshOb->data;
968         int i;
969
970         for (i=0; i<me->totface; i++) {
971                 MFace *mf = &((MFace*) me->mface)[i];
972
973                 if (enableSmooth) {
974                         mf->flag |= ME_SMOOTH;
975                 } else {
976                         mf->flag &= ~ME_SMOOTH;
977                 }
978         }
979
980 // XXX do this in caller        DAG_object_flush_update(scene, meshOb, OB_RECALC_DATA);
981 }
982
983 void mesh_calc_normals(MVert *mverts, int numVerts, MFace *mfaces, int numFaces, float **faceNors_r) 
984 {
985         float (*tnorms)[3]= MEM_callocN(numVerts*sizeof(*tnorms), "tnorms");
986         float *fnors= MEM_mallocN(sizeof(*fnors)*3*numFaces, "meshnormals");
987         int i;
988
989         for (i=0; i<numFaces; i++) {
990                 MFace *mf= &mfaces[i];
991                 float *f_no= &fnors[i*3];
992
993                 if (mf->v4)
994                         CalcNormFloat4(mverts[mf->v1].co, mverts[mf->v2].co, mverts[mf->v3].co, mverts[mf->v4].co, f_no);
995                 else
996                         CalcNormFloat(mverts[mf->v1].co, mverts[mf->v2].co, mverts[mf->v3].co, f_no);
997                 
998                 VecAddf(tnorms[mf->v1], tnorms[mf->v1], f_no);
999                 VecAddf(tnorms[mf->v2], tnorms[mf->v2], f_no);
1000                 VecAddf(tnorms[mf->v3], tnorms[mf->v3], f_no);
1001                 if (mf->v4)
1002                         VecAddf(tnorms[mf->v4], tnorms[mf->v4], f_no);
1003         }
1004         for (i=0; i<numVerts; i++) {
1005                 MVert *mv= &mverts[i];
1006                 float *no= tnorms[i];
1007                 
1008                 if (Normalize(no)==0.0) {
1009                         VECCOPY(no, mv->co);
1010                         Normalize(no);
1011                 }
1012
1013                 mv->no[0]= (short)(no[0]*32767.0);
1014                 mv->no[1]= (short)(no[1]*32767.0);
1015                 mv->no[2]= (short)(no[2]*32767.0);
1016         }
1017         
1018         MEM_freeN(tnorms);
1019
1020         if (faceNors_r) {
1021                 *faceNors_r = fnors;
1022         } else {
1023                 MEM_freeN(fnors);
1024         }
1025 }
1026
1027 float (*mesh_getVertexCos(Mesh *me, int *numVerts_r))[3]
1028 {
1029         int i, numVerts = me->totvert;
1030         float (*cos)[3] = MEM_mallocN(sizeof(*cos)*numVerts, "vertexcos1");
1031         
1032         if (numVerts_r) *numVerts_r = numVerts;
1033         for (i=0; i<numVerts; i++)
1034                 VECCOPY(cos[i], me->mvert[i].co);
1035         
1036         return cos;
1037 }
1038
1039 float (*mesh_getRefKeyCos(Mesh *me, int *numVerts_r))[3]
1040 {
1041         KeyBlock *kb;
1042         float (*cos)[3] = NULL;
1043         int totvert;
1044         
1045         if(me->key && me->key->refkey) {
1046                 if(numVerts_r) *numVerts_r= me->totvert;
1047                 
1048                 kb= me->key->refkey;
1049                 
1050                 /* prevent accessing invalid memory */
1051                 if (me->totvert > kb->totelem)          cos= MEM_callocN(sizeof(*cos)*me->totvert, "vertexcos1");
1052                 else                                                            cos= MEM_mallocN(sizeof(*cos)*me->totvert, "vertexcos1");
1053                 
1054                 totvert= MIN2(kb->totelem, me->totvert);
1055
1056                 memcpy(cos, kb->data, sizeof(*cos)*totvert);
1057         }
1058
1059         return cos;
1060 }
1061
1062 UvVertMap *make_uv_vert_map(struct MFace *mface, struct MTFace *tface, unsigned int totface, unsigned int totvert, int selected, float *limit)
1063 {
1064         UvVertMap *vmap;
1065         UvMapVert *buf;
1066         MFace *mf;
1067         MTFace *tf;
1068         unsigned int a;
1069         int     i, totuv, nverts;
1070
1071         totuv = 0;
1072
1073         /* generate UvMapVert array */
1074         mf= mface;
1075         tf= tface;
1076         for(a=0; a<totface; a++, mf++, tf++)
1077                 if(!selected || (!(mf->flag & ME_HIDE) && (mf->flag & ME_FACE_SEL)))
1078                         totuv += (mf->v4)? 4: 3;
1079                 
1080         if(totuv==0)
1081                 return NULL;
1082         
1083         vmap= (UvVertMap*)MEM_callocN(sizeof(*vmap), "UvVertMap");
1084         if (!vmap)
1085                 return NULL;
1086
1087         vmap->vert= (UvMapVert**)MEM_callocN(sizeof(*vmap->vert)*totvert, "UvMapVert*");
1088         buf= vmap->buf= (UvMapVert*)MEM_callocN(sizeof(*vmap->buf)*totuv, "UvMapVert");
1089
1090         if (!vmap->vert || !vmap->buf) {
1091                 free_uv_vert_map(vmap);
1092                 return NULL;
1093         }
1094
1095         mf= mface;
1096         tf= tface;
1097         for(a=0; a<totface; a++, mf++, tf++) {
1098                 if(!selected || (!(mf->flag & ME_HIDE) && (mf->flag & ME_FACE_SEL))) {
1099                         nverts= (mf->v4)? 4: 3;
1100
1101                         for(i=0; i<nverts; i++) {
1102                                 buf->tfindex= i;
1103                                 buf->f= a;
1104                                 buf->separate = 0;
1105                                 buf->next= vmap->vert[*(&mf->v1 + i)];
1106                                 vmap->vert[*(&mf->v1 + i)]= buf;
1107                                 buf++;
1108                         }
1109                 }
1110         }
1111         
1112         /* sort individual uvs for each vert */
1113         tf= tface;
1114         for(a=0; a<totvert; a++) {
1115                 UvMapVert *newvlist= NULL, *vlist=vmap->vert[a];
1116                 UvMapVert *iterv, *v, *lastv, *next;
1117                 float *uv, *uv2, uvdiff[2];
1118
1119                 while(vlist) {
1120                         v= vlist;
1121                         vlist= vlist->next;
1122                         v->next= newvlist;
1123                         newvlist= v;
1124
1125                         uv= (tf+v->f)->uv[v->tfindex];
1126                         lastv= NULL;
1127                         iterv= vlist;
1128
1129                         while(iterv) {
1130                                 next= iterv->next;
1131
1132                                 uv2= (tf+iterv->f)->uv[iterv->tfindex];
1133                                 Vec2Subf(uvdiff, uv2, uv);
1134
1135
1136                                 if(fabs(uv[0]-uv2[0]) < limit[0] && fabs(uv[1]-uv2[1]) < limit[1]) {
1137                                         if(lastv) lastv->next= next;
1138                                         else vlist= next;
1139                                         iterv->next= newvlist;
1140                                         newvlist= iterv;
1141                                 }
1142                                 else
1143                                         lastv=iterv;
1144
1145                                 iterv= next;
1146                         }
1147
1148                         newvlist->separate = 1;
1149                 }
1150
1151                 vmap->vert[a]= newvlist;
1152         }
1153         
1154         return vmap;
1155 }
1156
1157 UvMapVert *get_uv_map_vert(UvVertMap *vmap, unsigned int v)
1158 {
1159         return vmap->vert[v];
1160 }
1161
1162 void free_uv_vert_map(UvVertMap *vmap)
1163 {
1164         if (vmap) {
1165                 if (vmap->vert) MEM_freeN(vmap->vert);
1166                 if (vmap->buf) MEM_freeN(vmap->buf);
1167                 MEM_freeN(vmap);
1168         }
1169 }
1170
1171 /* Generates a map where the key is the vertex and the value is a list
1172    of faces that use that vertex as a corner. The lists are allocated
1173    from one memory pool. */
1174 void create_vert_face_map(ListBase **map, IndexNode **mem, const MFace *mface, const int totvert, const int totface)
1175 {
1176         int i,j;
1177         IndexNode *node = NULL;
1178         
1179         (*map) = MEM_callocN(sizeof(ListBase) * totvert, "vert face map");
1180         (*mem) = MEM_callocN(sizeof(IndexNode) * totface*4, "vert face map mem");
1181         node = *mem;
1182         
1183         /* Find the users */
1184         for(i = 0; i < totface; ++i){
1185                 for(j = 0; j < (mface[i].v4?4:3); ++j, ++node) {
1186                         node->index = i;
1187                         BLI_addtail(&(*map)[((unsigned int*)(&mface[i]))[j]], node);
1188                 }
1189         }
1190 }
1191
1192 /* Generates a map where the key is the vertex and the value is a list
1193    of edges that use that vertex as an endpoint. The lists are allocated
1194    from one memory pool. */
1195 void create_vert_edge_map(ListBase **map, IndexNode **mem, const MEdge *medge, const int totvert, const int totedge)
1196 {
1197         int i, j;
1198         IndexNode *node = NULL;
1199  
1200         (*map) = MEM_callocN(sizeof(ListBase) * totvert, "vert edge map");
1201         (*mem) = MEM_callocN(sizeof(IndexNode) * totedge * 2, "vert edge map mem");
1202         node = *mem;
1203        
1204         /* Find the users */
1205         for(i = 0; i < totedge; ++i){
1206                 for(j = 0; j < 2; ++j, ++node) {
1207                         node->index = i;
1208                         BLI_addtail(&(*map)[((unsigned int*)(&medge[i].v1))[j]], node);
1209                 }
1210         }
1211 }
1212
1213 /* Partial Mesh Visibility */
1214 PartialVisibility *mesh_pmv_copy(PartialVisibility *pmv)
1215 {
1216         PartialVisibility *n= MEM_dupallocN(pmv);
1217         n->vert_map= MEM_dupallocN(pmv->vert_map);
1218         n->edge_map= MEM_dupallocN(pmv->edge_map);
1219         n->old_edges= MEM_dupallocN(pmv->old_edges);
1220         n->old_faces= MEM_dupallocN(pmv->old_faces);
1221         return n;
1222 }
1223
1224 void mesh_pmv_free(PartialVisibility *pv)
1225 {
1226         MEM_freeN(pv->vert_map);
1227         MEM_freeN(pv->edge_map);
1228         MEM_freeN(pv->old_faces);
1229         MEM_freeN(pv->old_edges);
1230         MEM_freeN(pv);
1231 }
1232
1233 void mesh_pmv_revert(Object *ob, Mesh *me)
1234 {
1235         if(me->pv) {
1236                 unsigned i;
1237                 MVert *nve, *old_verts;
1238                 
1239                 /* Reorder vertices */
1240                 nve= me->mvert;
1241                 old_verts = MEM_mallocN(sizeof(MVert)*me->pv->totvert,"PMV revert verts");
1242                 for(i=0; i<me->pv->totvert; ++i)
1243                         old_verts[i]= nve[me->pv->vert_map[i]];
1244
1245                 /* Restore verts, edges and faces */
1246                 CustomData_free_layer_active(&me->vdata, CD_MVERT, me->totvert);
1247                 CustomData_free_layer_active(&me->edata, CD_MEDGE, me->totedge);
1248                 CustomData_free_layer_active(&me->fdata, CD_MFACE, me->totface);
1249
1250                 CustomData_add_layer(&me->vdata, CD_MVERT, CD_ASSIGN, old_verts, me->pv->totvert);
1251                 CustomData_add_layer(&me->edata, CD_MEDGE, CD_ASSIGN, me->pv->old_edges, me->pv->totedge);
1252                 CustomData_add_layer(&me->fdata, CD_MFACE, CD_ASSIGN, me->pv->old_faces, me->pv->totface);
1253                 mesh_update_customdata_pointers(me);
1254
1255                 me->totvert= me->pv->totvert;
1256                 me->totedge= me->pv->totedge;
1257                 me->totface= me->pv->totface;
1258
1259                 me->pv->old_edges= NULL;
1260                 me->pv->old_faces= NULL;
1261
1262                 /* Free maps */
1263                 MEM_freeN(me->pv->edge_map);
1264                 me->pv->edge_map= NULL;
1265                 MEM_freeN(me->pv->vert_map);
1266                 me->pv->vert_map= NULL;
1267
1268 // XXX do this in caller                DAG_object_flush_update(scene, ob, OB_RECALC_DATA);
1269         }
1270 }
1271
1272 void mesh_pmv_off(Object *ob, Mesh *me)
1273 {
1274         if(ob && me->pv) {
1275                 mesh_pmv_revert(ob, me);
1276                 MEM_freeN(me->pv);
1277                 me->pv= NULL;
1278         }
1279 }