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

Add background functionality

Browse Source
This commit is contained in:
Pim Nelissen
2026-03-30 08:24:07 +02:00
parent 0a60bb09e9
commit c635c7f594
6 changed files with 43 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/pg_rad/physics/fluence.py
View File

@ -2,6 +2,8 @@ from typing import Tuple, TYPE_CHECKING
import numpy as np
from pg_rad.background.background import generate_background
if TYPE_CHECKING:
from pg_rad.landscape.landscape import Landscape
from pg_rad.detector.detector import Detector
@ -146,12 +148,17 @@ def calculate_counts_along_path(
landscape, full_positions, detector
)
bkg = generate_background(
cps, detector, landscape.point_sources[0].isotope.E
)
cps_with_bg = cps# + bkg
# reshape so each segment is on a row
cps_per_seg = cps.reshape(num_segments, points_per_segment)
cps_per_seg = cps_with_bg.reshape(num_segments, points_per_segment)
du = s[1] - s[0]
integrated_counts = np.trapezoid(cps_per_seg, dx=du, axis=1) / velocity
int_counts_result = np.zeros(num_points)
int_counts_result[1:] = integrated_counts
return original_distances, s, cps, int_counts_result
return original_distances, s, cps_with_bg, int_counts_result, np.mean(bkg)