annotate gamelib/gui_base.py @ 71:20b01d100ff5

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