check mask #145

Author: AntSimi

src/py_eddy_tracker/dataset/grid.py

@@ -402,6 +402,14 @@ def load(self):
 
         self.setup_coordinates()
 
+    @staticmethod
+    def get_mask(a):
+        if len(a.mask.shape):
+            m = a.mask
+        else:
+            m = ones(a.shape, dtype='bool') if a.mask else zeros(a.shape, dtype='bool')
+        return m
+
     @staticmethod
     def c_to_bounds(c):
         """
@@ -1126,7 +1134,7 @@ def _low_filter(self, grid_name, w_cut, factor=8.0):
         bins = (x_array, y_array)
 
         x_flat, y_flat, z_flat = x.reshape((-1,)), y.reshape((-1,)), data.reshape((-1,))
-        m = ~z_flat.mask
+        m = ~self.get_mask(z_flat)
         x_flat, y_flat, z_flat = x_flat[m], y_flat[m], z_flat[m]
 
         nb_value, _, _ = histogram2d(x_flat, y_flat, bins=bins)
@@ -1936,14 +1944,6 @@ def regrid(self, other, grid_name, new_name=None):
         # self.variables_description[new_name]['infos'] = False
         # self.variables_description[new_name]['kwargs']['dimensions'] = ...
 
-    @staticmethod
-    def get_mask(a):
-        if len(a.mask.shape):
-            m = a.mask
-        else:
-            m = ones(a.shape) if a.mask else zeros(a.shape)
-        return m
-
     def interp(self, grid_name, lons, lats, method="bilinear"):
         """
         Compute z over lons, lats