From 76812b8aa44fbb6ff48fd7b8f04354893e90f0b9 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Wed, 17 Jul 2019 12:27:07 -0400
Subject: [PATCH] cleaned up the simple_examlpe_N files added reset() function
to rendertools
---
examples/simple_example_1.py | 1 -
examples/simple_example_2.py | 1 -
examples/simple_example_3.py | 1 -
flatland/utils/rendertools.py | 11 +++++++++++
4 files changed, 11 insertions(+), 3 deletions(-)
diff --git a/examples/simple_example_1.py b/examples/simple_example_1.py
index daf56a9b..93d52688 100644
--- a/examples/simple_example_1.py
+++ b/examples/simple_example_1.py
@@ -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...")
diff --git a/examples/simple_example_2.py b/examples/simple_example_2.py
index be6fdef3..f8612a39 100644
--- a/examples/simple_example_2.py
+++ b/examples/simple_example_2.py
@@ -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...")
diff --git a/examples/simple_example_3.py b/examples/simple_example_3.py
index 3a1c583f..5aa03d8f 100644
--- a/examples/simple_example_3.py
+++ b/examples/simple_example_3.py
@@ -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)")
diff --git a/flatland/utils/rendertools.py b/flatland/utils/rendertools.py
index 1e1a7c15..bc24a214 100644
--- a/flatland/utils/rendertools.py
+++ b/flatland/utils/rendertools.py
@@ -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 = {}
--
GitLab