e078fa8eda15ac809a251f863db22c0f0e982260
[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
32 #include <string.h>
33 #include <stdio.h>
34 #include <stdlib.h>
35
36 #include "DNA_scene_types.h"
37 #include "DNA_object_types.h"
38 #include "DNA_mesh_types.h"
39
40 #include "RNA_access.h"
41
42 #include "MEM_guardedalloc.h"
43
44 #include "BLI_math.h"
45 #include "BLI_blenlib.h"
46 #include "BLI_utildefines.h"
47
48 #include "BLF_translation.h"
49
50 #include "BKE_context.h"
51 #include "BKE_depsgraph.h"
52 #include "BKE_effect.h"
53 #include "BKE_main.h"
54 #include "BKE_mesh.h"
55 #include "BKE_modifier.h"
56 #include "BKE_paint.h"
57 #include "BKE_screen.h"
58 #include "BKE_tessmesh.h"
59
60 #include "ED_mesh.h"
61 #include "ED_util.h"
62 #include "ED_screen.h"
63 #include "ED_transform.h"
64 #include "ED_types.h"
65
66 #include "WM_api.h"
67 #include "WM_types.h"
68
69 #include "RNA_define.h"
70 #include "RNA_enum_types.h"
71
72 #include "UI_interface.h"
73 #include "UI_resources.h"
74
75 #include "view3d_intern.h"
76
77
78 /* View3d->modeselect 
79  * This is a bit of a dodgy hack to enable a 'mode' menu with icons+labels
80  * rather than those buttons.
81  * I know the implementation's not good - it's an experiment to see if this
82  * approach would work well
83  *
84  * This can be cleaned when I make some new 'mode' icons.
85  */
86
87 /* end XXX ************* */
88
89 static void do_view3d_header_buttons(bContext *C, void *arg, int event);
90
91 #define B_MODESELECT 108
92 #define B_SEL_VERT  110
93 #define B_SEL_EDGE  111
94 #define B_SEL_FACE  112
95 #define B_MAN_TRANS 116
96 #define B_MAN_ROT   117
97 #define B_MAN_SCALE 118
98 #define B_MAN_MODE  120
99
100 /* XXX quickly ported across */
101 static void handle_view3d_lock(bContext *C)
102 {
103         Main *bmain = CTX_data_main(C);
104         Scene *scene = CTX_data_scene(C);
105         ScrArea *sa = CTX_wm_area(C);
106         View3D *v3d = CTX_wm_view3d(C);
107         
108         if (v3d != NULL && sa != NULL) {
109                 if (v3d->localvd == NULL && v3d->scenelock && sa->spacetype == SPACE_VIEW3D) {
110                         /* copy to scene */
111                         scene->lay = v3d->lay;
112                         scene->layact = v3d->layact;
113                         scene->camera = v3d->camera;
114
115                         /* not through notifier, listener don't have context
116                          * and non-open screens or spaces need to be updated too */
117                         BKE_screen_view3d_main_sync(&bmain->screen, scene);
118                         
119                         /* notifiers for scene update */
120                         WM_event_add_notifier(C, NC_SCENE | ND_LAYER, scene);
121                 }
122         }
123 }
124
125 /**
126  * layer code is on three levels actually:
127  * - here for operator
128  * - uiTemplateLayers in interface/ code for buttons
129  * - ED_view3d_scene_layer_set for RNA
130  */
131 static void view3d_layers_editmode_ensure(Scene *scene, View3D *v3d)
132 {
133         /* sanity check - when in editmode disallow switching the editmode layer off since its confusing
134          * an alternative would be to always draw the editmode object. */
135         if (scene->obedit && (scene->obedit->lay & v3d->lay) == 0) {
136                 int bit;
137                 for (bit = 0; bit < 32; bit++) {
138                         if (scene->obedit->lay & (1 << bit)) {
139                                 v3d->lay |= 1 << bit;
140                                 break;
141                         }
142                 }
143         }
144 }
145
146 static int view3d_layers_exec(bContext *C, wmOperator *op)
147 {
148         Scene *scene = CTX_data_scene(C);
149         ScrArea *sa = CTX_wm_area(C);
150         View3D *v3d = sa->spacedata.first;
151         int nr = RNA_int_get(op->ptr, "nr");
152         int toggle = RNA_boolean_get(op->ptr, "toggle");
153         
154         if (nr < 0)
155                 return OPERATOR_CANCELLED;
156
157         if (nr == 0) {
158                 /* all layers */
159                 if (!v3d->layact)
160                         v3d->layact = 1;
161
162                 if (toggle && v3d->lay == ((1 << 20) - 1)) {
163                         /* return to active layer only */
164                         v3d->lay = v3d->layact;
165
166                         view3d_layers_editmode_ensure(scene, v3d);
167                 }
168                 else {
169                         v3d->lay |= (1 << 20) - 1;
170                 }
171         }
172         else {
173                 int bit;
174                 nr--;
175
176                 if (RNA_boolean_get(op->ptr, "extend")) {
177                         if (toggle && v3d->lay & (1 << nr) && (v3d->lay & ~(1 << nr)))
178                                 v3d->lay &= ~(1 << nr);
179                         else
180                                 v3d->lay |= (1 << nr);
181                 }
182                 else {
183                         v3d->lay = (1 << nr);
184                 }
185
186                 view3d_layers_editmode_ensure(scene, v3d);
187
188                 /* set active layer, ensure to always have one */
189                 if (v3d->lay & (1 << nr))
190                         v3d->layact = 1 << nr;
191                 else if ((v3d->lay & v3d->layact) == 0) {
192                         for (bit = 0; bit < 32; bit++) {
193                                 if (v3d->lay & (1 << bit)) {
194                                         v3d->layact = 1 << bit;
195                                         break;
196                                 }
197                         }
198                 }
199         }
200         
201         if (v3d->scenelock) handle_view3d_lock(C);
202         
203         DAG_on_visible_update(CTX_data_main(C), FALSE);
204
205         ED_area_tag_redraw(sa);
206         
207         return OPERATOR_FINISHED;
208 }
209
210 /* applies shift and alt, lazy coding or ok? :) */
211 /* the local per-keymap-entry keymap will solve it */
212 static int view3d_layers_invoke(bContext *C, wmOperator *op, wmEvent *event)
213 {
214         if (event->ctrl || event->oskey)
215                 return OPERATOR_PASS_THROUGH;
216         
217         if (event->shift)
218                 RNA_boolean_set(op->ptr, "extend", TRUE);
219         else
220                 RNA_boolean_set(op->ptr, "extend", FALSE);
221         
222         if (event->alt) {
223                 int nr = RNA_int_get(op->ptr, "nr") + 10;
224                 RNA_int_set(op->ptr, "nr", nr);
225         }
226         view3d_layers_exec(C, op);
227         
228         return OPERATOR_FINISHED;
229 }
230
231 static int view3d_layers_poll(bContext *C)
232 {
233         return (ED_operator_view3d_active(C) && CTX_wm_view3d(C)->localvd == NULL);
234 }
235
236 void VIEW3D_OT_layers(wmOperatorType *ot)
237 {
238         /* identifiers */
239         ot->name = "Layers";
240         ot->description = "Toggle layer(s) visibility";
241         ot->idname = "VIEW3D_OT_layers";
242         
243         /* api callbacks */
244         ot->invoke = view3d_layers_invoke;
245         ot->exec = view3d_layers_exec;
246         ot->poll = view3d_layers_poll;
247         
248         /* flags */
249         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
250         
251         RNA_def_int(ot->srna, "nr", 1, 0, 20, "Number", "The layer number to set, zero for all layers", 0, 20);
252         RNA_def_boolean(ot->srna, "extend", 0, "Extend", "Add this layer to the current view layers");
253         RNA_def_boolean(ot->srna, "toggle", 1, "Toggle", "Toggle the layer");
254 }
255
256 static int modeselect_addmode(char *str, const char *title, int id, int icon)
257 {
258         static char formatstr[] = "|%s %%x%d %%i%d";
259
260         return sprintf(str, formatstr, IFACE_(title), id, icon);
261 }
262
263 static char *view3d_modeselect_pup(Scene *scene)
264 {
265         Object *ob = OBACT;
266         static char string[512];
267         const char *title = IFACE_("Mode: %t");
268         char *str = string;
269
270         BLI_strncpy(str, title, sizeof(string));
271
272         str += modeselect_addmode(str, N_("Object Mode"), OB_MODE_OBJECT, ICON_OBJECT_DATA);
273
274         if (ob == NULL || ob->data == NULL) return string;
275         if (ob->id.lib) return string;
276
277         if (!((ID *)ob->data)->lib) {
278                 /* if active object is editable */
279                 if (ob->type == OB_ARMATURE) {
280                         if (ob->mode & OB_MODE_POSE)
281                                 str += modeselect_addmode(str, N_("Edit Mode"), OB_MODE_EDIT | OB_MODE_POSE, ICON_EDITMODE_HLT);
282                         else
283                                 str += modeselect_addmode(str, N_("Edit Mode"), OB_MODE_EDIT, ICON_EDITMODE_HLT);
284                 }
285                 else if (OB_TYPE_SUPPORT_EDITMODE(ob->type)) {
286                         str += modeselect_addmode(str, N_("Edit Mode"), OB_MODE_EDIT, ICON_EDITMODE_HLT);
287                 }
288
289                 if (ob->type == OB_MESH) {
290
291                         str += modeselect_addmode(str, N_("Sculpt Mode"), OB_MODE_SCULPT, ICON_SCULPTMODE_HLT);
292                         str += modeselect_addmode(str, N_("Vertex Paint"), OB_MODE_VERTEX_PAINT, ICON_VPAINT_HLT);
293                         str += modeselect_addmode(str, N_("Texture Paint"), OB_MODE_TEXTURE_PAINT, ICON_TPAINT_HLT);
294                         str += modeselect_addmode(str, N_("Weight Paint"), OB_MODE_WEIGHT_PAINT, ICON_WPAINT_HLT);
295                 }
296         }
297
298         /* if active object is an armature */
299         if (ob->type == OB_ARMATURE) {
300                 str += modeselect_addmode(str, N_("Pose Mode"), OB_MODE_POSE, ICON_POSE_HLT);
301         }
302
303         if (ob->particlesystem.first ||
304             modifiers_findByType(ob, eModifierType_Cloth) ||
305             modifiers_findByType(ob, eModifierType_Softbody))
306         {
307                 str += modeselect_addmode(str, N_("Particle Mode"), OB_MODE_PARTICLE_EDIT, ICON_PARTICLEMODE);
308         }
309         (void)str;
310         return (string);
311 }
312
313 static void do_view3d_header_buttons(bContext *C, void *UNUSED(arg), int event)
314 {
315         wmWindow *win = CTX_wm_window(C);
316         ScrArea *sa = CTX_wm_area(C);
317         View3D *v3d = sa->spacedata.first;
318         const int ctrl = win->eventstate->ctrl, shift = win->eventstate->shift;
319         PointerRNA props_ptr;
320
321         /* watch it: if sa->win does not exist, check that when calling direct drawing routines */
322
323         switch (event) {
324                 case B_MODESELECT:
325                         WM_operator_properties_create(&props_ptr, "OBJECT_OT_mode_set");
326                         RNA_enum_set(&props_ptr, "mode", v3d->modeselect);
327                         WM_operator_name_call(C, "OBJECT_OT_mode_set", WM_OP_EXEC_REGION_WIN, &props_ptr);
328                         WM_operator_properties_free(&props_ptr);
329                         break;
330
331                 case B_SEL_VERT:
332                         if (EDBM_selectmode_toggle(C, SCE_SELECT_VERTEX, -1, shift, ctrl)) {
333                                 ED_undo_push(C, "Selectmode Set: Vertex");
334                         }
335                         break;
336                 case B_SEL_EDGE:
337                         if (EDBM_selectmode_toggle(C, SCE_SELECT_EDGE, -1, shift, ctrl)) {
338                                 ED_undo_push(C, "Selectmode Set: Edge");
339                         }
340                         break;
341                 case B_SEL_FACE:
342                         if (EDBM_selectmode_toggle(C, SCE_SELECT_FACE, -1, shift, ctrl)) {
343                                 ED_undo_push(C, "Selectmode Set: Face");
344                         }
345                         break;
346
347                 case B_MAN_TRANS:
348                         if (shift == 0 || v3d->twtype == 0) {
349                                 v3d->twtype = V3D_MANIP_TRANSLATE;
350                         }
351                         ED_area_tag_redraw(sa);
352                         break;
353                 case B_MAN_ROT:
354                         if (shift == 0 || v3d->twtype == 0) {
355                                 v3d->twtype = V3D_MANIP_ROTATE;
356                         }
357                         ED_area_tag_redraw(sa);
358                         break;
359                 case B_MAN_SCALE:
360                         if (shift == 0 || v3d->twtype == 0) {
361                                 v3d->twtype = V3D_MANIP_SCALE;
362                         }
363                         ED_area_tag_redraw(sa);
364                         break;
365                 case B_MAN_MODE:
366                         ED_area_tag_redraw(sa);
367                         break;
368                 default:
369                         break;
370         }
371 }
372
373 /* Returns the icon associated with an object mode */
374 static int object_mode_icon(int mode)
375 {
376         EnumPropertyItem *item = object_mode_items;
377         
378         while (item->name != NULL) {
379                 if (item->value == mode) {
380                         return item->icon;
381                 }
382                 item++;
383         }
384
385         return ICON_OBJECT_DATAMODE;
386 }
387
388 void uiTemplateEditModeSelection(uiLayout *layout, struct bContext *C)
389 {
390         Object *obedit = CTX_data_edit_object(C);
391         uiBlock *block = uiLayoutGetBlock(layout);
392
393         uiBlockSetHandleFunc(block, do_view3d_header_buttons, NULL);
394
395         if (obedit && (obedit->type == OB_MESH)) {
396                 BMEditMesh *em = BMEdit_FromObject(obedit);
397                 uiLayout *row;
398
399                 row = uiLayoutRow(layout, TRUE);
400                 block = uiLayoutGetBlock(row);
401                 uiDefIconButBitS(block, TOG, SCE_SELECT_VERTEX, B_SEL_VERT, ICON_VERTEXSEL,
402                                  0, 0, UI_UNIT_X, UI_UNIT_Y, &em->selectmode, 1.0, 0.0, 0, 0,
403                                  TIP_("Vertex select - Shift-Click for multiple modes"));
404                 uiDefIconButBitS(block, TOG, SCE_SELECT_EDGE, B_SEL_EDGE, ICON_EDGESEL,
405                                  0, 0, UI_UNIT_X, UI_UNIT_Y, &em->selectmode, 1.0, 0.0, 0, 0,
406                                  TIP_("Edge select - Shift-Click for multiple modes, Ctrl-Click expands selection"));
407                 uiDefIconButBitS(block, TOG, SCE_SELECT_FACE, B_SEL_FACE, ICON_FACESEL,
408                                  0, 0, UI_UNIT_X, UI_UNIT_Y, &em->selectmode, 1.0, 0.0, 0, 0,
409                                  TIP_("Face select - Shift-Click for multiple modes, Ctrl-Click expands selection"));
410         }
411 }
412
413 void uiTemplateHeader3D(uiLayout *layout, struct bContext *C)
414 {
415         bScreen *screen = CTX_wm_screen(C);
416         ScrArea *sa = CTX_wm_area(C);
417         View3D *v3d = sa->spacedata.first;
418         Scene *scene = CTX_data_scene(C);
419         ToolSettings *ts = CTX_data_tool_settings(C);
420         PointerRNA v3dptr, toolsptr, sceneptr;
421         Object *ob = OBACT;
422         Object *obedit = CTX_data_edit_object(C);
423         uiBlock *block;
424         uiBut *but;
425         uiLayout *row;
426         const float dpi_fac = UI_DPI_FAC;
427         int is_paint = 0;
428         
429         RNA_pointer_create(&screen->id, &RNA_SpaceView3D, v3d, &v3dptr);
430         RNA_pointer_create(&scene->id, &RNA_ToolSettings, ts, &toolsptr);
431         RNA_pointer_create(&scene->id, &RNA_Scene, scene, &sceneptr);
432
433         block = uiLayoutGetBlock(layout);
434         uiBlockSetHandleFunc(block, do_view3d_header_buttons, NULL);
435
436         /* other buttons: */
437         uiBlockSetEmboss(block, UI_EMBOSS);
438         
439         /* mode */
440         if (ob) {
441                 v3d->modeselect = ob->mode;
442                 is_paint = ELEM4(ob->mode, OB_MODE_SCULPT, OB_MODE_VERTEX_PAINT, OB_MODE_WEIGHT_PAINT, OB_MODE_TEXTURE_PAINT);
443         }
444         else {
445                 v3d->modeselect = OB_MODE_OBJECT;
446         }
447
448         row = uiLayoutRow(layout, TRUE);
449         uiDefIconTextButS(block, MENU, B_MODESELECT, object_mode_icon(v3d->modeselect), view3d_modeselect_pup(scene),
450                           0, 0, 126 * dpi_fac, UI_UNIT_Y, &(v3d->modeselect), 0, 0, 0, 0, TIP_("Mode"));
451         
452         /* Draw type */
453         uiItemR(layout, &v3dptr, "viewport_shade", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
454
455         if (obedit == NULL && is_paint) {
456
457                 if (ob->mode & OB_MODE_WEIGHT_PAINT) {
458                         /* Only for Weight Paint. makes no sense in other paint modes. */
459                         row = uiLayoutRow(layout, TRUE);
460                         uiItemR(row, &v3dptr, "pivot_point", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
461                 }
462
463                 /* Manipulators aren't used in paint modes */
464                 if (!ELEM(ob->mode, OB_MODE_SCULPT, OB_MODE_PARTICLE_EDIT)) {
465                         /* masks aren't used for sculpt and particle painting */
466                         PointerRNA meshptr;
467
468                         RNA_pointer_create(&ob->id, &RNA_Mesh, ob->data, &meshptr);
469                         if (ob->mode & (OB_MODE_TEXTURE_PAINT | OB_MODE_VERTEX_PAINT)) {
470                                 uiItemR(layout, &meshptr, "use_paint_mask", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
471                         }
472                         else {
473                                 row = uiLayoutRow(layout, TRUE);
474                                 uiItemR(row, &meshptr, "use_paint_mask", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
475                                 uiItemR(row, &meshptr, "use_paint_mask_vertex", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
476                         }
477                 }
478         }
479         else {
480                 const char *str_menu;
481
482                 row = uiLayoutRow(layout, TRUE);
483                 uiItemR(row, &v3dptr, "pivot_point", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
484
485                 /* pose/object only however we want to allow in weight paint mode too
486                  * so don't be totally strict and just check not-editmode for now 
487                  * XXX We never get here when we are in Weight Paint mode
488                  */
489                 if (obedit == NULL) {
490                         uiItemR(row, &v3dptr, "use_pivot_point_align", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
491                 }
492
493                 /* Transform widget / manipulators */
494                 row = uiLayoutRow(layout, TRUE);
495                 uiItemR(row, &v3dptr, "show_manipulator", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
496                 block = uiLayoutGetBlock(row);
497                 
498                 if (v3d->twflag & V3D_USE_MANIPULATOR) {
499                         but = uiDefIconButBitC(block, TOG, V3D_MANIP_TRANSLATE, B_MAN_TRANS, ICON_MAN_TRANS,
500                                                0, 0, UI_UNIT_X, UI_UNIT_Y, &v3d->twtype, 1.0, 0.0, 0, 0,
501                                                TIP_("Translate manipulator - Shift-Click for multiple modes"));
502                         uiButClearFlag(but, UI_BUT_UNDO); /* skip undo on screen buttons */
503                         but = uiDefIconButBitC(block, TOG, V3D_MANIP_ROTATE, B_MAN_ROT, ICON_MAN_ROT,
504                                                0, 0, UI_UNIT_X, UI_UNIT_Y, &v3d->twtype, 1.0, 0.0, 0, 0,
505                                                TIP_("Rotate manipulator - Shift-Click for multiple modes"));
506                         uiButClearFlag(but, UI_BUT_UNDO); /* skip undo on screen buttons */
507                         but = uiDefIconButBitC(block, TOG, V3D_MANIP_SCALE, B_MAN_SCALE, ICON_MAN_SCALE,
508                                                0, 0, UI_UNIT_X, UI_UNIT_Y, &v3d->twtype, 1.0, 0.0, 0, 0,
509                                                TIP_("Scale manipulator - Shift-Click for multiple modes"));
510                         uiButClearFlag(but, UI_BUT_UNDO); /* skip undo on screen buttons */
511                 }
512                         
513                 if (v3d->twmode > (BIF_countTransformOrientation(C) - 1) + V3D_MANIP_CUSTOM) {
514                         v3d->twmode = 0;
515                 }
516                         
517                 str_menu = BIF_menustringTransformOrientation(C, "Orientation");
518                 but = uiDefButC(block, MENU, B_MAN_MODE, str_menu, 0, 0, 70 * dpi_fac, UI_UNIT_Y, &v3d->twmode, 0, 0, 0, 0,
519                                 TIP_("Transform Orientation"));
520                 uiButClearFlag(but, UI_BUT_UNDO); /* skip undo on screen buttons */
521                 MEM_freeN((void *)str_menu);
522         }
523
524         if (obedit == NULL && v3d->localvd == NULL) {
525                 unsigned int ob_lay = ob ? ob->lay : 0;
526
527                 /* Layers */
528                 uiTemplateLayers(layout, v3d->scenelock ? &sceneptr : &v3dptr, "layers", &v3dptr, "layers_used", ob_lay);
529
530                 /* Scene lock */
531                 uiItemR(layout, &v3dptr, "lock_camera_and_layers", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
532         }
533         
534         uiTemplateEditModeSelection(layout, C);
535 }