Skip to content

Commit

Permalink
Merge pull request #85 from ami-iit/feature/render_static
Browse files Browse the repository at this point in the history
Add static rendering in `MujocoVideoRecorder`
  • Loading branch information
flferretti committed Feb 20, 2024
2 parents adabdb9 + d441a5a commit 6e3da71
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 6e3da71

Please sign in to comment.