X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fscenes%2Fmenu.py;h=e0383b5ca3159a3d3abd63b4a0e04f56cb2c4107;hb=89e77ffa7dc2f2485049763b780fd5519541daa7;hp=b17f611234253b5ad9b69f1aacd1aebcf34eb5d9;hpb=9299925383d4746e7f9c4e948e083a1bff1e339a;p=naja.git

diff --git a/naja/scenes/menu.py b/naja/scenes/menu.py
index b17f611..e0383b5 100644
--- a/naja/scenes/menu.py
+++ b/naja/scenes/menu.py
@@ -4,26 +4,44 @@ Main menu scene.
 
 import pygame.locals as pgl
 
+from naja.constants import KEYS
+from naja.events import SceneChangeEvent, QuitGameEvent
 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.events import SceneChangeEvent
+from naja.scenes.game import GameScene
+from naja.scenes.load_save import SaveGameScene
+from naja.widgets.selector import SelectorWidget
+from naja.widgets.text import TextWidget
+
 
 class MenuScene(Scene):
+    def __init__(self, state):
+        super(MenuScene, self).__init__(state)
+        selector = SelectorWidget()
+        self.add(selector)
+
+        run_game = TextWidget((100, 100), 'Game', fontsize=32, colour='white')
+        run_game.add_callback(
+            'click', lambda event: SceneChangeEvent.post(GameScene))
+        selector.add(run_game)
+
+        credits = TextWidget(
+            (100, 150), 'Credits', fontsize=32, colour='white')
+        credits.add_callback(
+            'click', lambda event: SceneChangeEvent.post(CreditsScene))
+        selector.add(credits)
+
+        save = TextWidget((100, 250), 'Save', fontsize=32, colour='white')
+        save.add_callback(
+            'click', lambda event: SceneChangeEvent.post(SaveGameScene))
+        selector.add(save)
+
+        quit = TextWidget((100, 300), 'Quit', fontsize=32, colour='white')
+        quit.add_callback('click', lambda event: QuitGameEvent.post())
+        selector.add(quit)
 
-    def __init__(self):
-        super(MenuScene, self).__init__()
-        self.widgets.append(TextWidget((10, 10), 'Haai', fontsize=32,
-        self.widgets.append(TextWidget((10, 10), 'Naja',
-                            colour=(255, 255, 255)))
-        self.credits_menu = CreditsScene()
-        self.credits_menu.set_base_menu(self)
-
-    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:
-            SceneChangeEvent.post(self.credits_menu)
-            return
+    def handle_scene_event(self, ev):
+        if ev.type == pgl.KEYDOWN:
+            if ev.key in KEYS.QUIT:
+                QuitGameEvent.post()
+                return