system/wine
system/wine
1
0
Fork 0
mirror of https://github.com/wine-mirror/wine synced 2024-07-08 03:45:57 +00:00
Code Issues Projects Releases Wiki Activity

evr/filter: Initial implementation of sample rendering.

Browse Source 
Signed-off-by: Nikolay Sivov <nsivov@codeweavers.com>
This commit is contained in:
Nikolay Sivov 2022-09-06 17:09:51 +03:00 committed by Alexandre Julliard
parent 97b83a6dec
commit a1d2e52468
1 changed files with 141 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

144
dlls/evr/evr.c
View File

@ -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)