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

vectorize fluence calculation

Browse Source
This commit is contained in:
Pim Nelissen
2026-02-25 14:13:27 +01:00
parent c18f924348
commit bab41c128b
1 changed files with 38 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,4 +1,4 @@
from typing import TYPE_CHECKING from typing import Tuple, TYPE_CHECKING
import numpy as np import numpy as np
@ -42,10 +42,22 @@ def phi(
return phi_r return phi_r
def calculate_fluence_at(landscape: "Landscape", pos: tuple): def calculate_fluence_at(landscape: "Landscape", pos: np.ndarray):
total_phi = 0. """Compute fluence at an arbitrary position in the landscape.
Args:
landscape (Landscape): The landscape to compute.
pos (np.ndarray): (N, 3) array of positions.
Returns:
total_phi (np.ndarray): (N,) array of fluences.
"""
total_phi = np.zeros(pos.shape[0])
for source in landscape.point_sources: for source in landscape.point_sources:
r = source.distance_to(pos) r = np.linalg.norm(pos - np.array(source.pos), axis=1)
r = np.maximum(r, 1e-3) # enforce minimum distance of 1cm
phi_source = phi( phi_source = phi(
r=r, r=r,
activity=source.activity, activity=source.activity,
@ -53,30 +65,34 @@ def calculate_fluence_at(landscape: "Landscape", pos: tuple):
mu_mass_air=source.isotope.mu_mass_air, mu_mass_air=source.isotope.mu_mass_air,
air_density=landscape.air_density air_density=landscape.air_density
) )
total_phi += phi_source total_phi += phi_source
return total_phi return total_phi
def calculate_fluence_along_path( def calculate_fluence_along_path(
landscape: "Landscape", landscape: "Landscape",
points_per_segment: int = 10 points_per_segment: int = 10
): ) -> Tuple[np.ndarray, np.ndarray]:
phi_result = []
path = landscape.path path = landscape.path
waypoints = list(zip(path.x_list, path.y_list)) num_segments = len(path.segments)
for w, wp1 in zip(waypoints, waypoints[1:]): xnew = np.linspace(
x_ref, y_ref = zip(w, wp1) path.x_list[0],
path.x_list[-1],
num=num_segments*points_per_segment)
x = np.linspace(x_ref[0], x_ref[1], points_per_segment) ynew = np.interp(xnew, path.x_list, path.y_list)
y = np.interp(x, x_ref, y_ref)
z = np.full(x.shape, fill_value=path.z)
for pos in zip(x, y, z): z = np.full(xnew.shape, path.z)
phi_segment = calculate_fluence_at(landscape, pos) full_positions = np.c_[xnew, ynew, z]
phi_result.append(phi_segment) phi_result = calculate_fluence_at(landscape, full_positions)
return phi_result dist_travelled = np.linspace(
full_positions[0, 0],
path.length,
len(phi_result)
)
return dist_travelled, phi_result