Add a 'How To Play' screen

[naja.git] / naja / scenes / menu.py

diff --git a/naja/scenes/menu.py b/naja/scenes/menu.py
index 46b2b3b13eab19a19e18ea17d8a9f8ea4c1b2998..a94a0edd5e85342429702fbc6f700abf341adf6b 100644 (file)
--- a/naja/scenes/menu.py
+++ b/naja/scenes/menu.py
@@ -4,38 +4,73 @@ Main menu scene.
 
 import pygame.locals as pgl
 
+from naja.constants import KEYS, PALETTE
+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.scenes.game import GameScene
-from naja.events import SceneChangeEvent
+from naja.scenes.howto import HowtoScene
+from naja.scenes.load_save import LoadGameScene, SaveGameScene
+from naja.scenes.new_game import NewGameScene
+from naja.widgets.selector import SelectorWidget
+from naja.widgets.text import TextWidget
 
 
 class MenuScene(Scene):
-
     def __init__(self, state):
         super(MenuScene, self).__init__(state)
-        self.add(TextWidget((10, 10), 'Game', fontsize=32, colour='white'))
-        self.add(TextWidget((10, 40), 'Credits', fontsize=32, colour='white'))
-        self.add(TextWidget((10, 70), 'Quit', fontsize=32, colour='white'))
-        self.pos = 0
+        selector = SelectorWidget()
+        self.add(selector)
+        y_offset = 0
+
+        y_offset += 50
+        resume = TextWidget(
+            (100, y_offset), 'Resume Game', colour=PALETTE.WHITE,
+            unselectable_colour=PALETTE.GREY)
+        resume.add_callback('click', self.scene_callback(GameScene))
+        resume.set_selectable_callback(lambda: state is not None)
+        selector.add(resume)
+
+        y_offset += 50
+        new = TextWidget((100, y_offset), 'New Game', colour=PALETTE.WHITE)
+        new.add_callback('click', self.scene_callback(NewGameScene))
+        selector.add(new)
+
+        y_offset += 50
+        load = TextWidget((100, y_offset), 'Load', colour=PALETTE.WHITE)
+        load.add_callback('click', self.scene_callback(LoadGameScene))
+        selector.add(load)
+
+        y_offset += 50
+        save = TextWidget(
+            (100, y_offset), 'Save', colour=PALETTE.WHITE,
+            unselectable_colour=PALETTE.GREY)
+        save.add_callback('click', self.scene_callback(SaveGameScene))
+        save.set_selectable_callback(lambda: state is not None)
+        selector.add(save)
+
+        y_offset += 50
+        howto = TextWidget(
+            (100, y_offset), 'How To Play', fontsize=32, colour='white')
+        howto.add_callback(
+            'click', lambda event: SceneChangeEvent.post(HowtoScene))
+        selector.add(howto)
+
+        y_offset += 50
+        credits = TextWidget((100, y_offset), 'Credits', colour=PALETTE.WHITE)
+        credits.add_callback('click', self.scene_callback(CreditsScene))
+        selector.add(credits)
+
+        y_offset += 50
+        quit = TextWidget((100, y_offset), 'Quit', colour=PALETTE.WHITE)
+        quit.add_callback('click', lambda event: QuitGameEvent.post())
+        selector.add(quit)
+
+    def scene_callback(self, scene_class):
+        return lambda event: SceneChangeEvent.post(scene_class)
 
     def handle_scene_event(self, ev):
         if ev.type == pgl.KEYDOWN:
-            if ev.key in (pgl.K_q, pgl.K_ESCAPE):
+            if ev.key in KEYS.QUIT:
                 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)