Cleanup: style, use braces for editors
[blender.git] / source / blender / editors / mesh / editmesh_extrude_screw.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) 2004 by Blender Foundation.
17  * All rights reserved.
18  */
19
20 /** \file
21  * \ingroup edmesh
22  */
23
24 #include "MEM_guardedalloc.h"
25
26 #include "DNA_object_types.h"
27
28 #include "BLI_math.h"
29
30 #include "BKE_context.h"
31 #include "BKE_editmesh.h"
32 #include "BKE_layer.h"
33 #include "BKE_report.h"
34
35 #include "RNA_define.h"
36 #include "RNA_access.h"
37
38 #include "WM_types.h"
39
40 #include "ED_mesh.h"
41 #include "ED_screen.h"
42 #include "ED_view3d.h"
43
44 #include "mesh_intern.h" /* own include */
45
46 /* -------------------------------------------------------------------- */
47 /** \name Screw Operator
48  * \{ */
49
50 static int edbm_screw_exec(bContext *C, wmOperator *op)
51 {
52   BMEdge *eed;
53   BMVert *eve, *v1, *v2;
54   BMIter iter, eiter;
55   float dvec[3], nor[3], cent[3], axis[3], v1_co_global[3], v2_co_global[3];
56   int steps, turns;
57   int valence;
58   uint objects_empty_len = 0;
59   uint failed_axis_len = 0;
60   uint failed_vertices_len = 0;
61
62   turns = RNA_int_get(op->ptr, "turns");
63   steps = RNA_int_get(op->ptr, "steps");
64   RNA_float_get_array(op->ptr, "center", cent);
65   RNA_float_get_array(op->ptr, "axis", axis);
66
67   uint objects_len = 0;
68   ViewLayer *view_layer = CTX_data_view_layer(C);
69   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
70       view_layer, CTX_wm_view3d(C), &objects_len);
71
72   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
73     Object *obedit = objects[ob_index];
74     BMEditMesh *em = BKE_editmesh_from_object(obedit);
75     BMesh *bm = em->bm;
76
77     if (bm->totvertsel < 2) {
78       if (bm->totvertsel == 0) {
79         objects_empty_len++;
80       }
81       continue;
82     }
83
84     if (is_zero_v3(axis)) {
85       failed_axis_len++;
86       continue;
87     }
88
89     /* find two vertices with valence count == 1, more or less is wrong */
90     v1 = NULL;
91     v2 = NULL;
92
93     BM_ITER_MESH (eve, &iter, em->bm, BM_VERTS_OF_MESH) {
94       valence = 0;
95       BM_ITER_ELEM (eed, &eiter, eve, BM_EDGES_OF_VERT) {
96         if (BM_elem_flag_test(eed, BM_ELEM_SELECT)) {
97           valence++;
98         }
99       }
100
101       if (valence == 1) {
102         if (v1 == NULL) {
103           v1 = eve;
104         }
105         else if (v2 == NULL) {
106           v2 = eve;
107         }
108         else {
109           v1 = NULL;
110           break;
111         }
112       }
113     }
114
115     if (v1 == NULL || v2 == NULL) {
116       failed_vertices_len++;
117       continue;
118     }
119
120     copy_v3_v3(nor, obedit->obmat[2]);
121
122     /* calculate dvec */
123     mul_v3_m4v3(v1_co_global, obedit->obmat, v1->co);
124     mul_v3_m4v3(v2_co_global, obedit->obmat, v2->co);
125     sub_v3_v3v3(dvec, v1_co_global, v2_co_global);
126     mul_v3_fl(dvec, 1.0f / steps);
127
128     if (dot_v3v3(nor, dvec) > 0.0f) {
129       negate_v3(dvec);
130     }
131
132     BMOperator spinop;
133     if (!EDBM_op_init(
134             em,
135             &spinop,
136             op,
137             "spin geom=%hvef cent=%v axis=%v dvec=%v steps=%i angle=%f space=%m4 use_duplicate=%b",
138             BM_ELEM_SELECT,
139             cent,
140             axis,
141             dvec,
142             turns * steps,
143             DEG2RADF(360.0f * turns),
144             obedit->obmat,
145             false)) {
146       continue;
147     }
148
149     BMO_op_exec(bm, &spinop);
150     EDBM_flag_disable_all(em, BM_ELEM_SELECT);
151     BMO_slot_buffer_hflag_enable(
152         bm, spinop.slots_out, "geom_last.out", BM_ALL_NOLOOP, BM_ELEM_SELECT, true);
153
154     if (!EDBM_op_finish(em, &spinop, op, true)) {
155       continue;
156     }
157
158     EDBM_update_generic(em, true, true);
159   }
160   MEM_freeN(objects);
161
162   if (failed_axis_len == objects_len - objects_empty_len) {
163     BKE_report(op->reports, RPT_ERROR, "Invalid/unset axis");
164   }
165   else if (failed_vertices_len == objects_len - objects_empty_len) {
166     BKE_report(op->reports, RPT_ERROR, "You have to select a string of connected vertices too");
167   }
168
169   return OPERATOR_FINISHED;
170 }
171
172 /* get center and axis, in global coords */
173 static int edbm_screw_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
174 {
175   Scene *scene = CTX_data_scene(C);
176   RegionView3D *rv3d = ED_view3d_context_rv3d(C);
177
178   PropertyRNA *prop;
179   prop = RNA_struct_find_property(op->ptr, "center");
180   if (!RNA_property_is_set(op->ptr, prop)) {
181     RNA_property_float_set_array(op->ptr, prop, scene->cursor.location);
182   }
183   if (rv3d) {
184     prop = RNA_struct_find_property(op->ptr, "axis");
185     if (!RNA_property_is_set(op->ptr, prop)) {
186       RNA_property_float_set_array(op->ptr, prop, rv3d->viewinv[1]);
187     }
188   }
189
190   return edbm_screw_exec(C, op);
191 }
192
193 void MESH_OT_screw(wmOperatorType *ot)
194 {
195   /* identifiers */
196   ot->name = "Screw";
197   ot->description =
198       "Extrude selected vertices in screw-shaped rotation around the cursor in indicated viewport";
199   ot->idname = "MESH_OT_screw";
200
201   /* api callbacks */
202   ot->invoke = edbm_screw_invoke;
203   ot->exec = edbm_screw_exec;
204   ot->poll = ED_operator_editmesh;
205
206   /* flags */
207   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
208
209   /* props */
210   RNA_def_int(ot->srna, "steps", 9, 1, 100000, "Steps", "Steps", 3, 256);
211   RNA_def_int(ot->srna, "turns", 1, 1, 100000, "Turns", "Turns", 1, 256);
212
213   RNA_def_float_vector_xyz(ot->srna,
214                            "center",
215                            3,
216                            NULL,
217                            -1e12f,
218                            1e12f,
219                            "Center",
220                            "Center in global view space",
221                            -1e4f,
222                            1e4f);
223   RNA_def_float_vector(
224       ot->srna, "axis", 3, NULL, -1.0f, 1.0f, "Axis", "Axis in global view space", -1.0f, 1.0f);
225 }
226
227 /** \} */