From 5983257179f99a57d210cf0800ae78e240401629 Mon Sep 17 00:00:00 2001
From: hagrid67 <jdhwatson@gmail.com>
Date: Fri, 14 Aug 2020 10:22:55 +0100
Subject: [PATCH] adding jupyter_utils; adding ipycanvas to reqs_dev
---
flatland/utils/jupyter_utils.py | 48 +++++++++++++++++++++++++++++++++
requirements_dev.txt | 1 +
2 files changed, 49 insertions(+)
create mode 100644 flatland/utils/jupyter_utils.py
diff --git a/flatland/utils/jupyter_utils.py b/flatland/utils/jupyter_utils.py
new file mode 100644
index 00000000..a2f37dfb
--- /dev/null
+++ b/flatland/utils/jupyter_utils.py
@@ -0,0 +1,48 @@
+
+
+import PIL
+from IPython import display
+from ipycanvas import canvas
+import time
+
+from flatland.envs import malfunction_generators as malgen
+from flatland.envs.agent_utils import EnvAgent
+#from flatland.envs import sparse_rail_gen as spgen
+from flatland.envs import rail_generators as rail_gen
+from flatland.envs import agent_chains as ac
+from flatland.envs.rail_env import RailEnv, RailEnvActions
+from flatland.envs.persistence import RailEnvPersister
+from flatland.utils.rendertools import RenderTool
+from flatland.utils import env_edit_utils as eeu
+
+
+class Behaviour():
+ def __init__(self, env):
+ self.env = env
+
+ def getActions(self):
+ return {}
+
+
+class AlwaysForward():
+ pass
+
+
+class EnvCanvas():
+
+ def __init__(self, env):
+ self.env = env
+ self.oRT = RenderTool(env, show_debug=True)
+ self.render()
+ self.oCan = canvas.Canvas(size=(600,300))
+ self.oCan.put_image_data(self.oRT.get_image())
+
+ def render(self):
+ self.oRT.render_env(show_rowcols=True, show_inactive_agents=True, show_observations=False)
+
+ def show(self):
+ self.render()
+ self.oCan.put_image_data(self.oRT.get_image())
+ display.display(self.oCan)
+
+
diff --git a/requirements_dev.txt b/requirements_dev.txt
index 1008d6ef..b35f5d03 100644
--- a/requirements_dev.txt
+++ b/requirements_dev.txt
@@ -20,3 +20,4 @@ timeout-decorator>=0.4.1
attrs
gym==0.14.0
networkx
+ipycanvas
--
GitLab