Make scenes take the game state.

[naja.git] / naja / engine.py

diff --git a/naja/engine.py b/naja/engine.py
index d509ec20e2e914e958b172380c2f3842b8d6648c..cbe27ce0b9ffe7d56da3bc9814b111815a46d84b 100644 (file)
--- a/naja/engine.py
+++ b/naja/engine.py
@@ -1,15 +1,37 @@
 import pygame
 import pygame.locals as pgl
 
+from naja.constants import FPS
+from naja.events import SceneChangeEvent, QuitGameEvent
+
 
 class Engine(object):
-    def __init__(self, surface):
+    def __init__(self, surface, scene, state):
         self._surface = surface
+        self._scene = scene
+        self._state = state
 
     def run(self):
+        clock = pygame.time.Clock()
         while True:
+            self._surface.fill((0, 0, 0))
             for ev in pygame.event.get():
-                if ev.type == pgl.QUIT:
+                if ev.type == pgl.QUIT or QuitGameEvent.matches(ev):
+                    self.quit_game()
                     return
-            # TODO: work
+                elif SceneChangeEvent.matches(ev):
+                    self.change_scene(ev.scene_cls)
+                else:
+                    self._scene.handle_event(ev)
+            self._scene.render(self._surface)
             pygame.display.flip()
+            clock.tick(FPS)
+
+    def change_scene(self, scene_cls):
+        self._scene.exit()
+        self._scene = scene_cls(self._state)
+        self._scene.enter()
+
+    def quit_game(self):
+        self._scene.exit()
+        self._scene = None