koperkapel/scenes/menu.py

   1 """ Main menu scene. """
   2
   3 from pgzero.constants import keys
   4 from ..actors.buttons import TextButton
   5 from ..actors.navigator import ActorNavigator
   6 from .base import Scene, ChangeSceneEvent, QuitEvent
   7
   8
   9
  10
  11 class MenuScene(Scene):
  12     """ Main menu scene. """
  13
  14     def __init__(self):
  15         super().__init__()
  16         self._title = self.actors.add(TextButton("Main Menu"))
  17         self._title.pos = (300, 200)
  18         self._nav = ActorNavigator()
  19         self._menu = [
  20             TextButton("Play", action=self.change_to_level),
  21             TextButton("Credits", action=self.change_to_credits),
  22             TextButton("Quit", action=self.quit),
  23         ]
  24         for i, item in enumerate(self._menu):
  25             item.pos = (300, 300 + (i * 100))
  26             self.actors.add(item)
  27         self._nav.connect(
  28             self._menu, [keys.DOWN, keys.RIGHT], [keys.UP, keys.LEFT],
  29             wrap=True)
  30         self._nav.current.select()
  31
  32     def change_to_level(self):
  33         from .level import LevelScene
  34         return [ChangeSceneEvent(LevelScene('level1'))]
  35
  36     def change_to_credits(self):
  37         from .credits import CreditsScene
  38         return [ChangeSceneEvent(CreditsScene())]
  39
  40     def quit(self):
  41         return [QuitEvent()]
  42
  43     def on_key_down(self, key, mod, unicode):
  44         if key == keys.RETURN:
  45             current = self._nav.current
  46             if current is not None:
  47                 return current.action()
  48         else:
  49             prev, current = self._nav.follow(key)
  50             if prev is not None:
  51                 prev.deselect()
  52             if current is not None:
  53                 current.select()