From 4532fa0b00568c11632adffda87b707d205caee1 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Sun, 18 Aug 2019 12:42:48 -0400
Subject: [PATCH] increased feasibility probability of levels
---
flatland/core/transition_map.py | 11 +++++++++--
1 file changed, 9 insertions(+), 2 deletions(-)
diff --git a/flatland/core/transition_map.py b/flatland/core/transition_map.py
index dbb68a73..8c0bcb6d 100644
--- a/flatland/core/transition_map.py
+++ b/flatland/core/transition_map.py
@@ -455,8 +455,15 @@ class GridTransitionMap(TransitionMap):
self.set_transition((rcPos[0], rcPos[1], mirror(connect_directions[2])), connect_directions[0], 1)
# Make a cross
if number_of_incoming == 4:
- for direction in range(4):
- self.set_transition((grcPos[0], grcPos[1], direction), direction, 1)
+ connect_directions = np.arange(4)
+ self.set_transition((rcPos[0], rcPos[1], connect_directions[0]), connect_directions[0], 1)
+ self.set_transition((rcPos[0], rcPos[1], connect_directions[0]), connect_directions[1], 1)
+ self.set_transition((rcPos[0], rcPos[1], connect_directions[1]), connect_directions[0], 1)
+ self.set_transition((rcPos[0], rcPos[1], connect_directions[1]), connect_directions[1], 1)
+ self.set_transition((rcPos[0], rcPos[1], connect_directions[2]), connect_directions[2], 1)
+ self.set_transition((rcPos[0], rcPos[1], connect_directions[2]), connect_directions[3], 1)
+ self.set_transition((rcPos[0], rcPos[1], connect_directions[3]), connect_directions[2], 1)
+ self.set_transition((rcPos[0], rcPos[1], connect_directions[3]), connect_directions[3], 1)
return True
--
GitLab