X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fengine.py;h=cbe27ce0b9ffe7d56da3bc9814b111815a46d84b;hb=67e79ba3fbe96590efa50a73fe983c2810efdd0f;hp=a8e01460c8b345336f874f0c5fc104b7bda26289;hpb=7d477cf85abea0402d0a41fe81d912a57a107e27;p=naja.git

diff --git a/naja/engine.py b/naja/engine.py
index a8e0146..cbe27ce 100644
--- a/naja/engine.py
+++ b/naja/engine.py
@@ -6,9 +6,10 @@ from naja.events import SceneChangeEvent, QuitGameEvent
 
 
 class Engine(object):
-    def __init__(self, surface, scene):
+    def __init__(self, surface, scene, state):
         self._surface = surface
         self._scene = scene
+        self._state = state
 
     def run(self):
         clock = pygame.time.Clock()
@@ -19,16 +20,16 @@ class Engine(object):
                     self.quit_game()
                     return
                 elif SceneChangeEvent.matches(ev):
-                    self.change_scene(ev.scene)
+                    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):
+    def change_scene(self, scene_cls):
         self._scene.exit()
-        self._scene = scene
+        self._scene = scene_cls(self._state)
         self._scene.enter()
 
     def quit_game(self):