in mujoco_py/mjviewer.py [0:0]
def key_callback(self, window, key, scancode, action, mods):
if action != glfw.RELEASE:
return
elif key == glfw.KEY_TAB: # Switches cameras.
self.cam.fixedcamid += 1
self.cam.type = const.CAMERA_FIXED
if self.cam.fixedcamid >= self._ncam:
self.cam.fixedcamid = -1
self.cam.type = const.CAMERA_FREE
elif key == glfw.KEY_H: # hides all overlay.
self._hide_overlay = not self._hide_overlay
elif key == glfw.KEY_SPACE and self._paused is not None: # stops simulation.
self._paused = not self._paused
# Advances simulation by one step.
elif key == glfw.KEY_RIGHT and self._paused is not None:
self._advance_by_one_step = True
self._paused = True
elif key == glfw.KEY_V or \
(key == glfw.KEY_ESCAPE and self._record_video): # Records video. Trigers with V or if in progress by ESC.
self._record_video = not self._record_video
if self._record_video:
fps = (1 / self._time_per_render)
self._video_process = Process(target=save_video,
args=(self._video_queue, self._video_path % self._video_idx, fps))
self._video_process.start()
if not self._record_video:
self._video_queue.put(None)
self._video_process.join()
self._video_idx += 1
elif key == glfw.KEY_T: # capture screenshot
img = self._read_pixels_as_in_window()
imageio.imwrite(self._image_path % self._image_idx, img)
self._image_idx += 1
elif key == glfw.KEY_I: # drops in debugger.
print('You can access the simulator by self.sim')
import ipdb
ipdb.set_trace()
elif key == glfw.KEY_S: # Slows down simulation.
self._run_speed /= 2.0
elif key == glfw.KEY_F: # Speeds up simulation.
self._run_speed *= 2.0
elif key == glfw.KEY_C: # Displays contact forces.
vopt = self.vopt
vopt.flags[10] = vopt.flags[11] = not vopt.flags[10]
elif key == glfw.KEY_D: # turn off / turn on rendering every frame.
self._render_every_frame = not self._render_every_frame
elif key == glfw.KEY_E:
vopt = self.vopt
vopt.frame = 1 - vopt.frame
elif key == glfw.KEY_R: # makes everything little bit transparent.
self._transparent = not self._transparent
if self._transparent:
self.sim.model.geom_rgba[:, 3] /= 5.0
else:
self.sim.model.geom_rgba[:, 3] *= 5.0
elif key == glfw.KEY_M: # Shows / hides mocap bodies
self._show_mocap = not self._show_mocap
for body_idx1, val in enumerate(self.sim.model.body_mocapid):
if val != -1:
for geom_idx, body_idx2 in enumerate(self.sim.model.geom_bodyid):
if body_idx1 == body_idx2:
if not self._show_mocap:
# Store transparency for later to show it.
self.sim.extras[
geom_idx] = self.sim.model.geom_rgba[geom_idx, 3]
self.sim.model.geom_rgba[geom_idx, 3] = 0
else:
self.sim.model.geom_rgba[
geom_idx, 3] = self.sim.extras[geom_idx]
elif key in (glfw.KEY_0, glfw.KEY_1, glfw.KEY_2, glfw.KEY_3, glfw.KEY_4):
self.vopt.geomgroup[key - glfw.KEY_0] ^= 1
super().key_callback(window, key, scancode, action, mods)