.

pyvidplayer2/video_pyglet.py

@@ -0,0 +1,36 @@
+import cv2 
+import pyglet 
+import numpy
+from .video import Video
+from typing import Tuple
+from .post_processing import PostProcessing
+
+
+class VideoPyglet(Video):
+    def __init__(self, path: str, chunk_size=300, max_threads=1, max_chunks=1, post_process=PostProcessing.none, interp=cv2.INTER_LINEAR, use_pygame_audio=False) -> None:
+        Video.__init__(self, path, chunk_size, max_threads, max_chunks, None, post_process, interp, use_pygame_audio)
+
+    def __str__(self) -> str:
+        return f"<VideoPyglet(path={self.path})>"
+
+    def _create_frame(self, data: numpy.ndarray) -> pyglet.image.ImageData:
+        return pyglet.image.ImageData(*self.current_size, "BGR", cv2.flip(data, 0).tobytes())
+    
+    def _render_frame(self, pos: Tuple[int, int]) -> None:
+        self.frame_surf.blit(*pos)
+
+    def draw(self, pos: Tuple[int, int], force_draw=True) -> bool:
+        if (self._update() or force_draw) and self.frame_surf is not None:
+            self._render_frame(pos) # (0, 0) pos draws the video bottomleft
+            return True
+        return False
+    
+    def preview(self) -> None:
+        def update(dt):
+            self.draw((0, 0), force_draw=False)
+            if not self.active:
+                win.close()
+        win = pyglet.window.Window(width=self.current_size[0], height=self.current_size[1], config=pyglet.gl.Config(double_buffer=False), caption=f"pyglet - {self.name}")
+        pyglet.clock.schedule_interval(update, 1/60.0)
+        pyglet.app.run()
+        self.close()