menu: two rows

[pmachines.git] / pmachines / menu.py

diff --git a/pmachines/menu.py b/pmachines/menu.py
index 0fc6accf7f9fdb1136d52cce908e7d41d3cc59ae..494973d444e03ed816a06eff4ab2cefbd1344b8a 100644 (file)
--- a/pmachines/menu.py
+++ b/pmachines/menu.py
@@ -1,6 +1,9 @@
 from logging import info, debug
 from sys import platform, exit
 from os import environ, system
+from glob import glob
+from importlib import import_module
+from inspect import isclass
 from webbrowser import open_new_tab
 from panda3d.core import Texture, TextNode, WindowProperties, LVector2i, \
     TextProperties, TextPropertiesManager
@@ -9,6 +12,8 @@ from direct.gui.DirectGui import DirectButton, DirectCheckButton, \
 from direct.gui.DirectGuiGlobals import FLAT
 from direct.gui.OnscreenText import OnscreenText
 from lib.engine.gui.cursor import MouseCursor
+from pmachines.scene import Scene
+from panda3d.bullet import BulletWorld
 
 
 class Menu:
@@ -155,7 +160,7 @@ class Menu:
             text=_('Website'), pos=(-.6, 1, .29), command=self.on_website,
             **self._common_btn | {'scale': .08})]
         self._widgets += [OnscreenText(
-            _('Supporters\n  \1scale\1rdb\2\n  \1scale\1Luisa Tenuta\2\n  \1scale\1Damiana Ercolani\2'),
+            _('Special thanks to:\n  \1scale\1rdb\2\n  \1scale\1Luisa Tenuta\2\n  \1scale\1Damiana Ercolani\2'),
             pos=(.1, .55), font=self._common['text_font'],
             scale=self._common['scale'], fg=self._common['text_fg'],
             align=TextNode.A_left)]
@@ -164,14 +169,50 @@ class Menu:
             **self._common_btn)]
 
     def on_play(self):
-        self._fsm.demand('Scene')
+        scenes = []
+        for _file in glob('pmachines/scenes/*.py'):
+            _fn = _file.replace('.py', '').replace('/', '.')
+            for member in import_module(_fn).__dict__.values():
+                if isclass(member) and issubclass(member, Scene) and \
+                        member != Scene:
+                    scenes += [member]
+        scenes = sorted(scenes, key=lambda elm: elm.sorting)
+        self.destroy()
+        self._cursor = MouseCursor(
+            'assets/buttons/arrowUpLeft.png', (.04, 1, .04), (.5, .5, .5, 1),
+            (.01, .01))
+        self._widgets = []
+        cmn = self._common_btn.copy() | {
+            'frameSize': (-2.4, 2.4, -2.4, 2.4),
+            'frameColor': (1, 1, 1, .8)}
+        left = - (dx := .8) * (min(3, len(scenes)) - 1) / 2
+        for i, cls in enumerate(scenes):
+            top = .1 if len(scenes) < 4 else .6
+            row = 0 if i < 3 else 1
+            self._widgets += [DirectButton(
+                text=cls.name(), pos=(left + dx * (i % 3), 1, top - dx * row),
+                command=self.start, extraArgs=[cls], text_wordwrap=4,
+                frameTexture='assets/images/scenes/%s.png' % cls.__name__,
+                **cmn)]
+        self._widgets += [DirectButton(
+            text=_('Back'), pos=(0, 1, -.8), command=self.on_back,
+            **self._common_btn)]
+
+    def start(self, cls):
+        self._fsm.demand('Scene', cls)
 
     def on_options(self):
         self.destroy()
+        self._cursor = MouseCursor(
+            'assets/buttons/arrowUpLeft.png', (.04, 1, .04), (.5, .5, .5, 1),
+            (.01, .01))
         self._set_options()
 
     def on_credits(self):
         self.destroy()
+        self._cursor = MouseCursor(
+            'assets/buttons/arrowUpLeft.png', (.04, 1, .04), (.5, .5, .5, 1),
+            (.01, .01))
         self._set_credits()
 
     def _rearrange_width(self):
@@ -234,6 +275,9 @@ class Menu:
     def on_back(self):
         self._opt_file.store()
         self.destroy()
+        self._cursor = MouseCursor(
+            'assets/buttons/arrowUpLeft.png', (.04, 1, .04), (.5, .5, .5, 1),
+            (.01, .01))
         self._set_main()
 
     def destroy(self):