[PATCH 6/7] winedmo: Pass demuxer packets through the bitstream filters.

[Rémi Bernon]

From: Rémi Bernon

Wine-Bug: https://bugs.winehq.org/show_bug.cgi?id=57204
---
 dlls/winedmo/unix_demuxer.c | 68 ++++++++++++++++++++++++++++++-------
 1 file changed, 55 insertions(+), 13 deletions(-)

diff --git a/dlls/winedmo/unix_demuxer.c b/dlls/winedmo/unix_demuxer.c
index 3da4a495c95..f983b370a39 100644
--- a/dlls/winedmo/unix_demuxer.c
+++ b/dlls/winedmo/unix_demuxer.c
@@ -37,6 +37,7 @@ static inline const char *debugstr_averr( int err )
 struct stream
 {
     AVBSFContext *filter;
+    BOOL eos;
 };

struct demuxer
@@ -45,6 +46,7 @@ struct demuxer
     struct stream *streams;

AVPacket *last_packet; /* last read packet */
+    struct stream *last_stream; /* last read packet stream */
 };

static struct demuxer *get_demuxer( struct winedmo_demuxer demuxer )
@@ -202,6 +204,50 @@ NTSTATUS demuxer_destroy( void *arg )
     return STATUS_SUCCESS;
 }

+static NTSTATUS demuxer_filter_packet( struct demuxer *demuxer, AVPacket **packet )
+{
+    int i, ret;
+
+    do
+    {
+        if ((*packet = demuxer->last_packet)) return STATUS_SUCCESS;
+        if (!(*packet = av_packet_alloc())) return STATUS_NO_MEMORY;
+
+        if (demuxer->last_stream)
+        {
+            if (!(ret = av_bsf_receive_packet( demuxer->last_stream->filter, *packet ))) return STATUS_SUCCESS;
+            if (ret == AVERROR_EOF) demuxer->last_stream->eos = TRUE;
+            if (ret == AVERROR_EOF || ret == AVERROR(EAGAIN)) ret = 0;
+            if (ret < 0) WARN( "Failed to read packet from filter, error %s.\n", debugstr_averr( ret ) );
+            demuxer->last_stream = NULL;
+        }
+
+        if (!ret && !(ret = av_read_frame( demuxer->ctx, *packet )))
+        {
+            demuxer->last_stream = demuxer->streams + (*packet)->stream_index;
+            ret = av_bsf_send_packet( demuxer->last_stream->filter, *packet );
+            if (ret < 0) WARN( "Failed to send packet to filter, error %s.\n", debugstr_averr( ret ) );
+        }
+        else if (ret == AVERROR_EOF)
+        {
+            for (i = 0, ret = 0; !ret && i < demuxer->ctx->nb_streams; i++)
+            {
+                if (demuxer->streams[i].eos) continue;
+                demuxer->last_stream = demuxer->streams + i;
+                ret = av_bsf_send_packet( demuxer->streams[i].filter, NULL );
+            }
+
+            if (!demuxer->last_stream) return STATUS_END_OF_FILE;
+        }
+        if (ret == AVERROR_EOF || ret == AVERROR(EAGAIN)) ret = 0;
+
+        av_packet_free( packet );
+    } while (!ret);
+
+    ERR( "Failed to read packet from demuxer %p, error %s.\n", demuxer, debugstr_averr( ret ) );
+    return STATUS_UNSUCCESSFUL;
+}
+
 NTSTATUS demuxer_read( void *arg )
 {
     struct demuxer_read_params *params = arg;
@@ -210,21 +256,11 @@ NTSTATUS demuxer_read( void *arg )
     UINT capacity = params->sample.size;
     AVStream *stream;
     AVPacket *packet;
-    int ret;
+    NTSTATUS status;

TRACE( "demuxer %p, capacity %#x\n", demuxer, capacity );

-    if (!(packet = demuxer->last_packet))
-    {
-        if (!(packet = av_packet_alloc())) return STATUS_NO_MEMORY;
-        if ((ret = av_read_frame( demuxer->ctx, packet )) < 0)
-        {
-            TRACE( "Failed to read demuxer %p, error %s.\n", demuxer, debugstr_averr( ret ) );
-            av_packet_free( &packet );
-            if (ret == AVERROR_EOF) return STATUS_END_OF_FILE;
-            return STATUS_UNSUCCESSFUL;
-        }
-    }
+    if ((status = demuxer_filter_packet( demuxer, &packet ))) return status;

params->sample.size = packet->size;
     if ((capacity < packet->size))
@@ -251,7 +287,7 @@ NTSTATUS demuxer_seek( void *arg )
     struct demuxer_seek_params *params = arg;
     struct demuxer *demuxer = get_demuxer( params->demuxer );
     int64_t timestamp = params->timestamp * AV_TIME_BASE / 10000000;
-    int ret;
+    int i, ret;

TRACE( "demuxer %p, timestamp 0x%s\n", demuxer, wine_dbgstr_longlong( params->timestamp ) );

@@ -261,6 +297,12 @@ NTSTATUS demuxer_seek( void *arg )
         return STATUS_UNSUCCESSFUL;
     }

+    for (i = 0; i < demuxer->ctx->nb_streams; i++)
+    {
+        av_bsf_flush( demuxer->streams[i].filter );
+        demuxer->streams[i].eos = FALSE;
+    }
+
     return STATUS_SUCCESS;
 }