bmesh minor refactor
[blender.git] / source / blender / blenkernel / intern / modifiers_bmesh.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2005 by the Blender Foundation.
19  * All rights reserved.
20  *
21  * Contributor(s): Joseph Eagar
22  *
23  * ***** END GPL LICENSE BLOCK *****
24  *
25  */
26
27 /** \file blender/blenkernel/intern/modifiers_bmesh.c
28  *  \ingroup bke
29  */
30
31 #include "BLI_math.h"
32
33 #include "MEM_guardedalloc.h"
34
35 #include "DNA_object_types.h"
36
37 #include "BLI_array.h"
38
39 #include "BKE_bmesh.h"
40 #include "BKE_tessmesh.h"
41
42 /* main function for copying DerivedMesh data into BMesh */
43 void DM_to_bmesh_ex(DerivedMesh *dm, BMesh *bm)
44 {
45         MVert *mv, *mvert;
46         MEdge *me, *medge;
47         MPoly *mpoly, *mp;
48         MLoop *mloop, *ml;
49         BMVert *v, **vtable, **verts = NULL;
50         BMEdge *e, **etable, **edges = NULL;
51         BMFace *f;
52         BMIter liter;
53         BLI_array_declare(verts);
54         BLI_array_declare(edges);
55         int i, j, k, totvert, totedge, totface;
56
57         /*merge custom data layout*/
58         CustomData_bmesh_merge(&dm->vertData, &bm->vdata, CD_MASK_DERIVEDMESH, CD_CALLOC, bm, BM_VERT);
59         CustomData_bmesh_merge(&dm->edgeData, &bm->edata, CD_MASK_DERIVEDMESH, CD_CALLOC, bm, BM_EDGE);
60         CustomData_bmesh_merge(&dm->loopData, &bm->ldata, CD_MASK_DERIVEDMESH, CD_CALLOC, bm, BM_LOOP);
61         CustomData_bmesh_merge(&dm->polyData, &bm->pdata, CD_MASK_DERIVEDMESH, CD_CALLOC, bm, BM_FACE);
62
63         totvert = dm->getNumVerts(dm);
64         totedge = dm->getNumEdges(dm);
65         totface = dm->getNumPolys(dm);
66
67         vtable = MEM_callocN(sizeof(void**) * totvert, "vert table in BMDM_Copy");
68         etable = MEM_callocN(sizeof(void**) * totedge, "edge table in BMDM_Copy");
69
70         /*do verts*/
71         mv = mvert = dm->dupVertArray(dm);
72         for (i = 0; i < totvert; i++, mv++) {
73                 v = BM_vert_create(bm, mv->co, NULL);
74                 normal_short_to_float_v3(v->no, mv->no);
75                 v->head.hflag = BM_vert_flag_from_mflag(mv->flag);
76
77                 CustomData_to_bmesh_block(&dm->vertData, &bm->vdata, i, &v->head.data);
78                 vtable[i] = v;
79         }
80         MEM_freeN(mvert);
81
82         /*do edges*/
83         me = medge = dm->dupEdgeArray(dm);
84         for (i = 0; i < totedge; i++, me++) {
85                 e = BM_edge_create(bm, vtable[me->v1], vtable[me->v2], NULL, FALSE);
86
87                 e->head.hflag = BM_edge_flag_from_mflag(me->flag);
88
89                 CustomData_to_bmesh_block(&dm->edgeData, &bm->edata, i, &e->head.data);
90                 etable[i] = e;
91         }
92         MEM_freeN(medge);
93
94         /*do faces*/
95         mpoly = mp = dm->getPolyArray(dm);
96         mloop = dm->getLoopArray(dm);
97         for (i = 0; i < dm->numPolyData; i++, mp++) {
98                 BMLoop *l;
99
100                 BLI_array_empty(verts);
101                 BLI_array_empty(edges);
102
103                 BLI_array_growitems(verts, mp->totloop);
104                 BLI_array_growitems(edges, mp->totloop);
105
106                 ml = mloop + mp->loopstart;
107                 for (j = 0; j < mp->totloop; j++, ml++) {
108
109                         verts[j] = vtable[ml->v];
110                         edges[j] = etable[ml->e];
111                 }
112
113                 f = BM_face_create_ngon(bm, verts[0], verts[1], edges, mp->totloop, FALSE);
114
115                 if (!f)
116                         continue;
117
118                 f->head.hflag = BM_face_flag_from_mflag(mp->flag);
119                 f->mat_nr = mp->mat_nr;
120
121                 l = BM_iter_new(&liter, bm, BM_LOOPS_OF_FACE, f);
122                 k = mp->loopstart;
123
124                 for (j = 0; l; l = BM_iter_step(&liter), k++) {
125                         CustomData_to_bmesh_block(&dm->loopData, &bm->ldata, k, &l->head.data);
126                 }
127
128                 CustomData_to_bmesh_block(&dm->polyData, &bm->pdata, i, &f->head.data);
129         }
130
131         MEM_freeN(vtable);
132         MEM_freeN(etable);
133
134         BLI_array_free(verts);
135         BLI_array_free(edges);
136 }
137
138 /* converts a cddm to a BMEditMesh.  if existing is non-NULL, the
139  * new geometry will be put in there.*/
140 BMEditMesh *DM_to_editbmesh(Object *ob, DerivedMesh *dm, BMEditMesh *existing, int do_tesselate)
141 {
142         BMEditMesh *em = existing;
143         BMesh *bm;
144
145         if (em) bm = em->bm;
146         else bm = BM_mesh_create(ob, bm_mesh_allocsize_default);
147
148         DM_to_bmesh_ex(dm, bm);
149
150         if (!em) {
151                 em = BMEdit_Create(bm, do_tesselate);
152         }
153         else {
154                 if (do_tesselate) {
155                         BMEdit_RecalcTesselation(em);
156                 }
157         }
158
159         return em;
160 }
161
162 BMesh *DM_to_bmesh(Object *ob, DerivedMesh *dm)
163 {
164         BMesh *bm;
165
166         bm = BM_mesh_create(ob, bm_mesh_allocsize_default);
167
168         DM_to_bmesh_ex(dm, bm);
169
170         return bm;
171 }