jax_sn.analytical_solutions¶
- psi_isotropic_box(points: ndarray, box_size: float, omega: ndarray, sigma: float, q: float)[source]¶
Compute the analytical solution of the transport equation in a box with isotropic source and absorption.
- Parameters:
points (
ndarray) – The points at which to compute the solution.box_size (
float) – The size of the box (assumed to be a cube from (0,0,0) to (a,a,a)).omega (
ndarray) – The direction of travel.sigma (
float) – The absorption cross section.q (
float) – The isotropic source strength.
- Returns:
psi – The angular flux at the given points.
- Return type:
onp.ndarray [n_points]