pim/pg-rad
1
0
Fork 0
mirror of https://github.com/pim-n/pg-rad synced 2026-04-24 21:18:10 +02:00
Code Issues Packages Projects Releases Wiki Activity

improve plotting functionality

Browse Source
This commit is contained in:
Pim Nelissen
2026-03-30 08:23:02 +02:00
parent 8098ab9329
commit 0a60bb09e9
2 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/pg_rad/plotting/landscape_plotter.py
View File

@ -58,12 +58,7 @@ class LandscapeSlicePlotter:
ax.set_xlim(right=max(width, .5*height)) ax.set_xlim(right=max(width, .5*height))
# if the road is very flat, we center it vertically (looks better) # if the road is very flat, we center it vertically (looks better)
if median(landscape.path.y_list) == 0: ax.set_ylim(bottom=-.5*width, top=.5*width)
h = max(height, .5*width)
ax.set_ylim(bottom=-h//2,
top=h//2)
else:
ax.set_ylim(top=max(height, .5*width))
ax.set_xlabel("X [m]") ax.set_xlabel("X [m]")
ax.set_ylabel("Y [m]") ax.set_ylabel("Y [m]")

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

@ -37,7 +37,7 @@ class ResultPlotter:
self._draw_cps(ax_cps) self._draw_cps(ax_cps)
ax_counts = fig.add_subplot(gs[0, 1]) ax_counts = fig.add_subplot(gs[0, 1])
self._draw_count_rate(ax_counts) self._draw_counts(ax_counts)
ax_landscape = fig.add_subplot(gs[1, :]) ax_landscape = fig.add_subplot(gs[1, :])
self._plot_landscape(ax_landscape, landscape_z) self._plot_landscape(ax_landscape, landscape_z)
@ -77,15 +77,20 @@ class ResultPlotter:
def _draw_cps(self, ax): def _draw_cps(self, ax):
x = self.count_rate_res.sub_points x = self.count_rate_res.sub_points
y = self.count_rate_res.cps y = self.count_rate_res.cps
ax.plot(x, y, color='b') ax.plot(x, y, color='b', label=f'max(CPS) = {y.max():.2f}')
ax.legend(handlelength=0, handletextpad=0, fancybox=True)
ax.set_title('Counts per second (CPS)') ax.set_title('Counts per second (CPS)')
ax.set_xlabel('Arc length s [m]') ax.set_xlabel('Arc length s [m]')
ax.set_ylabel('CPS [s$^{-1}$]') ax.set_ylabel('CPS [s$^{-1}$]')
def _draw_count_rate(self, ax): def _draw_counts(self, ax):
x = self.count_rate_res.acquisition_points x = self.count_rate_res.acquisition_points[1:]
y = self.count_rate_res.integrated_counts y = self.count_rate_res.integrated_counts[1:]
ax.plot(x, y, color='r', linestyle='--', alpha=0.2) ax.plot(
x, y, color='r', linestyle='--',
alpha=0.2, label=f'max(counts) = {y.max():.2f}'
)
ax.legend(handlelength=0, handletextpad=0, fancybox=True)
ax.scatter(x, y, color='r', marker='x') ax.scatter(x, y, color='r', marker='x')
ax.set_title('Integrated counts') ax.set_title('Integrated counts')
ax.set_xlabel('Arc length s [m]') ax.set_xlabel('Arc length s [m]')
@ -99,6 +104,7 @@ class ResultPlotter:
["Air density (kg/m^3)", round(self.landscape.air_density, 3)], ["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)],
["Readout points", len(self.count_rate_res.integrated_counts)], ["Readout points", len(self.count_rate_res.integrated_counts)],
["Mean background cps", round(self.count_rate_res.mean_bkg_cps, 3)]
] ]
ax.table( ax.table(
@ -124,7 +130,7 @@ class ResultPlotter:
# list field efficiencies for each primary gamma in the landscape # list field efficiencies for each primary gamma in the landscape
effs = {e: det.get_efficiency(e) for e in source_energies} effs = {e: det.get_efficiency(e) for e in source_energies}
formatted_effs = ", ".join( formatted_effs = ", ".join(
f"{value:.3f} @ {key:.1f} keV" f"{value:.5f} @ {key:.1f} keV"
for key, value in effs.items() for key, value in effs.items()
) )
ax.set_axis_off() ax.set_axis_off()
@ -201,6 +207,5 @@ class ResultPlotter:
theta_rad = np.radians(theta_deg) theta_rad = np.radians(theta_deg)
print(theta_rad)
ax.plot(theta_rad, eff) ax.plot(theta_rad, eff)
ax.set_title(f"Rel. angular efficiency @ {energy_keV:.1f} keV") ax.set_title(f"Rel. angular efficiency @ {energy_keV:.1f} keV")