Rewrite command to store array variable

Author: adelepoulle

share/eddy_identification.yaml

@@ -90,6 +90,11 @@ EVOLVE_AREA_MAX: 500 # max change in area
 # - profiles of swirl velocity from effective contour inwards
 # Useful for working with ARGO data    
 TRACK_EXTRA_VARIABLES: No
+TRACK_ARRAY_VARIABLES:
+    NB_SAMPLES: 10
+    VARIABLES:
+        - contour_lon
+        - contour_lat
 
 # 'RectBivariate' is faster
 INTERP_METHOD: 'RectBivariate'

src/py_eddy_tracker/__init__.py

@@ -243,6 +243,26 @@ def parse_args(self, *args, **kwargs):
             description='observation sequence number (XX day intervals)',
             )
         ),
+    contour_lon=dict(
+        attr_name=None,
+        nc_name='contour_lon',
+        nc_type='f4',
+        nc_dims=('Nobs', 'NbSample'),
+        nc_attr=dict(
+            long_name='Longitude of contour',
+            description='lons of contour points',
+            )
+        ),
+    contour_lat=dict(
+        attr_name=None,
+        nc_name='contour_lat',
+        nc_type='f4',
+        nc_dims=('Nobs', 'NbSample'),
+        nc_attr=dict(
+            long_name='Latitude of contour',
+            description='lats of contour points',
+            )
+        ),
     contour_e=dict(
         attr_name=None,
         nc_name='contour_e',

src/py_eddy_tracker/make_eddy_tracker_list_obj.py

@@ -673,6 +673,9 @@ def __init__(self, sign_type, grd, date, **kwargs):
         self.diagnostic_type = kwargs.get('DIAGNOSTIC_TYPE', 'SLA')
         self.the_domain = kwargs.get('THE_DOMAIN', 'Regional')
         self.track_extra_variables = kwargs.get('TRACK_EXTRA_VARIABLES', False)
+        array_properties = kwargs.get('TRACK_ARRAY_VARIABLES', dict())
+        self.track_array_variables_sampling = array_properties.get('NB_SAMPLES', 0)
+        self.track_array_variables = array_properties.get('VARIABLES', [])
         self.smoothing = kwargs.get('SMOOTHING', True)
         self.max_local_extrema = kwargs.get('MAX_LOCAL_EXTREMA', 1)
         self.interp_method = kwargs.get('INTERP_METHOD', 'RectBivariate')

src/py_eddy_tracker/py_eddy_tracker_classes.py

@@ -467,6 +467,9 @@ def collection_loop(contours, grd, rtime, a_list_obj, c_list_obj,
                     args = (eddy, contours, centlon_e, centlat_e, cont, grd,
                             anticyclonic_search)
 
+                    #~ if eddy.track_array_variables > 10:
+                        #~ print ':)'
+                        #~ exit()
                     if not eddy.track_extra_variables:
                         # (uavg, centlon_s, centlat_s,
                         # eddy_radius_s, contlon_s, contlat_s,