Mercurial > mamba

annotate mamba/widgets/base.py @ 137:7fbeeb402685

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
PEP8 tidy mamba.widgets
author Stefano Rivera <stefano@rivera.za.net>
date Mon, 12 Sep 2011 13:09:09 +0200
parents d5aa5f805f00
children 79fdae806ca5
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
55
671d027f1b71 Callbacks for events in widgets
Stefano Rivera <stefano@rivera.za.net>
parents: 43
diff changeset
1 import collections
671d027f1b71 Callbacks for events in widgets
Stefano Rivera <stefano@rivera.za.net>
parents: 43
diff changeset
2
23
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
3 import pygame
115
d5aa5f805f00 Add Button class that buttons inherit from. It implements a 'clicked' callback
Stefano Rivera <stefano@rivera.za.net>
parents: 109
diff changeset
4 from pygame.constants import K_UP, K_DOWN, K_RETURN
58
7fa000c4b3c7 Dispatch key events
Stefano Rivera <stefano@rivera.za.net>
parents: 55
diff changeset
5 from pygame.locals import MOUSEMOTION, MOUSEBUTTONUP, MOUSEBUTTONDOWN, KEYDOWN
23
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
6
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
7
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
8 class Widget(object):
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
9
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
10 def __init__(self, rect):
38
047273a63054 Main menu says hello
Stefano Rivera <stefano@rivera.za.net>
parents: 24
diff changeset
11 if not isinstance(rect, pygame.Rect):
41
3f44c30a1c39 Correct Text placement
Stefano Rivera <stefano@rivera.za.net>
parents: 38
diff changeset
12 rect = pygame.Rect(rect, (0, 0))
3f44c30a1c39 Correct Text placement
Stefano Rivera <stefano@rivera.za.net>
parents: 38
diff changeset
13 self.rect = rect
58
7fa000c4b3c7 Dispatch key events
Stefano Rivera <stefano@rivera.za.net>
parents: 55
diff changeset
14 self.focussable = False
77
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
15 self.focussed = False
106
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
16 self.parent = None
55
671d027f1b71 Callbacks for events in widgets
Stefano Rivera <stefano@rivera.za.net>
parents: 43
diff changeset
17 self.callbacks = collections.defaultdict(list)
671d027f1b71 Callbacks for events in widgets
Stefano Rivera <stefano@rivera.za.net>
parents: 43
diff changeset
18
671d027f1b71 Callbacks for events in widgets
Stefano Rivera <stefano@rivera.za.net>
parents: 43
diff changeset
19 def add_callback(self, eventtype, callback, *args):
671d027f1b71 Callbacks for events in widgets
Stefano Rivera <stefano@rivera.za.net>
parents: 43
diff changeset
20 self.callbacks[eventtype].append((callback, args))
23
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
21
59
3cc917814579 Container subclasses Widget
Stefano Rivera <stefano@rivera.za.net>
parents: 58
diff changeset
22 def event(self, ev):
3cc917814579 Container subclasses Widget
Stefano Rivera <stefano@rivera.za.net>
parents: 58
diff changeset
23 for callback, args in self.callbacks[ev.type]:
77
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
24 if callback(ev, self, *args):
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
25 return True
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
26 return False
23
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
27
24
30d4f3e62bcf Refactor widgets *again*, add container to Habitat
Stefano Rivera <stefano@rivera.za.net>
parents: 23
diff changeset
28 def draw(self, surface):
23
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
29 "Override me"
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
30 pass
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
31
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
32
115
d5aa5f805f00 Add Button class that buttons inherit from. It implements a 'clicked' callback
Stefano Rivera <stefano@rivera.za.net>
parents: 109
diff changeset
33 class Button(Widget):
137
7fbeeb402685 PEP8 tidy mamba.widgets
Stefano Rivera <stefano@rivera.za.net>
parents: 115
diff changeset
34
115
d5aa5f805f00 Add Button class that buttons inherit from. It implements a 'clicked' callback
Stefano Rivera <stefano@rivera.za.net>
parents: 109
diff changeset
35 def event(self, ev):
d5aa5f805f00 Add Button class that buttons inherit from. It implements a 'clicked' callback
Stefano Rivera <stefano@rivera.za.net>
parents: 109
diff changeset
36 if super(Button, self).event(ev):
d5aa5f805f00 Add Button class that buttons inherit from. It implements a 'clicked' callback
Stefano Rivera <stefano@rivera.za.net>
parents: 109
diff changeset
37 return True
d5aa5f805f00 Add Button class that buttons inherit from. It implements a 'clicked' callback
Stefano Rivera <stefano@rivera.za.net>
parents: 109
diff changeset
38 if (ev.type == MOUSEBUTTONDOWN
d5aa5f805f00 Add Button class that buttons inherit from. It implements a 'clicked' callback
Stefano Rivera <stefano@rivera.za.net>
parents: 109
diff changeset
39 or (ev.type == KEYDOWN and ev.key == K_RETURN)):
d5aa5f805f00 Add Button class that buttons inherit from. It implements a 'clicked' callback
Stefano Rivera <stefano@rivera.za.net>
parents: 109
diff changeset
40 for callback, args in self.callbacks['clicked']:
d5aa5f805f00 Add Button class that buttons inherit from. It implements a 'clicked' callback
Stefano Rivera <stefano@rivera.za.net>
parents: 109
diff changeset
41 if callback(ev, self, *args):
d5aa5f805f00 Add Button class that buttons inherit from. It implements a 'clicked' callback
Stefano Rivera <stefano@rivera.za.net>
parents: 109
diff changeset
42 return True
d5aa5f805f00 Add Button class that buttons inherit from. It implements a 'clicked' callback
Stefano Rivera <stefano@rivera.za.net>
parents: 109
diff changeset
43 return False
d5aa5f805f00 Add Button class that buttons inherit from. It implements a 'clicked' callback
Stefano Rivera <stefano@rivera.za.net>
parents: 109
diff changeset
44
d5aa5f805f00 Add Button class that buttons inherit from. It implements a 'clicked' callback
Stefano Rivera <stefano@rivera.za.net>
parents: 109
diff changeset
45
59
3cc917814579 Container subclasses Widget
Stefano Rivera <stefano@rivera.za.net>
parents: 58
diff changeset
46 class Container(Widget):
23
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
47
106
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
48 def __init__(self, rect):
59
3cc917814579 Container subclasses Widget
Stefano Rivera <stefano@rivera.za.net>
parents: 58
diff changeset
49 super(Container, self).__init__(rect)
23
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
50 self.children = []
77
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
51 self.focussed_child = None
23
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
52
59
3cc917814579 Container subclasses Widget
Stefano Rivera <stefano@rivera.za.net>
parents: 58
diff changeset
53 def event(self, ev):
77
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
54 """Push an event down through the tree, and fire our own event as a
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
55 last resort
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
56 """
59
3cc917814579 Container subclasses Widget
Stefano Rivera <stefano@rivera.za.net>
parents: 58
diff changeset
57 if ev.type in (MOUSEMOTION, MOUSEBUTTONUP, MOUSEBUTTONDOWN):
58
7fa000c4b3c7 Dispatch key events
Stefano Rivera <stefano@rivera.za.net>
parents: 55
diff changeset
58 for child in self.children:
59
3cc917814579 Container subclasses Widget
Stefano Rivera <stefano@rivera.za.net>
parents: 58
diff changeset
59 if child.rect.collidepoint(ev.pos):
109
dc0b0be7e2f6 Move mouse focussing to within Container
Stefano Rivera <stefano@rivera.za.net>
parents: 106
diff changeset
60 if ev.type == MOUSEBUTTONDOWN and child.focussable:
dc0b0be7e2f6 Move mouse focussing to within Container
Stefano Rivera <stefano@rivera.za.net>
parents: 106
diff changeset
61 root = self
dc0b0be7e2f6 Move mouse focussing to within Container
Stefano Rivera <stefano@rivera.za.net>
parents: 106
diff changeset
62 while root.parent is not None:
dc0b0be7e2f6 Move mouse focussing to within Container
Stefano Rivera <stefano@rivera.za.net>
parents: 106
diff changeset
63 root = root.parent
dc0b0be7e2f6 Move mouse focussing to within Container
Stefano Rivera <stefano@rivera.za.net>
parents: 106
diff changeset
64 root.defocus()
dc0b0be7e2f6 Move mouse focussing to within Container
Stefano Rivera <stefano@rivera.za.net>
parents: 106
diff changeset
65 widget = child
dc0b0be7e2f6 Move mouse focussing to within Container
Stefano Rivera <stefano@rivera.za.net>
parents: 106
diff changeset
66 while widget.parent is not None:
dc0b0be7e2f6 Move mouse focussing to within Container
Stefano Rivera <stefano@rivera.za.net>
parents: 106
diff changeset
67 parent = widget.parent
dc0b0be7e2f6 Move mouse focussing to within Container
Stefano Rivera <stefano@rivera.za.net>
parents: 106
diff changeset
68 if isinstance(parent, Container):
dc0b0be7e2f6 Move mouse focussing to within Container
Stefano Rivera <stefano@rivera.za.net>
parents: 106
diff changeset
69 idx = parent.children.index(widget)
dc0b0be7e2f6 Move mouse focussing to within Container
Stefano Rivera <stefano@rivera.za.net>
parents: 106
diff changeset
70 parent.focussed_child = idx
dc0b0be7e2f6 Move mouse focussing to within Container
Stefano Rivera <stefano@rivera.za.net>
parents: 106
diff changeset
71 widget = parent
dc0b0be7e2f6 Move mouse focussing to within Container
Stefano Rivera <stefano@rivera.za.net>
parents: 106
diff changeset
72 child.focussed = True
dc0b0be7e2f6 Move mouse focussing to within Container
Stefano Rivera <stefano@rivera.za.net>
parents: 106
diff changeset
73
77
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
74 if child.event(ev):
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
75 return True
109
dc0b0be7e2f6 Move mouse focussing to within Container
Stefano Rivera <stefano@rivera.za.net>
parents: 106
diff changeset
76
59
3cc917814579 Container subclasses Widget
Stefano Rivera <stefano@rivera.za.net>
parents: 58
diff changeset
77 if ev.type == KEYDOWN:
58
7fa000c4b3c7 Dispatch key events
Stefano Rivera <stefano@rivera.za.net>
parents: 55
diff changeset
78 for child in self.children:
77
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
79 if child.focussed:
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
80 if child.event(ev):
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
81 return True
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
82 if super(Container, self).event(ev):
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
83 return True
106
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
84 if (self.parent is None and ev.type == KEYDOWN
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
85 and ev.key in (K_UP, K_DOWN)):
97
c65046b5bafd eat focus adjusting key events
Stefano Rivera <stefano@rivera.za.net>
parents: 95
diff changeset
86 return self.adjust_focus(1 if ev.key == K_DOWN else -1)
23
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
87
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
88 def add(self, widget):
77
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
89 widget.parent = self
23
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
90 self.children.append(widget)
106
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
91 self.rect = self.rect.union(widget.rect)
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
92
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
93 def defocus(self):
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
94 if self.focussed_child is not None:
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
95 child = self.children[self.focussed_child]
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
96 if isinstance(child, Container):
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
97 child.defocus()
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
98 child.focussed = False
23
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
99
77
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
100 def adjust_focus(self, direction):
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
101 """Try and adjust focus in direction (integer)
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
102 """
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
103 if self.focussed_child is not None:
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
104 child = self.children[self.focussed_child]
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
105 if isinstance(child, Container):
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
106 if child.adjust_focus(direction):
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
107 return True
106
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
108 else:
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
109 child.focussed = False
77
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
110
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
111 current = self.focussed_child
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
112 if current is None:
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
113 current = -1 if direction > 0 else len(self.children)
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
114 if direction > 0:
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
115 possibles = list(enumerate(self.children))[current + 1:]
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
116 else:
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
117 possibles = list(enumerate(self.children))[:current]
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
118 possibles.reverse()
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
119 for i, child in possibles:
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
120 if child.focussable:
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
121 child.focussed = True
106
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
122 self.focussed_child = i
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
123 return True
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
124 if isinstance(child, Container):
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
125 if child.adjust_focus(direction):
77
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
126 self.focussed_child = i
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
127 return True
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
128 else:
106
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
129 if self.parent is None:
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
130 if self.focussed_child is not None:
99
2b6626f417f2 Handle root containers where nothing is focussable
Stefano Rivera <stefano@rivera.za.net>
parents: 97
diff changeset
131 # At the end, mark the last one as focussed, again
2b6626f417f2 Handle root containers where nothing is focussable
Stefano Rivera <stefano@rivera.za.net>
parents: 97
diff changeset
132 child = self.children[self.focussed_child]
2b6626f417f2 Handle root containers where nothing is focussable
Stefano Rivera <stefano@rivera.za.net>
parents: 97
diff changeset
133 if isinstance(child, Container):
2b6626f417f2 Handle root containers where nothing is focussable
Stefano Rivera <stefano@rivera.za.net>
parents: 97
diff changeset
134 if child.adjust_focus(-direction):
2b6626f417f2 Handle root containers where nothing is focussable
Stefano Rivera <stefano@rivera.za.net>
parents: 97
diff changeset
135 return True
106
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
136 else:
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
137 child.focussed = True
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
138 return True
95
7a17c5b74148 Correctly handle nested Containers. Set focus on root container at first draw
Stefano Rivera <stefano@rivera.za.net>
parents: 77
diff changeset
139 else:
7a17c5b74148 Correctly handle nested Containers. Set focus on root container at first draw
Stefano Rivera <stefano@rivera.za.net>
parents: 77
diff changeset
140 self.focussed_child = None
77
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
141 return False
Stefano Rivera <stefano@rivera.za.net>
parents: 59
diff changeset
142
23
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
143 def draw(self, surface):
106
d1152f2a0e13 Mouse affects focus
Stefano Rivera <stefano@rivera.za.net>
parents: 99
diff changeset
144 if self.parent is None and not self.focussed:
95
7a17c5b74148 Correctly handle nested Containers. Set focus on root container at first draw
Stefano Rivera <stefano@rivera.za.net>
parents: 77
diff changeset
145 self.focussed = True
7a17c5b74148 Correctly handle nested Containers. Set focus on root container at first draw
Stefano Rivera <stefano@rivera.za.net>
parents: 77
diff changeset
146 self.adjust_focus(1)
23
d0e8940df703 Widgets -> base
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
147 for child in self.children:
24
30d4f3e62bcf Refactor widgets *again*, add container to Habitat
Stefano Rivera <stefano@rivera.za.net>
parents: 23
diff changeset
148 child.draw(surface)