Gizmo: match the DOF gizmo size with the non-gizmo indicator
[blender.git] / source / blender / editors / mesh / editmesh_extrude_spin.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 "DNA_object_types.h"
25
26 #include "BLI_math.h"
27
28 #include "BKE_context.h"
29 #include "BKE_report.h"
30 #include "BKE_editmesh.h"
31 #include "BKE_layer.h"
32
33 #include "RNA_define.h"
34 #include "RNA_access.h"
35 #include "RNA_enum_types.h"
36
37 #include "WM_api.h"
38 #include "WM_types.h"
39
40 #include "ED_mesh.h"
41 #include "ED_screen.h"
42 #include "ED_view3d.h"
43
44 #include "UI_resources.h"
45
46 #include "MEM_guardedalloc.h"
47
48 #include "mesh_intern.h" /* own include */
49
50 #define USE_GIZMO
51
52 /* -------------------------------------------------------------------- */
53 /** \name Spin Operator
54  * \{ */
55
56 static int edbm_spin_exec(bContext *C, wmOperator *op)
57 {
58   ViewLayer *view_layer = CTX_data_view_layer(C);
59   float cent[3], axis[3];
60   float d[3] = {0.0f, 0.0f, 0.0f};
61
62   RNA_float_get_array(op->ptr, "center", cent);
63   RNA_float_get_array(op->ptr, "axis", axis);
64   const int steps = RNA_int_get(op->ptr, "steps");
65   const float angle = RNA_float_get(op->ptr, "angle");
66   const bool use_normal_flip = RNA_boolean_get(op->ptr, "use_normal_flip");
67   const bool dupli = RNA_boolean_get(op->ptr, "dupli");
68   const bool use_auto_merge = (RNA_boolean_get(op->ptr, "use_auto_merge") && (dupli == false) &&
69                                (steps >= 3) && fabsf((fabsf(angle) - (float)(M_PI * 2))) <= 1e-6f);
70
71   if (is_zero_v3(axis)) {
72     BKE_report(op->reports, RPT_ERROR, "Invalid/unset axis");
73     return OPERATOR_CANCELLED;
74   }
75
76   uint objects_len = 0;
77   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
78       view_layer, CTX_wm_view3d(C), &objects_len);
79
80   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
81     Object *obedit = objects[ob_index];
82     BMEditMesh *em = BKE_editmesh_from_object(obedit);
83     BMesh *bm = em->bm;
84     BMOperator spinop;
85
86     /* keep the values in worldspace since we're passing the obmat */
87     if (!EDBM_op_init(em,
88                       &spinop,
89                       op,
90                       "spin geom=%hvef cent=%v axis=%v dvec=%v steps=%i angle=%f space=%m4 "
91                       "use_normal_flip=%b use_duplicate=%b use_merge=%b",
92                       BM_ELEM_SELECT,
93                       cent,
94                       axis,
95                       d,
96                       steps,
97                       -angle,
98                       obedit->obmat,
99                       use_normal_flip,
100                       dupli,
101                       use_auto_merge)) {
102       continue;
103     }
104     BMO_op_exec(bm, &spinop);
105     if (use_auto_merge == false) {
106       EDBM_flag_disable_all(em, BM_ELEM_SELECT);
107       BMO_slot_buffer_hflag_enable(
108           bm, spinop.slots_out, "geom_last.out", BM_ALL_NOLOOP, BM_ELEM_SELECT, true);
109     }
110     if (!EDBM_op_finish(em, &spinop, op, true)) {
111       continue;
112     }
113
114     EDBM_update_generic(em, true, true);
115   }
116
117   MEM_freeN(objects);
118
119   return OPERATOR_FINISHED;
120 }
121
122 /* get center and axis, in global coords */
123 static int edbm_spin_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
124 {
125   Scene *scene = CTX_data_scene(C);
126   View3D *v3d = CTX_wm_view3d(C);
127   RegionView3D *rv3d = ED_view3d_context_rv3d(C);
128
129   PropertyRNA *prop;
130   prop = RNA_struct_find_property(op->ptr, "center");
131   if (!RNA_property_is_set(op->ptr, prop)) {
132     RNA_property_float_set_array(op->ptr, prop, scene->cursor.location);
133   }
134   if (rv3d) {
135     prop = RNA_struct_find_property(op->ptr, "axis");
136     if (!RNA_property_is_set(op->ptr, prop)) {
137       RNA_property_float_set_array(op->ptr, prop, rv3d->viewinv[2]);
138     }
139   }
140
141 #ifdef USE_GIZMO
142   /* Start with zero angle, drag out the value. */
143   prop = RNA_struct_find_property(op->ptr, "angle");
144   if (!RNA_property_is_set(op->ptr, prop)) {
145     RNA_property_float_set(op->ptr, prop, 0.0f);
146   }
147 #endif
148
149   int ret = edbm_spin_exec(C, op);
150
151 #ifdef USE_GIZMO
152   if (ret & OPERATOR_FINISHED) {
153     /* Setup gizmos */
154     if (v3d && ((v3d->gizmo_flag & V3D_GIZMO_HIDE) == 0)) {
155       wmGizmoGroupType *gzgt = WM_gizmogrouptype_find("MESH_GGT_spin_redo", false);
156       if (!WM_gizmo_group_type_ensure_ptr(gzgt)) {
157         struct Main *bmain = CTX_data_main(C);
158         WM_gizmo_group_type_reinit_ptr(bmain, gzgt);
159       }
160     }
161   }
162 #endif
163
164   return ret;
165 }
166
167 static bool edbm_spin_poll_property(const bContext *UNUSED(C),
168                                     wmOperator *op,
169                                     const PropertyRNA *prop)
170 {
171   const char *prop_id = RNA_property_identifier(prop);
172   const bool dupli = RNA_boolean_get(op->ptr, "dupli");
173
174   if (dupli) {
175     if (STREQ(prop_id, "use_auto_merge") || STREQ(prop_id, "use_normal_flip")) {
176       return false;
177     }
178   }
179   return true;
180 }
181
182 void MESH_OT_spin(wmOperatorType *ot)
183 {
184   PropertyRNA *prop;
185
186   /* identifiers */
187   ot->name = "Spin";
188   ot->description =
189       "Extrude selected vertices in a circle around the cursor in indicated viewport";
190   ot->idname = "MESH_OT_spin";
191
192   /* api callbacks */
193   ot->invoke = edbm_spin_invoke;
194   ot->exec = edbm_spin_exec;
195   ot->poll = ED_operator_editmesh;
196   ot->poll_property = edbm_spin_poll_property;
197
198   /* flags */
199   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
200
201   /* props */
202   RNA_def_int(ot->srna, "steps", 9, 0, 1000000, "Steps", "Steps", 0, 1000);
203   RNA_def_boolean(ot->srna, "dupli", 0, "Duplicate", "Make Duplicates");
204   prop = RNA_def_float(ot->srna,
205                        "angle",
206                        DEG2RADF(90.0f),
207                        -1e12f,
208                        1e12f,
209                        "Angle",
210                        "Rotation for each step",
211                        DEG2RADF(-360.0f),
212                        DEG2RADF(360.0f));
213   RNA_def_property_subtype(prop, PROP_ANGLE);
214   RNA_def_boolean(ot->srna,
215                   "use_auto_merge",
216                   true,
217                   "Auto Merge",
218                   "Merge first/last when the angle is a full revolution");
219   RNA_def_boolean(ot->srna, "use_normal_flip", 0, "Flip Normals", "");
220
221   RNA_def_float_vector_xyz(ot->srna,
222                            "center",
223                            3,
224                            NULL,
225                            -1e12f,
226                            1e12f,
227                            "Center",
228                            "Center in global view space",
229                            -1e4f,
230                            1e4f);
231   RNA_def_float_vector(
232       ot->srna, "axis", 3, NULL, -1.0f, 1.0f, "Axis", "Axis in global view space", -1.0f, 1.0f);
233
234   WM_gizmogrouptype_append(MESH_GGT_spin);
235 #ifdef USE_GIZMO
236   WM_gizmogrouptype_append(MESH_GGT_spin_redo);
237 #endif
238 }
239
240 /** \} */