[PATCH 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   | 27 ++++++++++++++++++++++++++-
 4 files changed, 47 insertions(+), 1 deletion(-)

diff --git a/dlls/winegstreamer/gst_private.h b/dlls/winegstreamer/gst_private.h
index 5b4c01a3cd0..8627c1f0d97 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 19629d12fd0..42ce5ca598b 100644
--- a/dlls/winegstreamer/unixlib.h
+++ b/dlls/winegstreamer/unixlib.h
@@ -230,6 +230,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;
@@ -338,6 +344,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..1c2b5ac0fc2 100644
--- a/dlls/winegstreamer/wg_parser.c
+++ b/dlls/winegstreamer/wg_parser.c
@@ -96,6 +96,9 @@ struct wg_parser
     bool unlimited_buffering;

gchar *sink_caps;
+
+    GstCaps *video_codec_sink_caps;
+    GstCaps *audio_codec_sink_caps;
 };

struct wg_parser_stream
@@ -106,7 +109,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 +211,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;
@@ -499,6 +510,14 @@ static GstAutoplugSelectResult autoplug_select_cb(GstElement *bin, GstPad *pad,
     if (!parser->sink_caps && strstr(klass, GST_ELEMENT_FACTORY_KLASS_DEMUXER))
         parser->sink_caps = g_strdup(gst_structure_get_name(gst_caps_get_structure(caps, 0)));

+    if (strstr(klass, GST_ELEMENT_FACTORY_KLASS_DECODER) || strstr(klass, GST_ELEMENT_FACTORY_KLASS_ENCODER))
+    {
+        if (!parser->video_codec_sink_caps && strstr(klass, GST_ELEMENT_FACTORY_KLASS_MEDIA_VIDEO))
+            parser->video_codec_sink_caps = caps;
+        else if (!parser->audio_codec_sink_caps && strstr(klass, GST_ELEMENT_FACTORY_KLASS_MEDIA_AUDIO))
+            parser->audio_codec_sink_caps = caps;
+    }
+
     return GST_AUTOPLUG_SELECT_TRY;
 }

@@ -783,6 +802,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);

@@ -884,6 +904,8 @@ static void pad_added_cb(GstElement *element, GstPad *pad, gpointer user)
         stream->post_sink = gst_element_get_static_pad(deinterlace, "sink");
         stream->post_src = gst_element_get_static_pad(vconv2, "src");
         stream->flip = flip;
+        if (parser->video_codec_sink_caps)
+            wg_format_from_caps(&stream->codec_format, parser->video_codec_sink_caps);
     }
     else if (!strcmp(name, "audio/x-raw"))
     {
@@ -901,6 +923,8 @@ static void pad_added_cb(GstElement *element, GstPad *pad, gpointer user)

stream->post_sink = gst_element_get_static_pad(convert, "sink");
         stream->post_src = gst_element_get_static_pad(convert, "src");
+        if (parser->audio_codec_sink_caps)
+            wg_format_from_caps(&stream->codec_format, parser->audio_codec_sink_caps);
     }

if (stream->post_sink)
@@ -1797,6 +1821,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),