Skip to content
Snippets Groups Projects
Commit 76812b8a authored by Erik Nygren's avatar Erik Nygren
Browse files

cleaned up the simple_examlpe_N files

added reset() function to rendertools
parent 409797dd
No related branches found
No related tags found
No related merge requests found
...@@ -20,6 +20,5 @@ env.reset() ...@@ -20,6 +20,5 @@ env.reset()
env_renderer = RenderTool(env) env_renderer = RenderTool(env)
env_renderer.render_env(show=True) env_renderer.render_env(show=True)
env_renderer.render_env(show=True)
input("Press Enter to continue...") input("Press Enter to continue...")
...@@ -34,6 +34,5 @@ env.reset() ...@@ -34,6 +34,5 @@ env.reset()
env_renderer = RenderTool(env, gl="PIL") env_renderer = RenderTool(env, gl="PIL")
env_renderer.render_env(show=True) env_renderer.render_env(show=True)
env_renderer.render_env(show=True)
input("Press Enter to continue...") input("Press Enter to continue...")
...@@ -23,7 +23,6 @@ for i in range(env.get_num_agents()): ...@@ -23,7 +23,6 @@ for i in range(env.get_num_agents()):
env_renderer = RenderTool(env) env_renderer = RenderTool(env)
env_renderer.render_env(show=True, frames=True) env_renderer.render_env(show=True, frames=True)
env_renderer.render_env(show=True, frames=True)
print("Manual control: s=perform step, q=quit, [agent id] [1-2-3 action] \ print("Manual control: s=perform step, q=quit, [agent id] [1-2-3 action] \
(turnleft+move, move to front, turnright+move)") (turnleft+move, move to front, turnright+move)")
......
...@@ -57,6 +57,17 @@ class RenderTool(object): ...@@ -57,6 +57,17 @@ class RenderTool(object):
self.new_rail = True self.new_rail = True
self.update_background() self.update_background()
def reset(self):
"""
Resets the environment
:return:
"""
self.set_new_rail()
self.frame_nr = 0
self.start_time = time.time()
self.times_list = deque()
return
def update_background(self): def update_background(self):
# create background map # create background map
targets = {} targets = {}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment