Merge pull request #13 from rpapallas/main

Added ability to toggle on/off the small bottom-left menu

Author: rohanpsingh

README.md

@@ -54,3 +54,4 @@ Other key bindings are shown in the overlay menu (almost similar to `mujoco-py`)
 - `title`: set the title of the window, for example: `viewer = mujoco_viewer.MujocoViewer(model, data, title='My Demo')` (defaults to `mujoco-python-viewer`). 
 - `width`: set the window width, for example: `viewer = mujoco_viewer.MujocoViewer(model, data, width=300)` (defaults to full screen's width). 
 - `height`: set the window height, for example: `viewer = mujoco_viewer.MujocoViewer(model, data, height=300)` (defaults to full screen's height). 
+- `hide_menus`: set whether the overlay menus should be hidden or not (defaults to `True`).

mujoco_viewer/mujoco_viewer.py

@@ -14,7 +14,8 @@ def __init__(
             data,
             title="mujoco-python-viewer",
             width=None,
-            height=None):
+            height=None,
+            hide_menus=True):
         self.model = model
         self.data = data
 
@@ -41,7 +42,7 @@ def __init__(
         self._run_speed = 1.0
         self._loop_count = 0
         self._advance_by_one_step = False
-        self._hide_menu = True
+        self._hide_menus = hide_menus
 
         # glfw init
         glfw.init()
@@ -90,7 +91,7 @@ def __init__(
     def _key_callback(self, window, key, scancode, action, mods):
         if action != glfw.RELEASE:
             if key == glfw.KEY_LEFT_ALT:
-                self._hide_menu = False
+                self._hide_menus = False
             return
         # Switch cameras
         elif key == glfw.KEY_TAB:
@@ -137,9 +138,9 @@ def _key_callback(self, window, key, scancode, action, mods):
             self.vopt.frame = 1 - self.vopt.frame
         # Hide overlay menu
         elif key == glfw.KEY_LEFT_ALT:
-            self._hide_menu = True
+            self._hide_menus = True
         elif key == glfw.KEY_H:
-            self._hide_menu = not self._hide_menu
+            self._hide_menus = not self._hide_menus
         # Make transparent
         elif key == glfw.KEY_R:
             self._transparent = not self._transparent
@@ -457,7 +458,7 @@ def add_overlay(gridpos, text1, text2):
             topleft,
             "Referenc[e] frames",
             "On" if self.vopt.frame == 1 else "Off")
-        add_overlay(topleft, "[H]ide Menu", "")
+        add_overlay(topleft, "[H]ide Menus", "")
         if self._image_idx > 0:
             fname = self._image_path % (self._image_idx - 1)
             add_overlay(topleft, "Cap[t]ure frame", "Saved as %s" % fname)
@@ -513,8 +514,11 @@ def update():
                 mujoco.mjr_render(self.viewport, self.scn, self.ctx)
                 # overlay items
                 for gridpos, [t1, t2] in self._overlay.items():
-                    if gridpos == mujoco.mjtGridPos.mjGRID_TOPLEFT and self._hide_menu:
+                    menu_positions = [mujoco.mjtGridPos.mjGRID_TOPLEFT,
+                                      mujoco.mjtGridPos.mjGRID_BOTTOMLEFT]
+                    if gridpos in menu_positions and self._hide_menus:
                         continue
+
                     mujoco.mjr_overlay(
                         mujoco.mjtFontScale.mjFONTSCALE_150,
                         gridpos,