Ziqing Hui : winegstreamer: Implement wg_parser_stream_get_codec_format.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 13188bb14bec6052fa108f42a96d1e4964e507b4
URL:    https://gitlab.winehq.org/wine/wine/-/commit/13188bb14bec6052fa108f42a96d1e4...

Author: Ziqing Hui <zhui(a)codeweavers.com>
Date:   Mon Mar 13 12:13:52 2023 +0800

winegstreamer: Implement wg_parser_stream_get_codec_format.

---

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

diff --git a/dlls/winegstreamer/gst_private.h b/dlls/winegstreamer/gst_private.h
index 04ec84c7936..54f59aa708a 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 1e54edb2015..2e7763872d0 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, ¶ms);
 }
 
+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, ¶ms);
+}
+
 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 a01efb934f8..e20fdd7256b 100644
--- a/dlls/winegstreamer/unixlib.h
+++ b/dlls/winegstreamer/unixlib.h
@@ -240,6 +240,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;
@@ -350,6 +356,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 16867c0da7d..e80d7501f2c 100644
--- a/dlls/winegstreamer/wg_parser.c
+++ b/dlls/winegstreamer/wg_parser.c
@@ -107,7 +107,7 @@ struct wg_parser_stream
     GstPad *their_src, *my_sink;
     GstElement *flip, *decodebin;
     GstSegment segment;
-    struct wg_format preferred_format, current_format, stream_format;
+    struct wg_format preferred_format, current_format, codec_format;
 
     pthread_cond_t event_cond, event_empty_cond;
     GstBuffer *buffer;
@@ -216,6 +216,16 @@ 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 = format_is_compressed(¶ms->stream->codec_format) ?
+            params->stream->codec_format :
+            params->stream->preferred_format;
+    return S_OK;
+}
+
 static NTSTATUS wg_parser_stream_enable(void *args)
 {
     const struct wg_parser_stream_enable_params *params = args;
@@ -950,11 +960,11 @@ static void pad_added_cb(GstElement *element, GstPad *pad, gpointer user)
         return;
 
     caps = gst_pad_query_caps(pad, NULL);
-    wg_format_from_caps(&stream->stream_format, caps);
+    wg_format_from_caps(&stream->codec_format, caps);
     gst_caps_unref(caps);
 
     /* For compressed stream, create an extra decodebin to decode it. */
-    if (format_is_compressed(&stream->stream_format))
+    if (format_is_compressed(&stream->codec_format))
     {
         if (!stream_decodebin_create(stream))
         {
@@ -1945,6 +1955,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),