diff --git a/flatland/utils/graphics_pil.py b/flatland/utils/graphics_pil.py index 7bfc67c1a2f19c31a73a5c41dde0c8645649ed60..f8c96f1bd3d052aa5238ac563044acb6995e23a1 100644 --- a/flatland/utils/graphics_pil.py +++ b/flatland/utils/graphics_pil.py @@ -276,7 +276,7 @@ class PILSVG(PILGL): def loadBuildingSVGs(self): - dBuildingFiles = { + dBuildingFiles = [ "Buildings/Bank.svg", "Buildings/Bar.svg", "Buildings/Fabrik_A.svg", @@ -295,7 +295,7 @@ class PILSVG(PILGL): "Buildings/Post.svg", "Buildings/Supermarkt.svg", "Buildings/Tankstelle.svg", - "Buildings/Wohnhaus.svg"} + "Buildings/Wohnhaus.svg"] self.dBuildings = [] for sFile in dBuildingFiles: @@ -303,19 +303,19 @@ class PILSVG(PILGL): self.dBuildings.append(img) def loadScenerySVGs(self): - dSceneryFiles = { - "Scenery/Bergwelt_A_Teil_1_links.svg", - "Scenery/Bergwelt_A_Teil_2_mitte.svg", - "Scenery/Bergwelt_A_Teil_3_rechts.svg", - "Scenery/Bergwelt_B.svg", - "Scenery/Bergwelt_C_Teil_1_links.svg", - "Scenery/Bergwelt_C_Teil_2_rechts.svg", + dSceneryFiles = [ "Scenery/Laubbaume_A.svg", "Scenery/Laubbaume_B.svg", "Scenery/Laubbaume_C.svg", "Scenery/Nadelbaume_A.svg", "Scenery/Nadelbaume_B.svg", - } + "Scenery/Bergwelt_B.svg", + "Scenery/Bergwelt_C_Teil_1_links.svg", + "Scenery/Bergwelt_C_Teil_2_rechts.svg", + "Scenery/Bergwelt_A_Teil_1_links.svg", + "Scenery/Bergwelt_A_Teil_2_mitte.svg", + "Scenery/Bergwelt_A_Teil_3_rechts.svg", + ] self.dScenery = [] for sFile in dSceneryFiles: img = self.pilFromSvgFile('svg',sFile) @@ -418,7 +418,7 @@ class PILSVG(PILGL): a = a % len(self.dBuildings) pilTrack = self.dBuildings[a] else: - a = int(self.background_grid[col][row]) + a = int(self.background_grid[col][row]) - 4 a = a % len(self.dScenery) pilTrack = self.dScenery[a] diff --git a/flatland/utils/rendertools.py b/flatland/utils/rendertools.py index df0ca8a5a83bf52328ec7d1cd9b66e01d7f8ab7a..937580f4469aec68942f04dbaadf50b5ee2f40bb 100644 --- a/flatland/utils/rendertools.py +++ b/flatland/utils/rendertools.py @@ -658,6 +658,8 @@ class RenderTool(object): self.gl.setRailAt(r, c, binTrans, iTarget=target, isSelected=isSelected) + self.gl.build_background_map(dTargets) + for iAgent, agent in enumerate(self.env.agents): if agent is None: diff --git a/setup.py b/setup.py index cc4d9ee348065f633692a139feb21557009e7c38..72e19e3ce79adeafdc2034a42633d940e9d4c98e 100644 --- a/setup.py +++ b/setup.py @@ -63,10 +63,8 @@ else: def get_all_svg_files(directory='./svg/'): ret = [] for f in os.listdir(directory): - if f != '__pycache__': - if f != "Buildings" and f != "Scenery": - ret.append(directory + f) - print("install svg:" , ret) + if os.path.isfile(os.path.join(directory, f)): + ret.append(directory + f) return ret @@ -105,7 +103,7 @@ setup( keywords='flatland', name='flatland-rl', packages=find_packages('.'), - data_files=[('svg', get_all_svg_files()),('svg/Buildings', get_all_svg_files('svg/Buildings/')),('svg/Scenery', get_all_svg_files('svg/Scenery/'))], + data_files=[('svg', get_all_svg_files())], setup_requires=setup_requirements, test_suite='tests', tests_require=test_requirements,