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

fix interpolator to properly interpolate in 2D space

Browse Source
This commit is contained in:
Pim Nelissen
2026-03-02 12:58:12 +01:00
parent 92789c718f
commit 8e429fe636
1 changed files with 17 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -77,23 +77,27 @@ def calculate_fluence_along_path(
points_per_segment: int = 10
) -> Tuple[np.ndarray, np.ndarray]:
path = landscape.path
num_segments = len(path.segments)
num_points = len(path.x_list)
xnew = np.linspace(
path.x_list[0],
path.x_list[-1],
num=num_segments*points_per_segment)
dx = np.diff(path.x_list)
dy = np.diff(path.y_list)
segment_lengths = np.sqrt(dx**2 + dy**2)
ynew = np.interp(xnew, path.x_list, path.y_list)
original_distances = np.zeros(num_points)
original_distances[1:] = np.cumsum(segment_lengths)
# arc lengths at which to evaluate the path
s = np.linspace(
0,
original_distances[-1],
num=num_points * points_per_segment)
# Interpolate x and y as functions of arc length
xnew = np.interp(s, original_distances, path.x_list)
ynew = np.interp(s, original_distances, path.y_list)
z = np.full(xnew.shape, path.z)
full_positions = np.c_[xnew, ynew, z]
phi_result = calculate_fluence_at(landscape, full_positions)
dist_travelled = np.linspace(
full_positions[0, 0],
path.length,
len(phi_result)
)
return dist_travelled, phi_result
return s, phi_result