add a function to filter position

Author: AntSimi

examples/06_grid_manipulation/README.rst

@@ -0,0 +1,2 @@
+Grid Manipulation
+=======================

examples/06_grid_manipulation/pet_hide_pixel_out_eddies.py

@@ -0,0 +1,44 @@
+"""
+Select pixel in eddies
+======================
+
+"""
+
+from matplotlib import pyplot as plt
+from matplotlib.path import Path
+from numpy import ones
+from py_eddy_tracker.observations.observation import EddiesObservations, custom_concat
+from py_eddy_tracker.dataset.grid import RegularGridDataset
+from py_eddy_tracker import data
+
+a = EddiesObservations.load_file(data.get_path("Anticyclonic_20190223.nc"))
+g = RegularGridDataset(
+    data.get_path("nrt_global_allsat_phy_l4_20190223_20190226.nc"),
+    "longitude",
+    "latitude",
+)
+
+# Plot
+fig = plt.figure(figsize=(12, 6))
+ax = fig.add_axes((0.05, 0.05, 0.9, 0.9))
+ax.set_aspect("equal")
+ax.set_xlim(10, 70)
+ax.set_ylim(-50, -25)
+x_name, y_name = a.intern(False)
+adt = g.grid("adt")
+mask = ones(adt.shape, dtype='bool')
+for eddy in a:
+    i, j = Path(custom_concat(eddy[x_name], eddy[y_name])).pixels_in(g)
+    mask[i, j] = False
+adt.mask[:] += ~mask
+g.display(ax, "adt")
+a.display(ax, label="Anticyclonic", color="g", lw=1, extern_only=True)
+
+fig = plt.figure(figsize=(12, 6))
+ax = fig.add_axes((0.05, 0.05, 0.9, 0.9))
+ax.set_aspect("equal")
+ax.set_xlim(10, 70)
+ax.set_ylim(-50, -25)
+adt.mask[:] = mask
+g.display(ax, "adt")
+a.display(ax, label="Anticyclonic", color="g", lw=1, extern_only=True)

examples/00_tracking_manipulation/README.rst renamed to examples/08_tracking_manipulation/README.rst

@@ -17,11 +17,8 @@
 
 a = a.extract_with_length((7 * 20, -1))
 c = c.extract_with_length((7 * 20, -1))
-a.median_filter(1, 'time', 'lon').loess_filter(5, 'time', 'lon')
-a.median_filter(1, 'time', 'lat').loess_filter(5, 'time', 'lat')
-c.median_filter(1, 'time', 'lon').loess_filter(5, 'time', 'lon')
-c.median_filter(1, 'time', 'lat').loess_filter(5, 'time', 'lat')
-
+a.position_filter(median_half_window=1, loess_half_window=5)
+c.position_filter(median_half_window=1, loess_half_window=5)
 
 # Plot
 fig = plt.figure(figsize=(12, 5))

examples/00_tracking_manipulation/pet_display_field.py renamed to examples/08_tracking_manipulation/pet_display_field.py

@@ -13,13 +13,23 @@
 
 eddy = a.extract_ids([9672])
 eddy_f = a.extract_ids([9672])
-eddy_f.median_filter(1, "time", "lon").loess_filter(5, "time", "lon")
-eddy_f.median_filter(1, "time", "lat").loess_filter(5, "time", "lat")
+eddy_f.position_filter(median_half_window=1, loess_half_window=5)
+
 fig = plt.figure(figsize=(12, 5))
-ax = fig.add_axes((0.05, 0.1, 0.9, 0.9))
+ax = fig.add_axes((0.05, 0.05, 0.9, 0.9))
 ax.set_xlim(17.5, 22)
 ax.set_ylim(35, 36.5)
 ax.set_aspect("equal")
 ax.grid()
 eddy.plot(ax, color="r", lw=0.5)
-eddy_f.plot(ax, color="g", lw=1)
+eddy_f.scatter(ax, "n", cmap="jet", s=80)
+
+fig = plt.figure(figsize=(12, 5))
+ax = fig.add_axes((0.05, 0.05, 0.9, 0.9))
+ax.set_xlim(17, 23)
+ax.set_ylim(34.5, 37)
+ax.set_aspect("equal")
+ax.grid()
+eddy.plot(ax, color="r", lw=0.5, label='track')
+eddy.index(range(0, len(eddy), 40)).display(ax, intern_only=True, label='observations every 40')
+ax.legend()

examples/00_tracking_manipulation/pet_display_track.py renamed to examples/08_tracking_manipulation/pet_display_track.py

@@ -10,8 +10,7 @@
 c = TrackEddiesObservations.load_file(
     py_eddy_tracker_sample.get_path("eddies_med_adt_allsat_dt2018/Cyclonic.zarr")
 )
-c.median_filter(1, "time", "lon").loess_filter(5, "time", "lon")
-c.median_filter(1, "time", "lat").loess_filter(5, "time", "lat")
+c.position_filter(median_half_window=1, loess_half_window=5)
 x0, x1, y0, y1 = 3, 4, 37, 38
 area = dict(llcrnrlon=x0, llcrnrlat=y0, urcrnrlon=x1, urcrnrlat=y1)
 c_subset = c.extract_with_area(area, full_path=True)

examples/00_tracking_manipulation/pet_one_track.py renamed to examples/08_tracking_manipulation/pet_one_track.py

@@ -34,10 +34,8 @@ def cum_distance_by_track(distance, track):
     py_eddy_tracker_sample.get_path("eddies_med_adt_allsat_dt2018/Cyclonic.zarr")
 )
 
-a.median_filter(1, "time", "lon").loess_filter(5, "time", "lon")
-a.median_filter(1, "time", "lat").loess_filter(5, "time", "lat")
-c.median_filter(1, "time", "lon").loess_filter(5, "time", "lon")
-c.median_filter(1, "time", "lat").loess_filter(5, "time", "lat")
+a.position_filter(median_half_window=1, loess_half_window=5)
+c.position_filter(median_half_window=1, loess_half_window=5)
 
 d_a = distance(a.longitude[:-1], a.latitude[:-1], a.longitude[1:], a.latitude[1:])
 d_c = distance(c.longitude[:-1], c.latitude[:-1], c.longitude[1:], c.latitude[1:])

examples/00_tracking_manipulation/pet_select_track_across_area.py renamed to examples/08_tracking_manipulation/pet_select_track_across_area.py

@@ -1247,26 +1247,29 @@ def scatter(self, ax, name, ref=None, ** kwargs):
             x = (x - ref) % 360 + ref
         return ax.scatter(x, self.latitude, c=self[name], **kwargs)
 
-    def display(self, ax, ref=None, extern_only=False, **kwargs):
+    def display(self, ax, ref=None, extern_only=False, intern_only=False, **kwargs):
         if not extern_only:
             lon_s = flatten_line_matrix(self.obs["contour_lon_s"])
             lat_s = flatten_line_matrix(self.obs["contour_lat_s"])
-        lon_e = flatten_line_matrix(self.obs["contour_lon_e"])
-        lat_e = flatten_line_matrix(self.obs["contour_lat_e"])
+        if not intern_only:
+            lon_e = flatten_line_matrix(self.obs["contour_lon_e"])
+            lat_e = flatten_line_matrix(self.obs["contour_lat_e"])
         if "label" in kwargs:
-            kwargs["label"] += " (%s eddies)" % len(self)
+            kwargs["label"] += " (%s observations)" % len(self)
         kwargs_e = kwargs.copy()
         if not extern_only:
             kwargs_e.pop("label", None)
         if ref is None:
             if not extern_only:
                 ax.plot(lon_s, lat_s, **kwargs)
-            ax.plot(lon_e, lat_e, linestyle="-.", **kwargs_e)
+            if not intern_only:
+                ax.plot(lon_e, lat_e, linestyle="-.", **kwargs_e)
         else:
             # FIXME : ref could split eddies
             if not extern_only:
                 ax.plot((lon_s - ref) % 360 + ref, lat_s, **kwargs)
-            ax.plot((lon_e - ref) % 360 + ref, lat_e, linestyle="-.", **kwargs_e)
+            if not intern_only:
+                ax.plot((lon_e - ref) % 360 + ref, lat_e, linestyle="-.", **kwargs_e)
 
     def grid_count(self, bins, intern=False, center=False):
         x_name, y_name = self.intern(intern)