From 013c9db959e118baa5ad41b35b83147d0f307385 Mon Sep 17 00:00:00 2001 From: Anton Khirnov Date: Thu, 23 Feb 2023 20:10:32 +0100 Subject: doublenull: import functions from interpolate explicitly --- doublenull.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/doublenull.py b/doublenull.py index 78e5632..6d11bd4 100644 --- a/doublenull.py +++ b/doublenull.py @@ -1,5 +1,5 @@ import numpy as np -import scipy.interpolate as interp +from scipy.interpolate import RectBivariateSpline, interp1d from scipy.integrate import solve_ivp def calc_null_curves(times, spatial_coords, gXX, gXt, gtt, reverse = False): @@ -26,9 +26,9 @@ def calc_null_curves(times, spatial_coords, gXX, gXt, gtt, reverse = False): X=spatial_coords[i]) at time t=ray_times[j]. """ - gXX_interp = interp.RectBivariateSpline(times, spatial_coords, gXX) - gXt_interp = interp.RectBivariateSpline(times, spatial_coords, gXt) - gtt_interp = interp.RectBivariateSpline(times, spatial_coords, gtt) + gXX_interp = RectBivariateSpline(times, spatial_coords, gXX) + gXt_interp = RectBivariateSpline(times, spatial_coords, gXt) + gtt_interp = RectBivariateSpline(times, spatial_coords, gtt) if reverse: ray_times = times[::-1] @@ -109,7 +109,7 @@ def calc_null_coordinates(times, spatial_coords, u_rays, v_rays, for i, t in enumerate(times): Xu = X_of_ut[:, i] Xv = X_of_vt[:, i] - u_of_tx[i] = interp.interp1d(Xu, u_rays, fill_value = 'extrapolate')(spatial_coords) - v_of_tx[i] = interp.interp1d(Xv, v_rays, fill_value = 'extrapolate')(spatial_coords) + u_of_tx[i] = interp1d(Xu, u_rays, fill_value = 'extrapolate')(spatial_coords) + v_of_tx[i] = interp1d(Xv, v_rays, fill_value = 'extrapolate')(spatial_coords) return (u_of_tx, v_of_tx) -- cgit v1.2.3