From e504d2430b95e15683bd37295087c1d20bf69b93 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Thu, 22 Aug 2019 23:36:53 +0200
Subject: [PATCH] inital postion / target not equal
---
flatland/envs/generators.py | 15 +++++++++++++--
1 file changed, 13 insertions(+), 2 deletions(-)
diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index 90b05485..712debda 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -915,13 +915,24 @@ def realistic_rail_generator(nr_start_goal=1, seed=0, add_max_dead_end=4, two_tr
for a in range(min(len(agents_targets), num_agents)):
t = np.random.choice(range(len(agents_targets)))
- d = agents_targets[t]
+ tp = agents_targets[t]
agents_targets.pop(t)
- agents_target.append((d[0], d[1]))
+ agents_target.append((tp[0], tp[1]))
+
sel = np.random.choice(range(len(agents_positions)))
# backward
p = agents_positions[sel]
d = agents_directions[sel]
+ cnt = 0
+ while (p[0] == tp[0] and p[1] == tp[1]):
+ sel = np.random.choice(range(len(agents_positions)))
+ # backward
+ p = agents_positions[sel]
+ d = agents_directions[sel]
+ cnt += 1
+ if cnt > 10:
+ print("target postion == agent postion !")
+ break
agents_positions.pop(sel)
agents_directions.pop(sel)
agents_position.append((p[0], p[1]))
--
GitLab