From 564b1ce9d104c2b25b1a2aff719327efddc26bb3 Mon Sep 17 00:00:00 2001 From: David Hilditch Date: Mon, 3 Apr 2023 19:24:14 +0100 Subject: Null: Allow specifying geodesic start times. --- null.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/null.py b/null.py index 22f1860..1bdc5ab 100644 --- a/null.py +++ b/null.py @@ -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, -- cgit v1.2.3