"""Wrapper for include/libavutil/avutil.h """ from ctypes import c_char_p, c_void_p, POINTER, Structure from ctypes import c_int, c_int64, c_uint64 from ctypes import c_uint8, c_int8, c_uint, c_size_t import pyglet.lib from pyglet.util import debug_print from . import compat _debug = debug_print('debug_media') avutil = pyglet.lib.load_library( 'avutil', win32=('avutil-57', 'avutil-56'), darwin=('avutil.57', 'avutil-56') ) avutil.avutil_version.restype = c_int compat.set_version('avutil', avutil.avutil_version() >> 16) AVMEDIA_TYPE_UNKNOWN = -1 AVMEDIA_TYPE_VIDEO = 0 AVMEDIA_TYPE_AUDIO = 1 AVMEDIA_TYPE_DATA = 2 AVMEDIA_TYPE_SUBTITLE = 3 AVMEDIA_TYPE_ATTACHMENT = 4 AVMEDIA_TYPE_NB = 5 AV_SAMPLE_FMT_U8 = 0 AV_SAMPLE_FMT_S16 = 1 AV_SAMPLE_FMT_S32 = 2 AV_SAMPLE_FMT_FLT = 3 AV_SAMPLE_FORMAT_DOUBLE = 4 AV_SAMPLE_FMT_U8P = 5 AV_SAMPLE_FMT_S16P = 6 AV_SAMPLE_FMT_S32P = 7 AV_SAMPLE_FMT_FLTP = 8 AV_SAMPLE_FMT_DBLP = 9 AV_SAMPLE_FMT_S64 = 10 AV_SAMPLE_FMT_S64P = 11 AV_NUM_DATA_POINTERS = 8 AV_PIX_FMT_RGB24 = 2 AV_PIX_FMT_ARGB = 25 AV_PIX_FMT_RGBA = 26 AVChannelOrder = c_int class AVChannelLayout(Structure): _fields_ = [ ('order', c_int), ('nb_channels', c_int), # .. more ] class AVBuffer(Structure): _fields_ = [ ('data', POINTER(c_uint8)), ('size', c_int), # .. more ] class AVBufferRef(Structure): _fields_ = [ ('buffer', POINTER(AVBuffer)), ('data', POINTER(c_uint8)), ('size', c_int) ] class AVDictionaryEntry(Structure): _fields_ = [ ('key', c_char_p), ('value', c_char_p) ] class AVDictionary(Structure): _fields_ = [ ('count', c_int), ('elems', POINTER(AVDictionaryEntry)) ] class AVClass(Structure): pass class AVRational(Structure): _fields_ = [ ('num', c_int), ('den', c_int) ] def __repr__(self): return f"AVRational({self.num}/{self.den})" class AVFrameSideData(Structure): pass class AVFrame(Structure): pass AVFrame_Fields = [ ('data', POINTER(c_uint8) * AV_NUM_DATA_POINTERS), ('linesize', c_int * AV_NUM_DATA_POINTERS), ('extended_data', POINTER(POINTER(c_uint8))), ('width', c_int), ('height', c_int), ('nb_samples', c_int), ('format', c_int), ('key_frame', c_int), ('pict_type', c_int), ('sample_aspect_ratio', AVRational), ('pts', c_int64), ('pkt_pts', c_int64), # Deprecated. Removed in 57. ('pkt_dts', c_int64), ('time_base', AVRational), # (5.x) ('coded_picture_number', c_int), ('display_picture_number', c_int), ('quality', c_int), ('opaque', c_void_p), ('error', c_uint64 * AV_NUM_DATA_POINTERS), # Deprecated. Removed in 57. ('repeat_pict', c_int), ('interlaced_frame', c_int), ('top_field_first', c_int), ('palette_has_changed', c_int), ('reordered_opaque', c_int64), ('sample_rate', c_int), ('channel_layout', c_uint64), ('buf', POINTER(AVBufferRef) * AV_NUM_DATA_POINTERS), ('extended_buf', POINTER(POINTER(AVBufferRef))), ('nb_extended_buf', c_int), ('side_data', POINTER(POINTER(AVFrameSideData))), ('nb_side_data', c_int), ('flags', c_int), ('color_range', c_int), ('color_primaries', c_int), ('color_trc', c_int), ('colorspace', c_int), ('chroma_location', c_int), ('best_effort_timestamp', c_int64), ('pkt_pos', c_int64), ('pkt_duration', c_int64), # ! ('metadata', POINTER(AVDictionary)), ('decode_error_flags', c_int), ('channels', c_int), ('pkt_size', c_int), ('qscale_table', POINTER(c_int8)), # Deprecated. Removed in 57. ('qstride', c_int), # Deprecated. Removed in 57. ('qscale_type', c_int), # Deprecated. Removed in 57. ('qp_table_buf', POINTER(AVBufferRef)), # Deprecated. Removed in 57. ('hw_frames_ctx', POINTER(AVBufferRef)), ('opaque_ref', POINTER(AVBufferRef)), ('crop_top', c_size_t), # video frames only ('crop_bottom', c_size_t), # video frames only ('crop_left', c_size_t), # video frames only ('crop_right', c_size_t), # video frames only ('private_ref', POINTER(AVBufferRef)), ] compat.add_version_changes('avutil', 56, AVFrame, AVFrame_Fields, removals=('time_base',)) compat.add_version_changes('avutil', 57, AVFrame, AVFrame_Fields, removals=('pkt_pts', 'error', 'qscale_table', 'qstride', 'qscale_type', 'qp_table_buf')) AV_NOPTS_VALUE = -0x8000000000000000 AV_TIME_BASE = 1000000 AV_TIME_BASE_Q = AVRational(1, AV_TIME_BASE) avutil.av_version_info.restype = c_char_p avutil.av_dict_get.restype = POINTER(AVDictionaryEntry) avutil.av_dict_get.argtypes = [POINTER(AVDictionary), c_char_p, POINTER(AVDictionaryEntry), c_int] avutil.av_rescale_q.restype = c_int64 avutil.av_rescale_q.argtypes = [c_int64, AVRational, AVRational] avutil.av_samples_get_buffer_size.restype = c_int avutil.av_samples_get_buffer_size.argtypes = [POINTER(c_int), c_int, c_int, c_int] avutil.av_frame_alloc.restype = POINTER(AVFrame) avutil.av_frame_free.argtypes = [POINTER(POINTER(AVFrame))] avutil.av_get_default_channel_layout.restype = c_int64 avutil.av_get_default_channel_layout.argtypes = [c_int] avutil.av_get_bytes_per_sample.restype = c_int avutil.av_get_bytes_per_sample.argtypes = [c_int] avutil.av_strerror.restype = c_int avutil.av_strerror.argtypes = [c_int, c_char_p, c_size_t] avutil.av_image_fill_arrays.restype = c_int avutil.av_image_fill_arrays.argtypes = [POINTER(c_uint8) * 4, c_int * 4, POINTER(c_uint8), c_int, c_int, c_int, c_int] avutil.av_dict_set.restype = c_int avutil.av_dict_set.argtypes = [POINTER(POINTER(AVDictionary)), c_char_p, c_char_p, c_int] avutil.av_dict_free.argtypes = [POINTER(POINTER(AVDictionary))] avutil.av_log_set_level.restype = c_int avutil.av_log_set_level.argtypes = [c_uint] avutil.av_malloc.restype = c_void_p avutil.av_malloc.argtypes = [c_int] avutil.av_freep.restype = c_void_p avutil.av_freep.argtypes = [c_void_p] __all__ = [ 'avutil', 'AVMEDIA_TYPE_UNKNOWN', 'AVMEDIA_TYPE_VIDEO', 'AVMEDIA_TYPE_AUDIO', 'AVMEDIA_TYPE_DATA', 'AVMEDIA_TYPE_SUBTITLE', 'AVMEDIA_TYPE_ATTACHMENT', 'AVMEDIA_TYPE_NB', 'AV_SAMPLE_FMT_U8', 'AV_SAMPLE_FMT_S16', 'AV_SAMPLE_FMT_S32', 'AV_SAMPLE_FMT_FLT', 'AV_SAMPLE_FORMAT_DOUBLE', 'AV_SAMPLE_FMT_U8P', 'AV_SAMPLE_FMT_S16P', 'AV_SAMPLE_FMT_S32P', 'AV_SAMPLE_FMT_FLTP', 'AV_SAMPLE_FMT_DBLP', 'AV_SAMPLE_FMT_S64', 'AV_SAMPLE_FMT_S64P', 'AV_NUM_DATA_POINTERS', 'AV_PIX_FMT_RGB24', 'AV_PIX_FMT_ARGB', 'AV_PIX_FMT_RGBA', 'AV_NOPTS_VALUE', 'AV_TIME_BASE', 'AV_TIME_BASE_Q', 'AVFrame', 'AVRational', 'AVDictionary', ]