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