optimisation poly_indexs and bug correction with np.nan casting

Author: ludwigVonKoopa

src/py_eddy_tracker/observations/observation.py

@@ -2050,7 +2050,10 @@ def contains(self, x, y, intern=False):
         :rtype: array[int32]
         """
         xname, yname = self.intern(intern)
-        return poly_indexs(x, y, self[xname], self[yname])
+        m = ~ (isnan(x) + isnan(y))
+        i = -ones(x.shape, dtype='i4')
+        i[m] = poly_indexs(x[m], y[m], self[xname], self[yname])
+        return i
 
     def inside(self, x, y, intern=False):
         """

src/py_eddy_tracker/poly.py

@@ -815,7 +815,7 @@ def box_indexes(x, y, step):
 
 
 @njit(cache=True)
-def poly_indexs_(x_p, y_p, x_c, y_c):
+def poly_indexs(x_p, y_p, x_c, y_c):
     """
     Index of contour for each postion inside a contour, -1 in case of no contour
 
@@ -874,7 +874,7 @@ def poly_indexs_(x_p, y_p, x_c, y_c):
 
 
 @njit(cache=True)
-def poly_indexs(x_p, y_p, x_c, y_c):
+def poly_indexs_old(x_p, y_p, x_c, y_c):
     """
     index of contour for each postion inside a contour, -1 in case of no contour