From 84cd5f10db50e35eb92df850ddf5b38a9b09476b Mon Sep 17 00:00:00 2001
From: hagrid67 <jdhwatson@gmail.com>
Date: Sun, 5 May 2019 15:53:11 +0100
Subject: [PATCH] Added "remove_deadends" flag to set_transition in
GridTransitionMap.
---
flatland/core/transition_map.py | 12 +++++++-----
1 file changed, 7 insertions(+), 5 deletions(-)
diff --git a/flatland/core/transition_map.py b/flatland/core/transition_map.py
index bce795e..fd42787 100644
--- a/flatland/core/transition_map.py
+++ b/flatland/core/transition_map.py
@@ -220,7 +220,7 @@ class GridTransitionMap(TransitionMap):
return ()
return self.transitions.get_transition(self.grid[cell_id[0]][cell_id[1]], cell_id[2], transition_index)
- def set_transition(self, cell_id, transition_index, new_transition):
+ def set_transition(self, cell_id, transition_index, new_transition, remove_deadends=False):
"""
Replaces the validity of transition to `transition_index' in cell
`cell_id' with the new `new_transition'.
@@ -244,10 +244,12 @@ class GridTransitionMap(TransitionMap):
print('GridTransitionMap.set_transition() ERROR: \
wrong cell_id tuple.')
return
- self.grid[cell_id[0]][cell_id[1]] = self.transitions.set_transition(self.grid[cell_id[0]][cell_id[1]],
- cell_id[2],
- transition_index,
- new_transition)
+ self.grid[cell_id[0]][cell_id[1]] = self.transitions.set_transition(
+ self.grid[cell_id[0]][cell_id[1]],
+ cell_id[2],
+ transition_index,
+ new_transition,
+ remove_deadends)
def save_transition_map(self, filename):
"""
--
GitLab