Merge pull request #196 from USDA-ARS-NWRC/wind_ninja_interp

Wind ninja interpolation
USDA-ARS-NWRC · Nov 3, 2020 · 36d2c16 · 36d2c16
2 parents 736219b + 704961d
commit 36d2c16
Show file tree

Hide file tree

Showing 6 changed files with 388 additions and 12 deletions.
diff --git a/smrf/distribute/wind/wind_ninja.py b/smrf/distribute/wind/wind_ninja.py
@@ -264,20 +264,22 @@ def fill_data(self, g_vel):
         iy = np.sum(np.isnan(g_vel[:, ix+1]))
 
         #  first go in the Y direction
-        yi = g_vel[:, ix:ix+10]
-        xi = self.X[0, ix:ix+10]
-        x = self.X[0, :ix]
+        if ix > 0:
+            yi = g_vel[:, ix:ix+10]
+            xi = self.X[0, ix:ix+10]
+            x = self.X[0, :ix]
 
-        o = np.apply_along_axis(interpx, axis=1, arr=yi, xi=xi, x=x)
-        g_vel[:, :ix] = o
+            o = np.apply_along_axis(interpx, axis=1, arr=yi, xi=xi, x=x)
+            g_vel[:, :ix] = o
 
         #  first go in the X direction
-        yi = g_vel[-iy-10:-iy, :]
-        xi = self.Y[-iy-10:-iy, 0]
-        x = self.Y[-iy:, 0]
+        if iy > 0:
+            yi = g_vel[-iy-10:-iy, :]
+            xi = self.Y[-iy-10:-iy, 0]
+            x = self.Y[-iy:, 0]
 
-        o = np.apply_along_axis(interpx, axis=0, arr=yi, xi=xi, x=x)
-        g_vel[-iy:, :] = o
+            o = np.apply_along_axis(interpx, axis=0, arr=yi, xi=xi, x=x)
+            g_vel[-iy:, :] = o
 
         if np.any(np.isnan(g_vel)):
             raise ValueError('WindNinja data still has NaN values')