Merge branch 'master' into blender2.8
[blender.git] / source / blender / editors / interface / interface_region_menu_pie.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  * The Original Code is Copyright (C) 2008 Blender Foundation.
19  * All rights reserved.
20  *
21  * Contributor(s): Blender Foundation
22  *
23  * ***** END GPL LICENSE BLOCK *****
24  */
25
26 /** \file blender/editors/interface/interface_region_menu_pie.c
27  *  \ingroup edinterface
28  *
29  * Pie Menu Region
30  */
31
32 #include <stdarg.h>
33 #include <stdlib.h>
34 #include <string.h>
35 #include <assert.h>
36
37 #include "MEM_guardedalloc.h"
38
39 #include "DNA_userdef_types.h"
40
41 #include "BLI_blenlib.h"
42 #include "BLI_utildefines.h"
43
44 #include "PIL_time.h"
45
46 #include "BKE_context.h"
47 #include "BKE_screen.h"
48
49 #include "WM_api.h"
50 #include "WM_types.h"
51
52 #include "RNA_access.h"
53
54 #include "UI_interface.h"
55
56 #include "BLT_translation.h"
57
58 #include "ED_screen.h"
59
60 #include "interface_intern.h"
61 #include "interface_regions_intern.h"
62
63 /* -------------------------------------------------------------------- */
64 /** \name Pie Menu
65  * \{ */
66
67 struct uiPieMenu {
68         uiBlock *block_radial; /* radial block of the pie menu (more could be added later) */
69         uiLayout *layout;
70         int mx, my;
71 };
72
73 static uiBlock *ui_block_func_PIE(bContext *UNUSED(C), uiPopupBlockHandle *handle, void *arg_pie)
74 {
75         uiBlock *block;
76         uiPieMenu *pie = arg_pie;
77         int minwidth, width, height;
78
79         minwidth = UI_MENU_WIDTH_MIN;
80         block = pie->block_radial;
81
82         /* in some cases we create the block before the region,
83          * so we set it delayed here if necessary */
84         if (BLI_findindex(&handle->region->uiblocks, block) == -1)
85                 UI_block_region_set(block, handle->region);
86
87         UI_block_layout_resolve(block, &width, &height);
88
89         UI_block_flag_enable(block, UI_BLOCK_LOOP | UI_BLOCK_NUMSELECT);
90
91         block->minbounds = minwidth;
92         block->bounds = 1;
93         block->mx = 0;
94         block->my = 0;
95         block->bounds_type = UI_BLOCK_BOUNDS_PIE_CENTER;
96
97         block->pie_data.pie_center_spawned[0] = pie->mx;
98         block->pie_data.pie_center_spawned[1] = pie->my;
99
100         return pie->block_radial;
101 }
102
103 static float ui_pie_menu_title_width(const char *name, int icon)
104 {
105         const uiFontStyle *fstyle = UI_FSTYLE_WIDGET;
106         return (UI_fontstyle_string_width(fstyle, name) +
107                  (UI_UNIT_X * (1.50f + (icon ? 0.25f : 0.0f))));
108 }
109
110 uiPieMenu *UI_pie_menu_begin(struct bContext *C, const char *title, int icon, const wmEvent *event)
111 {
112         uiStyle *style;
113         uiPieMenu *pie;
114         short event_type;
115
116         wmWindow *win = CTX_wm_window(C);
117
118         style = UI_style_get_dpi();
119         pie = MEM_callocN(sizeof(*pie), "pie menu");
120
121         pie->block_radial = UI_block_begin(C, NULL, __func__, UI_EMBOSS);
122         /* may be useful later to allow spawning pies
123          * from old positions */
124         /* pie->block_radial->flag |= UI_BLOCK_POPUP_MEMORY; */
125         pie->block_radial->puphash = ui_popup_menu_hash(title);
126         pie->block_radial->flag |= UI_BLOCK_RADIAL;
127
128         /* if pie is spawned by a left click, release or click event, it is always assumed to be click style */
129         if (event->type == LEFTMOUSE || ELEM(event->val, KM_RELEASE, KM_CLICK)) {
130                 pie->block_radial->pie_data.flags |= UI_PIE_CLICK_STYLE;
131                 pie->block_radial->pie_data.event = EVENT_NONE;
132                 win->lock_pie_event = EVENT_NONE;
133         }
134         else {
135                 if (win->last_pie_event != EVENT_NONE) {
136                         /* original pie key has been released, so don't propagate the event */
137                         if (win->lock_pie_event == EVENT_NONE) {
138                                 event_type = EVENT_NONE;
139                                 pie->block_radial->pie_data.flags |= UI_PIE_CLICK_STYLE;
140                         }
141                         else
142                                 event_type = win->last_pie_event;
143                 }
144                 else {
145                         event_type = event->type;
146                 }
147
148                 pie->block_radial->pie_data.event = event_type;
149                 win->lock_pie_event = event_type;
150         }
151
152         pie->layout = UI_block_layout(pie->block_radial, UI_LAYOUT_VERTICAL, UI_LAYOUT_PIEMENU, 0, 0, 200, 0, 0, style);
153
154         /* Open from where we started dragging. */
155         if (event->val == KM_CLICK_DRAG) {
156                 pie->mx = event->prevclickx;
157                 pie->my = event->prevclicky;
158         }
159         else {
160                 pie->mx = event->x;
161                 pie->my = event->y;
162         }
163
164         /* create title button */
165         if (title[0]) {
166                 uiBut *but;
167                 char titlestr[256];
168                 int w;
169                 if (icon) {
170                         BLI_snprintf(titlestr, sizeof(titlestr), " %s", title);
171                         w = ui_pie_menu_title_width(titlestr, icon);
172                         but = uiDefIconTextBut(
173                                 pie->block_radial, UI_BTYPE_LABEL, 0, icon, titlestr, 0, 0, w, UI_UNIT_Y,
174                                 NULL, 0.0, 0.0, 0, 0, "");
175                 }
176                 else {
177                         w = ui_pie_menu_title_width(title, 0);
178                         but = uiDefBut(
179                                 pie->block_radial, UI_BTYPE_LABEL, 0, title, 0, 0, w, UI_UNIT_Y,
180                                 NULL, 0.0, 0.0, 0, 0, "");
181                 }
182                 /* do not align left */
183                 but->drawflag &= ~UI_BUT_TEXT_LEFT;
184                 pie->block_radial->pie_data.title = but->str;
185                 pie->block_radial->pie_data.icon = icon;
186         }
187
188         return pie;
189 }
190
191 void UI_pie_menu_end(bContext *C, uiPieMenu *pie)
192 {
193         wmWindow *window = CTX_wm_window(C);
194         uiPopupBlockHandle *menu;
195
196         menu = ui_popup_block_create(C, NULL, NULL, NULL, ui_block_func_PIE, pie);
197         menu->popup = true;
198         menu->towardstime = PIL_check_seconds_timer();
199
200         UI_popup_handlers_add(
201                 C, &window->modalhandlers,
202                 menu, WM_HANDLER_ACCEPT_DBL_CLICK);
203         WM_event_add_mousemove(C);
204
205         MEM_freeN(pie);
206 }
207
208 uiLayout *UI_pie_menu_layout(uiPieMenu *pie)
209 {
210         return pie->layout;
211 }
212
213 int UI_pie_menu_invoke(struct bContext *C, const char *idname, const wmEvent *event)
214 {
215         uiPieMenu *pie;
216         uiLayout *layout;
217         MenuType *mt = WM_menutype_find(idname, true);
218
219         if (mt == NULL) {
220                 printf("%s: named menu \"%s\" not found\n", __func__, idname);
221                 return OPERATOR_CANCELLED;
222         }
223
224         if (WM_menutype_poll(C, mt) == false) {
225                 /* cancel but allow event to pass through, just like operators do */
226                 return (OPERATOR_CANCELLED | OPERATOR_PASS_THROUGH);
227         }
228
229         pie = UI_pie_menu_begin(C, IFACE_(mt->label), ICON_NONE, event);
230         layout = UI_pie_menu_layout(pie);
231
232         UI_menutype_draw(C, mt, layout);
233
234         UI_pie_menu_end(C, pie);
235
236         return OPERATOR_INTERFACE;
237 }
238
239 int UI_pie_menu_invoke_from_operator_enum(
240         struct bContext *C, const char *title, const char *opname,
241         const char *propname, const wmEvent *event)
242 {
243         uiPieMenu *pie;
244         uiLayout *layout;
245
246         pie = UI_pie_menu_begin(C, IFACE_(title), ICON_NONE, event);
247         layout = UI_pie_menu_layout(pie);
248
249         layout = uiLayoutRadial(layout);
250         uiItemsEnumO(layout, opname, propname);
251
252         UI_pie_menu_end(C, pie);
253
254         return OPERATOR_INTERFACE;
255 }
256
257 int UI_pie_menu_invoke_from_rna_enum(
258         struct bContext *C, const char *title, const char *path,
259         const wmEvent *event)
260 {
261         PointerRNA ctx_ptr;
262         PointerRNA r_ptr;
263         PropertyRNA *r_prop;
264         uiPieMenu *pie;
265         uiLayout *layout;
266
267         RNA_pointer_create(NULL, &RNA_Context, C, &ctx_ptr);
268
269         if (!RNA_path_resolve(&ctx_ptr, path, &r_ptr, &r_prop)) {
270                 return OPERATOR_CANCELLED;
271         }
272
273         /* invalid property, only accept enums */
274         if (RNA_property_type(r_prop) != PROP_ENUM) {
275                 BLI_assert(0);
276                 return OPERATOR_CANCELLED;
277         }
278
279         pie = UI_pie_menu_begin(C, IFACE_(title), ICON_NONE, event);
280
281         layout = UI_pie_menu_layout(pie);
282
283         layout = uiLayoutRadial(layout);
284         uiItemFullR(layout, &r_ptr, r_prop, RNA_NO_INDEX, 0, UI_ITEM_R_EXPAND, NULL, 0);
285
286         UI_pie_menu_end(C, pie);
287
288         return OPERATOR_INTERFACE;
289 }
290
291 /** \} */
292
293 /* -------------------------------------------------------------------- */
294 /**
295  * \name Pie Menu Levels
296  *
297  * Pie menus can't contain more than 8 items (yet). When using #uiItemsFullEnumO, a "More" button is created that calls
298  * a new pie menu if the enum has too many items. We call this a new "level".
299  * Indirect recursion is used, so that a theoretically unlimited number of items is supported.
300  *
301  * This is a implementation specifically for operator enums, needed since the object mode pie now has more than 8
302  * items. Ideally we'd have some way of handling this for all kinds of pie items, but that's tricky.
303  *
304  * - Julian (Feb 2016)
305  *
306  * \{ */
307
308 typedef struct PieMenuLevelData {
309         char title[UI_MAX_NAME_STR]; /* parent pie title, copied for level */
310         int icon; /* parent pie icon, copied for level */
311         int totitem; /* total count of *remaining* items */
312
313         /* needed for calling uiItemsFullEnumO_array again for new level */
314         wmOperatorType *ot;
315         const char *propname;
316         IDProperty *properties;
317         int context, flag;
318 } PieMenuLevelData;
319
320 /**
321  * Invokes a new pie menu for a new level.
322  */
323 static void ui_pie_menu_level_invoke(bContext *C, void *argN, void *arg2)
324 {
325         EnumPropertyItem *item_array = (EnumPropertyItem *)argN;
326         PieMenuLevelData *lvl = (PieMenuLevelData *)arg2;
327         wmWindow *win = CTX_wm_window(C);
328
329         uiPieMenu *pie = UI_pie_menu_begin(C, IFACE_(lvl->title), lvl->icon, win->eventstate);
330         uiLayout *layout = UI_pie_menu_layout(pie);
331
332         layout = uiLayoutRadial(layout);
333
334         PointerRNA ptr;
335
336         WM_operator_properties_create_ptr(&ptr, lvl->ot);
337         /* so the context is passed to itemf functions (some need it) */
338         WM_operator_properties_sanitize(&ptr, false);
339         PropertyRNA *prop = RNA_struct_find_property(&ptr, lvl->propname);
340
341         if (prop) {
342                 uiItemsFullEnumO_items(
343                         layout, lvl->ot, ptr, prop, lvl->properties, lvl->context, lvl->flag,
344                         item_array, lvl->totitem);
345         }
346         else {
347                 RNA_warning("%s.%s not found", RNA_struct_identifier(ptr.type), lvl->propname);
348         }
349
350         UI_pie_menu_end(C, pie);
351 }
352
353 /**
354  * Set up data for defining a new pie menu level and add button that invokes it.
355  */
356 void ui_pie_menu_level_create(
357         uiBlock *block, wmOperatorType *ot, const char *propname, IDProperty *properties,
358         const EnumPropertyItem *items, int totitem, int context, int flag)
359 {
360         const int totitem_parent = PIE_MAX_ITEMS - 1;
361         const int totitem_remain = totitem - totitem_parent;
362         size_t array_size = sizeof(EnumPropertyItem) * totitem_remain;
363
364         /* used as but->func_argN so freeing is handled elsewhere */
365         EnumPropertyItem *remaining = MEM_mallocN(array_size + sizeof(EnumPropertyItem), "pie_level_item_array");
366         memcpy(remaining, items + totitem_parent, array_size);
367         /* a NULL terminating sentinal element is required */
368         memset(&remaining[totitem_remain], 0, sizeof(EnumPropertyItem));
369
370
371         /* yuk, static... issue is we can't reliably free this without doing dangerous changes */
372         static PieMenuLevelData lvl;
373         BLI_strncpy(lvl.title, block->pie_data.title, UI_MAX_NAME_STR);
374         lvl.totitem    = totitem_remain;
375         lvl.ot         = ot;
376         lvl.propname   = propname;
377         lvl.properties = properties;
378         lvl.context    = context;
379         lvl.flag       = flag;
380
381         /* add a 'more' menu entry */
382         uiBut *but = uiDefIconTextBut(block, UI_BTYPE_BUT, 0, ICON_PLUS, "More", 0, 0, UI_UNIT_X * 3, UI_UNIT_Y, NULL,
383                                       0.0f, 0.0f, 0.0f, 0.0f, "Show more items of this menu");
384         UI_but_funcN_set(but, ui_pie_menu_level_invoke, remaining, &lvl);
385 }
386
387 /** \} */ /* Pie Menu Levels */