X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fscenes%2Fgame.py;h=c0b4b270ffd5cca556309d28406854e4508e11e4;hb=d56932610ab443154ece192dac89b9a16de73eaa;hp=9d8b29046a5c8b7ee81f79b5702a5b994973519c;hpb=baa600e7e8557e51c6deddff16988a5e1779f379;p=naja.git

diff --git a/naja/scenes/game.py b/naja/scenes/game.py
index 9d8b290..c0b4b27 100644
--- a/naja/scenes/game.py
+++ b/naja/scenes/game.py
@@ -3,13 +3,18 @@ Gameboard scene.
 """
 
 import pygame.locals as pgl
+import random
 
-from naja.events import SceneChangeEvent
+from naja.constants import KEYS, TUNES
+from naja.events import SceneChangeEvent, finish_event
+from naja.options import options
 from naja.scenes.scene import Scene
+from naja.sound import sound
 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 +24,40 @@ class GameScene(Scene):
 
     def __init__(self, state):
         super(GameScene, self).__init__(state)
-        self.widgets.append(PlayerBitsWidget((0, 0)))
-        self.widgets.append(BoardWidget((0, 60)))
-        self.widgets.append(GameBitsWidget((0, 540)))
-        self.widgets.append(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))
+        sound.play_sound('startup.ogg')
+        background_track = random.choice(TUNES)
+        sound.play_music(background_track, 0.25)
 
-    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(MenuScene)
-            return
+        if ev.type == pgl.KEYDOWN:
+            if ev.key in KEYS.QUIT:
+                sound.stop()
+                SceneChangeEvent.post(MenuScene)
+                return
+            if options.cheat_enabled:
+                if ev.key == pgl.K_0:
+                    self.state.player.bits.toggle_bit(0)
+                elif ev.key == pgl.K_1:
+                    self.state.player.bits.toggle_bit(1)
+                elif ev.key == pgl.K_2:
+                    self.state.player.bits.toggle_bit(2)
+                elif ev.key == pgl.K_3:
+                    self.state.player.bits.toggle_bit(3)
+                elif ev.key == pgl.K_4:
+                    self.state.player.bits.toggle_bit(4)
+                elif ev.key == pgl.K_5:
+                    self.state.player.bits.toggle_bit(5)
+                elif ev.key == pgl.K_6:
+                    self.state.player.bits.toggle_bit(6)
+                elif ev.key == pgl.K_7:
+                    self.state.player.bits.toggle_bit(7)
+                elif ev.key == pgl.K_EQUALS:
+                    self.state.gameboard.gain_health()
+                return finish_event()