X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;ds=sidebyside;f=koperkapel%2Fscenes%2Fmenu.py;h=eefd5515fc5cf6e51daadc13548c2cf920b41424;hb=723e1584330175d8156805bff09c334bb31a2c57;hp=e5746e80463a82d87b1e7075115ce45c0c6621bc;hpb=e97ad367d4a1f7f9abcd790636b5b72ea312cb29;p=koperkapel.git

diff --git a/koperkapel/scenes/menu.py b/koperkapel/scenes/menu.py
index e5746e8..eefd551 100644
--- a/koperkapel/scenes/menu.py
+++ b/koperkapel/scenes/menu.py
@@ -2,7 +2,8 @@
 
 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):
@@ -11,16 +12,54 @@ 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_roaches),
+            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 LevelScene
+        return [ChangeSceneEvent(LevelScene())]
+
+    def change_to_viewer(self):
+        from .viewlevel import ViewLevelScene
+        return [
+            WorldEvent("set", {"level.name": "map"}),
+            ChangeSceneEvent(ViewLevelScene())
+        ]
+
+    def change_to_roaches(self):
+        from .roaches 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.L:
-            from .level import LevelScene
-            return [ChangeSceneEvent(LevelScene('level1'))]
+        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()