Subtitles: make text bytes, type hints

av/packet.pyi

@@ -1,6 +1,5 @@
 from collections.abc import Buffer
 from fractions import Fraction
-from typing import Iterator
 
 from av.subtitles.subtitle import SubtitleSet
 
@@ -22,5 +21,5 @@ class Packet(Buffer):
     is_disposable: bool
 
     def __init__(self, input: int | bytes | None = None) -> None: ...
-    def decode(self) -> Iterator[SubtitleSet]: ...
+    def decode(self) -> list[SubtitleSet]: ...
     def __buffer__(self, arg1) -> memoryview: ...

av/subtitles/codeccontext.pyx

@@ -8,13 +8,14 @@ from av.subtitles.subtitle cimport SubtitleProxy, SubtitleSet
 cdef class SubtitleCodecContext(CodecContext):
     cdef _send_packet_and_recv(self, Packet packet):
         cdef SubtitleProxy proxy = SubtitleProxy()
-
         cdef int got_frame = 0
-        err_check(lib.avcodec_decode_subtitle2(
-            self.ptr,
-            &proxy.struct,
-            &got_frame,
-            packet.ptr if packet else NULL))
+
+        err_check(
+            lib.avcodec_decode_subtitle2(
+                self.ptr, &proxy.struct, &got_frame, packet.ptr if packet else NULL
+            )
+        )
+
         if got_frame:
             return [SubtitleSet(proxy)]
         else:

av/subtitles/subtitle.pxd

@@ -4,19 +4,16 @@ from av.packet cimport Packet
 
 
 cdef class SubtitleProxy:
-
     cdef lib.AVSubtitle struct
 
 
 cdef class SubtitleSet:
-
     cdef readonly Packet packet
     cdef SubtitleProxy proxy
     cdef readonly tuple rects
 
 
 cdef class Subtitle:
-
     cdef SubtitleProxy proxy
     cdef lib.AVSubtitleRect *ptr
     cdef readonly bytes type
@@ -28,11 +25,9 @@ cdef class ASSSubtitle(Subtitle):
     pass
 
 cdef class BitmapSubtitle(Subtitle):
-
     cdef readonly planes
 
 cdef class BitmapSubtitlePlane:
-
     cdef readonly BitmapSubtitle subtitle
     cdef readonly int index
     cdef readonly long buffer_size

av/subtitles/subtitle.pyi

@@ -5,6 +5,7 @@ class SubtitleSet:
     start_display_time: int
     end_display_time: int
     pts: int
+    rects: tuple[Subtitle]
 
     def __len__(self) -> int: ...
     def __iter__(self) -> Iterator[Subtitle]: ...
@@ -29,7 +30,7 @@ class BitmapSubtitlePlane:
 
 class TextSubtitle(Subtitle):
     type: Literal[b"text"]
-    text: str
+    text: bytes
 
 class AssSubtitle(Subtitle):
     type: Literal[b"ass"]

av/subtitles/subtitle.pyx

@@ -145,7 +145,9 @@ cdef class TextSubtitle(Subtitle):
 
     @property
     def text(self):
-        return self.ptr.text
+        if self.ptr.text is not NULL:
+            return PyBytes_FromString(self.ptr.text)
+        return b""
 
 
 cdef class AssSubtitle(Subtitle):