NEW Grid.getRegion() -> Generator over all Coordinates of connected cells with the same value inside Grid borders
This commit is contained in:
parent
b2cc1e814c
commit
02623f8c9a
@ -229,6 +229,20 @@ class Grid:
|
|||||||
else:
|
else:
|
||||||
return self.__grid.keys()
|
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(
|
def getActiveRegion(
|
||||||
self,
|
self,
|
||||||
start: Coordinate,
|
start: Coordinate,
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user