Skip to content

Commit

Permalink
Add static rendering in MujocoVideoRecorder
Browse files Browse the repository at this point in the history
Co-authored-by: Diego Ferigo <diego.ferigo@iit.it>
  • Loading branch information
flferretti and diegoferigo committed Feb 19, 2024
1 parent adabdb9 commit d441a5a
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions src/jaxsim/mujoco/visualizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,13 +52,19 @@ def reset(
self.data = data if data is not None else self.data
self.model = model if model is not None else self.model

def render_frame(self, camera_name: str | None = None) -> None:
def render_frame(self, camera_name: str | None = None) -> npt.NDArray:
""""""

mujoco.mj_forward(self.model, self.data)
self.renderer.update_scene(data=self.data) # TODO camera name

self.frames.append(self.renderer.render())
return self.renderer.render()

def record_frame(self, camera_name: str | None = None) -> None:
""""""

frame = self.render_frame(camera_name=camera_name)
self.frames.append(frame)

def write_video(self, path: pathlib.Path, exist_ok: bool = False) -> None:
""""""
Expand Down

0 comments on commit d441a5a

Please sign in to comment.