mirror of
git://source.winehq.org/git/wine.git
synced 2024-10-06 08:54:05 +00:00
evr/filter: Initial implementation of sample rendering.
Signed-off-by: Nikolay Sivov <nsivov@codeweavers.com>
This commit is contained in:
parent
97b83a6dec
commit
a1d2e52468
144
dlls/evr/evr.c
144
dlls/evr/evr.c
|
@ -51,6 +51,8 @@ struct evr
|
|||
|
||||
IMFTransform *mixer;
|
||||
IMFVideoPresenter *presenter;
|
||||
IMFVideoSampleAllocator *allocator;
|
||||
IMFMediaType *media_type;
|
||||
unsigned int flags;
|
||||
};
|
||||
|
||||
|
@ -148,6 +150,20 @@ static BOOL evr_is_mixer_d3d_aware(const struct evr *filter)
|
|||
return ret;
|
||||
}
|
||||
|
||||
static HRESULT evr_get_service(void *unk, REFGUID service, REFIID riid, void **obj)
|
||||
{
|
||||
IMFGetService *gs;
|
||||
HRESULT hr;
|
||||
|
||||
if (SUCCEEDED(hr = IUnknown_QueryInterface((IUnknown *)unk, &IID_IMFGetService, (void **)&gs)))
|
||||
{
|
||||
hr = IMFGetService_GetService(gs, service, riid, obj);
|
||||
IMFGetService_Release(gs);
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
static HRESULT evr_init_services(struct evr *filter)
|
||||
{
|
||||
IMFTopologyServiceLookupClient *lookup_client;
|
||||
|
@ -224,6 +240,16 @@ static void evr_release_services(struct evr *filter)
|
|||
}
|
||||
}
|
||||
|
||||
static void evr_set_input_type(struct evr *filter, IMFMediaType *media_type)
|
||||
{
|
||||
if (filter->media_type)
|
||||
IMFMediaType_Release(filter->media_type);
|
||||
if ((filter->media_type = media_type))
|
||||
IMFMediaType_AddRef(filter->media_type);
|
||||
if (!media_type && filter->allocator)
|
||||
IMFVideoSampleAllocator_UninitializeSampleAllocator(filter->allocator);
|
||||
}
|
||||
|
||||
static HRESULT evr_test_input_type(struct evr *filter, const AM_MEDIA_TYPE *mt, IMFMediaType **ret)
|
||||
{
|
||||
IMFMediaType *media_type;
|
||||
|
@ -259,7 +285,9 @@ static HRESULT evr_test_input_type(struct evr *filter, const AM_MEDIA_TYPE *mt,
|
|||
static HRESULT evr_connect(struct strmbase_renderer *iface, const AM_MEDIA_TYPE *mt)
|
||||
{
|
||||
struct evr *filter = impl_from_strmbase_renderer(iface);
|
||||
IMFVideoSampleAllocator *allocator;
|
||||
IMFMediaType *media_type;
|
||||
IUnknown *device_manager;
|
||||
HRESULT hr;
|
||||
|
||||
if (SUCCEEDED(hr = evr_test_input_type(filter, mt, &media_type)))
|
||||
|
@ -267,6 +295,27 @@ static HRESULT evr_connect(struct strmbase_renderer *iface, const AM_MEDIA_TYPE
|
|||
if (SUCCEEDED(hr = IMFTransform_SetInputType(filter->mixer, 0, media_type, 0)))
|
||||
hr = IMFVideoPresenter_ProcessMessage(filter->presenter, MFVP_MESSAGE_INVALIDATEMEDIATYPE, 0);
|
||||
|
||||
if (SUCCEEDED(hr = MFCreateVideoSampleAllocator(&IID_IMFVideoSampleAllocator, (void **)&allocator)))
|
||||
{
|
||||
if (SUCCEEDED(hr = evr_get_service(filter->presenter, &MR_VIDEO_RENDER_SERVICE,
|
||||
&IID_IDirect3DDeviceManager9, (void **)&device_manager)))
|
||||
{
|
||||
if (SUCCEEDED(hr = IMFVideoSampleAllocator_SetDirectXManager(allocator, device_manager)))
|
||||
{
|
||||
if (SUCCEEDED(hr = IMFVideoSampleAllocator_InitializeSampleAllocator(allocator, 2, media_type)))
|
||||
{
|
||||
IMFVideoSampleAllocator_AddRef((filter->allocator = allocator));
|
||||
}
|
||||
}
|
||||
IUnknown_Release(device_manager);
|
||||
}
|
||||
|
||||
IMFVideoSampleAllocator_Release(allocator);
|
||||
}
|
||||
|
||||
if (SUCCEEDED(hr))
|
||||
evr_set_input_type(filter, media_type);
|
||||
|
||||
IMFMediaType_Release(media_type);
|
||||
}
|
||||
|
||||
|
@ -279,6 +328,7 @@ static void evr_disconnect(struct strmbase_renderer *iface)
|
|||
|
||||
if (filter->mixer)
|
||||
IMFTransform_SetInputType(filter->mixer, 0, NULL, 0);
|
||||
evr_set_input_type(filter, NULL);
|
||||
evr_release_services(filter);
|
||||
}
|
||||
|
||||
|
@ -287,14 +337,102 @@ static void evr_destroy(struct strmbase_renderer *iface)
|
|||
struct evr *filter = impl_from_strmbase_renderer(iface);
|
||||
|
||||
evr_uninitialize(filter);
|
||||
evr_set_input_type(filter, NULL);
|
||||
if (filter->allocator)
|
||||
IMFVideoSampleAllocator_Release(filter->allocator);
|
||||
strmbase_renderer_cleanup(&filter->renderer);
|
||||
free(filter);
|
||||
}
|
||||
|
||||
static HRESULT evr_render(struct strmbase_renderer *iface, IMediaSample *sample)
|
||||
static HRESULT evr_copy_sample_buffer(struct evr *filter, IMediaSample *input_sample, IMFSample **sample)
|
||||
{
|
||||
FIXME("Not implemented.\n");
|
||||
return E_NOTIMPL;
|
||||
IDirect3DSurface9 *surface;
|
||||
D3DLOCKED_RECT locked_rect;
|
||||
IMFMediaBuffer *buffer;
|
||||
UINT64 frame_size = 0;
|
||||
UINT32 width, lines;
|
||||
LONG src_stride;
|
||||
HRESULT hr;
|
||||
BYTE *src;
|
||||
|
||||
if (FAILED(hr = IMFMediaType_GetUINT32(filter->media_type, &MF_MT_DEFAULT_STRIDE, (UINT32 *)&src_stride)))
|
||||
{
|
||||
WARN("Unknown input buffer stride.\n");
|
||||
return hr;
|
||||
}
|
||||
IMFMediaType_GetUINT64(filter->media_type, &MF_MT_FRAME_SIZE, &frame_size);
|
||||
width = frame_size >> 32;
|
||||
lines = frame_size;
|
||||
|
||||
if (FAILED(hr = IMediaSample_GetPointer(input_sample, &src)))
|
||||
{
|
||||
WARN("Failed to get pointer to sample data, hr %#lx.\n", hr);
|
||||
return hr;
|
||||
}
|
||||
|
||||
if (FAILED(hr = IMFVideoSampleAllocator_AllocateSample(filter->allocator, sample)))
|
||||
{
|
||||
WARN("Failed to allocate a sample, hr %#lx.\n", hr);
|
||||
return hr;
|
||||
}
|
||||
|
||||
if (SUCCEEDED(hr = IMFSample_GetBufferByIndex(*sample, 0, &buffer)))
|
||||
{
|
||||
if (SUCCEEDED(hr = evr_get_service(buffer, &MR_BUFFER_SERVICE, &IID_IDirect3DSurface9, (void **)&surface)))
|
||||
{
|
||||
if (SUCCEEDED(hr = IDirect3DSurface9_LockRect(surface, &locked_rect, NULL, D3DLOCK_DISCARD)))
|
||||
{
|
||||
if (src_stride < 0) src -= src_stride * (lines - 1);
|
||||
MFCopyImage(locked_rect.pBits, locked_rect.Pitch, src, src_stride, width * 4, lines);
|
||||
IDirect3DSurface9_UnlockRect(surface);
|
||||
}
|
||||
|
||||
IDirect3DSurface9_Release(surface);
|
||||
}
|
||||
IMFMediaBuffer_Release(buffer);
|
||||
}
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
IMFSample_Release(*sample);
|
||||
*sample = NULL;
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
static HRESULT evr_render(struct strmbase_renderer *iface, IMediaSample *input_sample)
|
||||
{
|
||||
struct evr *filter = impl_from_strmbase_renderer(iface);
|
||||
HRESULT hr = E_NOTIMPL;
|
||||
GUID subtype = { 0 };
|
||||
IMFSample *sample;
|
||||
|
||||
if (!filter->media_type)
|
||||
{
|
||||
WARN("Media type wasn't set.\n");
|
||||
return E_UNEXPECTED;
|
||||
}
|
||||
|
||||
IMFMediaType_GetGUID(filter->media_type, &MF_MT_SUBTYPE, &subtype);
|
||||
|
||||
if (IsEqualGUID(&subtype, &MFVideoFormat_ARGB32)
|
||||
|| IsEqualGUID(&subtype, &MFVideoFormat_RGB32))
|
||||
{
|
||||
if (SUCCEEDED(hr = evr_copy_sample_buffer(filter, input_sample, &sample)))
|
||||
{
|
||||
if (SUCCEEDED(IMFTransform_ProcessInput(filter->mixer, 0, sample, 0)))
|
||||
IMFVideoPresenter_ProcessMessage(filter->presenter, MFVP_MESSAGE_PROCESSINPUTNOTIFY, 0);
|
||||
|
||||
IMFSample_Release(sample);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
FIXME("Unhandled input type %s.\n", debugstr_guid(&subtype));
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
static HRESULT evr_query_accept(struct strmbase_renderer *iface, const AM_MEDIA_TYPE *mt)
|
||||
|
|
Loading…
Reference in a new issue