561 lines
16 KiB
C
561 lines
16 KiB
C
/*
|
|
* Copyright 2020 Nikolay Sivov
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#define COBJMACROS
|
|
|
|
#include "wine/debug.h"
|
|
#include "evr.h"
|
|
#include "d3d9.h"
|
|
#include "mferror.h"
|
|
|
|
#include "evr_classes.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(evr);
|
|
|
|
#define MAX_MIXER_INPUT_STREAMS 16
|
|
|
|
struct input_stream
|
|
{
|
|
unsigned int id;
|
|
};
|
|
|
|
struct video_mixer
|
|
{
|
|
IMFTransform IMFTransform_iface;
|
|
IMFVideoDeviceID IMFVideoDeviceID_iface;
|
|
IMFTopologyServiceLookupClient IMFTopologyServiceLookupClient_iface;
|
|
LONG refcount;
|
|
|
|
struct input_stream inputs[MAX_MIXER_INPUT_STREAMS];
|
|
unsigned int input_ids[MAX_MIXER_INPUT_STREAMS];
|
|
unsigned int input_count;
|
|
CRITICAL_SECTION cs;
|
|
};
|
|
|
|
static struct video_mixer *impl_from_IMFTransform(IMFTransform *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, struct video_mixer, IMFTransform_iface);
|
|
}
|
|
|
|
static struct video_mixer *impl_from_IMFVideoDeviceID(IMFVideoDeviceID *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, struct video_mixer, IMFVideoDeviceID_iface);
|
|
}
|
|
|
|
static struct video_mixer *impl_from_IMFTopologyServiceLookupClient(IMFTopologyServiceLookupClient *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, struct video_mixer, IMFTopologyServiceLookupClient_iface);
|
|
}
|
|
|
|
static int video_mixer_compare_input_id(const void *a, const void *b)
|
|
{
|
|
const unsigned int *key = a;
|
|
const struct input_stream *input = b;
|
|
|
|
if (*key > input->id) return 1;
|
|
if (*key < input->id) return -1;
|
|
return 0;
|
|
}
|
|
|
|
static struct input_stream * video_mixer_get_input(const struct video_mixer *mixer, unsigned int id)
|
|
{
|
|
return bsearch(&id, mixer->inputs, mixer->input_count, sizeof(*mixer->inputs), video_mixer_compare_input_id);
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_QueryInterface(IMFTransform *iface, REFIID riid, void **obj)
|
|
{
|
|
struct video_mixer *mixer = impl_from_IMFTransform(iface);
|
|
|
|
TRACE("%p, %s, %p.\n", iface, debugstr_guid(riid), obj);
|
|
|
|
if (IsEqualIID(riid, &IID_IMFTransform) ||
|
|
IsEqualIID(riid, &IID_IUnknown))
|
|
{
|
|
*obj = iface;
|
|
}
|
|
else if (IsEqualIID(riid, &IID_IMFVideoDeviceID))
|
|
{
|
|
*obj = &mixer->IMFVideoDeviceID_iface;
|
|
}
|
|
else if (IsEqualIID(riid, &IID_IMFTopologyServiceLookupClient))
|
|
{
|
|
*obj = &mixer->IMFTopologyServiceLookupClient_iface;
|
|
}
|
|
else
|
|
{
|
|
WARN("Unsupported interface %s.\n", debugstr_guid(riid));
|
|
*obj = NULL;
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
IUnknown_AddRef((IUnknown *)*obj);
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI video_mixer_transform_AddRef(IMFTransform *iface)
|
|
{
|
|
struct video_mixer *mixer = impl_from_IMFTransform(iface);
|
|
ULONG refcount = InterlockedIncrement(&mixer->refcount);
|
|
|
|
TRACE("%p, refcount %u.\n", iface, refcount);
|
|
|
|
return refcount;
|
|
}
|
|
|
|
static ULONG WINAPI video_mixer_transform_Release(IMFTransform *iface)
|
|
{
|
|
struct video_mixer *mixer = impl_from_IMFTransform(iface);
|
|
ULONG refcount = InterlockedDecrement(&mixer->refcount);
|
|
|
|
TRACE("%p, refcount %u.\n", iface, refcount);
|
|
|
|
if (!refcount)
|
|
{
|
|
DeleteCriticalSection(&mixer->cs);
|
|
free(mixer);
|
|
}
|
|
|
|
return refcount;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_GetStreamLimits(IMFTransform *iface, DWORD *input_minimum,
|
|
DWORD *input_maximum, DWORD *output_minimum, DWORD *output_maximum)
|
|
{
|
|
TRACE("%p, %p, %p, %p, %p.\n", iface, input_minimum, input_maximum, output_minimum, output_maximum);
|
|
|
|
*input_minimum = 1;
|
|
*input_maximum = 16;
|
|
*output_minimum = 1;
|
|
*output_maximum = 1;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_GetStreamCount(IMFTransform *iface, DWORD *inputs, DWORD *outputs)
|
|
{
|
|
struct video_mixer *mixer = impl_from_IMFTransform(iface);
|
|
|
|
TRACE("%p, %p, %p.\n", iface, inputs, outputs);
|
|
|
|
EnterCriticalSection(&mixer->cs);
|
|
if (inputs) *inputs = mixer->input_count;
|
|
if (outputs) *outputs = 1;
|
|
LeaveCriticalSection(&mixer->cs);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_GetStreamIDs(IMFTransform *iface, DWORD input_size, DWORD *inputs,
|
|
DWORD output_size, DWORD *outputs)
|
|
{
|
|
struct video_mixer *mixer = impl_from_IMFTransform(iface);
|
|
HRESULT hr = S_OK;
|
|
|
|
TRACE("%p, %u, %p, %u, %p.\n", iface, input_size, inputs, output_size, outputs);
|
|
|
|
EnterCriticalSection(&mixer->cs);
|
|
if (mixer->input_count > input_size || !output_size)
|
|
hr = MF_E_BUFFERTOOSMALL;
|
|
else if (inputs)
|
|
memcpy(inputs, mixer->input_ids, mixer->input_count * sizeof(*inputs));
|
|
if (outputs) *outputs = 0;
|
|
LeaveCriticalSection(&mixer->cs);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_GetInputStreamInfo(IMFTransform *iface, DWORD id, MFT_INPUT_STREAM_INFO *info)
|
|
{
|
|
struct video_mixer *mixer = impl_from_IMFTransform(iface);
|
|
struct input_stream *input;
|
|
HRESULT hr = S_OK;
|
|
|
|
TRACE("%p, %u, %p.\n", iface, id, info);
|
|
|
|
EnterCriticalSection(&mixer->cs);
|
|
if (!(input = video_mixer_get_input(mixer, id)))
|
|
hr = MF_E_INVALIDSTREAMNUMBER;
|
|
else
|
|
{
|
|
memset(info, 0, sizeof(*info));
|
|
if (id)
|
|
info->dwFlags |= MFT_INPUT_STREAM_REMOVABLE | MFT_INPUT_STREAM_OPTIONAL;
|
|
}
|
|
LeaveCriticalSection(&mixer->cs);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_GetOutputStreamInfo(IMFTransform *iface, DWORD id, MFT_OUTPUT_STREAM_INFO *info)
|
|
{
|
|
TRACE("%p, %u, %p.\n", iface, id, info);
|
|
|
|
if (id)
|
|
return MF_E_INVALIDSTREAMNUMBER;
|
|
|
|
memset(info, 0, sizeof(*info));
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_GetAttributes(IMFTransform *iface, IMFAttributes **attributes)
|
|
{
|
|
FIXME("%p, %p.\n", iface, attributes);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_GetInputStreamAttributes(IMFTransform *iface, DWORD id,
|
|
IMFAttributes **attributes)
|
|
{
|
|
FIXME("%p, %u, %p.\n", iface, id, attributes);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_GetOutputStreamAttributes(IMFTransform *iface, DWORD id,
|
|
IMFAttributes **attributes)
|
|
{
|
|
TRACE("%p, %u, %p.\n", iface, id, attributes);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_DeleteInputStream(IMFTransform *iface, DWORD id)
|
|
{
|
|
struct video_mixer *mixer = impl_from_IMFTransform(iface);
|
|
struct input_stream *input;
|
|
HRESULT hr = S_OK;
|
|
unsigned int idx;
|
|
|
|
TRACE("%p, %u.\n", iface, id);
|
|
|
|
EnterCriticalSection(&mixer->cs);
|
|
|
|
/* Can't delete reference stream. */
|
|
if (!id || !(input = video_mixer_get_input(mixer, id)))
|
|
hr = MF_E_INVALIDSTREAMNUMBER;
|
|
else
|
|
{
|
|
mixer->input_count--;
|
|
idx = input - mixer->inputs;
|
|
if (idx < mixer->input_count)
|
|
{
|
|
memmove(&mixer->inputs[idx], &mixer->inputs[idx + 1], (mixer->input_count - idx) * sizeof(*mixer->inputs));
|
|
memmove(&mixer->input_ids[idx], &mixer->input_ids[idx + 1], (mixer->input_count - idx) *
|
|
sizeof(*mixer->input_ids));
|
|
}
|
|
}
|
|
|
|
LeaveCriticalSection(&mixer->cs);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static int video_mixer_add_input_sort_compare(const void *a, const void *b)
|
|
{
|
|
const struct input_stream *left = a, *right = b;
|
|
return left->id != right->id ? (left->id < right->id ? -1 : 1) : 0;
|
|
};
|
|
|
|
static HRESULT WINAPI video_mixer_transform_AddInputStreams(IMFTransform *iface, DWORD count, DWORD *ids)
|
|
{
|
|
struct video_mixer *mixer = impl_from_IMFTransform(iface);
|
|
struct input_stream inputs[MAX_MIXER_INPUT_STREAMS] = { {0} };
|
|
unsigned int i, len;
|
|
HRESULT hr = S_OK;
|
|
|
|
TRACE("%p, %u, %p.\n", iface, count, ids);
|
|
|
|
if (!ids)
|
|
return E_POINTER;
|
|
|
|
EnterCriticalSection(&mixer->cs);
|
|
if (count > ARRAY_SIZE(mixer->inputs) - mixer->input_count)
|
|
hr = E_INVALIDARG;
|
|
else
|
|
{
|
|
/* Test for collisions. */
|
|
memcpy(inputs, mixer->inputs, mixer->input_count * sizeof(*inputs));
|
|
for (i = 0; i < count; ++i)
|
|
inputs[i + mixer->input_count].id = ids[i];
|
|
|
|
len = mixer->input_count + count;
|
|
|
|
qsort(inputs, len, sizeof(*inputs), video_mixer_add_input_sort_compare);
|
|
|
|
for (i = 1; i < len; ++i)
|
|
{
|
|
if (inputs[i - 1].id == inputs[i].id)
|
|
{
|
|
hr = E_INVALIDARG;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
memcpy(&mixer->input_ids[mixer->input_count], ids, count * sizeof(*ids));
|
|
memcpy(mixer->inputs, inputs, len * sizeof(*inputs));
|
|
mixer->input_count += count;
|
|
}
|
|
}
|
|
LeaveCriticalSection(&mixer->cs);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_GetInputAvailableType(IMFTransform *iface, DWORD id, DWORD index,
|
|
IMFMediaType **type)
|
|
{
|
|
FIXME("%p, %u, %u, %p.\n", iface, id, index, type);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_GetOutputAvailableType(IMFTransform *iface, DWORD id, DWORD index,
|
|
IMFMediaType **type)
|
|
{
|
|
FIXME("%p, %u, %u, %p.\n", iface, id, index, type);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_SetInputType(IMFTransform *iface, DWORD id, IMFMediaType *type, DWORD flags)
|
|
{
|
|
FIXME("%p, %u, %p, %#x.\n", iface, id, type, flags);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_SetOutputType(IMFTransform *iface, DWORD id, IMFMediaType *type, DWORD flags)
|
|
{
|
|
FIXME("%p, %u, %p, %#x.\n", iface, id, type, flags);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_GetInputCurrentType(IMFTransform *iface, DWORD id, IMFMediaType **type)
|
|
{
|
|
FIXME("%p, %u, %p.\n", iface, id, type);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_GetOutputCurrentType(IMFTransform *iface, DWORD id, IMFMediaType **type)
|
|
{
|
|
FIXME("%p, %u, %p.\n", iface, id, type);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_GetInputStatus(IMFTransform *iface, DWORD id, DWORD *flags)
|
|
{
|
|
FIXME("%p, %u, %p.\n", iface, id, flags);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_GetOutputStatus(IMFTransform *iface, DWORD *flags)
|
|
{
|
|
FIXME("%p, %p.\n", iface, flags);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_SetOutputBounds(IMFTransform *iface, LONGLONG lower, LONGLONG upper)
|
|
{
|
|
FIXME("%p, %s, %s.\n", iface, wine_dbgstr_longlong(lower), wine_dbgstr_longlong(upper));
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_ProcessEvent(IMFTransform *iface, DWORD id, IMFMediaEvent *event)
|
|
{
|
|
FIXME("%p, %u, %p.\n", iface, id, event);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_ProcessMessage(IMFTransform *iface, MFT_MESSAGE_TYPE message, ULONG_PTR param)
|
|
{
|
|
FIXME("%p, %u, %#lx.\n", iface, message, param);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_ProcessInput(IMFTransform *iface, DWORD id, IMFSample *sample, DWORD flags)
|
|
{
|
|
FIXME("%p, %u, %p, %#x.\n", iface, id, sample, flags);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_transform_ProcessOutput(IMFTransform *iface, DWORD flags, DWORD count,
|
|
MFT_OUTPUT_DATA_BUFFER *samples, DWORD *status)
|
|
{
|
|
FIXME("%p, %#x, %u, %p, %p.\n", iface, flags, count, samples, status);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static const IMFTransformVtbl video_mixer_transform_vtbl =
|
|
{
|
|
video_mixer_transform_QueryInterface,
|
|
video_mixer_transform_AddRef,
|
|
video_mixer_transform_Release,
|
|
video_mixer_transform_GetStreamLimits,
|
|
video_mixer_transform_GetStreamCount,
|
|
video_mixer_transform_GetStreamIDs,
|
|
video_mixer_transform_GetInputStreamInfo,
|
|
video_mixer_transform_GetOutputStreamInfo,
|
|
video_mixer_transform_GetAttributes,
|
|
video_mixer_transform_GetInputStreamAttributes,
|
|
video_mixer_transform_GetOutputStreamAttributes,
|
|
video_mixer_transform_DeleteInputStream,
|
|
video_mixer_transform_AddInputStreams,
|
|
video_mixer_transform_GetInputAvailableType,
|
|
video_mixer_transform_GetOutputAvailableType,
|
|
video_mixer_transform_SetInputType,
|
|
video_mixer_transform_SetOutputType,
|
|
video_mixer_transform_GetInputCurrentType,
|
|
video_mixer_transform_GetOutputCurrentType,
|
|
video_mixer_transform_GetInputStatus,
|
|
video_mixer_transform_GetOutputStatus,
|
|
video_mixer_transform_SetOutputBounds,
|
|
video_mixer_transform_ProcessEvent,
|
|
video_mixer_transform_ProcessMessage,
|
|
video_mixer_transform_ProcessInput,
|
|
video_mixer_transform_ProcessOutput,
|
|
};
|
|
|
|
static HRESULT WINAPI video_mixer_device_id_QueryInterface(IMFVideoDeviceID *iface, REFIID riid, void **obj)
|
|
{
|
|
struct video_mixer *mixer = impl_from_IMFVideoDeviceID(iface);
|
|
return IMFTransform_QueryInterface(&mixer->IMFTransform_iface, riid, obj);
|
|
}
|
|
|
|
static ULONG WINAPI video_mixer_device_id_AddRef(IMFVideoDeviceID *iface)
|
|
{
|
|
struct video_mixer *mixer = impl_from_IMFVideoDeviceID(iface);
|
|
return IMFTransform_AddRef(&mixer->IMFTransform_iface);
|
|
}
|
|
|
|
static ULONG WINAPI video_mixer_device_id_Release(IMFVideoDeviceID *iface)
|
|
{
|
|
struct video_mixer *mixer = impl_from_IMFVideoDeviceID(iface);
|
|
return IMFTransform_Release(&mixer->IMFTransform_iface);
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_device_id_GetDeviceID(IMFVideoDeviceID *iface, IID *device_id)
|
|
{
|
|
TRACE("%p, %p.\n", iface, device_id);
|
|
|
|
if (!device_id)
|
|
return E_POINTER;
|
|
|
|
memcpy(device_id, &IID_IDirect3DDevice9, sizeof(*device_id));
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static const IMFVideoDeviceIDVtbl video_mixer_device_id_vtbl =
|
|
{
|
|
video_mixer_device_id_QueryInterface,
|
|
video_mixer_device_id_AddRef,
|
|
video_mixer_device_id_Release,
|
|
video_mixer_device_id_GetDeviceID,
|
|
};
|
|
|
|
static HRESULT WINAPI video_mixer_service_client_QueryInterface(IMFTopologyServiceLookupClient *iface,
|
|
REFIID riid, void **obj)
|
|
{
|
|
struct video_mixer *mixer = impl_from_IMFTopologyServiceLookupClient(iface);
|
|
return IMFTransform_QueryInterface(&mixer->IMFTransform_iface, riid, obj);
|
|
}
|
|
|
|
static ULONG WINAPI video_mixer_service_client_AddRef(IMFTopologyServiceLookupClient *iface)
|
|
{
|
|
struct video_mixer *mixer = impl_from_IMFTopologyServiceLookupClient(iface);
|
|
return IMFTransform_AddRef(&mixer->IMFTransform_iface);
|
|
}
|
|
|
|
static ULONG WINAPI video_mixer_service_client_Release(IMFTopologyServiceLookupClient *iface)
|
|
{
|
|
struct video_mixer *mixer = impl_from_IMFTopologyServiceLookupClient(iface);
|
|
return IMFTransform_Release(&mixer->IMFTransform_iface);
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_service_client_InitServicePointers(IMFTopologyServiceLookupClient *iface,
|
|
IMFTopologyServiceLookup *service_lookup)
|
|
{
|
|
FIXME("%p, %p.\n", iface, service_lookup);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI video_mixer_service_client_ReleaseServicePointers(IMFTopologyServiceLookupClient *iface)
|
|
{
|
|
FIXME("%p.\n", iface);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static const IMFTopologyServiceLookupClientVtbl video_mixer_service_client_vtbl =
|
|
{
|
|
video_mixer_service_client_QueryInterface,
|
|
video_mixer_service_client_AddRef,
|
|
video_mixer_service_client_Release,
|
|
video_mixer_service_client_InitServicePointers,
|
|
video_mixer_service_client_ReleaseServicePointers,
|
|
};
|
|
|
|
HRESULT WINAPI MFCreateVideoMixer(IUnknown *owner, REFIID riid_device, REFIID riid, void **obj)
|
|
{
|
|
TRACE("%p, %s, %s, %p.\n", owner, debugstr_guid(riid_device), debugstr_guid(riid), obj);
|
|
|
|
*obj = NULL;
|
|
|
|
if (!IsEqualIID(riid_device, &IID_IDirect3DDevice9))
|
|
return E_INVALIDARG;
|
|
|
|
return CoCreateInstance(&CLSID_MFVideoMixer9, owner, CLSCTX_INPROC_SERVER, riid, obj);
|
|
}
|
|
|
|
HRESULT evr_mixer_create(IUnknown *outer, void **out)
|
|
{
|
|
struct video_mixer *object;
|
|
|
|
if (outer)
|
|
return E_NOINTERFACE;
|
|
|
|
if (!(object = calloc(1, sizeof(*object))))
|
|
return E_OUTOFMEMORY;
|
|
|
|
object->IMFTransform_iface.lpVtbl = &video_mixer_transform_vtbl;
|
|
object->IMFVideoDeviceID_iface.lpVtbl = &video_mixer_device_id_vtbl;
|
|
object->IMFTopologyServiceLookupClient_iface.lpVtbl = &video_mixer_service_client_vtbl;
|
|
object->refcount = 1;
|
|
object->input_count = 1;
|
|
InitializeCriticalSection(&object->cs);
|
|
|
|
*out = &object->IMFTransform_iface;
|
|
|
|
return S_OK;
|
|
}
|