yta-video-frame-time 0.0.2__tar.gz → 0.0.4__tar.gz

{yta_video_frame_time-0.0.2 → yta_video_frame_time-0.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: yta-video-frame-time
-Version: 0.0.2
+Version: 0.0.4
 Summary: Youtube Autonomous Video Frame Time Module
 Author: danialcala94
 Author-email: danielalcalavalera@gmail.com

{yta_video_frame_time-0.0.2 → yta_video_frame_time-0.0.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "yta-video-frame-time"
-version = "0.0.2"
+version = "0.0.4"
 description = "Youtube Autonomous Video Frame Time Module"
 authors = [
     {name = "danialcala94",email = "danielalcalavalera@gmail.com"}

{yta_video_frame_time-0.0.2 → yta_video_frame_time-0.0.4}/src/yta_video_frame_time/__init__.py

@@ -2,6 +2,7 @@
 Module to handle frame indexes and times.
 """
 from yta_validation.parameter import ParameterValidator
+from typing import Union
 
 
 SMALL_AMOUNT_TO_FIX = 0.0000000001
@@ -36,11 +37,13 @@ class T:
         ParameterValidator.validate_mandatory_positive_number('t', t, do_include_zero = True)
         ParameterValidator.validate_mandatory_positive_number('fps', fps, do_include_zero = False)
 
-        return T.frame_time_to_frame_index(t, fps) / fps + SMALL_AMOUNT_TO_FIX
-
+        return T.video_frame_time_to_video_frame_index(t, fps) / fps + SMALL_AMOUNT_TO_FIX
+    
     def get_frame_indexes(
         duration: float,
         fps: float,
+        start: Union[float, None] = None,
+        end: Union[float, None] = None,
         do_invert_order: bool = False
     ):
         """
@@ -59,6 +62,28 @@ class T:
         # If you remove 'list' you get a generator instead
         frame_indexes = list(range(get_number_of_frames(duration, fps)))
 
+        # TODO: Maybe move this to a helper (?)
+        # Limit to ['start', 'end'] range
+        if (
+            start is not None or
+            end is not None
+        ):
+            start = (
+                T.video_frame_time_to_video_frame_index(start, fps)
+                if start is not None else
+                0.0
+            )
+            end = (
+                T.video_frame_time_to_video_frame_index(end, fps)
+                if end is not None else
+                9999999999
+            )
+            frame_indexes = [
+                frame_index
+                for frame_index in frame_indexes
+                if start <= frame_index < end
+            ]
+
         return (
             frame_indexes[::-1]
             if do_invert_order else
@@ -286,6 +311,8 @@ class T:
     def get_frame_time_moments(
         duration: float,
         fps: float,
+        start: Union[float, None] = None,
+        end: Union[float, None] = None,
         do_invert_order: bool = False
     ):
         """
@@ -313,6 +340,28 @@ class T:
             for i in range(get_number_of_frames(duration, fps) + 1)
         ][:-1]
 
+        # TODO: Maybe move this to a helper (?)
+        # Limit to ['start', 'end'] range
+        if (
+            start is not None or
+            end is not None
+        ):
+            start = (
+                start
+                if start is not None else
+                0.0
+            )
+            end = (
+                end
+                if end is not None else
+                9999999999
+            )
+            frame_time_moments = [
+                frame_time_moment
+                for frame_time_moment in frame_time_moments
+                if start <= frame_time_moment < end
+            ]
+
         return (
             frame_time_moments[::-1]
             if do_invert_order else
@@ -409,19 +458,20 @@ class T:
         ParameterValidator.validate_mandatory_positive_number('audio_fps', audio_fps, do_include_zero = False)
         ParameterValidator.validate_mandatory_bool('do_invert_order', do_invert_order)
 
-        from yta_general_utils.math.progression import Progression
-
         audio_frames_per_video_frame = int(audio_fps / video_fps)
         audio_frame_duration = 1 / audio_fps
         video_frame_duration = 1 / video_fps
 
         t = T.get_frame_time_base(video_t, video_fps)
 
-        audio_time_moments = Progression(
-            start = t, 
+        # This is replacing a 'Progression' from
+        # the 'yta_general_utils' library because
+        # it has many imports we don't want here...
+        audio_time_moments = _linspace(
+            start = t,
             end = t + video_frame_duration - audio_frame_duration,
             n = audio_frames_per_video_frame
-        ).values
+        )
 
         return (
             audio_time_moments[::-1]
@@ -540,4 +590,29 @@ def get_number_of_frames(
     The formula is:
     - `int(duration * fps + SMALL_AMOUNT)`
     """
-    return int(duration * fps + SMALL_AMOUNT_TO_FIX)
+    return int(duration * fps + SMALL_AMOUNT_TO_FIX)
+
+def _linspace(
+    start: float,
+    end: float,
+    n: int,
+):
+    """
+    *For internal use only*
+
+    A simplified version of the 'Progression'
+    class we have in 'yta_general_utils', to
+    avoid importing that library that has other
+    dependencies I don't need here.
+
+    This method should be removed when the
+    Progression class is refactored and all 
+    those dependencies are removed.
+    """
+    if n < 2:
+        raise Exception('The "n" parameter must be greater or equal than 2.')
+
+    return [
+        start + i * ((end - start) / (n - 1))
+        for i in range(n)
+    ]