Move keys to constants

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

diff --git a/naja/scenes/menu.py b/naja/scenes/menu.py
index 7b42ad7b730d90959f078e89c4ffe98ed13207d5..91f858b0d6304ef62e55e6142c40e9737b66d5f0 100644 (file)
--- a/naja/scenes/menu.py
+++ b/naja/scenes/menu.py
@@ -2,17 +2,37 @@
 Main menu scene.
 """
 
+import pygame.locals as pgl
+
+from naja.constants import KEYS
 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
 
 
 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, 200), 'Credits', fontsize=32, colour='white')
+        credits.add_callback('click',
+                             lambda event: SceneChangeEvent.post(CreditsScene))
+        selector.add(credits)
+        quit = TextWidget((100, 300), 'Quit', fontsize=32, colour='white')
+        quit.add_callback('click', lambda event: QuitGameEvent.post())
+        selector.add(quit)
 
-    def __init__(self):
-        self.tw = TextWidget((10, 10), 'Haai', colour=(255, 255, 255))
-
-    def render(self, surface):
-        self.tw.render(surface)
-
-    def handle_event(self, ev):
-        pass
+    def handle_scene_event(self, ev):
+        if ev.type == pgl.KEYDOWN:
+            if ev.key in KEYS.QUIT:
+                QuitGameEvent.post()
+                return