yta-video-frame-time 0.0.1__tar.gz → 0.0.3__tar.gz

{yta_video_frame_time-0.0.1 → yta_video_frame_time-0.0.3}/PKG-INFO

@@ -1,12 +1,13 @@
 Metadata-Version: 2.3
 Name: yta-video-frame-time
-Version: 0.0.1
+Version: 0.0.3
 Summary: Youtube Autonomous Video Frame Time Module
 Author: danialcala94
 Author-email: danielalcalavalera@gmail.com
 Requires-Python: ==3.9
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.9
+Requires-Dist: yta_general_utils (>=0.0.1,<1.0.0)
 Requires-Dist: yta_validation (>=0.0.1,<1.0.0)
 Description-Content-Type: text/markdown
 

{yta_video_frame_time-0.0.1 → yta_video_frame_time-0.0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "yta-video-frame-time"
-version = "0.0.1"
+version = "0.0.3"
 description = "Youtube Autonomous Video Frame Time Module"
 authors = [
     {name = "danialcala94",email = "danielalcalavalera@gmail.com"}
@@ -9,6 +9,7 @@ readme = "README.md"
 requires-python = "==3.9"
 dependencies = [
     "yta_validation (>=0.0.1,<1.0.0)",
+    "yta_general_utils (>=0.0.1,<1.0.0)"
 ]
 
 [tool.poetry]

{yta_video_frame_time-0.0.1 → yta_video_frame_time-0.0.3}/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,6 +458,7 @@ 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)
 
+        # TODO: This import is bringin many libraries...
         from yta_general_utils.math.progression import Progression
 
         audio_frames_per_video_frame = int(audio_fps / video_fps)