static HRESULT STDMETHODCALLTYPE d3d11_query_QueryInterface(ID3D11Query *iface, REFIID riid, void **object)
{
struct d3d_query *query = impl_from_ID3D11Query(iface);
TRACE("iface %p, riid %s, object %p.\n", iface, debugstr_guid(riid), object);
if ((IsEqualGUID(riid, &IID_ID3D11Predicate) && query->predicate)
|| IsEqualGUID(riid, &IID_ID3D11Query)
|| IsEqualGUID(riid, &IID_ID3D11Asynchronous)
|| IsEqualGUID(riid, &IID_ID3D11DeviceChild)
|| IsEqualGUID(riid, &IID_IUnknown))
{
ID3D11Query_AddRef(iface);
*object = iface;
return S_OK;
}
if ((IsEqualGUID(riid, &IID_ID3D10Predicate) && query->predicate)
|| IsEqualGUID(riid, &IID_ID3D10Query)
|| IsEqualGUID(riid, &IID_ID3D10Asynchronous)
|| IsEqualGUID(riid, &IID_ID3D10DeviceChild))
{
ID3D10Query_AddRef(&query->ID3D10Query_iface);
*object = &query->ID3D10Query_iface;
return S_OK;
}
WARN("%s not implemented, returning E_NOINTERFACE.\n", debugstr_guid(riid));
*object = NULL;
return E_NOINTERFACE;
}
static void STDMETHODCALLTYPE d3d11_query_GetDesc(ID3D11Query *iface, D3D11_QUERY_DESC *desc)
{
struct d3d_query *query = impl_from_ID3D11Query(iface);
TRACE("iface %p, desc %p.\n", iface, desc);
*desc = query->desc;
}
static void STDMETHODCALLTYPE d3d11_query_GetDevice(ID3D11Query *iface, ID3D11Device **device)
{
struct d3d_query *query = impl_from_ID3D11Query(iface);
TRACE("iface %p, device %p.\n", iface, device);
*device = query->device;
ID3D11Device_AddRef(*device);
}
static ULONG STDMETHODCALLTYPE d3d11_query_AddRef(ID3D11Query *iface)
{
struct d3d_query *query = impl_from_ID3D11Query(iface);
ULONG refcount = InterlockedIncrement(&query->refcount);
TRACE("%p increasing refcount to %u.\n", query, refcount);
return refcount;
}
static HRESULT STDMETHODCALLTYPE d3d11_query_SetPrivateDataInterface(ID3D11Query *iface,
REFGUID guid, const IUnknown *data)
{
struct d3d_query *query = impl_from_ID3D11Query(iface);
TRACE("iface %p, guid %s, data %p.\n", iface, debugstr_guid(guid), data);
return d3d_set_private_data_interface(&query->private_store, guid, data);
}
static HRESULT STDMETHODCALLTYPE d3d11_query_GetPrivateData(ID3D11Query *iface,
REFGUID guid, UINT *data_size, void *data)
{
struct d3d_query *query = impl_from_ID3D11Query(iface);
TRACE("iface %p, guid %s, data_size %p, data %p.\n", iface, debugstr_guid(guid), data_size, data);
return d3d_get_private_data(&query->private_store, guid, data_size, data);
}
static UINT STDMETHODCALLTYPE d3d11_query_GetDataSize(ID3D11Query *iface)
{
struct d3d_query *query = impl_from_ID3D11Query(iface);
unsigned int data_size;
TRACE("iface %p.\n", iface);
wined3d_mutex_lock();
data_size = wined3d_query_get_data_size(query->wined3d_query);
wined3d_mutex_unlock();
return data_size;
}
static ULONG STDMETHODCALLTYPE d3d11_query_AddRef(ID3D11Query *iface)
{
struct d3d_query *query = impl_from_ID3D11Query(iface);
ULONG refcount = InterlockedIncrement(&query->refcount);
TRACE("%p increasing refcount to %u.\n", query, refcount);
if (refcount == 1)
{
ID3D11Device_AddRef(query->device);
wined3d_mutex_lock();
wined3d_query_incref(query->wined3d_query);
wined3d_mutex_unlock();
}
return refcount;
}
static ULONG STDMETHODCALLTYPE d3d11_query_Release(ID3D11Query *iface)
{
struct d3d_query *query = impl_from_ID3D11Query(iface);
ULONG refcount = InterlockedDecrement(&query->refcount);
TRACE("%p decreasing refcount to %u.\n", query, refcount);
if (!refcount)
{
ID3D11Device_Release(query->device);
wined3d_mutex_lock();
wined3d_query_decref(query->wined3d_query);
wined3d_private_store_cleanup(&query->private_store);
wined3d_mutex_unlock();
HeapFree(GetProcessHeap(), 0, query);
}
return refcount;
}
static ULONG STDMETHODCALLTYPE d3d11_query_Release(ID3D11Query *iface)
{
struct d3d_query *query = impl_from_ID3D11Query(iface);
ULONG refcount = InterlockedDecrement(&query->refcount);
TRACE("%p decreasing refcount to %u.\n", query, refcount);
if (!refcount)
{
ID3D11Device *device = query->device;
wined3d_mutex_lock();
wined3d_query_decref(query->wined3d_query);
wined3d_mutex_unlock();
ID3D11Device_Release(device);
}
return refcount;
}
