[PATCH v3 05/10] mf/session: Push transform input samples one by one to ProcessInput.

[Rémi Bernon]

From: Rémi Bernon

---
 dlls/mf/session.c | 71 ++++++++++++++++-------------------------------
 1 file changed, 24 insertions(+), 47 deletions(-)

diff --git a/dlls/mf/session.c b/dlls/mf/session.c
index 268a930e21d..863ff81b29f 100644
--- a/dlls/mf/session.c
+++ b/dlls/mf/session.c
@@ -673,14 +673,6 @@ static void session_set_caps(struct media_session *session, DWORD caps)
     IMFMediaEvent_Release(event);
 }

-static void transform_release_sample(struct sample *sample)
-{
-    list_remove(&sample->entry);
-    if (sample->sample)
-        IMFSample_Release(sample->sample);
-    free(sample);
-}
-
 static HRESULT transform_stream_push_sample(struct transform_stream *stream, IMFSample *sample)
 {
     struct sample *entry;
@@ -3176,6 +3168,29 @@ static BOOL transform_node_all_streams_ended(struct topo_node *topo_node)
     return TRUE;
 }

+static void transform_node_push_sample(const struct media_session *session, struct topo_node *topo_node,
+        UINT input, IMFSample *sample)
+{
+    struct transform_stream *stream = &topo_node->u.transform.inputs[input];
+    HRESULT hr;
+
+    if ((stream->stream_ended = !sample))
+    {
+        if (!transform_node_all_streams_ended(topo_node))
+            return;
+
+        if (FAILED(hr = IMFTransform_ProcessMessage(topo_node->object.transform, MFT_MESSAGE_COMMAND_DRAIN, 0)))
+            WARN("Drain command failed for transform, hr %#lx.\n", hr);
+    }
+    else
+    {
+        UINT id = transform_node_get_stream_id(topo_node, FALSE, input);
+        if (FAILED(hr = IMFTransform_ProcessInput(topo_node->object.transform, id, sample, 0))
+                && FAILED(hr = transform_stream_push_sample(stream, sample)))
+            WARN("Failed to queue delayed input sample, hr %#lx\n", hr);
+    }
+}
+
 static void session_deliver_sample_to_node(struct media_session *session, IMFTopologyNode *node, unsigned int input,
         IMFSample *sample);

@@ -3218,12 +3233,9 @@ static void transform_node_deliver_samples(struct media_session *session, struct
 static void session_deliver_sample_to_node(struct media_session *session, IMFTopologyNode *node, unsigned int input,
         IMFSample *sample)
 {
-    struct sample *sample_entry, *sample_entry2;
-    DWORD stream_id;
     struct topo_node *topo_node;
     MF_TOPOLOGY_TYPE node_type;
     TOPOID node_id;
-    unsigned int i;
     HRESULT hr;

if (session->quality_manager)
@@ -3253,45 +3265,10 @@ static void session_deliver_sample_to_node(struct media_session *session, IMFTop
             }
             break;
         case MF_TOPOLOGY_TRANSFORM_NODE:
-        {
-            struct transform_stream *input_stream = &topo_node->u.transform.inputs[input];
-
-            transform_node_pull_samples(session, topo_node);
-
-            transform_stream_push_sample(input_stream, sample);
-
-            for (i = 0; i < topo_node->u.transform.input_count; ++i)
-            {
-                struct transform_stream *stream = &topo_node->u.transform.inputs[i];
-
-                stream_id = transform_node_get_stream_id(topo_node, FALSE, i);
-
-                LIST_FOR_EACH_ENTRY_SAFE(sample_entry, sample_entry2, &stream->samples, struct sample, entry)
-                {
-                    if ((stream->stream_ended = !sample_entry->sample))
-                        transform_stream_drop_samples(stream);
-                    else
-                    {
-                        if ((hr = IMFTransform_ProcessInput(topo_node->object.transform, stream_id,
-                                sample_entry->sample, 0)) == MF_E_NOTACCEPTING)
-                            break;
-                        if (FAILED(hr))
-                            WARN("Failed to process input for stream %u/%lu, hr %#lx.\n", i, stream_id, hr);
-                        transform_release_sample(sample_entry);
-                    }
-                }
-            }
-
-            if (transform_node_all_streams_ended(topo_node))
-            {
-                if (FAILED(hr = IMFTransform_ProcessMessage(topo_node->object.transform, MFT_MESSAGE_COMMAND_DRAIN, 0)))
-                    WARN("Drain command failed for transform, hr %#lx.\n", hr);
-            }
-
+            transform_node_push_sample(session, topo_node, input, sample);
             transform_node_pull_samples(session, topo_node);
             transform_node_deliver_samples(session, topo_node);
             break;
-        }
         case MF_TOPOLOGY_TEE_NODE:
             FIXME("Unhandled downstream node type %d.\n", node_type);
             break;