feat: propogate "mode" argument to np.pad

Author: william-silversmith

kimimaro/utility.py

@@ -189,7 +189,7 @@ def cross_sectional_area(
   return skeletons
 
 # From SO: https://stackoverflow.com/questions/14313510/how-to-calculate-rolling-moving-average-using-python-numpy-scipy
-def moving_average(a:np.ndarray, n:int) -> np.ndarray:
+def moving_average(a:np.ndarray, n:int, mode:str = "symmetric") -> np.ndarray:
   if n <= 0:
     raise ValueError(f"Window size ({n}), must be >= 1.")
   elif n == 1:
@@ -199,9 +199,9 @@ def moving_average(a:np.ndarray, n:int) -> np.ndarray:
     return a
 
   if a.ndim == 2:
-    a = np.pad(a, [[n, n],[0,0]], mode="symmetric")
+    a = np.pad(a, [[n, n],[0,0]], mode=mode)
   else:
-    a = np.pad(a, [n, n], mode="symmetric")
+    a = np.pad(a, [n, n], mode=mode)
 
   ret = np.cumsum(a, dtype=float, axis=0)
   ret = (ret[n:] - ret[:-n])[:-n]