Null: Allow specifying geodesic start times.

1 files 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,