def __init__(self, env):
self.env = env
super(GatherViewer, self).__init__()
green_ball_model = MjModel(
osp.abspath(osp.join(MODEL_DIR, 'green_ball.xml')))
self.green_ball_renderer = EmbeddedViewer()
self.green_ball_model = green_ball_model
self.green_ball_renderer.set_model(green_ball_model)
red_ball_model = MjModel(
osp.abspath(osp.join(MODEL_DIR, 'red_ball.xml')))
self.red_ball_renderer = EmbeddedViewer()
self.red_ball_model = red_ball_model
self.red_ball_renderer.set_model(red_ball_model)
#"""
green_ball_highlighted_model = MjModel(
osp.abspath(osp.join(MODEL_DIR, 'green_ball_highlighted.xml')))
self.green_ball_highlighted_renderer = EmbeddedViewer()
self.green_ball_highlighted_model = green_ball_highlighted_model
self.green_ball_highlighted_renderer.set_model(
green_ball_highlighted_model)
red_ball_highlighted_model = MjModel(
osp.abspath(osp.join(MODEL_DIR, 'red_ball_highlighted.xml')))
self.red_ball_highlighted_renderer = EmbeddedViewer()
self.red_ball_highlighted_model = red_ball_highlighted_model
self.red_ball_highlighted_renderer.set_model(
red_ball_highlighted_model)
