annotate pyntnclick/widgets/base.py @ 717:3b2d1adca59c pyntnclick

Pull a bunch of draw logic out of rect_drawer and over to the widgets
author Neil Muller <neil@dip.sun.ac.za>
date Sun, 05 Aug 2012 21:50:34 +0200
parents e9265818a96c
children efa58c92b304
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
1 import collections
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
2
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
3 import pygame
559
b7d8b89de71a Rip out keyboard handling
Stefano Rivera <stefano@rivera.za.net>
parents: 558
diff changeset
4 from pygame.locals import (MOUSEBUTTONDOWN, MOUSEBUTTONUP,
717
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
5 MOUSEMOTION, SRCALPHA, USEREVENT,
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
6 BLEND_RGBA_MIN)
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
7
558
f9f04cb35697 s/mamba/pyntnclick/
Stefano Rivera <stefano@rivera.za.net>
parents: 555
diff changeset
8 from pyntnclick.engine import UserEvent
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
9
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
10
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
11 class Widget(object):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
12
692
d6ded808cc33 Much scene management refactoring.
Jeremy Thurgood <firxen@gmail.com>
parents: 682
diff changeset
13 highlight_cursor = False
d6ded808cc33 Much scene management refactoring.
Jeremy Thurgood <firxen@gmail.com>
parents: 682
diff changeset
14
639
fdc63049b08c Add game_description to the widgets,, since it looks a useful thing to do
Neil Muller <neil@dip.sun.ac.za>
parents: 602
diff changeset
15 def __init__(self, rect, gd):
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
16 if not isinstance(rect, pygame.Rect):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
17 rect = pygame.Rect(rect, (0, 0))
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
18 self.rect = rect
639
fdc63049b08c Add game_description to the widgets,, since it looks a useful thing to do
Neil Muller <neil@dip.sun.ac.za>
parents: 602
diff changeset
19 self.gd = gd
fdc63049b08c Add game_description to the widgets,, since it looks a useful thing to do
Neil Muller <neil@dip.sun.ac.za>
parents: 602
diff changeset
20 self.resource = gd.resource
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
21 self.modal = False
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
22 self.parent = None
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
23 self.disabled = False
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
24 self.callbacks = collections.defaultdict(list)
692
d6ded808cc33 Much scene management refactoring.
Jeremy Thurgood <firxen@gmail.com>
parents: 682
diff changeset
25 # To track which widget the mouse is over
d6ded808cc33 Much scene management refactoring.
Jeremy Thurgood <firxen@gmail.com>
parents: 682
diff changeset
26 self.mouseover_widget = self
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
27
694
58d78038a197 Better modal wrapping.
Jeremy Thurgood <firxen@gmail.com>
parents: 693
diff changeset
28 def set_parent(self, parent):
58d78038a197 Better modal wrapping.
Jeremy Thurgood <firxen@gmail.com>
parents: 693
diff changeset
29 self.parent = parent
58d78038a197 Better modal wrapping.
Jeremy Thurgood <firxen@gmail.com>
parents: 693
diff changeset
30
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
31 def add_callback(self, eventtype, callback, *args):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
32 self.callbacks[eventtype].append((callback, args))
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
33
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
34 def event(self, ev):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
35 "Don't override this without damn good reason"
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
36 if self.disabled:
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
37 return True
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
38
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
39 type_ = ev.type
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
40 if type_ == USEREVENT:
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
41 for k in self.callbacks.iterkeys():
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
42 if (isinstance(k, type) and issubclass(k, UserEvent)
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
43 and k.matches(ev)):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
44 type_ = k
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
45 break
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
46
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
47 for callback, args in self.callbacks[type_]:
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
48 if callback(ev, self, *args):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
49 return True
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
50 return False
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
51
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
52 def draw(self, surface):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
53 "Override me"
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
54 pass
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
55
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
56 def disable(self):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
57 if not self.disabled:
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
58 self.disabled = True
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
59 if hasattr(self, 'prepare'):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
60 self.prepare()
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
61
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
62 def enable(self):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
63 if self.disabled:
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
64 self.disabled = False
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
65 if hasattr(self, 'prepare'):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
66 self.prepare()
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
67
652
Jeremy Thurgood <firxen@gmail.com>
parents: 639
diff changeset
68 def global_to_local(self, pos):
Jeremy Thurgood <firxen@gmail.com>
parents: 639
diff changeset
69 x, y = pos
Jeremy Thurgood <firxen@gmail.com>
parents: 639
diff changeset
70 return (x - self.rect.left, y - self.rect.top)
Jeremy Thurgood <firxen@gmail.com>
parents: 639
diff changeset
71
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
72
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
73 class Button(Widget):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
74
692
d6ded808cc33 Much scene management refactoring.
Jeremy Thurgood <firxen@gmail.com>
parents: 682
diff changeset
75 highlight_cursor = True
d6ded808cc33 Much scene management refactoring.
Jeremy Thurgood <firxen@gmail.com>
parents: 682
diff changeset
76
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
77 def event(self, ev):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
78 if super(Button, self).event(ev):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
79 return True
559
b7d8b89de71a Rip out keyboard handling
Stefano Rivera <stefano@rivera.za.net>
parents: 558
diff changeset
80 if ev.type == MOUSEBUTTONDOWN:
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
81 for callback, args in self.callbacks['clicked']:
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
82 if callback(ev, self, *args):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
83 return True
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
84 return False
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
85
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
86 def forced_click(self):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
87 """Force calling the clicked handler"""
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
88 for callback, args in self.callbacks['clicked']:
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
89 if callback(None, self, *args):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
90 return True
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
91 return False
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
92
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
93
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
94 class Container(Widget):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
95
639
fdc63049b08c Add game_description to the widgets,, since it looks a useful thing to do
Neil Muller <neil@dip.sun.ac.za>
parents: 602
diff changeset
96 def __init__(self, rect, gd):
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
97 if rect is None:
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
98 rect = pygame.Rect(0, 0, 0, 0)
639
fdc63049b08c Add game_description to the widgets,, since it looks a useful thing to do
Neil Muller <neil@dip.sun.ac.za>
parents: 602
diff changeset
99 super(Container, self).__init__(rect, gd)
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
100 self.children = []
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
101
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
102 def event(self, ev):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
103 """Push an event down through the tree, and fire our own event as a
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
104 last resort
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
105 """
692
d6ded808cc33 Much scene management refactoring.
Jeremy Thurgood <firxen@gmail.com>
parents: 682
diff changeset
106 self.mouseover_widget = self
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
107 if ev.type in (MOUSEMOTION, MOUSEBUTTONUP, MOUSEBUTTONDOWN):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
108 for child in self.children[:]:
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
109 if child.rect.collidepoint(ev.pos):
692
d6ded808cc33 Much scene management refactoring.
Jeremy Thurgood <firxen@gmail.com>
parents: 682
diff changeset
110 result = child.event(ev)
d6ded808cc33 Much scene management refactoring.
Jeremy Thurgood <firxen@gmail.com>
parents: 682
diff changeset
111 self.mouseover_widget = child.mouseover_widget
d6ded808cc33 Much scene management refactoring.
Jeremy Thurgood <firxen@gmail.com>
parents: 682
diff changeset
112 if result:
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
113 return True
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
114
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
115 else:
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
116 # Other events go to all children first
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
117 for child in self.children[:]:
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
118 if child.event(ev):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
119 return True
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
120 if super(Container, self).event(ev):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
121 return True
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
122
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
123 def add(self, widget):
694
58d78038a197 Better modal wrapping.
Jeremy Thurgood <firxen@gmail.com>
parents: 693
diff changeset
124 widget.set_parent(self)
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
125 self.children.append(widget)
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
126 self.rect = self.rect.union(widget.rect)
672
55d5d384fc16 Widget-based inventory view.
Jeremy Thurgood <firxen@gmail.com>
parents: 670
diff changeset
127 return widget
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
128
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
129 def remove(self, widget):
694
58d78038a197 Better modal wrapping.
Jeremy Thurgood <firxen@gmail.com>
parents: 693
diff changeset
130 widget.set_parent(None)
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
131 self.children.remove(widget)
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
132
682
0749438e325e Unified Scene widgetry.
Jeremy Thurgood <firxen@gmail.com>
parents: 679
diff changeset
133 def remove_all(self):
0749438e325e Unified Scene widgetry.
Jeremy Thurgood <firxen@gmail.com>
parents: 679
diff changeset
134 for widget in reversed(self.children[:]):
0749438e325e Unified Scene widgetry.
Jeremy Thurgood <firxen@gmail.com>
parents: 679
diff changeset
135 self.remove(widget)
0749438e325e Unified Scene widgetry.
Jeremy Thurgood <firxen@gmail.com>
parents: 679
diff changeset
136
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
137 def draw(self, surface):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
138 for child in self.children:
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
139 child.draw(surface)
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
140
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
141
678
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
142 class ModalStackContainer(Container):
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
143
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
144 def __init__(self, rect, gd, obscure_color=None):
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
145 super(ModalStackContainer, self).__init__(rect, gd)
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
146 if obscure_color is None:
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
147 obscure_color = gd.constants.modal_obscure_color
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
148 self.obscure_color = convert_color(obscure_color)
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
149
693
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
150 @property
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
151 def top(self):
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
152 if self.children:
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
153 return self.children[-1]
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
154 return None
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
155
678
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
156 def event(self, ev):
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
157 """Only the topmost child gets events.
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
158 """
693
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
159 self.mouseover_widget = self
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
160 if self.top:
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
161 self.mouseover_widget = self.top.mouseover_widget
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
162 if self.top.event(ev):
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
163 return True
678
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
164
693
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
165 # We skip Container's event() method and hop straight to its parent's.
678
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
166 if super(Container, self).event(ev):
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
167 return True
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
168
679
fa168b5e2624 The return of the message dialog (and there was much rejoicing)
Neil Muller <neil@dip.sun.ac.za>
parents: 678
diff changeset
169 def is_top(self, widget):
693
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
170 return self.top is widget
679
fa168b5e2624 The return of the message dialog (and there was much rejoicing)
Neil Muller <neil@dip.sun.ac.za>
parents: 678
diff changeset
171
678
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
172 def draw(self, surface):
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
173 obscure = pygame.Surface(self.rect.size, SRCALPHA)
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
174 obscure.fill(self.obscure_color)
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
175 for child in self.children:
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
176 surface.blit(obscure, self.rect)
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
177 child.draw(surface)
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
178
36d7f7e9650e Modal magic. (And yes, it is quite magical.)
Jeremy Thurgood <firxen@gmail.com>
parents: 672
diff changeset
179
555
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
180 class GridContainer(Container):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
181 """Hacky container that only supports grids, won't work with Container
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
182 children, or modal children.
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
183 """
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
184
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
185 def __init__(self, width, rect=None):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
186 super(GridContainer, self).__init__(rect)
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
187 self.width = width
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
188
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
189 def add(self, widget):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
190 assert not isinstance(widget, Container)
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
191 assert not widget.modal
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
192 super(GridContainer, self).add(widget)
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
193
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
194
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
195 class Box(Container):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
196 """A container that draws a filled background with a border"""
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
197 padding = 4
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
198
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
199 def draw(self, surface):
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
200 expandrect = self.rect.move((-self.padding, -self.padding))
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
201 expandrect.width = self.rect.width + 2 * self.padding
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
202 expandrect.height = self.rect.height + 2 * self.padding
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
203 border = pygame.Surface(expandrect.size, SRCALPHA)
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
204 border.fill(pygame.Color('black'))
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
205 surface.blit(border, expandrect)
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
206 background = pygame.Surface(self.rect.size, SRCALPHA)
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
207 background.fill(pygame.Color('gray'))
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
208 surface.blit(background, self.rect)
c0474fe18b96 Copy in widgets from mamba (currently unused)
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
209 super(Box, self).draw(surface)
670
fe7023750b20 Add convert_color function
Stefano Rivera <stefano@rivera.za.net>
parents: 652
diff changeset
210
fe7023750b20 Add convert_color function
Stefano Rivera <stefano@rivera.za.net>
parents: 652
diff changeset
211
fe7023750b20 Add convert_color function
Stefano Rivera <stefano@rivera.za.net>
parents: 652
diff changeset
212 def convert_color(color):
fe7023750b20 Add convert_color function
Stefano Rivera <stefano@rivera.za.net>
parents: 652
diff changeset
213 """Give me a pygame Color, dammit"""
fe7023750b20 Add convert_color function
Stefano Rivera <stefano@rivera.za.net>
parents: 652
diff changeset
214 if isinstance(color, pygame.Color):
fe7023750b20 Add convert_color function
Stefano Rivera <stefano@rivera.za.net>
parents: 652
diff changeset
215 return color
fe7023750b20 Add convert_color function
Stefano Rivera <stefano@rivera.za.net>
parents: 652
diff changeset
216 if isinstance(color, basestring):
fe7023750b20 Add convert_color function
Stefano Rivera <stefano@rivera.za.net>
parents: 652
diff changeset
217 return pygame.Color(color)
fe7023750b20 Add convert_color function
Stefano Rivera <stefano@rivera.za.net>
parents: 652
diff changeset
218 return pygame.Color(*color)
693
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
219
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
220
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
221 class ModalWrapper(Container):
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
222 "A wrapper around a widget that removes itself when a mouse click occurs"
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
223
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
224 def __init__(self, widget, close_callback=None):
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
225 super(ModalWrapper, self).__init__(widget.rect, widget.gd)
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
226 self.close_callback = close_callback
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
227 self.add(widget)
694
58d78038a197 Better modal wrapping.
Jeremy Thurgood <firxen@gmail.com>
parents: 693
diff changeset
228 self.add_callback(MOUSEBUTTONDOWN, self.close)
693
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
229 widget.add_callback(MOUSEBUTTONDOWN, self.close)
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
230
694
58d78038a197 Better modal wrapping.
Jeremy Thurgood <firxen@gmail.com>
parents: 693
diff changeset
231 def set_parent(self, parent):
58d78038a197 Better modal wrapping.
Jeremy Thurgood <firxen@gmail.com>
parents: 693
diff changeset
232 super(ModalWrapper, self).set_parent(parent)
58d78038a197 Better modal wrapping.
Jeremy Thurgood <firxen@gmail.com>
parents: 693
diff changeset
233 if parent:
58d78038a197 Better modal wrapping.
Jeremy Thurgood <firxen@gmail.com>
parents: 693
diff changeset
234 self.rect = self.parent.rect
58d78038a197 Better modal wrapping.
Jeremy Thurgood <firxen@gmail.com>
parents: 693
diff changeset
235
693
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
236 def close(self, ev, widget):
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
237 if self.parent:
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
238 self.parent.remove(self)
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
239 if self.close_callback:
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
240 self.close_callback()
c8b683dd56d3 Better modal message and result handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 692
diff changeset
241 return True
711
e9265818a96c Add image widget for rect_drawer
Neil Muller <neil@dip.sun.ac.za>
parents: 694
diff changeset
242
e9265818a96c Add image widget for rect_drawer
Neil Muller <neil@dip.sun.ac.za>
parents: 694
diff changeset
243
e9265818a96c Add image widget for rect_drawer
Neil Muller <neil@dip.sun.ac.za>
parents: 694
diff changeset
244 class Image(Widget):
e9265818a96c Add image widget for rect_drawer
Neil Muller <neil@dip.sun.ac.za>
parents: 694
diff changeset
245 """Basic widget that draws an image, with an associated rect"""
e9265818a96c Add image widget for rect_drawer
Neil Muller <neil@dip.sun.ac.za>
parents: 694
diff changeset
246
e9265818a96c Add image widget for rect_drawer
Neil Muller <neil@dip.sun.ac.za>
parents: 694
diff changeset
247 def __init__(self, rect, gd, image):
e9265818a96c Add image widget for rect_drawer
Neil Muller <neil@dip.sun.ac.za>
parents: 694
diff changeset
248 super(Image, self).__init__(rect, gd)
e9265818a96c Add image widget for rect_drawer
Neil Muller <neil@dip.sun.ac.za>
parents: 694
diff changeset
249 self.image = image
e9265818a96c Add image widget for rect_drawer
Neil Muller <neil@dip.sun.ac.za>
parents: 694
diff changeset
250 self.rect.width = image.get_rect().width
e9265818a96c Add image widget for rect_drawer
Neil Muller <neil@dip.sun.ac.za>
parents: 694
diff changeset
251 self.rect.height = image.get_rect().height
e9265818a96c Add image widget for rect_drawer
Neil Muller <neil@dip.sun.ac.za>
parents: 694
diff changeset
252 self.visible = True
e9265818a96c Add image widget for rect_drawer
Neil Muller <neil@dip.sun.ac.za>
parents: 694
diff changeset
253
e9265818a96c Add image widget for rect_drawer
Neil Muller <neil@dip.sun.ac.za>
parents: 694
diff changeset
254 def draw(self, surface):
e9265818a96c Add image widget for rect_drawer
Neil Muller <neil@dip.sun.ac.za>
parents: 694
diff changeset
255 if self.visible:
e9265818a96c Add image widget for rect_drawer
Neil Muller <neil@dip.sun.ac.za>
parents: 694
diff changeset
256 surface.blit(self.image, self.rect)
717
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
257
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
258
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
259 class TranslucentImage(Image):
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
260 """Image that can also be translucent"""
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
261
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
262 def __init__(self, rect, gd, image):
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
263 super(TranslucentImage, self).__init__(rect, gd, image)
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
264 self.translucent = False
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
265 surf = pygame.surface.Surface((self.rect.width, self.rect.height),
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
266 SRCALPHA).convert_alpha()
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
267 surf.fill(pygame.color.Color(255, 255, 255, 96))
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
268 surf.blit(self.image, (0, 0), None, BLEND_RGBA_MIN)
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
269 self.trans_image = surf
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
270
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
271 def draw(self, surface):
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
272 if self.visible:
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
273 if self.translucent:
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
274 surface.blit(self.trans_image, self.rect)
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
275 else:
3b2d1adca59c Pull a bunch of draw logic out of rect_drawer and over to the widgets
Neil Muller <neil@dip.sun.ac.za>
parents: 711
diff changeset
276 surface.blit(self.image, self.rect)