public/py-tools
1
0
Fork 0
Code Issues Pull Requests Actions Releases 8 Activity

Grid.range[XYZ]: make ranges reverable

Browse Source
This commit is contained in:
Stefan Harmuth 2021-12-25 06:48:17 +01:00
parent 709b0f471b
commit 5bf2ec1c47
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tools/grid.py
View File

@ -51,16 +51,25 @@ class Grid:
self.minZ = min(pos.z, self.minZ) self.minZ = min(pos.z, self.minZ)
self.maxZ = max(pos.z, self.maxZ) self.maxZ = max(pos.z, self.maxZ)
def rangeX(self, pad: int = 0): def rangeX(self, pad: int = 0, reverse=False):
return range(self.minX - pad, self.maxX + pad + 1) if reverse:
return range(self.maxX + pad, self.minX - pad - 1, -1)
else:
return range(self.minX - pad, self.maxX + pad + 1)
def rangeY(self, pad: int = 0): def rangeY(self, pad: int = 0, reverse=False):
return range(self.minY - pad, self.maxY + pad + 1) if reverse:
return range(self.maxY + pad, self.minY - pad - 1, -1)
else:
return range(self.minY - pad, self.maxY + pad + 1)
def rangeZ(self, pad: int = 0): def rangeZ(self, pad: int = 0, reverse=False):
if not self.mode3D: if not self.mode3D:
raise ValueError("rangeZ not available in 2D space") raise ValueError("rangeZ not available in 2D space")
return range(self.minZ - pad, self.maxZ + pad + 1) if reverse:
return range(self.maxZ + pad, self.minZ - pad - 1, -1)
else:
return range(self.minZ - pad, self.maxZ + pad + 1)
def toggle(self, pos: Coordinate): def toggle(self, pos: Coordinate):
if pos in self.__grid: if pos in self.__grid: