début d'organisation du logging.
[minwii.git] / src / app / eventutils.py
1 # -*- coding: utf-8 -*-
2 """
3 Décorateur, métaclasse et classe mixin pour faciliter
4 l'implémentation des gestionnaires d'événements pygame.
5
6 Utilisation :
7 - hériter de EventHandlerMixin
8 - décorer les méthodes gestionnaires d'événements avec le décorateur 'event_handler'
9 en passant en paramètre un ou plusieurs codes d'événement pygame
10
11 Il n'existe aucune contrainte sur le nommage de la méthode décorée.
12 La méthode gestionnaire d'événement reçoit l'événement pygame comme unique paramètre.
13
14 par exemple :
15
16 import pygame
17 class Machin(pygame.sprite.Sprite, EventHandlerMixin) :
18
19 # .../...
20
21 @event_handler(pygame.KEYDOWN)
22 def handleKeyDown(self, event) :
23 pass
24
25
26 $Id$
27 $URL$
28 """
29 import types
30 import pygame
31 from StringIO import StringIO
32 from log import console, eventLog, hiddenEvents, pprinters
33
34 class _EventDispatcher :
35 def __init__(self) :
36 self.registry = {}
37
38 def addEventListener(self, eventType, listener) :
39 if self.registry.has_key(eventType) :
40 self.registry[eventType][listener] = True
41 else :
42 self.registry[eventType] = {listener:True}
43
44 def removeEventListener(self, eventType, listener) :
45 try :
46 del self.registry[eventType][listener]
47 except KeyError :
48 console.warn("no listener to remove")
49
50 def dispatchEvents(self) :
51 events = pygame.event.get()
52 for event in events :
53 listeners = self.registry.get(event.type, {})
54 if not hiddenEvents.has_key(event.type):
55 #if listeners and not hiddenEvents.has_key(event.type) :
56 eventLog.info(pprinters[event.type](event))
57 for listener in listeners.keys() :
58 listener(event)
59
60 def reset(self) :
61 self.registry = {}
62
63 def __repr__(self) :
64 out = StringIO()
65 keys = self.registry.keys()
66 keys.sort()
67 for k in keys :
68 print >> out, "event", k
69 for listener in self.registry[k].keys() :
70 print >> out, listener.__name__
71 out.seek(0)
72 return out.read()
73
74
75 EventDispatcher = _EventDispatcher()
76
77 def event_handler(*eventTypes) :
78 def markFunctionAsListener(m) :
79 m.__islistener__ = True
80 m.__eventtypes__ = eventTypes
81 return m
82 return markFunctionAsListener
83
84
85 class EventInitializer(type):
86
87 def __init__(cls, name, bases, dict) :
88 def init_listeners(self) :
89 for k, v in dict.items() :
90 if isinstance(v, types.FunctionType) and hasattr(v, '__islistener__') :
91 listener = getattr(self, k)
92 for eventType in v.__eventtypes__ :
93 EventDispatcher.addEventListener(eventType, listener)
94
95 def ctor(self, *args, **kw) :
96 default_ctor = dict.get('__init__')
97 if not default_ctor :
98 super(cls, self).__init__(*args, **kw)
99 else :
100 default_ctor(self, *args, **kw)
101 init_listeners(self)
102
103 cls.__init__ = ctor
104
105
106 class EventHandlerMixin(object) :
107 __metaclass__ = EventInitializer