code cleanup: redundant includes and add minor comments.
[blender.git] / source / blender / bmesh / operators / bmo_symmetrize.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_symmetrize.c
24  *  \ingroup bmesh
25  *
26  * Makes the mesh symmetrical by splitting along an axis and duplicating the geometry.
27  */
28
29 #include "BLI_utildefines.h"
30 #include "BLI_math.h"
31
32 #include "bmesh.h"
33 #include "intern/bmesh_operators_private.h"
34
35 #define ELE_OUT 1
36
37 void bmo_symmetrize_exec(BMesh *bm, BMOperator *op)
38 {
39         const float dist = BMO_slot_float_get(op->slots_in, "dist");
40         const int direction = BMO_slot_int_get(op->slots_in, "direction");
41         const int axis = direction % 3;
42
43         BMOperator op_bisect;
44         BMOperator op_dupe;
45         BMOperator op_weld;
46
47         BMOpSlot *slot_vertmap;
48         BMOpSlot *slot_targetmap;
49
50         float plane_no[3];
51         float scale[3];
52
53         BMOIter siter;
54         BMVert *v;
55
56         copy_v3_fl(plane_no, 0.0f);
57         copy_v3_fl(scale, 1.0f);
58
59         plane_no[axis] = direction > 2 ? 1.0f : -1.0f;
60         scale[axis] *= -1.0f;
61
62         /* Cut in half */
63         BMO_op_initf(bm, &op_bisect, op->flag,
64                      "bisect_plane geom=%s plane_no=%v dist=%f clear_outer=%b use_snap_center=%b",
65                      op, "input", plane_no, dist, true, true);
66
67         BMO_op_exec(bm, &op_bisect);
68
69         /* Duplicate */
70         BMO_op_initf(bm, &op_dupe, op->flag,
71                      "duplicate geom=%S",
72                      &op_bisect, "geom.out");
73
74         BMO_op_exec(bm, &op_dupe);
75
76         /* Flag for output (some will be merged) */
77         BMO_slot_buffer_flag_enable(bm, op_bisect.slots_out, "geom.out", BM_ALL_NOLOOP, ELE_OUT);
78         BMO_slot_buffer_flag_enable(bm, op_dupe.slots_out, "geom.out", BM_ALL_NOLOOP, ELE_OUT);
79
80
81         BMO_op_callf(bm, op->flag, "scale verts=%S vec=%v", &op_dupe, "geom.out", scale);
82         BMO_op_callf(bm, op->flag, "reverse_faces faces=%S", &op_dupe, "geom.out");
83
84
85         /* Weld verts */
86         BMO_op_init(bm, &op_weld, op->flag, "weld_verts");
87
88         slot_vertmap = BMO_slot_get(op_dupe.slots_out, "vert_map.out");
89         slot_targetmap = BMO_slot_get(op_weld.slots_in, "targetmap");
90
91         BMO_ITER (v, &siter, op_bisect.slots_out, "geom_cut.out", BM_VERT) {
92                 BMVert *v_dupe = BMO_slot_map_elem_get(slot_vertmap, v);
93                 BMO_slot_map_elem_insert(&op_weld, slot_targetmap, v_dupe, v);
94         }
95
96         BMO_op_exec(bm, &op_weld);
97
98         /* Cleanup */
99         BMO_op_finish(bm, &op_weld);
100
101         BMO_op_finish(bm, &op_dupe);
102         BMO_op_finish(bm, &op_bisect);
103
104         /* Create output */
105         BMO_slot_buffer_from_enabled_flag(bm, op, op->slots_out, "geom.out", BM_ALL_NOLOOP, ELE_OUT);
106 }