koperkapel/scenes/menu.py

   1 """ Main menu scene. """
   2
   3 from pgzero.constants import keys
   4 from pgzero.loaders import images
   5 from ..actors.navigator import ActorNavigator
   6 from ..actors.orientatedsurf import OrientatedSurfActor, SelectableSurfActor
   7 from .base import Scene, ChangeSceneEvent, WorldEvent, QuitEvent
   8
   9
  10 class MenuScene(Scene):
  11     """ Main menu scene. """
  12
  13     def __init__(self):
  14         super().__init__()
  15         self.actors.add_layer("buttons", level=10)
  16         self._init_bg()
  17         self._init_buttons()
  18
  19     def _init_bg(self):
  20         self.actors.default.clear()
  21         self.actors.default.add(OrientatedSurfActor(
  22             images.load("splash/background")))
  23
  24     def _init_buttons(self):
  25         self._nav = ActorNavigator()
  26         menu = [
  27             self._build_button("new_game", action=self.change_to_new_game),
  28             self._build_button("resume_game", action=self.change_to_resume),
  29             self._build_button("credits", action=self.change_to_credits),
  30             self._build_button("quit", action=self.quit),
  31         ]
  32         for i, item in enumerate(menu):
  33             item.pos = (300, 200 + (i * 100))
  34             self.actors.add(item)
  35         self._nav.connect(
  36             menu, [keys.DOWN, keys.RIGHT], [keys.UP, keys.LEFT],
  37             wrap=True)
  38         self._nav.current.selected = True
  39
  40     def _build_button(self, name, action):
  41         actor = SelectableSurfActor(
  42             images.load("splash/%s" % (name,)),
  43             images.load("splash/%s_selected" % (name,)))
  44         actor.action = action
  45         return actor
  46
  47     def change_to_resume(self):
  48         from .level import GameLevelScene
  49         return [ChangeSceneEvent(GameLevelScene())]
  50
  51     def change_to_new_game(self):
  52         from .level import GameLevelScene
  53         return [
  54             WorldEvent("reset"),
  55             ChangeSceneEvent(GameLevelScene())
  56         ]
  57
  58     def change_to_credits(self):
  59         from .credits import CreditsScene
  60         return [ChangeSceneEvent(CreditsScene())]
  61
  62     def quit(self):
  63         return [QuitEvent()]
  64
  65     def on_key_down(self, key, mod, unicode):
  66         if key == keys.RETURN:
  67             current = self._nav.current
  68             if current is not None:
  69                 return current.action()
  70         else:
  71             prev, current = self._nav.follow(key)
  72             if prev is not None:
  73                 prev.selected = False
  74             if current is not None:
  75                 current.selected = True