changes to the sequencer in trunk between 20855:20928, (excluding UI changes)
[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
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                 if(dl->type==DL_SEGM) {
799                         startvert= vertcount;
800                         a= dl->parts*dl->nr;
801                         data= dl->verts;
802                         while(a--) {
803                                 VECCOPY(mvert->co, data);
804                                 data+=3;
805                                 vertcount++;
806                                 mvert++;
807                         }
808
809                         for(a=0; a<dl->parts; a++) {
810                                 ofs= a*dl->nr;
811                                 for(b=1; b<dl->nr; b++) {
812                                         mface->v1= startvert+ofs+b-1;
813                                         mface->v2= startvert+ofs+b;
814                                         mface++;
815                                 }
816                         }
817
818                 }
819                 else if(dl->type==DL_POLY) {
820                         /* 3d polys are not filled */
821                         if(cu->flag & CU_3D) {
822                                 startvert= vertcount;
823                                 a= dl->parts*dl->nr;
824                                 data= dl->verts;
825                                 while(a--) {
826                                         VECCOPY(mvert->co, data);
827                                         data+=3;
828                                         vertcount++;
829                                         mvert++;
830                                 }
831         
832                                 for(a=0; a<dl->parts; a++) {
833                                         ofs= a*dl->nr;
834                                         for(b=0; b<dl->nr; b++) {
835                                                 mface->v1= startvert+ofs+b;
836                                                 if(b==dl->nr-1) mface->v2= startvert+ofs;
837                                                 else mface->v2= startvert+ofs+b+1;
838                                                 mface++;
839                                         }
840                                 }
841                         }
842                 }
843                 else if(dl->type==DL_INDEX3) {
844                         startvert= vertcount;
845                         a= dl->nr;
846                         data= dl->verts;
847                         while(a--) {
848                                 VECCOPY(mvert->co, data);
849                                 data+=3;
850                                 vertcount++;
851                                 mvert++;
852                         }
853
854                         a= dl->parts;
855                         index= dl->index;
856                         while(a--) {
857                                 mface->v1= startvert+index[0];
858                                 mface->v2= startvert+index[2];
859                                 mface->v3= startvert+index[1];
860                                 mface->v4= 0;
861                                 test_index_face(mface, NULL, 0, 3);
862                                 
863                                 mface++;
864                                 index+= 3;
865                         }
866         
867         
868                 }
869                 else if(dl->type==DL_SURF) {
870                         startvert= vertcount;
871                         a= dl->parts*dl->nr;
872                         data= dl->verts;
873                         while(a--) {
874                                 VECCOPY(mvert->co, data);
875                                 data+=3;
876                                 vertcount++;
877                                 mvert++;
878                         }
879
880                         for(a=0; a<dl->parts; a++) {
881
882                                 if( (dl->flag & DL_CYCL_V)==0 && a==dl->parts-1) break;
883
884                                 if(dl->flag & DL_CYCL_U) {                      /* p2 -> p1 -> */
885                                         p1= startvert+ dl->nr*a;        /* p4 -> p3 -> */
886                                         p2= p1+ dl->nr-1;               /* -----> next row */
887                                         p3= p1+ dl->nr;
888                                         p4= p2+ dl->nr;
889                                         b= 0;
890                                 }
891                                 else {
892                                         p2= startvert+ dl->nr*a;
893                                         p1= p2+1;
894                                         p4= p2+ dl->nr;
895                                         p3= p1+ dl->nr;
896                                         b= 1;
897                                 }
898                                 if( (dl->flag & DL_CYCL_V) && a==dl->parts-1) {
899                                         p3-= dl->parts*dl->nr;
900                                         p4-= dl->parts*dl->nr;
901                                 }
902
903                                 for(; b<dl->nr; b++) {
904                                         mface->v1= p1;
905                                         mface->v2= p3;
906                                         mface->v3= p4;
907                                         mface->v4= p2;
908                                         mface->mat_nr= (unsigned char)dl->col;
909                                         test_index_face(mface, NULL, 0, 4);
910                                         mface++;
911
912                                         p4= p3; 
913                                         p3++;
914                                         p2= p1; 
915                                         p1++;
916                                 }
917                         }
918
919                 }
920
921                 dl= dl->next;
922         }
923
924         make_edges(me, 0);      // all edges
925         mesh_calc_normals(me->mvert, me->totvert, me->mface, me->totface, NULL);
926
927         if(ob->data) {
928                 free_libblock(&G.main->curve, ob->data);
929         }
930         ob->data= me;
931         ob->type= OB_MESH;
932         
933         /* other users */
934         ob1= G.main->object.first;
935         while(ob1) {
936                 if(ob1->data==cu) {
937                         ob1->type= OB_MESH;
938                 
939                         ob1->data= ob->data;
940                         id_us_plus((ID *)ob->data);
941                 }
942                 ob1= ob1->id.next;
943         }
944
945 }
946
947 void mesh_delete_material_index(Mesh *me, int index)
948 {
949         int i;
950
951         for (i=0; i<me->totface; i++) {
952                 MFace *mf = &((MFace*) me->mface)[i];
953                 if (mf->mat_nr && mf->mat_nr>=index) 
954                         mf->mat_nr--;
955         }
956 }
957
958 void mesh_set_smooth_flag(Object *meshOb, int enableSmooth) 
959 {
960         Mesh *me = meshOb->data;
961         int i;
962
963         for (i=0; i<me->totface; i++) {
964                 MFace *mf = &((MFace*) me->mface)[i];
965
966                 if (enableSmooth) {
967                         mf->flag |= ME_SMOOTH;
968                 } else {
969                         mf->flag &= ~ME_SMOOTH;
970                 }
971         }
972
973 // XXX do this in caller        DAG_object_flush_update(scene, meshOb, OB_RECALC_DATA);
974 }
975
976 void mesh_calc_normals(MVert *mverts, int numVerts, MFace *mfaces, int numFaces, float **faceNors_r) 
977 {
978         float (*tnorms)[3]= MEM_callocN(numVerts*sizeof(*tnorms), "tnorms");
979         float *fnors= MEM_mallocN(sizeof(*fnors)*3*numFaces, "meshnormals");
980         int i;
981
982         for (i=0; i<numFaces; i++) {
983                 MFace *mf= &mfaces[i];
984                 float *f_no= &fnors[i*3];
985
986                 if (mf->v4)
987                         CalcNormFloat4(mverts[mf->v1].co, mverts[mf->v2].co, mverts[mf->v3].co, mverts[mf->v4].co, f_no);
988                 else
989                         CalcNormFloat(mverts[mf->v1].co, mverts[mf->v2].co, mverts[mf->v3].co, f_no);
990                 
991                 VecAddf(tnorms[mf->v1], tnorms[mf->v1], f_no);
992                 VecAddf(tnorms[mf->v2], tnorms[mf->v2], f_no);
993                 VecAddf(tnorms[mf->v3], tnorms[mf->v3], f_no);
994                 if (mf->v4)
995                         VecAddf(tnorms[mf->v4], tnorms[mf->v4], f_no);
996         }
997         for (i=0; i<numVerts; i++) {
998                 MVert *mv= &mverts[i];
999                 float *no= tnorms[i];
1000                 
1001                 if (Normalize(no)==0.0) {
1002                         VECCOPY(no, mv->co);
1003                         Normalize(no);
1004                 }
1005
1006                 mv->no[0]= (short)(no[0]*32767.0);
1007                 mv->no[1]= (short)(no[1]*32767.0);
1008                 mv->no[2]= (short)(no[2]*32767.0);
1009         }
1010         
1011         MEM_freeN(tnorms);
1012
1013         if (faceNors_r) {
1014                 *faceNors_r = fnors;
1015         } else {
1016                 MEM_freeN(fnors);
1017         }
1018 }
1019
1020 float (*mesh_getVertexCos(Mesh *me, int *numVerts_r))[3]
1021 {
1022         int i, numVerts = me->totvert;
1023         float (*cos)[3] = MEM_mallocN(sizeof(*cos)*numVerts, "vertexcos1");
1024         
1025         if (numVerts_r) *numVerts_r = numVerts;
1026         for (i=0; i<numVerts; i++)
1027                 VECCOPY(cos[i], me->mvert[i].co);
1028         
1029         return cos;
1030 }
1031
1032 float (*mesh_getRefKeyCos(Mesh *me, int *numVerts_r))[3]
1033 {
1034         KeyBlock *kb;
1035         float (*cos)[3] = NULL;
1036         int totvert;
1037         
1038         if(me->key && me->key->refkey) {
1039                 if(numVerts_r) *numVerts_r= me->totvert;
1040                 
1041                 kb= me->key->refkey;
1042                 
1043                 /* prevent accessing invalid memory */
1044                 if (me->totvert > kb->totelem)          cos= MEM_callocN(sizeof(*cos)*me->totvert, "vertexcos1");
1045                 else                                                            cos= MEM_mallocN(sizeof(*cos)*me->totvert, "vertexcos1");
1046                 
1047                 totvert= MIN2(kb->totelem, me->totvert);
1048
1049                 memcpy(cos, kb->data, sizeof(*cos)*totvert);
1050         }
1051
1052         return cos;
1053 }
1054
1055 UvVertMap *make_uv_vert_map(struct MFace *mface, struct MTFace *tface, unsigned int totface, unsigned int totvert, int selected, float *limit)
1056 {
1057         UvVertMap *vmap;
1058         UvMapVert *buf;
1059         MFace *mf;
1060         MTFace *tf;
1061         unsigned int a;
1062         int     i, totuv, nverts;
1063
1064         totuv = 0;
1065
1066         /* generate UvMapVert array */
1067         mf= mface;
1068         tf= tface;
1069         for(a=0; a<totface; a++, mf++, tf++)
1070                 if(!selected || (!(mf->flag & ME_HIDE) && (mf->flag & ME_FACE_SEL)))
1071                         totuv += (mf->v4)? 4: 3;
1072                 
1073         if(totuv==0)
1074                 return NULL;
1075         
1076         vmap= (UvVertMap*)MEM_callocN(sizeof(*vmap), "UvVertMap");
1077         if (!vmap)
1078                 return NULL;
1079
1080         vmap->vert= (UvMapVert**)MEM_callocN(sizeof(*vmap->vert)*totvert, "UvMapVert*");
1081         buf= vmap->buf= (UvMapVert*)MEM_callocN(sizeof(*vmap->buf)*totuv, "UvMapVert");
1082
1083         if (!vmap->vert || !vmap->buf) {
1084                 free_uv_vert_map(vmap);
1085                 return NULL;
1086         }
1087
1088         mf= mface;
1089         tf= tface;
1090         for(a=0; a<totface; a++, mf++, tf++) {
1091                 if(!selected || (!(mf->flag & ME_HIDE) && (mf->flag & ME_FACE_SEL))) {
1092                         nverts= (mf->v4)? 4: 3;
1093
1094                         for(i=0; i<nverts; i++) {
1095                                 buf->tfindex= i;
1096                                 buf->f= a;
1097                                 buf->separate = 0;
1098                                 buf->next= vmap->vert[*(&mf->v1 + i)];
1099                                 vmap->vert[*(&mf->v1 + i)]= buf;
1100                                 buf++;
1101                         }
1102                 }
1103         }
1104         
1105         /* sort individual uvs for each vert */
1106         tf= tface;
1107         for(a=0; a<totvert; a++) {
1108                 UvMapVert *newvlist= NULL, *vlist=vmap->vert[a];
1109                 UvMapVert *iterv, *v, *lastv, *next;
1110                 float *uv, *uv2, uvdiff[2];
1111
1112                 while(vlist) {
1113                         v= vlist;
1114                         vlist= vlist->next;
1115                         v->next= newvlist;
1116                         newvlist= v;
1117
1118                         uv= (tf+v->f)->uv[v->tfindex];
1119                         lastv= NULL;
1120                         iterv= vlist;
1121
1122                         while(iterv) {
1123                                 next= iterv->next;
1124
1125                                 uv2= (tf+iterv->f)->uv[iterv->tfindex];
1126                                 Vec2Subf(uvdiff, uv2, uv);
1127
1128
1129                                 if(fabs(uv[0]-uv2[0]) < limit[0] && fabs(uv[1]-uv2[1]) < limit[1]) {
1130                                         if(lastv) lastv->next= next;
1131                                         else vlist= next;
1132                                         iterv->next= newvlist;
1133                                         newvlist= iterv;
1134                                 }
1135                                 else
1136                                         lastv=iterv;
1137
1138                                 iterv= next;
1139                         }
1140
1141                         newvlist->separate = 1;
1142                 }
1143
1144                 vmap->vert[a]= newvlist;
1145         }
1146         
1147         return vmap;
1148 }
1149
1150 UvMapVert *get_uv_map_vert(UvVertMap *vmap, unsigned int v)
1151 {
1152         return vmap->vert[v];
1153 }
1154
1155 void free_uv_vert_map(UvVertMap *vmap)
1156 {
1157         if (vmap) {
1158                 if (vmap->vert) MEM_freeN(vmap->vert);
1159                 if (vmap->buf) MEM_freeN(vmap->buf);
1160                 MEM_freeN(vmap);
1161         }
1162 }
1163
1164 /* Generates a map where the key is the vertex and the value is a list
1165    of faces that use that vertex as a corner. The lists are allocated
1166    from one memory pool. */
1167 void create_vert_face_map(ListBase **map, IndexNode **mem, const MFace *mface, const int totvert, const int totface)
1168 {
1169         int i,j;
1170         IndexNode *node = NULL;
1171         
1172         (*map) = MEM_callocN(sizeof(ListBase) * totvert, "vert face map");
1173         (*mem) = MEM_callocN(sizeof(IndexNode) * totface*4, "vert face map mem");
1174         node = *mem;
1175         
1176         /* Find the users */
1177         for(i = 0; i < totface; ++i){
1178                 for(j = 0; j < (mface[i].v4?4:3); ++j, ++node) {
1179                         node->index = i;
1180                         BLI_addtail(&(*map)[((unsigned int*)(&mface[i]))[j]], node);
1181                 }
1182         }
1183 }
1184
1185 /* Generates a map where the key is the vertex and the value is a list
1186    of edges that use that vertex as an endpoint. The lists are allocated
1187    from one memory pool. */
1188 void create_vert_edge_map(ListBase **map, IndexNode **mem, const MEdge *medge, const int totvert, const int totedge)
1189 {
1190         int i, j;
1191         IndexNode *node = NULL;
1192  
1193         (*map) = MEM_callocN(sizeof(ListBase) * totvert, "vert edge map");
1194         (*mem) = MEM_callocN(sizeof(IndexNode) * totedge * 2, "vert edge map mem");
1195         node = *mem;
1196        
1197         /* Find the users */
1198         for(i = 0; i < totedge; ++i){
1199                 for(j = 0; j < 2; ++j, ++node) {
1200                         node->index = i;
1201                         BLI_addtail(&(*map)[((unsigned int*)(&medge[i].v1))[j]], node);
1202                 }
1203         }
1204 }
1205
1206 /* Partial Mesh Visibility */
1207 PartialVisibility *mesh_pmv_copy(PartialVisibility *pmv)
1208 {
1209         PartialVisibility *n= MEM_dupallocN(pmv);
1210         n->vert_map= MEM_dupallocN(pmv->vert_map);
1211         n->edge_map= MEM_dupallocN(pmv->edge_map);
1212         n->old_edges= MEM_dupallocN(pmv->old_edges);
1213         n->old_faces= MEM_dupallocN(pmv->old_faces);
1214         return n;
1215 }
1216
1217 void mesh_pmv_free(PartialVisibility *pv)
1218 {
1219         MEM_freeN(pv->vert_map);
1220         MEM_freeN(pv->edge_map);
1221         MEM_freeN(pv->old_faces);
1222         MEM_freeN(pv->old_edges);
1223         MEM_freeN(pv);
1224 }
1225
1226 void mesh_pmv_revert(Object *ob, Mesh *me)
1227 {
1228         if(me->pv) {
1229                 unsigned i;
1230                 MVert *nve, *old_verts;
1231                 
1232                 /* Reorder vertices */
1233                 nve= me->mvert;
1234                 old_verts = MEM_mallocN(sizeof(MVert)*me->pv->totvert,"PMV revert verts");
1235                 for(i=0; i<me->pv->totvert; ++i)
1236                         old_verts[i]= nve[me->pv->vert_map[i]];
1237
1238                 /* Restore verts, edges and faces */
1239                 CustomData_free_layer_active(&me->vdata, CD_MVERT, me->totvert);
1240                 CustomData_free_layer_active(&me->edata, CD_MEDGE, me->totedge);
1241                 CustomData_free_layer_active(&me->fdata, CD_MFACE, me->totface);
1242
1243                 CustomData_add_layer(&me->vdata, CD_MVERT, CD_ASSIGN, old_verts, me->pv->totvert);
1244                 CustomData_add_layer(&me->edata, CD_MEDGE, CD_ASSIGN, me->pv->old_edges, me->pv->totedge);
1245                 CustomData_add_layer(&me->fdata, CD_MFACE, CD_ASSIGN, me->pv->old_faces, me->pv->totface);
1246                 mesh_update_customdata_pointers(me);
1247
1248                 me->totvert= me->pv->totvert;
1249                 me->totedge= me->pv->totedge;
1250                 me->totface= me->pv->totface;
1251
1252                 me->pv->old_edges= NULL;
1253                 me->pv->old_faces= NULL;
1254
1255                 /* Free maps */
1256                 MEM_freeN(me->pv->edge_map);
1257                 me->pv->edge_map= NULL;
1258                 MEM_freeN(me->pv->vert_map);
1259                 me->pv->vert_map= NULL;
1260
1261 // XXX do this in caller                DAG_object_flush_update(scene, ob, OB_RECALC_DATA);
1262         }
1263 }
1264
1265 void mesh_pmv_off(Object *ob, Mesh *me)
1266 {
1267         if(ob && me->pv) {
1268                 mesh_pmv_revert(ob, me);
1269                 MEM_freeN(me->pv);
1270                 me->pv= NULL;
1271         }
1272 }