Notebook test

src/post_processing/dataclass/data_aplose.py

@@ -431,14 +431,15 @@ def plot(
             season = kwargs.get("season", False)
             effort = kwargs.get("effort")
 
-            return scatter(df=df_filtered,
-                           ax=ax,
-                           time_range=time,
-                           show_rise_set=show_rise_set,
-                           season=season,
-                           coordinates=self.coordinates,
-                           effort=effort,
-                           )
+            return scatter(
+                df=df_filtered,
+                ax=ax,
+                time_range=time,
+                show_rise_set=show_rise_set,
+                season=season,
+                coordinates=self.coordinates,
+                effort=effort,
+            )
 
         if mode == "agreement":
             bin_size = kwargs.get("bin_size")
@@ -452,7 +453,11 @@ def plot(
                 label,
             )
 
-            return timeline(df=df_filtered, ax=ax, color=color)
+            return timeline(
+                df=df_filtered,
+                ax=ax,
+                color=color,
+            )
 
         msg = f"Unsupported plot mode: {mode}"
         raise ValueError(msg)

src/post_processing/utils/core_utils.py

@@ -8,7 +8,8 @@
 import astral
 import easygui
 import numpy as np
-from astral.sun import sunrise, sunset
+from astral import LocationInfo
+from astral.sun import sunrise, sunset, sun
 from matplotlib import pyplot as plt
 from osekit.config import TIMESTAMP_FORMAT_AUDIO_FILE
 from osekit.utils.timestamp_utils import strftime_osmose_format, strptime_from_text
@@ -579,3 +580,44 @@ def timedelta_to_str(td: Timedelta) -> str:
     if seconds % 60 == 0:
         return f"{seconds // 60}min"
     return f"{seconds}s"
+
+
+def assign_light_regime(
+    ts: Timestamp,
+    lat: float | None = None,
+    lon: float | None = None,
+) -> DataFrame:
+    """Assign daylight regime to temporal events.
+
+    Parameters
+    ----------
+    ts: Timestamp
+        Timestamp to assign a light regime to.
+    lat: float
+        The latitude of corresponding point.
+    lon: float
+        The longitude of corresponding point.
+
+    Returns
+    -------
+    DataFrame
+        The same dataframe with the column daytime.
+
+    """
+    if not all([lat, lon]):
+        lat, lon = get_coordinates()
+
+    # Get sun times for given location
+    location = LocationInfo(latitude=lat, longitude=lon)
+    s = sun(location.observer, date=ts.date())
+
+    if ts < s['dawn']:
+        return 'night'
+    elif ts < s['sunrise']:
+        return 'dawn'
+    elif ts < s['sunset']:
+        return 'day'
+    elif ts < s['dusk']:
+        return 'dusk'
+    else:
+        return 'night'

src/post_processing/utils/filtering_utils.py

@@ -486,6 +486,7 @@ def _process_annotator_label_pair(
     ]
     file_vector = [
         filename_vector[i] for i, detected in enumerate(detect_vec) if detected
+        # filename_vector[i + 1] for i, detected in enumerate(detect_vec) if detected
     ]
 
     if not start_datetime:
@@ -510,8 +511,8 @@ def reshape_timebin(
     timebin_new: Timedelta
         The size of the new time bin.
     timestamp_audio: list[Timestamp]
-        A list of Timestamp objects corresponding to the start of each wav
-         that corresponds to a detection
+        A list of Timestamp objects corresponding to the shape
+        in which the data should be reshaped.
 
     Returns
     -------
@@ -572,10 +573,10 @@ def get_filename_timestamps(df: DataFrame, date_parser: str) -> list[Timestamp]:
     """
     tz = get_timezone(df)
     timestamps = [
-    strptime_from_text(
-        ts,
-        datetime_template=date_parser,
-    ) for ts in df["filename"]
+        strptime_from_text(
+            ts,
+            datetime_template=date_parser,
+        ) for ts in df["filename"]
     ]
 
     if all(t.tz is None for t in timestamps):