source/blender/editors/armature/armature_ops.c

   1 /**
   2  * $Id:
   3  *
   4  * ***** BEGIN GPL LICENSE BLOCK *****
   5  *
   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.
  10  *
  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.
  15  *
  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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  19  *
  20  * The Original Code is Copyright (C) 2009 Blender Foundation.
  21  * All rights reserved.
  22  *
  23  *
  24  * Contributor(s): Blender Foundation
  25  *
  26  * ***** END GPL LICENSE BLOCK *****
  27  */
  28
  29 #include <stdlib.h>
  30 #include <math.h>
  31
  32 #include "MEM_guardedalloc.h"
  33
  34 #include "DNA_object_types.h"
  35 #include "DNA_scene_types.h"
  36 #include "DNA_screen_types.h"
  37 #include "DNA_space_types.h"
  38 #include "DNA_userdef_types.h"
  39 #include "DNA_view3d_types.h"
  40 #include "DNA_windowmanager_types.h"
  41
  42 #include "BLI_arithb.h"
  43 #include "BLI_blenlib.h"
  44
  45 #include "BKE_context.h"
  46 #include "BKE_global.h"
  47 #include "BKE_utildefines.h"
  48
  49 #include "RNA_access.h"
  50 #include "RNA_define.h"
  51
  52 #include "WM_api.h"
  53 #include "WM_types.h"
  54
  55 #include "ED_armature.h"
  56 #include "ED_screen.h"
  57 #include "ED_object.h"
  58
  59 #include "armature_intern.h"
  60
  61 /* ************************** quick tests **********************************/
  62
  63 /*  XXX This is a quick test operator to print names of all EditBones in context
  64  *              that should be removed once tool coding starts...
  65  */
  66
  67 static int armature_test_exec (bContext *C, wmOperator *op)
  68 {
  69         printf("EditMode Armature Test: \n");
  70
  71         printf("\tSelected Bones \n");
  72         CTX_DATA_BEGIN(C, EditBone*, ebone, selected_bones)
  73         {
  74                 printf("\t\tEditBone '%s' \n", ebone->name);
  75         }
  76         CTX_DATA_END;
  77
  78         printf("\tEditable Bones \n");
  79         CTX_DATA_BEGIN(C, EditBone*, ebone, selected_editable_bones)
  80         {
  81                 printf("\t\tEditBone '%s' \n", ebone->name);
  82         }
  83         CTX_DATA_END;
  84
  85         printf("\tActive Bone \n");
  86         {
  87                 EditBone *ebone= CTX_data_active_bone(C);
  88                 if (ebone) printf("\t\tEditBone '%s' \n", ebone->name);
  89                 else printf("\t\t<None> \n");
  90         }
  91
  92         return OPERATOR_FINISHED;
  93 }
  94
  95 void ARMATURE_OT_test(wmOperatorType *ot)
  96 {
  97         /* identifiers */
  98         ot->name= "Test Context";
  99         ot->idname= "ARMATURE_OT_test";
 100
 101         /* api callbacks */
 102         ot->exec= armature_test_exec;
 103 }
 104
 105 /* ************************** registration **********************************/
 106
 107 /* Both operators ARMATURE_OT_xxx and POSE_OT_xxx here */
 108 void ED_operatortypes_armature(void)
 109 {
 110         /* EDIT ARMATURE */
 111         WM_operatortype_append(ARMATURE_OT_bone_primitive_add);
 112
 113         WM_operatortype_append(ARMATURE_OT_bones_align);
 114         WM_operatortype_append(ARMATURE_OT_calculate_roll);
 115         WM_operatortype_append(ARMATURE_OT_switch_direction);
 116         WM_operatortype_append(ARMATURE_OT_subdivs);
 117         WM_operatortype_append(ARMATURE_OT_subdivide_simple);
 118         WM_operatortype_append(ARMATURE_OT_subdivide_multi);
 119
 120         WM_operatortype_append(ARMATURE_OT_parent_set);
 121         WM_operatortype_append(ARMATURE_OT_parent_clear);
 122
 123         WM_operatortype_append(ARMATURE_OT_select_all_toggle);
 124         WM_operatortype_append(ARMATURE_OT_select_inverse);
 125         WM_operatortype_append(ARMATURE_OT_select_hierarchy);
 126         WM_operatortype_append(ARMATURE_OT_select_linked);
 127
 128         WM_operatortype_append(ARMATURE_OT_delete);
 129         WM_operatortype_append(ARMATURE_OT_duplicate_selected);
 130         WM_operatortype_append(ARMATURE_OT_extrude);
 131         WM_operatortype_append(ARMATURE_OT_click_extrude);
 132
 133         /* SKETCH */
 134         WM_operatortype_append(SKETCH_OT_gesture);
 135         WM_operatortype_append(SKETCH_OT_delete);
 136         WM_operatortype_append(SKETCH_OT_draw_stroke);
 137         WM_operatortype_append(SKETCH_OT_draw_preview);
 138         WM_operatortype_append(SKETCH_OT_finish_stroke);
 139         WM_operatortype_append(SKETCH_OT_cancel_stroke);
 140         WM_operatortype_append(SKETCH_OT_select);
 141
 142         /* POSE */
 143         WM_operatortype_append(POSE_OT_hide);
 144         WM_operatortype_append(POSE_OT_reveal);
 145
 146         WM_operatortype_append(POSE_OT_rot_clear);
 147         WM_operatortype_append(POSE_OT_loc_clear);
 148         WM_operatortype_append(POSE_OT_scale_clear);
 149
 150         WM_operatortype_append(POSE_OT_select_all_toggle);
 151         WM_operatortype_append(POSE_OT_select_inverse);
 152
 153         WM_operatortype_append(POSE_OT_select_parent);
 154         WM_operatortype_append(POSE_OT_select_hierarchy);
 155         WM_operatortype_append(POSE_OT_select_linked);
 156         WM_operatortype_append(POSE_OT_select_constraint_target);
 157
 158         /* POSELIB */
 159         WM_operatortype_append(POSELIB_OT_browse_interactive);
 160
 161         WM_operatortype_append(POSELIB_OT_pose_add);
 162         WM_operatortype_append(POSELIB_OT_pose_remove);
 163         WM_operatortype_append(POSELIB_OT_pose_rename);
 164
 165         /* TESTS */
 166         WM_operatortype_append(ARMATURE_OT_test); // XXX temp test for context iterators... to be removed
 167 }
 168
 169 void ED_keymap_armature(wmWindowManager *wm)
 170 {
 171         ListBase *keymap;
 172         wmKeymapItem *kmi;
 173
 174         /* Armature ------------------------ */
 175         keymap= WM_keymap_listbase(wm, "Armature", 0, 0);
 176
 177         /* only set in editmode armature, by space_view3d listener */
 178 //      WM_keymap_add_item(keymap, "ARMATURE_OT_hide", HKEY, KM_PRESS, 0, 0);
 179         WM_keymap_add_item(keymap, "ARMATURE_OT_bones_align", AKEY, KM_PRESS, KM_CTRL|KM_ALT, 0);
 180         WM_keymap_add_item(keymap, "ARMATURE_OT_calculate_roll", NKEY, KM_PRESS, KM_CTRL, 0);
 181
 182         WM_keymap_add_item(keymap, "ARMATURE_OT_switch_direction", FKEY, KM_PRESS, KM_ALT, 0);
 183
 184         WM_keymap_add_item(keymap, "ARMATURE_OT_bone_primitive_add", AKEY, KM_PRESS, KM_SHIFT, 0);
 185                 /* only the menu-version of subdivide is registered in keymaps for now */
 186         WM_keymap_add_item(keymap, "ARMATURE_OT_subdivs", SKEY, KM_PRESS, KM_ALT, 0);
 187
 188         WM_keymap_add_item(keymap, "ARMATURE_OT_parent_set", PKEY, KM_PRESS, KM_CTRL, 0);
 189         WM_keymap_add_item(keymap, "ARMATURE_OT_parent_clear", PKEY, KM_PRESS, KM_ALT, 0);
 190
 191         WM_keymap_add_item(keymap, "ARMATURE_OT_select_all_toggle", AKEY, KM_PRESS, 0, 0);
 192         WM_keymap_add_item(keymap, "ARMATURE_OT_select_inverse", IKEY, KM_PRESS, KM_CTRL, 0);
 193
 194         WM_keymap_add_item(keymap, "ARMATURE_OT_test", TKEY, KM_PRESS, 0, 0);  // XXX temp test for context iterators... to be removed
 195
 196         kmi= WM_keymap_add_item(keymap, "ARMATURE_OT_select_hierarchy", LEFTBRACKETKEY, KM_PRESS, 0, 0);
 197                 RNA_enum_set(kmi->ptr, "direction", BONE_SELECT_PARENT);
 198         kmi= WM_keymap_add_item(keymap, "ARMATURE_OT_select_hierarchy", LEFTBRACKETKEY, KM_PRESS, KM_SHIFT, 0);
 199                 RNA_enum_set(kmi->ptr, "direction", BONE_SELECT_PARENT);
 200                 RNA_boolean_set(kmi->ptr, "extend", 1);
 201
 202         kmi= WM_keymap_add_item(keymap, "ARMATURE_OT_select_hierarchy", RIGHTBRACKETKEY, KM_PRESS, 0, 0);
 203                 RNA_enum_set(kmi->ptr, "direction", BONE_SELECT_CHILD);
 204         kmi= WM_keymap_add_item(keymap, "ARMATURE_OT_select_hierarchy", RIGHTBRACKETKEY, KM_PRESS, KM_SHIFT, 0);
 205                 RNA_enum_set(kmi->ptr, "direction", BONE_SELECT_CHILD);
 206                 RNA_boolean_set(kmi->ptr, "extend", 1);
 207
 208         WM_keymap_add_item(keymap, "ARMATURE_OT_select_linked", LKEY, KM_PRESS, 0, 0);
 209         WM_keymap_add_item(keymap, "ARMATURE_OT_delete", XKEY, KM_PRESS, 0, 0);
 210         WM_keymap_add_item(keymap, "ARMATURE_OT_duplicate_selected", DKEY, KM_PRESS, KM_SHIFT, 0);
 211         WM_keymap_add_item(keymap, "ARMATURE_OT_extrude", EKEY, KM_PRESS, 0, 0);
 212         kmi= WM_keymap_add_item(keymap, "ARMATURE_OT_extrude", EKEY, KM_PRESS, KM_SHIFT, 0);
 213                 RNA_boolean_set(kmi->ptr, "forked", 1);
 214         WM_keymap_add_item(keymap, "ARMATURE_OT_click_extrude", LEFTMOUSE, KM_PRESS, KM_CTRL, 0);
 215
 216         /* Armature -> Etch-A-Ton ------------------------ */
 217         WM_keymap_add_item(keymap, "SKETCH_OT_delete", XKEY, KM_PRESS, 0, 0);
 218         WM_keymap_add_item(keymap, "SKETCH_OT_finish_stroke", SELECTMOUSE, KM_PRESS, 0, 0);
 219         WM_keymap_add_item(keymap, "SKETCH_OT_cancel_stroke", ESCKEY, KM_PRESS, 0, 0);
 220         WM_keymap_add_item(keymap, "SKETCH_OT_select", SELECTMOUSE, KM_PRESS, 0, 0);
 221
 222         /* Pose ------------------------ */
 223         /* only set in posemode, by space_view3d listener */
 224         keymap= WM_keymap_listbase(wm, "Pose", 0, 0);
 225
 226         WM_keymap_add_item(keymap, "POSE_OT_hide", HKEY, KM_PRESS, 0, 0);
 227         kmi= WM_keymap_add_item(keymap, "POSE_OT_hide", HKEY, KM_PRESS, KM_SHIFT, 0);
 228                 RNA_boolean_set(kmi->ptr, "unselected", 1);
 229         WM_keymap_add_item(keymap, "POSE_OT_reveal", HKEY, KM_PRESS, KM_ALT, 0);
 230         /*clear pose*/
 231         WM_keymap_add_item(keymap, "POSE_OT_rot_clear", RKEY, KM_PRESS, KM_ALT, 0);
 232         WM_keymap_add_item(keymap, "POSE_OT_loc_clear", GKEY, KM_PRESS, KM_ALT, 0);
 233         WM_keymap_add_item(keymap, "POSE_OT_scale_clear", SKEY, KM_PRESS, KM_ALT, 0);
 234
 235         WM_keymap_add_item(keymap, "POSE_OT_select_all_toggle", AKEY, KM_PRESS, 0, 0);
 236         WM_keymap_add_item(keymap, "POSE_OT_select_inverse", IKEY, KM_PRESS, KM_CTRL, 0);
 237
 238         WM_keymap_add_item(keymap, "POSE_OT_select_parent", PKEY, KM_PRESS, KM_SHIFT, 0);
 239
 240         kmi= WM_keymap_add_item(keymap, "POSE_OT_select_hierarchy", LEFTBRACKETKEY, KM_PRESS, 0, 0);
 241                 RNA_enum_set(kmi->ptr, "direction", BONE_SELECT_PARENT);
 242         kmi= WM_keymap_add_item(keymap, "POSE_OT_select_hierarchy", LEFTBRACKETKEY, KM_PRESS, KM_SHIFT, 0);
 243                 RNA_enum_set(kmi->ptr, "direction", BONE_SELECT_PARENT);
 244                 RNA_boolean_set(kmi->ptr, "extend", 1);
 245
 246         kmi= WM_keymap_add_item(keymap, "POSE_OT_select_hierarchy", RIGHTBRACKETKEY, KM_PRESS, 0, 0);
 247                 RNA_enum_set(kmi->ptr, "direction", BONE_SELECT_CHILD);
 248         kmi= WM_keymap_add_item(keymap, "POSE_OT_select_hierarchy", RIGHTBRACKETKEY, KM_PRESS, KM_SHIFT, 0);
 249                 RNA_enum_set(kmi->ptr, "direction", BONE_SELECT_CHILD);
 250                 RNA_boolean_set(kmi->ptr, "extend", 1);
 251
 252         WM_keymap_add_item(keymap, "POSE_OT_select_linked", LKEY, KM_PRESS, 0, 0);
 253
 254         WM_keymap_add_item(keymap, "POSE_OT_constraint_add_with_targets", CKEY, KM_PRESS, KM_CTRL|KM_SHIFT, 0);
 255         WM_keymap_add_item(keymap, "POSE_OT_constraints_clear", CKEY, KM_PRESS, KM_CTRL|KM_ALT, 0);
 256         WM_keymap_add_item(keymap, "POSE_OT_ik_add", IKEY, KM_PRESS, /*KM_CTRL|*/KM_SHIFT, 0);
 257         WM_keymap_add_item(keymap, "POSE_OT_ik_clear", IKEY, KM_PRESS, KM_CTRL|KM_ALT, 0);
 258
 259         // XXX this should probably be in screen instead... here for testing purposes in the meantime... - Aligorith
 260         WM_keymap_verify_item(keymap, "ANIM_OT_insert_keyframe_menu", IKEY, KM_PRESS, 0, 0);
 261         WM_keymap_verify_item(keymap, "ANIM_OT_delete_keyframe_old", IKEY, KM_PRESS, KM_ALT, 0);
 262
 263         /* Pose -> PoseLib ------------- */
 264         /* only set in posemode, by space_view3d listener */
 265         WM_keymap_add_item(keymap, "POSELIB_OT_browse_interactive", LKEY, KM_PRESS, KM_CTRL, 0);
 266
 267         WM_keymap_add_item(keymap, "POSELIB_OT_pose_add", LKEY, KM_PRESS, KM_SHIFT, 0);
 268         WM_keymap_add_item(keymap, "POSELIB_OT_pose_remove", LKEY, KM_PRESS, KM_ALT, 0);
 269         WM_keymap_add_item(keymap, "POSELIB_OT_pose_rename", LKEY, KM_PRESS, KM_CTRL|KM_SHIFT, 0);
 270 }
 271