Nikolay Sivov : mfmediaengine: Keep latest video frame buffer.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: c7023042e109d7592433b61cb25ca0d3f77d048b
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=c7023042e109d7592433b61cb25ca0d3f77d048b

Author: Nikolay Sivov <nsivov at codeweavers.com>
Date:   Mon May 17 12:26:48 2021 +0300

mfmediaengine: Keep latest video frame buffer.

Signed-off-by: Nikolay Sivov <nsivov at codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 dlls/mfmediaengine/main.c | 29 +++++++++++++++++++----------
 1 file changed, 19 insertions(+), 10 deletions(-)

diff --git a/dlls/mfmediaengine/main.c b/dlls/mfmediaengine/main.c
index c6c73b625f1..0678768cf08 100644
--- a/dlls/mfmediaengine/main.c
+++ b/dlls/mfmediaengine/main.c
@@ -84,14 +84,6 @@ enum media_engine_flags
     FLAGS_ENGINE_IS_ENDED = 0x4000,
 };
 
-struct video_frame
-{
-    LONGLONG pts;
-    SIZE size;
-    SIZE ratio;
-    TOPOID node_id;
-};
-
 struct media_engine
 {
     IMFMediaEngine IMFMediaEngine_iface;
@@ -115,7 +107,15 @@ struct media_engine
     IMFPresentationClock *clock;
     IMFSourceResolver *resolver;
     BSTR current_source;
-    struct video_frame video_frame;
+    struct
+    {
+        LONGLONG pts;
+        SIZE size;
+        SIZE ratio;
+        TOPOID node_id;
+        BYTE *buffer;
+        UINT buffer_size;
+    } video_frame;
     CRITICAL_SECTION cs;
 };
 
@@ -976,6 +976,7 @@ static void free_media_engine(struct media_engine *engine)
         IMFSourceResolver_Release(engine->resolver);
     SysFreeString(engine->current_source);
     DeleteCriticalSection(&engine->cs);
+    free(engine->video_frame.buffer);
     free(engine);
 }
 
@@ -1781,7 +1782,7 @@ static HRESULT WINAPI media_engine_grabber_callback_OnSetPresentationClock(IMFSa
 
 static HRESULT WINAPI media_engine_grabber_callback_OnProcessSample(IMFSampleGrabberSinkCallback *iface,
         REFGUID major_type, DWORD sample_flags, LONGLONG sample_time, LONGLONG sample_duration,
-        const BYTE *buffer, DWORD sample_size)
+        const BYTE *buffer, DWORD buffer_size)
 {
     struct media_engine *engine = impl_from_IMFSampleGrabberSinkCallback(iface);
 
@@ -1793,6 +1794,14 @@ static HRESULT WINAPI media_engine_grabber_callback_OnProcessSample(IMFSampleGra
         media_engine_set_flag(engine, FLAGS_ENGINE_FIRST_FRAME, TRUE);
     }
     engine->video_frame.pts = sample_time;
+    if (engine->video_frame.buffer_size < buffer_size)
+    {
+        free(engine->video_frame.buffer);
+        if ((engine->video_frame.buffer = malloc(buffer_size)))
+            engine->video_frame.buffer_size = buffer_size;
+    }
+    if (engine->video_frame.buffer)
+        memcpy(engine->video_frame.buffer, buffer, buffer_size);
 
     LeaveCriticalSection(&engine->cs);