X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fscenes%2Fmenu.py;h=a73c2ecb498ff94d5d76a2e9bd7d06a08ff3fa82;hb=b3db4400695249420f5f941744c2a2692a1bad98;hp=96c8c7a5677e8374458a1f2d1694d0209f6d274a;hpb=0f233e6d450d863f06f5acddee3bd94f53594d08;p=naja.git

diff --git a/naja/scenes/menu.py b/naja/scenes/menu.py
index 96c8c7a..a73c2ec 100644
--- a/naja/scenes/menu.py
+++ b/naja/scenes/menu.py
@@ -8,20 +8,37 @@ from naja.scenes.scene import Scene
 from naja.widgets.text import TextWidget
 from naja.events import QuitGameEvent
 from naja.scenes.credits import CreditsScene
+from naja.scenes.game import GameScene
 from naja.events import SceneChangeEvent
 
+
 class MenuScene(Scene):
 
-    def __init__(self):
-        super(MenuScene, self).__init__()
-        self.widgets.append(TextWidget((10, 10), 'Haai', fontsize=32,
+    def __init__(self, state):
+        super(MenuScene, self).__init__(state)
+        self.widgets.append(TextWidget((10, 10), 'Game', fontsize=32,
+                            colour='white'))
+        self.widgets.append(TextWidget((10, 40), 'Credits', fontsize=32,
+                            colour='white'))
+        self.widgets.append(TextWidget((10, 70), 'Quit', fontsize=32,
                             colour='white'))
+        self.pos = 0
 
     def handle_event(self, ev):
-        if ev.type == pgl.KEYUP and ev.key in (pgl.K_q, pgl.K_ESCAPE):
-            QuitGameEvent.post()
-            return
-        if ev.type == pgl.KEYUP and ev.key is pgl.K_RETURN:
-            credits_menu = CreditsScene()
-            SceneChangeEvent.post(credits_menu)
-            return
+        if ev.type == pgl.KEYDOWN:
+            if ev.key in (pgl.K_q, pgl.K_ESCAPE):
+                QuitGameEvent.post()
+                return
+            if ev.key == pgl.K_DOWN:
+                self.pos = (self.pos + 1) % 3
+            elif ev.key == pgl.K_UP:
+                self.pos = (self.pos - 1) % 3
+            elif ev.key == pgl.K_RETURN:
+                if self.pos == 2:
+                    QuitGameEvent.post()
+                    return
+                elif self.pos == 0:
+                    scene = GameScene
+                elif self.pos == 1:
+                    scene = CreditsScene
+                SceneChangeEvent.post(scene)