Rémi Bernon : mf/session: Introduce new (allocate|release)_output_samples helpers.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 8ad97c1c0f6646f9ec7ad6f6f5e329263a8df0c9
URL:    https://gitlab.winehq.org/wine/wine/-/commit/8ad97c1c0f6646f9ec7ad6f6f5e3292...

Author: Rémi Bernon <rbernon(a)codeweavers.com>
Date:   Tue Apr  2 11:19:31 2024 +0200

mf/session: Introduce new (allocate|release)_output_samples helpers.

---

 dlls/mf/session.c | 69 +++++++++++++++++++++++++++++++++----------------------
 1 file changed, 41 insertions(+), 28 deletions(-)

diff --git a/dlls/mf/session.c b/dlls/mf/session.c
index ba18261674d..301dba1de11 100644
--- a/dlls/mf/session.c
+++ b/dlls/mf/session.c
@@ -3172,45 +3172,59 @@ static HRESULT transform_get_external_output_sample(const struct media_session *
     return hr;
 }
 
-static HRESULT transform_node_pull_samples(const struct media_session *session, struct topo_node *node)
+static HRESULT allocate_output_samples(const struct media_session *session, struct topo_node *node,
+        MFT_OUTPUT_DATA_BUFFER *buffers)
 {
-    MFT_OUTPUT_STREAM_INFO stream_info;
-    MFT_OUTPUT_DATA_BUFFER *buffers;
-    HRESULT hr = E_UNEXPECTED;
-    DWORD status = 0;
-    unsigned int i;
-
-    if (!(buffers = calloc(node->u.transform.output_count, sizeof(*buffers))))
-        return E_OUTOFMEMORY;
+    HRESULT hr;
+    UINT i;
 
     for (i = 0; i < node->u.transform.output_count; ++i)
     {
+        MFT_OUTPUT_STREAM_INFO stream_info = {0};
+
         buffers[i].dwStreamID = transform_node_get_stream_id(node, TRUE, i);
-        buffers[i].pSample = NULL;
-        buffers[i].dwStatus = 0;
-        buffers[i].pEvents = NULL;
 
-        memset(&stream_info, 0, sizeof(stream_info));
         if (FAILED(hr = IMFTransform_GetOutputStreamInfo(node->object.transform, buffers[i].dwStreamID, &stream_info)))
-            break;
-
-        if (!(stream_info.dwFlags & (MFT_OUTPUT_STREAM_PROVIDES_SAMPLES | MFT_OUTPUT_STREAM_CAN_PROVIDE_SAMPLES)))
-        {
-            if (FAILED(hr = transform_get_external_output_sample(session, node, i, &stream_info, &buffers[i].pSample)))
-                break;
-        }
+            return hr;
+        if (!(stream_info.dwFlags & (MFT_OUTPUT_STREAM_PROVIDES_SAMPLES | MFT_OUTPUT_STREAM_CAN_PROVIDE_SAMPLES))
+                && FAILED(hr = transform_get_external_output_sample(session, node, i, &stream_info, &buffers[i].pSample)))
+            return hr;
     }
 
-    if (SUCCEEDED(hr))
-        hr = IMFTransform_ProcessOutput(node->object.transform, 0, node->u.transform.output_count, buffers, &status);
+    return S_OK;
+}
+
+static void release_output_samples(struct topo_node *node, MFT_OUTPUT_DATA_BUFFER *buffers)
+{
+    UINT i;
 
-    /* Collect returned samples for all streams. */
     for (i = 0; i < node->u.transform.output_count; ++i)
     {
-        struct transform_stream *stream = &node->u.transform.outputs[i];
-
+        if (buffers[i].pSample)
+            IMFSample_Release(buffers[i].pSample);
         if (buffers[i].pEvents)
             IMFCollection_Release(buffers[i].pEvents);
+    }
+}
+
+static HRESULT transform_node_pull_samples(const struct media_session *session, struct topo_node *node)
+{
+    MFT_OUTPUT_DATA_BUFFER *buffers;
+    DWORD status;
+    HRESULT hr;
+    UINT i;
+
+    if (!(buffers = calloc(node->u.transform.output_count, sizeof(*buffers))))
+        return E_OUTOFMEMORY;
+    if (FAILED(hr = allocate_output_samples(session, node, buffers)))
+        goto done;
+
+    status = 0;
+    hr = IMFTransform_ProcessOutput(node->object.transform, 0, node->u.transform.output_count, buffers, &status);
+
+    for (i = 0; i < node->u.transform.output_count; ++i)
+    {
+        struct transform_stream *stream = &node->u.transform.outputs[i];
 
         if (SUCCEEDED(hr) && !(buffers[i].dwStatus & MFT_OUTPUT_DATA_BUFFER_NO_SAMPLE))
         {
@@ -3219,11 +3233,10 @@ static HRESULT transform_node_pull_samples(const struct media_session *session,
             if (FAILED(hr = transform_stream_push_sample(stream, buffers[i].pSample)))
                 WARN("Failed to queue output sample, hr %#lx\n", hr);
         }
-
-        if (buffers[i].pSample)
-            IMFSample_Release(buffers[i].pSample);
     }
 
+done:
+    release_output_samples(node, buffers);
     free(buffers);
 
     return hr;