plexus-python-common 1.0.63__py3-none-any.whl → 1.0.64__py3-none-any.whl

plexus/common/utils/tagutils.py

@@ -1,6 +1,7 @@
 import contextlib
 import dataclasses
 import datetime
+import math
 import os
 import pathlib
 import textwrap
@@ -16,6 +17,7 @@ import sqlalchemy as sa
 import sqlalchemy.dialects.sqlite as sa_sqlite
 import sqlalchemy.orm as sa_orm
 from iker.common.utils.dbutils import ConnectionMaker
+from iker.common.utils.dtutils import dt_from_ts_us, dt_to_ts_us
 from iker.common.utils.funcutils import memorized, singleton
 from iker.common.utils.iterutils import batched, head_or_none
 from iker.common.utils.iterutils import dicttree
@@ -48,6 +50,8 @@ __all__ = [
     "tag_cache_file_path",
     "TagCache",
     "tag_cache",
+    "standard_clip_duration_us",
+    "populate_clip_ranges",
 ]
 
 
@@ -1093,3 +1097,52 @@ def tag_cache(*, identifier: str | None = None, file_path: str | None = None) ->
     if file_path is not None:
         return TagCache(file_path=file_path)
     return TagCache(file_path=tag_cache_file_path())
+
+
+
+@singleton
+def standard_clip_duration_us() -> int:
+    """Duration of clips to split samples into, in microseconds, which is fixed to 20 seconds for now."""
+    return 20 * 1_000_000  # 20 seconds
+
+
+
+def populate_clip_ranges(
+    data_begin_dt: datetime.datetime,
+    data_end_dt: datetime.datetime,
+    *,
+    clip_duration_us: int = None
+) -> Generator[tuple[datetime.datetime, datetime.datetime, datetime.datetime, datetime.datetime]]:
+    """
+    Generate clip begin datetime, clip data begin datetime, and clip data end datetime for each clip slot that overlaps
+    with the given data begin and end datetimes.
+    The clip begin datetime is the beginning of the clip slot, the clip data begin datetime is the maximum of the clip
+    begin datetime and the data begin datetime, and the clip data end datetime is the minimum of the clip end datetime
+    and the data end datetime.
+
+    :param data_begin_dt: The beginning datetime of the data.
+    :param data_end_dt: The end datetime of the data.
+    :param clip_duration_us: The duration of each clip in microseconds. If not provided,
+                             the standard clip duration will be used.
+    :return: A generator of tuples containing the clip begin datetime, clip data begin datetime,
+             and clip data end datetime
+    """
+    clip_duration_us = clip_duration_us or standard_clip_duration_us()
+
+    if data_begin_dt == data_end_dt:
+        clip_slot = math.floor(dt_to_ts_us(data_begin_dt) / clip_duration_us)
+        clip_begin_dt = dt_from_ts_us(clip_slot * clip_duration_us)
+        clip_end_dt = dt_from_ts_us((clip_slot + 1) * clip_duration_us)
+        yield clip_begin_dt, clip_end_dt, data_begin_dt, data_end_dt
+        return
+
+    begin_clip_slot = math.floor(dt_to_ts_us(data_begin_dt) / clip_duration_us)
+    end_clip_slot = math.ceil(dt_to_ts_us(data_end_dt)  / clip_duration_us)
+
+    for clip_slot in range(begin_clip_slot, end_clip_slot):
+        clip_begin_dt = dt_from_ts_us(clip_slot * clip_duration_us)
+        clip_end_dt = dt_from_ts_us((clip_slot + 1) * clip_duration_us)
+        clip_data_begin_dt = max(clip_begin_dt, data_begin_dt)
+        clip_data_end_dt = min(clip_end_dt, data_end_dt)
+
+        yield clip_begin_dt, clip_end_dt, clip_data_begin_dt, clip_data_end_dt

{plexus_python_common-1.0.63.dist-info → plexus_python_common-1.0.64.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plexus-python-common
-Version: 1.0.63
+Version: 1.0.64
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13

{plexus_python_common-1.0.63.dist-info → plexus_python_common-1.0.64.dist-info}/RECORD

@@ -21,9 +21,9 @@ plexus/common/utils/pathutils.py,sha256=hGJqSLj08tuOeZ7WeC5d4BtjnPI732BuntVQBQsq
 plexus/common/utils/s3utils.py,sha256=zlO4kGs-c2gUeOfPfiKIE5liQZsbYxqAZYCwA8kL0Lo,36017
 plexus/common/utils/sqlutils.py,sha256=D6kTBjhO5YlNRt3uFlPt6z3uH61m9ajEzPYmsI6NoFc,231
 plexus/common/utils/strutils.py,sha256=O9Inv4ffUTf6Xjc5ftoZwbIua1NeG7itCT9S3zjZxBc,16436
-plexus/common/utils/tagutils.py,sha256=Ga22WnUQxVbzFtGR8XyiRiZZnaddOhdHqBNVof40pqs,44514
+plexus/common/utils/tagutils.py,sha256=7qBLW848fr64Jq8ogUUuUY15mT1KnNZbtEbb3tKosXQ,46930
 plexus/common/utils/testutils.py,sha256=GyrKOKfrl1Go8Q7tCZLybxYvVqyox1AtEFWzWoecNwg,6163
-plexus_python_common-1.0.63.dist-info/METADATA,sha256=x1Njz90tSzZxpbdTNZHMD2IPGqmSiZxFFzoSLa9Ae40,1481
-plexus_python_common-1.0.63.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-plexus_python_common-1.0.63.dist-info/top_level.txt,sha256=ug_g7CVwaMQuas5UzAXbHUrQvKGCn8ezc6ZNvvRlJOE,7
-plexus_python_common-1.0.63.dist-info/RECORD,,
+plexus_python_common-1.0.64.dist-info/METADATA,sha256=J8wWo3Q3ffOL79INfOk4lyJAcDolpsf3FXz5lIZ95eY,1481
+plexus_python_common-1.0.64.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+plexus_python_common-1.0.64.dist-info/top_level.txt,sha256=ug_g7CVwaMQuas5UzAXbHUrQvKGCn8ezc6ZNvvRlJOE,7
+plexus_python_common-1.0.64.dist-info/RECORD,,