annotate gamelib/gamegui.py @ 65:5c4c67673112

Display available points
author Neil Muller <drnlmuller@gmail.com>
date Tue, 08 May 2012 21:06:08 +0200
parents 364ff3479ef2
children 43b34b013462
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
53
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
1 # -*- coding: utf-8 -*-
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
2 # vim:fileencoding=utf-8 ai ts=4 sts=4 et sw=4
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
3
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
4 """Gui for the actual game"""
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
5
57
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
6 from pygame import image
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
7
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
8 from gamelib.data import filepath
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
9 from gamelib.gui_base import Window, TextLabel, font_small, font_medium
53
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
10 from gamelib.gui import BigButton
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
11 from gamelib.engine import PopWindow
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
12 from gamelib.constants import WIDTH
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
13 from gamelib.gamestate import Game
53
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
14
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
15
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
16 class ExitGameButton(BigButton):
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
17
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
18 def __init__(self):
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
19 super(ExitGameButton, self).__init__(((WIDTH - 128), 10), 'Exit')
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
20
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
21 def on_click(self):
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
22 PopWindow.post()
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
23
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
24
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
25 class EndTurnButton(BigButton):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
26
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
27 def __init__(self, parent):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
28 super(EndTurnButton, self).__init__(((WIDTH - 256), 10), 'End Turn')
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
29 self.parent = parent
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
30
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
31 def on_click(self):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
32 self.parent.end_turn()
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
33
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
34
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
35 class ResetButton(BigButton):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
36
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
37 def __init__(self, parent):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
38 super(ResetButton, self).__init__((10, 10), 'Reset points')
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
39 self.parent = parent
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
40
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
41 def on_click(self):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
42 self.parent.reset()
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
43
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
44
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
45 class ScienceWidget(BigButton):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
46
57
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
47 WIDTH = 200
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
48
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
49 BG_IMAGE_NORMAL = image.load(filepath('images/science_normal.png'))
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
50 BG_IMAGE_DOWN = image.load(filepath('images/science_down.png'))
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
51
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
52 def __init__(self, science, pos, parent):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
53 self.science = science
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
54 self.points = 0
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
55 self.parent = parent
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
56 super(ScienceWidget, self).__init__(pos, '%s: %d' % (science.NAME,
62
38f41d046c6f Make button font a parameter. Tweak indication of points to be spent
Neil Muller <drnlmuller@gmail.com>
parents: 60
diff changeset
57 science.points), font_small)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
58
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
59 def on_click(self):
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
60 if (self.parent.game.get_available_points() > 0 and
59
977224a5c663 Check can_spend in science button
Neil Muller <drnlmuller@gmail.com>
parents: 57
diff changeset
61 self.science.can_spend(self.parent.game.lab, self.points + 1)):
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
62 self.points += 1
62
38f41d046c6f Make button font a parameter. Tweak indication of points to be spent
Neil Muller <drnlmuller@gmail.com>
parents: 60
diff changeset
63 self.text = '%s: %d + %d' % (self.science.NAME,
38f41d046c6f Make button font a parameter. Tweak indication of points to be spent
Neil Muller <drnlmuller@gmail.com>
parents: 60
diff changeset
64 self.science.points, self.points)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
65 self._draw_text()
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
66 self.parent.game.cur_allocation.append(self.science)
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
67 self.parent.update_points()
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
68
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
69 def reset(self):
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
70 while self.points > 0:
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
71 self.parent.game.cur_allocation.remove(self.science)
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
72 self.points -= 1
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
73 self.text = '%s: %d' % (self.science.NAME, self.science.points)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
74 self._draw_text()
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
75 self.parent.update_points()
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
76
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
77
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
78 class PointsLabel(TextLabel):
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
79
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
80 def __init__(self):
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
81 super(PointsLabel, self).__init__((10, 70, 300, 50),
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
82 'Available Human Resources : 0', font_medium, (255, 255, 0))
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
83
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
84 def set_value(self, value):
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
85 self.text = 'Available Human Resource : %d' % value
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
86 self._draw_text()
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
87
60
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
88
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
89 class LabWindow(Window):
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
90 """Window for the research lab"""
53
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
91
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
92 def __init__(self, screen):
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
93 super(LabWindow, self).__init__(screen)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
94 self.game = Game()
53
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
95 exit = ExitGameButton()
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
96 self.add_child(exit)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
97 end_turn = EndTurnButton(self)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
98 self.add_child(end_turn)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
99 reset = ResetButton(self)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
100 self.add_child(reset)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
101
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
102 self.points = PointsLabel()
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
103 self.add_child(self.points)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
104
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
105 self._sciences = []
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
106 # Setup for the first turn
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
107 self.game.start_turn()
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
108 self.update_points()
60
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
109 self._make_science_widgets()
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
110
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
111 def _make_science_widgets(self):
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
112 x = 0
57
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
113 y = 150
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
114
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
115 for science in self.game.lab.science:
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
116 widget = ScienceWidget(science, (x, y), self)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
117 self.add_child(widget)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
118 self._sciences.append(widget)
57
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
119 x += 200
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
120 if x >= WIDTH:
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
121 x = 0
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
122 y += 100
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
123
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
124 def end_turn(self):
60
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
125 self.game.end_turn()
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
126 # FIXME: Horrible hackery
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
127 for widget in self._sciences:
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
128 self.remove_child(widget)
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
129 self._sciences = []
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
130 self._make_science_widgets()
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
131 self.game.start_turn()
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
132 self.update_points()
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
133
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
134 def update_points(self):
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
135 self.points.set_value(self.game.get_available_points())
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
136
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
137 def reset(self):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
138 for widget in self._sciences:
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
139 widget.reset()