aa9af4af154ceb60252247e26e2b5e742647a2c4
[blender.git] / source / blender / windowmanager / intern / wm_keymap.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) 2007 Blender Foundation.
19  * All rights reserved.
20  *
21  * 
22  * Contributor(s): Blender Foundation
23  *
24  * ***** END GPL LICENSE BLOCK *****
25  */
26
27 /** \file blender/windowmanager/intern/wm_keymap.c
28  *  \ingroup wm
29  */
30
31
32 #include <string.h>
33
34 #include "DNA_object_types.h"
35 #include "DNA_screen_types.h"
36 #include "DNA_space_types.h"
37 #include "DNA_userdef_types.h"
38 #include "DNA_windowmanager_types.h"
39
40 #include "MEM_guardedalloc.h"
41
42 #include "BLI_blenlib.h"
43 #include "BLI_utildefines.h"
44
45 #include "BKE_blender.h"
46 #include "BKE_context.h"
47 #include "BKE_idprop.h"
48 #include "BKE_global.h"
49 #include "BKE_library.h"
50 #include "BKE_main.h"
51 #include "BKE_screen.h"
52
53
54 #include "RNA_access.h"
55 #include "RNA_enum_types.h"
56
57 #include "WM_api.h"
58 #include "WM_types.h"
59 #include "wm_window.h"
60 #include "wm_event_system.h"
61 #include "wm_event_types.h"
62
63 /******************************* Keymap Item **********************************
64  * Item in a keymap, that maps from an event to an operator or modal map item */
65
66 static wmKeyMapItem *wm_keymap_item_copy(wmKeyMapItem *kmi)
67 {
68         wmKeyMapItem *kmin = MEM_dupallocN(kmi);
69
70         kmin->prev = kmin->next = NULL;
71         kmin->flag &= ~KMI_UPDATE;
72
73         if (kmin->properties) {
74                 kmin->ptr = MEM_callocN(sizeof(PointerRNA), "UserKeyMapItemPtr");
75                 WM_operator_properties_create(kmin->ptr, kmin->idname);
76
77                 kmin->properties = IDP_CopyProperty(kmin->properties);
78                 kmin->ptr->data = kmin->properties;
79         }
80
81         return kmin;
82 }
83
84 static void wm_keymap_item_free(wmKeyMapItem *kmi)
85 {
86         /* not kmi itself */
87         if (kmi->ptr) {
88                 WM_operator_properties_free(kmi->ptr);
89                 MEM_freeN(kmi->ptr);
90         }
91 }
92
93 static void wm_keymap_item_properties_set(wmKeyMapItem *kmi)
94 {
95         WM_operator_properties_alloc(&(kmi->ptr), &(kmi->properties), kmi->idname);
96         WM_operator_properties_sanitize(kmi->ptr, 1);
97 }
98
99 static int wm_keymap_item_equals_result(wmKeyMapItem *a, wmKeyMapItem *b)
100 {
101         if (strcmp(a->idname, b->idname) != 0)
102                 return 0;
103         
104         if (!RNA_struct_equals(a->ptr, b->ptr, true))
105                 return 0;
106         
107         if ((a->flag & KMI_INACTIVE) != (b->flag & KMI_INACTIVE))
108                 return 0;
109         
110         return (a->propvalue == b->propvalue);
111 }
112
113 static int wm_keymap_item_equals(wmKeyMapItem *a, wmKeyMapItem *b)
114 {
115         return (wm_keymap_item_equals_result(a, b) &&
116                 a->type == b->type &&
117                 a->val == b->val &&
118                 a->shift == b->shift &&
119                 a->ctrl == b->ctrl &&
120                 a->alt == b->alt &&
121                 a->oskey == b->oskey &&
122                 a->keymodifier == b->keymodifier &&
123                 a->maptype == b->maptype);
124 }
125
126 /* properties can be NULL, otherwise the arg passed is used and ownership is given to the kmi */
127 void WM_keymap_properties_reset(wmKeyMapItem *kmi, struct IDProperty *properties)
128 {
129         if (LIKELY(kmi->ptr)) {
130                 WM_operator_properties_free(kmi->ptr);
131                 MEM_freeN(kmi->ptr);
132
133                 kmi->ptr = NULL;
134         }
135
136         kmi->properties = properties;
137
138         wm_keymap_item_properties_set(kmi);
139 }
140
141 /**************************** Keymap Diff Item *********************************
142  * Item in a diff keymap, used for saving diff of keymaps in user preferences */
143
144 static wmKeyMapDiffItem *wm_keymap_diff_item_copy(wmKeyMapDiffItem *kmdi)
145 {
146         wmKeyMapDiffItem *kmdin = MEM_dupallocN(kmdi);
147
148         kmdin->next = kmdin->prev = NULL;
149         if (kmdi->add_item)
150                 kmdin->add_item = wm_keymap_item_copy(kmdi->add_item);
151         if (kmdi->remove_item)
152                 kmdin->remove_item = wm_keymap_item_copy(kmdi->remove_item);
153         
154         return kmdin;
155 }
156
157 static void wm_keymap_diff_item_free(wmKeyMapDiffItem *kmdi)
158 {
159         if (kmdi->remove_item) {
160                 wm_keymap_item_free(kmdi->remove_item);
161                 MEM_freeN(kmdi->remove_item);
162         }
163         if (kmdi->add_item) {
164                 wm_keymap_item_free(kmdi->add_item);
165                 MEM_freeN(kmdi->add_item);
166         }
167 }
168
169 /***************************** Key Configuration ******************************
170  * List of keymaps for all editors, modes, ... . There is a builtin default key
171  * configuration, a user key configuration, and other preset configurations. */
172
173 wmKeyConfig *WM_keyconfig_new(wmWindowManager *wm, const char *idname)
174 {
175         wmKeyConfig *keyconf;
176         
177         keyconf = MEM_callocN(sizeof(wmKeyConfig), "wmKeyConfig");
178         BLI_strncpy(keyconf->idname, idname, sizeof(keyconf->idname));
179         BLI_addtail(&wm->keyconfigs, keyconf);
180
181         return keyconf;
182 }
183
184 wmKeyConfig *WM_keyconfig_new_user(wmWindowManager *wm, const char *idname)
185 {
186         wmKeyConfig *keyconf = WM_keyconfig_new(wm, idname);
187
188         keyconf->flag |= KEYCONF_USER;
189
190         return keyconf;
191 }
192
193 int WM_keyconfig_remove(wmWindowManager *wm, wmKeyConfig *keyconf)
194 {
195         if (BLI_findindex(&wm->keyconfigs, keyconf) != -1) {
196                 if (strncmp(U.keyconfigstr, keyconf->idname, sizeof(U.keyconfigstr)) == 0) {
197                         BLI_strncpy(U.keyconfigstr, wm->defaultconf->idname, sizeof(U.keyconfigstr));
198                         WM_keyconfig_update_tag(NULL, NULL);
199                 }
200
201                 BLI_remlink(&wm->keyconfigs, keyconf);
202                 WM_keyconfig_free(keyconf);
203
204                 return TRUE;
205         }
206         else {
207                 return FALSE;
208         }
209 }
210
211 void WM_keyconfig_free(wmKeyConfig *keyconf)
212 {
213         wmKeyMap *km;
214
215         while ((km = keyconf->keymaps.first)) {
216                 WM_keymap_free(km);
217                 BLI_freelinkN(&keyconf->keymaps, km);
218         }
219
220         MEM_freeN(keyconf);
221 }
222
223 static wmKeyConfig *wm_keyconfig_list_find(ListBase *lb, char *idname)
224 {
225         wmKeyConfig *kc;
226
227         for (kc = lb->first; kc; kc = kc->next)
228                 if (0 == strncmp(idname, kc->idname, KMAP_MAX_NAME))
229                         return kc;
230         
231         return NULL;
232 }
233
234 static wmKeyConfig *WM_keyconfig_active(wmWindowManager *wm)
235 {
236         wmKeyConfig *keyconf;
237
238         /* first try from preset */
239         keyconf = wm_keyconfig_list_find(&wm->keyconfigs, U.keyconfigstr);
240         if (keyconf)
241                 return keyconf;
242         
243         /* otherwise use default */
244         return wm->defaultconf;
245 }
246
247 void WM_keyconfig_set_active(wmWindowManager *wm, const char *idname)
248 {
249         /* setting a different key configuration as active: we ensure all is
250          * updated properly before and after making the change */
251
252         WM_keyconfig_update(wm);
253
254         BLI_strncpy(U.keyconfigstr, idname, sizeof(U.keyconfigstr));
255
256         WM_keyconfig_update_tag(NULL, NULL);
257         WM_keyconfig_update(wm);
258 }
259
260 /********************************** Keymap *************************************
261  * List of keymap items for one editor, mode, modal operator, ... */
262
263 static wmKeyMap *wm_keymap_new(const char *idname, int spaceid, int regionid)
264 {
265         wmKeyMap *km = MEM_callocN(sizeof(struct wmKeyMap), "keymap list");
266
267         BLI_strncpy(km->idname, idname, KMAP_MAX_NAME);
268         km->spaceid = spaceid;
269         km->regionid = regionid;
270
271         return km;
272 }
273
274 static wmKeyMap *wm_keymap_copy(wmKeyMap *keymap)
275 {
276         wmKeyMap *keymapn = MEM_dupallocN(keymap);
277         wmKeyMapItem *kmi, *kmin;
278         wmKeyMapDiffItem *kmdi, *kmdin;
279
280         keymapn->modal_items = keymap->modal_items;
281         keymapn->poll = keymap->poll;
282         keymapn->items.first = keymapn->items.last = NULL;
283         keymapn->flag &= ~(KEYMAP_UPDATE | KEYMAP_EXPANDED);
284
285         for (kmdi = keymap->diff_items.first; kmdi; kmdi = kmdi->next) {
286                 kmdin = wm_keymap_diff_item_copy(kmdi);
287                 BLI_addtail(&keymapn->items, kmdin);
288         }
289
290         for (kmi = keymap->items.first; kmi; kmi = kmi->next) {
291                 kmin = wm_keymap_item_copy(kmi);
292                 BLI_addtail(&keymapn->items, kmin);
293         }
294
295         return keymapn;
296 }
297
298 void WM_keymap_free(wmKeyMap *keymap)
299 {
300         wmKeyMapItem *kmi;
301         wmKeyMapDiffItem *kmdi;
302
303         for (kmdi = keymap->diff_items.first; kmdi; kmdi = kmdi->next)
304                 wm_keymap_diff_item_free(kmdi);
305
306         for (kmi = keymap->items.first; kmi; kmi = kmi->next)
307                 wm_keymap_item_free(kmi);
308
309         BLI_freelistN(&keymap->diff_items);
310         BLI_freelistN(&keymap->items);
311 }
312
313 static void keymap_event_set(wmKeyMapItem *kmi, short type, short val, int modifier, short keymodifier)
314 {
315         kmi->type = type;
316         kmi->val = val;
317         kmi->keymodifier = keymodifier;
318
319         if (modifier == KM_ANY) {
320                 kmi->shift = kmi->ctrl = kmi->alt = kmi->oskey = KM_ANY;
321         }
322         else {
323                 kmi->shift = (modifier & KM_SHIFT) ? KM_MOD_FIRST : ((modifier & KM_SHIFT2) ? KM_MOD_SECOND : FALSE);
324                 kmi->ctrl =  (modifier & KM_CTRL)  ? KM_MOD_FIRST : ((modifier & KM_CTRL2)  ? KM_MOD_SECOND : FALSE);
325                 kmi->alt =   (modifier & KM_ALT)   ? KM_MOD_FIRST : ((modifier & KM_ALT2)   ? KM_MOD_SECOND : FALSE);
326                 kmi->oskey = (modifier & KM_OSKEY) ? KM_MOD_FIRST : ((modifier & KM_OSKEY2) ? KM_MOD_SECOND : FALSE);
327         }
328 }
329
330 static void keymap_item_set_id(wmKeyMap *keymap, wmKeyMapItem *kmi)
331 {
332         keymap->kmi_id++;
333         if ((keymap->flag & KEYMAP_USER) == 0) {
334                 kmi->id = keymap->kmi_id;
335         }
336         else {
337                 kmi->id = -keymap->kmi_id; /* User defined keymap entries have negative ids */
338         }
339 }
340
341 /* if item was added, then bail out */
342 wmKeyMapItem *WM_keymap_verify_item(wmKeyMap *keymap, const char *idname, int type, int val, int modifier, int keymodifier)
343 {
344         wmKeyMapItem *kmi;
345         
346         for (kmi = keymap->items.first; kmi; kmi = kmi->next)
347                 if (strncmp(kmi->idname, idname, OP_MAX_TYPENAME) == 0)
348                         break;
349         if (kmi == NULL) {
350                 kmi = MEM_callocN(sizeof(wmKeyMapItem), "keymap entry");
351                 
352                 BLI_addtail(&keymap->items, kmi);
353                 BLI_strncpy(kmi->idname, idname, OP_MAX_TYPENAME);
354                 
355                 keymap_item_set_id(keymap, kmi);
356
357                 keymap_event_set(kmi, type, val, modifier, keymodifier);
358                 wm_keymap_item_properties_set(kmi);
359         }
360         return kmi;
361 }
362
363 /* always add item */
364 wmKeyMapItem *WM_keymap_add_item(wmKeyMap *keymap, const char *idname, int type, int val, int modifier, int keymodifier)
365 {
366         wmKeyMapItem *kmi = MEM_callocN(sizeof(wmKeyMapItem), "keymap entry");
367         
368         BLI_addtail(&keymap->items, kmi);
369         BLI_strncpy(kmi->idname, idname, OP_MAX_TYPENAME);
370
371         keymap_event_set(kmi, type, val, modifier, keymodifier);
372         wm_keymap_item_properties_set(kmi);
373
374         keymap_item_set_id(keymap, kmi);
375
376         WM_keyconfig_update_tag(keymap, kmi);
377
378         return kmi;
379 }
380
381 /* menu wrapper for WM_keymap_add_item */
382 wmKeyMapItem *WM_keymap_add_menu(wmKeyMap *keymap, const char *idname, int type, int val, int modifier, int keymodifier)
383 {
384         wmKeyMapItem *kmi = WM_keymap_add_item(keymap, "WM_OT_call_menu", type, val, modifier, keymodifier);
385         RNA_string_set(kmi->ptr, "name", idname);
386         return kmi;
387 }
388
389 int WM_keymap_remove_item(wmKeyMap *keymap, wmKeyMapItem *kmi)
390 {
391         if (BLI_findindex(&keymap->items, kmi) != -1) {
392                 if (kmi->ptr) {
393                         WM_operator_properties_free(kmi->ptr);
394                         MEM_freeN(kmi->ptr);
395                 }
396                 BLI_freelinkN(&keymap->items, kmi);
397
398                 WM_keyconfig_update_tag(keymap, NULL);
399                 return TRUE;
400         }
401         else {
402                 return FALSE;
403         }
404 }
405
406 /************************** Keymap Diff and Patch ****************************
407  * Rather than saving the entire keymap for user preferences, we only save a
408  * diff so that changes in the defaults get synced. This system is not perfect
409  * but works better than overriding the keymap entirely when only few items
410  * are changed. */
411
412 static void wm_keymap_addon_add(wmKeyMap *keymap, wmKeyMap *addonmap)
413 {
414         wmKeyMapItem *kmi, *kmin;
415
416         for (kmi = addonmap->items.first; kmi; kmi = kmi->next) {
417                 kmin = wm_keymap_item_copy(kmi);
418                 keymap_item_set_id(keymap, kmin);
419                 BLI_addhead(&keymap->items, kmin);
420         }
421 }
422
423 static wmKeyMapItem *wm_keymap_find_item_equals(wmKeyMap *km, wmKeyMapItem *needle)
424 {
425         wmKeyMapItem *kmi;
426
427         for (kmi = km->items.first; kmi; kmi = kmi->next)
428                 if (wm_keymap_item_equals(kmi, needle))
429                         return kmi;
430         
431         return NULL;
432 }
433
434 static wmKeyMapItem *wm_keymap_find_item_equals_result(wmKeyMap *km, wmKeyMapItem *needle)
435 {
436         wmKeyMapItem *kmi;
437
438         for (kmi = km->items.first; kmi; kmi = kmi->next)
439                 if (wm_keymap_item_equals_result(kmi, needle))
440                         return kmi;
441         
442         return NULL;
443 }
444
445 static void wm_keymap_diff(wmKeyMap *diff_km, wmKeyMap *from_km, wmKeyMap *to_km, wmKeyMap *orig_km, wmKeyMap *addon_km)
446 {
447         wmKeyMapItem *kmi, *to_kmi, *orig_kmi;
448         wmKeyMapDiffItem *kmdi;
449
450         for (kmi = from_km->items.first; kmi; kmi = kmi->next) {
451                 to_kmi = WM_keymap_item_find_id(to_km, kmi->id);
452
453                 if (!to_kmi) {
454                         /* remove item */
455                         kmdi = MEM_callocN(sizeof(wmKeyMapDiffItem), "wmKeyMapDiffItem");
456                         kmdi->remove_item = wm_keymap_item_copy(kmi);
457                         BLI_addtail(&diff_km->diff_items, kmdi);
458                 }
459                 else if (to_kmi && !wm_keymap_item_equals(kmi, to_kmi)) {
460                         /* replace item */
461                         kmdi = MEM_callocN(sizeof(wmKeyMapDiffItem), "wmKeyMapDiffItem");
462                         kmdi->remove_item = wm_keymap_item_copy(kmi);
463                         kmdi->add_item = wm_keymap_item_copy(to_kmi);
464                         BLI_addtail(&diff_km->diff_items, kmdi);
465                 }
466
467                 /* sync expanded flag back to original so we don't loose it on repatch */
468                 if (to_kmi) {
469                         orig_kmi = WM_keymap_item_find_id(orig_km, kmi->id);
470
471                         if (!orig_kmi)
472                                 orig_kmi = wm_keymap_find_item_equals(addon_km, kmi);
473
474                         if (orig_kmi) {
475                                 orig_kmi->flag &= ~KMI_EXPANDED;
476                                 orig_kmi->flag |= (to_kmi->flag & KMI_EXPANDED);
477                         }
478                 }
479         }
480
481         for (kmi = to_km->items.first; kmi; kmi = kmi->next) {
482                 if (kmi->id < 0) {
483                         /* add item */
484                         kmdi = MEM_callocN(sizeof(wmKeyMapDiffItem), "wmKeyMapDiffItem");
485                         kmdi->add_item = wm_keymap_item_copy(kmi);
486                         BLI_addtail(&diff_km->diff_items, kmdi);
487                 }
488         }
489 }
490
491 static void wm_keymap_patch(wmKeyMap *km, wmKeyMap *diff_km)
492 {
493         wmKeyMapDiffItem *kmdi;
494         wmKeyMapItem *kmi_remove, *kmi_add;
495
496         for (kmdi = diff_km->diff_items.first; kmdi; kmdi = kmdi->next) {
497                 /* find item to remove */
498                 kmi_remove = NULL;
499                 if (kmdi->remove_item) {
500                         kmi_remove = wm_keymap_find_item_equals(km, kmdi->remove_item);
501                         if (!kmi_remove)
502                                 kmi_remove = wm_keymap_find_item_equals_result(km, kmdi->remove_item);
503                 }
504
505                 /* add item */
506                 if (kmdi->add_item) {
507                         /* only if nothing to remove or item to remove found */
508                         if (!kmdi->remove_item || kmi_remove) {
509                                 kmi_add = wm_keymap_item_copy(kmdi->add_item);
510                                 kmi_add->flag |= KMI_USER_MODIFIED;
511
512                                 if (kmi_remove) {
513                                         kmi_add->flag &= ~KMI_EXPANDED;
514                                         kmi_add->flag |= (kmi_remove->flag & KMI_EXPANDED);
515                                         kmi_add->id = kmi_remove->id;
516                                         BLI_insertlinkbefore(&km->items, kmi_remove, kmi_add);
517                                 }
518                                 else {
519                                         keymap_item_set_id(km, kmi_add);
520                                         BLI_addtail(&km->items, kmi_add);
521                                 }
522                         }
523                 }
524
525                 /* remove item */
526                 if (kmi_remove) {
527                         wm_keymap_item_free(kmi_remove);
528                         BLI_freelinkN(&km->items, kmi_remove);
529                 }
530         }
531 }
532
533 static wmKeyMap *wm_keymap_patch_update(ListBase *lb, wmKeyMap *defaultmap, wmKeyMap *addonmap, wmKeyMap *usermap)
534 {
535         wmKeyMap *km;
536         int expanded = 0;
537
538         /* remove previous keymap in list, we will replace it */
539         km = WM_keymap_list_find(lb, defaultmap->idname, defaultmap->spaceid, defaultmap->regionid);
540         if (km) {
541                 expanded = (km->flag & (KEYMAP_EXPANDED | KEYMAP_CHILDREN_EXPANDED));
542                 WM_keymap_free(km);
543                 BLI_freelinkN(lb, km);
544         }
545
546         /* copy new keymap from an existing one */
547         if (usermap && !(usermap->flag & KEYMAP_DIFF)) {
548                 /* for compatibility with old user preferences with non-diff
549                  * keymaps we override the original entirely */
550                 wmKeyMapItem *kmi, *orig_kmi;
551
552                 km = wm_keymap_copy(usermap);
553
554                 /* try to find corresponding id's for items */
555                 for (kmi = km->items.first; kmi; kmi = kmi->next) {
556                         orig_kmi = wm_keymap_find_item_equals(defaultmap, kmi);
557                         if (!orig_kmi)
558                                 orig_kmi = wm_keymap_find_item_equals_result(defaultmap, kmi);
559
560                         if (orig_kmi)
561                                 kmi->id = orig_kmi->id;
562                         else
563                                 kmi->id = -(km->kmi_id++);
564                 }
565
566                 km->flag |= KEYMAP_UPDATE; /* update again to create diff */
567         }
568         else
569                 km = wm_keymap_copy(defaultmap);
570
571         /* add addon keymap items */
572         if (addonmap)
573                 wm_keymap_addon_add(km, addonmap);
574
575         /* tag as being user edited */
576         if (usermap)
577                 km->flag |= KEYMAP_USER_MODIFIED;
578         km->flag |= KEYMAP_USER | expanded;
579
580         /* apply user changes of diff keymap */
581         if (usermap && (usermap->flag & KEYMAP_DIFF))
582                 wm_keymap_patch(km, usermap);
583
584         /* add to list */
585         BLI_addtail(lb, km);
586         
587         return km;
588 }
589
590 static void wm_keymap_diff_update(ListBase *lb, wmKeyMap *defaultmap, wmKeyMap *addonmap, wmKeyMap *km)
591 {
592         wmKeyMap *diffmap, *prevmap, *origmap;
593
594         /* create temporary default + addon keymap for diff */
595         origmap = defaultmap;
596
597         if (addonmap) {
598                 defaultmap = wm_keymap_copy(defaultmap);
599                 wm_keymap_addon_add(defaultmap, addonmap);
600         }
601
602         /* remove previous diff keymap in list, we will replace it */
603         prevmap = WM_keymap_list_find(lb, km->idname, km->spaceid, km->regionid);
604         if (prevmap) {
605                 WM_keymap_free(prevmap);
606                 BLI_freelinkN(lb, prevmap);
607         }
608
609         /* create diff keymap */
610         diffmap = wm_keymap_new(km->idname, km->spaceid, km->regionid);
611         diffmap->flag |= KEYMAP_DIFF;
612         if (defaultmap->flag & KEYMAP_MODAL)
613                 diffmap->flag |= KEYMAP_MODAL;
614         wm_keymap_diff(diffmap, defaultmap, km, origmap, addonmap);
615
616         /* add to list if not empty */
617         if (diffmap->diff_items.first) {
618                 BLI_addtail(lb, diffmap);
619         }
620         else {
621                 WM_keymap_free(diffmap);
622                 MEM_freeN(diffmap);
623         }
624
625         /* free temporary default map */
626         if (addonmap) {
627                 WM_keymap_free(defaultmap);
628                 MEM_freeN(defaultmap);
629         }
630 }
631
632 /* ****************** storage in WM ************ */
633
634 /* name id's are for storing general or multiple keymaps, 
635  * space/region ids are same as DNA_space_types.h */
636 /* gets freed in wm.c */
637
638 wmKeyMap *WM_keymap_list_find(ListBase *lb, const char *idname, int spaceid, int regionid)
639 {
640         wmKeyMap *km;
641
642         for (km = lb->first; km; km = km->next)
643                 if (km->spaceid == spaceid && km->regionid == regionid)
644                         if (0 == strncmp(idname, km->idname, KMAP_MAX_NAME))
645                                 return km;
646         
647         return NULL;
648 }
649
650 wmKeyMap *WM_keymap_find(wmKeyConfig *keyconf, const char *idname, int spaceid, int regionid)
651 {
652         wmKeyMap *km = WM_keymap_list_find(&keyconf->keymaps, idname, spaceid, regionid);
653         
654         if (km == NULL) {
655                 km = wm_keymap_new(idname, spaceid, regionid);
656                 BLI_addtail(&keyconf->keymaps, km);
657
658                 WM_keyconfig_update_tag(km, NULL);
659         }
660         
661         return km;
662 }
663
664 wmKeyMap *WM_keymap_find_all(const bContext *C, const char *idname, int spaceid, int regionid)
665 {
666         wmWindowManager *wm = CTX_wm_manager(C);
667
668         return WM_keymap_list_find(&wm->userconf->keymaps, idname, spaceid, regionid);
669 }
670
671 /* ****************** modal keymaps ************ */
672
673 /* modal maps get linked to a running operator, and filter the keys before sending to modal() callback */
674
675 wmKeyMap *WM_modalkeymap_add(wmKeyConfig *keyconf, const char *idname, EnumPropertyItem *items)
676 {
677         wmKeyMap *km = WM_keymap_find(keyconf, idname, 0, 0);
678         km->flag |= KEYMAP_MODAL;
679         km->modal_items = items;
680
681         if (!items) {
682                 /* init modal items from default config */
683                 wmWindowManager *wm = G.main->wm.first;
684                 wmKeyMap *defaultkm = WM_keymap_list_find(&wm->defaultconf->keymaps, km->idname, 0, 0);
685
686                 if (defaultkm) {
687                         km->modal_items = defaultkm->modal_items;
688                         km->poll = defaultkm->poll;
689                 }
690         }
691         
692         return km;
693 }
694
695 wmKeyMap *WM_modalkeymap_get(wmKeyConfig *keyconf, const char *idname)
696 {
697         wmKeyMap *km;
698         
699         for (km = keyconf->keymaps.first; km; km = km->next)
700                 if (km->flag & KEYMAP_MODAL)
701                         if (0 == strncmp(idname, km->idname, KMAP_MAX_NAME))
702                                 break;
703         
704         return km;
705 }
706
707
708 wmKeyMapItem *WM_modalkeymap_add_item(wmKeyMap *km, int type, int val, int modifier, int keymodifier, int value)
709 {
710         wmKeyMapItem *kmi = MEM_callocN(sizeof(wmKeyMapItem), "keymap entry");
711         
712         BLI_addtail(&km->items, kmi);
713         kmi->propvalue = value;
714         
715         keymap_event_set(kmi, type, val, modifier, keymodifier);
716
717         keymap_item_set_id(km, kmi);
718
719         WM_keyconfig_update_tag(km, kmi);
720
721         return kmi;
722 }
723
724 wmKeyMapItem *WM_modalkeymap_add_item_str(wmKeyMap *km, int type, int val, int modifier, int keymodifier, const char *value)
725 {
726         wmKeyMapItem *kmi = MEM_callocN(sizeof(wmKeyMapItem), "keymap entry");
727
728         BLI_addtail(&km->items, kmi);
729         BLI_strncpy(kmi->propvalue_str, value, sizeof(kmi->propvalue_str));
730
731         keymap_event_set(kmi, type, val, modifier, keymodifier);
732
733         keymap_item_set_id(km, kmi);
734
735         WM_keyconfig_update_tag(km, kmi);
736
737         return kmi;
738 }
739
740 wmKeyMapItem *WM_modalkeymap_find_propvalue(wmKeyMap *km, const int propvalue)
741 {
742
743         if (km->flag & KEYMAP_MODAL) {
744                 wmKeyMapItem *kmi;
745                 for (kmi = km->items.first; kmi; kmi = kmi->next) {
746                         if (kmi->propvalue == propvalue) {
747                                 return kmi;
748                         }
749                 }
750         }
751         else {
752                 BLI_assert(!"called with non modal keymap");
753         }
754
755         return NULL;
756 }
757
758 void WM_modalkeymap_assign(wmKeyMap *km, const char *opname)
759 {
760         wmOperatorType *ot = WM_operatortype_find(opname, 0);
761         
762         if (ot)
763                 ot->modalkeymap = km;
764         else
765                 printf("error: modalkeymap_assign, unknown operator %s\n", opname);
766 }
767
768 static void wm_user_modal_keymap_set_items(wmWindowManager *wm, wmKeyMap *km)
769 {
770         /* here we convert propvalue string values delayed, due to python keymaps
771          * being created before the actual modal keymaps, so no modal_items */
772         wmKeyMap *defaultkm;
773         wmKeyMapItem *kmi;
774         int propvalue;
775
776         if (km && (km->flag & KEYMAP_MODAL) && !km->modal_items) {
777                 defaultkm = WM_keymap_list_find(&wm->defaultconf->keymaps, km->idname, 0, 0);
778
779                 if (!defaultkm)
780                         return;
781
782                 km->modal_items = defaultkm->modal_items;
783                 km->poll = defaultkm->poll;
784
785                 for (kmi = km->items.first; kmi; kmi = kmi->next) {
786                         if (kmi->propvalue_str[0]) {
787                                 if (RNA_enum_value_from_id(km->modal_items, kmi->propvalue_str, &propvalue))
788                                         kmi->propvalue = propvalue;
789                                 kmi->propvalue_str[0] = '\0';
790                         }
791                 }
792         }
793 }
794
795 /* ***************** get string from key events **************** */
796
797 const char *WM_key_event_string(short type)
798 {
799         const char *name = NULL;
800         if (RNA_enum_name(event_type_items, (int)type, &name))
801                 return name;
802         
803         return "";
804 }
805
806 int WM_keymap_item_to_string(wmKeyMapItem *kmi, char *str, int len)
807 {
808         char buf[128];
809
810         buf[0] = 0;
811
812         if (kmi->shift == KM_ANY &&
813             kmi->ctrl == KM_ANY &&
814             kmi->alt == KM_ANY &&
815             kmi->oskey == KM_ANY)
816         {
817                 strcat(buf, "Any ");
818         }
819         else {
820                 if (kmi->shift)
821                         strcat(buf, "Shift ");
822
823                 if (kmi->ctrl)
824                         strcat(buf, "Ctrl ");
825
826                 if (kmi->alt)
827                         strcat(buf, "Alt ");
828
829                 if (kmi->oskey)
830                         strcat(buf, "Cmd ");
831         }
832                 
833         if (kmi->keymodifier) {
834                 strcat(buf, WM_key_event_string(kmi->keymodifier));
835                 strcat(buf, " ");
836         }
837
838         strcat(buf, WM_key_event_string(kmi->type));
839         return BLI_strncpy_rlen(str, buf, len);
840 }
841
842 static wmKeyMapItem *wm_keymap_item_find_handlers(
843         const bContext *C, ListBase *handlers, const char *opname, int UNUSED(opcontext),
844         IDProperty *properties, int is_strict, int hotkey, wmKeyMap **keymap_r)
845 {
846         wmWindowManager *wm = CTX_wm_manager(C);
847         wmEventHandler *handler;
848         wmKeyMap *keymap;
849         wmKeyMapItem *kmi;
850
851         /* find keymap item in handlers */
852         for (handler = handlers->first; handler; handler = handler->next) {
853                 keymap = WM_keymap_active(wm, handler->keymap);
854
855                 if (keymap && (!keymap->poll || keymap->poll((bContext *)C))) {
856                         for (kmi = keymap->items.first; kmi; kmi = kmi->next) {
857                                 
858                                 if (strcmp(kmi->idname, opname) == 0 && WM_key_event_string(kmi->type)[0]) {
859                                         if (hotkey)
860                                                 if (!ISHOTKEY(kmi->type))
861                                                         continue;
862
863                                         if (properties) {
864
865                                                 /* example of debugging keymaps */
866 #if 0
867                                                 if (kmi->ptr) {
868                                                         if (strcmp("MESH_OT_rip_move", opname) == 0) {
869                                                                 printf("OPERATOR\n");
870                                                                 IDP_spit(properties);
871                                                                 printf("KEYMAP\n");
872                                                                 IDP_spit(kmi->ptr->data);
873                                                         }
874                                                 }
875 #endif
876
877                                                 if (kmi->ptr) {
878                                                         PointerRNA properties_ptr;
879                                                         RNA_pointer_create(NULL, kmi->ptr->type, properties, &properties_ptr);
880
881                                                         if (RNA_struct_equals(&properties_ptr, kmi->ptr, is_strict)) {
882                                                                 if (keymap_r) *keymap_r = keymap;
883                                                                 return kmi;
884                                                         }
885                                                 }
886                                         }
887                                         else {
888                                                 if (keymap_r) *keymap_r = keymap;
889                                                 return kmi;
890                                         }
891                                 }
892                         }
893                 }
894         }
895         
896         /* ensure un-initialized keymap is never used */
897         if (keymap_r) *keymap_r = NULL;
898         return NULL;
899 }
900
901 static wmKeyMapItem *wm_keymap_item_find_props(
902         const bContext *C, const char *opname, int opcontext,
903         IDProperty *properties, int is_strict, int hotkey, wmKeyMap **keymap_r)
904 {
905         wmWindow *win = CTX_wm_window(C);
906         ScrArea *sa = CTX_wm_area(C);
907         ARegion *ar = CTX_wm_region(C);
908         wmKeyMapItem *found = NULL;
909
910         /* look into multiple handler lists to find the item */
911         if (win)
912                 found = wm_keymap_item_find_handlers(C, &win->handlers, opname, opcontext, properties, is_strict, hotkey, keymap_r);
913
914         if (sa && found == NULL)
915                 found = wm_keymap_item_find_handlers(C, &sa->handlers, opname, opcontext, properties, is_strict, hotkey, keymap_r);
916
917         if (found == NULL) {
918                 if (ELEM(opcontext, WM_OP_EXEC_REGION_WIN, WM_OP_INVOKE_REGION_WIN)) {
919                         if (sa) {
920                                 if (!(ar && ar->regiontype == RGN_TYPE_WINDOW))
921                                         ar = BKE_area_find_region_type(sa, RGN_TYPE_WINDOW);
922                                 
923                                 if (ar)
924                                         found = wm_keymap_item_find_handlers(C, &ar->handlers, opname, opcontext, properties, is_strict, hotkey, keymap_r);
925                         }
926                 }
927                 else if (ELEM(opcontext, WM_OP_EXEC_REGION_CHANNELS, WM_OP_INVOKE_REGION_CHANNELS)) {
928                         if (!(ar && ar->regiontype == RGN_TYPE_CHANNELS))
929                                 ar = BKE_area_find_region_type(sa, RGN_TYPE_CHANNELS);
930
931                         if (ar)
932                                 found = wm_keymap_item_find_handlers(C, &ar->handlers, opname, opcontext, properties, is_strict, hotkey, keymap_r);
933                 }
934                 else if (ELEM(opcontext, WM_OP_EXEC_REGION_PREVIEW, WM_OP_INVOKE_REGION_PREVIEW)) {
935                         if (!(ar && ar->regiontype == RGN_TYPE_PREVIEW))
936                                 ar = BKE_area_find_region_type(sa, RGN_TYPE_PREVIEW);
937
938                         if (ar)
939                                 found = wm_keymap_item_find_handlers(C, &ar->handlers, opname, opcontext, properties, is_strict, hotkey, keymap_r);
940                 }
941                 else {
942                         if (ar)
943                                 found = wm_keymap_item_find_handlers(C, &ar->handlers, opname, opcontext, properties, is_strict, hotkey, keymap_r);
944                 }
945         }
946
947         return found;
948 }
949
950 static wmKeyMapItem *wm_keymap_item_find(
951         const bContext *C, const char *opname, int opcontext,
952         IDProperty *properties, const short hotkey, const short UNUSED(sloppy), wmKeyMap **keymap_r)
953 {
954         wmKeyMapItem *found = wm_keymap_item_find_props(C, opname, opcontext, properties, 1, hotkey, keymap_r);
955
956         if (!found && properties) {
957                 wmOperatorType *ot = WM_operatortype_find(opname, TRUE);
958                 if (ot) {
959                         /* make a copy of the properties and set any unset props
960                          * to their default values, so the ID property compare function succeeds */
961                         PointerRNA opptr;
962                         IDProperty *properties_default = IDP_CopyProperty(properties);
963
964                         RNA_pointer_create(NULL, ot->srna, properties_default, &opptr);
965
966                         if (WM_operator_properties_default(&opptr, true) ||
967                             (ot->prop && RNA_property_is_set(&opptr, ot->prop))) {
968                                 /* for operator that has enum menu, unset it so it always matches */
969                                 if (ot->prop)
970                                         RNA_property_unset(&opptr, ot->prop);
971
972                                 found = wm_keymap_item_find_props(C, opname, opcontext, properties_default, 0, hotkey, keymap_r);
973                         }
974
975                         IDP_FreeProperty(properties_default);
976                         MEM_freeN(properties_default);
977                 }
978         }
979
980         return found;
981 }
982
983 char *WM_key_event_operator_string(
984         const bContext *C, const char *opname, int opcontext,
985         IDProperty *properties, const bool sloppy, char *str, int len)
986 {
987         wmKeyMapItem *kmi = wm_keymap_item_find(C, opname, opcontext, properties, 0, sloppy, NULL);
988         
989         if (kmi) {
990                 WM_keymap_item_to_string(kmi, str, len);
991                 return str;
992         }
993
994         return NULL;
995 }
996
997 int WM_key_event_operator_id(
998         const bContext *C, const char *opname, int opcontext,
999         IDProperty *properties, int hotkey, wmKeyMap **keymap_r)
1000 {
1001         wmKeyMapItem *kmi = wm_keymap_item_find(C, opname, opcontext, properties, hotkey, TRUE, keymap_r);
1002         
1003         if (kmi)
1004                 return kmi->id;
1005         else
1006                 return 0;
1007 }
1008
1009 int WM_keymap_item_compare(wmKeyMapItem *k1, wmKeyMapItem *k2)
1010 {
1011         int k1type, k2type;
1012
1013         if (k1->flag & KMI_INACTIVE || k2->flag & KMI_INACTIVE)
1014                 return 0;
1015
1016         /* take event mapping into account */
1017         k1type = WM_userdef_event_map(k1->type);
1018         k2type = WM_userdef_event_map(k2->type);
1019
1020         if (k1type != KM_ANY && k2type != KM_ANY && k1type != k2type)
1021                 return 0;
1022
1023         if (k1->val != KM_ANY && k2->val != KM_ANY) {
1024                 /* take click, press, release conflict into account */
1025                 if (k1->val == KM_CLICK && ELEM3(k2->val, KM_PRESS, KM_RELEASE, KM_CLICK) == 0)
1026                         return 0;
1027                 if (k2->val == KM_CLICK && ELEM3(k1->val, KM_PRESS, KM_RELEASE, KM_CLICK) == 0)
1028                         return 0;
1029                 if (k1->val != k2->val)
1030                         return 0;
1031         }
1032
1033         if (k1->shift != KM_ANY && k2->shift != KM_ANY && k1->shift != k2->shift)
1034                 return 0;
1035
1036         if (k1->ctrl != KM_ANY && k2->ctrl != KM_ANY && k1->ctrl != k2->ctrl)
1037                 return 0;
1038
1039         if (k1->alt != KM_ANY && k2->alt != KM_ANY && k1->alt != k2->alt)
1040                 return 0;
1041
1042         if (k1->oskey != KM_ANY && k2->oskey != KM_ANY && k1->oskey != k2->oskey)
1043                 return 0;
1044
1045         if (k1->keymodifier != k2->keymodifier)
1046                 return 0;
1047
1048         return 1;
1049 }
1050
1051 /************************* Update Final Configuration *************************
1052  * On load or other changes, the final user key configuration is rebuilt from
1053  * the preset, addon and user preferences keymaps. We also test if the final
1054  * configuration changed and write the changes to the user preferences. */
1055
1056 static int WM_KEYMAP_UPDATE = 0;
1057
1058 void WM_keyconfig_update_tag(wmKeyMap *km, wmKeyMapItem *kmi)
1059 {
1060         /* quick tag to do delayed keymap updates */
1061         WM_KEYMAP_UPDATE = 1;
1062
1063         if (km)
1064                 km->flag |= KEYMAP_UPDATE;
1065         if (kmi)
1066                 kmi->flag |= KMI_UPDATE;
1067 }
1068
1069 static int wm_keymap_test_and_clear_update(wmKeyMap *km)
1070 {
1071         wmKeyMapItem *kmi;
1072         int update;
1073         
1074         update = (km->flag & KEYMAP_UPDATE);
1075         km->flag &= ~KEYMAP_UPDATE;
1076
1077         for (kmi = km->items.first; kmi; kmi = kmi->next) {
1078                 update = update || (kmi->flag & KMI_UPDATE);
1079                 kmi->flag &= ~KMI_UPDATE;
1080         }
1081         
1082         return update;
1083 }
1084
1085 static wmKeyMap *wm_keymap_preset(wmWindowManager *wm, wmKeyMap *km)
1086 {
1087         wmKeyConfig *keyconf = WM_keyconfig_active(wm);
1088         wmKeyMap *keymap;
1089
1090         keymap = WM_keymap_list_find(&keyconf->keymaps, km->idname, km->spaceid, km->regionid);
1091         if (!keymap)
1092                 keymap = WM_keymap_list_find(&wm->defaultconf->keymaps, km->idname, km->spaceid, km->regionid);
1093
1094         return keymap;
1095 }
1096
1097 void WM_keyconfig_update(wmWindowManager *wm)
1098 {
1099         wmKeyMap *km, *defaultmap, *addonmap, *usermap, *kmn;
1100         wmKeyMapItem *kmi;
1101         wmKeyMapDiffItem *kmdi;
1102         int compat_update = 0;
1103
1104         if (G.background)
1105                 return;
1106         if (!WM_KEYMAP_UPDATE)
1107                 return;
1108         
1109         /* update operator properties for non-modal user keymaps */
1110         for (km = U.user_keymaps.first; km; km = km->next) {
1111                 if ((km->flag & KEYMAP_MODAL) == 0) {
1112                         for (kmdi = km->diff_items.first; kmdi; kmdi = kmdi->next) {
1113                                 if (kmdi->add_item)
1114                                         wm_keymap_item_properties_set(kmdi->add_item);
1115                                 if (kmdi->remove_item)
1116                                         wm_keymap_item_properties_set(kmdi->remove_item);
1117                         }
1118
1119                         for (kmi = km->items.first; kmi; kmi = kmi->next)
1120                                 wm_keymap_item_properties_set(kmi);
1121                 }
1122         }
1123
1124         /* update U.user_keymaps with user key configuration changes */
1125         for (km = wm->userconf->keymaps.first; km; km = km->next) {
1126                 /* only diff if the user keymap was modified */
1127                 if (wm_keymap_test_and_clear_update(km)) {
1128                         /* find keymaps */
1129                         defaultmap = wm_keymap_preset(wm, km);
1130                         addonmap = WM_keymap_list_find(&wm->addonconf->keymaps, km->idname, km->spaceid, km->regionid);
1131
1132                         /* diff */
1133                         if (defaultmap)
1134                                 wm_keymap_diff_update(&U.user_keymaps, defaultmap, addonmap, km);
1135                 }
1136         }
1137
1138         /* create user key configuration from preset + addon + user preferences */
1139         for (km = wm->defaultconf->keymaps.first; km; km = km->next) {
1140                 /* find keymaps */
1141                 defaultmap = wm_keymap_preset(wm, km);
1142                 addonmap = WM_keymap_list_find(&wm->addonconf->keymaps, km->idname, km->spaceid, km->regionid);
1143                 usermap = WM_keymap_list_find(&U.user_keymaps, km->idname, km->spaceid, km->regionid);
1144
1145                 wm_user_modal_keymap_set_items(wm, defaultmap);
1146
1147                 /* add */
1148                 kmn = wm_keymap_patch_update(&wm->userconf->keymaps, defaultmap, addonmap, usermap);
1149
1150                 if (kmn) {
1151                         kmn->modal_items = km->modal_items;
1152                         kmn->poll = km->poll;
1153                 }
1154
1155                 /* in case of old non-diff keymaps, force extra update to create diffs */
1156                 compat_update = compat_update || (usermap && !(usermap->flag & KEYMAP_DIFF));
1157         }
1158
1159         WM_KEYMAP_UPDATE = 0;
1160
1161         if (compat_update) {
1162                 WM_keyconfig_update_tag(NULL, NULL);
1163                 WM_keyconfig_update(wm);
1164         }
1165 }
1166
1167 /********************************* Event Handling *****************************
1168  * Handlers have pointers to the keymap in the default configuration. During
1169  * event handling this function is called to get the keymap from the final
1170  * configuration. */
1171
1172 wmKeyMap *WM_keymap_active(wmWindowManager *wm, wmKeyMap *keymap)
1173 {
1174         wmKeyMap *km;
1175
1176         if (!keymap)
1177                 return NULL;
1178         
1179         /* first user defined keymaps */
1180         km = WM_keymap_list_find(&wm->userconf->keymaps, keymap->idname, keymap->spaceid, keymap->regionid);
1181
1182         if (km)
1183                 return km;
1184
1185         return keymap;
1186 }
1187
1188 /******************************* Keymap Editor ********************************
1189  * In the keymap editor the user key configuration is edited. */
1190
1191 void WM_keymap_restore_item_to_default(bContext *C, wmKeyMap *keymap, wmKeyMapItem *kmi)
1192 {
1193         wmWindowManager *wm = CTX_wm_manager(C);
1194         wmKeyMap *defaultmap, *addonmap;
1195         wmKeyMapItem *orig;
1196
1197         if (!keymap)
1198                 return;
1199
1200         /* construct default keymap from preset + addons */
1201         defaultmap = wm_keymap_preset(wm, keymap);
1202         addonmap = WM_keymap_list_find(&wm->addonconf->keymaps, keymap->idname, keymap->spaceid, keymap->regionid);
1203
1204         if (addonmap) {
1205                 defaultmap = wm_keymap_copy(defaultmap);
1206                 wm_keymap_addon_add(defaultmap, addonmap);
1207         }
1208
1209         /* find original item */
1210         orig = WM_keymap_item_find_id(defaultmap, kmi->id);
1211
1212         if (orig) {
1213                 /* restore to original */
1214                 if (strcmp(orig->idname, kmi->idname) != 0) {
1215                         BLI_strncpy(kmi->idname, orig->idname, sizeof(kmi->idname));
1216                         WM_keymap_properties_reset(kmi, NULL);
1217                 }
1218
1219                 if (orig->properties) {
1220                         if (kmi->properties) {
1221                                 IDP_FreeProperty(kmi->properties);
1222                                 MEM_freeN(kmi->properties);
1223                                 kmi->properties = NULL;
1224                         }
1225
1226                         kmi->properties = IDP_CopyProperty(orig->properties);
1227                         kmi->ptr->data = kmi->properties;
1228                 }
1229
1230                 kmi->propvalue = orig->propvalue;
1231                 kmi->type = orig->type;
1232                 kmi->val = orig->val;
1233                 kmi->shift = orig->shift;
1234                 kmi->ctrl = orig->ctrl;
1235                 kmi->alt = orig->alt;
1236                 kmi->oskey = orig->oskey;
1237                 kmi->keymodifier = orig->keymodifier;
1238                 kmi->maptype = orig->maptype;
1239
1240                 WM_keyconfig_update_tag(keymap, kmi);
1241         }
1242
1243         /* free temporary keymap */
1244         if (addonmap) {
1245                 WM_keymap_free(defaultmap);
1246                 MEM_freeN(defaultmap);
1247         }
1248 }
1249
1250 void WM_keymap_restore_to_default(wmKeyMap *keymap, bContext *C)
1251 {
1252         wmWindowManager *wm = CTX_wm_manager(C);
1253         wmKeyMap *usermap;
1254
1255         /* remove keymap from U.user_keymaps and update */
1256         usermap = WM_keymap_list_find(&U.user_keymaps, keymap->idname, keymap->spaceid, keymap->regionid);
1257
1258         if (usermap) {
1259                 WM_keymap_free(usermap);
1260                 BLI_freelinkN(&U.user_keymaps, usermap);
1261
1262                 WM_keyconfig_update_tag(NULL, NULL);
1263                 WM_keyconfig_update(wm);
1264         }
1265 }
1266
1267 wmKeyMapItem *WM_keymap_item_find_id(wmKeyMap *keymap, int id)
1268 {
1269         wmKeyMapItem *kmi;
1270         
1271         for (kmi = keymap->items.first; kmi; kmi = kmi->next) {
1272                 if (kmi->id == id) {
1273                         return kmi;
1274                 }
1275         }
1276         
1277         return NULL;
1278 }
1279
1280 /* Guess an appropriate keymap from the operator name */
1281 /* Needs to be kept up to date with Keymap and Operator naming */
1282 wmKeyMap *WM_keymap_guess_opname(const bContext *C, const char *opname)
1283 {
1284         wmKeyMap *km = NULL;
1285         SpaceLink *sl = CTX_wm_space_data(C);
1286         
1287         /* Window */
1288         if (strstr(opname, "WM_OT")) {
1289                 km = WM_keymap_find_all(C, "Window", 0, 0);
1290         }
1291         /* Screen */
1292         else if (strstr(opname, "SCREEN_OT")) {
1293                 km = WM_keymap_find_all(C, "Screen", 0, 0);
1294         }
1295         /* Grease Pencil */
1296         else if (strstr(opname, "GPENCIL_OT")) {
1297                 km = WM_keymap_find_all(C, "Grease Pencil", 0, 0);
1298         }
1299         /* Markers */
1300         else if (strstr(opname, "MARKER_OT")) {
1301                 km = WM_keymap_find_all(C, "Markers", 0, 0);
1302         }
1303         /* Import/Export*/
1304         else if (strstr(opname, "IMPORT_") || strstr(opname, "EXPORT_")) {
1305                 km = WM_keymap_find_all(C, "Window", 0, 0);
1306         }
1307         
1308         
1309         /* 3D View */
1310         else if (strstr(opname, "VIEW3D_OT")) {
1311                 km = WM_keymap_find_all(C, "3D View", sl->spacetype, 0);
1312         }
1313         else if (strstr(opname, "OBJECT_OT")) {
1314                 km = WM_keymap_find_all(C, "Object Mode", 0, 0);
1315         }
1316
1317         
1318         /* Editing Modes */
1319         else if (strstr(opname, "MESH_OT")) {
1320                 km = WM_keymap_find_all(C, "Mesh", 0, 0);
1321                 
1322                 /* some mesh operators are active in object mode too, like add-prim */
1323                 if (km && km->poll && km->poll((bContext *)C) == 0) {
1324                         km = WM_keymap_find_all(C, "Object Mode", 0, 0);
1325                 }
1326         }
1327         else if (strstr(opname, "CURVE_OT")) {
1328                 km = WM_keymap_find_all(C, "Curve", 0, 0);
1329                 
1330                 /* some curve operators are active in object mode too, like add-prim */
1331                 if (km && km->poll && km->poll((bContext *)C) == 0) {
1332                         km = WM_keymap_find_all(C, "Object Mode", 0, 0);
1333                 }
1334         }
1335         else if (strstr(opname, "ARMATURE_OT")) {
1336                 km = WM_keymap_find_all(C, "Armature", 0, 0);
1337         }
1338         else if (strstr(opname, "POSE_OT")) {
1339                 km = WM_keymap_find_all(C, "Pose", 0, 0);
1340         }
1341         else if (strstr(opname, "SCULPT_OT")) {
1342                 switch (CTX_data_mode_enum(C)) {
1343                         case OB_MODE_SCULPT:
1344                                 km = WM_keymap_find_all(C, "Sculpt", 0, 0);
1345                                 break;
1346                         case OB_MODE_EDIT:
1347                                 km = WM_keymap_find_all(C, "UV Sculpt", 0, 0);
1348                                 break;
1349                 }
1350         }
1351         else if (strstr(opname, "MBALL_OT")) {
1352                 km = WM_keymap_find_all(C, "Metaball", 0, 0);
1353                 
1354                 /* some mball operators are active in object mode too, like add-prim */
1355                 if (km && km->poll && km->poll((bContext *)C) == 0) {
1356                         km = WM_keymap_find_all(C, "Object Mode", 0, 0);
1357                 }
1358         }
1359         else if (strstr(opname, "LATTICE_OT")) {
1360                 km = WM_keymap_find_all(C, "Lattice", 0, 0);
1361         }
1362         else if (strstr(opname, "PARTICLE_OT")) {
1363                 km = WM_keymap_find_all(C, "Particle", 0, 0);
1364         }
1365         else if (strstr(opname, "FONT_OT")) {
1366                 km = WM_keymap_find_all(C, "Font", 0, 0);
1367         }
1368         else if (strstr(opname, "PAINT_OT")) {
1369                 
1370                 /* check for relevant mode */
1371                 switch (CTX_data_mode_enum(C)) {
1372                         case OB_MODE_WEIGHT_PAINT:
1373                                 km = WM_keymap_find_all(C, "Weight Paint", 0, 0);
1374                                 break;
1375                         case OB_MODE_VERTEX_PAINT:
1376                                 km = WM_keymap_find_all(C, "Vertex Paint", 0, 0);
1377                                 break;
1378                         case OB_MODE_TEXTURE_PAINT:
1379                                 km = WM_keymap_find_all(C, "Image Paint", 0, 0);
1380                                 break;
1381                 }
1382         }
1383         /* Paint Face Mask */
1384         else if (strstr(opname, "PAINT_OT_face_select")) {
1385                 km = WM_keymap_find_all(C, "Face Mask", sl->spacetype, 0);
1386         }
1387         /* Timeline */
1388         else if (strstr(opname, "TIME_OT")) {
1389                 km = WM_keymap_find_all(C, "Timeline", sl->spacetype, 0);
1390         }
1391         /* Image Editor */
1392         else if (strstr(opname, "IMAGE_OT")) {
1393                 km = WM_keymap_find_all(C, "Image", sl->spacetype, 0);
1394         }
1395         /* UV Editor */
1396         else if (strstr(opname, "UV_OT")) {
1397                 km = WM_keymap_find_all(C, "UV Editor", sl->spacetype, 0);
1398         }
1399         /* Node Editor */
1400         else if (strstr(opname, "NODE_OT")) {
1401                 km = WM_keymap_find_all(C, "Node Editor", sl->spacetype, 0);
1402         }
1403         /* Animation Editor Channels */
1404         else if (strstr(opname, "ANIM_OT_channels")) {
1405                 km = WM_keymap_find_all(C, "Animation Channels", sl->spacetype, 0);
1406         }
1407         /* Animation Generic - after channels */
1408         else if (strstr(opname, "ANIM_OT")) {
1409                 km = WM_keymap_find_all(C, "Animation", 0, 0);
1410         }
1411         /* Graph Editor */
1412         else if (strstr(opname, "GRAPH_OT")) {
1413                 km = WM_keymap_find_all(C, "Graph Editor", sl->spacetype, 0);
1414         }
1415         /* Dopesheet Editor */
1416         else if (strstr(opname, "ACTION_OT")) {
1417                 km = WM_keymap_find_all(C, "Dopesheet", sl->spacetype, 0);
1418         }
1419         /* NLA Editor */
1420         else if (strstr(opname, "NLA_OT")) {
1421                 km = WM_keymap_find_all(C, "NLA Editor", sl->spacetype, 0);
1422         }
1423         /* Script */
1424         else if (strstr(opname, "SCRIPT_OT")) {
1425                 km = WM_keymap_find_all(C, "Script", sl->spacetype, 0);
1426         }
1427         /* Text */
1428         else if (strstr(opname, "TEXT_OT")) {
1429                 km = WM_keymap_find_all(C, "Text", sl->spacetype, 0);
1430         }
1431         /* Sequencer */
1432         else if (strstr(opname, "SEQUENCER_OT")) {
1433                 km = WM_keymap_find_all(C, "Sequencer", sl->spacetype, 0);
1434         }
1435         /* Console */
1436         else if (strstr(opname, "CONSOLE_OT")) {
1437                 km = WM_keymap_find_all(C, "Console", sl->spacetype, 0);
1438         }
1439         /* Console */
1440         else if (strstr(opname, "INFO_OT")) {
1441                 km = WM_keymap_find_all(C, "Info", sl->spacetype, 0);
1442         }
1443         
1444         /* Transform */
1445         else if (strstr(opname, "TRANSFORM_OT")) {
1446                 
1447                 /* check for relevant editor */
1448                 switch (sl->spacetype) {
1449                         case SPACE_VIEW3D:
1450                                 km = WM_keymap_find_all(C, "3D View", sl->spacetype, 0);
1451                                 break;
1452                         case SPACE_IPO:
1453                                 km = WM_keymap_find_all(C, "Graph Editor", sl->spacetype, 0);
1454                                 break;
1455                         case SPACE_ACTION:
1456                                 km = WM_keymap_find_all(C, "Dopesheet", sl->spacetype, 0);
1457                                 break;
1458                         case SPACE_NLA:
1459                                 km = WM_keymap_find_all(C, "NLA Editor", sl->spacetype, 0);
1460                                 break;
1461                         case SPACE_IMAGE:
1462                                 km = WM_keymap_find_all(C, "UV Editor", sl->spacetype, 0);
1463                                 break;
1464                         case SPACE_NODE:
1465                                 km = WM_keymap_find_all(C, "Node Editor", sl->spacetype, 0);
1466                                 break;
1467                         case SPACE_SEQ:
1468                                 km = WM_keymap_find_all(C, "Sequencer", sl->spacetype, 0);
1469                                 break;
1470                 }
1471         }
1472         
1473         return km;
1474 }
1475