modifications network file

Author: JulietteGamot

src/py_eddy_tracker/appli/gui.py

@@ -11,7 +11,7 @@
 from matplotlib.animation import FuncAnimation
 from matplotlib.axes import Axes
 from matplotlib.collections import LineCollection
-from numpy import arange, where
+from numpy import arange, where, nan
 
 from .. import EddyParser
 from ..gui import GUI
@@ -58,7 +58,10 @@ def setup(
         self.kw_label["fontweight"] = kwargs.pop("fontweight", "demibold")
         # To text each visible eddy
         if field_txt:
-            self.field_txt = self.eddy[field_txt]
+            if isinstance(field_txt,str):
+                self.field_txt = self.eddy[field_txt]
+            else :
+                self.field_txt=field_txt
         if field_color:
             # To color each visible eddy
             self.field_color = self.eddy[field_color].astype("f4")

src/py_eddy_tracker/observations/network.py

@@ -5,7 +5,8 @@
 from glob import glob
 import logging
 import time
-
+from datetime import timedelta, datetime
+import os
 import netCDF4
 from numba import njit, types as nb_types
 from numba.typed import List
@@ -16,6 +17,7 @@
     bincount,
     bool_,
     concatenate,
+
     empty,
     nan,
     ones,
@@ -120,13 +122,15 @@ def __repr__(self):
         period = (self.period[1] - self.period[0]) / 365.25
         nb_by_network = self.network_size()
         nb_trash = 0 if self.ref_index != 0 else nb_by_network[0]
+        lifetime=self.lifetime
         big = 50_000
         infos = [
             f"Atlas with {self.nb_network} networks ({self.nb_network / period:0.0f} networks/year),"
             f" {self.nb_segment} segments ({self.nb_segment / period:0.0f} segments/year), {len(self)} observations ({len(self) / period:0.0f} observations/year)",
             f"    {m_event.size} merging ({m_event.size / period:0.0f} merging/year), {s_event.size} splitting ({s_event.size / period:0.0f} splitting/year)",
             f"    with {(nb_by_network > big).sum()} network with more than {big} obs and the biggest have {nb_by_network.max()} observations ({nb_by_network[nb_by_network > big].sum()} observations cumulate)",
             f"    {nb_trash} observations in trash",
+            f" {lifetime.max()} days max of lifetime",
         ]
         return "\n".join(infos)
 
@@ -201,6 +205,13 @@ def ref_segment_track_index(self):
     @property
     def ref_index(self):
         return self.index_network[2]
+    
+    @property
+    def lifetime(self):
+        """Return lifetime for each observation"""
+        lt=self.networks_period.astype("int")
+        nb_by_network=self.network_size()
+        return lt.repeat(nb_by_network)
 
     def network_segment_size(self, id_networks=None):
         """Get number of segment by network
@@ -226,12 +237,15 @@ def network_size(self, id_networks=None):
             i = id_networks - self.index_network[2]
             return self.index_network[1][i] - self.index_network[0][i]
 
+    @property
     def networks_period(self):
         """
         Return period for each network
         """
         return get_period_with_index(self.time, *self.index_network[:2])
 
+    
+    
     def unique_segment_to_id(self, id_unique):
         """Return id network and id segment for a unique id
 
@@ -281,7 +295,7 @@ def astype(self, cls):
                 new[k][:] = self[k][:]
         new.sign_type = self.sign_type
         return new
-
+    
     def longer_than(self, nb_day_min=-1, nb_day_max=-1):
         """
         Select network on time duration
@@ -1132,23 +1146,29 @@ def segment_track_array(self):
             self._segment_track_array = build_unique_array(self.segment, self.track)
         return self._segment_track_array
 
-    def birth_event(self):
+    def birth_event(self, only_index=False):
         """Extract birth events."""
         i_start, _, _ = self.index_segment_track
         indices = i_start[self.previous_obs[i_start] == -1]
         if self.first_is_trash():
             indices = indices[1:]
-        return self.extract_event(indices)
-
+        if only_index: 
+            return indices
+        else : 
+            return self.extract_event(indices)
+        
     generation_event = birth_event
 
-    def death_event(self):
+    def death_event(self, only_index=False):
         """Extract death events."""
         _, i_stop, _ = self.index_segment_track
         indices = i_stop[self.next_obs[i_stop - 1] == -1] - 1
         if self.first_is_trash():
             indices = indices[1:]
-        return self.extract_event(indices)
+        if only_index: 
+            return indices
+        else : 
+            return self.extract_event(indices)
 
     dissipation_event = death_event
 
@@ -1459,7 +1479,7 @@ def plot(self, ax, ref=None, color_cycle=None, **kwargs):
             j += 1
         return mappables
 
-    def remove_dead_end(self, nobs=3, ndays=0, recursive=0, mask=None):
+    def remove_dead_end(self, nobs=3, ndays=0, recursive=0, mask=None, return_mask=False):
         """
         Remove short segments that don't connect several segments
 
@@ -1485,6 +1505,8 @@ def remove_dead_end(self, nobs=3, ndays=0, recursive=0, mask=None):
             )
         # get mask for selected obs
         m = ~self.segment_mask(segments_keep)
+        if return_mask:
+            return ~m
         self.track[m] = 0
         self.segment[m] = 0
         self.previous_obs[m] = -1
@@ -1502,6 +1524,8 @@ def remove_dead_end(self, nobs=3, ndays=0, recursive=0, mask=None):
         self.sort()
         if recursive > 0:
             self.remove_dead_end(nobs, ndays, recursive - 1)
+        
+        
 
     def extract_segment(self, segments, absolute=False):
         """Extract given segments
@@ -2042,6 +2066,29 @@ def group_observations(self, min_overlap=0.2, minimal_area=False, **kwargs):
         results, nb_obs = list(), list()
         # To display print only in INFO
         display_iteration = logger.getEffectiveLevel() == logging.INFO
+        
+        
+        # Trier les fichiers par date
+        def extract_date(file):
+            filename = os.path.basename(file)
+            date_str = filename.split('_')[-1].split('.')[0]  # Extraire la partie date (ex : "20180101")
+            return datetime.strptime(date_str, "%Y%m%d")  # Convertir en objet datetime
+        self.filenames = sorted(self.filenames, key=extract_date)
+
+        # Detect missing date and print them to inform the user which files are missing
+        missing_dates = []
+        dates_list = [extract_date(self.filenames[i]) for i in range(len(self.filenames))]
+        for i in range(len(dates_list) - 1):
+            expected_date = dates_list[i] + timedelta(days=1)
+            while expected_date < dates_list[i + 1]:  
+                missing_dates.append(expected_date)
+                expected_date += timedelta(days=1)
+        if missing_dates:
+            missing_str = ', '.join(date.strftime("%Y-%m-%d") for date in missing_dates)
+            raise Exception(f"Following files missing : {missing_str}")
+        else:
+            print("No missing files")
+
         for i, filename in enumerate(self.filenames):
             if display_iteration:
                 print(f"{filename} compared to {self.window} next", end="\r")