Render bit pattern markers on tile widgets

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

diff --git a/naja/scenes/menu.py b/naja/scenes/menu.py
index 27b7e9c23b629235c21decfc774ba0578d3d4134..46b2b3b13eab19a19e18ea17d8a9f8ea4c1b2998 100644 (file)
--- a/naja/scenes/menu.py
+++ b/naja/scenes/menu.py
@@ -2,10 +2,40 @@
 Main menu scene.
 """
 
-from .scene import Scene
+import pygame.locals as pgl
+
+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):
-    """
-    Main menu.
-    """
+
+    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
+
+    def handle_scene_event(self, ev):
+        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)