annotate gamelib/gamegui.py @ 67:06c8539df478

Tweak label handling. Show money as well
author Neil Muller <drnlmuller@gmail.com>
date Tue, 08 May 2012 22:28:00 +0200
parents 43b34b013462
children 8b06b07a1477
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
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
11 from gamelib.engine import PopWindow, AddWindow
53
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):
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
42 self.parent.do_reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
43
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
44
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
45 class SwitchWinButton(BigButton):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
46
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
47 def __init__(self, pos, text, new_window):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
48 super(SwitchWinButton, self).__init__(pos, text)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
49 self.new_window = new_window
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
50
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
51 def on_click(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
52 PopWindow.post()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
53 # Refresh state
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
54 self.new_window.update()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
55 AddWindow.post(self.new_window)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
57
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
58 class ScienceWidget(BigButton):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
59
57
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
60 WIDTH = 200
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
61
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
62 BG_IMAGE_NORMAL = image.load(filepath('images/science_normal.png'))
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
63 BG_IMAGE_DOWN = image.load(filepath('images/science_down.png'))
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
64
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
65 def __init__(self, science, pos, parent):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
66 self.science = science
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
67 self.points = 0
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
68 self.parent = parent
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
69 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
70 science.points), font_small)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
71
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
72 def on_click(self):
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
73 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
74 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
75 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
76 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
77 self.science.points, self.points)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
78 self._draw_text()
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
79 self.parent.game.cur_allocation.append(self.science)
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
80 self.parent.update_labels()
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
81
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
82 def reset(self):
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
83 while self.points > 0:
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
84 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
85 self.points -= 1
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
86 self.text = '%s: %d' % (self.science.NAME, self.science.points)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
87 self._draw_text()
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
88 self.parent.update_labels()
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
89
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
90
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
91 class ValueLabel(TextLabel):
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
92
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
93 def __init__(self, pos, description):
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
94 self.description = description
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
95 rect = (pos[0], pos[1], 300, 50)
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
96 super(ValueLabel, self).__init__(rect,
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
97 '%s : 0' % description, font_medium, (255, 255, 0))
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
98
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
99 def set_value(self, value):
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
100 self.text = '%s : %d' % (self.description, value)
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
101 self._draw_text()
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
102
60
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
103
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
104 class ActivityWindow(Window):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
105
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
106 def __init__(self, screen, lab, develop):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
107 super(ActivityWindow, self).__init__(screen)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
108 self.lab = lab
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
109 self.develop = develop
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
110 self.game = lab.game
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
111 exitbut = ExitGameButton()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
112 self.add_child(exitbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
113 end_turn = EndTurnButton(self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
114 self.add_child(end_turn)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
115 reset = ResetButton(self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
116 self.add_child(reset)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
117
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
118 self.points = ValueLabel((10, 75), 'Available Human Resources')
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
119 self.add_child(self.points)
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
120 self.money = ValueLabel((310, 75), 'Money')
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
121 self.add_child(self.money)
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
122
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
123 labbut = SwitchWinButton((150, 10), 'Research', lab)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
124 self.add_child(labbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
125
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
126 devbut = SwitchWinButton((300, 10), 'Development', develop)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
127 self.add_child(devbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
128
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
129 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
130 self._make_widgets()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
131
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
132 def _make_widgets(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
133 pass
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
134
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
135 def end_turn(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
136 # Drop back to the research screen
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
137 PopWindow.post()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
138 self.lab.update()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
139 AddWindow.post(self.lab)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
140 self.lab.end_turn()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
141
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
142 def update_widgets(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
143 pass
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
144
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
145 def update(self):
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
146 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
147
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
148 def update_labels(self):
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
149 self.points.set_value(self.game.get_available_points())
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
150 self.money.set_value(self.game.money)
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
151
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
152 def do_reset(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
153 self.lab.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
154 self.develop.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
155 self.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
156
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
157 def reset(self):
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
158 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
159
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
160
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
161 class DevelopmentWindow(Window):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
162 """Window for handling schematics research"""
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
163
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
164 def __init__(self, screen, lab):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
165 super(DevelopmentWindow, self).__init__(screen)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
166 self.lab = lab
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
167 self.game = lab.game
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
168 exitbut = ExitGameButton()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
169 self.add_child(exitbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
170 end_turn = EndTurnButton(self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
171 self.add_child(end_turn)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
172 reset = ResetButton(self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
173 self.add_child(reset)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
174
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
175 self.points = ValueLabel((10, 75), 'Available Human Resources')
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
176 self.add_child(self.points)
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
177 self.money = ValueLabel((310, 75), 'Money')
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
178 self.add_child(self.money)
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
179
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
180 labbut = SwitchWinButton((150, 10), 'Research', lab)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
181 self.add_child(labbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
182 self.activity = None
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
183
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
184 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
185 self._sciences = []
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
186 self._make_science_widgets()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
187
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
188 def set_activity_window(self, activity):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
189 # Oh, what tangled webs we weave
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
190 if not self.activity:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
191 self.activity = activity
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
192 actbut = SwitchWinButton((300, 10), 'Activities', activity)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
193 self.add_child(actbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
194
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
195 def _make_science_widgets(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
196 x = 0
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
197 y = 150
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
198
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
199 for science in self.game.lab.science:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
200 if science.SCIENCE_TYPE == 'schematic':
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
201 widget = ScienceWidget(science, (x, y), self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
202 self.add_child(widget)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
203 self._sciences.append(widget)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
204 x += 200
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
205 if x >= WIDTH:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
206 x = 0
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
207 y += 100
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
208
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
209 def end_turn(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
210 # Drop back to the research screen
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
211 PopWindow.post()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
212 self.lab.update()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
213 AddWindow.post(self.lab)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
214 self.lab.end_turn()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
215
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
216 def update_widgets(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
217 for widget in self._sciences:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
218 self.remove_child(widget)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
219 self._sciences = []
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
220 self._make_science_widgets()
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
221 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
222
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
223 def update(self):
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
224 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
225
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
226 def update_labels(self):
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
227 self.points.set_value(self.game.get_available_points())
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
228 self.money.set_value(self.game.money)
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
229
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
230 def do_reset(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
231 self.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
232 self.lab.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
233 self.activity.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
234
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
235 def reset(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
236 for widget in self._sciences:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
237 widget.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
238
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
239
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
240 class LabWindow(Window):
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
241 """Window for the research lab"""
53
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
242
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
243 def __init__(self, screen):
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
244 super(LabWindow, self).__init__(screen)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
245 self.game = Game()
53
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
246 exit = ExitGameButton()
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
247 self.add_child(exit)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
248 end_turn = EndTurnButton(self)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
249 self.add_child(end_turn)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
250 reset = ResetButton(self)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
251 self.add_child(reset)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
252
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
253 self.points = ValueLabel((10, 75), 'Available Human Resources')
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
254 self.add_child(self.points)
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
255 self.money = ValueLabel((310, 75), 'Money')
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
256 self.add_child(self.money)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
257
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
258 self.develop = DevelopmentWindow(screen, self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
259 self.activity = ActivityWindow(screen, self, self.develop)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
260 self.develop.set_activity_window(self.activity)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
261
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
262 devbut = SwitchWinButton((150, 10), 'Development', self.develop)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
263 self.add_child(devbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
264 actbut = SwitchWinButton((300, 10), 'Activities', self.activity)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
265 self.add_child(actbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
266
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
267 self._sciences = []
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
268 # Setup for the first turn
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
269 self.game.start_turn()
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
270 self.update_labels()
60
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
271 self._make_science_widgets()
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
272
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
273 def _make_science_widgets(self):
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
274 x = 0
57
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
275 y = 150
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
276
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
277 for science in self.game.lab.science:
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
278 if science.SCIENCE_TYPE == 'research':
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
279 widget = ScienceWidget(science, (x, y), self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
280 self.add_child(widget)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
281 self._sciences.append(widget)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
282 x += 200
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
283 if x >= WIDTH:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
284 x = 0
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
285 y += 100
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
286
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
287 def update_widgets(self):
60
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
288 # FIXME: Horrible hackery
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
289 for widget in self._sciences:
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
290 self.remove_child(widget)
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
291 self._sciences = []
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
292 self._make_science_widgets()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
293
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
294 def end_turn(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
295 self.game.end_turn()
60
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
296 self.game.start_turn()
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
297 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
298 self.update_widgets()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
299 self.develop.update_widgets()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
300 self.activity.update_widgets()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
301
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
302 def update(self):
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
303 self.update_labels()
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
304
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
305 def update_labels(self):
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
306 self.points.set_value(self.game.get_available_points())
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
307 self.money.set_value(self.game.money)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
308
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
309 def do_reset(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
310 self.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
311 self.develop.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
312 self.activity.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
313
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
314 def reset(self):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
315 for widget in self._sciences:
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
316 widget.reset()