[PATCH v2 5/5] winegstreamer/video_encoder: Implement GetOutputStreamInfo.

[Ziqing Hui]

From: Ziqing Hui

---
 dlls/winegstreamer/video_encoder.c | 26 +++++++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/dlls/winegstreamer/video_encoder.c b/dlls/winegstreamer/video_encoder.c
index 512a4dcfa4d..40e803a9073 100644
--- a/dlls/winegstreamer/video_encoder.c
+++ b/dlls/winegstreamer/video_encoder.c
@@ -46,6 +46,7 @@ struct video_encoder
     IMFMediaType *input_type;
     MFT_INPUT_STREAM_INFO input_info;
     IMFMediaType *output_type;
+    MFT_OUTPUT_STREAM_INFO output_info;

IMFAttributes *attributes;
 };
@@ -94,6 +95,20 @@ done:
     return hr;
 }

+static HRESULT update_output_info(struct video_encoder *encoder)
+{
+    UINT64 frame_size;
+    HRESULT hr;
+
+    if (FAILED(hr = IMFMediaType_GetUINT64(encoder->output_type, &MF_MT_FRAME_SIZE, &frame_size)))
+        return hr;
+
+    /* FIXME: It's hard to calculate the encoded output size,
+     * use a uncompressed size here and hope it will work. */
+    return MFCalculateImageSize(&MFVideoFormat_NV12, frame_size >> 32, frame_size,
+            (UINT32 *)&encoder->output_info.cbSize);
+}
+
 static HRESULT WINAPI transform_QueryInterface(IMFTransform *iface, REFIID iid, void **out)
 {
     struct video_encoder *encoder = impl_from_IMFTransform(iface);
@@ -179,8 +194,12 @@ static HRESULT WINAPI transform_GetInputStreamInfo(IMFTransform *iface, DWORD id

static HRESULT WINAPI transform_GetOutputStreamInfo(IMFTransform *iface, DWORD id, MFT_OUTPUT_STREAM_INFO *info)
 {
-    FIXME("iface %p, id %#lx, info %p.\n", iface, id, info);
-    return E_NOTIMPL;
+    struct video_encoder *encoder = impl_from_IMFTransform(iface);
+
+    TRACE("iface %p, id %#lx, info %p.\n", iface, id, info);
+
+    *info = encoder->output_info;
+    return S_OK;
 }

static HRESULT WINAPI transform_GetAttributes(IMFTransform *iface, IMFAttributes **attributes)
@@ -327,6 +346,7 @@ static HRESULT WINAPI transform_SetOutputType(IMFTransform *iface, DWORD id, IMF
         {
             IMFMediaType_Release(encoder->output_type);
             encoder->output_type = NULL;
+            memset(&encoder->output_info, 0, sizeof(encoder->output_info));
         }
         return S_OK;
     }
@@ -361,7 +381,7 @@ static HRESULT WINAPI transform_SetOutputType(IMFTransform *iface, DWORD id, IMF

/* FIXME: Add MF_MT_MPEG_SEQUENCE_HEADER attribute. */

-    return S_OK;
+    return update_output_info(encoder);
 }

static HRESULT WINAPI transform_GetInputCurrentType(IMFTransform *iface, DWORD id, IMFMediaType **type)