2.5 - Constraints Editing + Keyframe Drawing Tweaks
[blender-staging.git] / 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