8160d28d7a7ba2d1c9d038f88fe6118c8d24823e
[blender.git] / source / blender / bmesh / operators / bmo_fill_holes.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  * Contributor(s): Campbell Barton.
19  *
20  * ***** END GPL LICENSE BLOCK *****
21  */
22
23 /** \file blender/bmesh/operators/bmo_fill_holes.c
24  *  \ingroup bmesh
25  *
26  * Fill boundary edge loop(s) with faces.
27  */
28
29 #include "MEM_guardedalloc.h"
30
31 #include "BLI_utildefines.h"
32
33 #include "bmesh.h"
34 #include "bmesh_tools.h"
35
36 #include "intern/bmesh_operators_private.h" /* own include */
37
38 void bmo_holes_fill_exec(BMesh *bm, BMOperator *op)
39 {
40         BMOperator op_attr;
41         const unsigned int sides = BMO_slot_int_get(op->slots_in,  "sides");
42
43
44         BM_mesh_elem_hflag_disable_all(bm, BM_EDGE | BM_FACE, BM_ELEM_TAG, false);
45         BMO_slot_buffer_hflag_enable(bm, op->slots_in, "edges", BM_EDGE, BM_ELEM_TAG, false);
46
47         BM_mesh_edgenet(bm, true, true);  // TODO, sides
48
49
50         /* bad - remove faces after as a workaround */
51         if (sides != 0) {
52                 BMOIter siter;
53                 BMFace *f;
54
55                 BMO_slot_buffer_from_enabled_hflag(bm, op, op->slots_out, "faces.out", BM_FACE, BM_ELEM_TAG);
56                 BMO_ITER (f, &siter, op->slots_out, "faces.out", BM_FACE) {
57                         if (f->len > sides) {
58                                 BM_face_kill(bm, f);
59                         }
60                 }
61         }
62
63         BMO_slot_buffer_from_enabled_hflag(bm, op, op->slots_out, "faces.out", BM_FACE, BM_ELEM_TAG);
64
65         /* --- Attribute Fill --- */
66         /* may as well since we have the faces already in a buffer */
67         BMO_op_initf(bm, &op_attr, op->flag,
68                      "face_attribute_fill faces=%S use_normals=%b use_data=%b",
69                      op, "faces.out", true, true);
70
71         BMO_op_exec(bm, &op_attr);
72
73         /* check if some faces couldn't be touched */
74         if (BMO_slot_buffer_count(op_attr.slots_out, "faces_fail.out")) {
75                 BMOIter siter;
76                 BMFace *f;
77
78                 BMO_ITER (f, &siter, op_attr.slots_out, "faces_fail.out", BM_FACE) {
79                         BM_face_normal_update(f);  /* normals are zero'd */
80                 }
81
82                 BMO_op_callf(bm, op->flag, "recalc_face_normals faces=%S", &op_attr, "faces_fail.out");
83         }
84         BMO_op_finish(bm, &op_attr);
85 }