Extract with a bounding box

Author: AntSimi

src/py_eddy_tracker/observations/observation.py

@@ -174,6 +174,14 @@ def __init__(
         self.active = True
         self.sign_type = None
 
+    @property
+    def longitude(self):
+        return self.observations["lon"]
+
+    @property
+    def latitude(self):
+        return self.observations["lat"]
+
     @property
     def time(self):
         return self.observations["time"]
@@ -211,7 +219,7 @@ def dtype(self):
             data_type = VAR_DESCR[elt][
                 "compute_type"
                 if "compute_type" in VAR_DESCR[elt] and not self.raw_data
-                else "nc_type"
+                else "output_type"
             ]
             if elt in self.array_variables:
                 dtype.append((elt, data_type, (self.track_array_variables,)))

src/py_eddy_tracker/observations/tracking.py

@@ -122,6 +122,22 @@ def set_global_attr_netcdf(self, h_nc):
         h_nc.time_coverage_start = d_start.strftime('%Y-%m-%dT00:00:00Z')
         h_nc.time_coverage_end = d_end.strftime('%Y-%m-%dT00:00:00Z')
 
+    def extract_with_area(self, area, **kwargs):
+        """
+        Extract with a bounding box
+        Args:
+            area: 4 coordinates in a dictionary to specify bounding box (lower left corner and upper right corner)
+            **kwargs:
+
+        Returns:
+
+        """
+        mask = (self.latitude > area['llcrnrlat']) * (self.latitude < area['urcrnrlat'])
+        lon0 = area['llcrnrlon']
+        lon = (self.longitude - lon0) % 360 + lon0
+        mask *= (lon > lon0) * (lon < area['urcrnrlon'])
+        return self.__extract_with_mask(mask, **kwargs)
+
     def extract_with_period(self, period, **kwargs):
         """
         Extract with a period

src/scripts/EddySubSetter

@@ -18,6 +18,10 @@ def id_parser():
                         help='Extract path, if one obs or more are selected')
     parser.add_argument('-d', '--remove_incomplete', action='store_true',
                         help='Extract path only if all obs are selected')
+    parser.add_argument('-a', '--area', nargs=4, type=float,
+                        metavar=('llcrnrlon', 'llcrnrlat', 'urcrnrlon', 'urcrnrlat'),
+                        help='Coordinates of bounding to extract'
+                        )
     return parser
 
 
@@ -29,5 +33,14 @@ if __name__ == '__main__':
         dataset = dataset.extract_with_period(args.period, full_path=args.full_path,
                                               remove_incomplete=args.remove_incomplete)
 
+    if args.area is not None:
+        area = dict(llcrnrlon=args.area[0],
+                    llcrnrlat=args.area[1],
+                    urcrnrlon=args.area[2],
+                    urcrnrlat=args.area[3],
+                    )
+        dataset = dataset.extract_with_area(area, full_path=args.full_path,
+                                            remove_incomplete=args.remove_incomplete)
+
     if len(dataset) != 0:
         dataset.write_netcdf(filename=args.filename_out)