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