Merge branch 'master' into blender2.8
[blender.git] / source / blender / editors / space_view3d / view3d_header.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) 2004-2008 Blender Foundation.
19  * All rights reserved.
20  *
21  *
22  * Contributor(s): Blender Foundation
23  *
24  * ***** END GPL LICENSE BLOCK *****
25  */
26
27 /** \file blender/editors/space_view3d/view3d_header.c
28  *  \ingroup spview3d
29  */
30
31 #include <string.h>
32 #include <stdio.h>
33 #include <stdlib.h>
34
35 #include "DNA_scene_types.h"
36 #include "DNA_object_types.h"
37 #include "DNA_gpencil_types.h"
38
39 #include "BLI_utildefines.h"
40
41 #include "BLT_translation.h"
42
43 #include "BKE_context.h"
44 #include "BKE_main.h"
45 #include "BKE_screen.h"
46 #include "BKE_editmesh.h"
47
48 #include "DEG_depsgraph.h"
49
50 #include "RNA_access.h"
51 #include "RNA_define.h"
52 #include "RNA_enum_types.h"
53
54 #include "WM_api.h"
55 #include "WM_types.h"
56
57 #include "ED_mesh.h"
58 #include "ED_undo.h"
59 #include "ED_screen.h"
60
61 #include "UI_interface.h"
62 #include "UI_resources.h"
63
64 #include "view3d_intern.h"
65
66 static void do_view3d_header_buttons(bContext *C, void *arg, int event);
67
68 #define B_SEL_VERT  110
69 #define B_SEL_EDGE  111
70 #define B_SEL_FACE  112
71
72 /* XXX quickly ported across */
73 static void handle_view3d_lock(bContext *C)
74 {
75         Scene *scene = CTX_data_scene(C);
76         ScrArea *sa = CTX_wm_area(C);
77         View3D *v3d = CTX_wm_view3d(C);
78
79         if (v3d != NULL && sa != NULL) {
80                 if (v3d->localvd == NULL && v3d->scenelock && sa->spacetype == SPACE_VIEW3D) {
81                         /* copy to scene */
82                         scene->lay = v3d->lay;
83                         scene->layact = v3d->layact;
84                         scene->camera = v3d->camera;
85
86                         /* notifiers for scene update */
87                         WM_event_add_notifier(C, NC_SCENE | ND_LAYER, scene);
88                 }
89         }
90 }
91
92 /**
93  * layer code is on three levels actually:
94  * - here for operator
95  * - uiTemplateLayers in interface/ code for buttons
96  * - ED_view3d_view_layer_set for RNA
97  */
98 static void view3d_layers_editmode_ensure(View3D *v3d, Object *obedit)
99 {
100         /* sanity check - when in editmode disallow switching the editmode layer off since its confusing
101          * an alternative would be to always draw the editmode object. */
102         if (obedit && (obedit->lay & v3d->lay) == 0) {
103                 int bit;
104                 for (bit = 0; bit < 32; bit++) {
105                         if (obedit->lay & (1u << bit)) {
106                                 v3d->lay |= (1u << bit);
107                                 break;
108                         }
109                 }
110         }
111 }
112
113 static int view3d_layers_exec(bContext *C, wmOperator *op)
114 {
115         ScrArea *sa = CTX_wm_area(C);
116         View3D *v3d = sa->spacedata.first;
117         Object *obedit = CTX_data_edit_object(C);
118         int nr = RNA_int_get(op->ptr, "nr");
119         const bool toggle = RNA_boolean_get(op->ptr, "toggle");
120
121         if (nr < 0)
122                 return OPERATOR_CANCELLED;
123
124         if (nr == 0) {
125                 /* all layers */
126                 if (!v3d->lay_prev)
127                         v3d->lay_prev = 1;
128
129                 if (toggle && v3d->lay == ((1 << 20) - 1)) {
130                         /* return to active layer only */
131                         v3d->lay = v3d->lay_prev;
132
133                         view3d_layers_editmode_ensure(v3d, obedit);
134                 }
135                 else {
136                         v3d->lay_prev = v3d->lay;
137                         v3d->lay |= (1 << 20) - 1;
138                 }
139         }
140         else {
141                 int bit;
142                 nr--;
143
144                 if (RNA_boolean_get(op->ptr, "extend")) {
145                         if (toggle && v3d->lay & (1 << nr) && (v3d->lay & ~(1 << nr)))
146                                 v3d->lay &= ~(1 << nr);
147                         else
148                                 v3d->lay |= (1 << nr);
149                 }
150                 else {
151                         v3d->lay = (1 << nr);
152                 }
153
154                 view3d_layers_editmode_ensure(v3d, obedit);
155
156                 /* set active layer, ensure to always have one */
157                 if (v3d->lay & (1 << nr))
158                         v3d->layact = 1 << nr;
159                 else if ((v3d->lay & v3d->layact) == 0) {
160                         for (bit = 0; bit < 32; bit++) {
161                                 if (v3d->lay & (1u << bit)) {
162                                         v3d->layact = (1u << bit);
163                                         break;
164                                 }
165                         }
166                 }
167         }
168
169         if (v3d->scenelock) handle_view3d_lock(C);
170
171         DEG_on_visible_update(CTX_data_main(C), false);
172
173         ED_area_tag_redraw(sa);
174
175         return OPERATOR_FINISHED;
176 }
177
178 /* applies shift and alt, lazy coding or ok? :) */
179 /* the local per-keymap-entry keymap will solve it */
180 static int view3d_layers_invoke(bContext *C, wmOperator *op, const wmEvent *event)
181 {
182         if (event->ctrl || event->oskey)
183                 return OPERATOR_PASS_THROUGH;
184
185         if (event->shift)
186                 RNA_boolean_set(op->ptr, "extend", true);
187         else
188                 RNA_boolean_set(op->ptr, "extend", false);
189
190         if (event->alt) {
191                 const int nr = RNA_int_get(op->ptr, "nr") + 10;
192                 RNA_int_set(op->ptr, "nr", nr);
193         }
194         view3d_layers_exec(C, op);
195
196         return OPERATOR_FINISHED;
197 }
198
199 static int view3d_layers_poll(bContext *C)
200 {
201         return (ED_operator_view3d_active(C) && CTX_wm_view3d(C)->localvd == NULL);
202 }
203
204 void VIEW3D_OT_layers(wmOperatorType *ot)
205 {
206         /* identifiers */
207         ot->name = "Layers";
208         ot->description = "Toggle layer(s) visibility";
209         ot->idname = "VIEW3D_OT_layers";
210
211         /* api callbacks */
212         ot->invoke = view3d_layers_invoke;
213         ot->exec = view3d_layers_exec;
214         ot->poll = view3d_layers_poll;
215
216         /* flags */
217         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
218
219         RNA_def_int(ot->srna, "nr", 1, 0, 20, "Number", "The layer number to set, zero for all layers", 0, 20);
220         RNA_def_boolean(ot->srna, "extend", 0, "Extend", "Add this layer to the current view layers");
221         RNA_def_boolean(ot->srna, "toggle", 1, "Toggle", "Toggle the layer");
222 }
223
224 /* -------------------------------------------------------------------- */
225 /** \name Toggle Bone selection Overlay Operator
226  * \{ */
227
228 static int toggle_show_xray(bContext *C, wmOperator *UNUSED(op))
229 {
230         View3D *v3d = CTX_wm_view3d(C);
231         v3d->shading.flag ^= V3D_SHADING_XRAY;
232         ED_view3d_shade_update(CTX_data_main(C), v3d, CTX_wm_area(C));
233         WM_event_add_notifier(C, NC_SPACE | ND_SPACE_VIEW3D, v3d);
234         return OPERATOR_FINISHED;
235 }
236
237 static int toggle_show_xray_poll(bContext *C)
238 {
239         bool result = (ED_operator_view3d_active(C) && !ED_operator_posemode(C) && !ED_operator_editmesh(C));
240         if (result) {
241                 // Additional test for SOLID or TEXTURE mode
242                 View3D *v3d = CTX_wm_view3d(C);
243                 result = (v3d->drawtype & (OB_SOLID | OB_TEXTURE)) > 0;
244         }
245         return result;
246 }
247
248 void VIEW3D_OT_toggle_xray_draw_option(wmOperatorType *ot)
249 {
250         /* identifiers */
251         ot->name = "Toggle Show X-Ray";
252         ot->description = "Toggle show X-Ray";
253         ot->idname = "VIEW3D_OT_toggle_xray_draw_option";
254
255         /* api callbacks */
256         ot->exec = toggle_show_xray;
257         ot->poll = toggle_show_xray_poll;
258 }
259
260 /** \} */
261
262
263 static void do_view3d_header_buttons(bContext *C, void *UNUSED(arg), int event)
264 {
265         wmWindow *win = CTX_wm_window(C);
266         const int ctrl = win->eventstate->ctrl, shift = win->eventstate->shift;
267
268         /* watch it: if sa->win does not exist, check that when calling direct drawing routines */
269
270         switch (event) {
271                 case B_SEL_VERT:
272                         if (EDBM_selectmode_toggle(C, SCE_SELECT_VERTEX, -1, shift, ctrl)) {
273                                 ED_undo_push(C, "Selectmode Set: Vertex");
274                         }
275                         break;
276                 case B_SEL_EDGE:
277                         if (EDBM_selectmode_toggle(C, SCE_SELECT_EDGE, -1, shift, ctrl)) {
278                                 ED_undo_push(C, "Selectmode Set: Edge");
279                         }
280                         break;
281                 case B_SEL_FACE:
282                         if (EDBM_selectmode_toggle(C, SCE_SELECT_FACE, -1, shift, ctrl)) {
283                                 ED_undo_push(C, "Selectmode Set: Face");
284                         }
285                         break;
286                 default:
287                         break;
288         }
289 }
290
291 void uiTemplateEditModeSelection(uiLayout *layout, struct bContext *C)
292 {
293         Object *obedit = CTX_data_edit_object(C);
294         uiBlock *block = uiLayoutGetBlock(layout);
295
296         UI_block_func_handle_set(block, do_view3d_header_buttons, NULL);
297
298         if (obedit && (obedit->type == OB_MESH)) {
299                 BMEditMesh *em = BKE_editmesh_from_object(obedit);
300                 uiLayout *row;
301
302                 row = uiLayoutRow(layout, true);
303                 block = uiLayoutGetBlock(row);
304                 uiDefIconButBitS(block, UI_BTYPE_TOGGLE, SCE_SELECT_VERTEX, B_SEL_VERT, ICON_VERTEXSEL,
305                                  0, 0, UI_UNIT_X, UI_UNIT_Y, &em->selectmode, 1.0, 0.0, 0, 0,
306                                  TIP_("Vertex select - Shift-Click for multiple modes, Ctrl-Click contracts selection"));
307                 uiDefIconButBitS(block, UI_BTYPE_TOGGLE, SCE_SELECT_EDGE, B_SEL_EDGE, ICON_EDGESEL,
308                                  0, 0, UI_UNIT_X, UI_UNIT_Y, &em->selectmode, 1.0, 0.0, 0, 0,
309                                  TIP_("Edge select - Shift-Click for multiple modes, Ctrl-Click expands/contracts selection"));
310                 uiDefIconButBitS(block, UI_BTYPE_TOGGLE, SCE_SELECT_FACE, B_SEL_FACE, ICON_FACESEL,
311                                  0, 0, UI_UNIT_X, UI_UNIT_Y, &em->selectmode, 1.0, 0.0, 0, 0,
312                                  TIP_("Face select - Shift-Click for multiple modes, Ctrl-Click expands selection"));
313         }
314 }
315
316 static void uiTemplatePaintModeSelection(uiLayout *layout, struct bContext *C)
317 {
318         ViewLayer *view_layer = CTX_data_view_layer(C);
319         Object *ob = OBACT(view_layer);
320
321         /* Manipulators aren't used in paint modes */
322         if (!ELEM(ob->mode, OB_MODE_SCULPT, OB_MODE_PARTICLE_EDIT)) {
323                 /* masks aren't used for sculpt and particle painting */
324                 PointerRNA meshptr;
325
326                 RNA_pointer_create(ob->data, &RNA_Mesh, ob->data, &meshptr);
327                 if (ob->mode & (OB_MODE_TEXTURE_PAINT)) {
328                         uiItemR(layout, &meshptr, "use_paint_mask", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
329                 }
330                 else {
331                         uiLayout *row = uiLayoutRow(layout, true);
332                         uiItemR(row, &meshptr, "use_paint_mask", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
333                         uiItemR(row, &meshptr, "use_paint_mask_vertex", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
334                 }
335         }
336 }
337
338 void uiTemplateHeader3D_mode(uiLayout *layout, struct bContext *C)
339 {
340         /* Extracted from: uiTemplateHeader3D */
341         ViewLayer *view_layer = CTX_data_view_layer(C);
342         Object *ob = OBACT(view_layer);
343         Object *obedit = CTX_data_edit_object(C);
344         bGPdata *gpd = CTX_data_gpencil_data(C);
345
346         bool is_paint = (
347                 ob && !(gpd && (gpd->flag & GP_DATA_STROKE_EDITMODE)) &&
348                 ELEM(ob->mode,
349                      OB_MODE_SCULPT, OB_MODE_VERTEX_PAINT, OB_MODE_WEIGHT_PAINT, OB_MODE_TEXTURE_PAINT));
350
351         uiTemplateEditModeSelection(layout, C);
352         if ((obedit == NULL) && is_paint) {
353                 uiTemplatePaintModeSelection(layout, C);
354         }
355 }
356
357 void uiTemplateHeader3D(uiLayout *layout, struct bContext *C)
358 {
359         bScreen *screen = CTX_wm_screen(C);
360         ScrArea *sa = CTX_wm_area(C);
361         View3D *v3d = sa->spacedata.first;
362         Scene *scene = CTX_data_scene(C);
363         ViewLayer *view_layer = CTX_data_view_layer(C);
364         ToolSettings *ts = CTX_data_tool_settings(C);
365         PointerRNA v3dptr, toolsptr, sceneptr;
366         Object *ob = OBACT(view_layer);
367         Object *obedit = CTX_data_edit_object(C);
368         bGPdata *gpd = CTX_data_gpencil_data(C);
369         uiBlock *block;
370         bool is_paint = (
371                 ob && !(gpd && (gpd->flag & GP_DATA_STROKE_EDITMODE)) &&
372                 ELEM(ob->mode,
373                      OB_MODE_SCULPT, OB_MODE_VERTEX_PAINT, OB_MODE_WEIGHT_PAINT, OB_MODE_TEXTURE_PAINT));
374
375         RNA_pointer_create(&screen->id, &RNA_SpaceView3D, v3d, &v3dptr);
376         RNA_pointer_create(&scene->id, &RNA_ToolSettings, ts, &toolsptr);
377         RNA_pointer_create(&scene->id, &RNA_Scene, scene, &sceneptr);
378
379         block = uiLayoutGetBlock(layout);
380         UI_block_func_handle_set(block, do_view3d_header_buttons, NULL);
381
382         /* other buttons: */
383         UI_block_emboss_set(block, UI_EMBOSS);
384
385         /* moved to topbar */
386 #if 0
387         uiLayout *row = uiLayoutRow(layout, true);
388         uiItemR(row, &v3dptr, "pivot_point", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
389         if (!ob || ELEM(ob->mode, OB_MODE_OBJECT, OB_MODE_POSE, OB_MODE_WEIGHT_PAINT)) {
390                 uiItemR(row, &v3dptr, "use_pivot_point_align", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
391         }
392 #endif
393
394         if (obedit == NULL && is_paint) {
395                 /* Currently Python calls this directly. */
396 #if 0
397                 uiTemplatePaintModeSelection(layout, C);
398 #endif
399
400         }
401         else {
402                 /* Moved to popover and topbar. */
403 #if 0
404                 /* Transform widget / manipulators */
405                 row = uiLayoutRow(layout, true);
406                 uiItemR(row, &v3dptr, "show_manipulator", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
407                 uiItemR(row, &sceneptr, "transform_orientation", 0, "", ICON_NONE);
408 #endif
409         }
410
411         if (obedit == NULL && v3d->localvd == NULL) {
412                 /* Scene lock */
413                 uiItemR(layout, &v3dptr, "lock_camera_and_layers", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
414         }
415
416         /* Currently Python calls this directly. */
417 #if 0
418         uiTemplateEditModeSelection(layout, C);
419 #endif
420 }