diff --git a/src/tools/grid.py b/src/tools/grid.py index 8647561..0cd5fef 100644 --- a/src/tools/grid.py +++ b/src/tools/grid.py @@ -102,6 +102,12 @@ class Grid: else: return range(self.minZ - pad, self.maxZ + pad + 1) + def get_column(self, column: int) -> list[Any]: + return [self.get(Coordinate(column, y)) for y in self.rangeY()] + + def get_row(self, row: int) -> list[Any]: + return [self.get(Coordinate(x, row)) for x in self.rangeX()] + def toggle(self, pos: Coordinate): if pos in self.__grid: del self.__grid[pos]