[PATCH v9 12/12] windows.media.speech: Store recorded audio in a temporary ringbuffer.

[Bernhard Kölbl]

From: Bernhard Kölbl

Signed-off-by: Bernhard Kölbl 
---
 dlls/windows.media.speech/recognizer.c | 28 +++++++++++++++++++++-----
 1 file changed, 23 insertions(+), 5 deletions(-)

diff --git a/dlls/windows.media.speech/recognizer.c b/dlls/windows.media.speech/recognizer.c
index 6864ae02da8..cf9428fa7db 100644
--- a/dlls/windows.media.speech/recognizer.c
+++ b/dlls/windows.media.speech/recognizer.c
@@ -192,16 +192,19 @@ static DWORD CALLBACK session_worker_thread_cb( void *args )
     ISpeechContinuousRecognitionSession *iface = args;
     struct session *impl = impl_from_ISpeechContinuousRecognitionSession(iface);
     BOOLEAN running = TRUE, paused = FALSE;
+    UINT32 frame_count, tmp_buf_size;
+    BYTE *audio_buf, *tmp_buf;
     DWORD flags, status;
-    UINT32 frame_count;
     HANDLE events[2];
-    BYTE *audio_buf;

SetThreadDescription(GetCurrentThread(), L"wine_speech_recognition_session_worker");

IAudioClient_Start(impl->audio_client);
     IAudioClient_GetBufferSize(impl->audio_client, &frame_count);

+    tmp_buf_size = sizeof(*tmp_buf) * frame_count * impl->capture_wfx.nBlockAlign;
+    tmp_buf = malloc(tmp_buf_size);
+
     while (running)
     {
         BOOLEAN old_paused = paused;
@@ -232,13 +235,27 @@ static DWORD CALLBACK session_worker_thread_cb( void *args )
         }
         else if (status == 1) /* audio_buf_event signaled */
         {
-            UINT32 frames_available = 0;
+            UINT32 frames_available = 0, tmp_buf_offset = 0;
+            SIZE_T packet_size = frames_available * impl->capture_wfx.nBlockAlign;

-            while (IAudioCaptureClient_GetBuffer(impl->capture_client, &audio_buf, &frames_available, &flags, NULL, NULL) == S_OK)
+            while (tmp_buf_offset < tmp_buf_size
+                   && IAudioCaptureClient_GetBuffer(impl->capture_client, &audio_buf, &frames_available, &flags, NULL, NULL) == S_OK)
             {
-                /* TODO: Send mic data to recognizer and handle results. */
+                if (tmp_buf_offset + packet_size > tmp_buf_size)
+                {
+                    /* Defer processing until the next iteration of the worker loop. */
+                    IAudioCaptureClient_ReleaseBuffer(impl->capture_client, 0);
+                    SetEvent(impl->audio_buf_event);
+                    break;
+                }
+
+                memcpy(tmp_buf + tmp_buf_offset, audio_buf, packet_size);
+                tmp_buf_offset += packet_size;
+
                 IAudioCaptureClient_ReleaseBuffer(impl->capture_client, frames_available);
             }
+
+            /* TODO: Send mic data to recognizer and handle results. */
         }
         else
         {
@@ -248,6 +265,7 @@ static DWORD CALLBACK session_worker_thread_cb( void *args )
     }

IAudioClient_Stop(impl->audio_client);
+    free(tmp_buf);

return 0;
 }