pim/pg-rad
1
0
Fork 0
mirror of https://github.com/pim-n/pg-rad synced 2026-03-11 19:58:11 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add LandscapeDirector with a test case to build landscape using LandscapeBuilder

Browse Source
This commit is contained in:
Pim Nelissen
2026-02-13 14:57:37 +01:00
parent e8bf687563
commit a4fb4a7c57
3 changed files with 44 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/pg_rad/landscape/landscape.py
View File

@ -16,6 +16,7 @@ class Landscape:
"""
def __init__(
self,
name: str,
path: Path,
point_sources: list[PointSource] = [],
size: tuple[int, int, int] = [500, 500, 50],
@ -35,12 +36,13 @@ class Landscape:
TypeError: _description_
"""
self.name = name
self.path = path
self.point_sources = point_sources
self.size = size
self.air_density = air_density
logger.debug("Landscape initialized.")
logger.debug(f"Landscape created: {self.name}")
def calculate_fluence_at(self, pos: tuple):
total_phi = 0.
@ -61,12 +63,15 @@ class Landscape:
class LandscapeBuilder:
def __init__(self):
def __init__(self, name: str = "Unnamed landscape"):
self.name = name
self._path = None
self._point_sources = []
self._size = None
self._air_density = None
logger.debug(f"LandscapeBuilder initialized: {self.name}")
def set_air_density(self, air_density) -> Self:
"""Set the air density of the world."""
self._air_density = air_density
@ -95,7 +100,8 @@ class LandscapeBuilder:
self._path = path_from_RT90(
df=df,
east_col=east_col,
north_col=north_col
north_col=north_col,
z=z
)
# The size of the landscape will be updated if
@ -108,11 +114,12 @@ class LandscapeBuilder:
if needs_resize:
if not self._size:
logger.info("Landscape size set to path dimensions.")
logger.debug("Because no Landscape size was set, "
"it will now set to path dimensions.")
else:
logger.warning(
"Path exceeds current landscape size. "
"Expanding landscape to accommodate it."
"Landscape size will be expanded to accommodate path."
)
self.set_landscape_size(self._path.size)
@ -142,9 +149,13 @@ class LandscapeBuilder:
self._point_sources = sources
def build(self):
return Landscape(
landscape = Landscape(
name=self.name,
path=self._path,
point_sources=self._point_sources,
size=self._size,
air_density=self._air_density
)
logger.info(f"Landscape built successfully: {landscape.name}")
return landscape