Mercurial > sypikslang
annotate gamelib/gui_base.py @ 69:626b1eaec13c
fucking pep8
| author | Rizmari Versfeld <rizziepit@gmail.com> |
|---|---|
| date | Tue, 08 May 2012 23:08:51 +0200 |
| parents | 7309392d9ca9 |
| children | 442a1bf9b14e |
| rev | line source |
|---|---|
| 37 | 1 import pygame |
|
39
d82d3e54a4ef
fixed more pep8 issues
Rizmari Versfeld <rizziepit@gmail.com>
parents:
38
diff
changeset
|
2 from pygame import Surface, Rect |
| 37 | 3 from pygame.font import Font |
| 4 | |
| 5 from gamelib import data | |
| 6 | |
| 7 | |
| 8 # different font sizes | |
| 9 pygame.font.init() | |
| 10 font_small = Font(data.filepath('fonts/DejaVuSans.ttf'), 10) | |
| 11 font_medium = Font(data.filepath('fonts/DejaVuSans.ttf'), 14) | |
| 12 font_large = Font(data.filepath('fonts/DejaVuSans.ttf'), 18) | |
| 13 | |
| 14 | |
| 15 class Drawable(object): | |
| 38 | 16 |
| 17 def __init__(self, rect): | |
| 18 if isinstance(rect, Rect): | |
| 19 self.rect = rect | |
| 20 else: | |
| 21 self.rect = Rect(rect[0], rect[1], rect[2], rect[3]) | |
| 22 | |
| 23 def draw(self, surface): | |
| 24 pass | |
| 25 | |
| 26 | |
| 37 | 27 class Clickable(object): |
| 38 | 28 |
| 29 def on_mouse_down(self, pos): | |
| 30 pass | |
| 31 | |
| 32 def on_mouse_up(self, pos): | |
| 33 pass | |
| 34 | |
| 35 def on_mouse_move(self, pos): | |
| 36 pass | |
| 37 | |
| 38 def on_mouse_cancel(self): | |
| 39 pass | |
| 40 | |
| 41 def on_click(self): | |
| 42 pass | |
| 43 | |
| 37 | 44 |
| 45 class ContainerView(Drawable): | |
| 38 | 46 |
| 47 def __init__(self): | |
| 48 self.children = [] | |
| 49 | |
| 50 def add_child(self, child): | |
| 51 self.children.append(child) | |
| 52 | |
| 53 def remove_child(self, child): | |
| 54 self.children.remove(child) | |
| 55 | |
| 56 def get_child_by_pos(self, pos): | |
| 57 for child in self.children: | |
| 58 if child.rect.collidepoint(pos): | |
| 59 if isinstance(child, ContainerView): | |
| 60 # calculates position relative to child | |
| 61 x = pos[0] - child.rect[0] | |
| 62 y = pos[1] - child.rect[1] | |
| 63 return child.get_child_by_pos((x, y)) | |
| 64 else: | |
| 65 return child | |
| 66 | |
| 67 def draw_children(self, surface): | |
| 68 for child in self.children: | |
| 69 child.draw(surface) | |
| 70 | |
| 71 | |
| 37 | 72 class Window(Clickable, ContainerView): |
| 38 | 73 |
| 74 def __init__(self, screen): | |
| 75 super(Window, self).__init__() | |
| 76 self.surface = Surface((screen.get_width(), screen.get_height())) | |
| 77 self.background_colour = None | |
| 78 self.background_image = None | |
| 79 self.pressed_child = None | |
| 80 | |
| 81 def on_mouse_down(self, pos): | |
| 82 child = self.get_child_by_pos(pos) | |
| 83 if isinstance(child, Clickable): | |
| 84 # calculates position relative to child | |
| 85 x = pos[0] - child.rect[0] | |
| 86 y = pos[1] - child.rect[1] | |
| 87 child.on_mouse_down((x, y)) | |
| 88 self.pressed_child = child | |
| 89 | |
| 90 def on_mouse_up(self, pos): | |
| 91 if self.pressed_child: | |
| 92 child = self.pressed_child | |
| 93 x = pos[0] - child.rect[0] | |
| 94 y = pos[1] - child.rect[1] | |
| 95 child.on_mouse_up((x, y)) | |
| 96 self.pressed_child = None | |
| 97 | |
| 98 def on_mouse_move(self, pos): | |
| 99 if self.pressed_child: | |
| 100 if self.pressed_child != self.get_child_by_pos(pos): | |
| 101 self.pressed_child.on_mouse_cancel() | |
| 102 self.pressed_child = None | |
| 103 | |
| 104 def draw(self, screen): | |
| 105 if self.background_colour: | |
| 106 self.surface.fill(self.background_colour) | |
|
68
7309392d9ca9
fixed widget drawing bugs
Rizmari Versfeld <rizziepit@gmail.com>
parents:
64
diff
changeset
|
107 else: |
|
7309392d9ca9
fixed widget drawing bugs
Rizmari Versfeld <rizziepit@gmail.com>
parents:
64
diff
changeset
|
108 self.surface.fill((0, 0, 0, 255)) |
| 38 | 109 if self.background_image: |
| 110 self.surface.blit(self.background_image, (0, 0)) | |
| 111 self.draw_children(self.surface) | |
| 112 screen.blit(self.surface, (0, 0)) | |
| 113 | |
| 37 | 114 |
| 115 class StateDrawable(Drawable): | |
| 38 | 116 |
| 117 def __init__(self, rect): | |
| 118 super(StateDrawable, self).__init__(rect) | |
| 119 self.state = -1 | |
| 120 self.states = {} | |
| 121 self.drawables = [] | |
| 122 self.surface = Surface((self.rect[2], self.rect[3])) | |
| 123 | |
| 124 def draw(self, surface): | |
| 125 if self.state != -1 and self.drawables[self.state]: | |
| 126 self.drawables[self.state].draw(surface) | |
| 127 | |
| 128 def add_state(self, state_name, drawable): | |
| 129 self.states[state_name] = len(self.drawables) | |
| 130 self.drawables.append(drawable) | |
| 131 | |
| 132 def set_state(self, state_name): | |
| 133 self.state = self.states[state_name] | |
| 134 | |
| 135 | |
| 37 | 136 class Button(Clickable, StateDrawable): |
| 38 | 137 |
| 138 def __init__(self, rect, normal_drawable, down_drawable): | |
| 139 super(Button, self).__init__(rect) | |
| 140 self.add_state('NORMAL', normal_drawable) | |
| 141 self.add_state('DOWN', down_drawable) | |
| 142 self.set_state('NORMAL') | |
| 143 | |
| 144 def on_mouse_down(self, pos): | |
| 145 self.set_state('DOWN') | |
| 146 | |
| 147 def on_mouse_up(self, pos): | |
| 148 self.set_state('NORMAL') | |
| 149 self.on_click() | |
| 150 | |
| 151 def on_mouse_cancel(self): | |
| 152 self.set_state('NORMAL') | |
| 153 | |
| 154 | |
| 37 | 155 class TextButton(Button): |
| 156 | |
|
62
38f41d046c6f
Make button font a parameter. Tweak indication of points to be spent
Neil Muller <drnlmuller@gmail.com>
parents:
55
diff
changeset
|
157 def __init__(self, rect, normal_drawable, down_drawable, text, font, |
|
38f41d046c6f
Make button font a parameter. Tweak indication of points to be spent
Neil Muller <drnlmuller@gmail.com>
parents:
55
diff
changeset
|
158 shadow): |
| 38 | 159 super(TextButton, self).__init__(rect, normal_drawable, down_drawable) |
|
68
7309392d9ca9
fixed widget drawing bugs
Rizmari Versfeld <rizziepit@gmail.com>
parents:
64
diff
changeset
|
160 self.surface = Surface((rect[2], rect[3]), pygame.SRCALPHA) |
| 38 | 161 self.text = text |
|
62
38f41d046c6f
Make button font a parameter. Tweak indication of points to be spent
Neil Muller <drnlmuller@gmail.com>
parents:
55
diff
changeset
|
162 self.font = font |
|
38f41d046c6f
Make button font a parameter. Tweak indication of points to be spent
Neil Muller <drnlmuller@gmail.com>
parents:
55
diff
changeset
|
163 self.shadow = shadow |
|
55
86d83dcb7d42
Make text drawing easier to poke at for subclasses
Neil Muller <drnlmuller@gmail.com>
parents:
39
diff
changeset
|
164 self._draw_text() |
|
86d83dcb7d42
Make text drawing easier to poke at for subclasses
Neil Muller <drnlmuller@gmail.com>
parents:
39
diff
changeset
|
165 |
|
86d83dcb7d42
Make text drawing easier to poke at for subclasses
Neil Muller <drnlmuller@gmail.com>
parents:
39
diff
changeset
|
166 def _draw_text(self): |
|
62
38f41d046c6f
Make button font a parameter. Tweak indication of points to be spent
Neil Muller <drnlmuller@gmail.com>
parents:
55
diff
changeset
|
167 self.font.set_bold(True) |
|
68
7309392d9ca9
fixed widget drawing bugs
Rizmari Versfeld <rizziepit@gmail.com>
parents:
64
diff
changeset
|
168 self.text_surface = self.font.render(self.text, True, (0, 0, 0)) |
| 69 | 169 size = self.font.size(self.text) |
|
62
38f41d046c6f
Make button font a parameter. Tweak indication of points to be spent
Neil Muller <drnlmuller@gmail.com>
parents:
55
diff
changeset
|
170 if self.shadow: |
| 69 | 171 s = self.font.render(self.text, True, (128, 128, 128)) |
| 172 temp_surface = Surface((s.get_width() + 2, s.get_width() + 2), pygame.SRCALPHA) | |
|
68
7309392d9ca9
fixed widget drawing bugs
Rizmari Versfeld <rizziepit@gmail.com>
parents:
64
diff
changeset
|
173 temp_surface.fill((255, 255, 255, 0)) |
| 69 | 174 temp_surface.blit(s, (2, 2)) |
|
68
7309392d9ca9
fixed widget drawing bugs
Rizmari Versfeld <rizziepit@gmail.com>
parents:
64
diff
changeset
|
175 temp_surface.blit(self.text_surface, (0, 0)) |
|
7309392d9ca9
fixed widget drawing bugs
Rizmari Versfeld <rizziepit@gmail.com>
parents:
64
diff
changeset
|
176 self.text_surface = temp_surface |
|
7309392d9ca9
fixed widget drawing bugs
Rizmari Versfeld <rizziepit@gmail.com>
parents:
64
diff
changeset
|
177 size = [s + 2 for s in size] |
|
55
86d83dcb7d42
Make text drawing easier to poke at for subclasses
Neil Muller <drnlmuller@gmail.com>
parents:
39
diff
changeset
|
178 self.text_offset = ((self.rect[2] - size[0]) / 2, |
|
86d83dcb7d42
Make text drawing easier to poke at for subclasses
Neil Muller <drnlmuller@gmail.com>
parents:
39
diff
changeset
|
179 (self.rect[3] - size[1]) / 2) |
|
68
7309392d9ca9
fixed widget drawing bugs
Rizmari Versfeld <rizziepit@gmail.com>
parents:
64
diff
changeset
|
180 self.font.set_bold(False) |
| 37 | 181 |
| 38 | 182 def draw(self, surface): |
| 183 self.surface.fill((0, 0, 0, 0)) | |
| 184 super(TextButton, self).draw(self.surface) | |
| 185 self.surface.blit(self.text_surface, self.text_offset) | |
| 186 surface.blit(self.surface, self.rect) | |
| 64 | 187 |
| 188 | |
| 189 class TextLabel(Drawable): | |
| 190 | |
| 191 def __init__(self, rect, text, font, color): | |
| 192 super(TextLabel, self).__init__(rect) | |
|
68
7309392d9ca9
fixed widget drawing bugs
Rizmari Versfeld <rizziepit@gmail.com>
parents:
64
diff
changeset
|
193 self.surface = Surface((rect[2], rect[3]), pygame.SRCALPHA) |
| 64 | 194 self.text = text |
| 195 self.font = font | |
| 196 self.color = color | |
| 197 self._draw_text() | |
| 198 | |
| 199 def _draw_text(self): | |
| 200 self.text_surface = self.font.render(self.text, True, self.color) | |
| 201 size = self.font.size(self.text) | |
| 202 # We centre vertically, but start at left edge | |
| 203 self.text_offset = (0, (self.rect[3] - size[1]) / 2) | |
| 204 | |
| 205 def draw(self, surface): | |
| 206 self.surface.fill((0, 0, 0, 0)) | |
| 207 super(TextLabel, self).draw(self.surface) | |
| 208 self.surface.blit(self.text_surface, self.text_offset) | |
| 209 surface.blit(self.surface, self.rect) |