pim/pg-rad
1
0
Fork 0
mirror of https://github.com/pim-n/pg-rad synced 2026-02-02 14:33:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

rename units to scale for the size of the world

Browse Source
This commit is contained in:
Pim Nelissen
2026-01-27 21:09:13 +01:00
parent 7a03251c6f
commit 5a7c2338f3
1 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/pg_rad/landscape.py
View File

@ -6,7 +6,12 @@ from pg_rad.path import Path
from pg_rad.objects import Source
class Landscape:
def __init__(self, size: int | tuple[int, int, int] = 500, unit = 'meters'):
def __init__(
self,
size: int | tuple[int, int, int] = 500,
scale = 'meters',
):
if isinstance(size, int):
self.world = np.zeros((size, size, size))
elif isinstance(size, tuple) and len(size) == 3:
@ -14,7 +19,7 @@ class Landscape:
else:
raise TypeError("size must be an integer or a tuple of 3 integers.")
self.unit = unit
self.scale = scale
self.path: Path = None
self.sources: list[Source] = []
@ -33,8 +38,8 @@ class Landscape:
fig, ax = plt.subplots()
ax.set_xlim(right=x_lim)
ax.set_ylim(top=y_lim)
ax.set_xlabel(f"X [{self.unit}]")
ax.set_ylabel(f"Y [{self.unit}]")
ax.set_xlabel(f"X [{self.scale}]")
ax.set_ylabel(f"Y [{self.scale}]")
if not self.path == None:
ax.plot(self.path.x_list, self.path.y_list, 'bo-')