[PATCH v5 1/4] winegstreamer: Implement wg_parser_stream_get_codec_format.

[Ziqing Hui]

From: Ziqing Hui

---
 dlls/winegstreamer/gst_private.h |  1 +
 dlls/winegstreamer/main.c        | 13 +++++++++++
 dlls/winegstreamer/unixlib.h     |  7 ++++++
 dlls/winegstreamer/wg_parser.c   | 37 +++++++++++++++++++++++++++++---
 4 files changed, 55 insertions(+), 3 deletions(-)

diff --git a/dlls/winegstreamer/gst_private.h b/dlls/winegstreamer/gst_private.h
index c59d3f1bb66..6b7d7a82848 100644
--- a/dlls/winegstreamer/gst_private.h
+++ b/dlls/winegstreamer/gst_private.h
@@ -82,6 +82,7 @@ uint32_t wg_parser_get_stream_count(struct wg_parser *parser);
 struct wg_parser_stream *wg_parser_get_stream(struct wg_parser *parser, uint32_t index);

void wg_parser_stream_get_preferred_format(struct wg_parser_stream *stream, struct wg_format *format);
+void wg_parser_stream_get_codec_format(struct wg_parser_stream *stream, struct wg_format *format);
 void wg_parser_stream_enable(struct wg_parser_stream *stream, const struct wg_format *format);
 void wg_parser_stream_disable(struct wg_parser_stream *stream);

diff --git a/dlls/winegstreamer/main.c b/dlls/winegstreamer/main.c
index 1afa51ac0aa..ba8b01497a3 100644
--- a/dlls/winegstreamer/main.c
+++ b/dlls/winegstreamer/main.c
@@ -185,6 +185,19 @@ void wg_parser_stream_get_preferred_format(struct wg_parser_stream *stream, stru
     WINE_UNIX_CALL(unix_wg_parser_stream_get_preferred_format, &params);
 }

+void wg_parser_stream_get_codec_format(struct wg_parser_stream *stream, struct wg_format *format)
+{
+    struct wg_parser_stream_get_codec_format_params params =
+    {
+        .stream = stream,
+        .format = format,
+    };
+
+    TRACE("stream %p, format %p.\n", stream, format);
+
+    WINE_UNIX_CALL(unix_wg_parser_stream_get_codec_format, &params);
+}
+
 void wg_parser_stream_enable(struct wg_parser_stream *stream, const struct wg_format *format)
 {
     struct wg_parser_stream_enable_params params =
diff --git a/dlls/winegstreamer/unixlib.h b/dlls/winegstreamer/unixlib.h
index 539ccd3e12a..9b0ba090f22 100644
--- a/dlls/winegstreamer/unixlib.h
+++ b/dlls/winegstreamer/unixlib.h
@@ -238,6 +238,12 @@ struct wg_parser_stream_get_preferred_format_params
     struct wg_format *format;
 };

+struct wg_parser_stream_get_codec_format_params
+{
+    struct wg_parser_stream *stream;
+    struct wg_format *format;
+};
+
 struct wg_parser_stream_enable_params
 {
     struct wg_parser_stream *stream;
@@ -346,6 +352,7 @@ enum unix_funcs
     unix_wg_parser_get_stream,

unix_wg_parser_stream_get_preferred_format,
+    unix_wg_parser_stream_get_codec_format,
     unix_wg_parser_stream_enable,
     unix_wg_parser_stream_disable,

diff --git a/dlls/winegstreamer/wg_parser.c b/dlls/winegstreamer/wg_parser.c
index 5bb824f4399..15b0e9ceddd 100644
--- a/dlls/winegstreamer/wg_parser.c
+++ b/dlls/winegstreamer/wg_parser.c
@@ -106,7 +106,7 @@ struct wg_parser_stream
     GstPad *their_src, *post_sink, *post_src, *my_sink;
     GstElement *flip;
     GstSegment segment;
-    struct wg_format preferred_format, current_format;
+    struct wg_format preferred_format, current_format, codec_format;

pthread_cond_t event_cond, event_empty_cond;
     GstBuffer *buffer;
@@ -208,6 +208,14 @@ static NTSTATUS wg_parser_stream_get_preferred_format(void *args)
     return S_OK;
 }

+static NTSTATUS wg_parser_stream_get_codec_format(void *args)
+{
+    struct wg_parser_stream_get_codec_format_params *params = args;
+
+    *params->format = params->stream->codec_format;
+    return S_OK;
+}
+
 static NTSTATUS wg_parser_stream_enable(void *args)
 {
     const struct wg_parser_stream_enable_params *params = args;
@@ -783,6 +791,7 @@ static struct wg_parser_stream *create_stream(struct wg_parser *parser)
     stream->parser = parser;
     stream->number = parser->stream_count;
     stream->current_format.major_type = WG_MAJOR_TYPE_UNKNOWN;
+    stream->codec_format.major_type = WG_MAJOR_TYPE_UNKNOWN;
     pthread_cond_init(&stream->event_cond, NULL);
     pthread_cond_init(&stream->event_empty_cond, NULL);

@@ -826,12 +835,15 @@ static void free_stream(struct wg_parser_stream *stream)

static void pad_added_cb(GstElement *element, GstPad *pad, gpointer user)
 {
+    GstPad *decoder_src_pad, *decoder_sink_pad;
+    GstCaps *caps, *decoder_sink_caps;
     struct wg_parser *parser = user;
     struct wg_parser_stream *stream;
-    const char *name;
-    GstCaps *caps;
+    const char *name, *klass;
+    GstElement *decoder;
     int ret;

+
     GST_LOG("parser %p, element %p, pad %p.", parser, element, pad);

if (gst_pad_is_linked(pad))
@@ -843,6 +855,24 @@ static void pad_added_cb(GstElement *element, GstPad *pad, gpointer user)
     if (!(stream = create_stream(parser)))
         goto out;

+    /* Get decoder format for decodebin. */
+    if (!strcmp(gst_element_get_metadata(element, GST_ELEMENT_METADATA_LONGNAME), "Decoder Bin"))
+    {
+        decoder_src_pad = gst_ghost_pad_get_target(GST_GHOST_PAD(pad));
+        decoder = gst_pad_get_parent_element(decoder_src_pad);
+        klass = gst_element_get_metadata(decoder, GST_ELEMENT_METADATA_KLASS);
+        if (strstr(klass, GST_ELEMENT_FACTORY_KLASS_DECODER))
+        {
+            decoder_sink_pad = gst_element_get_static_pad(decoder, "sink");
+            decoder_sink_caps = gst_pad_get_current_caps(decoder_sink_pad);
+            wg_format_from_caps(&stream->codec_format, decoder_sink_caps);
+            gst_caps_unref(decoder_sink_caps);
+            gst_object_unref(decoder_sink_pad);
+        }
+        gst_object_unref(decoder);
+        gst_object_unref(decoder_src_pad);
+    }
+
     if (!strcmp(name, "video/x-raw"))
     {
         GstElement *deinterlace, *vconv, *flip, *vconv2;
@@ -1797,6 +1827,7 @@ const unixlib_entry_t __wine_unix_call_funcs[] =
     X(wg_parser_get_stream),

X(wg_parser_stream_get_preferred_format),
+    X(wg_parser_stream_get_codec_format),
     X(wg_parser_stream_enable),
     X(wg_parser_stream_disable),