yta-video-opengl 0.0.5__py3-none-any.whl → 0.0.6__py3-none-any.whl

yta_video_opengl/utils.py

@@ -0,0 +1,24 @@
+from yta_validation import PythonValidator
+from typing import Union
+
+import numpy as np
+import moderngl
+
+
+def frame_to_texture(
+    frame: Union['VideoFrame', 'np.ndarray'],
+    context: moderngl.Context,
+    numpy_format: str = 'rgb24'
+):
+    """
+    Transform the given 'frame' to an opengl
+    texture.
+    """
+    # To numpy RGB inverted for opengl
+    frame: np.ndarray = (
+        np.flipud(frame.to_ndarray(format = numpy_format))
+        if PythonValidator.is_instance_of(frame, 'VideoFrame') else
+        np.flipud(frame)
+    )
+
+    return context.texture((frame.shape[1], frame.shape[0]), 3, frame.tobytes())

yta_video_opengl/writer.py

@@ -70,7 +70,22 @@ class VideoWriter:
 
         return self
 
-    # TODO: Maybe 'add_video_stream_from_template' (?)
+    def set_video_stream_from_template(
+        self,
+        template: Stream
+    ) -> 'VideoWriter':
+        """
+        Set the video stream, that will overwrite any other
+        previous video stream set.
+
+        You can pass the video stream as it was
+        obtained from the reader.
+        """
+        self.video_stream: VideoStream = self.output.add_stream_from_template(
+            template
+        )
+
+        return self
 
     def set_audio_stream(
         self,
@@ -156,6 +171,30 @@ class VideoWriter:
         ParameterValidator.validate_mandatory_instance_of('packet', packet, Packet)
 
         if packet.size > 0:
+            try:
+                self.output.mux(packet)
+            except Exception as e:
+                # TODO: What strategy should we adopt with
+                # the packets that cannot be handled
+                # properly (?)
+                print(packet)
+                pass
+
+        return self
+    
+        # TODO: This below is to test fails
+        if (
+            packet.size > 0 and
+            # This is a new special case
+            packet.dts % 1024 == 0 and
+            packet.dts > 0
+        ):
+            # av.Packet of #0, dts=-2, pts=0; 1225 bytes at 0x1ef2d8e0680>
+            # av.Packet of #0, dts=0, pts=2; 110 bytes at 0x1e1feb8c6d0>
+            # av.Packet of #0, dts=1, pts=1; 182 bytes at 0x153f6b3b400>
+            # are failing
+            print(packet)
+            print(packet.stream.type)
             self.output.mux(packet)
 
         return self

{yta_video_opengl-0.0.5.dist-info → yta_video_opengl-0.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: yta-video-opengl
-Version: 0.0.5
+Version: 0.0.6
 Summary: Youtube Autonomous Video OpenGL Module
 Author: danialcala94
 Author-email: danielalcalavalera@gmail.com

yta_video_opengl-0.0.6.dist-info/RECORD

@@ -0,0 +1,11 @@
+yta_video_opengl/__init__.py,sha256=ycAx_XYMVDfkuObSvtW6irQ0Wo-fgxEz3fjIRMe8PpY,205
+yta_video_opengl/classes.py,sha256=fYPdqMRtyC_Latn1eAXIMWRCsaMZhq7uHlnJPvZe8K0,32584
+yta_video_opengl/reader/__init__.py,sha256=aKB5Ik9oR9L05-Lpwnl0YI3lK2RtmmhXIY7jGg9Hc1I,11320
+yta_video_opengl/reader/cache.py,sha256=Y3lQrirQJz7zFeiJQeJnkzyghYeMahkpKzsouzB90VI,4421
+yta_video_opengl/tests.py,sha256=00LEGdwjy1QfuHmWhIFJMNJixhc7gcAqb9ESAXH7Oag,24956
+yta_video_opengl/utils.py,sha256=T1Rj36yk1wnSfFhisD4Gn8ZOaJXwnNVUeenC_dLcrP4,641
+yta_video_opengl/writer.py,sha256=G4Rk6gcuPsF6H7PjXSRc3wGOY9pdiJdNCO9GdM6v4mE,7606
+yta_video_opengl-0.0.6.dist-info/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
+yta_video_opengl-0.0.6.dist-info/METADATA,sha256=OvvH6wKPtmPIZdOoMNeOCxEM7xWieoR10NkmdtBZxTU,670
+yta_video_opengl-0.0.6.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+yta_video_opengl-0.0.6.dist-info/RECORD,,

yta_video_opengl-0.0.5.dist-info/RECORD

@@ -1,8 +0,0 @@
-yta_video_opengl/__init__.py,sha256=ycAx_XYMVDfkuObSvtW6irQ0Wo-fgxEz3fjIRMe8PpY,205
-yta_video_opengl/reader.py,sha256=ROjx5jSi_M2KWcp1Fksi4y-apb1IvItPhpLF32u5DpU,9869
-yta_video_opengl/tests.py,sha256=7a_n-xW2xBVYR1sj-IuGm8c8F6It2LHZ8GqCH_SeY-U,6081
-yta_video_opengl/writer.py,sha256=9i1SSaOxWF-uBTWvpToP8juIqSWhbIh_Nfyi69adWIw,6356
-yta_video_opengl-0.0.5.dist-info/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
-yta_video_opengl-0.0.5.dist-info/METADATA,sha256=hH7uBbRBkNQ44kIEi1oKB_s3fGLtwcnd0rdZB_M3bVw,670
-yta_video_opengl-0.0.5.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-yta_video_opengl-0.0.5.dist-info/RECORD,,