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

update plotter to add direction of travel arrow and detector info

Browse Source
This commit is contained in:
Pim Nelissen
2026-03-10 20:45:35 +01:00
parent 938b3a7afc
commit 7dec54fa1c
2 changed files with 62 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
src/pg_rad/plotting/result_plotter.py
View File

@ -16,10 +16,10 @@ class ResultPlotter:
fig = plt.figure(figsize=(12, 10), constrained_layout=True)
fig.suptitle(self.landscape.name)
gs = GridSpec(
3,
4,
2,
width_ratios=[0.5, 0.5],
height_ratios=[0.7, 0.15, 0.15],
height_ratios=[0.7, 0.1, 0.1, 0.1],
hspace=0.2)
ax1 = fig.add_subplot(gs[0, 0])
@ -32,9 +32,11 @@ class ResultPlotter:
self._draw_table(ax3)
ax4 = fig.add_subplot(gs[2, :])
self._draw_source_table(ax4)
self._draw_detector_table(ax4)
ax5 = fig.add_subplot(gs[3, :])
self._draw_source_table(ax5)
plt.tight_layout()
plt.show()
def _plot_landscape(self, ax, z):
@ -57,7 +59,26 @@ class ResultPlotter:
cols = ('Parameter', 'Value')
data = [
["Air density (kg/m^3)", round(self.landscape.air_density, 3)],
["Total path length (m)", round(self.landscape.path.length, 3)]
["Total path length (m)", round(self.landscape.path.length, 3)],
]
ax.table(
cellText=data,
colLabels=cols,
loc='center'
)
return ax
def _draw_detector_table(self, ax):
det = self.landscape.detector
print(det)
ax.set_axis_off()
ax.set_title('Detector')
cols = ('Parameter', 'Value')
data = [
["Name", det.name],
["Type", det.efficiency],
]
ax.table(