Use cpdef for encode/decode

PyAV-Org · Mar 9, 2024 · 57f1b92 · 57f1b92
1 parent 23ce2a3
commit 57f1b92
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 7 deletions.
diff --git a/av/audio/stream.pxd b/av/audio/stream.pxd
@@ -1,5 +1,8 @@
 from av.stream cimport Stream
+from av.packet cimport Packet
 
+from .frame cimport AudioFrame
 
 cdef class AudioStream(Stream):
-    pass
+    cpdef encode(self, AudioFrame frame=?)
+    cpdef decode(self, Packet packet=?)
diff --git a/av/audio/stream.pyx b/av/audio/stream.pyx
@@ -1,5 +1,7 @@
 from av.packet cimport Packet
 
+from .frame cimport AudioFrame
+
 
 cdef class AudioStream(Stream):
     def __repr__(self):
@@ -9,7 +11,7 @@ cdef class AudioStream(Stream):
             f" {self.layout.name}, {form} at 0x{id(self):x}>"
         )
 
-    def encode(self, frame=None):
+    cpdef encode(self, AudioFrame frame=None):
         """
         Encode an :class:`.AudioFrame` and return a list of :class:`.Packet`.
 
@@ -26,7 +28,7 @@ cdef class AudioStream(Stream):
 
         return packets
 
-    def decode(self, packet=None):
+    cpdef decode(self, Packet packet=None):
         """
         Decode a :class:`.Packet` and return a list of :class:`.AudioFrame`.
         :return: :class:`list` of :class:`.AudioFrame`

diff --git a/av/video/stream.pxd b/av/video/stream.pxd
@@ -1,5 +1,8 @@
 from av.stream cimport Stream
+from .frame cimport VideoFrame
+from av.packet cimport Packet
 
 
 cdef class VideoStream(Stream):
-    pass
+    cpdef encode(self, VideoFrame frame=?)
+    cpdef decode(self, Packet packet=?)
diff --git a/av/video/stream.pyx b/av/video/stream.pyx
@@ -1,5 +1,5 @@
 from av.packet cimport Packet
-
+from .frame cimport VideoFrame
 
 cdef class VideoStream(Stream):
     def __repr__(self):
@@ -9,7 +9,7 @@ cdef class VideoStream(Stream):
             f"{self.codec_context.height} at 0x{id(self):x}>"
         )
 
-    def encode(self, frame=None):
+    cpdef encode(self, VideoFrame frame=None):
         """
         Encode an :class:`.VideoFrame` and return a list of :class:`.Packet`.
 
@@ -27,7 +27,7 @@ cdef class VideoStream(Stream):
         return packets
 
 
-    def decode(self, packet=None):
+    cpdef decode(self, Packet packet=None):
         """
         Decode a :class:`.Packet` and return a list of :class:`.VideoFrame`.
         :return: :class:`list` of :class:`.Frame` subclasses.