diff options
Diffstat (limited to 'null.py')
-rw-r--r-- | null.py | 6 |
1 files changed, 4 insertions, 2 deletions
@@ -164,7 +164,7 @@ def _events_bnd_null_geodesics(times, spatial_coords): def null_geodesics(times, X, gXX, gtt, gtX = None, - pu_t_0 = None, interp_order = 6): + pu_t_0 = None, interp_order = 6, integrate_times=None): """ Compute null geodesics along a axis, assuming zero shift. @@ -222,7 +222,9 @@ def null_geodesics(times, X, gXX, gtt, events = _events_bnd_null_geodesics(times, X) ret = [] - for i, t0 in enumerate(times): + if integrate_times is None: + integrate_times = times + for i, t0 in enumerate(integrate_times): pu_t_0_val = 1.0 if pu_t_0 is None else pu_t_0[i] ret.append(_null_geodesic_kernel(times, t0, events, pu_t_0_val, gtt_interp, gtt_dX_interp, gtt_dt_interp, |