X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=tabakrolletjie%2Fscenes%2Fmenu.py;h=89b531e2621d78f69ec75e7869fb880104c5e3b5;hb=503ee94a5c586cad2d9a79d140c15f25a8a1bac1;hp=33bb0ef8ae393d6f9d12e788c2ee3e5a5f76cef2;hpb=589be35175056d9b7307b0a1f4c310110ab2980e;p=tabakrolletjie.git

diff --git a/tabakrolletjie/scenes/menu.py b/tabakrolletjie/scenes/menu.py
index 33bb0ef..89b531e 100644
--- a/tabakrolletjie/scenes/menu.py
+++ b/tabakrolletjie/scenes/menu.py
@@ -4,11 +4,64 @@ 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, SpacerButton
+from ..utils import save_file_exists, get_save_data
 
 
 class MenuScene(BaseScene):
+    def enter(self, gamestate):
+        if gamestate.station is None:
+            print "Loading Station Alpha ..."
+            gamestate.set_station(loader.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._cont_but = TextButton("Continue / Retry", (255, 255, 255),
+                                    name='continue')
+        if not save_file_exists():
+            self._cont_but.disable()
+        self._menu = [
+            TextButton("New Game", (255, 255, 255), name='new game'),
+            self._cont_but,
+            TextButton("Help", (255, 255, 255), name='help'),
+            SpacerButton((50, 30)),
+            TextButton("Quit", (255, 255, 255), name='quit'),
+        ]
+
     def render(self, surface, gamestate):
-        surface.fill((0, 255, 0))
+        surface.fill((0, 128, 0))
+
+        pos = ((surface.get_width() - self._title.get_width()) / 2, 50)
+        surface.blit(self._title, pos, None)
+
+        height = 150
+        for item in self._menu:
+            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 button in self._menu:
+            if button.pressed(ev):
+                return button.name
+        return None
+
+    def _do_load_save(self, gamestate):
+        gamestate.load_from_json(get_save_data())
+        from .day import DayScene
+        SceneChangeEvent.post(scene=DayScene())
+
+    def _start_new_game(self):
+        from .load_level import LoadLevelScene
+        SceneChangeEvent.post(scene=LoadLevelScene())
+
+    def _do_help(self):
+        from .help import HelpScene
+        SceneChangeEvent.post(scene=HelpScene())
 
     def event(self, ev, gamestate):
         if ev.type == pgl.KEYDOWN:
@@ -18,8 +71,20 @@ class MenuScene(BaseScene):
                 from .night import NightScene
                 SceneChangeEvent.post(scene=NightScene())
             elif ev.key == pgl.K_d:
-                from .day import DayScene
-                SceneChangeEvent.post(scene=DayScene())
+                self._do_day()
             elif ev.key == pgl.K_l:
-                print "Loading Station Alpha ..."
-                gamestate.load_station("station-alpha.json")
+                self._start_new_game()
+            elif ev.key == pgl.K_h:
+                self._do_help()
+        elif ev.type == pgl.MOUSEBUTTONDOWN:
+            pressed = self._get_pressed(ev)
+            if pressed == 'new game':
+                self._start_new_game()
+            elif pressed == 'continue':
+                self._do_load_save(gamestate)
+            elif pressed == 'help':
+                self._do_help()
+            elif pressed == 'quit':
+                QuitEvent.post()
+            elif pressed:
+                print 'Pressed', pressed