Saved games can now be loaded as well.

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

diff --git a/naja/scenes/menu.py b/naja/scenes/menu.py
index 91f858b0d6304ef62e55e6142c40e9737b66d5f0..a06e061f0d7c03afec99b22122841420f2d1f7d8 100644 (file)
--- a/naja/scenes/menu.py
+++ b/naja/scenes/menu.py
@@ -5,13 +5,13 @@ 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.widgets.selector import SelectorWidget
-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.load_save import LoadGameScene, SaveGameScene
+from naja.widgets.selector import SelectorWidget
+from naja.widgets.text import TextWidget
 
 
 class MenuScene(Scene):
@@ -19,14 +19,28 @@ class MenuScene(Scene):
         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))
+        run_game.add_callback(
+            'click', lambda event: SceneChangeEvent.post(GameScene))
         selector.add(run_game)
-        credits = TextWidget((100, 200), 'Credits', fontsize=32, colour='white')
-        credits.add_callback('click',
-                             lambda event: SceneChangeEvent.post(CreditsScene))
+
+        credits = TextWidget(
+            (100, 150), 'Credits', fontsize=32, colour='white')
+        credits.add_callback(
+            'click', lambda event: SceneChangeEvent.post(CreditsScene))
         selector.add(credits)
+
+        load = TextWidget((100, 200), 'Load', fontsize=32, colour='white')
+        load.add_callback(
+            'click', lambda event: SceneChangeEvent.post(LoadGameScene))
+        selector.add(load)
+
+        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)