source: gamelib/engine.py @ 312:dd1ffee5ccf5

Last change on this file since 312:dd1ffee5ccf5 was 312:dd1ffee5ccf5, checked in by Neil Muller <drnlmuller@…>, 11 years ago

Use different score tables fot the different modes. Refactor game modes code as a result

File size: 8.7 KB
Line 
1"""Game engine and states."""
2from pgu.engine import Game, State, Quit
3import pygame
4from pygame.locals import USEREVENT, QUIT, KEYDOWN, K_ESCAPE, K_n, K_d, K_s, K_i
5
6import gameboard
7import gameover
8import sound
9import constants
10import mainmenu
11import helpscreen
12
13class Engine(Game):
14    def __init__(self, main_app):
15        self.main_app = main_app
16        self.clock = pygame.time.Clock()
17        self.main_menu = mainmenu.make_main_menu()
18        self._open_window = None
19        self.scoreboard = gameover.ScoreTable()
20        self.gameboard = None
21
22    def tick(self):
23        """Tic toc."""
24        pygame.time.wait(10)
25
26    def open_window(self, window):
27        """Open a widget as the main window."""
28        if self._open_window is not None:
29            self.main_app.close(self._open_window)
30        self.main_app.open(window)
31        self._open_window = window
32
33    def create_game_board(self):
34        """Create and open a gameboard window."""
35        self.gameboard = gameboard.GameBoard(self.main_app)
36        self.open_window(self.gameboard.get_top_widget())
37
38    def set_main_menu(self):
39        """Open the main menu"""
40        self.open_window(self.main_menu)
41
42    def set_help_screen(self):
43        """Open the main menu"""
44        help_screen = helpscreen.make_help_screen()
45        self.open_window(help_screen)
46
47    def create_game_over(self):
48        """Create and open the Game Over window"""
49        for mode, days in constants.TURN_LIMITS.iteritems():
50            if days == constants.TURN_LIMIT:
51                game_over = gameover.create_game_over(self.gameboard,
52                        self.scoreboard[mode], mode)
53        self.gameboard = None
54        self.open_window(game_over)
55
56    def event(self, e):
57        if not Game.event(self, e):
58            if self.gameboard:
59                return self.gameboard.event(e)
60            return False
61        return True
62       
63
64class MainMenuState(State):
65    def init(self):
66        sound.stop_background_music()
67        self.game.set_main_menu()
68
69    def event(self, e):
70        if events_equal(e, START_DAY):
71            self.game.create_game_board()
72            return DayState(self.game)
73        elif events_equal(e, GO_HELP_SCREEN):
74            return HelpScreenState(self.game)
75        elif e.type is KEYDOWN:
76            if e.key == K_ESCAPE:
77                return Quit(self.game)
78            elif e.key == K_s:
79                self.game.create_game_board()
80                return DayState(self.game)
81            elif e.key == K_i:
82                return HelpScreenState(self.game)
83        elif e.type is not QUIT:
84            self.game.main_app.event(e)
85
86    def paint(self, screen):
87        screen.fill((0,0,0))
88        self.game.main_app.paint(screen)
89        pygame.display.flip()
90
91    def update(self, screen):
92        update = self.game.main_app.update(screen)
93        pygame.display.update(update)
94
95class HelpScreenState(State):
96    def init(self):
97        sound.stop_background_music()
98        self.game.set_help_screen()
99
100    def event(self, e):
101        if e.type is KEYDOWN and e.key == K_ESCAPE:
102            return MainMenuState(self.game)
103        elif events_equal(e, GO_MAIN_MENU):
104            return MainMenuState(self.game)
105        elif e.type is not QUIT:
106            self.game.main_app.event(e)
107
108    def paint(self, screen):
109        screen.fill((0,0,0))
110        self.game.main_app.paint(screen)
111        pygame.display.flip()
112
113    def update(self, screen):
114        update = self.game.main_app.update(screen)
115        pygame.display.update(update)
116
117class DayState(State):
118    def init(self):
119        """Add some chickens to the farm"""
120        sound.stop_background_music()
121        self.game.gameboard.start_day()
122
123        sound.play_sound("daybreak.ogg")
124        # disable timer
125        pygame.time.set_timer(MOVE_FOX_ID, 0)
126        pygame.time.set_timer(ANIM_ID, SLOW_ANIM_SPEED)
127        self.game.gameboard.advance_day()
128        self.game.gameboard.clear_foxes()
129        sound.background_music("daytime.ogg")
130        self.game.gameboard.hatch_eggs()
131
132    def event(self, e):
133        if events_equal(e, START_NIGHT):
134            self.game.gameboard.reset_states()
135            return NightState(self.game)
136        elif e.type is KEYDOWN and e.key == K_ESCAPE:
137            self.game.gameboard.reset_states()
138            return GameOver(self.game)
139        elif e.type is ANIM_ID:
140            self.game.gameboard.run_animations()
141        elif e.type is KEYDOWN and e.key == K_n:
142            return pygame.event.post(START_NIGHT)
143        elif events_equal(e, GO_MAIN_MENU):
144            return MainMenuState(self.game)
145        elif e.type is not QUIT:
146            self.game.main_app.event(e)
147
148    def paint(self, screen):
149        self.game.main_app.paint(screen)
150        pygame.display.flip()
151
152    def update(self, screen):
153        self.game.gameboard.update()
154        update = self.game.main_app.update(screen)
155        pygame.display.update(update)
156
157    def loop(self):
158        self.game.gameboard.loop()
159
160class NightState(State):
161    def init(self):
162        """Add some foxes to the farm"""
163        sound.stop_background_music()
164        self.game.gameboard.start_night()
165
166        sound.play_sound("nightfall.ogg")
167        # Add a timer to the event queue
168        self.cycle_count = 0
169        self.cycle_time = SLOW_ANIM_SPEED
170        pygame.time.set_timer(MOVE_FOX_ID, 4*self.cycle_time)
171        pygame.time.set_timer(ANIM_ID, self.cycle_time)
172        self.game.gameboard.spawn_foxes()
173        sound.background_music("nighttime.ogg")
174
175        self.game.gameboard.lay_eggs()
176
177    def event(self, e):
178        if events_equal(e, START_DAY):
179            if self.game.gameboard.is_game_over():
180                return GameOver(self.game)
181            return DayState(self.game)
182        elif e.type is KEYDOWN and e.key == K_d:
183            if self.cycle_time > FAST_ANIM_SPEED:
184                self.cycle_time = FAST_ANIM_SPEED
185            else:
186                self.cycle_time = SLOW_ANIM_SPEED
187            pygame.time.set_timer(ANIM_ID, self.cycle_time)
188            pygame.time.set_timer(MOVE_FOX_ID, 4*self.cycle_time)
189        elif e.type is KEYDOWN and e.key == K_ESCAPE:
190            self.game.gameboard.set_cursor()
191            return GameOver(self.game)
192        elif e.type is ANIM_ID:
193            self.game.gameboard.run_animations()
194        elif e.type is MOVE_FOX_ID:
195            # Timer aren't nessecairly ordered, so we make sure
196            # we don't get a ANIM event until at least cycle after this
197            pygame.time.set_timer(ANIM_ID, self.cycle_time)
198            # Ensure any outstanding animitions get cleaned up
199            self.game.gameboard.run_animations()
200            self.cycle_count += 1
201            if self.cycle_count > constants.NIGHT_LENGTH:
202                return pygame.event.post(START_DAY)
203            if self.game.gameboard.move_foxes():
204                # All foxes are gone/safe, so dawn happens
205                return pygame.event.post(START_DAY)
206        elif e.type is not QUIT:
207            self.game.main_app.event(e)
208
209    def loop(self):
210        self.game.gameboard.loop()
211
212    def paint(self, screen):
213        self.game.main_app.paint(screen)
214        pygame.display.flip()
215
216    def update(self, screen):
217        self.game.gameboard.update()
218        update = self.game.main_app.update(screen)
219        pygame.display.update(update)
220
221class GameOver(State):
222    def init(self):
223        """Setup everything"""
224        sound.stop_background_music()
225        self.game.create_game_over()
226        pygame.time.set_timer(MOVE_FOX_ID, 0)
227        pygame.time.set_timer(ANIM_ID, 0)
228
229    def event(self, e):
230        if e.type is KEYDOWN:
231            if e.key == K_ESCAPE:
232                return MainMenuState(self.game)
233        elif events_equal(e, GO_MAIN_MENU):
234            return MainMenuState(self.game)
235        elif e.type is not QUIT:
236            self.game.main_app.event(e)
237
238    def paint(self, screen):
239        screen.fill((0,0,0))
240        self.game.main_app.paint(screen)
241        pygame.display.flip()
242
243    def update(self, screen):
244        update = self.game.main_app.update(screen)
245        pygame.display.update(update)
246
247# pygame events
248
249def events_equal(e1, e2):
250    """Compare two user events."""
251    return (e1.type is e2.type and e1.name == e2.name)
252
253START_DAY = pygame.event.Event(USEREVENT, name="START_DAY")
254START_NIGHT = pygame.event.Event(USEREVENT, name="START_NIGHT")
255GO_MAIN_MENU = pygame.event.Event(USEREVENT, name="GO_MAIN_MENU")
256GO_HELP_SCREEN = pygame.event.Event(USEREVENT, name="GO_HELP_SCREEN")
257MOVE_FOX_ID = USEREVENT + 1
258ANIM_ID = USEREVENT + 6
259MOVE_FOXES = pygame.event.Event(MOVE_FOX_ID, name="MOVE_FOXES")
260QUIT = pygame.event.Event(QUIT)
261
262# Due to the way pgu's loop timing works, these will only get proceesed
263# at intervals of 10ms, so there's no point in them not being multiples of 10
264FAST_ANIM_SPEED=20
265SLOW_ANIM_SPEED=50
Note: See TracBrowser for help on using the repository browser.