Add method to found close center

Author: AntSimi

src/py_eddy_tracker/observations/observation.py

@@ -893,24 +893,38 @@ def intern(flag, public_label=False):
             labels = [VAR_DESCR[label]["nc_name"] for label in labels]
         return labels
 
-    def match(self, other, intern=False, cmin=0):
+    def match(self, other, method="overlap", intern=False, cmin=0, **kwargs):
         """return index and score compute with area
 
         :param EddiesObservations other: Observations to compare
+        :param str method:
+            if method is "overlap" method will use contour to compute score,
+            if method is "circle" method will apply a formula of circle overlap
         :param bool intern: if True, speed contour will be used
         :param float cmin: 0 < cmin < 1, return only couple above cmin
+        :param dict kwargs: look at :py:meth:`vertice_overlap`
         :return: return index of couple in self and other and cost value
         :rtype: (array(int), array(int), array(float))
 
         .. minigallery:: py_eddy_tracker.EddiesObservations.match
         """
         x_name, y_name = self.intern(intern)
-        i, j = bbox_intersection(
-            self[x_name], self[y_name], other[x_name], other[y_name]
-        )
-        c = vertice_overlap(
-            self[x_name][i], self[y_name][i], other[x_name][j], other[y_name][j]
-        )
+        if method == "overlap":
+            i, j = bbox_intersection(
+                self[x_name], self[y_name], other[x_name], other[y_name]
+            )
+            c = vertice_overlap(
+                self[x_name][i],
+                self[y_name][i],
+                other[x_name][j],
+                other[y_name][j],
+                **kwargs,
+            )
+        elif method == "close_center":
+            i, j, c = close_center(
+                self.latitude, self.longitude, other.latitude, other.longitude, **kwargs
+            )
+
         m = c > cmin
         return i[m], j[m], c[m]
 

src/py_eddy_tracker/poly.py

@@ -171,6 +171,31 @@ def winding_number_grid_in_poly(x_1d, y_1d, i_x0, i_x1, x_size, i_y0, xy_poly):
     return i_x, i_y
 
 
+@njit(cache=True, fastmath=True)
+def close_center(x0, y0, x1, y1, delta=.1):
+    """
+    Compute an overlap with circle parameter and return a percentage
+
+    :param array x0: x centers of dataset 0
+    :param array y0: y centers of dataset 0
+    :param array x1: x centers of dataset 1
+    :param array y1: y centers of dataset 1
+    :return: Result of cost function
+    :rtype: array
+    """
+    nb0, nb1 = x0.shape[0], x1.shape[0]
+    i, j, c = list(), list(), list()
+    for i0 in range(nb0):
+        xi0, yi0 = x0[i0], y0[i0]
+        for i1 in range(nb1):
+            if abs(x1[i1] - xi0) > delta:
+                continue
+            if abs(y1[i1] - yi0) > delta:
+                continue
+            i.append(i0), j.append(i1), c.append(1)
+    return array(i), array(j), array(c)
+
+
 @njit(cache=True, fastmath=True)
 def bbox_intersection(x0, y0, x1, y1):
     """