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