Add static rendering in MujocoVisualizer

src/jaxsim/mujoco/visualizer.py

@@ -1,6 +1,6 @@
 import contextlib
 import pathlib
-from typing import ContextManager
+from typing import ContextManager, Sequence, Union
 
 import mediapy as media
 import mujoco as mj
@@ -150,3 +150,18 @@ def open(
             yield handle
         finally:
             handle.close() if close_on_exit else None
+
+    def render_static(
+        self,
+        height: int = 240,
+        width: int = 320,
+    ) -> Union[Sequence[npt.NDArray], npt.NDArray]:
+        """"""
+
+        renderer = mujoco.Renderer(model=self.model, height=height, width=width)
+
+        data = mujoco.MjData(self.model) or self.data
+
+        mujoco.mj_forward(self.model, data)
+        renderer.update_scene(data, camera_name)
+        return renderer.render()