Mercurial > sypikslang

annotate gamelib/gamegui.py @ 80:a40a76012bd7

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Refactor message handling so we can deal with in the gui
author Neil Muller <drnlmuller@gmail.com>
date Wed, 09 May 2012 12:08:44 +0200
parents 8b06b07a1477
children 59afe9f92383
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
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
121 self.mode = 'NORMAL'
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
122 self.set_state('NORMAL')
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
123 else:
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
124 # select mission and equipment
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
125 self.mode = 'SELECTED'
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
126 self.set_state('SELECTED')
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
127
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
128 def selected(self):
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
129 return self.mode == 'SELECTED'
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
130
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
131 def get_equipment(self):
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
132 return self.equipment
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
133
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
134 def reset(self):
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
135 self.equipment = []
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
136 self.mode = 'NORMAL'
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
137 self.set_state('NORMAL')
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
138
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
139
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
140 class ValueLabel(TextLabel):
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
141
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
142 def __init__(self, pos, description):
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
143 self.description = description
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
144 rect = (pos[0], pos[1], 300, 50)
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
145 super(ValueLabel, self).__init__(rect,
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
146 '%s : 0' % description, font_medium, (255, 255, 0))
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
147
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
148 def set_value(self, value):
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
149 self.text = '%s : %d' % (self.description, value)
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
150 self._draw_text()
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
151
60
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
152
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
153 class ActivityWindow(Window):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
154
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
155 def __init__(self, screen, lab, develop):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
156 super(ActivityWindow, self).__init__(screen)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
157 self.lab = lab
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
158 self.develop = develop
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
159 self.game = lab.game
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
160 exitbut = ExitGameButton()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
161 self.add_child(exitbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
162 end_turn = EndTurnButton(self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
163 self.add_child(end_turn)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
164 reset = ResetButton(self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
165 self.add_child(reset)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
166
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
167 self.points = ValueLabel((10, 75), 'Available Human Resources')
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
168 self.add_child(self.points)
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
169 self.money = ValueLabel((310, 75), 'Money')
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
170 self.add_child(self.money)
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
171
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
172 labbut = SwitchWinButton((150, 10), 'Research', lab)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
173 self.add_child(labbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
174
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
175 devbut = SwitchWinButton((300, 10), 'Development', develop)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
176 self.add_child(devbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
177
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
178 self.update_labels()
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
179 self._missions = []
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
180 self._make_widgets()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
181
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
182 def _make_widgets(self):
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
183 x = 0
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
184 y = 150
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
185 for mission in self.game.get_available_missions():
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
186 widget = MissionWidget(mission, (x, y), self)
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
187 self._missions.append(widget)
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
188 self.add_child(widget)
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
189 x += 200
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
190 if x >= WIDTH:
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
191 x = 0
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
192 y += 100
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
193
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
194 def end_turn(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
195 # Drop back to the research screen
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
196 PopWindow.post()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
197 self.lab.update()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
198 AddWindow.post(self.lab)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
199 self.lab.end_turn()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
200
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
201 def update_widgets(self):
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
202 for widget in self._missions:
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
203 self.remove_child(widget)
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
204 self._missions = []
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
205 self._make_widgets()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
206
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
207 def update(self):
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
208 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
209
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
210 def update_labels(self):
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
211 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
212 self.money.set_value(self.game.money)
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
213
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
214 def do_reset(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
215 self.lab.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
216 self.develop.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
217 self.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
218
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
219 def get_mission_list(self):
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
220 selected_missions = []
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
221 for widget in self._missions:
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
222 if widget.selected():
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
223 equipment = widget.get_equipment()
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
224 mission = widget.mission
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
225 selected_missions.append((mission, equipment))
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
226 return selected_missions
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
227
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
228 def reset(self):
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
229 for widget in self._missions:
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
230 widget.reset()
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
231 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
232
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
233
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
234 class DevelopmentWindow(Window):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
235 """Window for handling schematics research"""
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
236
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
237 def __init__(self, screen, lab):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
238 super(DevelopmentWindow, self).__init__(screen)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
239 self.lab = lab
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
240 self.game = lab.game
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
241 exitbut = ExitGameButton()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
242 self.add_child(exitbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
243 end_turn = EndTurnButton(self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
244 self.add_child(end_turn)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
245 reset = ResetButton(self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
246 self.add_child(reset)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
247
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
248 self.points = ValueLabel((10, 75), 'Available Human Resources')
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
249 self.add_child(self.points)
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
250 self.money = ValueLabel((310, 75), 'Money')
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
251 self.add_child(self.money)
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
252
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
253 labbut = SwitchWinButton((150, 10), 'Research', lab)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
254 self.add_child(labbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
255 self.activity = None
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
256
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
257 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
258 self._sciences = []
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
259 self._make_science_widgets()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
260
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
261 def set_activity_window(self, activity):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
262 # Oh, what tangled webs we weave
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
263 if not self.activity:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
264 self.activity = activity
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
265 actbut = SwitchWinButton((300, 10), 'Activities', activity)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
266 self.add_child(actbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
267
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
268 def _make_science_widgets(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
269 x = 0
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
270 y = 150
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
271
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
272 for science in self.game.lab.science:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
273 if science.SCIENCE_TYPE == 'schematic':
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
274 widget = ScienceWidget(science, (x, y), self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
275 self.add_child(widget)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
276 self._sciences.append(widget)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
277 x += 200
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
278 if x >= WIDTH:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
279 x = 0
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
280 y += 100
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
281
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
282 def end_turn(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
283 # Drop back to the research screen
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
284 PopWindow.post()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
285 self.lab.update()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
286 AddWindow.post(self.lab)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
287 self.lab.end_turn()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
288
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
289 def update_widgets(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
290 for widget in self._sciences:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
291 self.remove_child(widget)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
292 self._sciences = []
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
293 self._make_science_widgets()
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
294 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
295
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
296 def update(self):
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
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
299 def update_labels(self):
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
300 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
301 self.money.set_value(self.game.money)
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
302
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
303 def do_reset(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
304 self.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
305 self.lab.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
306 self.activity.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
307
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
308 def reset(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
309 for widget in self._sciences:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
310 widget.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
311
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
312
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
313 class LabWindow(Window):
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
314 """Window for the research lab"""
53
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
315
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
316 def __init__(self, screen):
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 62
diff changeset
317 super(LabWindow, self).__init__(screen)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
318 self.game = Game()
53
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
319 exit = ExitGameButton()
655a6912e0ae Split gui stuff out of main.py
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
320 self.add_child(exit)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
321 end_turn = EndTurnButton(self)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
322 self.add_child(end_turn)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
323 reset = ResetButton(self)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
324 self.add_child(reset)
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
325
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
326 self.points = ValueLabel((10, 75), 'Available Human Resources')
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
327 self.add_child(self.points)
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
328 self.money = ValueLabel((310, 75), 'Money')
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
329 self.add_child(self.money)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
330
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
331 self.develop = DevelopmentWindow(screen, self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
332 self.activity = ActivityWindow(screen, self, self.develop)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
333 self.develop.set_activity_window(self.activity)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
334
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
335 devbut = SwitchWinButton((150, 10), 'Development', self.develop)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
336 self.add_child(devbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
337 actbut = SwitchWinButton((300, 10), 'Activities', self.activity)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
338 self.add_child(actbut)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
339
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
340 self._sciences = []
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
341 # Setup for the first turn
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
342 self.game.start_turn()
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
343 self.update_labels()
60
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
344 self._make_science_widgets()
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
345
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
346 def _make_science_widgets(self):
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
347 x = 0
57
9aa0252fb6e4 Tweak science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 56
diff changeset
348 y = 150
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
349
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
350 for science in self.game.lab.science:
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
351 if science.SCIENCE_TYPE == 'research':
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
352 widget = ScienceWidget(science, (x, y), self)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
353 self.add_child(widget)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
354 self._sciences.append(widget)
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
355 x += 200
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
356 if x >= WIDTH:
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
357 x = 0
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
358 y += 100
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
359
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
360 def update_widgets(self):
60
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
361 # FIXME: Horrible hackery
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
362 for widget in self._sciences:
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
363 self.remove_child(widget)
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
364 self._sciences = []
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
365 self._make_science_widgets()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
366
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
367 def end_turn(self):
72
8b06b07a1477 Very basic mission support
Neil Muller <drnlmuller@gmail.com>
parents: 67
diff changeset
368 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
369 messages = self.game.end_turn()
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 72
diff changeset
370 # FIXME: Hook this up to a report window or something
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 72
diff changeset
371 for msg_type, msg in messages:
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 72
diff changeset
372 if msg_type == NEW_SCIENCE or msg_type == NEW_SCHEMATIC:
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 72
diff changeset
373 print msg
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 72
diff changeset
374 elif msg_type == FAILURE:
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 72
diff changeset
375 print 'Mission failure: %s' % msg
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 72
diff changeset
376 elif msg_type == SUCCESS:
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 72
diff changeset
377 print 'Success: %s' % msg
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 72
diff changeset
378 else:
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 72
diff changeset
379 print 'Unhandled result %d: %s' % (msg_type, msg)
60
f9d2ba74723d Hack in turn progression
Neil Muller <drnlmuller@gmail.com>
parents: 59
diff changeset
380 self.game.start_turn()
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
381 self.update_labels()
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
382 self.update_widgets()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
383 self.develop.update_widgets()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
384 self.activity.update_widgets()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
385
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
386 def update(self):
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
387 self.update_labels()
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
388
67
06c8539df478 Tweak label handling. Show money as well
Neil Muller <drnlmuller@gmail.com>
parents: 66
diff changeset
389 def update_labels(self):
65
5c4c67673112 Display available points
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
390 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
391 self.money.set_value(self.game.money)
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
392
66
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
393 def do_reset(self):
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
394 self.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
395 self.develop.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
396 self.activity.reset()
43b34b013462 Add development and activity screens
Neil Muller <drnlmuller@gmail.com>
parents: 65
diff changeset
397
56
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
398 def reset(self):
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
399 for widget in self._sciences:
78dfd429b9a6 Start adding science buttons
Neil Muller <drnlmuller@gmail.com>
parents: 53
diff changeset
400 widget.reset()