From 2ceb60445d8bea21112c495a2ffc82bbe0574fa6 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Tue, 28 May 2019 16:30:29 +0200
Subject: [PATCH] pil rendering bug fix
---
examples/demo.py | 2 +-
flatland/utils/graphics_pil.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/examples/demo.py b/examples/demo.py
index dae2118c..86e986f6 100644
--- a/examples/demo.py
+++ b/examples/demo.py
@@ -127,7 +127,7 @@ class Demo:
self.action_size = 4
def create_renderer(self):
- self.renderer = RenderTool(self.env, gl="QTSVG")
+ self.renderer = RenderTool(self.env, gl="PILSVG")
handle = self.env.get_agent_handles()
return handle
diff --git a/flatland/utils/graphics_pil.py b/flatland/utils/graphics_pil.py
index fe5e74b5..63a4be28 100644
--- a/flatland/utils/graphics_pil.py
+++ b/flatland/utils/graphics_pil.py
@@ -351,7 +351,7 @@ class PILSVG(PILGL):
rgbaZug3[xy_color_mask, 0:3] = tnNewColor
self.dPilZug[(iDirIn2, iDirOut2, iColor)] = Image.fromarray(rgbaZug3)
- def setAgentAt(self, iAgent, row, col, iDirIn, iDirOut):
+ def setAgentAt(self, iAgent, row, col, iDirIn, iDirOut, color=None):
delta_dir = (iDirOut - iDirIn) % 4
iColor = iAgent % self.nAgentColors
# when flipping direction at a dead end, use the "iDirOut" direction.
--
GitLab