Merge branch 'blender2.7'
[blender.git] / source / blender / bmesh / intern / bmesh_construct.c
1 /*
2  * This program is free software; you can redistribute it and/or
3  * modify it under the terms of the GNU General Public License
4  * as published by the Free Software Foundation; either version 2
5  * of the License, or (at your option) any later version.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * along with this program; if not, write to the Free Software Foundation,
14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
15  *
16  * The Original Code is Copyright (C) 2007 Blender Foundation.
17  * All rights reserved.
18  */
19
20 /** \file \ingroup bmesh
21  *
22  * BM construction functions.
23  */
24
25 #include "MEM_guardedalloc.h"
26
27 #include "BLI_alloca.h"
28 #include "BLI_math.h"
29 #include "BLI_sort_utils.h"
30
31 #include "BKE_customdata.h"
32
33 #include "DNA_meshdata_types.h"
34
35 #include "bmesh.h"
36 #include "intern/bmesh_private.h"
37
38 #define SELECT 1
39
40
41 /**
42  * Fill in a vertex array from an edge array.
43  *
44  * \returns false if any verts aren't found.
45  */
46 bool BM_verts_from_edges(BMVert **vert_arr, BMEdge **edge_arr, const int len)
47 {
48         int i, i_prev = len - 1;
49         for (i = 0; i < len; i++) {
50                 vert_arr[i] = BM_edge_share_vert(edge_arr[i_prev], edge_arr[i]);
51                 if (vert_arr[i] == NULL) {
52                         return false;
53                 }
54                 i_prev = i;
55         }
56         return true;
57 }
58
59 /**
60  * Fill in an edge array from a vertex array (connected polygon loop).
61  *
62  * \returns false if any edges aren't found.
63  */
64 bool BM_edges_from_verts(BMEdge **edge_arr, BMVert **vert_arr, const int len)
65 {
66         int i, i_prev = len - 1;
67         for (i = 0; i < len; i++) {
68                 edge_arr[i_prev] = BM_edge_exists(vert_arr[i_prev], vert_arr[i]);
69                 if (edge_arr[i_prev] == NULL) {
70                         return false;
71                 }
72                 i_prev = i;
73         }
74         return true;
75 }
76
77 /**
78  * Fill in an edge array from a vertex array (connected polygon loop).
79  * Creating edges as-needed.
80  */
81 void BM_edges_from_verts_ensure(BMesh *bm, BMEdge **edge_arr, BMVert **vert_arr, const int len)
82 {
83         int i, i_prev = len - 1;
84         for (i = 0; i < len; i++) {
85                 edge_arr[i_prev] = BM_edge_create(bm, vert_arr[i_prev], vert_arr[i], NULL, BM_CREATE_NO_DOUBLE);
86                 i_prev = i;
87         }
88 }
89
90 /* prototypes */
91 static void bm_loop_attrs_copy(
92         BMesh *source_mesh, BMesh *target_mesh,
93         const BMLoop *source_loop, BMLoop *target_loop, uint64_t cd_mask);
94
95 /**
96  * \brief Make Quad/Triangle
97  *
98  * Creates a new quad or triangle from a list of 3 or 4 vertices.
99  * If \a no_double is true, then a check is done to see if a face
100  * with these vertices already exists and returns it instead.
101  *
102  * If a pointer to an example face is provided, it's custom data
103  * and properties will be copied to the new face.
104  *
105  * \note The winding of the face is determined by the order
106  * of the vertices in the vertex array.
107  */
108
109 BMFace *BM_face_create_quad_tri(
110         BMesh *bm,
111         BMVert *v1, BMVert *v2, BMVert *v3, BMVert *v4,
112         const BMFace *f_example, const eBMCreateFlag create_flag)
113 {
114         BMVert *vtar[4] = {v1, v2, v3, v4};
115         return BM_face_create_verts(bm, vtar, v4 ? 4 : 3, f_example, create_flag, true);
116 }
117
118 /**
119  * \brief copies face loop data from shared adjacent faces.
120  *
121  * \param filter_fn: A function that filters the source loops before copying (don't always want to copy all)
122  *
123  * \note when a matching edge is found, both loops of that edge are copied
124  * this is done since the face may not be completely surrounded by faces,
125  * this way: a quad with 2 connected quads on either side will still get all 4 loops updated
126  */
127 void BM_face_copy_shared(
128         BMesh *bm, BMFace *f,
129         BMLoopFilterFunc filter_fn, void *user_data)
130 {
131         BMLoop *l_first;
132         BMLoop *l_iter;
133
134 #ifdef DEBUG
135         l_iter = l_first = BM_FACE_FIRST_LOOP(f);
136         do {
137                 BLI_assert(BM_ELEM_API_FLAG_TEST(l_iter, _FLAG_OVERLAP) == 0);
138         } while ((l_iter = l_iter->next) != l_first);
139 #endif
140
141         l_iter = l_first = BM_FACE_FIRST_LOOP(f);
142         do {
143                 BMLoop *l_other = l_iter->radial_next;
144
145                 if (l_other && l_other != l_iter) {
146                         BMLoop *l_src[2];
147                         BMLoop *l_dst[2] = {l_iter, l_iter->next};
148                         uint j;
149
150                         if (l_other->v == l_iter->v) {
151                                 l_src[0] = l_other;
152                                 l_src[1] = l_other->next;
153                         }
154                         else {
155                                 l_src[0] = l_other->next;
156                                 l_src[1] = l_other;
157                         }
158
159                         for (j = 0; j < 2; j++) {
160                                 BLI_assert(l_dst[j]->v == l_src[j]->v);
161                                 if (BM_ELEM_API_FLAG_TEST(l_dst[j], _FLAG_OVERLAP) == 0) {
162                                         if ((filter_fn == NULL) || filter_fn(l_src[j], user_data)) {
163                                                 bm_loop_attrs_copy(bm, bm, l_src[j], l_dst[j], 0x0);
164                                                 BM_ELEM_API_FLAG_ENABLE(l_dst[j], _FLAG_OVERLAP);
165                                         }
166                                 }
167                         }
168                 }
169         } while ((l_iter = l_iter->next) != l_first);
170
171
172         l_iter = l_first = BM_FACE_FIRST_LOOP(f);
173         do {
174                 BM_ELEM_API_FLAG_DISABLE(l_iter, _FLAG_OVERLAP);
175         } while ((l_iter = l_iter->next) != l_first);
176 }
177
178 /**
179  * Given an array of edges,
180  * order them using the winding defined by \a v1 & \a v2
181  * into \a edges_sort & \a verts_sort.
182  *
183  * All arrays must be \a len long.
184  */
185 static bool bm_edges_sort_winding(
186         BMVert *v1, BMVert *v2,
187         BMEdge **edges, const int len,
188         BMEdge **edges_sort, BMVert **verts_sort)
189 {
190         BMEdge *e_iter, *e_first;
191         BMVert *v_iter;
192         int i;
193
194         /* all flags _must_ be cleared on exit! */
195         for (i = 0; i < len; i++) {
196                 BM_ELEM_API_FLAG_ENABLE(edges[i], _FLAG_MF);
197                 BM_ELEM_API_FLAG_ENABLE(edges[i]->v1, _FLAG_MV);
198                 BM_ELEM_API_FLAG_ENABLE(edges[i]->v2, _FLAG_MV);
199         }
200
201         /* find first edge */
202         i = 0;
203         v_iter = v1;
204         e_iter = e_first = v1->e;
205         do {
206                 if (BM_ELEM_API_FLAG_TEST(e_iter, _FLAG_MF) &&
207                     (BM_edge_other_vert(e_iter, v_iter) == v2))
208                 {
209                         i = 1;
210                         break;
211                 }
212         } while ((e_iter = bmesh_disk_edge_next(e_iter, v_iter)) != e_first);
213         if (i == 0) {
214                 goto error;
215         }
216
217         i = 0;
218         do {
219                 /* entering loop will always succeed */
220                 if (BM_ELEM_API_FLAG_TEST(e_iter, _FLAG_MF)) {
221                         if (UNLIKELY(BM_ELEM_API_FLAG_TEST(v_iter, _FLAG_MV) == false)) {
222                                 /* vert is in loop multiple times */
223                                 goto error;
224                         }
225
226                         BM_ELEM_API_FLAG_DISABLE(e_iter, _FLAG_MF);
227                         edges_sort[i] = e_iter;
228
229                         BM_ELEM_API_FLAG_DISABLE(v_iter, _FLAG_MV);
230                         verts_sort[i] = v_iter;
231
232                         i += 1;
233
234                         /* walk onto the next vertex */
235                         v_iter = BM_edge_other_vert(e_iter, v_iter);
236                         if (i == len) {
237                                 if (UNLIKELY(v_iter != verts_sort[0])) {
238                                         goto error;
239                                 }
240                                 break;
241                         }
242
243                         e_first = e_iter;
244                 }
245         } while ((e_iter = bmesh_disk_edge_next(e_iter, v_iter)) != e_first);
246
247         if (i == len) {
248                 return true;
249         }
250
251 error:
252         for (i = 0; i < len; i++) {
253                 BM_ELEM_API_FLAG_DISABLE(edges[i], _FLAG_MF);
254                 BM_ELEM_API_FLAG_DISABLE(edges[i]->v1, _FLAG_MV);
255                 BM_ELEM_API_FLAG_DISABLE(edges[i]->v2, _FLAG_MV);
256         }
257
258         return false;
259 }
260
261 /**
262  * \brief Make NGon
263  *
264  * Makes an ngon from an unordered list of edges.
265  * Verts \a v1 and \a v2 define the winding of the new face.
266  *
267  * \a edges are not required to be ordered, simply to form
268  * a single closed loop as a whole.
269  *
270  * \note While this function will work fine when the edges
271  * are already sorted, if the edges are always going to be sorted,
272  * #BM_face_create should be considered over this function as it
273  * avoids some unnecessary work.
274  */
275 BMFace *BM_face_create_ngon(
276         BMesh *bm, BMVert *v1, BMVert *v2, BMEdge **edges, const int len,
277         const BMFace *f_example, const eBMCreateFlag create_flag)
278 {
279         BMEdge **edges_sort = BLI_array_alloca(edges_sort, len);
280         BMVert **verts_sort = BLI_array_alloca(verts_sort, len);
281
282         BLI_assert(len && v1 && v2 && edges && bm);
283
284         if (bm_edges_sort_winding(v1, v2, edges, len, edges_sort, verts_sort)) {
285                 return BM_face_create(bm, verts_sort, edges_sort, len, f_example, create_flag);
286         }
287
288         return NULL;
289 }
290
291 /**
292  * Create an ngon from an array of sorted verts
293  *
294  * Special features this has over other functions.
295  * - Optionally calculate winding based on surrounding edges.
296  * - Optionally create edges between vertices.
297  * - Uses verts so no need to find edges (handy when you only have verts)
298  */
299 BMFace *BM_face_create_ngon_verts(
300         BMesh *bm, BMVert **vert_arr, const int len,
301         const BMFace *f_example, const eBMCreateFlag create_flag,
302         const bool calc_winding, const bool create_edges)
303 {
304         BMEdge **edge_arr = BLI_array_alloca(edge_arr, len);
305         uint winding[2] = {0, 0};
306         int i, i_prev = len - 1;
307         BMVert *v_winding[2] = {vert_arr[i_prev], vert_arr[0]};
308
309         BLI_assert(len > 2);
310
311         for (i = 0; i < len; i++) {
312                 if (create_edges) {
313                         edge_arr[i] = BM_edge_create(bm, vert_arr[i_prev], vert_arr[i], NULL, BM_CREATE_NO_DOUBLE);
314                 }
315                 else {
316                         edge_arr[i] = BM_edge_exists(vert_arr[i_prev], vert_arr[i]);
317                         if (edge_arr[i] == NULL) {
318                                 return NULL;
319                         }
320                 }
321
322                 if (calc_winding) {
323                         /* the edge may exist already and be attached to a face
324                          * in this case we can find the best winding to use for the new face */
325                         if (edge_arr[i]->l) {
326                                 BMVert *test_v1, *test_v2;
327                                 /* we want to use the reverse winding to the existing order */
328                                 BM_edge_ordered_verts(edge_arr[i], &test_v2, &test_v1);
329                                 winding[(vert_arr[i_prev] == test_v2)]++;
330                                 BLI_assert(vert_arr[i_prev] == test_v2 || vert_arr[i_prev] == test_v1);
331                         }
332                 }
333
334                 i_prev = i;
335         }
336
337         /* --- */
338
339         if (calc_winding) {
340                 if (winding[0] < winding[1]) {
341                         winding[0] = 1;
342                         winding[1] = 0;
343                 }
344                 else {
345                         winding[0] = 0;
346                         winding[1] = 1;
347                 }
348         }
349         else {
350                 winding[0] = 0;
351                 winding[1] = 1;
352         }
353
354         /* --- */
355
356         /* create the face */
357         return BM_face_create_ngon(
358                 bm,
359                 v_winding[winding[0]],
360                 v_winding[winding[1]],
361                 edge_arr, len,
362                 f_example, create_flag);
363 }
364
365 /**
366  * Makes an NGon from an un-ordered set of verts
367  *
368  * assumes...
369  * - that verts are only once in the list.
370  * - that the verts have roughly planer bounds
371  * - that the verts are roughly circular
372  * there can be concave areas but overlapping folds from the center point will fail.
373  *
374  * a brief explanation of the method used
375  * - find the center point
376  * - find the normal of the vcloud
377  * - order the verts around the face based on their angle to the normal vector at the center point.
378  *
379  * \note Since this is a vcloud there is no direction.
380  */
381 void BM_verts_sort_radial_plane(BMVert **vert_arr, int len)
382 {
383         struct SortIntByFloat *vang = BLI_array_alloca(vang, len);
384         BMVert **vert_arr_map = BLI_array_alloca(vert_arr_map, len);
385
386         float totv_inv = 1.0f / (float)len;
387         int i = 0;
388
389         float cent[3], nor[3];
390
391         const float *far = NULL, *far_cross = NULL;
392
393         float far_vec[3];
394         float far_cross_vec[3];
395         float sign_vec[3]; /* work out if we are pos/neg angle */
396
397         float far_dist_sq, far_dist_max_sq;
398         float far_cross_dist, far_cross_best = 0.0f;
399
400         /* get the center point and collect vector array since we loop over these a lot */
401         zero_v3(cent);
402         for (i = 0; i < len; i++) {
403                 madd_v3_v3fl(cent, vert_arr[i]->co, totv_inv);
404         }
405
406
407         /* find the far point from cent */
408         far_dist_max_sq = 0.0f;
409         for (i = 0; i < len; i++) {
410                 far_dist_sq = len_squared_v3v3(vert_arr[i]->co, cent);
411                 if (far_dist_sq > far_dist_max_sq || far == NULL) {
412                         far = vert_arr[i]->co;
413                         far_dist_max_sq = far_dist_sq;
414                 }
415         }
416
417         sub_v3_v3v3(far_vec, far, cent);
418         // far_dist = len_v3(far_vec); /* real dist */ /* UNUSED */
419
420         /* --- */
421
422         /* find a point 90deg about to compare with */
423         far_cross_best = 0.0f;
424         for (i = 0; i < len; i++) {
425
426                 if (far == vert_arr[i]->co) {
427                         continue;
428                 }
429
430                 sub_v3_v3v3(far_cross_vec, vert_arr[i]->co, cent);
431                 far_cross_dist = normalize_v3(far_cross_vec);
432
433                 /* more of a weight then a distance */
434                 far_cross_dist = (/* first we want to have a value close to zero mapped to 1 */
435                                   1.0f - fabsf(dot_v3v3(far_vec, far_cross_vec)) *
436
437                                   /* second  we multiply by the distance
438                                    * so points close to the center are not preferred */
439                                   far_cross_dist);
440
441                 if (far_cross_dist > far_cross_best || far_cross == NULL) {
442                         far_cross = vert_arr[i]->co;
443                         far_cross_best = far_cross_dist;
444                 }
445         }
446
447         sub_v3_v3v3(far_cross_vec, far_cross, cent);
448
449         /* --- */
450
451         /* now we have 2 vectors we can have a cross product */
452         cross_v3_v3v3(nor, far_vec, far_cross_vec);
453         normalize_v3(nor);
454         cross_v3_v3v3(sign_vec, far_vec, nor); /* this vector should match 'far_cross_vec' closely */
455
456         /* --- */
457
458         /* now calculate every points angle around the normal (signed) */
459         for (i = 0; i < len; i++) {
460                 vang[i].sort_value = angle_signed_on_axis_v3v3v3_v3(far, cent, vert_arr[i]->co, nor);
461                 vang[i].data = i;
462                 vert_arr_map[i] = vert_arr[i];
463         }
464
465         /* sort by angle and magic! - we have our ngon */
466         qsort(vang, len, sizeof(*vang), BLI_sortutil_cmp_float);
467
468         /* --- */
469
470         for (i = 0; i < len; i++) {
471                 vert_arr[i] = vert_arr_map[vang[i].data];
472         }
473 }
474
475 /*************************************************************/
476
477
478 static void bm_vert_attrs_copy(
479         BMesh *source_mesh, BMesh *target_mesh,
480         const BMVert *source_vertex, BMVert *target_vertex, uint64_t cd_mask)
481 {
482         if ((source_mesh == target_mesh) && (source_vertex == target_vertex)) {
483                 BLI_assert(!"BMVert: source and targer match");
484                 return;
485         }
486         if ((cd_mask & CD_MASK_NORMAL) == 0) {
487                 copy_v3_v3(target_vertex->no, source_vertex->no);
488         }
489         CustomData_bmesh_free_block_data(&target_mesh->vdata, target_vertex->head.data);
490         CustomData_bmesh_copy_data(&source_mesh->vdata, &target_mesh->vdata,
491                                    source_vertex->head.data, &target_vertex->head.data);
492 }
493
494 static void bm_edge_attrs_copy(
495         BMesh *source_mesh, BMesh *target_mesh,
496         const BMEdge *source_edge, BMEdge *target_edge, uint64_t UNUSED(cd_mask))
497 {
498         if ((source_mesh == target_mesh) && (source_edge == target_edge)) {
499                 BLI_assert(!"BMEdge: source and targer match");
500                 return;
501         }
502         CustomData_bmesh_free_block_data(&target_mesh->edata, target_edge->head.data);
503         CustomData_bmesh_copy_data(&source_mesh->edata, &target_mesh->edata,
504                                    source_edge->head.data, &target_edge->head.data);
505 }
506
507 static void bm_loop_attrs_copy(
508         BMesh *source_mesh, BMesh *target_mesh,
509         const BMLoop *source_loop, BMLoop *target_loop, uint64_t UNUSED(cd_mask))
510 {
511         if ((source_mesh == target_mesh) && (source_loop == target_loop)) {
512                 BLI_assert(!"BMLoop: source and targer match");
513                 return;
514         }
515         CustomData_bmesh_free_block_data(&target_mesh->ldata, target_loop->head.data);
516         CustomData_bmesh_copy_data(&source_mesh->ldata, &target_mesh->ldata,
517                                    source_loop->head.data, &target_loop->head.data);
518 }
519
520 static void bm_face_attrs_copy(
521         BMesh *source_mesh, BMesh *target_mesh,
522         const BMFace *source_face, BMFace *target_face, uint64_t cd_mask)
523 {
524         if ((source_mesh == target_mesh) && (source_face == target_face)) {
525                 BLI_assert(!"BMFace: source and targer match");
526                 return;
527         }
528         if ((cd_mask & CD_MASK_NORMAL) == 0) {
529                 copy_v3_v3(target_face->no, source_face->no);
530         }
531         CustomData_bmesh_free_block_data(&target_mesh->pdata, target_face->head.data);
532         CustomData_bmesh_copy_data(&source_mesh->pdata, &target_mesh->pdata,
533                                    source_face->head.data, &target_face->head.data);
534         target_face->mat_nr = source_face->mat_nr;
535 }
536
537 /* BMESH_TODO: Special handling for hide flags? */
538 /* BMESH_TODO: swap src/dst args, everywhere else in bmesh does other way round */
539
540 /**
541  * Copies attributes, e.g. customdata, header flags, etc, from one element
542  * to another of the same type.
543  */
544 void BM_elem_attrs_copy_ex(
545         BMesh *bm_src, BMesh *bm_dst, const void *ele_src_v, void *ele_dst_v,
546         const char hflag_mask, const uint64_t cd_mask)
547 {
548         const BMHeader *ele_src = ele_src_v;
549         BMHeader *ele_dst = ele_dst_v;
550
551         BLI_assert(ele_src->htype == ele_dst->htype);
552         BLI_assert(ele_src != ele_dst);
553
554         /* Only support normal layer at the moment. */
555         BLI_assert((cd_mask & ~CD_MASK_NORMAL) == 0);
556
557         if ((hflag_mask & BM_ELEM_SELECT) == 0) {
558                 /* First we copy select */
559                 if (BM_elem_flag_test((BMElem *)ele_src, BM_ELEM_SELECT)) {
560                         BM_elem_select_set(bm_dst, (BMElem *)ele_dst, true);
561                 }
562         }
563
564         /* Now we copy flags */
565         if (hflag_mask == 0) {
566                 ele_dst->hflag = ele_src->hflag;
567         }
568         else if (hflag_mask == 0xff) {
569                 /* pass */
570         }
571         else {
572                 ele_dst->hflag = ((ele_dst->hflag & hflag_mask) | (ele_src->hflag & ~hflag_mask));
573         }
574
575         /* Copy specific attributes */
576         switch (ele_dst->htype) {
577                 case BM_VERT:
578                         bm_vert_attrs_copy(bm_src, bm_dst, (const BMVert *)ele_src, (BMVert *)ele_dst, cd_mask);
579                         break;
580                 case BM_EDGE:
581                         bm_edge_attrs_copy(bm_src, bm_dst, (const BMEdge *)ele_src, (BMEdge *)ele_dst, cd_mask);
582                         break;
583                 case BM_LOOP:
584                         bm_loop_attrs_copy(bm_src, bm_dst, (const BMLoop *)ele_src, (BMLoop *)ele_dst, cd_mask);
585                         break;
586                 case BM_FACE:
587                         bm_face_attrs_copy(bm_src, bm_dst, (const BMFace *)ele_src, (BMFace *)ele_dst, cd_mask);
588                         break;
589                 default:
590                         BLI_assert(0);
591                         break;
592         }
593 }
594
595 void BM_elem_attrs_copy(BMesh *bm_src, BMesh *bm_dst, const void *ele_src, void *ele_dst)
596 {
597         /* BMESH_TODO, default 'use_flags' to false */
598         BM_elem_attrs_copy_ex(bm_src, bm_dst, ele_src, ele_dst, BM_ELEM_SELECT, 0x0);
599 }
600
601 void BM_elem_select_copy(BMesh *bm_dst, void *ele_dst_v, const void *ele_src_v)
602 {
603         BMHeader *ele_dst = ele_dst_v;
604         const BMHeader *ele_src = ele_src_v;
605
606         BLI_assert(ele_src->htype == ele_dst->htype);
607
608         if ((ele_src->hflag & BM_ELEM_SELECT) != (ele_dst->hflag & BM_ELEM_SELECT)) {
609                 BM_elem_select_set(bm_dst, (BMElem *)ele_dst, (ele_src->hflag & BM_ELEM_SELECT) != 0);
610         }
611 }
612
613 /* helper function for 'BM_mesh_copy' */
614 static BMFace *bm_mesh_copy_new_face(
615         BMesh *bm_new, BMesh *bm_old,
616         BMVert **vtable, BMEdge **etable,
617         BMFace *f)
618 {
619         BMLoop **loops = BLI_array_alloca(loops, f->len);
620         BMVert **verts = BLI_array_alloca(verts, f->len);
621         BMEdge **edges = BLI_array_alloca(edges, f->len);
622
623         BMFace *f_new;
624         BMLoop *l_iter, *l_first;
625         int j;
626
627         j = 0;
628         l_iter = l_first = BM_FACE_FIRST_LOOP(f);
629         do {
630                 loops[j] = l_iter;
631                 verts[j] = vtable[BM_elem_index_get(l_iter->v)];
632                 edges[j] = etable[BM_elem_index_get(l_iter->e)];
633                 j++;
634         } while ((l_iter = l_iter->next) != l_first);
635
636         f_new = BM_face_create(bm_new, verts, edges, f->len, NULL, BM_CREATE_SKIP_CD);
637
638         if (UNLIKELY(f_new == NULL)) {
639                 return NULL;
640         }
641
642         /* use totface in case adding some faces fails */
643         BM_elem_index_set(f_new, (bm_new->totface - 1)); /* set_inline */
644
645         BM_elem_attrs_copy_ex(bm_old, bm_new, f, f_new, 0xff, 0x0);
646         f_new->head.hflag = f->head.hflag;  /* low level! don't do this for normal api use */
647
648         j = 0;
649         l_iter = l_first = BM_FACE_FIRST_LOOP(f_new);
650         do {
651                 BM_elem_attrs_copy(bm_old, bm_new, loops[j], l_iter);
652                 j++;
653         } while ((l_iter = l_iter->next) != l_first);
654
655         return f_new;
656 }
657
658 void BM_mesh_copy_init_customdata(BMesh *bm_dst, BMesh *bm_src, const BMAllocTemplate *allocsize)
659 {
660         if (allocsize == NULL) {
661                 allocsize = &bm_mesh_allocsize_default;
662         }
663
664         CustomData_copy(&bm_src->vdata, &bm_dst->vdata, CD_MASK_BMESH, CD_CALLOC, 0);
665         CustomData_copy(&bm_src->edata, &bm_dst->edata, CD_MASK_BMESH, CD_CALLOC, 0);
666         CustomData_copy(&bm_src->ldata, &bm_dst->ldata, CD_MASK_BMESH, CD_CALLOC, 0);
667         CustomData_copy(&bm_src->pdata, &bm_dst->pdata, CD_MASK_BMESH, CD_CALLOC, 0);
668
669         CustomData_bmesh_init_pool(&bm_dst->vdata, allocsize->totvert, BM_VERT);
670         CustomData_bmesh_init_pool(&bm_dst->edata, allocsize->totedge, BM_EDGE);
671         CustomData_bmesh_init_pool(&bm_dst->ldata, allocsize->totloop, BM_LOOP);
672         CustomData_bmesh_init_pool(&bm_dst->pdata, allocsize->totface, BM_FACE);
673 }
674
675
676 BMesh *BM_mesh_copy(BMesh *bm_old)
677 {
678         BMesh *bm_new;
679         BMVert *v, *v_new, **vtable = NULL;
680         BMEdge *e, *e_new, **etable = NULL;
681         BMFace *f, *f_new, **ftable = NULL;
682         BMElem **eletable;
683         BMEditSelection *ese;
684         BMIter iter;
685         int i;
686         const BMAllocTemplate allocsize = BMALLOC_TEMPLATE_FROM_BM(bm_old);
687
688         /* allocate a bmesh */
689         bm_new = BM_mesh_create(
690                 &allocsize,
691                 &((struct BMeshCreateParams){.use_toolflags = bm_old->use_toolflags,}));
692
693         BM_mesh_copy_init_customdata(bm_new, bm_old, &allocsize);
694
695         vtable = MEM_mallocN(sizeof(BMVert *) * bm_old->totvert, "BM_mesh_copy vtable");
696         etable = MEM_mallocN(sizeof(BMEdge *) * bm_old->totedge, "BM_mesh_copy etable");
697         ftable = MEM_mallocN(sizeof(BMFace *) * bm_old->totface, "BM_mesh_copy ftable");
698
699         BM_ITER_MESH_INDEX (v, &iter, bm_old, BM_VERTS_OF_MESH, i) {
700                 /* copy between meshes so cant use 'example' argument */
701                 v_new = BM_vert_create(bm_new, v->co, NULL, BM_CREATE_SKIP_CD);
702                 BM_elem_attrs_copy_ex(bm_old, bm_new, v, v_new, 0xff, 0x0);
703                 v_new->head.hflag = v->head.hflag;  /* low level! don't do this for normal api use */
704                 vtable[i] = v_new;
705                 BM_elem_index_set(v, i); /* set_inline */
706                 BM_elem_index_set(v_new, i); /* set_inline */
707         }
708         bm_old->elem_index_dirty &= ~BM_VERT;
709         bm_new->elem_index_dirty &= ~BM_VERT;
710
711         /* safety check */
712         BLI_assert(i == bm_old->totvert);
713
714         BM_ITER_MESH_INDEX (e, &iter, bm_old, BM_EDGES_OF_MESH, i) {
715                 e_new = BM_edge_create(bm_new,
716                                        vtable[BM_elem_index_get(e->v1)],
717                                        vtable[BM_elem_index_get(e->v2)],
718                                        e, BM_CREATE_SKIP_CD);
719
720                 BM_elem_attrs_copy_ex(bm_old, bm_new, e, e_new, 0xff, 0x0);
721                 e_new->head.hflag = e->head.hflag;  /* low level! don't do this for normal api use */
722                 etable[i] = e_new;
723                 BM_elem_index_set(e, i); /* set_inline */
724                 BM_elem_index_set(e_new, i); /* set_inline */
725         }
726         bm_old->elem_index_dirty &= ~BM_EDGE;
727         bm_new->elem_index_dirty &= ~BM_EDGE;
728
729         /* safety check */
730         BLI_assert(i == bm_old->totedge);
731
732         BM_ITER_MESH_INDEX (f, &iter, bm_old, BM_FACES_OF_MESH, i) {
733                 BM_elem_index_set(f, i); /* set_inline */
734
735                 f_new = bm_mesh_copy_new_face(bm_new, bm_old, vtable, etable, f);
736
737                 ftable[i] = f_new;
738
739                 if (f == bm_old->act_face) bm_new->act_face = f_new;
740         }
741         bm_old->elem_index_dirty &= ~BM_FACE;
742         bm_new->elem_index_dirty &= ~BM_FACE;
743
744
745         /* low level! don't do this for normal api use */
746         bm_new->totvertsel = bm_old->totvertsel;
747         bm_new->totedgesel = bm_old->totedgesel;
748         bm_new->totfacesel = bm_old->totfacesel;
749
750         /* safety check */
751         BLI_assert(i == bm_old->totface);
752
753         /* copy over edit selection history */
754         for (ese = bm_old->selected.first; ese; ese = ese->next) {
755                 BMElem *ele = NULL;
756
757                 switch (ese->htype) {
758                         case BM_VERT:
759                                 eletable = (BMElem **)vtable;
760                                 break;
761                         case BM_EDGE:
762                                 eletable = (BMElem **)etable;
763                                 break;
764                         case BM_FACE:
765                                 eletable = (BMElem **)ftable;
766                                 break;
767                         default:
768                                 eletable = NULL;
769                                 break;
770                 }
771
772                 if (eletable) {
773                         ele = eletable[BM_elem_index_get(ese->ele)];
774                         if (ele) {
775                                 BM_select_history_store(bm_new, ele);
776                         }
777                 }
778         }
779
780         MEM_freeN(etable);
781         MEM_freeN(vtable);
782         MEM_freeN(ftable);
783
784         return bm_new;
785 }
786
787 /* ME -> BM */
788 char BM_vert_flag_from_mflag(const char  meflag)
789 {
790         return ( ((meflag & SELECT)       ? BM_ELEM_SELECT : 0) |
791                  ((meflag & ME_HIDE)      ? BM_ELEM_HIDDEN : 0)
792                  );
793 }
794 char BM_edge_flag_from_mflag(const short meflag)
795 {
796         return ( ((meflag & SELECT)        ? BM_ELEM_SELECT : 0) |
797                  ((meflag & ME_SEAM)       ? BM_ELEM_SEAM   : 0) |
798                  ((meflag & ME_EDGEDRAW)   ? BM_ELEM_DRAW   : 0) |
799                  ((meflag & ME_SHARP) == 0 ? BM_ELEM_SMOOTH : 0) | /* invert */
800                  ((meflag & ME_HIDE)       ? BM_ELEM_HIDDEN : 0)
801                  );
802 }
803 char BM_face_flag_from_mflag(const char  meflag)
804 {
805         return ( ((meflag & ME_FACE_SEL)  ? BM_ELEM_SELECT : 0) |
806                  ((meflag & ME_SMOOTH)    ? BM_ELEM_SMOOTH : 0) |
807                  ((meflag & ME_HIDE)      ? BM_ELEM_HIDDEN : 0)
808                  );
809 }
810
811 /* BM -> ME */
812 char  BM_vert_flag_to_mflag(BMVert *eve)
813 {
814         const char hflag = eve->head.hflag;
815
816         return ( ((hflag & BM_ELEM_SELECT)  ? SELECT  : 0) |
817                  ((hflag & BM_ELEM_HIDDEN)  ? ME_HIDE : 0)
818                  );
819 }
820
821 short BM_edge_flag_to_mflag(BMEdge *eed)
822 {
823         const char hflag = eed->head.hflag;
824
825         return ( ((hflag & BM_ELEM_SELECT)       ? SELECT       : 0) |
826                  ((hflag & BM_ELEM_SEAM)         ? ME_SEAM      : 0) |
827                  ((hflag & BM_ELEM_DRAW)         ? ME_EDGEDRAW  : 0) |
828                  ((hflag & BM_ELEM_SMOOTH) == 0  ? ME_SHARP     : 0) |
829                  ((hflag & BM_ELEM_HIDDEN)       ? ME_HIDE      : 0) |
830                  ((BM_edge_is_wire(eed))         ? ME_LOOSEEDGE : 0) | /* not typical */
831                  ME_EDGERENDER
832                  );
833 }
834 char  BM_face_flag_to_mflag(BMFace *efa)
835 {
836         const char hflag = efa->head.hflag;
837
838         return ( ((hflag & BM_ELEM_SELECT) ? ME_FACE_SEL : 0) |
839                  ((hflag & BM_ELEM_SMOOTH) ? ME_SMOOTH   : 0) |
840                  ((hflag & BM_ELEM_HIDDEN) ? ME_HIDE     : 0)
841                  );
842 }