demuxer, decoder not working

mp4/libav.py

@@ -3,9 +3,28 @@
 
 import ctypes
 
+_avutil = ctypes.cdll.LoadLibrary('libavutil.so')
 _avformat = ctypes.cdll.LoadLibrary('libavformat.so')
 _avcodec = ctypes.cdll.LoadLibrary('libavcodec.so')
 
+class AVFrame(ctypes.Structure):
+    pass
+
+AVFrame_p = ctypes.POINTER(AVFrame)
+AVFrame_pp = ctypes.POINTER(AVFrame_p)
+
+_avutil.av_frame_alloc.restype = AVFrame_p
+_avutil.av_frame_alloc.argtypes = None
+
+def frame_alloc():
+    return _avutil.av_frame_alloc()
+
+_avutil.av_frame_free.restype = None
+_avutil.av_frame_free.argtypes = [AVFrame_pp]
+
+def frame_free(frame):
+    _avutil.av_frame_free(ctypes.byref(frame))
+
 class AVFormatContext(ctypes.Structure):
     pass
 
@@ -15,37 +34,37 @@ AVFormatContext_pp = ctypes.POINTER(AVFormatContext_p)
 _avformat.avformat_alloc_context.restype = AVFormatContext_p
 _avformat.avformat_alloc_context.argtypes = None
 
-def alloc_context():
+def format_alloc_context():
     return _avformat.avformat_alloc_context()
 
 _avformat.avformat_free_context.restype = None
-_avformat.avformat_free_context.argtypes = [AVFormatContext_p] # context
+_avformat.avformat_free_context.argtypes = [AVFormatContext_p]
 
-def free_context(context):
+def format_free_context(context):
     _avformat.avformat_free_context(context)
 
 _avformat.avformat_open_input.restype = ctypes.c_int
 _avformat.avformat_open_input.argtypes = [
-    AVFormatContext_pp, # context
+    AVFormatContext_pp,
     ctypes.c_char_p, # url
     ctypes.c_void_p, # format
     ctypes.POINTER(ctypes.c_void_p)] # options
 
-def open_input(context, url):
+def format_open_input(context, url):
     return _avformat.avformat_open_input(ctypes.byref(context), url.encode('ascii', 'ignore'), None, None)
 
 _avformat.avformat_close_input.restype = None
-_avformat.avformat_close_input.argtypes = [AVFormatContext_pp] # context
+_avformat.avformat_close_input.argtypes = [AVFormatContext_pp]
 
-def close_input(context):
+def format_close_input(context):
     _avformat.avformat_close_input(ctypes.byref(context))
 
 _avformat.avformat_find_stream_info.restype = ctypes.c_int
 _avformat.avformat_find_stream_info.argtypes = [
-    AVFormatContext_p, # context
+    AVFormatContext_p,
     ctypes.POINTER(ctypes.c_void_p)] # options
 
-def find_stream_info(context):
+def format_find_stream_info(context):
     return _avformat.avformat_find_stream_info(context, None)
 
 AVMEDIA_TYPE_UNKNOWN = -1
@@ -66,14 +85,14 @@ AVCodec_pp = ctypes.POINTER(AVCodec_p)
 
 _avformat.av_find_best_stream.restype = ctypes.c_int
 _avformat.av_find_best_stream.argtypes = [
-    AVFormatContext_p, # context
+    AVFormatContext_p,
     ctypes.c_int, # type
     ctypes.c_int, # wanted stream
     ctypes.c_int, # related stream
-    AVCodec_pp, # decoder
+    AVCodec_pp,
     ctypes.c_int] # flags
 
-def find_best_stream(context, type):
+def format_find_best_stream(context, type):
     codec = AVCodec_p()
     index = _avformat.av_find_best_stream(context, type, -1, -1, ctypes.byref(codec), 0)
     return index, codec
@@ -98,15 +117,52 @@ def packet_alloc():
     return _avformat.av_packet_alloc()
 
 _avformat.av_packet_free.restype = None
-_avformat.av_packet_free.argtypes = [AVPacket_pp] # packet
+_avformat.av_packet_free.argtypes = [AVPacket_pp]
 
 def packet_free(packet):
     _avformat.av_packet_free(ctypes.byref(packet))
 
 _avformat.av_read_frame.restype = ctypes.c_int
-_avformat.av_read_frame.argtypes = [
-    AVFormatContext_p, # context
-    AVPacket_p] # packet
+_avformat.av_read_frame.argtypes = [AVFormatContext_p, AVPacket_p]
 
 def read_frame(context, packet):
     return _avformat.av_read_frame(context, packet)
+
+class AVCodecContext(ctypes.Structure):
+    pass
+
+AVCodecContext_p = ctypes.POINTER(AVCodecContext)
+AVCodecContext_pp = ctypes.POINTER(AVCodecContext_p)
+
+_avcodec.avcodec_alloc_context3.restype = AVCodecContext_p
+_avcodec.avcodec_alloc_context3.argtypes = [AVCodec_p]
+
+def codec_alloc_context(codec):
+    return _avcodec.avcodec_alloc_context3(codec)
+
+_avcodec.avcodec_free_context.restype = None
+_avcodec.avcodec_free_context.argtypes = [AVCodecContext_pp]
+
+def codec_free_context(context):
+    _avcodec.avcodec_free_context(ctypes.byref(context))
+
+_avcodec.avcodec_open2.restype = ctypes.c_int
+_avcodec.avcodec_open2.argtypes = [
+    AVCodecContext_p,
+    AVCodec_p,
+    ctypes.POINTER(ctypes.c_void_p)] # options
+
+def codec_open(context, codec):
+    return _avcodec.avcodec_open2(context, codec, None)
+
+_avcodec.avcodec_send_packet.restype = ctypes.c_int
+_avcodec.avcodec_send_packet.argtypes = [AVCodecContext_p, AVPacket_p]
+
+def codec_send_packet(context, packet):
+    return _avcodec.avcodec_send_packet(context, packet)
+
+_avcodec.avcodec_receive_frame.restype = ctypes.c_int
+_avcodec.avcodec_receive_frame.argtypes = [AVCodecContext_p, AVFrame_p]
+
+def codec_receive_frame(context, frame):
+    return _avcodec.avcodec_receive_frame(context, frame)