widgets start arriving

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

diff --git a/tabakrolletjie/scenes/menu.py b/tabakrolletjie/scenes/menu.py
index e7092d58637889e3941e939841cfad24e1a6d52b..e87005491f9fd79867b1fbc30b5c9242f298a08c 100644 (file)
--- a/tabakrolletjie/scenes/menu.py
+++ b/tabakrolletjie/scenes/menu.py
@@ -1,4 +1,71 @@
 """ Menu scene. """
 
-class MenuScene(object):
-    pass
+import pygame.locals as pgl
+
+from .base import BaseScene
+from ..events import QuitEvent, SceneChangeEvent
+from ..loader import loader
+from ..constants import FONTS
+from ..widgets import TextButton
+
+
+class MenuScene(BaseScene):
+    def enter(self, gamestate):
+        if gamestate.station is None:
+            print "Loading Station Alpha ..."
+            gamestate.load_station("station-alpha.json")
+        font_title = loader.load_font(FONTS['bold'], size=32)
+        self._title = font_title.render('A Game with a title', True,
+                                        (255, 255, 255))
+        self._menu = {
+            'load level': TextButton("Load Level", (255, 255, 255)),
+            'saved game': TextButton("Load Saved Game", (255, 255, 255)),
+            'start game': TextButton("Start Game (Day)", (255, 255, 255)),
+        }
+
+    def render(self, surface, gamestate):
+        surface.fill((0, 128, 0))
+
+        pos = ((surface.get_width() - self._title.get_width()) / 2, 50)
+        surface.blit(self._title, pos, None)
+
+        height = 150
+        for label in sorted(self._menu):
+            item = self._menu[label]
+            if not item.position:
+                item.position = ((surface.get_width() - item.get_width()) / 2,
+                                 height)
+            item.render(surface)
+            height += item.get_height() + 30
+
+    def _get_pressed(self, ev):
+        for label, button in self._menu.items():
+            if button.pressed(ev):
+                return label
+        return None
+
+    def _do_day(self):
+        from .day import DayScene
+        SceneChangeEvent.post(scene=DayScene())
+
+    def event(self, ev, gamestate):
+        if ev.type == pgl.KEYDOWN:
+            if ev.key in (pgl.K_q, pgl.K_ESCAPE):
+                QuitEvent.post()
+            elif ev.key == pgl.K_n:
+                from .night import NightScene
+                SceneChangeEvent.post(scene=NightScene())
+            elif ev.key == pgl.K_d:
+                self._do_day()
+            elif ev.key == pgl.K_l:
+                print "Loading Station Alpha ..."
+                gamestate.load_station("station-alpha.json")
+        elif ev.type == pgl.MOUSEBUTTONDOWN:
+            pressed = self._get_pressed(ev)
+            if pressed == 'load level':
+                print "Loading Station Alpha ..."
+                gamestate.load_station("station-alpha.json")
+            elif pressed == 'start game':
+                self._do_day()
+            elif pressed:
+                print 'Pressed', pressed