center robot again
[naja.git] / naja / engine.py
1 import pygame
2 import pygame.locals as pgl
3
4 from naja.constants import FPS
5 from naja.events import SceneChangeEvent, QuitGameEvent
6
7
8 class Engine(object):
9     def __init__(self, surface, scene, state):
10         self._surface = surface
11         self._scene = scene
12         self._state = state
13
14     def run(self):
15         clock = pygame.time.Clock()
16         while True:
17             self._surface.fill((0, 0, 0))
18             for ev in pygame.event.get():
19                 if ev.type == pgl.QUIT or QuitGameEvent.matches(ev):
20                     self.quit_game()
21                     return
22                 elif SceneChangeEvent.matches(ev):
23                     self.change_scene(ev.scene_cls)
24                 else:
25                     self._scene.handle_event(ev)
26             self._scene.render(self._surface)
27             pygame.display.flip()
28             clock.tick(FPS)
29
30     def change_scene(self, scene_cls):
31         self._scene.exit()
32         self._scene = scene_cls(self._state)
33         self._scene.enter()
34
35     def quit_game(self):
36         self._scene.exit()
37         self._scene = None