From b31704420fe9c0a05f8bc9fb5add99a6a5f98b54 Mon Sep 17 00:00:00 2001
From: Roz K <roz@rozk.net>
Date: Fri, 3 Oct 2025 17:07:17 +0200
Subject: [PATCH] decoder

---
 mp4/decoder.py | 6 ++++--
 mp4/demuxer.py | 6 ++++++
 pve.py         | 9 +++++----
 3 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/mp4/decoder.py b/mp4/decoder.py
index c244bc1..52da118 100644
--- a/mp4/decoder.py
+++ b/mp4/decoder.py
@@ -5,7 +5,7 @@ from .packet import Packet
 from .frame import Frame
 
 class Decoder:
-    __slots__ = '_context', '_index'
+    __slots__ = '_context'
 
     def __init__(self, stream):
         self._context = libav.codec_alloc_context(stream.codec)
@@ -19,7 +19,6 @@ class Decoder:
         if errcode < 0:
             libav.codec_free_context(self._context)
             raise Exception("Failed to open codec context")
-        self._index = 0
 
     def __del__(self):
         if self._context:
@@ -46,3 +45,6 @@ class Decoder:
             errstring = libav.strerror(errcode)
             raise Exception(f"Failed to send packet: {errstring}")
         return self._receive()
+
+    def flush(self):
+        return self.decode(None)
diff --git a/mp4/demuxer.py b/mp4/demuxer.py
index 4627767..fbb9c9c 100644
--- a/mp4/demuxer.py
+++ b/mp4/demuxer.py
@@ -22,6 +22,12 @@ class Demuxer:
         self.video_stream = self._find_stream(libav.AVMEDIA_TYPE_VIDEO)
         self.audio_stream = self._find_stream(libav.AVMEDIA_TYPE_AUDIO)
 
+    @property
+    def nb_streams(self):
+        if not self._context:
+            return 0
+        return self._context.contents.nb_streams
+
     def _find_stream(self, type):
         index, codec_ref = libav.format_find_best_stream(self._context, type)
         if index < 0 or not codec_ref:
diff --git a/pve.py b/pve.py
index 73ab92d..9e97c74 100644
--- a/pve.py
+++ b/pve.py
@@ -5,8 +5,9 @@ from mp4.decoder import Decoder
 
 demuxer = Demuxer('test.mp4')
 
-print(demuxer.video_stream.codec.name)
-print(demuxer.audio_stream.codec.name)
+print(f"nb_streams  = {demuxer.nb_streams}")
+print(f"video codec = {demuxer.video_stream.codec.name}")
+print(f"audio codec = {demuxer.audio_stream.codec.name}")
 
 video_decoder = Decoder(demuxer.video_stream)
 audio_decoder = Decoder(demuxer.audio_stream)
@@ -26,10 +27,10 @@ while True:
     else:
         print("unkown packet")
 
-video_frames = video_decoder.decode(None)
+video_frames = video_decoder.flush()
 print(f"flushed {len(video_frames)} video frames")
 
-audio_frames = audio_decoder.decode(None)
+audio_frames = audio_decoder.flush()
 print(f"flushed {len(audio_frames)} audio frames")
 
 demuxer.close()