Skip to content
Snippets Groups Projects
Commit 9064fde5 authored by Erik Nygren's avatar Erik Nygren :bullettrain_front:
Browse files

Merge branch '109_refactor_rendertools' into 'master'

cleaned up the simple_examlpe_N files

Closes #111 and #109

See merge request flatland/flatland!110
parents f6a4c7df 76812b8a
No related branches found
No related tags found
No related merge requests found
......@@ -20,6 +20,5 @@ env.reset()
env_renderer = RenderTool(env)
env_renderer.render_env(show=True)
env_renderer.render_env(show=True)
input("Press Enter to continue...")
......@@ -34,6 +34,5 @@ env.reset()
env_renderer = RenderTool(env, gl="PIL")
env_renderer.render_env(show=True)
env_renderer.render_env(show=True)
input("Press Enter to continue...")
......@@ -23,7 +23,6 @@ for i in range(env.get_num_agents()):
env_renderer = RenderTool(env)
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] \
(turnleft+move, move to front, turnright+move)")
......
......@@ -57,6 +57,17 @@ class RenderTool(object):
self.new_rail = True
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):
# create background map
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