RK/rk_pve
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

decoder

Browse Source
This commit is contained in:
Roz K
2025-10-03 16:55:38 +02:00
parent 2316435af8
commit dfd0902256
7 changed files with 122 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
mp4/demuxer.py
View File

@ -6,37 +6,38 @@ from .stream import NullStream, Stream
from .packet import Packet
class Demuxer:
__slots__ = '_ref', 'video_stream', 'audio_stream'
__slots__ = '_context', 'video_stream', 'audio_stream'
def __init__(self, path):
self._ref = libav.format_alloc_context()
if not self._ref:
self._context = libav.format_alloc_context()
if not self._context:
raise MemoryError
errcode = libav.format_open_input(self._ref, "file:" + path)
errcode = libav.format_open_input(self._context, "file:" + path)
if errcode < 0:
raise Exception(f"Failed to open: {path}")
errcode = libav.format_find_stream_info(self._ref)
errcode = libav.format_find_stream_info(self._context)
if errcode < 0:
libav.format_close_input(self._ref)
libav.format_close_input(self._context)
raise Exception("Failed to find stream info")
self.video_stream = self._find_stream(libav.AVMEDIA_TYPE_VIDEO)
self.audio_stream = self._find_stream(libav.AVMEDIA_TYPE_AUDIO)
def _find_stream(self, type):
index, codec_ref = libav.format_find_best_stream(self._ref, type)
index, codec_ref = libav.format_find_best_stream(self._context, type)
if index < 0 or not codec_ref:
return NullStream()
return Stream(index, Codec(codec_ref))
parameters = self._context.contents.streams[index].contents.codecpar
return Stream(index, Codec(codec_ref), parameters)
def read_packet(self):
if not self._ref:
if not self._context:
return None
packet = Packet()
errcode = libav.read_frame(self._ref, packet)
errcode = libav.read_frame(self._context, packet)
if errcode < 0:
return None
return packet
def close(self):
if self._ref:
libav.format_close_input(self._ref)
if self._context:
libav.format_close_input(self._context)