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