mirror of
https://github.com/pim-n/pg-rad
synced 2026-03-11 19:58:11 +01:00
76 lines
2.1 KiB
Python
76 lines
2.1 KiB
Python
import logging
|
|
|
|
from matplotlib import pyplot as plt
|
|
from matplotlib.patches import Circle
|
|
|
|
from pg_rad.landscape import Landscape
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class LandscapeSlicePlotter:
|
|
def plot(self, landscape: Landscape, z: int = 0):
|
|
"""Plot a top-down slice of the landscape at a height z.
|
|
|
|
Args:
|
|
landscape (Landscape): the landscape to plot
|
|
z (int, optional): Height at which to plot slice. Defaults to 0.
|
|
""" """
|
|
|
|
"""
|
|
self.z = z
|
|
fig, ax = plt.subplots()
|
|
|
|
self._draw_base(ax, landscape)
|
|
self._draw_path(ax, landscape)
|
|
self._draw_point_sources(ax, landscape)
|
|
|
|
ax.set_aspect("equal")
|
|
plt.show()
|
|
|
|
def _draw_base(self, ax, landscape):
|
|
width, height = landscape.size[:2]
|
|
ax.set_xlim(right=width)
|
|
ax.set_ylim(top=height)
|
|
ax.set_xlabel("X [m]")
|
|
ax.set_ylabel("Y [m]")
|
|
ax.set_title(f"Landscape (top-down, z = {self.z})")
|
|
|
|
def _draw_path(self, ax, landscape):
|
|
if landscape.path.z < self.z:
|
|
ax.plot(landscape.path.x_list, landscape.path.y_list, 'bo-')
|
|
else:
|
|
logger.warning(
|
|
"Path is above the slice height z."
|
|
"It will not show on the plot."
|
|
)
|
|
|
|
def _draw_point_sources(self, ax, landscape):
|
|
for s in landscape.point_sources:
|
|
if s.z <= self.z:
|
|
dot = Circle(
|
|
(s.x, s.y),
|
|
radius=5,
|
|
color=s.color,
|
|
zorder=5
|
|
)
|
|
|
|
ax.text(
|
|
s.x + 0.06,
|
|
s.y + 0.06,
|
|
s.name,
|
|
color=s.color,
|
|
fontsize=10,
|
|
ha="left",
|
|
va="bottom",
|
|
zorder=6
|
|
)
|
|
|
|
ax.add_patch(dot)
|
|
else:
|
|
logger.warning(
|
|
f"Source {s.name} is above slice height z."
|
|
"It will not show on the plot."
|
|
)
|