tabakrolletjie/scenes/menu.py

   1 """ Menu scene. """
   2
   3 import pygame.locals as pgl
   4
   5 from .base import BaseScene
   6 from ..events import QuitEvent, SceneChangeEvent
   7 from ..loader import loader
   8 from ..constants import FONTS
   9 from ..widgets import TextButton, SpacerButton
  10 from ..utils import save_file_exists, get_save_data
  11
  12
  13 class MenuScene(BaseScene):
  14     def enter(self, gamestate):
  15         if gamestate.station is None:
  16             print "Loading Station Alpha ..."
  17             gamestate.set_station(loader.load_station("station-alpha.json"))
  18         font_title = loader.load_font(FONTS['bold'], size=32)
  19         self._title = font_title.render('A Game with a title', True,
  20                                         (255, 255, 255))
  21         self._cont_but = TextButton("Continue / Retry", (255, 255, 255),
  22                                     name='continue')
  23         if not save_file_exists():
  24             self._cont_but.disable()
  25         self._menu = [
  26             TextButton("New Game", (255, 255, 255), name='new game'),
  27             self._cont_but,
  28             TextButton("Help", (255, 255, 255), name='help'),
  29             SpacerButton((50, 30)),
  30             TextButton("Quit", (255, 255, 255), name='quit'),
  31         ]
  32
  33     def render(self, surface, gamestate):
  34         surface.fill((0, 128, 0))
  35
  36         pos = ((surface.get_width() - self._title.get_width()) / 2, 50)
  37         surface.blit(self._title, pos, None)
  38
  39         height = 150
  40         for item in self._menu:
  41             if not item.position:
  42                 item.position = ((surface.get_width() - item.get_width()) / 2,
  43                                  height)
  44             item.render(surface)
  45             height += item.get_height() + 30
  46
  47     def _get_pressed(self, ev):
  48         for button in self._menu:
  49             if button.pressed(ev):
  50                 return button.name
  51         return None
  52
  53     def _do_load_save(self, gamestate):
  54         gamestate.load_from_json(get_save_data())
  55         from .day import DayScene
  56         SceneChangeEvent.post(scene=DayScene())
  57
  58     def _start_new_game(self):
  59         from .load_level import LoadLevelScene
  60         SceneChangeEvent.post(scene=LoadLevelScene())
  61
  62     def _do_help(self):
  63         from .help import HelpScene
  64         SceneChangeEvent.post(scene=HelpScene())
  65
  66     def event(self, ev, gamestate):
  67         if ev.type == pgl.KEYDOWN:
  68             if ev.key in (pgl.K_q, pgl.K_ESCAPE):
  69                 QuitEvent.post()
  70             elif ev.key == pgl.K_n:
  71                 from .night import NightScene
  72                 SceneChangeEvent.post(scene=NightScene())
  73             elif ev.key == pgl.K_d:
  74                 self._do_day()
  75             elif ev.key == pgl.K_l:
  76                 self._start_new_game()
  77             elif ev.key == pgl.K_h:
  78                 self._do_help()
  79         elif ev.type == pgl.MOUSEBUTTONDOWN:
  80             pressed = self._get_pressed(ev)
  81             if pressed == 'new game':
  82                 self._start_new_game()
  83             elif pressed == 'continue':
  84                 self._do_load_save(gamestate)
  85             elif pressed == 'help':
  86                 self._do_help()
  87             elif pressed == 'quit':
  88                 QuitEvent.post()
  89             elif pressed:
  90                 print 'Pressed', pressed