Blender  V3.3
editmesh_extrude_spin.c
Go to the documentation of this file.
1 /* SPDX-License-Identifier: GPL-2.0-or-later
2  * Copyright 2004 Blender Foundation. All rights reserved. */
3 
8 #include "DNA_object_types.h"
9 
10 #include "BLI_math.h"
11 #include "BLI_string.h"
12 
13 #include "BKE_context.h"
14 #include "BKE_editmesh.h"
15 #include "BKE_layer.h"
16 #include "BKE_report.h"
17 
18 #include "RNA_access.h"
19 #include "RNA_define.h"
20 
21 #include "WM_types.h"
22 
23 #include "ED_mesh.h"
24 #include "ED_screen.h"
25 #include "ED_view3d.h"
26 
27 #include "MEM_guardedalloc.h"
28 
29 #include "mesh_intern.h" /* own include */
30 
31 #define USE_GIZMO
32 
33 /* -------------------------------------------------------------------- */
38 {
39  ViewLayer *view_layer = CTX_data_view_layer(C);
40  float cent[3], axis[3];
41  const float d[3] = {0.0f, 0.0f, 0.0f};
42 
43  RNA_float_get_array(op->ptr, "center", cent);
44  RNA_float_get_array(op->ptr, "axis", axis);
45  const int steps = RNA_int_get(op->ptr, "steps");
46  const float angle = RNA_float_get(op->ptr, "angle");
47  const bool use_normal_flip = RNA_boolean_get(op->ptr, "use_normal_flip");
48  const bool dupli = RNA_boolean_get(op->ptr, "dupli");
49  const bool use_auto_merge = (RNA_boolean_get(op->ptr, "use_auto_merge") && (dupli == false) &&
50  (steps >= 3) && fabsf((fabsf(angle) - (float)(M_PI * 2))) <= 1e-6f);
51 
52  if (is_zero_v3(axis)) {
53  BKE_report(op->reports, RPT_ERROR, "Invalid/unset axis");
54  return OPERATOR_CANCELLED;
55  }
56 
57  uint objects_len = 0;
59  view_layer, CTX_wm_view3d(C), &objects_len);
60 
61  for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
62  Object *obedit = objects[ob_index];
64  BMesh *bm = em->bm;
65  BMOperator spinop;
66 
67  /* Keep the values in world-space since we're passing the `obmat`. */
68  if (!EDBM_op_init(em,
69  &spinop,
70  op,
71  "spin geom=%hvef cent=%v axis=%v dvec=%v steps=%i angle=%f space=%m4 "
72  "use_normal_flip=%b use_duplicate=%b use_merge=%b",
74  cent,
75  axis,
76  d,
77  steps,
78  -angle,
79  obedit->obmat,
80  use_normal_flip,
81  dupli,
82  use_auto_merge)) {
83  continue;
84  }
85  BMO_op_exec(bm, &spinop);
86  if (use_auto_merge == false) {
89  bm, spinop.slots_out, "geom_last.out", BM_ALL_NOLOOP, BM_ELEM_SELECT, true);
90  }
91  if (!EDBM_op_finish(em, &spinop, op, true)) {
92  continue;
93  }
94 
95  EDBM_update(obedit->data,
96  &(const struct EDBMUpdate_Params){
97  .calc_looptri = true,
98  .calc_normals = false,
99  .is_destructive = true,
100  });
101  }
102 
103  MEM_freeN(objects);
104 
105  return OPERATOR_FINISHED;
106 }
107 
108 /* get center and axis, in global coords */
109 static int edbm_spin_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
110 {
112  View3D *v3d = CTX_wm_view3d(C);
114 
115  PropertyRNA *prop;
116  prop = RNA_struct_find_property(op->ptr, "center");
117  if (!RNA_property_is_set(op->ptr, prop)) {
119  }
120  if (rv3d) {
121  prop = RNA_struct_find_property(op->ptr, "axis");
122  if (!RNA_property_is_set(op->ptr, prop)) {
123  RNA_property_float_set_array(op->ptr, prop, rv3d->viewinv[2]);
124  }
125  }
126 
127 #ifdef USE_GIZMO
128  /* Start with zero angle, drag out the value. */
129  prop = RNA_struct_find_property(op->ptr, "angle");
130  if (!RNA_property_is_set(op->ptr, prop)) {
131  RNA_property_float_set(op->ptr, prop, 0.0f);
132  }
133 #endif
134 
135  int ret = edbm_spin_exec(C, op);
136 
137 #ifdef USE_GIZMO
138  if (ret & OPERATOR_FINISHED) {
139  /* Setup gizmos */
140  if (v3d && ((v3d->gizmo_flag & V3D_GIZMO_HIDE) == 0)) {
141  wmGizmoGroupType *gzgt = WM_gizmogrouptype_find("MESH_GGT_spin_redo", false);
142  if (!WM_gizmo_group_type_ensure_ptr(gzgt)) {
143  struct Main *bmain = CTX_data_main(C);
144  WM_gizmo_group_type_reinit_ptr(bmain, gzgt);
145  }
146  }
147  }
148 #endif
149 
150  return ret;
151 }
152 
154  wmOperator *op,
155  const PropertyRNA *prop)
156 {
157  const char *prop_id = RNA_property_identifier(prop);
158  const bool dupli = RNA_boolean_get(op->ptr, "dupli");
159 
160  if (dupli) {
161  if (STR_ELEM(prop_id, "use_auto_merge", "use_normal_flip")) {
162  return false;
163  }
164  }
165  return true;
166 }
167 
169 {
170  PropertyRNA *prop;
171 
172  /* identifiers */
173  ot->name = "Spin";
174  ot->description =
175  "Extrude selected vertices in a circle around the cursor in indicated viewport";
176  ot->idname = "MESH_OT_spin";
177 
178  /* api callbacks */
183 
184  /* flags */
186 
187  /* props */
188  RNA_def_int(ot->srna, "steps", 12, 0, 1000000, "Steps", "Steps", 0, 1000);
189 
190  prop = RNA_def_boolean(ot->srna, "dupli", 0, "Use Duplicates", "");
192 
193  prop = RNA_def_float(ot->srna,
194  "angle",
195  DEG2RADF(90.0f),
196  -1e12f,
197  1e12f,
198  "Angle",
199  "Rotation for each step",
200  DEG2RADF(-360.0f),
201  DEG2RADF(360.0f));
204  "use_auto_merge",
205  true,
206  "Auto Merge",
207  "Merge first/last when the angle is a full revolution");
208  RNA_def_boolean(ot->srna, "use_normal_flip", 0, "Flip Normals", "");
209 
211  "center",
212  3,
213  NULL,
214  -1e12f,
215  1e12f,
216  "Center",
217  "Center in global view space",
218  -1e4f,
219  1e4f);
221  ot->srna, "axis", 3, NULL, -1.0f, 1.0f, "Axis", "Axis in global view space", -1.0f, 1.0f);
222 
224 #ifdef USE_GIZMO
226 #endif
227 }
228 
struct Scene * CTX_data_scene(const bContext *C)
Definition: context.c:1090
struct ViewLayer * CTX_data_view_layer(const bContext *C)
Definition: context.c:1100
struct View3D * CTX_wm_view3d(const bContext *C)
Definition: context.c:784
struct Main * CTX_data_main(const bContext *C)
Definition: context.c:1074
BMEditMesh * BKE_editmesh_from_object(struct Object *ob)
Return the BMEditMesh for a given object.
Definition: editmesh.c:58
#define BKE_view_layer_array_from_objects_in_edit_mode_unique_data(view_layer, v3d, r_len)
Definition: BKE_layer.h:542
void BKE_report(ReportList *reports, eReportType type, const char *message)
Definition: report.c:83
#define M_PI
Definition: BLI_math_base.h:20
#define DEG2RADF(_deg)
MINLINE bool is_zero_v3(const float a[3]) ATTR_WARN_UNUSED_RESULT
#define STR_ELEM(...)
Definition: BLI_string.h:539
unsigned int uint
Definition: BLI_sys_types.h:67
#define UNUSED(x)
Object is a sort of wrapper for general info.
@ V3D_GIZMO_HIDE
@ OPERATOR_CANCELLED
@ OPERATOR_FINISHED
void EDBM_update(struct Mesh *me, const struct EDBMUpdate_Params *params)
void EDBM_flag_disable_all(struct BMEditMesh *em, char hflag)
bool ED_operator_editmesh(struct bContext *C)
Definition: screen_ops.c:433
struct RegionView3D * ED_view3d_context_rv3d(struct bContext *C)
Definition: space_view3d.c:82
Read Guarded memory(de)allocation.
@ PROP_SKIP_SAVE
Definition: RNA_types.h:218
@ PROP_ANGLE
Definition: RNA_types.h:145
#define C
Definition: RandGen.cpp:25
@ OPTYPE_UNDO
Definition: WM_types.h:148
@ OPTYPE_REGISTER
Definition: WM_types.h:146
#define BM_ALL_NOLOOP
Definition: bmesh_class.h:411
@ BM_ELEM_SELECT
Definition: bmesh_class.h:471
ATTR_WARN_UNUSED_RESULT BMesh * bm
void BMO_slot_buffer_hflag_enable(BMesh *bm, BMOpSlot slot_args[BMO_OP_MAX_SLOTS], const char *slot_name, char htype, char hflag, bool do_flush)
BMO_FLAG_BUFFER.
void BMO_op_exec(BMesh *bm, BMOperator *op)
BMESH OPSTACK EXEC OP.
SIMD_FORCE_INLINE btScalar angle(const btVector3 &v) const
Return the angle between this and another vector.
Definition: btVector3.h:356
Scene scene
static int edbm_spin_exec(bContext *C, wmOperator *op)
static int edbm_spin_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
static bool edbm_spin_poll_property(const bContext *UNUSED(C), wmOperator *op, const PropertyRNA *prop)
void MESH_OT_spin(wmOperatorType *ot)
void MESH_GGT_spin(struct wmGizmoGroupType *gzgt)
void MESH_GGT_spin_redo(struct wmGizmoGroupType *gzgt)
bool EDBM_op_init(BMEditMesh *em, BMOperator *bmop, wmOperator *op, const char *fmt,...)
bool EDBM_op_finish(BMEditMesh *em, BMOperator *bmop, wmOperator *op, const bool do_report)
void(* MEM_freeN)(void *vmemh)
Definition: mallocn.c:27
#define fabsf(x)
Definition: metal/compat.h:219
return ret
const char * RNA_property_identifier(const PropertyRNA *prop)
Definition: rna_access.c:1000
bool RNA_property_is_set(PointerRNA *ptr, PropertyRNA *prop)
Definition: rna_access.c:5271
PropertyRNA * RNA_struct_find_property(PointerRNA *ptr, const char *identifier)
Definition: rna_access.c:717
void RNA_float_get_array(PointerRNA *ptr, const char *name, float *values)
Definition: rna_access.c:4980
int RNA_int_get(PointerRNA *ptr, const char *name)
Definition: rna_access.c:4910
float RNA_float_get(PointerRNA *ptr, const char *name)
Definition: rna_access.c:4957
void RNA_property_float_set_array(PointerRNA *ptr, PropertyRNA *prop, const float *values)
Definition: rna_access.c:2978
void RNA_property_float_set(PointerRNA *ptr, PropertyRNA *prop, float value)
Definition: rna_access.c:2790
bool RNA_boolean_get(PointerRNA *ptr, const char *name)
Definition: rna_access.c:4863
PropertyRNA * RNA_def_float(StructOrFunctionRNA *cont_, const char *identifier, float default_value, float hardmin, float hardmax, const char *ui_name, const char *ui_description, float softmin, float softmax)
Definition: rna_define.c:3836
PropertyRNA * RNA_def_boolean(StructOrFunctionRNA *cont_, const char *identifier, bool default_value, const char *ui_name, const char *ui_description)
Definition: rna_define.c:3493
PropertyRNA * RNA_def_float_vector(StructOrFunctionRNA *cont_, const char *identifier, int len, const float *default_value, float hardmin, float hardmax, const char *ui_name, const char *ui_description, float softmin, float softmax)
Definition: rna_define.c:3862
PropertyRNA * RNA_def_float_vector_xyz(StructOrFunctionRNA *cont_, const char *identifier, int len, const float *default_value, float hardmin, float hardmax, const char *ui_name, const char *ui_description, float softmin, float softmax)
Definition: rna_define.c:3894
void RNA_def_property_flag(PropertyRNA *prop, PropertyFlag flag)
Definition: rna_define.c:1490
PropertyRNA * RNA_def_int(StructOrFunctionRNA *cont_, const char *identifier, int default_value, int hardmin, int hardmax, const char *ui_name, const char *ui_description, int softmin, int softmax)
Definition: rna_define.c:3597
void RNA_def_property_subtype(PropertyRNA *prop, PropertySubType subtype)
Definition: rna_define.c:1534
static const int steps
Definition: sky_nishita.cpp:19
struct BMesh * bm
Definition: BKE_editmesh.h:40
struct BMOpSlot slots_out[BMO_OP_MAX_SLOTS]
Definition: BKE_main.h:121
float obmat[4][4]
void * data
float viewinv[4][4]
View3DCursor cursor
char gizmo_flag
int(* invoke)(struct bContext *, struct wmOperator *, const struct wmEvent *) ATTR_WARN_UNUSED_RESULT
Definition: WM_types.h:919
const char * name
Definition: WM_types.h:888
const char * idname
Definition: WM_types.h:890
bool(* poll)(struct bContext *) ATTR_WARN_UNUSED_RESULT
Definition: WM_types.h:943
bool(* poll_property)(const struct bContext *C, struct wmOperator *op, const PropertyRNA *prop) ATTR_WARN_UNUSED_RESULT
Definition: WM_types.h:949
struct StructRNA * srna
Definition: WM_types.h:969
const char * description
Definition: WM_types.h:893
int(* exec)(struct bContext *, struct wmOperator *) ATTR_WARN_UNUSED_RESULT
Definition: WM_types.h:903
struct ReportList * reports
struct PointerRNA * ptr
wmOperatorType * ot
Definition: wm_files.c:3479
bool WM_gizmo_group_type_ensure_ptr(wmGizmoGroupType *gzgt)
void WM_gizmo_group_type_reinit_ptr(struct Main *bmain, wmGizmoGroupType *gzgt)
wmGizmoGroupType * WM_gizmogrouptype_find(const char *idname, bool quiet)
wmGizmoGroupType * WM_gizmogrouptype_append(void(*wtfunc)(struct wmGizmoGroupType *))