Add support for user-defined figures in the MuJoCo viewer.

PiperOrigin-RevId: 726983516 Change-Id: Ifb821f3cafb163d4f3077d76b0668037b4bc5a9c
2025-02-14 10:44:50 -08:00
parent 6f0bd0a5f8
commit 37d7591ce0
5 changed files with 72 additions and 19 deletions
@@ -108,6 +108,23 @@ class Handle:
      return sim.d
    return None

+  @property
+  def viewport(self):
+    sim = self._sim()
+    if sim is not None:
+      return sim.viewport
+    return None
+
+  def set_figures(self, viewports_figures):
+    sim = self._sim()
+    if sim is not None:
+      sim.set_figures(viewports_figures)
+
+  def clear_figures(self):
+    sim = self._sim()
+    if sim is not None:
+      sim.clear_figures()
+
  def close(self):
    sim = self._sim()
    if sim is not None: