4 * ***** BEGIN GPL LICENSE BLOCK *****
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version 2
9 * of the License, or (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software Foundation,
18 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
20 * The Original Code is Copyright (C) 2004-2008 Blender Foundation.
21 * All rights reserved.
24 * Contributor(s): Blender Foundation
26 * ***** END GPL LICENSE BLOCK *****
29 /** \file blender/editors/space_view3d/view3d_header.c
38 #include "DNA_scene_types.h"
39 #include "DNA_object_types.h"
41 #include "RNA_access.h"
43 #include "MEM_guardedalloc.h"
46 #include "BLI_blenlib.h"
47 #include "BLI_editVert.h"
48 #include "BLI_utildefines.h"
50 #include "BKE_context.h"
51 #include "BKE_depsgraph.h"
52 #include "BKE_effect.h"
55 #include "BKE_modifier.h"
56 #include "BKE_paint.h"
57 #include "BKE_screen.h"
61 #include "ED_screen.h"
62 #include "ED_transform.h"
68 #include "RNA_define.h"
69 #include "RNA_enum_types.h"
71 #include "UI_interface.h"
72 #include "UI_resources.h"
74 #include "view3d_intern.h"
78 * This is a bit of a dodgy hack to enable a 'mode' menu with icons+labels
79 * rather than those buttons.
80 * I know the implementation's not good - it's an experiment to see if this
81 * approach would work well
83 * This can be cleaned when I make some new 'mode' icons.
86 #define TEST_EDITMESH if(obedit==0) return; \
87 if( (v3d->lay & obedit->lay)==0 ) return;
89 /* view3d handler codes */
90 #define VIEW3D_HANDLER_BACKGROUND 1
91 #define VIEW3D_HANDLER_PROPERTIES 2
92 #define VIEW3D_HANDLER_OBJECT 3
93 #define VIEW3D_HANDLER_PREVIEW 4
94 #define VIEW3D_HANDLER_MULTIRES 5
95 #define VIEW3D_HANDLER_TRANSFORM 6
96 #define VIEW3D_HANDLER_GREASEPENCIL 7
97 #define VIEW3D_HANDLER_BONESKETCH 8
99 /* end XXX ************* */
101 static void do_view3d_header_buttons(bContext *C, void *arg, int event);
103 #define B_SCENELOCK 101
106 #define B_VIEWBUT 104
108 #define B_MODESELECT 108
109 #define B_SEL_VERT 110
110 #define B_SEL_EDGE 111
111 #define B_SEL_FACE 112
112 #define B_MAN_TRANS 116
113 #define B_MAN_ROT 117
114 #define B_MAN_SCALE 118
116 #define B_MAN_MODE 120
120 // XXX quickly ported across
121 static void handle_view3d_lock(bContext *C)
123 Main *bmain= CTX_data_main(C);
124 Scene *scene= CTX_data_scene(C);
125 ScrArea *sa= CTX_wm_area(C);
126 View3D *v3d= CTX_wm_view3d(C);
128 if (v3d != NULL && sa != NULL) {
129 if(v3d->localvd==NULL && v3d->scenelock && sa->spacetype==SPACE_VIEW3D) {
131 scene->lay= v3d->lay;
132 scene->layact= v3d->layact;
133 scene->camera= v3d->camera;
135 /* not through notifiery, listener don't have context
136 and non-open screens or spaces need to be updated too */
137 BKE_screen_view3d_main_sync(&bmain->screen, scene);
139 /* notifiers for scene update */
140 WM_event_add_notifier(C, NC_SCENE|ND_LAYER, scene);
145 /* layer code is on three levels actually:
147 - uiTemplateLayers in interface/ code for buttons
148 - ED_view3d_scene_layer_set for RNA
150 static void view3d_layers_editmode_ensure(Scene *scene, View3D *v3d)
152 /* sanity check - when in editmode disallow switching the editmode layer off since its confusing
153 * an alternative would be to always draw the editmode object. */
154 if(scene->obedit && (scene->obedit->lay & v3d->lay)==0) {
156 for(bit=0; bit<32; bit++) {
157 if(scene->obedit->lay & (1<<bit)) {
165 static int view3d_layers_exec(bContext *C, wmOperator *op)
167 Scene *scene= CTX_data_scene(C);
168 ScrArea *sa= CTX_wm_area(C);
169 View3D *v3d= sa->spacedata.first;
170 int nr= RNA_int_get(op->ptr, "nr");
171 int toggle= RNA_boolean_get(op->ptr, "toggle");
174 return OPERATOR_CANCELLED;
181 if (toggle && v3d->lay == ((1<<20)-1)) {
182 /* return to active layer only */
183 v3d->lay = v3d->layact;
185 view3d_layers_editmode_ensure(scene, v3d);
188 v3d->lay |= (1<<20)-1;
195 if(RNA_boolean_get(op->ptr, "extend")) {
196 if(toggle && v3d->lay & (1<<nr) && (v3d->lay & ~(1<<nr)))
197 v3d->lay &= ~(1<<nr);
204 view3d_layers_editmode_ensure(scene, v3d);
206 /* set active layer, ensure to always have one */
207 if(v3d->lay & (1<<nr))
209 else if((v3d->lay & v3d->layact)==0) {
210 for(bit=0; bit<32; bit++) {
211 if(v3d->lay & (1<<bit)) {
219 if(v3d->scenelock) handle_view3d_lock(C);
221 DAG_on_visible_update(CTX_data_main(C), FALSE);
223 ED_area_tag_redraw(sa);
225 return OPERATOR_FINISHED;
228 /* applies shift and alt, lazy coding or ok? :) */
229 /* the local per-keymap-entry keymap will solve it */
230 static int view3d_layers_invoke(bContext *C, wmOperator *op, wmEvent *event)
232 if(event->ctrl || event->oskey)
233 return OPERATOR_PASS_THROUGH;
236 RNA_boolean_set(op->ptr, "extend", 1);
239 int nr= RNA_int_get(op->ptr, "nr") + 10;
240 RNA_int_set(op->ptr, "nr", nr);
242 view3d_layers_exec(C, op);
244 return OPERATOR_FINISHED;
247 static int view3d_layers_poll(bContext *C)
249 return (ED_operator_view3d_active(C) && CTX_wm_view3d(C)->localvd==NULL);
252 void VIEW3D_OT_layers(wmOperatorType *ot)
256 ot->description= "Toggle layer(s) visibility";
257 ot->idname= "VIEW3D_OT_layers";
260 ot->invoke= view3d_layers_invoke;
261 ot->exec= view3d_layers_exec;
262 ot->poll= view3d_layers_poll;
265 ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
267 RNA_def_int(ot->srna, "nr", 1, 0, 20, "Number", "The layer number to set, zero for all layers", 0, 20);
268 RNA_def_boolean(ot->srna, "extend", 0, "Extend", "Add this layer to the current view layers");
269 RNA_def_boolean(ot->srna, "toggle", 1, "Toggle", "Toggle the layer");
272 static char *view3d_modeselect_pup(Scene *scene)
275 static char string[256];
276 static char formatstr[] = "|%s %%x%d %%i%d";
279 str += sprintf(str, "Mode: %%t");
281 str += sprintf(str, formatstr, "Object Mode", OB_MODE_OBJECT, ICON_OBJECT_DATA);
283 if(ob==NULL) return string;
285 /* if active object is editable */
286 if ( ((ob->type == OB_MESH)
287 || (ob->type == OB_CURVE) || (ob->type == OB_SURF) || (ob->type == OB_FONT)
288 || (ob->type == OB_MBALL) || (ob->type == OB_LATTICE))) {
290 str += sprintf(str, formatstr, "Edit Mode", OB_MODE_EDIT, ICON_EDITMODE_HLT);
292 else if (ob->type == OB_ARMATURE) {
293 if (ob->mode & OB_MODE_POSE)
294 str += sprintf(str, formatstr, "Edit Mode", OB_MODE_EDIT|OB_MODE_POSE, ICON_EDITMODE_HLT);
296 str += sprintf(str, formatstr, "Edit Mode", OB_MODE_EDIT, ICON_EDITMODE_HLT);
299 if (ob->type == OB_MESH) {
301 str += sprintf(str, formatstr, "Sculpt Mode", OB_MODE_SCULPT, ICON_SCULPTMODE_HLT);
302 str += sprintf(str, formatstr, "Vertex Paint", OB_MODE_VERTEX_PAINT, ICON_VPAINT_HLT);
303 str += sprintf(str, formatstr, "Texture Paint", OB_MODE_TEXTURE_PAINT, ICON_TPAINT_HLT);
304 str += sprintf(str, formatstr, "Weight Paint", OB_MODE_WEIGHT_PAINT, ICON_WPAINT_HLT);
308 /* if active object is an armature */
309 if (ob->type==OB_ARMATURE) {
310 str += sprintf(str, formatstr, "Pose Mode", OB_MODE_POSE, ICON_POSE_HLT);
313 if (ob->particlesystem.first || modifiers_findByType(ob, eModifierType_Cloth) || modifiers_findByType(ob, eModifierType_Softbody)) {
314 str += sprintf(str, formatstr, "Particle Mode", OB_MODE_PARTICLE_EDIT, ICON_PARTICLEMODE);
321 static void do_view3d_header_buttons(bContext *C, void *UNUSED(arg), int event)
323 wmWindow *win= CTX_wm_window(C);
324 ToolSettings *ts= CTX_data_tool_settings(C);
325 ScrArea *sa= CTX_wm_area(C);
326 View3D *v3d= sa->spacedata.first;
327 Object *obedit = CTX_data_edit_object(C);
329 int ctrl= win->eventstate->ctrl, shift= win->eventstate->shift;
330 PointerRNA props_ptr;
332 if(obedit && obedit->type==OB_MESH) {
333 em= BKE_mesh_get_editmesh((Mesh *)obedit->data);
335 /* watch it: if sa->win does not exist, check that when calling direct drawing routines */
339 ED_area_tag_redraw(sa);
343 WM_operator_properties_create(&props_ptr, "OBJECT_OT_mode_set");
344 RNA_enum_set(&props_ptr, "mode", v3d->modeselect);
345 WM_operator_name_call(C, "OBJECT_OT_mode_set", WM_OP_EXEC_REGION_WIN, &props_ptr);
346 WM_operator_properties_free(&props_ptr);
351 if(shift==0 || em->selectmode==0)
352 em->selectmode= SCE_SELECT_VERTEX;
353 ts->selectmode= em->selectmode;
354 EM_selectmode_set(em);
355 WM_event_add_notifier(C, NC_GEOM|ND_SELECT, obedit->data);
356 ED_undo_push(C, "Selectmode Set: Vertex");
361 if(shift==0 || em->selectmode==0){
362 if( (em->selectmode ^ SCE_SELECT_EDGE) == SCE_SELECT_VERTEX){
363 if(ctrl) EM_convertsel(em, SCE_SELECT_VERTEX,SCE_SELECT_EDGE);
365 em->selectmode = SCE_SELECT_EDGE;
367 ts->selectmode= em->selectmode;
368 EM_selectmode_set(em);
369 WM_event_add_notifier(C, NC_GEOM|ND_SELECT, obedit->data);
370 ED_undo_push(C, "Selectmode Set: Edge");
375 if( shift==0 || em->selectmode==0){
376 if( ((em->selectmode ^ SCE_SELECT_FACE) == SCE_SELECT_VERTEX) || ((em->selectmode ^ SCE_SELECT_FACE) == SCE_SELECT_EDGE)){
378 EM_convertsel(em, (em->selectmode ^ SCE_SELECT_FACE),SCE_SELECT_FACE);
380 em->selectmode = SCE_SELECT_FACE;
382 ts->selectmode= em->selectmode;
383 EM_selectmode_set(em);
384 WM_event_add_notifier(C, NC_GEOM|ND_SELECT, obedit->data);
385 ED_undo_push(C, "Selectmode Set: Face");
390 if( shift==0 || v3d->twtype==0) {
391 v3d->twtype= V3D_MANIP_TRANSLATE;
393 ED_area_tag_redraw(sa);
396 if( shift==0 || v3d->twtype==0) {
397 v3d->twtype= V3D_MANIP_ROTATE;
399 ED_area_tag_redraw(sa);
402 if( shift==0 || v3d->twtype==0) {
403 v3d->twtype= V3D_MANIP_SCALE;
405 ED_area_tag_redraw(sa);
408 ED_area_tag_redraw(sa);
411 ED_area_tag_redraw(sa);
417 if(obedit && obedit->type==OB_MESH)
418 BKE_mesh_end_editmesh(obedit->data, em);
421 /* Returns the icon associated with an object mode */
422 static int object_mode_icon(int mode)
424 EnumPropertyItem *item = object_mode_items;
426 while(item->name != NULL) {
427 if(item->value == mode)
432 return ICON_OBJECT_DATAMODE;
435 void uiTemplateEditModeSelection(uiLayout *layout, struct bContext *C)
437 Object *obedit = CTX_data_edit_object(C);
438 uiBlock *block= uiLayoutGetBlock(layout);
440 uiBlockSetHandleFunc(block, do_view3d_header_buttons, NULL);
442 if(obedit && (obedit->type == OB_MESH)) {
443 EditMesh *em= BKE_mesh_get_editmesh((Mesh *)obedit->data);
446 row= uiLayoutRow(layout, 1);
447 block= uiLayoutGetBlock(row);
448 uiDefIconButBitS(block, TOG, SCE_SELECT_VERTEX, B_SEL_VERT, ICON_VERTEXSEL, 0,0,UI_UNIT_X,UI_UNIT_Y, &em->selectmode, 1.0, 0.0, 0, 0, "Vertex select mode");
449 uiDefIconButBitS(block, TOG, SCE_SELECT_EDGE, B_SEL_EDGE, ICON_EDGESEL, 0,0,UI_UNIT_X,UI_UNIT_Y, &em->selectmode, 1.0, 0.0, 0, 0, "Edge select mode");
450 uiDefIconButBitS(block, TOG, SCE_SELECT_FACE, B_SEL_FACE, ICON_FACESEL, 0,0,UI_UNIT_X,UI_UNIT_Y, &em->selectmode, 1.0, 0.0, 0, 0, "Face select mode");
452 BKE_mesh_end_editmesh(obedit->data, em);
456 void uiTemplateHeader3D(uiLayout *layout, struct bContext *C)
458 bScreen *screen= CTX_wm_screen(C);
459 ScrArea *sa= CTX_wm_area(C);
460 View3D *v3d= sa->spacedata.first;
461 Scene *scene= CTX_data_scene(C);
462 ToolSettings *ts= CTX_data_tool_settings(C);
463 PointerRNA v3dptr, toolsptr, sceneptr;
465 Object *obedit = CTX_data_edit_object(C);
468 const float dpi_fac= UI_DPI_FAC;
470 RNA_pointer_create(&screen->id, &RNA_SpaceView3D, v3d, &v3dptr);
471 RNA_pointer_create(&scene->id, &RNA_ToolSettings, ts, &toolsptr);
472 RNA_pointer_create(&scene->id, &RNA_Scene, scene, &sceneptr);
474 block= uiLayoutGetBlock(layout);
475 uiBlockSetHandleFunc(block, do_view3d_header_buttons, NULL);
478 uiBlockSetEmboss(block, UI_EMBOSS);
482 v3d->modeselect = ob->mode;
484 v3d->modeselect = OB_MODE_OBJECT;
486 uiBlockBeginAlign(block);
487 uiDefIconTextButS(block, MENU, B_MODESELECT, object_mode_icon(v3d->modeselect), view3d_modeselect_pup(scene) ,
488 0,0,126 * dpi_fac, UI_UNIT_Y, &(v3d->modeselect), 0, 0, 0, 0, "Mode");
489 uiBlockEndAlign(block);
492 uiItemR(layout, &v3dptr, "viewport_shade", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
494 if (obedit==NULL && ((ob && ob->mode & (OB_MODE_VERTEX_PAINT|OB_MODE_WEIGHT_PAINT|OB_MODE_TEXTURE_PAINT)))) {
495 /* Manipulators aren't used in weight paint mode */
498 RNA_pointer_create(&ob->id, &RNA_Mesh, ob->data, &meshptr);
499 if(ob->mode & (OB_MODE_TEXTURE_PAINT|OB_MODE_VERTEX_PAINT)) {
500 uiItemR(layout, &meshptr, "use_paint_mask", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
503 row= uiLayoutRow(layout, 1);
504 uiItemR(row, &meshptr, "use_paint_mask", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
505 uiItemR(row, &meshptr, "wp_vert_sel", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
508 const char *str_menu;
510 row= uiLayoutRow(layout, 1);
511 uiItemR(row, &v3dptr, "pivot_point", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
512 uiItemR(row, &v3dptr, "use_pivot_point_align", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
514 /* Transform widget / manipulators */
515 row= uiLayoutRow(layout, 1);
516 uiItemR(row, &v3dptr, "show_manipulator", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
517 block= uiLayoutGetBlock(row);
519 if(v3d->twflag & V3D_USE_MANIPULATOR) {
520 uiDefIconButBitC(block, TOG, V3D_MANIP_TRANSLATE, B_MAN_TRANS, ICON_MAN_TRANS, 0,0,UI_UNIT_X,UI_UNIT_Y, &v3d->twtype, 1.0, 0.0, 0, 0, "Translate manipulator mode");
521 uiDefIconButBitC(block, TOG, V3D_MANIP_ROTATE, B_MAN_ROT, ICON_MAN_ROT, 0,0,UI_UNIT_X,UI_UNIT_Y, &v3d->twtype, 1.0, 0.0, 0, 0, "Rotate manipulator mode");
522 uiDefIconButBitC(block, TOG, V3D_MANIP_SCALE, B_MAN_SCALE, ICON_MAN_SCALE, 0,0,UI_UNIT_X,UI_UNIT_Y, &v3d->twtype, 1.0, 0.0, 0, 0, "Scale manipulator mode");
525 if (v3d->twmode > (BIF_countTransformOrientation(C) - 1) + V3D_MANIP_CUSTOM) {
529 str_menu = BIF_menustringTransformOrientation(C, "Orientation");
530 uiDefButC(block, MENU, B_MAN_MODE, str_menu,0,0,70 * dpi_fac, UI_UNIT_Y, &v3d->twmode, 0, 0, 0, 0, "Transform Orientation");
531 MEM_freeN((void *)str_menu);
534 if(obedit==NULL && v3d->localvd==NULL) {
535 unsigned int ob_lay = ob ? ob->lay : 0;
539 uiTemplateLayers(layout, &sceneptr, "layers", &v3dptr, "layers_used", ob_lay);
541 uiTemplateLayers(layout, &v3dptr, "layers", &v3dptr, "layers_used", ob_lay);
544 uiItemR(layout, &v3dptr, "lock_camera_and_layers", UI_ITEM_R_ICON_ONLY, "", ICON_NONE);
547 uiTemplateEditModeSelection(layout, C);