evr/filter: Implement media type testing for QueryAccept().

Signed-off-by: Nikolay Sivov <nsivov@codeweavers.com>
This commit is contained in:
Nikolay Sivov 2022-08-29 15:36:47 +03:00 committed by Alexandre Julliard
parent 4d7b3c7861
commit f143af50a3
2 changed files with 213 additions and 4 deletions

View file

@ -25,12 +25,20 @@
#include "evr_private.h"
#include "d3d9.h"
#include "mferror.h"
#include "mfapi.h"
#include "initguid.h"
#include "dxva2api.h"
WINE_DEFAULT_DEBUG_CHANNEL(evr);
enum evr_flags
{
EVR_INIT_SERVICES = 0x1, /* Currently in InitServices() call. */
EVR_MIXER_INITED_SERVICES = 0x2,
EVR_PRESENTER_INITED_SERVICES = 0x4,
};
struct evr
{
struct strmbase_renderer renderer;
@ -43,6 +51,7 @@ struct evr
IMFTransform *mixer;
IMFVideoPresenter *presenter;
unsigned int flags;
};
static void evr_uninitialize(struct evr *filter)
@ -125,6 +134,96 @@ static HRESULT evr_query_interface(struct strmbase_renderer *iface, REFIID iid,
return S_OK;
}
static BOOL evr_is_mixer_d3d_aware(const struct evr *filter)
{
IMFAttributes *attributes;
unsigned int value = 0;
BOOL ret;
if (FAILED(IMFTransform_QueryInterface(filter->mixer, &IID_IMFAttributes, (void **)&attributes)))
return FALSE;
ret = SUCCEEDED(IMFAttributes_GetUINT32(attributes, &MF_SA_D3D_AWARE, &value)) && value;
IMFAttributes_Release(attributes);
return ret;
}
static HRESULT evr_init_services(struct evr *filter)
{
IMFTopologyServiceLookupClient *lookup_client;
HRESULT hr;
if (SUCCEEDED(hr = IMFTransform_QueryInterface(filter->mixer, &IID_IMFTopologyServiceLookupClient,
(void **)&lookup_client)))
{
filter->flags |= EVR_INIT_SERVICES;
if (SUCCEEDED(hr = IMFTopologyServiceLookupClient_InitServicePointers(lookup_client,
&filter->IMFTopologyServiceLookup_iface)))
{
filter->flags |= EVR_MIXER_INITED_SERVICES;
}
filter->flags &= ~EVR_INIT_SERVICES;
IMFTopologyServiceLookupClient_Release(lookup_client);
}
if (FAILED(hr)) return hr;
/* Set device manager that presenter should have created. */
if (evr_is_mixer_d3d_aware(filter))
{
IUnknown *device_manager;
IMFGetService *gs;
if (SUCCEEDED(IUnknown_QueryInterface(filter->presenter, &IID_IMFGetService, (void **)&gs)))
{
if (SUCCEEDED(IMFGetService_GetService(gs, &MR_VIDEO_RENDER_SERVICE, &IID_IDirect3DDeviceManager9,
(void **)&device_manager)))
{
IMFTransform_ProcessMessage(filter->mixer, MFT_MESSAGE_SET_D3D_MANAGER, (ULONG_PTR)device_manager);
IUnknown_Release(device_manager);
}
IMFGetService_Release(gs);
}
}
if (SUCCEEDED(hr = IMFVideoPresenter_QueryInterface(filter->presenter, &IID_IMFTopologyServiceLookupClient,
(void **)&lookup_client)))
{
filter->flags |= EVR_INIT_SERVICES;
if (SUCCEEDED(hr = IMFTopologyServiceLookupClient_InitServicePointers(lookup_client,
&filter->IMFTopologyServiceLookup_iface)))
{
filter->flags |= EVR_PRESENTER_INITED_SERVICES;
}
filter->flags &= ~EVR_INIT_SERVICES;
IMFTopologyServiceLookupClient_Release(lookup_client);
}
return hr;
}
static void evr_release_services(struct evr *filter)
{
IMFTopologyServiceLookupClient *lookup_client;
if (filter->flags & EVR_MIXER_INITED_SERVICES && SUCCEEDED(IMFTransform_QueryInterface(filter->mixer,
&IID_IMFTopologyServiceLookupClient, (void **)&lookup_client)))
{
IMFTopologyServiceLookupClient_ReleaseServicePointers(lookup_client);
IMFTopologyServiceLookupClient_Release(lookup_client);
filter->flags &= ~EVR_MIXER_INITED_SERVICES;
}
if (filter->flags & EVR_PRESENTER_INITED_SERVICES && SUCCEEDED(IMFVideoPresenter_QueryInterface(filter->presenter,
&IID_IMFTopologyServiceLookupClient, (void **)&lookup_client)))
{
IMFTopologyServiceLookupClient_ReleaseServicePointers(lookup_client);
IMFTopologyServiceLookupClient_Release(lookup_client);
filter->flags &= ~EVR_PRESENTER_INITED_SERVICES;
}
}
static void evr_destroy(struct strmbase_renderer *iface)
{
struct evr *filter = impl_from_strmbase_renderer(iface);
@ -142,8 +241,37 @@ static HRESULT evr_render(struct strmbase_renderer *iface, IMediaSample *sample)
static HRESULT evr_query_accept(struct strmbase_renderer *iface, const AM_MEDIA_TYPE *mt)
{
FIXME("Not implemented.\n");
return E_NOTIMPL;
struct evr *filter = impl_from_strmbase_renderer(iface);
IMFMediaType *media_type;
HRESULT hr = S_OK;
EnterCriticalSection(&filter->renderer.filter.filter_cs);
if (!filter->presenter)
hr = evr_initialize(filter, NULL, NULL);
if (SUCCEEDED(hr))
hr = evr_init_services(filter);
if (SUCCEEDED(hr))
hr = MFCreateMediaType(&media_type);
if (SUCCEEDED(hr))
{
if (SUCCEEDED(hr = MFInitMediaTypeFromAMMediaType(media_type, mt)))
{
/* TODO: some pin -> mixer input mapping is necessary to test the substreams. */
hr = IMFTransform_SetInputType(filter->mixer, 0, media_type, MFT_SET_TYPE_TEST_ONLY);
}
IMFMediaType_Release(media_type);
evr_release_services(filter);
}
LeaveCriticalSection(&filter->renderer.filter.filter_cs);
return hr;
}
static const struct strmbase_renderer_ops renderer_ops =
@ -419,10 +547,37 @@ static ULONG WINAPI filter_service_lookup_Release(IMFTopologyServiceLookup *ifac
static HRESULT WINAPI filter_service_lookup_LookupService(IMFTopologyServiceLookup *iface, MF_SERVICE_LOOKUP_TYPE lookup_type,
DWORD index, REFGUID service, REFIID riid, void **objects, DWORD *num_objects)
{
FIXME("iface %p, lookup_type %d, index %lu, service %s, riid %s, objects %p, num_objects %p.\n",
struct evr *filter = impl_from_IMFTopologyServiceLookup(iface);
HRESULT hr = S_OK;
TRACE("iface %p, lookup_type %d, index %lu, service %s, riid %s, objects %p, num_objects %p.\n",
iface, lookup_type, index, debugstr_guid(service), debugstr_guid(riid), objects, num_objects);
return MF_E_NOTACCEPTING;
EnterCriticalSection(&filter->renderer.filter.filter_cs);
if (!(filter->flags & EVR_INIT_SERVICES))
hr = MF_E_NOTACCEPTING;
else if (IsEqualGUID(service, &MR_VIDEO_RENDER_SERVICE))
{
if (IsEqualIID(riid, &IID_IMediaEventSink))
{
*objects = &filter->IMediaEventSink_iface;
IUnknown_AddRef((IUnknown *)*objects);
}
}
else if (IsEqualGUID(service, &MR_VIDEO_MIXER_SERVICE))
{
hr = IMFTransform_QueryInterface(filter->mixer, riid, objects);
}
else
{
WARN("Unsupported service %s.\n", debugstr_guid(service));
hr = MF_E_UNSUPPORTED_SERVICE;
}
LeaveCriticalSection(&filter->renderer.filter.filter_cs);
return hr;
}
static const IMFTopologyServiceLookupVtbl filter_service_lookup_vtbl =

View file

@ -637,6 +637,59 @@ static void test_service_lookup(void)
ok(!ref, "Got outstanding refcount %ld.\n", ref);
}
static void test_query_accept(void)
{
IBaseFilter *filter = create_evr();
AM_MEDIA_TYPE req_mt = {{0}};
VIDEOINFOHEADER vih =
{
{0}, {0}, 0, 0, 0,
{sizeof(BITMAPINFOHEADER), 32, 24, 1, 0, 0xdeadbeef}
};
unsigned int i;
HRESULT hr;
ULONG ref;
IPin *pin;
static const GUID *subtype_tests[] =
{
&MEDIASUBTYPE_RGB32,
&MEDIASUBTYPE_YUY2,
};
static const GUID *unsupported_subtype_tests[] =
{
&MEDIASUBTYPE_RGB8,
};
IBaseFilter_FindPin(filter, L"EVR Input0", &pin);
req_mt.majortype = MEDIATYPE_Video;
req_mt.formattype = FORMAT_VideoInfo;
req_mt.cbFormat = sizeof(VIDEOINFOHEADER);
req_mt.pbFormat = (BYTE *)&vih;
for (i = 0; i < ARRAY_SIZE(subtype_tests); ++i)
{
memcpy(&req_mt.subtype, subtype_tests[i], sizeof(GUID));
hr = IPin_QueryAccept(pin, &req_mt);
todo_wine_if(i)
ok(hr == S_OK, "Got hr %#lx.\n", hr);
}
for (i = 0; i < ARRAY_SIZE(unsupported_subtype_tests); ++i)
{
memcpy(&req_mt.subtype, unsupported_subtype_tests[i], sizeof(GUID));
hr = IPin_QueryAccept(pin, &req_mt);
ok(hr == S_FALSE, "Got hr %#lx.\n", hr);
}
IPin_Release(pin);
ref = IBaseFilter_Release(filter);
ok(!ref, "Got outstanding refcount %ld.\n", ref);
}
static IMFMediaType * create_video_type(const GUID *subtype)
{
IMFMediaType *video_type;
@ -3311,6 +3364,7 @@ START_TEST(evr)
test_misc_flags();
test_display_control();
test_service_lookup();
test_query_accept();
test_default_mixer();
test_default_mixer_type_negotiation();