pim/pg-rad
1
0
Fork 0
mirror of https://github.com/pim-n/pg-rad synced 2026-05-14 07:18:11 +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

18
src/pg_rad/background/background.py
View File

@ -9,11 +9,18 @@ from pg_rad.detector.detector import Detector
def generate_background(
cps_array: np.ndarray,
detector: Detector,
energy_keV: float,
) -> np.ndarray:
"""
Generate synthetic background cps for a given detector and energy.
"""
pass
ROI_lo, ROI_hi = get_roi_from_fwhm(detector, energy_keV)
lam = get_cps_from_roi(detector, ROI_lo, ROI_hi)
rng = np.random.default_rng()
return rng.poisson(lam=lam, size=cps_array.shape)
def fwhm(A: float, B: float, C: float, E: float) -> float:
@ -39,8 +46,13 @@ def get_cps_from_roi(
roi_hi: float
) -> float:
csv = files('pg_rad.data.backgrounds').joinpath(detector.name+'.csv')
data = read_csv(csv)
try:
csv = files('pg_rad.data.backgrounds').joinpath(detector.name+'.csv')
data = read_csv(csv)
except FileNotFoundError:
raise NotImplementedError(
f"Detector {detector.name} does not have backgrounds implemented."
)
# get indices of nearest bins
idx_min = (data["Energy"] - roi_lo).abs().idxmin()