from pgzero.constants import keys
from ..actors.buttons import TextButton
-from .base import Scene, ChangeSceneEvent
+from ..actors.navigator import ActorNavigator
+from .base import Scene, ChangeSceneEvent, WorldEvent, QuitEvent
class MenuScene(Scene):
def __init__(self):
super().__init__()
self._title = self.actors.add(TextButton("Main Menu"))
- self._title.pos = (300, 200)
- self._level = self.actors.add(TextButton("Level"))
- self._level.pos = (300, 300)
- self._credits = self.actors.add(TextButton("Credits"))
- self._credits.pos = (300, 400)
+ self._title.pos = (300, 100)
+ self._nav = ActorNavigator()
+ self._menu = [
+ TextButton("Play", action=self.change_to_level),
+ TextButton("View Last Generated Level",
+ action=self.change_to_viewer),
+ TextButton("Manage Roaches",
+ action=self.change_to_roach_management),
+ TextButton("Credits", action=self.change_to_credits),
+ TextButton("Quit", action=self.quit),
+ ]
+ for i, item in enumerate(self._menu):
+ item.pos = (300, 200 + (i * 100))
+ self.actors.add(item)
+ self._nav.connect(
+ self._menu, [keys.DOWN, keys.RIGHT], [keys.UP, keys.LEFT],
+ wrap=True)
+ self._nav.current.select()
+
+ def change_to_level(self):
+ from .level import GameLevelScene
+ return [ChangeSceneEvent(GameLevelScene())]
+
+ def change_to_viewer(self):
+ from .viewlevel import ViewLevelScene
+ return [
+ WorldEvent("set", {"level.name": "map"}),
+ ChangeSceneEvent(ViewLevelScene())
+ ]
+
+ def change_to_roach_management(self):
+ from .roach_management import RoachesScene
+ return [ChangeSceneEvent(RoachesScene())]
+
+ def change_to_credits(self):
+ from .credits import CreditsScene
+ return [ChangeSceneEvent(CreditsScene())]
+
+ def quit(self):
+ return [QuitEvent()]
def on_key_down(self, key, mod, unicode):
- if key == keys.C:
- from .credits import CreditsScene
- return [ChangeSceneEvent(CreditsScene())]
+ if key == keys.RETURN:
+ current = self._nav.current
+ if current is not None:
+ return current.action()
+ else:
+ prev, current = self._nav.follow(key)
+ if prev is not None:
+ prev.deselect()
+ if current is not None:
+ current.select()