2 * Copyright 2009 Henri Verbeet for CodeWeavers
3 * Copyright 2015-2017 Józef Kucia for CodeWeavers
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2.1 of the License, or (at your option) any later version.
10 * This library is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
15 * You should have received a copy of the GNU Lesser General Public
16 * License along with this library; if not, write to the Free Software
17 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 #include "d3d11_private.h"
23 WINE_DEFAULT_DEBUG_CHANNEL(d3d11
);
25 /* ID3D11Query methods */
27 static inline struct d3d_query
*impl_from_ID3D11Query(ID3D11Query
*iface
)
29 return CONTAINING_RECORD(iface
, struct d3d_query
, ID3D11Query_iface
);
32 static HRESULT STDMETHODCALLTYPE
d3d11_query_QueryInterface(ID3D11Query
*iface
, REFIID riid
, void **object
)
34 struct d3d_query
*query
= impl_from_ID3D11Query(iface
);
36 TRACE("iface %p, riid %s, object %p.\n", iface
, debugstr_guid(riid
), object
);
38 if ((IsEqualGUID(riid
, &IID_ID3D11Predicate
) && query
->predicate
)
39 || IsEqualGUID(riid
, &IID_ID3D11Query
)
40 || IsEqualGUID(riid
, &IID_ID3D11Asynchronous
)
41 || IsEqualGUID(riid
, &IID_ID3D11DeviceChild
)
42 || IsEqualGUID(riid
, &IID_IUnknown
))
44 ID3D11Query_AddRef(iface
);
49 if ((IsEqualGUID(riid
, &IID_ID3D10Predicate
) && query
->predicate
)
50 || IsEqualGUID(riid
, &IID_ID3D10Query
)
51 || IsEqualGUID(riid
, &IID_ID3D10Asynchronous
)
52 || IsEqualGUID(riid
, &IID_ID3D10DeviceChild
))
54 ID3D10Query_AddRef(&query
->ID3D10Query_iface
);
55 *object
= &query
->ID3D10Query_iface
;
59 WARN("%s not implemented, returning E_NOINTERFACE.\n", debugstr_guid(riid
));
65 static ULONG STDMETHODCALLTYPE
d3d11_query_AddRef(ID3D11Query
*iface
)
67 struct d3d_query
*query
= impl_from_ID3D11Query(iface
);
68 ULONG refcount
= InterlockedIncrement(&query
->refcount
);
70 TRACE("%p increasing refcount to %lu.\n", query
, refcount
);
74 ID3D11Device2_AddRef(query
->device
);
75 wined3d_query_incref(query
->wined3d_query
);
81 static ULONG STDMETHODCALLTYPE
d3d11_query_Release(ID3D11Query
*iface
)
83 struct d3d_query
*query
= impl_from_ID3D11Query(iface
);
84 ULONG refcount
= InterlockedDecrement(&query
->refcount
);
86 TRACE("%p decreasing refcount to %lu.\n", query
, refcount
);
90 ID3D11Device2
*device
= query
->device
;
91 wined3d_query_decref(query
->wined3d_query
);
92 ID3D11Device2_Release(device
);
98 static void STDMETHODCALLTYPE
d3d11_query_GetDevice(ID3D11Query
*iface
, ID3D11Device
**device
)
100 struct d3d_query
*query
= impl_from_ID3D11Query(iface
);
102 TRACE("iface %p, device %p.\n", iface
, device
);
104 *device
= (ID3D11Device
*)query
->device
;
105 ID3D11Device_AddRef(*device
);
108 static HRESULT STDMETHODCALLTYPE
d3d11_query_GetPrivateData(ID3D11Query
*iface
,
109 REFGUID guid
, UINT
*data_size
, void *data
)
111 struct d3d_query
*query
= impl_from_ID3D11Query(iface
);
113 TRACE("iface %p, guid %s, data_size %p, data %p.\n", iface
, debugstr_guid(guid
), data_size
, data
);
115 return d3d_get_private_data(&query
->private_store
, guid
, data_size
, data
);
118 static HRESULT STDMETHODCALLTYPE
d3d11_query_SetPrivateData(ID3D11Query
*iface
,
119 REFGUID guid
, UINT data_size
, const void *data
)
121 struct d3d_query
*query
= impl_from_ID3D11Query(iface
);
123 TRACE("iface %p, guid %s, data_size %u, data %p.\n", iface
, debugstr_guid(guid
), data_size
, data
);
125 return d3d_set_private_data(&query
->private_store
, guid
, data_size
, data
);
128 static HRESULT STDMETHODCALLTYPE
d3d11_query_SetPrivateDataInterface(ID3D11Query
*iface
,
129 REFGUID guid
, const IUnknown
*data
)
131 struct d3d_query
*query
= impl_from_ID3D11Query(iface
);
133 TRACE("iface %p, guid %s, data %p.\n", iface
, debugstr_guid(guid
), data
);
135 return d3d_set_private_data_interface(&query
->private_store
, guid
, data
);
138 static UINT STDMETHODCALLTYPE
d3d11_query_GetDataSize(ID3D11Query
*iface
)
140 struct d3d_query
*query
= impl_from_ID3D11Query(iface
);
141 unsigned int data_size
;
143 TRACE("iface %p.\n", iface
);
145 wined3d_mutex_lock();
146 data_size
= wined3d_query_get_data_size(query
->wined3d_query
);
147 wined3d_mutex_unlock();
152 static void STDMETHODCALLTYPE
d3d11_query_GetDesc(ID3D11Query
*iface
, D3D11_QUERY_DESC
*desc
)
154 struct d3d_query
*query
= impl_from_ID3D11Query(iface
);
156 TRACE("iface %p, desc %p.\n", iface
, desc
);
161 static const struct ID3D11QueryVtbl d3d11_query_vtbl
=
163 /* IUnknown methods */
164 d3d11_query_QueryInterface
,
167 /* ID3D11DeviceChild methods */
168 d3d11_query_GetDevice
,
169 d3d11_query_GetPrivateData
,
170 d3d11_query_SetPrivateData
,
171 d3d11_query_SetPrivateDataInterface
,
172 /* ID3D11Asynchronous methods */
173 d3d11_query_GetDataSize
,
174 /* ID3D11Query methods */
178 static void STDMETHODCALLTYPE
d3d_query_wined3d_object_destroyed(void *parent
)
180 struct d3d_query
*query
= parent
;
182 wined3d_private_store_cleanup(&query
->private_store
);
186 static const struct wined3d_parent_ops d3d_query_wined3d_parent_ops
=
188 d3d_query_wined3d_object_destroyed
,
191 struct d3d_query
*unsafe_impl_from_ID3D11Query(ID3D11Query
*iface
)
195 assert(iface
->lpVtbl
== &d3d11_query_vtbl
);
196 return CONTAINING_RECORD(iface
, struct d3d_query
, ID3D11Query_iface
);
199 struct d3d_query
*unsafe_impl_from_ID3D11Asynchronous(ID3D11Asynchronous
*iface
)
201 return unsafe_impl_from_ID3D11Query((ID3D11Query
*)iface
);
204 /* ID3D10Query methods */
206 static inline struct d3d_query
*impl_from_ID3D10Query(ID3D10Query
*iface
)
208 return CONTAINING_RECORD(iface
, struct d3d_query
, ID3D10Query_iface
);
211 /* IUnknown methods */
213 static HRESULT STDMETHODCALLTYPE
d3d10_query_QueryInterface(ID3D10Query
*iface
, REFIID riid
, void **object
)
215 struct d3d_query
*query
= impl_from_ID3D10Query(iface
);
217 TRACE("iface %p, riid %s, object %p.\n", iface
, debugstr_guid(riid
), object
);
219 return d3d11_query_QueryInterface(&query
->ID3D11Query_iface
, riid
, object
);
222 static ULONG STDMETHODCALLTYPE
d3d10_query_AddRef(ID3D10Query
*iface
)
224 struct d3d_query
*query
= impl_from_ID3D10Query(iface
);
226 TRACE("iface %p.\n", iface
);
228 return d3d11_query_AddRef(&query
->ID3D11Query_iface
);
231 static ULONG STDMETHODCALLTYPE
d3d10_query_Release(ID3D10Query
*iface
)
233 struct d3d_query
*query
= impl_from_ID3D10Query(iface
);
235 TRACE("iface %p.\n", iface
);
237 return d3d11_query_Release(&query
->ID3D11Query_iface
);
240 /* ID3D10DeviceChild methods */
242 static void STDMETHODCALLTYPE
d3d10_query_GetDevice(ID3D10Query
*iface
, ID3D10Device
**device
)
244 struct d3d_query
*query
= impl_from_ID3D10Query(iface
);
246 TRACE("iface %p, device %p.\n", iface
, device
);
248 ID3D11Device2_QueryInterface(query
->device
, &IID_ID3D10Device
, (void **)device
);
251 static HRESULT STDMETHODCALLTYPE
d3d10_query_GetPrivateData(ID3D10Query
*iface
,
252 REFGUID guid
, UINT
*data_size
, void *data
)
254 struct d3d_query
*query
= impl_from_ID3D10Query(iface
);
256 TRACE("iface %p, guid %s, data_size %p, data %p.\n",
257 iface
, debugstr_guid(guid
), data_size
, data
);
259 return d3d_get_private_data(&query
->private_store
, guid
, data_size
, data
);
262 static HRESULT STDMETHODCALLTYPE
d3d10_query_SetPrivateData(ID3D10Query
*iface
,
263 REFGUID guid
, UINT data_size
, const void *data
)
265 struct d3d_query
*query
= impl_from_ID3D10Query(iface
);
267 TRACE("iface %p, guid %s, data_size %u, data %p.\n",
268 iface
, debugstr_guid(guid
), data_size
, data
);
270 return d3d_set_private_data(&query
->private_store
, guid
, data_size
, data
);
273 static HRESULT STDMETHODCALLTYPE
d3d10_query_SetPrivateDataInterface(ID3D10Query
*iface
,
274 REFGUID guid
, const IUnknown
*data
)
276 struct d3d_query
*query
= impl_from_ID3D10Query(iface
);
278 TRACE("iface %p, guid %s, data %p.\n", iface
, debugstr_guid(guid
), data
);
280 return d3d_set_private_data_interface(&query
->private_store
, guid
, data
);
283 /* ID3D10Asynchronous methods */
285 static void STDMETHODCALLTYPE
d3d10_query_Begin(ID3D10Query
*iface
)
287 struct d3d_query
*query
= impl_from_ID3D10Query(iface
);
290 TRACE("iface %p.\n", iface
);
292 if (FAILED(hr
= wined3d_query_issue(query
->wined3d_query
, WINED3DISSUE_BEGIN
)))
293 ERR("Failed to issue query, hr %#lx.\n", hr
);
296 static void STDMETHODCALLTYPE
d3d10_query_End(ID3D10Query
*iface
)
298 struct d3d_query
*query
= impl_from_ID3D10Query(iface
);
301 TRACE("iface %p.\n", iface
);
303 if (FAILED(hr
= wined3d_query_issue(query
->wined3d_query
, WINED3DISSUE_END
)))
304 ERR("Failed to issue query, hr %#lx.\n", hr
);
307 static HRESULT STDMETHODCALLTYPE
d3d10_query_GetData(ID3D10Query
*iface
, void *data
, UINT data_size
, UINT flags
)
309 struct d3d_query
*query
= impl_from_ID3D10Query(iface
);
310 D3D11_QUERY_DATA_PIPELINE_STATISTICS d3d11_data
;
311 void *d3d10_data_pointer
= NULL
;
312 unsigned int wined3d_flags
;
315 TRACE("iface %p, data %p, data_size %u, flags %#x.\n", iface
, data
, data_size
, flags
);
317 if (!data
&& data_size
)
320 if (query
->desc
.Query
== D3D11_QUERY_PIPELINE_STATISTICS
321 && data_size
== sizeof(D3D10_QUERY_DATA_PIPELINE_STATISTICS
))
323 data_size
= sizeof(D3D11_QUERY_DATA_PIPELINE_STATISTICS
);
324 d3d10_data_pointer
= data
;
328 wined3d_flags
= wined3d_getdata_flags_from_d3d11_async_getdata_flags(flags
);
330 wined3d_mutex_lock();
331 if (!data_size
|| wined3d_query_get_data_size(query
->wined3d_query
) == data_size
)
333 hr
= wined3d_query_get_data(query
->wined3d_query
, data
, data_size
, wined3d_flags
);
334 if (hr
== WINED3DERR_INVALIDCALL
)
335 hr
= DXGI_ERROR_INVALID_CALL
;
339 WARN("Invalid data size %u.\n", data_size
);
342 wined3d_mutex_unlock();
344 if (d3d10_data_pointer
&& hr
== S_OK
)
345 memcpy(d3d10_data_pointer
, data
, sizeof(D3D10_QUERY_DATA_PIPELINE_STATISTICS
));
350 static UINT STDMETHODCALLTYPE
d3d10_query_GetDataSize(ID3D10Query
*iface
)
352 struct d3d_query
*query
= impl_from_ID3D10Query(iface
);
353 unsigned int data_size
;
355 TRACE("iface %p.\n", iface
);
357 wined3d_mutex_lock();
358 data_size
= wined3d_query_get_data_size(query
->wined3d_query
);
359 wined3d_mutex_unlock();
361 if (query
->desc
.Query
== D3D11_QUERY_PIPELINE_STATISTICS
)
362 data_size
= sizeof(D3D10_QUERY_DATA_PIPELINE_STATISTICS
);
367 /* ID3D10Query methods */
369 static void STDMETHODCALLTYPE
d3d10_query_GetDesc(ID3D10Query
*iface
, D3D10_QUERY_DESC
*desc
)
371 struct d3d_query
*query
= impl_from_ID3D10Query(iface
);
373 TRACE("iface %p, desc %p.\n", iface
, desc
);
375 memcpy(desc
, &query
->desc
, sizeof(*desc
));
378 static const struct ID3D10QueryVtbl d3d10_query_vtbl
=
380 /* IUnknown methods */
381 d3d10_query_QueryInterface
,
384 /* ID3D10DeviceChild methods */
385 d3d10_query_GetDevice
,
386 d3d10_query_GetPrivateData
,
387 d3d10_query_SetPrivateData
,
388 d3d10_query_SetPrivateDataInterface
,
389 /* ID3D10Asynchronous methods */
393 d3d10_query_GetDataSize
,
394 /* ID3D10Query methods */
398 struct d3d_query
*unsafe_impl_from_ID3D10Query(ID3D10Query
*iface
)
402 assert(iface
->lpVtbl
== &d3d10_query_vtbl
);
403 return CONTAINING_RECORD(iface
, struct d3d_query
, ID3D10Query_iface
);
406 static HRESULT
d3d_query_init(struct d3d_query
*query
, struct d3d_device
*device
,
407 const D3D11_QUERY_DESC
*desc
, BOOL predicate
)
411 static const enum wined3d_query_type query_type_map
[] =
413 /* D3D11_QUERY_EVENT */ WINED3D_QUERY_TYPE_EVENT
,
414 /* D3D11_QUERY_OCCLUSION */ WINED3D_QUERY_TYPE_OCCLUSION
,
415 /* D3D11_QUERY_TIMESTAMP */ WINED3D_QUERY_TYPE_TIMESTAMP
,
416 /* D3D11_QUERY_TIMESTAMP_DISJOINT */ WINED3D_QUERY_TYPE_TIMESTAMP_DISJOINT
,
417 /* D3D11_QUERY_PIPELINE_STATISTICS */ WINED3D_QUERY_TYPE_PIPELINE_STATISTICS
,
418 /* D3D11_QUERY_OCCLUSION_PREDICATE */ WINED3D_QUERY_TYPE_OCCLUSION
,
419 /* D3D11_QUERY_SO_STATISTICS */ WINED3D_QUERY_TYPE_SO_STATISTICS
,
420 /* D3D11_QUERY_SO_OVERFLOW_PREDICATE */ WINED3D_QUERY_TYPE_SO_OVERFLOW
,
421 /* D3D11_QUERY_SO_STATISTICS_STREAM0 */ WINED3D_QUERY_TYPE_SO_STATISTICS_STREAM0
,
422 /* D3D11_QUERY_SO_OVERFLOW_PREDICATE_STREAM0 */ WINED3D_QUERY_TYPE_SO_OVERFLOW_STREAM0
,
423 /* D3D11_QUERY_SO_STATISTICS_STREAM1 */ WINED3D_QUERY_TYPE_SO_STATISTICS_STREAM1
,
424 /* D3D11_QUERY_SO_OVERFLOW_PREDICATE_STREAM1 */ WINED3D_QUERY_TYPE_SO_OVERFLOW_STREAM1
,
425 /* D3D11_QUERY_SO_STATISTICS_STREAM2 */ WINED3D_QUERY_TYPE_SO_STATISTICS_STREAM2
,
426 /* D3D11_QUERY_SO_OVERFLOW_PREDICATE_STREAM2 */ WINED3D_QUERY_TYPE_SO_OVERFLOW_STREAM2
,
427 /* D3D11_QUERY_SO_STATISTICS_STREAM3 */ WINED3D_QUERY_TYPE_SO_STATISTICS_STREAM3
,
428 /* D3D11_QUERY_SO_OVERFLOW_PREDICATE_STREAM3 */ WINED3D_QUERY_TYPE_SO_OVERFLOW_STREAM3
,
431 if (desc
->Query
>= ARRAY_SIZE(query_type_map
))
433 FIXME("Unhandled query type %#x.\n", desc
->Query
);
438 FIXME("Ignoring MiscFlags %#x.\n", desc
->MiscFlags
);
440 query
->ID3D11Query_iface
.lpVtbl
= &d3d11_query_vtbl
;
441 query
->ID3D10Query_iface
.lpVtbl
= &d3d10_query_vtbl
;
446 wined3d_mutex_lock();
447 wined3d_private_store_init(&query
->private_store
);
449 if (FAILED(hr
= wined3d_query_create(device
->wined3d_device
, query_type_map
[desc
->Query
],
450 query
, &d3d_query_wined3d_parent_ops
, &query
->wined3d_query
)))
452 WARN("Failed to create wined3d query, hr %#lx.\n", hr
);
453 wined3d_private_store_cleanup(&query
->private_store
);
454 wined3d_mutex_unlock();
457 wined3d_mutex_unlock();
459 query
->predicate
= predicate
;
460 ID3D11Device2_AddRef(query
->device
= &device
->ID3D11Device2_iface
);
465 HRESULT
d3d_query_create(struct d3d_device
*device
, const D3D11_QUERY_DESC
*desc
, BOOL predicate
,
466 struct d3d_query
**query
)
468 struct d3d_query
*object
;
469 BOOL is_predicate_type
;
475 is_predicate_type
= desc
->Query
== D3D11_QUERY_OCCLUSION_PREDICATE
476 || desc
->Query
== D3D11_QUERY_SO_OVERFLOW_PREDICATE
477 || desc
->Query
== D3D11_QUERY_SO_OVERFLOW_PREDICATE_STREAM0
478 || desc
->Query
== D3D11_QUERY_SO_OVERFLOW_PREDICATE_STREAM1
479 || desc
->Query
== D3D11_QUERY_SO_OVERFLOW_PREDICATE_STREAM2
480 || desc
->Query
== D3D11_QUERY_SO_OVERFLOW_PREDICATE_STREAM3
;
482 if (!is_predicate_type
&& predicate
)
484 WARN("Query type %u is not a predicate.\n", desc
->Query
);
488 if (is_predicate_type
)
491 if (!(object
= calloc(1, sizeof(*object
))))
492 return E_OUTOFMEMORY
;
494 if (FAILED(hr
= d3d_query_init(object
, device
, desc
, predicate
)))
496 WARN("Failed to initialise predicate, hr %#lx.\n", hr
);
501 TRACE("Created query %p.\n", object
);