From d59bf10593047e074b33d169799886db0e2b75b3 Mon Sep 17 00:00:00 2001
From: hagrid67 <jdhwatson@gmail.com>
Date: Tue, 21 May 2019 21:21:28 +0100
Subject: [PATCH] Added empty_rail_generator for editor
---
flatland/envs/generators.py | 15 +++++++++++++++
1 file changed, 15 insertions(+)
diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index c1578a8..f396164 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -9,6 +9,21 @@ from flatland.envs.env_utils import distance_on_rail, connect_rail, get_directio
from flatland.envs.env_utils import get_rnd_agents_pos_tgt_dir_on_rail
+def empty_rail_generator():
+ """
+ Returns a generator which returns an empty rail mail with no agents.
+ Primarily used by the editor
+ """
+ def generator(width, height, num_agents=0, num_resets=0):
+ rail_trans = RailEnvTransitions()
+ grid_map = GridTransitionMap(width=width, height=height, transitions=rail_trans)
+ rail_array = grid_map.grid
+ rail_array.fill(0)
+
+ return grid_map, [], [], []
+ return generator
+
+
def complex_rail_generator(nr_start_goal=1, nr_extra=100, min_dist=20, max_dist=99999, seed=0):
"""
Parameters
--
GitLab