Add options to speed up tracking

Author: AntSimi

src/py_eddy_tracker/featured_tracking/area_tracker.py

@@ -6,6 +6,13 @@
 
 
 class AreaTracker(Model):
+
+    @classmethod
+    def needed_variable(cls):
+        vars = ['longitude', 'latitude']
+        vars.extend(cls.intern(False, public_label=True))
+        return vars
+
     def tracking(self, other):
         shape = (self.shape[0], other.shape[0])
         i, j, c = self.match(other, intern=False)

src/py_eddy_tracker/observations/observation.py

@@ -282,6 +282,10 @@ def __getitem__(self, attr):
             return self.observations[attr]
         raise KeyError("%s unknown" % attr)
 
+    @classmethod
+    def needed_variable(cls):
+        return None
+
     @classmethod
     def obs_dimension(cls, handler):
         for candidate in ("obs", "Nobs", "observation", "i"):

src/py_eddy_tracker/tracking.py

@@ -363,12 +363,16 @@ def track(self):
         self.reset_dataset_cache()
         first_dataset, flg_virtual = self.load_state()
 
-        self.swap_dataset(self.datasets[first_dataset - 1])
+        kwargs = dict()
+        needed_variable = self.class_method.needed_variable()
+        if needed_variable is not None:
+            kwargs['include_vars'] = needed_variable
+        self.swap_dataset(self.datasets[first_dataset - 1], **kwargs)
         # We begin with second file, first one is in previous
         for file_name in self.datasets[first_dataset:]:
-            self.swap_dataset(file_name)
-            logger.info('%s match with previous state', file_name)
-            logger.debug('%d obs to match', len(self.current_obs))
+            self.swap_dataset(file_name, **kwargs)
+            logger.info("%s match with previous state", file_name)
+            logger.debug("%d obs to match", len(self.current_obs))
 
             nb_real_obs = len(self.previous_obs)
             if flg_virtual: