X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fscenes%2Fgame.py;h=c0f52a8be47b9eff08aa4568da6102ddf123207e;hb=0cef06f581d48315a59a19ac06ca1d2be14cbb4c;hp=2b6bbf2481dd8065a0e0392df12bf1c8876fb075;hpb=ee32cda1beebfdc07234f393bf39bb948b36a39a;p=naja.git

diff --git a/naja/scenes/game.py b/naja/scenes/game.py
index 2b6bbf2..c0f52a8 100644
--- a/naja/scenes/game.py
+++ b/naja/scenes/game.py
@@ -4,12 +4,14 @@ Gameboard scene.
 
 import pygame.locals as pgl
 
+from naja.constants import KEYS
 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.widgets.robot import RobotWidget
 
 
 class GameScene(Scene):
@@ -19,13 +21,15 @@ class GameScene(Scene):
 
     def __init__(self, state):
         super(GameScene, self).__init__(state)
-        self.add(PlayerBitsWidget((0, 0)))
-        self.add(BoardWidget((0, 60), state))
-        self.add(GameBitsWidget((0, 540)))
-        self.add(InfoAreaWidget((480, 0)))
+        self.add(PlayerBitsWidget((0, 0), state))
+        info = InfoAreaWidget((480, 0), state)
+        self.add(BoardWidget((0, 60), state, info))
+        self.add(GameBitsWidget((0, 540), state))
+        self.add(info)
+        self.add(RobotWidget(state))
 
     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):
+        if ev.type == pgl.KEYDOWN and ev.key in KEYS.QUIT:
             SceneChangeEvent.post(MenuScene)
             return