NEW Grid.getRegion() -> Generator over all Coordinates of connected cells with the same value inside Grid borders

This commit is contained in:
Stefan Harmuth 2023-12-18 07:00:05 +01:00
parent b2cc1e814c
commit 02623f8c9a

View File

@ -229,6 +229,20 @@ class Grid:
else:
return self.__grid.keys()
def getRegion(self, start: Coordinate, includeDiagonal: bool = False) -> Iterable[Coordinate]:
start_value = self.get(start)
queue = deque()
queue.append(start)
visited = set()
while queue:
next_coord = queue.popleft()
if next_coord in visited or not self.isWithinBoundaries(next_coord) or self.get(next_coord) != start_value:
continue
visited.add(next_coord)
yield next_coord
for n in self.getNeighboursOf(next_coord, includeDefault=True, includeDiagonal=includeDiagonal):
queue.append(n)
def getActiveRegion(
self,
start: Coordinate,