Signed-off-by: Zebediah Figura z.figura12@gmail.com --- dlls/amstream/multimedia.c | 113 +++++++++++++++++---------------- dlls/amstream/tests/amstream.c | 54 +++++++--------- 2 files changed, 83 insertions(+), 84 deletions(-)
diff --git a/dlls/amstream/multimedia.c b/dlls/amstream/multimedia.c index ac61010698..a14771ec75 100644 --- a/dlls/amstream/multimedia.c +++ b/dlls/amstream/multimedia.c @@ -34,12 +34,12 @@ struct multimedia_stream { IAMMultiMediaStream IAMMultiMediaStream_iface; LONG ref; - IGraphBuilder* pFilterGraph; + IGraphBuilder *graph; IMediaSeeking* media_seeking; IMediaControl* media_control; IMediaStreamFilter *filter; IPin* ipin; - STREAM_TYPE StreamType; + STREAM_TYPE type; OAEVENT event; };
@@ -95,8 +95,8 @@ static ULONG WINAPI multimedia_stream_Release(IAMMultiMediaStream *iface) IMediaSeeking_Release(This->media_seeking); if (This->media_control) IMediaControl_Release(This->media_control); - if (This->pFilterGraph) - IGraphBuilder_Release(This->pFilterGraph); + if (This->graph) + IGraphBuilder_Release(This->graph); HeapFree(GetProcessHeap(), 0, This); }
@@ -194,62 +194,64 @@ static HRESULT WINAPI multimedia_stream_GetEndOfStream(IAMMultiMediaStream *ifac return E_NOTIMPL; }
-/*** IAMMultiMediaStream methods ***/ -static HRESULT WINAPI multimedia_stream_Initialize(IAMMultiMediaStream *iface, - STREAM_TYPE StreamType, DWORD dwFlags, IGraphBuilder *pFilterGraph) +static HRESULT create_graph(struct multimedia_stream *mmstream, IGraphBuilder *graph) { - struct multimedia_stream *This = impl_from_IAMMultiMediaStream(iface); - HRESULT hr = S_OK; - - TRACE("(%p/%p)->(%x,%x,%p)\n", This, iface, (DWORD)StreamType, dwFlags, pFilterGraph); + IMediaEventEx *eventsrc; + HRESULT hr;
- if (pFilterGraph) - { - This->pFilterGraph = pFilterGraph; - IGraphBuilder_AddRef(This->pFilterGraph); - } - else - { - hr = CoCreateInstance(&CLSID_FilterGraph, NULL, CLSCTX_INPROC_SERVER, &IID_IGraphBuilder, (LPVOID*)&This->pFilterGraph); - } + if (graph) + IGraphBuilder_AddRef(mmstream->graph = graph); + else if (FAILED(hr = CoCreateInstance(&CLSID_FilterGraph, NULL, + CLSCTX_INPROC_SERVER, &IID_IGraphBuilder, (void **)&mmstream->graph))) + return hr;
+ hr = IGraphBuilder_QueryInterface(mmstream->graph, &IID_IMediaSeeking, (void **)&mmstream->media_seeking); + if (SUCCEEDED(hr)) + hr = IGraphBuilder_QueryInterface(mmstream->graph, &IID_IMediaControl, (void **)&mmstream->media_control); + if (SUCCEEDED(hr)) + hr = IGraphBuilder_AddFilter(mmstream->graph, (IBaseFilter *)mmstream->filter, L"MediaStreamFilter"); + if (SUCCEEDED(hr)) + hr = IGraphBuilder_QueryInterface(mmstream->graph, &IID_IMediaEventEx, (void **)&eventsrc); if (SUCCEEDED(hr)) { - This->StreamType = StreamType; - hr = IGraphBuilder_QueryInterface(This->pFilterGraph, &IID_IMediaSeeking, (void**)&This->media_seeking); - if (SUCCEEDED(hr)) - hr = IGraphBuilder_QueryInterface(This->pFilterGraph, &IID_IMediaControl, (void**)&This->media_control); + hr = IMediaEventEx_GetEventHandle(eventsrc, &mmstream->event); if (SUCCEEDED(hr)) - hr = IGraphBuilder_AddFilter(This->pFilterGraph, (IBaseFilter*)This->filter, L"MediaStreamFilter"); - if (SUCCEEDED(hr)) - { - IMediaEventEx* media_event = NULL; - hr = IGraphBuilder_QueryInterface(This->pFilterGraph, &IID_IMediaEventEx, (void**)&media_event); - if (SUCCEEDED(hr)) - hr = IMediaEventEx_GetEventHandle(media_event, &This->event); - if (SUCCEEDED(hr)) - hr = IMediaEventEx_SetNotifyFlags(media_event, AM_MEDIAEVENT_NONOTIFY); - if (media_event) - IMediaEventEx_Release(media_event); - } + hr = IMediaEventEx_SetNotifyFlags(eventsrc, AM_MEDIAEVENT_NONOTIFY); + IMediaEventEx_Release(eventsrc); }
if (FAILED(hr)) { - if (This->media_seeking) - IMediaSeeking_Release(This->media_seeking); - This->media_seeking = NULL; - if (This->media_control) - IMediaControl_Release(This->media_control); - This->media_control = NULL; - if (This->pFilterGraph) - IGraphBuilder_Release(This->pFilterGraph); - This->pFilterGraph = NULL; + if (mmstream->media_seeking) + IMediaSeeking_Release(mmstream->media_seeking); + mmstream->media_seeking = NULL; + if (mmstream->media_control) + IMediaControl_Release(mmstream->media_control); + mmstream->media_control = NULL; + if (mmstream->graph) + IGraphBuilder_Release(mmstream->graph); + mmstream->graph = NULL; }
return hr; }
+static HRESULT WINAPI multimedia_stream_Initialize(IAMMultiMediaStream *iface, + STREAM_TYPE type, DWORD flags, IGraphBuilder *graph) +{ + struct multimedia_stream *mmstream = impl_from_IAMMultiMediaStream(iface); + HRESULT hr; + + TRACE("mmstream %p, type %u, flags %#x, graph %p.\n", mmstream, type, flags, graph); + + if (FAILED(hr = create_graph(mmstream, graph))) + return hr; + + mmstream->type = type; + + return S_OK; +} + static HRESULT WINAPI multimedia_stream_GetFilterGraph(IAMMultiMediaStream *iface, IGraphBuilder **graph) { struct multimedia_stream *mmstream = impl_from_IAMMultiMediaStream(iface); @@ -259,8 +261,8 @@ static HRESULT WINAPI multimedia_stream_GetFilterGraph(IAMMultiMediaStream *ifac if (!graph) return E_POINTER;
- if (mmstream->pFilterGraph) - IGraphBuilder_AddRef(*graph = mmstream->pFilterGraph); + if (mmstream->graph) + IGraphBuilder_AddRef(*graph = mmstream->graph); else *graph = NULL;
@@ -302,6 +304,9 @@ static HRESULT WINAPI multimedia_stream_AddMediaStream(IAMMultiMediaStream *ifac return MS_E_PURPOSEID; }
+ if (!This->graph && FAILED(hr = create_graph(This, NULL))) + return hr; + if (dwFlags & AMMSF_ADDDEFAULTRENDERER) { IBaseFilter *dsound_render; @@ -318,16 +323,16 @@ static HRESULT WINAPI multimedia_stream_AddMediaStream(IAMMultiMediaStream *ifac if (SUCCEEDED(hr = CoCreateInstance(&CLSID_DSoundRender, NULL, CLSCTX_INPROC_SERVER, &IID_IBaseFilter, (void **)&dsound_render))) { - hr = IGraphBuilder_AddFilter(This->pFilterGraph, dsound_render, NULL); + hr = IGraphBuilder_AddFilter(This->graph, dsound_render, NULL); IBaseFilter_Release(dsound_render); } return hr; }
if (IsEqualGUID(PurposeId, &MSPID_PrimaryVideo)) - hr = ddraw_stream_create((IMultiMediaStream*)iface, PurposeId, stream_object, This->StreamType, &pStream); + hr = ddraw_stream_create((IMultiMediaStream*)iface, PurposeId, stream_object, This->type, &pStream); else - hr = audio_stream_create((IMultiMediaStream*)iface, PurposeId, stream_object, This->StreamType, &pStream); + hr = audio_stream_create((IMultiMediaStream*)iface, PurposeId, stream_object, This->type, &pStream);
if (SUCCEEDED(hr)) { @@ -358,11 +363,11 @@ static HRESULT WINAPI multimedia_stream_OpenFile(IAMMultiMediaStream *iface, return E_POINTER;
/* If Initialize was not called before, we do it here */ - if (!This->pFilterGraph) + if (!This->graph) ret = IAMMultiMediaStream_Initialize(iface, STREAMTYPE_READ, 0, NULL);
if (SUCCEEDED(ret)) - ret = IGraphBuilder_AddSourceFilter(This->pFilterGraph, filename, L"Source", &BaseFilter); + ret = IGraphBuilder_AddSourceFilter(This->graph, filename, L"Source", &BaseFilter);
if (SUCCEEDED(ret)) ret = IBaseFilter_EnumPins(BaseFilter, &EnumPins); @@ -378,7 +383,7 @@ static HRESULT WINAPI multimedia_stream_OpenFile(IAMMultiMediaStream *iface, }
if (SUCCEEDED(ret) && !(flags & AMMSF_NORENDER)) - ret = IGraphBuilder_Render(This->pFilterGraph, This->ipin); + ret = IGraphBuilder_Render(This->graph, This->ipin);
if (EnumPins) IEnumPins_Release(EnumPins); @@ -406,7 +411,7 @@ static HRESULT WINAPI multimedia_stream_Render(IAMMultiMediaStream *iface, DWORD if(dwFlags != AMMSF_NOCLOCK) return E_INVALIDARG;
- return IGraphBuilder_Render(This->pFilterGraph, This->ipin); + return IGraphBuilder_Render(This->graph, This->ipin); }
static const IAMMultiMediaStreamVtbl multimedia_stream_vtbl = diff --git a/dlls/amstream/tests/amstream.c b/dlls/amstream/tests/amstream.c index f9d348c6e0..511cd1eadd 100644 --- a/dlls/amstream/tests/amstream.c +++ b/dlls/amstream/tests/amstream.c @@ -705,20 +705,18 @@ static void test_add_stream(void)
hr = IAMMultiMediaStream_GetFilterGraph(mmstream, &graph); ok(hr == S_OK, "Got hr %#x.\n", hr); - todo_wine ok(!!graph, "Expected a non-NULL graph.\n"); - if (graph) - { - hr = IGraphBuilder_EnumFilters(graph, &enum_filters); - ok(hr == S_OK, "Got hr %#x.\n", hr); - hr = IEnumFilters_Next(enum_filters, 3, filters, &count); - ok(hr == S_FALSE, "Got hr %#x.\n", hr); - ok(count == 1, "Got count %u.\n", count); - ok(filters[0] == (IBaseFilter *)stream_filter, - "Expected filter %p, got %p.\n", stream_filter, filters[0]); - IBaseFilter_Release(filters[0]); - IEnumFilters_Release(enum_filters); - IGraphBuilder_Release(graph); - } + ok(!!graph, "Expected a non-NULL graph.\n"); + + hr = IGraphBuilder_EnumFilters(graph, &enum_filters); + ok(hr == S_OK, "Got hr %#x.\n", hr); + hr = IEnumFilters_Next(enum_filters, 3, filters, &count); + ok(hr == S_FALSE, "Got hr %#x.\n", hr); + ok(count == 1, "Got count %u.\n", count); + ok(filters[0] == (IBaseFilter *)stream_filter, + "Expected filter %p, got %p.\n", stream_filter, filters[0]); + IBaseFilter_Release(filters[0]); + IEnumFilters_Release(enum_filters); + IGraphBuilder_Release(graph);
IMediaStreamFilter_Release(stream_filter); ref = IAMMultiMediaStream_Release(mmstream); @@ -792,10 +790,6 @@ static void test_add_stream(void) hr = IAMMultiMediaStream_GetFilter(mmstream, &stream_filter); ok(hr == S_OK, "Got hr %#x.\n", hr);
- /* FIXME: This call should not be necessary. */ - hr = IAMMultiMediaStream_Initialize(mmstream, STREAMTYPE_READ, 0, NULL); - ok(hr == S_OK, "Got hr %#x.\n", hr); - hr = IAMMultiMediaStream_AddMediaStream(mmstream, NULL, &MSPID_PrimaryVideo, AMMSF_ADDDEFAULTRENDERER, &video_stream); ok(hr == E_INVALIDARG, "Got hr %#x.\n", hr); @@ -1520,6 +1514,12 @@ static void test_initialize(void) todo_wine ok(type == STREAMTYPE_READ, "Got type %u.\n", type); IMediaStream_Release(stream);
+ ret_graph = NULL; + hr = IAMMultiMediaStream_GetFilterGraph(mmstream, &ret_graph); + ok(hr == S_OK, "Got hr %#x.\n", hr); + ok(!!ret_graph, "Got unexpected graph %p.\n", ret_graph); + IGraphBuilder_Release(ret_graph); + ref = IAMMultiMediaStream_Release(mmstream); ok(!ref, "Got outstanding refcount %d.\n", ref);
@@ -1589,6 +1589,12 @@ static void test_initialize(void) ok(hr == S_OK, "Got hr %#x.\n", hr); ok(type == STREAMTYPE_READ, "Got type %u.\n", type);
+ ret_graph = NULL; + hr = IAMMultiMediaStream_GetFilterGraph(mmstream, &ret_graph); + ok(hr == S_OK, "Got hr %#x.\n", hr); + ok(!!ret_graph, "Got unexpected graph %p.\n", ret_graph); + IGraphBuilder_Release(ret_graph); + hr = IAMMultiMediaStream_Initialize(mmstream, STREAMTYPE_TRANSFORM, 0, NULL); ok(hr == S_OK, "Got hr %#x.\n", hr);
@@ -1658,10 +1664,6 @@ static void test_enum_media_types(void) HRESULT hr; IPin *pin;
- /* FIXME: This call should not be necessary. */ - hr = IAMMultiMediaStream_Initialize(mmstream, STREAMTYPE_READ, 0, NULL); - ok(hr == S_OK, "Got hr %#x.\n", hr); - hr = IAMMultiMediaStream_AddMediaStream(mmstream, NULL, &MSPID_PrimaryVideo, 0, &stream); ok(hr == S_OK, "Got hr %#x.\n", hr); hr = IMediaStream_QueryInterface(stream, &IID_IPin, (void **)&pin); @@ -2377,9 +2379,6 @@ static void test_audiostream_get_format(void) ULONG ref; IPin *pin;
- hr = IAMMultiMediaStream_Initialize(mmstream, STREAMTYPE_READ, 0, NULL); - ok(hr == S_OK, "Got hr %#x.\n", hr); - hr = IAMMultiMediaStream_AddMediaStream(mmstream, NULL, &MSPID_PrimaryAudio, 0, &stream); ok(hr == S_OK, "Got hr %#x.\n", hr); hr = IMediaStream_QueryInterface(stream, &IID_IAudioMediaStream, (void **)&audio_stream); @@ -2570,9 +2569,6 @@ static void test_audiostream_set_format(void)
mmstream = create_ammultimediastream();
- hr = IAMMultiMediaStream_Initialize(mmstream, STREAMTYPE_READ, 0, NULL); - ok(hr == S_OK, "Got hr %#x.\n", hr); - hr = IAMMultiMediaStream_AddMediaStream(mmstream, NULL, &MSPID_PrimaryAudio, 0, &stream); ok(hr == S_OK, "Got hr %#x.\n", hr); hr = IMediaStream_QueryInterface(stream, &IID_IAudioMediaStream, (void **)&audio_stream); @@ -2653,8 +2649,6 @@ static void test_audiostream_receive_connection(void) IPin *pin;
mmstream = create_ammultimediastream(); - hr = IAMMultiMediaStream_Initialize(mmstream, STREAMTYPE_READ, 0, NULL); - ok(hr == S_OK, "Got hr %#x.\n", hr); hr = IAMMultiMediaStream_AddMediaStream(mmstream, NULL, &MSPID_PrimaryAudio, 0, &stream); ok(hr == S_OK, "Got hr %#x.\n", hr); hr = IMediaStream_QueryInterface(stream, &IID_IAudioMediaStream, (void **)&audio_stream);