457 lines
14 KiB
C
457 lines
14 KiB
C
/*
|
|
* Copyright 2009 Henri Verbeet for CodeWeavers
|
|
*
|
|
* 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
|
|
*
|
|
*/
|
|
|
|
#include "config.h"
|
|
#include "wine/port.h"
|
|
|
|
#include "d3d11_private.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(d3d11);
|
|
|
|
/* ID3D11Query methods */
|
|
|
|
static inline struct d3d_query *impl_from_ID3D11Query(ID3D11Query *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, struct d3d_query, ID3D11Query_iface);
|
|
}
|
|
|
|
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 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 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 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 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 HRESULT STDMETHODCALLTYPE d3d11_query_SetPrivateData(ID3D11Query *iface,
|
|
REFGUID guid, UINT data_size, const void *data)
|
|
{
|
|
struct d3d_query *query = impl_from_ID3D11Query(iface);
|
|
|
|
TRACE("iface %p, guid %s, data_size %u, data %p.\n", iface, debugstr_guid(guid), data_size, data);
|
|
|
|
return d3d_set_private_data(&query->private_store, guid, data_size, data);
|
|
}
|
|
|
|
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 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 void STDMETHODCALLTYPE d3d11_query_GetDesc(ID3D11Query *iface, D3D11_QUERY_DESC *desc)
|
|
{
|
|
FIXME("iface %p, desc %p stub!\n", iface, desc);
|
|
}
|
|
|
|
static const struct ID3D11QueryVtbl d3d11_query_vtbl =
|
|
{
|
|
/* IUnknown methods */
|
|
d3d11_query_QueryInterface,
|
|
d3d11_query_AddRef,
|
|
d3d11_query_Release,
|
|
/* ID3D11DeviceChild methods */
|
|
d3d11_query_GetDevice,
|
|
d3d11_query_GetPrivateData,
|
|
d3d11_query_SetPrivateData,
|
|
d3d11_query_SetPrivateDataInterface,
|
|
/* ID3D11Asynchronous methods */
|
|
d3d11_query_GetDataSize,
|
|
/* ID3D11Query methods */
|
|
d3d11_query_GetDesc,
|
|
};
|
|
|
|
struct d3d_query *unsafe_impl_from_ID3D11Query(ID3D11Query *iface)
|
|
{
|
|
if (!iface)
|
|
return NULL;
|
|
assert(iface->lpVtbl == &d3d11_query_vtbl);
|
|
return CONTAINING_RECORD(iface, struct d3d_query, ID3D11Query_iface);
|
|
}
|
|
|
|
struct d3d_query *unsafe_impl_from_ID3D11Asynchronous(ID3D11Asynchronous *iface)
|
|
{
|
|
return unsafe_impl_from_ID3D11Query((ID3D11Query *)iface);
|
|
}
|
|
|
|
/* ID3D10Query methods */
|
|
|
|
static inline struct d3d_query *impl_from_ID3D10Query(ID3D10Query *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, struct d3d_query, ID3D10Query_iface);
|
|
}
|
|
|
|
/* IUnknown methods */
|
|
|
|
static HRESULT STDMETHODCALLTYPE d3d10_query_QueryInterface(ID3D10Query *iface, REFIID riid, void **object)
|
|
{
|
|
struct d3d_query *query = impl_from_ID3D10Query(iface);
|
|
|
|
TRACE("iface %p, riid %s, object %p.\n", iface, debugstr_guid(riid), object);
|
|
|
|
return d3d11_query_QueryInterface(&query->ID3D11Query_iface, riid, object);
|
|
}
|
|
|
|
static ULONG STDMETHODCALLTYPE d3d10_query_AddRef(ID3D10Query *iface)
|
|
{
|
|
struct d3d_query *query = impl_from_ID3D10Query(iface);
|
|
|
|
TRACE("iface %p.\n", iface);
|
|
|
|
return d3d11_query_AddRef(&query->ID3D11Query_iface);
|
|
}
|
|
|
|
static ULONG STDMETHODCALLTYPE d3d10_query_Release(ID3D10Query *iface)
|
|
{
|
|
struct d3d_query *query = impl_from_ID3D10Query(iface);
|
|
|
|
TRACE("iface %p.\n", iface);
|
|
|
|
return d3d11_query_Release(&query->ID3D11Query_iface);
|
|
}
|
|
|
|
/* ID3D10DeviceChild methods */
|
|
|
|
static void STDMETHODCALLTYPE d3d10_query_GetDevice(ID3D10Query *iface, ID3D10Device **device)
|
|
{
|
|
struct d3d_query *query = impl_from_ID3D10Query(iface);
|
|
|
|
TRACE("iface %p, device %p.\n", iface, device);
|
|
|
|
ID3D11Device_QueryInterface(query->device, &IID_ID3D10Device, (void **)device);
|
|
}
|
|
|
|
static HRESULT STDMETHODCALLTYPE d3d10_query_GetPrivateData(ID3D10Query *iface,
|
|
REFGUID guid, UINT *data_size, void *data)
|
|
{
|
|
struct d3d_query *query = impl_from_ID3D10Query(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 HRESULT STDMETHODCALLTYPE d3d10_query_SetPrivateData(ID3D10Query *iface,
|
|
REFGUID guid, UINT data_size, const void *data)
|
|
{
|
|
struct d3d_query *query = impl_from_ID3D10Query(iface);
|
|
|
|
TRACE("iface %p, guid %s, data_size %u, data %p.\n",
|
|
iface, debugstr_guid(guid), data_size, data);
|
|
|
|
return d3d_set_private_data(&query->private_store, guid, data_size, data);
|
|
}
|
|
|
|
static HRESULT STDMETHODCALLTYPE d3d10_query_SetPrivateDataInterface(ID3D10Query *iface,
|
|
REFGUID guid, const IUnknown *data)
|
|
{
|
|
struct d3d_query *query = impl_from_ID3D10Query(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);
|
|
}
|
|
|
|
/* ID3D10Asynchronous methods */
|
|
|
|
static void STDMETHODCALLTYPE d3d10_query_Begin(ID3D10Query *iface)
|
|
{
|
|
struct d3d_query *query = impl_from_ID3D10Query(iface);
|
|
HRESULT hr;
|
|
|
|
TRACE("iface %p.\n", iface);
|
|
|
|
wined3d_mutex_lock();
|
|
if (FAILED(hr = wined3d_query_issue(query->wined3d_query, WINED3DISSUE_BEGIN)))
|
|
ERR("Failed to issue query, hr %#x.\n", hr);
|
|
wined3d_mutex_unlock();
|
|
}
|
|
|
|
static void STDMETHODCALLTYPE d3d10_query_End(ID3D10Query *iface)
|
|
{
|
|
struct d3d_query *query = impl_from_ID3D10Query(iface);
|
|
HRESULT hr;
|
|
|
|
TRACE("iface %p.\n", iface);
|
|
|
|
wined3d_mutex_lock();
|
|
if (FAILED(hr = wined3d_query_issue(query->wined3d_query, WINED3DISSUE_END)))
|
|
ERR("Failed to issue query, hr %#x.\n", hr);
|
|
wined3d_mutex_unlock();
|
|
}
|
|
|
|
static HRESULT STDMETHODCALLTYPE d3d10_query_GetData(ID3D10Query *iface, void *data, UINT data_size, UINT flags)
|
|
{
|
|
struct d3d_query *query = impl_from_ID3D10Query(iface);
|
|
unsigned int wined3d_flags;
|
|
HRESULT hr;
|
|
|
|
TRACE("iface %p, data %p, data_size %u, flags %#x.\n", iface, data, data_size, flags);
|
|
|
|
wined3d_flags = wined3d_getdata_flags_from_d3d11_async_getdata_flags(flags);
|
|
|
|
wined3d_mutex_lock();
|
|
if (!data_size || wined3d_query_get_data_size(query->wined3d_query) == data_size)
|
|
{
|
|
hr = wined3d_query_get_data(query->wined3d_query, data, data_size, wined3d_flags);
|
|
}
|
|
else
|
|
{
|
|
WARN("Invalid data size %u.\n", data_size);
|
|
hr = E_INVALIDARG;
|
|
}
|
|
wined3d_mutex_unlock();
|
|
|
|
return hr;
|
|
}
|
|
|
|
static UINT STDMETHODCALLTYPE d3d10_query_GetDataSize(ID3D10Query *iface)
|
|
{
|
|
struct d3d_query *query = impl_from_ID3D10Query(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;
|
|
}
|
|
|
|
/* ID3D10Query methods */
|
|
|
|
static void STDMETHODCALLTYPE d3d10_query_GetDesc(ID3D10Query *iface, D3D10_QUERY_DESC *desc)
|
|
{
|
|
FIXME("iface %p, desc %p stub!\n", iface, desc);
|
|
}
|
|
|
|
static const struct ID3D10QueryVtbl d3d10_query_vtbl =
|
|
{
|
|
/* IUnknown methods */
|
|
d3d10_query_QueryInterface,
|
|
d3d10_query_AddRef,
|
|
d3d10_query_Release,
|
|
/* ID3D10DeviceChild methods */
|
|
d3d10_query_GetDevice,
|
|
d3d10_query_GetPrivateData,
|
|
d3d10_query_SetPrivateData,
|
|
d3d10_query_SetPrivateDataInterface,
|
|
/* ID3D10Asynchronous methods */
|
|
d3d10_query_Begin,
|
|
d3d10_query_End,
|
|
d3d10_query_GetData,
|
|
d3d10_query_GetDataSize,
|
|
/* ID3D10Query methods */
|
|
d3d10_query_GetDesc,
|
|
};
|
|
|
|
struct d3d_query *unsafe_impl_from_ID3D10Query(ID3D10Query *iface)
|
|
{
|
|
if (!iface)
|
|
return NULL;
|
|
assert(iface->lpVtbl == &d3d10_query_vtbl);
|
|
return CONTAINING_RECORD(iface, struct d3d_query, ID3D10Query_iface);
|
|
}
|
|
|
|
static HRESULT d3d_query_init(struct d3d_query *query, struct d3d_device *device,
|
|
const D3D11_QUERY_DESC *desc, BOOL predicate)
|
|
{
|
|
HRESULT hr;
|
|
|
|
static const enum wined3d_query_type query_type_map[] =
|
|
{
|
|
/* D3D11_QUERY_EVENT */ WINED3D_QUERY_TYPE_EVENT,
|
|
/* D3D11_QUERY_OCCLUSION */ WINED3D_QUERY_TYPE_OCCLUSION,
|
|
/* D3D11_QUERY_TIMESTAMP */ WINED3D_QUERY_TYPE_TIMESTAMP,
|
|
/* D3D11_QUERY_TIMESTAMP_DISJOINT */ WINED3D_QUERY_TYPE_TIMESTAMP_DISJOINT,
|
|
/* D3D11_QUERY_PIPELINE_STATISTICS */ WINED3D_QUERY_TYPE_PIPELINE_STATISTICS,
|
|
/* D3D11_QUERY_OCCLUSION_PREDICATE */ WINED3D_QUERY_TYPE_OCCLUSION,
|
|
/* D3D11_QUERY_SO_STATISTICS */ WINED3D_QUERY_TYPE_SO_STATISTICS,
|
|
/* D3D11_QUERY_SO_OVERFLOW_PREDICATE */ WINED3D_QUERY_TYPE_SO_OVERFLOW,
|
|
};
|
|
|
|
if (desc->Query >= sizeof(query_type_map) / sizeof(*query_type_map))
|
|
{
|
|
FIXME("Unhandled query type %#x.\n", desc->Query);
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
if (desc->MiscFlags)
|
|
FIXME("Ignoring MiscFlags %#x.\n", desc->MiscFlags);
|
|
|
|
query->ID3D11Query_iface.lpVtbl = &d3d11_query_vtbl;
|
|
query->ID3D10Query_iface.lpVtbl = &d3d10_query_vtbl;
|
|
query->refcount = 1;
|
|
wined3d_mutex_lock();
|
|
wined3d_private_store_init(&query->private_store);
|
|
|
|
if (FAILED(hr = wined3d_query_create(device->wined3d_device,
|
|
query_type_map[desc->Query], query, &query->wined3d_query)))
|
|
{
|
|
WARN("Failed to create wined3d query, hr %#x.\n", hr);
|
|
wined3d_private_store_cleanup(&query->private_store);
|
|
wined3d_mutex_unlock();
|
|
return hr;
|
|
}
|
|
wined3d_mutex_unlock();
|
|
|
|
query->predicate = predicate;
|
|
query->device = &device->ID3D11Device_iface;
|
|
ID3D11Device_AddRef(query->device);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT d3d_query_create(struct d3d_device *device, const D3D11_QUERY_DESC *desc, BOOL predicate,
|
|
struct d3d_query **query)
|
|
{
|
|
struct d3d_query *object;
|
|
BOOL is_predicate_type;
|
|
HRESULT hr;
|
|
|
|
if (!desc)
|
|
return E_INVALIDARG;
|
|
|
|
is_predicate_type = desc->Query == D3D11_QUERY_OCCLUSION_PREDICATE
|
|
|| desc->Query == D3D11_QUERY_SO_OVERFLOW_PREDICATE
|
|
|| desc->Query == D3D11_QUERY_SO_OVERFLOW_PREDICATE_STREAM0
|
|
|| desc->Query == D3D11_QUERY_SO_OVERFLOW_PREDICATE_STREAM1
|
|
|| desc->Query == D3D11_QUERY_SO_OVERFLOW_PREDICATE_STREAM2
|
|
|| desc->Query == D3D11_QUERY_SO_OVERFLOW_PREDICATE_STREAM3;
|
|
|
|
if (!is_predicate_type && predicate)
|
|
{
|
|
WARN("Query type %u is not a predicate.\n", desc->Query);
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
if (is_predicate_type)
|
|
predicate = TRUE;
|
|
|
|
if (!(object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*object))))
|
|
return E_OUTOFMEMORY;
|
|
|
|
if (FAILED(hr = d3d_query_init(object, device, desc, predicate)))
|
|
{
|
|
WARN("Failed to initialize predicate, hr %#x.\n", hr);
|
|
HeapFree(GetProcessHeap(), 0, object);
|
|
return hr;
|
|
}
|
|
|
|
TRACE("Created predicate %p.\n", object);
|
|
*query = object;
|
|
|
|
return S_OK;
|
|
}
|