in safety_gym/envs/world.py [0:0]
def render(self, mode='human'):
''' Render the environment to the screen '''
if self.viewer is None:
self.viewer = MjViewer(self.sim)
# Turn all the geom groups on
self.viewer.vopt.geomgroup[:] = 1
# Set camera if specified
if mode == 'human':
self.viewer.cam.fixedcamid = -1
self.viewer.cam.type = const.CAMERA_FREE
else:
self.viewer.cam.fixedcamid = self.model.camera_name2id(mode)
self.viewer.cam.type = const.CAMERA_FIXED
if self.update_viewer_sim:
self.viewer.update_sim(self.sim)
self.update_viewer_sim = False
self.viewer.render()