X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fscenes%2Fgame.py;h=1f95a77402c53333d9a82489d084022d103b09fb;hb=4df5d499b14782109585a95f6c96d56963fb8231;hp=48625007fb34934711ce4cd497567987f8c383fe;hpb=c783c52af634dc5a11dc25ac45f2ad79f0d3d62d;p=naja.git

diff --git a/naja/scenes/game.py b/naja/scenes/game.py
index 4862500..1f95a77 100644
--- a/naja/scenes/game.py
+++ b/naja/scenes/game.py
@@ -4,15 +4,13 @@ Gameboard scene.
 
 import pygame.locals as pgl
 
-from .scene import Scene
-from .menu import MenuScene
-
+from naja.events import SceneChangeEvent
+from naja.scenes.scene import Scene
 from naja.widgets.board import BoardWidget
 from naja.widgets.player_bits import PlayerBitsWidget
 from naja.widgets.game_bits import GameBitsWidget
 from naja.widgets.info_area import InfoAreaWidget
-
-from naja.events import SceneChangeEvent
+from naja.widgets.robot import RobotWidget
 
 
 class GameScene(Scene):
@@ -20,14 +18,16 @@ class GameScene(Scene):
     Gameboard scene.
     """
 
-    def __init__(self):
-        super(GameScene, self).__init__()
-        self.widgets.append(PlayerBitsWidget((0, 0)))
-        self.widgets.append(BoardWidget((0, 60)))
-        self.widgets.append(GameBitsWidget((0, 540)))
-        self.widgets.append(InfoAreaWidget((480, 0)))
+    def __init__(self, state):
+        super(GameScene, self).__init__(state)
+        self.add(PlayerBitsWidget((0, 0), state))
+        self.add(BoardWidget((0, 60), state))
+        self.add(GameBitsWidget((0, 540), state))
+        self.add(InfoAreaWidget((480, 0)))
+        self.add(RobotWidget(state))
 
-    def handle_event(self, ev):
+    def handle_scene_event(self, ev):
+        from naja.scenes.menu import MenuScene
         if ev.type == pgl.KEYUP and ev.key in (pgl.K_q, pgl.K_ESCAPE):
-            SceneChangeEvent.post(scene=MenuScene())
+            SceneChangeEvent.post(MenuScene)
             return