[PATCH 03/10] winegstreamer: Implement thinning in wg_parser.

[Charlotte Pabst]

From: Charlotte Pabst

---
 dlls/winegstreamer/gst_private.h |  2 ++
 dlls/winegstreamer/main.c        | 12 +++++++
 dlls/winegstreamer/unixlib.h     |  8 +++++
 dlls/winegstreamer/wg_parser.c   | 62 ++++++++++++++++++++++++++++++++
 4 files changed, 84 insertions(+)

diff --git a/dlls/winegstreamer/gst_private.h b/dlls/winegstreamer/gst_private.h
index 60e0c3af8bc..dac5a88bc5c 100644
--- a/dlls/winegstreamer/gst_private.h
+++ b/dlls/winegstreamer/gst_private.h
@@ -81,6 +81,8 @@ void wg_parser_disconnect(wg_parser_t parser);
 bool wg_parser_get_next_read_offset(wg_parser_t parser, uint64_t *offset, uint32_t *size);
 void wg_parser_push_data(wg_parser_t parser, const void *data, uint32_t size);

+void wg_parser_set_thin(wg_parser_t parser, BOOL thin);
+
 uint32_t wg_parser_get_stream_count(wg_parser_t parser);
 wg_parser_stream_t wg_parser_get_stream(wg_parser_t parser, uint32_t index);

diff --git a/dlls/winegstreamer/main.c b/dlls/winegstreamer/main.c
index d18db8b21e8..94a779bf4fb 100644
--- a/dlls/winegstreamer/main.c
+++ b/dlls/winegstreamer/main.c
@@ -240,6 +240,18 @@ void wg_parser_push_data(wg_parser_t parser, const void *data, uint32_t size)
     WINE_UNIX_CALL(unix_wg_parser_push_data, &params);
 }

+void wg_parser_set_thin(wg_parser_t parser, BOOL thin)
+{
+    struct wg_parser_set_thin_params params =
+    {
+        .parser = parser,
+        .thin = thin,
+    };
+    TRACE("parser %#I64x, thin %d.\n", parser, thin);
+
+    WINE_UNIX_CALL(unix_wg_parser_set_thin, &params);
+}
+
 uint32_t wg_parser_get_stream_count(wg_parser_t parser)
 {
     struct wg_parser_get_stream_count_params params =
diff --git a/dlls/winegstreamer/unixlib.h b/dlls/winegstreamer/unixlib.h
index 179f15f78f7..bde18afd77e 100644
--- a/dlls/winegstreamer/unixlib.h
+++ b/dlls/winegstreamer/unixlib.h
@@ -246,6 +246,12 @@ struct wg_parser_push_data_params
     UINT32 size;
 };

+struct wg_parser_set_thin_params
+{
+    wg_parser_t parser;
+    BOOL thin;
+};
+
 struct wg_parser_get_stream_count_params
 {
     wg_parser_t parser;
@@ -430,6 +436,8 @@ enum unix_funcs
     unix_wg_parser_get_next_read_offset,
     unix_wg_parser_push_data,

+    unix_wg_parser_set_thin,
+
     unix_wg_parser_get_stream_count,
     unix_wg_parser_get_stream,

diff --git a/dlls/winegstreamer/wg_parser.c b/dlls/winegstreamer/wg_parser.c
index e806298fb57..a7988513107 100644
--- a/dlls/winegstreamer/wg_parser.c
+++ b/dlls/winegstreamer/wg_parser.c
@@ -80,6 +80,7 @@ struct wg_parser
     bool output_compressed;
     bool no_more_pads, has_duration, error;
     bool err_on, warn_on;
+    bool thin;

pthread_cond_t read_cond, read_done_cond;
     struct
@@ -121,6 +122,11 @@ struct wg_parser_stream
     gchar *tags[WG_PARSER_TAG_COUNT];
 };

+struct decoder_pad_probe_param
+{
+    struct wg_parser *parser;
+};
+
 static struct wg_parser *get_parser(wg_parser_t parser)
 {
     return (struct wg_parser *)(ULONG_PTR)parser;
@@ -226,6 +232,18 @@ static NTSTATUS wg_parser_push_data(void *args)
     return S_OK;
 }

+static NTSTATUS wg_parser_set_thin(void *args)
+{
+    const struct wg_parser_set_thin_params *params = args;
+    struct wg_parser *parser = get_parser(params->parser);
+
+    pthread_mutex_lock(&parser->mutex);
+    parser->thin = params->thin;
+    pthread_mutex_unlock(&parser->mutex);
+
+    return S_OK;
+}
+
 static NTSTATUS wg_parser_stream_get_current_format(void *args)
 {
     const struct wg_parser_stream_get_current_format_params *params = args;
@@ -600,10 +618,50 @@ static void no_more_pads_cb(GstElement *element, gpointer user)
     pthread_cond_signal(&parser->init_cond);
 }

+static GstPadProbeReturn decoder_pad_probe(GstPad *pad, GstPadProbeInfo *info, gpointer user)
+{
+    struct decoder_pad_probe_param *param = user;
+    GstBuffer *buffer;
+    bool thin;
+
+    pthread_mutex_lock(&param->parser->mutex);
+    thin = param->parser->thin;
+    pthread_mutex_unlock(&param->parser->mutex);
+
+    if (thin && (buffer = gst_pad_probe_info_get_buffer(info)))
+    {
+        if (GST_BUFFER_FLAG_IS_SET(buffer, GST_BUFFER_FLAG_DELTA_UNIT))
+        {
+            GST_DEBUG("Stream is thinned; discarding delta frame.");
+            return GST_PAD_PROBE_DROP;
+        }
+        else
+        {
+            GST_DEBUG("Stream is thinned; found key frame.");
+        }
+    }
+
+    return GST_PAD_PROBE_OK;
+}
+
 static void deep_element_added_cb(GstBin *self, GstBin *sub_bin, GstElement *element, gpointer user)
 {
+    struct wg_parser *parser = user;
+
     if (element)
+    {
+        GstElementFactory *fact = GST_ELEMENT_GET_CLASS(element)->elementfactory;
+        const char *klass = gst_element_factory_get_klass(fact);
+        GstPad *sink;
+        if (strstr(klass, GST_ELEMENT_FACTORY_KLASS_DECODER) && (sink = gst_element_get_static_pad(element, "sink")))
+        {
+            struct decoder_pad_probe_param *param = malloc(sizeof *param);
+            param->parser = parser;
+            gst_pad_add_probe(sink, GST_PAD_PROBE_TYPE_BUFFER, decoder_pad_probe, param, free);
+        }
+
         set_max_threads(element);
+    }
 }

static gboolean sink_event_cb(GstPad *pad, GstObject *parent, GstEvent *event)
@@ -1911,6 +1969,8 @@ const unixlib_entry_t __wine_unix_call_funcs[] =
     X(wg_parser_get_next_read_offset),
     X(wg_parser_push_data),

+    X(wg_parser_set_thin),
+
     X(wg_parser_get_stream_count),
     X(wg_parser_get_stream),

@@ -2308,6 +2368,8 @@ const unixlib_entry_t __wine_unix_call_wow64_funcs[] =
     X(wg_parser_get_next_read_offset),
     X64(wg_parser_push_data),

+    X(wg_parser_set_thin),
+
     X(wg_parser_get_stream_count),
     X(wg_parser_get_stream),