From c4b9b10b710f3f357b1b72557bf6b01675a1e869 Mon Sep 17 00:00:00 2001 From: Stefan Harmuth Date: Wed, 13 Dec 2023 06:37:24 +0100 Subject: [PATCH] Grid.get_column() and Grid.get_row() --- src/tools/grid.py | 6 ++++++ 1 file changed, 6 insertions(+) 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]