annotate gamelib/gamegui.py @ 82:b0d97d51df51

Hook up simplistic equipment screen
author Neil Muller <drnlmuller@gmail.com>
date Wed, 09 May 2012 16:51:26 +0200
parents 59afe9f92383
children 74ce25ec2073
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
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
10 from gamelib.gui import BigButton, ImageDrawable
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
11 from gamelib.engine import PopWindow, AddWindow
80
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 72
diff changeset
12 from gamelib.constants import (WIDTH, FAILURE, SUCCESS, NEW_SCIENCE,
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 72
diff changeset
13 NEW_SCHEMATIC)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
14 from gamelib.gamestate import Game
53
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
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
17 class ExitGameButton(BigButton):
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
18
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
19 def __init__(self):
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
20 super(ExitGameButton, self).__init__(((WIDTH - 128), 10), 'Exit')
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
21
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
22 def on_click(self):
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
23 PopWindow.post()
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
24
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
25
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
26 class EndTurnButton(BigButton):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
27
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
28 def __init__(self, parent):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
29 super(EndTurnButton, self).__init__(((WIDTH - 256), 10), 'End Turn')
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
30 self.parent = parent
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
31
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
32 def on_click(self):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
33 self.parent.end_turn()
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
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
36 class ResetButton(BigButton):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
37
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
38 def __init__(self, parent):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
39 super(ResetButton, self).__init__((10, 10), 'Reset points')
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
40 self.parent = parent
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
41
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
42 def on_click(self):
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
43 self.parent.do_reset()
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
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
46 class SwitchWinButton(BigButton):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
47
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
48 def __init__(self, pos, text, new_window):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
49 super(SwitchWinButton, self).__init__(pos, text)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
50 self.new_window = new_window
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
51
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
52 def on_click(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
53 PopWindow.post()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
54 # Refresh state
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
55 self.new_window.update()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
56 AddWindow.post(self.new_window)
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
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
59 class ScienceWidget(BigButton):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
60
57
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
61 WIDTH = 200
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
62
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
63 BG_IMAGE_NORMAL = image.load(filepath('images/science_normal.png'))
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
64 BG_IMAGE_DOWN = image.load(filepath('images/science_down.png'))
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
65
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
66 def __init__(self, science, pos, parent):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
67 self.science = science
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
68 self.points = 0
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
69 self.parent = parent
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
70 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
71 science.points), font_small)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
72
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
73 def on_click(self):
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
74 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
75 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
76 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
77 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
78 self.science.points, self.points)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
79 self._draw_text()
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
80 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
81 self.parent.update_labels()
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
82
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
83 def reset(self):
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
84 while self.points > 0:
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
85 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
86 self.points -= 1
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
87 self.text = '%s: %d' % (self.science.NAME, self.science.points)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
88 self._draw_text()
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
89 self.parent.update_labels()
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
90
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
91
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
92 class MissionWidget(BigButton):
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
93
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
94 WIDTH = 200
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
95
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
96 BG_IMAGE_NORMAL = image.load(filepath('images/science_normal.png'))
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
97 BG_IMAGE_DOWN = image.load(filepath('images/science_down.png'))
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
98 BG_IMAGE_SELECTED = image.load(filepath('images/mission_selected.png'))
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
99
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
100 def __init__(self, mission, pos, parent):
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
101 self.mission = mission
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
102 self.parent = parent
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
103 super(MissionWidget, self).__init__(pos, '%s' % mission.NAME,
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
104 font_small)
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
105 selected = ImageDrawable((0, 0, self.WIDTH, self.HEIGHT),
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
106 self.BG_IMAGE_SELECTED)
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
107 self.add_state('SELECTED', selected)
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
108 self.mode = 'NORMAL'
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
109 self.equipment = []
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
110
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
111 def on_mouse_cancel(self, pos):
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
112 self.set_state(self.mode)
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
113
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
114 def on_mouse_up(self, pos):
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
115 self.set_state(self.mode)
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
116 self.on_click()
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
117
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
118 def on_click(self):
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
119 if self.mode == 'SELECTED':
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
120 # unselect mission
82
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
121 self.reset()
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
122 else:
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
123 # select mission and equipment
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
124 self.mode = 'SELECTED'
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
125 self.set_state('SELECTED')
82
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
126 equip = EquipWindow(self.parent.screen, self.mission,
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
127 self.parent.game, self)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
128 AddWindow.post(equip)
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
129
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
130 def selected(self):
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
131 return self.mode == 'SELECTED'
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
132
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
133 def get_equipment(self):
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
134 return self.equipment
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
135
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
136 def reset(self):
82
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
137 for equip in self.equipment:
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
138 # Release funds
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
139 self.parent.game.money += equip.COST
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
140 self.parent.update()
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
141 self.equipment = []
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
142 self.mode = 'NORMAL'
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
143 self.set_state('NORMAL')
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
144
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
145
82
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
146 class EquipWidget(BigButton):
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
147
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
148 WIDTH = 200
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
149
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
150 BG_IMAGE_NORMAL = image.load(filepath('images/science_normal.png'))
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
151 BG_IMAGE_DOWN = image.load(filepath('images/science_down.png'))
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
152 BG_IMAGE_UNAVAILABLE = image.load(filepath('images/equip_grey.png'))
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
153
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
154 def __init__(self, equip, pos, parent, copies, available):
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
155 self.equip = equip
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
156 self.parent = parent
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
157 self.available = available
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
158 super(EquipWidget, self).__init__(pos, '[%s : %d] - %d (x %d)'
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
159 % (equip.NAME, equip.points, equip.COST, copies), font_small)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
160 if not self.available:
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
161 unavailable = ImageDrawable((0, 0, self.WIDTH, self.HEIGHT),
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
162 self.BG_IMAGE_UNAVAILABLE)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
163 # Override the default
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
164 self.add_state('NORMAL', unavailable)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
165 self.set_state('NORMAL')
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
166 self.equipment = []
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
167
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
168 def on_mouse_down(self, pos):
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
169 if self.available:
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
170 super(EquipWidget, self).on_mouse_down(pos)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
171
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
172 def on_mouse_up(self, pos):
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
173 if self.available:
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
174 super(EquipWidget, self).on_mouse_up(pos)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
175
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
176 def on_click(self):
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
177 self.parent.buy(self.equip)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
178
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
179
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
180 class ValueLabel(TextLabel):
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
181
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
182 def __init__(self, pos, description):
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
183 self.description = description
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
184 rect = (pos[0], pos[1], 300, 50)
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
185 super(ValueLabel, self).__init__(rect,
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
186 '%s : 0' % description, font_medium, (255, 255, 0))
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
187
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
188 def set_value(self, value):
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
189 self.text = '%s : %d' % (self.description, value)
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
190 self._draw_text()
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
191
60
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
192
82
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
193 class EquipWindow(Window):
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
194
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
195 def __init__(self, screen, mission, game, parent):
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
196 super(EquipWindow, self).__init__(screen)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
197 exitbut = ExitGameButton()
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
198 self.add_child(exitbut)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
199 reset = ResetButton(self)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
200 self.add_child(reset)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
201 title = TextLabel((200, 20, 400, 50), "Choose equipment for %s"
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
202 % mission.NAME, font_medium, (255, 255, 255))
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
203 self.add_child(title)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
204 self.parent = parent
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
205 self.game = game
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
206 self.money = ValueLabel((10, 75), 'Money')
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
207 self.money.set_value(self.game.money)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
208 self.add_child(self.money)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
209 self._equip = []
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
210 self._make_equip_widgets()
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
211
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
212 def _make_equip_widgets(self):
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
213 for widget in self._equip:
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
214 self.remove_child(widget)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
215 self._equip = []
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
216 x = 0
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
217 y = 150
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
218 available = self.game.get_available_equipment()
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
219 for equip in self.game.get_all_equipment():
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
220 copies = self.parent.equipment.count(equip)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
221 widget = EquipWidget(equip, (x, y), self, copies,
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
222 equip in available)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
223 self._equip.append(widget)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
224 self.add_child(widget)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
225 x += 200
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
226 if x >= WIDTH:
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
227 x = 0
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
228 y += 100
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
229
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
230 def buy(self, equip):
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
231 self.game.money -= equip.COST
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
232 self.money.set_value(self.game.money)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
233 self.parent.equipment.append(equip)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
234 self.parent.parent.update()
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
235 self._make_equip_widgets()
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
236
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
237 def do_reset(self):
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
238 for equip in self.parent.equipment:
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
239 # Release funds
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
240 self.game.money += equip.COST
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
241 self.money.set_value(self.game.money)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
242 self.parent.equipment = []
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
243 self.parent.parent.update()
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
244 self._make_equip_widgets()
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
245
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
246
81
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
247 class ResultsWindow(Window):
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
248
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
249 def __init__(self, screen, messages, turn):
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
250 super(ResultsWindow, self).__init__(screen)
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
251 exitbut = ExitGameButton()
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
252 self.add_child(exitbut)
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
253 title = TextLabel((200, 20, 400, 50), "Results for turn %d" % turn,
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
254 font_medium, (255, 255, 255))
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
255 self.add_child(title)
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
256 if not messages:
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
257 results = TextLabel((200, 200, 400, 50),
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
258 "Nothing of interest happened", font_medium,
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
259 (255, 255, 255))
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
260 self.add_child(results)
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
261 else:
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
262 y = 200
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
263 for msg_type, msg in messages:
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
264 # FIXME: Better widgets
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
265 if msg_type in [NEW_SCHEMATIC, NEW_SCIENCE]:
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
266 text = TextLabel((50, y, 750, 25),
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
267 msg, font_medium, (60, 60, 255))
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
268 elif msg_type == FAILURE:
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
269 text = TextLabel((50, y, 750, 25),
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
270 msg, font_medium, (255, 60, 60))
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
271 elif msg_type == SUCCESS:
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
272 text = TextLabel((50, y, 750, 25),
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
273 msg, font_medium, (60, 255, 60))
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
274 else:
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
275 text = TextLabel((50, y, 750, 25),
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
276 msg, font_medium, (255, 255, 0))
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
277 y += 50
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
278 self.add_child(text)
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
279
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
280
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
281 class ActivityWindow(Window):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
282
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
283 def __init__(self, screen, lab, develop):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
284 super(ActivityWindow, self).__init__(screen)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
285 self.lab = lab
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
286 self.develop = develop
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
287 self.game = lab.game
82
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
288 self.screen = screen
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
289 exitbut = ExitGameButton()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
290 self.add_child(exitbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
291 end_turn = EndTurnButton(self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
292 self.add_child(end_turn)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
293 reset = ResetButton(self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
294 self.add_child(reset)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
295
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
296 self.points = ValueLabel((10, 75), 'Available Human Resources')
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
297 self.add_child(self.points)
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
298 self.money = ValueLabel((310, 75), 'Money')
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
299 self.add_child(self.money)
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
300
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
301 labbut = SwitchWinButton((150, 10), 'Research', lab)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
302 self.add_child(labbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
303
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
304 devbut = SwitchWinButton((300, 10), 'Development', develop)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
305 self.add_child(devbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
306
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
307 self.update_labels()
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
308 self._missions = []
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
309 self._make_widgets()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
310
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
311 def _make_widgets(self):
82
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
312 for widget in self._missions:
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
313 self.remove_child(widget)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
314 self._missions = []
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
315 x = 0
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
316 y = 150
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
317 for mission in self.game.get_available_missions():
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
318 widget = MissionWidget(mission, (x, y), self)
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
319 self._missions.append(widget)
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
320 self.add_child(widget)
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
321 x += 200
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
322 if x >= WIDTH:
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
323 x = 0
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
324 y += 100
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
325
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
326 def end_turn(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
327 # Drop back to the research screen
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
328 PopWindow.post()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
329 self.lab.update()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
330 AddWindow.post(self.lab)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
331 self.lab.end_turn()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
332
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
333 def update_widgets(self):
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
334 self._make_widgets()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
335
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
336 def update(self):
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
337 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
338
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
339 def update_labels(self):
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
340 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
341 self.money.set_value(self.game.money)
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
342
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
343 def do_reset(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
344 self.lab.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
345 self.develop.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
346 self.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
347
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
348 def get_mission_list(self):
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
349 selected_missions = []
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
350 for widget in self._missions:
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
351 if widget.selected():
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
352 equipment = widget.get_equipment()
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
353 mission = widget.mission
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
354 selected_missions.append((mission, equipment))
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
355 return selected_missions
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
356
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
357 def reset(self):
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
358 for widget in self._missions:
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
359 widget.reset()
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
360 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
361
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
362
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
363 class DevelopmentWindow(Window):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
364 """Window for handling schematics research"""
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
365
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
366 def __init__(self, screen, lab):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
367 super(DevelopmentWindow, self).__init__(screen)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
368 self.lab = lab
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
369 self.game = lab.game
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
370 exitbut = ExitGameButton()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
371 self.add_child(exitbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
372 end_turn = EndTurnButton(self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
373 self.add_child(end_turn)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
374 reset = ResetButton(self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
375 self.add_child(reset)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
376
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
377 self.points = ValueLabel((10, 75), 'Available Human Resources')
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
378 self.add_child(self.points)
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
379 self.money = ValueLabel((310, 75), 'Money')
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
380 self.add_child(self.money)
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
381
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
382 labbut = SwitchWinButton((150, 10), 'Research', lab)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
383 self.add_child(labbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
384 self.activity = None
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
385
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
386 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
387 self._sciences = []
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
388 self._make_science_widgets()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
389
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
390 def set_activity_window(self, activity):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
391 # Oh, what tangled webs we weave
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
392 if not self.activity:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
393 self.activity = activity
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
394 actbut = SwitchWinButton((300, 10), 'Activities', activity)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
395 self.add_child(actbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
396
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
397 def _make_science_widgets(self):
82
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
398 for widget in self._sciences:
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
399 self.remove_child(widget)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
400 self._sciences = []
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
401 x = 0
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
402 y = 150
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
403 for science in self.game.lab.science:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
404 if science.SCIENCE_TYPE == 'schematic':
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
405 widget = ScienceWidget(science, (x, y), self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
406 self.add_child(widget)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
407 self._sciences.append(widget)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
408 x += 200
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
409 if x >= WIDTH:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
410 x = 0
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
411 y += 100
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
412
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
413 def end_turn(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
414 # Drop back to the research screen
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
415 PopWindow.post()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
416 self.lab.update()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
417 AddWindow.post(self.lab)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
418 self.lab.end_turn()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
419
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
420 def update_widgets(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
421 self._make_science_widgets()
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
422 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
423
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
424 def update(self):
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
425 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
426
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
427 def update_labels(self):
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
428 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
429 self.money.set_value(self.game.money)
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
430
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
431 def do_reset(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
432 self.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
433 self.lab.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
434 self.activity.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
435
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
436 def reset(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
437 for widget in self._sciences:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
438 widget.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
439
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
440
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
441 class LabWindow(Window):
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
442 """Window for the research lab"""
53
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
443
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
444 def __init__(self, screen):
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
445 super(LabWindow, self).__init__(screen)
81
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
446 self.screen = screen
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
447 self.game = Game()
53
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
448 exit = ExitGameButton()
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
449 self.add_child(exit)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
450 end_turn = EndTurnButton(self)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
451 self.add_child(end_turn)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
452 reset = ResetButton(self)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
453 self.add_child(reset)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
454
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
455 self.points = ValueLabel((10, 75), 'Available Human Resources')
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
456 self.add_child(self.points)
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
457 self.money = ValueLabel((310, 75), 'Money')
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
458 self.add_child(self.money)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
459
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
460 self.develop = DevelopmentWindow(screen, self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
461 self.activity = ActivityWindow(screen, self, self.develop)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
462 self.develop.set_activity_window(self.activity)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
463
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
464 devbut = SwitchWinButton((150, 10), 'Development', self.develop)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
465 self.add_child(devbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
466 actbut = SwitchWinButton((300, 10), 'Activities', self.activity)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
467 self.add_child(actbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
468
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
469 self._sciences = []
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
470 # Setup for the first turn
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
471 self.game.start_turn()
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
472 self.update_labels()
60
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
473 self._make_science_widgets()
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
474
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
475 def _make_science_widgets(self):
82
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
476 # FIXME: Horrible hackery
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
477 for widget in self._sciences:
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
478 self.remove_child(widget)
b0d97d51df51 Hook up simplistic equipment screen
Neil Muller <drnlmuller@gmail.com>
parents: 81
diff changeset
479 self._sciences = []
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
480 x = 0
57
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
481 y = 150
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
482 for science in self.game.lab.science:
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
483 if science.SCIENCE_TYPE == 'research':
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
484 widget = ScienceWidget(science, (x, y), self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
485 self.add_child(widget)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
486 self._sciences.append(widget)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
487 x += 200
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
488 if x >= WIDTH:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
489 x = 0
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
490 y += 100
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
491
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
492 def update_widgets(self):
60
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
493 self._make_science_widgets()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
494
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
495 def end_turn(self):
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
496 self.game.cur_missions = self.activity.get_mission_list()
80
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 72
diff changeset
497 messages = self.game.end_turn()
81
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
498 results = ResultsWindow(self.screen, messages, self.game.turn)
60
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
499 self.game.start_turn()
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
500 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
501 self.update_widgets()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
502 self.develop.update_widgets()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
503 self.activity.update_widgets()
81
59afe9f92383 Hook up basic results screen
Neil Muller <drnlmuller@gmail.com>
parents: 80
diff changeset
504 AddWindow.post(results)
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
505
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
506 def update(self):
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
507 self.update_labels()
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
508
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
509 def update_labels(self):
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
510 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
511 self.money.set_value(self.game.money)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
512
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
513 def do_reset(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
514 self.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
515 self.develop.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
516 self.activity.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
517
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
518 def reset(self):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
519 for widget in self._sciences:
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
520 widget.reset()