interp: allow extrapolation in interp1d

1 files changed, 1 insertions, 0 deletions

diff --git a/interp.py b/interp.py
index e326e6d..778fc52 100644
--- a/interp.py
+++ b/interp.py
@@ -4,6 +4,7 @@ import numpy as np
 
 def interp1d(src_start, src_step, src_val, dst_coords, stencil):
     idx_src   = ((dst_coords - src_start) / src_step - (stencil / 2 - 1)).astype(int)
+    idx_src   = np.minimum(np.maximum(idx_src, 0), len(src_val) - stencil)
     src_coord = np.linspace(src_start, src_start + src_step * (src_val.shape[0] - 1),
                             src_val.shape[0], dtype = src_val.dtype)