naja/scenes/menu.py

   1 """
   2 Main menu scene.
   3 """
   4
   5 import pygame.locals as pgl
   6
   7 from naja.constants import KEYS, PALETTE, KONAMI_CODE
   8 from naja.events import SceneChangeEvent, QuitGameEvent
   9 from naja.options import options
  10 from naja.scenes.scene import Scene
  11 from naja.scenes.credits import CreditsScene
  12 from naja.scenes.game import GameScene
  13 from naja.scenes.howto import HowtoScene
  14 from naja.scenes.load_save import LoadGameScene, SaveGameScene
  15 from naja.scenes.new_game import NewGameScene
  16 from naja.widgets.selector import SelectorWidget
  17 from naja.widgets.text import TextWidget
  18
  19
  20 class MenuScene(Scene):
  21     def __init__(self, state):
  22         super(MenuScene, self).__init__(state)
  23         selector = SelectorWidget()
  24         self.add(selector)
  25         y_offset = 0
  26
  27         y_offset += 50
  28         resume = TextWidget(
  29             (100, y_offset), 'Resume Game', colour=PALETTE.WHITE,
  30             unselectable_colour=PALETTE.GREY)
  31         resume.add_callback('click', self.scene_callback(GameScene))
  32         resume.set_selectable_callback(lambda: state is not None)
  33         selector.add(resume)
  34
  35         y_offset += 50
  36         new = TextWidget((100, y_offset), 'New Game', colour=PALETTE.WHITE)
  37         new.add_callback('click', self.scene_callback(NewGameScene))
  38         selector.add(new)
  39
  40         y_offset += 50
  41         load = TextWidget((100, y_offset), 'Load', colour=PALETTE.WHITE)
  42         load.add_callback('click', self.scene_callback(LoadGameScene))
  43         selector.add(load)
  44
  45         y_offset += 50
  46         save = TextWidget(
  47             (100, y_offset), 'Save', colour=PALETTE.WHITE,
  48             unselectable_colour=PALETTE.GREY)
  49         save.add_callback('click', self.scene_callback(SaveGameScene))
  50         save.set_selectable_callback(lambda: state is not None)
  51         selector.add(save)
  52
  53         y_offset += 50
  54         howto = TextWidget(
  55             (100, y_offset), 'How To Play', colour=PALETTE.WHITE)
  56         howto.add_callback('click', self.scene_callback(HowtoScene))
  57         selector.add(howto)
  58
  59         y_offset += 50
  60         credits = TextWidget((100, y_offset), 'Credits', colour=PALETTE.WHITE)
  61         credits.add_callback('click', self.scene_callback(CreditsScene))
  62         selector.add(credits)
  63
  64         y_offset += 50
  65         quit = TextWidget((100, y_offset), 'Quit', colour=PALETTE.WHITE)
  66         quit.add_callback('click', lambda event: QuitGameEvent.post())
  67         selector.add(quit)
  68
  69         self.konami = []
  70
  71     def scene_callback(self, scene_class):
  72         return lambda event: SceneChangeEvent.post(scene_class)
  73
  74     def handle_event(self, ev):
  75         if ev.type == pgl.KEYDOWN and self.konami is not None:
  76             if ev.key in KEYS.KONAMI:
  77                 self.konami.append(ev.key)
  78                 if self.konami == KONAMI_CODE:
  79                     options['cheat_enabled'] = True
  80                     print "Cheats enabled."
  81                     self.konami = None
  82                 elif self.konami != KONAMI_CODE[:len(self.konami)]:
  83                     self.konami = None
  84         return super(MenuScene, self).handle_event(ev)
  85
  86     def handle_scene_event(self, ev):
  87         if ev.type == pgl.KEYDOWN:
  88             if ev.key in KEYS.QUIT:
  89                 QuitGameEvent.post()
  90                 return