[PATCH v3 1/4] sapi: Implement ISpVoice::SetOutput.

[Shaun Ren]

From: Shaun Ren

---
 dlls/sapi/tests/tts.c | 30 +++++++++++++++++++++++
 dlls/sapi/tts.c       | 56 ++++++++++++++++++++++++++++++++++++++++---
 2 files changed, 83 insertions(+), 3 deletions(-)

diff --git a/dlls/sapi/tests/tts.c b/dlls/sapi/tests/tts.c
index 8303dfc6ebc..5d0cfd1bc1b 100644
--- a/dlls/sapi/tests/tts.c
+++ b/dlls/sapi/tests/tts.c
@@ -93,9 +93,39 @@ static void test_interfaces(void)
     ISpeechVoice_Release(speech_voice);
 }

+static void test_spvoice(void)
+{
+    ISpVoice *voice;
+    ISpMMSysAudio *audio_out;
+    HRESULT hr;
+
+    if (waveOutGetNumDevs() == 0) {
+        skip("no wave out devices.\n");
+        return;
+    }
+
+    hr = CoCreateInstance(&CLSID_SpVoice, NULL, CLSCTX_INPROC_SERVER,
+                          &IID_ISpVoice, (void **)&voice);
+    ok(hr == S_OK, "Failed to create SpVoice: %#lx.\n", hr);
+
+    hr = ISpVoice_SetOutput(voice, NULL, TRUE);
+    ok(hr == S_OK, "got %#lx.\n", hr);
+
+    hr = CoCreateInstance(&CLSID_SpMMAudioOut, NULL, CLSCTX_INPROC_SERVER,
+                          &IID_ISpMMSysAudio, (void **)&audio_out);
+    ok(hr == S_OK, "Failed to create SpMMAudioOut: %#lx.\n", hr);
+
+    hr = ISpVoice_SetOutput(voice, (IUnknown *)audio_out, TRUE);
+    todo_wine ok(hr == S_FALSE, "got %#lx.\n", hr);
+
+    ISpVoice_Release(voice);
+    ISpMMSysAudio_Release(audio_out);
+}
+
 START_TEST(tts)
 {
     CoInitialize(NULL);
     test_interfaces();
+    test_spvoice();
     CoUninitialize();
 }
diff --git a/dlls/sapi/tts.c b/dlls/sapi/tts.c
index 9f60c70e6c4..bebda86f09d 100644
--- a/dlls/sapi/tts.c
+++ b/dlls/sapi/tts.c
@@ -40,6 +40,9 @@ struct speech_voice
     ISpVoice ISpVoice_iface;
     IConnectionPointContainer IConnectionPointContainer_iface;
     LONG ref;
+
+    ISpStreamFormat *output;
+    CRITICAL_SECTION cs;
 };

static inline struct speech_voice *impl_from_ISpeechVoice(ISpeechVoice *iface)
@@ -102,6 +105,9 @@ static ULONG WINAPI speech_voice_Release(ISpeechVoice *iface)

if (!ref)
     {
+        if (This->output) ISpStreamFormat_Release(This->output);
+        DeleteCriticalSection(&This->cs);
+
         heap_free(This);
     }

@@ -515,11 +521,51 @@ static HRESULT WINAPI spvoice_GetInfo(ISpVoice *iface, SPEVENTSOURCEINFO *info)
     return E_NOTIMPL;
 }

-static HRESULT WINAPI spvoice_SetOutput(ISpVoice *iface, IUnknown *unk, BOOL changes)
+static HRESULT WINAPI spvoice_SetOutput(ISpVoice *iface, IUnknown *unk, BOOL allow_format_changes)
 {
-    FIXME("(%p, %p, %d): stub.\n", iface, unk, changes);
+    struct speech_voice *This = impl_from_ISpVoice(iface);
+    ISpStreamFormat *stream = NULL;
+    ISpObjectToken *token = NULL;
+    HRESULT hr;

-    return E_NOTIMPL;
+    TRACE("(%p, %p, %d).\n", iface, unk, allow_format_changes);
+
+    if (!allow_format_changes)
+        FIXME("ignoring allow_format_changes = FALSE.\n");
+
+    if (!unk)
+    {
+        /* TODO: Create the default SpAudioOut token here once SpMMAudioEnum is implemented. */
+        if (FAILED(hr = CoCreateInstance(&CLSID_SpMMAudioOut, NULL, CLSCTX_INPROC_SERVER,
+                                         &IID_ISpStreamFormat, (void **)&stream)))
+            return hr;
+    }
+    else
+    {
+        if (FAILED(IUnknown_QueryInterface(unk, &IID_ISpStreamFormat, (void **)&stream)))
+        {
+            if (FAILED(IUnknown_QueryInterface(unk, &IID_ISpObjectToken, (void **)&token)))
+                return E_INVALIDARG;
+        }
+    }
+
+    if (!stream)
+    {
+        hr = ISpObjectToken_CreateInstance(token, NULL, CLSCTX_ALL, &IID_ISpStreamFormat, (void **)&stream);
+        ISpObjectToken_Release(token);
+        if (FAILED(hr))
+            return hr;
+    }
+
+    EnterCriticalSection(&This->cs);
+
+    if (This->output)
+        ISpStreamFormat_Release(This->output);
+    This->output = stream;
+
+    LeaveCriticalSection(&This->cs);
+
+    return S_OK;
 }

static HRESULT WINAPI spvoice_GetOutputObjectToken(ISpVoice *iface, ISpObjectToken **token)
@@ -798,6 +844,10 @@ HRESULT speech_voice_create(IUnknown *outer, REFIID iid, void **obj)
     This->IConnectionPointContainer_iface.lpVtbl = &container_vtbl;
     This->ref = 1;

+    This->output = NULL;
+
+    InitializeCriticalSection(&This->cs);
+
     hr = ISpeechVoice_QueryInterface(&This->ISpeechVoice_iface, iid, obj);

ISpeechVoice_Release(&This->ISpeechVoice_iface);