Commit 16579ffb authored by Alfred Agrell's avatar Alfred Agrell Committed by Alexandre Julliard

quartz/tests: Test that compressed formats are offered for MPEGs.

parent e870f31d
......@@ -817,6 +817,7 @@ static void test_render_with_multithread(void)
struct testpin
{
IPin IPin_iface;
IMemInputPin IMemInputPin_iface;
LONG ref;
PIN_DIRECTION dir;
struct testfilter *filter;
......@@ -836,6 +837,10 @@ struct testpin
HRESULT Connect_hr;
HRESULT EnumMediaTypes_hr;
HRESULT QueryInternalConnections_hr;
BYTE input[64];
ULONG input_size;
HANDLE on_input_full;
};
struct testfilter
......@@ -870,7 +875,6 @@ struct testfilter
IReferenceClock IReferenceClock_iface;
IFileSourceFilter IFileSourceFilter_iface;
WCHAR filename[MAX_PATH];
};
static inline struct testpin *impl_from_IEnumMediaTypes(IEnumMediaTypes *iface)
......@@ -955,6 +959,11 @@ static inline struct testpin *impl_from_IPin(IPin *iface)
return CONTAINING_RECORD(iface, struct testpin, IPin_iface);
}
static inline struct testpin *impl_from_IMemInputPin(IMemInputPin *iface)
{
return CONTAINING_RECORD(iface, struct testpin, IMemInputPin_iface);
}
static HRESULT WINAPI testpin_QueryInterface(IPin *iface, REFIID iid, void **out)
{
struct testpin *pin = impl_from_IPin(iface);
......@@ -966,6 +975,12 @@ static HRESULT WINAPI testpin_QueryInterface(IPin *iface, REFIID iid, void **out
IPin_AddRef(*out);
return S_OK;
}
if (pin->IMemInputPin_iface.lpVtbl && IsEqualGUID(iid, &IID_IMemInputPin))
{
*out = &pin->IMemInputPin_iface;
IMemInputPin_AddRef(*out);
return S_OK;
}
*out = NULL;
return E_NOINTERFACE;
......@@ -1045,7 +1060,7 @@ static HRESULT WINAPI testpin_QueryId(IPin *iface, WCHAR **id)
{
struct testpin *pin = impl_from_IPin(iface);
if (winetest_debug > 1) trace("%p->QueryId()\n", iface);
*id = CoTaskMemAlloc(11);
*id = CoTaskMemAlloc(sizeof(WCHAR)*11);
wcscpy(*id, pin->id);
return S_OK;
}
......@@ -1169,6 +1184,8 @@ static void testpin_init(struct testpin *pin, const IPinVtbl *vtbl, PIN_DIRECTIO
pin->Connect_hr = S_OK;
pin->EnumMediaTypes_hr = S_OK;
pin->QueryInternalConnections_hr = E_NOTIMPL;
wsprintfW(pin->name, L"name%.4x", 0xFFFF&(intptr_t)pin);
wsprintfW(pin->id, L"id%.4x", 0xFFFF&(intptr_t)pin);
}
static void testsink_init(struct testpin *pin)
......@@ -1476,8 +1493,20 @@ static HRESULT WINAPI testfilter_EnumPins(IBaseFilter *iface, IEnumPins **out)
static HRESULT WINAPI testfilter_FindPin(IBaseFilter *iface, const WCHAR *id, IPin **pin)
{
ok(0, "Unexpected call.\n");
return E_NOTIMPL;
struct testfilter *filter = impl_from_IBaseFilter(iface);
int i;
if (winetest_debug > 1) trace("%p->FindPin(%ls)\n", filter, id);
for (i = 0; i < filter->pin_count; ++i)
{
if (!wcscmp(id, filter->pins[i].id))
{
*pin = &filter->pins[i].IPin_iface;
IPin_AddRef(*pin);
return S_OK;
}
}
return VFW_E_NOT_FOUND;
}
static HRESULT WINAPI testfilter_QueryFilterInfo(IBaseFilter *iface, FILTER_INFO *info)
......@@ -1970,11 +1999,7 @@ static ULONG WINAPI testfilesource_Release(IFileSourceFilter *iface)
static HRESULT WINAPI testfilesource_Load(IFileSourceFilter *iface,
const WCHAR *filename, const AM_MEDIA_TYPE *mt)
{
struct testfilter *filter = impl_from_IFileSourceFilter(iface);
if (winetest_debug > 1) trace("%p->Load()\n", iface);
wcscpy(filter->filename, filename);
if (winetest_debug > 1) trace("%p->Load(%ls)\n", iface, filename);
return S_OK;
}
......@@ -4193,6 +4218,182 @@ static void test_ec_complete(void)
ok(filter3.ref == 1, "Got outstanding refcount %ld.\n", filter3.ref);
}
static HRESULT WINAPI mpegtestsink_ReceiveConnection(IPin *iface, IPin *peer, const AM_MEDIA_TYPE *mt)
{
struct testpin *pin = impl_from_IPin(iface);
if (winetest_debug > 1) trace("%p->ReceiveConnection(%p)\n", pin, peer);
if (!IsEqualGUID(&mt->majortype, &MEDIATYPE_Video))
return VFW_E_TYPE_NOT_ACCEPTED;
if (!IsEqualGUID(&mt->subtype, &MEDIASUBTYPE_MPEG1Payload))
return VFW_E_TYPE_NOT_ACCEPTED;
if (!IsEqualGUID(&mt->formattype, &FORMAT_MPEGVideo))
return VFW_E_TYPE_NOT_ACCEPTED;
pin->peer = peer;
IPin_AddRef(peer);
return S_OK;
}
static HRESULT WINAPI mpegtestpin_EndOfStream(IPin *iface)
{
return S_OK;
}
static HRESULT WINAPI mpegtestpin_NewSegment(IPin *iface, REFERENCE_TIME start, REFERENCE_TIME stop, double rate)
{
return S_OK;
}
static const IPinVtbl mpegtestsink_vtbl =
{
testpin_QueryInterface,
testpin_AddRef,
testpin_Release,
no_Connect,
mpegtestsink_ReceiveConnection,
testpin_Disconnect,
testpin_ConnectedTo,
testpin_ConnectionMediaType,
testpin_QueryPinInfo,
testpin_QueryDirection,
testpin_QueryId,
testpin_QueryAccept,
testpin_EnumMediaTypes,
testpin_QueryInternalConnections,
mpegtestpin_EndOfStream,
testpin_BeginFlush,
testpin_EndFlush,
mpegtestpin_NewSegment
};
static HRESULT WINAPI meminput_QueryInterface(IMemInputPin *iface, REFIID iid, void **out)
{
struct testpin *pin = impl_from_IMemInputPin(iface);
if (winetest_debug > 1) trace("%p->QueryInterface(%s)\n", pin, wine_dbgstr_guid(iid));
*out = NULL;
return E_NOINTERFACE;
}
static ULONG WINAPI meminput_AddRef(IMemInputPin *iface)
{
struct testpin *pin = impl_from_IMemInputPin(iface);
return InterlockedIncrement(&pin->ref);
}
static ULONG WINAPI meminput_Release(IMemInputPin *iface)
{
struct testpin *pin = impl_from_IMemInputPin(iface);
return InterlockedDecrement(&pin->ref);
}
static HRESULT STDMETHODCALLTYPE meminput_GetAllocator(IMemInputPin *iface, IMemAllocator **allocator)
{
return VFW_E_NO_ALLOCATOR;
}
static HRESULT STDMETHODCALLTYPE meminput_NotifyAllocator(IMemInputPin *iface, IMemAllocator *allocator, BOOL readonly)
{
return S_OK;
}
static HRESULT STDMETHODCALLTYPE meminput_GetAllocatorRequirements(IMemInputPin *iface, ALLOCATOR_PROPERTIES *props)
{
return E_NOTIMPL;
}
static HRESULT STDMETHODCALLTYPE meminput_Receive(IMemInputPin *iface, IMediaSample *sample)
{
struct testpin *pin = impl_from_IMemInputPin(iface);
size_t new_bytes;
HRESULT hr;
BYTE *ptr;
long len;
len = IMediaSample_GetActualDataLength(sample);
hr = IMediaSample_GetPointer(sample, &ptr);
ok(hr == S_OK, "Got hr %#lx.\n", hr);
ok(len > 0, "Got %ld.\n", len);
new_bytes = min(sizeof(pin->input) - pin->input_size, len);
if (new_bytes)
{
memcpy(pin->input + pin->input_size, ptr, new_bytes);
pin->input_size += new_bytes;
if (pin->input_size == sizeof(pin->input))
SetEvent(pin->on_input_full);
}
return S_OK;
}
static HRESULT STDMETHODCALLTYPE meminput_ReceiveMultiple(IMemInputPin *iface, IMediaSample **samples, LONG n_samples, LONG *n_samples_processed)
{
*n_samples_processed = 1;
return IMemInputPin_Receive(iface, samples[0]);
}
static HRESULT STDMETHODCALLTYPE meminput_ReceiveCanBlock(IMemInputPin *iface) { return S_OK; }
static const IMemInputPinVtbl mpegtestsink_meminput_vtbl =
{
meminput_QueryInterface,
meminput_AddRef,
meminput_Release,
meminput_GetAllocator,
meminput_NotifyAllocator,
meminput_GetAllocatorRequirements,
meminput_Receive,
meminput_ReceiveMultiple,
meminput_ReceiveCanBlock
};
static void test_renderfile_compressed(void)
{
WCHAR *filename = load_resource(L"test.mpg");
struct testpin sink_pin;
struct testfilter sink;
IMediaControl *control;
IFilterGraph2 *graph;
HRESULT hr;
DWORD ret;
graph = create_graph();
testpin_init(&sink_pin, &mpegtestsink_vtbl, PINDIR_INPUT);
sink_pin.IMemInputPin_iface.lpVtbl = &mpegtestsink_meminput_vtbl;
testfilter_init(&sink, &sink_pin, 1);
hr = IFilterGraph2_AddFilter(graph, &sink.IBaseFilter_iface, L"sink");
ok(hr == S_OK, "Got hr %#lx.\n", hr);
hr = IFilterGraph2_RenderFile(graph, filename, NULL);
ok(hr == S_OK || hr == VFW_S_AUDIO_NOT_RENDERED, "Got hr %#lx.\n", hr);
ok(sink_pin.peer != NULL, "Expected connection.\n");
sink_pin.on_input_full = CreateEventW(NULL, FALSE, FALSE, NULL);
hr = IFilterGraph2_QueryInterface(graph, &IID_IMediaControl, (void **)&control);
ok(hr == S_OK, "Got hr %#lx.\n", hr);
hr = IMediaControl_Pause(control);
ok(SUCCEEDED(hr), "Got hr %#lx.\n", hr);
ret = WaitForSingleObject(sink_pin.on_input_full, 1000);
ok(ret == WAIT_OBJECT_0, "Got %#lx.\n", ret);
ok(sink_pin.input[0] == 0x00, "Expected MPEG sequence header.\n");
ok(sink_pin.input[1] == 0x00, "Expected MPEG sequence header.\n");
ok(sink_pin.input[2] == 0x01, "Expected MPEG sequence header.\n");
ok(sink_pin.input[3] == 0xB3, "Expected MPEG sequence header.\n");
IMediaControl_Stop(control);
IFilterGraph2_Release(graph);
DeleteFileW(filename);
}
static void test_renderfile_failure(void)
{
static const char bogus_data[20] = {0xde, 0xad, 0xbe, 0xef};
......@@ -5751,6 +5952,7 @@ START_TEST(filtergraph)
test_sync_source();
test_filter_state();
test_ec_complete();
test_renderfile_compressed();
test_renderfile_failure();
test_graph_seeking();
test_default_sync_source();
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment