Allow to keep value on masked area

Author: AntSimi

src/py_eddy_tracker/dataset/grid.py

@@ -1124,14 +1124,14 @@ def convolve_filter_with_dynamic_kernel(self, grid, kernel_func, lat_max=85, ext
             return out.astype(data.dtype)
         return out
 
-    def lanczos_high_filter(self, grid_name, wave_length, order=1, lat_max=85):
+    def lanczos_high_filter(self, grid_name, wave_length, order=1, lat_max=85, **kwargs):
         data_out = self.convolve_filter_with_dynamic_kernel(
-            grid_name, self.kernel_lanczos, lat_max=lat_max, wave_length=wave_length, order=order)
+            grid_name, self.kernel_lanczos, lat_max=lat_max, wave_length=wave_length, order=order, **kwargs)
         self.vars[grid_name] -= data_out
 
-    def lanczos_low_filter(self, grid_name, wave_length, order=1, lat_max=85):
+    def lanczos_low_filter(self, grid_name, wave_length, order=1, lat_max=85, **kwargs):
         data_out = self.convolve_filter_with_dynamic_kernel(
-            grid_name, self.kernel_lanczos, lat_max=lat_max, wave_length=wave_length, order=order)
+            grid_name, self.kernel_lanczos, lat_max=lat_max, wave_length=wave_length, order=order, **kwargs)
         self.vars[grid_name] = data_out
 
     def bessel_band_filter(self, grid_name, wave_length_inf, wave_length_sup, **kwargs):
@@ -1142,17 +1142,17 @@ def bessel_band_filter(self, grid_name, wave_length_inf, wave_length_sup, **kwar
             grid_name, self.kernel_bessel, wave_length=wave_length_sup, **kwargs)
         self.vars[grid_name] -= data_out
 
-    def bessel_high_filter(self, grid_name, wave_length, order=1, lat_max=85):
+    def bessel_high_filter(self, grid_name, wave_length, order=1, lat_max=85, **kwargs):
         logger.debug('Run filtering with wave of %(wave_length)s km and order of %(order)s ...',
                       dict(wave_length=wave_length, order=order))
         data_out = self.convolve_filter_with_dynamic_kernel(
-            grid_name, self.kernel_bessel, lat_max=lat_max, wave_length=wave_length, order=order)
+            grid_name, self.kernel_bessel, lat_max=lat_max, wave_length=wave_length, order=order, **kwargs)
         logger.debug('Filtering done')
         self.vars[grid_name] -= data_out
 
-    def bessel_low_filter(self, grid_name, wave_length, order=1, lat_max=85):
+    def bessel_low_filter(self, grid_name, wave_length, order=1, lat_max=85, **kwargs):
         data_out = self.convolve_filter_with_dynamic_kernel(
-            grid_name, self.kernel_bessel, lat_max=lat_max, wave_length=wave_length, order=order)
+            grid_name, self.kernel_bessel, lat_max=lat_max, wave_length=wave_length, order=order, **kwargs)
         self.vars[grid_name] = data_out
 
     def spectrum_lonlat(self, grid_name, area=None, ref=None, **kwargs):

src/scripts/GridFiltering

@@ -18,6 +18,8 @@ def id_parser():
                         help='Wavelength for mesoscale filter in km')
     parser.add_argument('--filter_order', default=3, type=int)
     parser.add_argument('--low', action='store_true')
+    parser.add_argument('--extend', default=0, type=float,
+                        help='Keep pixel compute by filtering on mask')
     return parser
 
 
@@ -26,7 +28,7 @@ if __name__ == '__main__':
 
     h = RegularGridDataset(args.filename, args.longitude, args.latitude)
     if args.low:
-        h.bessel_low_filter(args.grid, args.cut_wavelength, order=args.filter_order)
+        h.bessel_low_filter(args.grid, args.cut_wavelength, order=args.filter_order, extend=args.extend)
     else:
-        h.bessel_high_filter(args.grid, args.cut_wavelength, order=args.filter_order)
+        h.bessel_high_filter(args.grid, args.cut_wavelength, order=args.filter_order, extend=args.extend)
     h.write(args.filename_out)