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
  10
  11
  12 class MenuScene(BaseScene):
  13     def enter(self, gamestate):
  14         if gamestate.station is None:
  15             print "Loading Station Alpha ..."
  16             gamestate.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._menu = {
  21             'load level': TextButton("Load Level", (255, 255, 255)),
  22             'saved game': TextButton("Load Saved Game", (255, 255, 255)),
  23             'start game': TextButton("Start Game (Day)", (255, 255, 255)),
  24         }
  25
  26     def render(self, surface, gamestate):
  27         surface.fill((0, 128, 0))
  28
  29         pos = ((surface.get_width() - self._title.get_width()) / 2, 50)
  30         surface.blit(self._title, pos, None)
  31
  32         height = 150
  33         for label in sorted(self._menu):
  34             item = self._menu[label]
  35             if not item.position:
  36                 item.position = ((surface.get_width() - item.get_width()) / 2,
  37                                  height)
  38             item.render(surface)
  39             height += item.get_height() + 30
  40
  41     def _get_pressed(self, ev):
  42         for label, button in self._menu.items():
  43             if button.pressed(ev):
  44                 return label
  45         return None
  46
  47     def _do_day(self):
  48         from .day import DayScene
  49         SceneChangeEvent.post(scene=DayScene())
  50
  51     def event(self, ev, gamestate):
  52         if ev.type == pgl.KEYDOWN:
  53             if ev.key in (pgl.K_q, pgl.K_ESCAPE):
  54                 QuitEvent.post()
  55             elif ev.key == pgl.K_n:
  56                 from .night import NightScene
  57                 SceneChangeEvent.post(scene=NightScene())
  58             elif ev.key == pgl.K_d:
  59                 self._do_day()
  60             elif ev.key == pgl.K_l:
  61                 print "Loading Station Alpha ..."
  62                 gamestate.load_station("station-alpha.json")
  63         elif ev.type == pgl.MOUSEBUTTONDOWN:
  64             pressed = self._get_pressed(ev)
  65             if pressed == 'load level':
  66                 print "Loading Station Alpha ..."
  67                 gamestate.load_station("station-alpha.json")
  68             elif pressed == 'start game':
  69                 self._do_day()
  70             elif pressed:
  71                 print 'Pressed', pressed