dlls/d3d11/async.c

   1 /*
   2  * Copyright 2009 Henri Verbeet for CodeWeavers
   3  * Copyright 2015-2017 Józef Kucia for CodeWeavers
   4  *
   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.
   9  *
  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.
  14  *
  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
  18  *
  19  */
  20
  21 #include "d3d11_private.h"
  22
  23 WINE_DEFAULT_DEBUG_CHANNEL(d3d11);
  24
  25 /* ID3D11Query methods */
  26
  27 static inline struct d3d_query *impl_from_ID3D11Query(ID3D11Query *iface)
  28 {
  29     return CONTAINING_RECORD(iface, struct d3d_query, ID3D11Query_iface);
  30 }
  31
  32 static HRESULT STDMETHODCALLTYPE d3d11_query_QueryInterface(ID3D11Query *iface, REFIID riid, void **object)
  33 {
  34     struct d3d_query *query = impl_from_ID3D11Query(iface);
  35
  36     TRACE("iface %p, riid %s, object %p.\n", iface, debugstr_guid(riid), object);
  37
  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))
  43     {
  44         ID3D11Query_AddRef(iface);
  45         *object = iface;
  46         return S_OK;
  47     }
  48
  49     if ((IsEqualGUID(riid, &IID_ID3D10Predicate) && query->predicate)
  50             || IsEqualGUID(riid, &IID_ID3D10Query)
  51             || IsEqualGUID(riid, &IID_ID3D10Asynchronous)
  52             || IsEqualGUID(riid, &IID_ID3D10DeviceChild))
  53     {
  54         ID3D10Query_AddRef(&query->ID3D10Query_iface);
  55         *object = &query->ID3D10Query_iface;
  56         return S_OK;
  57     }
  58
  59     WARN("%s not implemented, returning E_NOINTERFACE.\n", debugstr_guid(riid));
  60
  61     *object = NULL;
  62     return E_NOINTERFACE;
  63 }
  64
  65 static ULONG STDMETHODCALLTYPE d3d11_query_AddRef(ID3D11Query *iface)
  66 {
  67     struct d3d_query *query = impl_from_ID3D11Query(iface);
  68     ULONG refcount = InterlockedIncrement(&query->refcount);
  69
  70     TRACE("%p increasing refcount to %u.\n", query, refcount);
  71
  72     if (refcount == 1)
  73     {
  74         ID3D11Device2_AddRef(query->device);
  75         wined3d_query_incref(query->wined3d_query);
  76     }
  77
  78     return refcount;
  79 }
  80
  81 static ULONG STDMETHODCALLTYPE d3d11_query_Release(ID3D11Query *iface)
  82 {
  83     struct d3d_query *query = impl_from_ID3D11Query(iface);
  84     ULONG refcount = InterlockedDecrement(&query->refcount);
  85
  86     TRACE("%p decreasing refcount to %u.\n", query, refcount);
  87
  88     if (!refcount)
  89     {
  90         ID3D11Device2 *device = query->device;
  91         wined3d_query_decref(query->wined3d_query);
  92         ID3D11Device2_Release(device);
  93     }
  94
  95     return refcount;
  96 }
  97
  98 static void STDMETHODCALLTYPE d3d11_query_GetDevice(ID3D11Query *iface, ID3D11Device **device)
  99 {
 100     struct d3d_query *query = impl_from_ID3D11Query(iface);
 101
 102     TRACE("iface %p, device %p.\n", iface, device);
 103
 104     *device = (ID3D11Device *)query->device;
 105     ID3D11Device_AddRef(*device);
 106 }
 107
 108 static HRESULT STDMETHODCALLTYPE d3d11_query_GetPrivateData(ID3D11Query *iface,
 109         REFGUID guid, UINT *data_size, void *data)
 110 {
 111     struct d3d_query *query = impl_from_ID3D11Query(iface);
 112
 113     TRACE("iface %p, guid %s, data_size %p, data %p.\n", iface, debugstr_guid(guid), data_size, data);
 114
 115     return d3d_get_private_data(&query->private_store, guid, data_size, data);
 116 }
 117
 118 static HRESULT STDMETHODCALLTYPE d3d11_query_SetPrivateData(ID3D11Query *iface,
 119         REFGUID guid, UINT data_size, const void *data)
 120 {
 121     struct d3d_query *query = impl_from_ID3D11Query(iface);
 122
 123     TRACE("iface %p, guid %s, data_size %u, data %p.\n", iface, debugstr_guid(guid), data_size, data);
 124
 125     return d3d_set_private_data(&query->private_store, guid, data_size, data);
 126 }
 127
 128 static HRESULT STDMETHODCALLTYPE d3d11_query_SetPrivateDataInterface(ID3D11Query *iface,
 129         REFGUID guid, const IUnknown *data)
 130 {
 131     struct d3d_query *query = impl_from_ID3D11Query(iface);
 132
 133     TRACE("iface %p, guid %s, data %p.\n", iface, debugstr_guid(guid), data);
 134
 135     return d3d_set_private_data_interface(&query->private_store, guid, data);
 136 }
 137
 138 static UINT STDMETHODCALLTYPE d3d11_query_GetDataSize(ID3D11Query *iface)
 139 {
 140     struct d3d_query *query = impl_from_ID3D11Query(iface);
 141     unsigned int data_size;
 142
 143     TRACE("iface %p.\n", iface);
 144
 145     wined3d_mutex_lock();
 146     data_size = wined3d_query_get_data_size(query->wined3d_query);
 147     wined3d_mutex_unlock();
 148
 149     return data_size;
 150 }
 151
 152 static void STDMETHODCALLTYPE d3d11_query_GetDesc(ID3D11Query *iface, D3D11_QUERY_DESC *desc)
 153 {
 154     struct d3d_query *query = impl_from_ID3D11Query(iface);
 155
 156     TRACE("iface %p, desc %p.\n", iface, desc);
 157
 158     *desc = query->desc;
 159 }
 160
 161 static const struct ID3D11QueryVtbl d3d11_query_vtbl =
 162 {
 163     /* IUnknown methods */
 164     d3d11_query_QueryInterface,
 165     d3d11_query_AddRef,
 166     d3d11_query_Release,
 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 */
 175     d3d11_query_GetDesc,
 176 };
 177
 178 static void STDMETHODCALLTYPE d3d_query_wined3d_object_destroyed(void *parent)
 179 {
 180     struct d3d_query *query = parent;
 181
 182     wined3d_private_store_cleanup(&query->private_store);
 183     heap_free(parent);
 184 }
 185
 186 static const struct wined3d_parent_ops d3d_query_wined3d_parent_ops =
 187 {
 188     d3d_query_wined3d_object_destroyed,
 189 };
 190
 191 struct d3d_query *unsafe_impl_from_ID3D11Query(ID3D11Query *iface)
 192 {
 193     if (!iface)
 194         return NULL;
 195     assert(iface->lpVtbl == &d3d11_query_vtbl);
 196     return CONTAINING_RECORD(iface, struct d3d_query, ID3D11Query_iface);
 197 }
 198
 199 struct d3d_query *unsafe_impl_from_ID3D11Asynchronous(ID3D11Asynchronous *iface)
 200 {
 201     return unsafe_impl_from_ID3D11Query((ID3D11Query *)iface);
 202 }
 203
 204 /* ID3D10Query methods */
 205
 206 static inline struct d3d_query *impl_from_ID3D10Query(ID3D10Query *iface)
 207 {
 208     return CONTAINING_RECORD(iface, struct d3d_query, ID3D10Query_iface);
 209 }
 210
 211 /* IUnknown methods */
 212
 213 static HRESULT STDMETHODCALLTYPE d3d10_query_QueryInterface(ID3D10Query *iface, REFIID riid, void **object)
 214 {
 215     struct d3d_query *query = impl_from_ID3D10Query(iface);
 216
 217     TRACE("iface %p, riid %s, object %p.\n", iface, debugstr_guid(riid), object);
 218
 219     return d3d11_query_QueryInterface(&query->ID3D11Query_iface, riid, object);
 220 }
 221
 222 static ULONG STDMETHODCALLTYPE d3d10_query_AddRef(ID3D10Query *iface)
 223 {
 224     struct d3d_query *query = impl_from_ID3D10Query(iface);
 225
 226     TRACE("iface %p.\n", iface);
 227
 228     return d3d11_query_AddRef(&query->ID3D11Query_iface);
 229 }
 230
 231 static ULONG STDMETHODCALLTYPE d3d10_query_Release(ID3D10Query *iface)
 232 {
 233     struct d3d_query *query = impl_from_ID3D10Query(iface);
 234
 235     TRACE("iface %p.\n", iface);
 236
 237     return d3d11_query_Release(&query->ID3D11Query_iface);
 238 }
 239
 240 /* ID3D10DeviceChild methods */
 241
 242 static void STDMETHODCALLTYPE d3d10_query_GetDevice(ID3D10Query *iface, ID3D10Device **device)
 243 {
 244     struct d3d_query *query = impl_from_ID3D10Query(iface);
 245
 246     TRACE("iface %p, device %p.\n", iface, device);
 247
 248     ID3D11Device2_QueryInterface(query->device, &IID_ID3D10Device, (void **)device);
 249 }
 250
 251 static HRESULT STDMETHODCALLTYPE d3d10_query_GetPrivateData(ID3D10Query *iface,
 252         REFGUID guid, UINT *data_size, void *data)
 253 {
 254     struct d3d_query *query = impl_from_ID3D10Query(iface);
 255
 256     TRACE("iface %p, guid %s, data_size %p, data %p.\n",
 257             iface, debugstr_guid(guid), data_size, data);
 258
 259     return d3d_get_private_data(&query->private_store, guid, data_size, data);
 260 }
 261
 262 static HRESULT STDMETHODCALLTYPE d3d10_query_SetPrivateData(ID3D10Query *iface,
 263         REFGUID guid, UINT data_size, const void *data)
 264 {
 265     struct d3d_query *query = impl_from_ID3D10Query(iface);
 266
 267     TRACE("iface %p, guid %s, data_size %u, data %p.\n",
 268             iface, debugstr_guid(guid), data_size, data);
 269
 270     return d3d_set_private_data(&query->private_store, guid, data_size, data);
 271 }
 272
 273 static HRESULT STDMETHODCALLTYPE d3d10_query_SetPrivateDataInterface(ID3D10Query *iface,
 274         REFGUID guid, const IUnknown *data)
 275 {
 276     struct d3d_query *query = impl_from_ID3D10Query(iface);
 277
 278     TRACE("iface %p, guid %s, data %p.\n", iface, debugstr_guid(guid), data);
 279
 280     return d3d_set_private_data_interface(&query->private_store, guid, data);
 281 }
 282
 283 /* ID3D10Asynchronous methods */
 284
 285 static void STDMETHODCALLTYPE d3d10_query_Begin(ID3D10Query *iface)
 286 {
 287     struct d3d_query *query = impl_from_ID3D10Query(iface);
 288     HRESULT hr;
 289
 290     TRACE("iface %p.\n", iface);
 291
 292     if (FAILED(hr = wined3d_query_issue(query->wined3d_query, WINED3DISSUE_BEGIN)))
 293         ERR("Failed to issue query, hr %#x.\n", hr);
 294 }
 295
 296 static void STDMETHODCALLTYPE d3d10_query_End(ID3D10Query *iface)
 297 {
 298     struct d3d_query *query = impl_from_ID3D10Query(iface);
 299     HRESULT hr;
 300
 301     TRACE("iface %p.\n", iface);
 302
 303     if (FAILED(hr = wined3d_query_issue(query->wined3d_query, WINED3DISSUE_END)))
 304         ERR("Failed to issue query, hr %#x.\n", hr);
 305 }
 306
 307 static HRESULT STDMETHODCALLTYPE d3d10_query_GetData(ID3D10Query *iface, void *data, UINT data_size, UINT flags)
 308 {
 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;
 313     HRESULT hr;
 314
 315     TRACE("iface %p, data %p, data_size %u, flags %#x.\n", iface, data, data_size, flags);
 316
 317     if (!data && data_size)
 318         return E_INVALIDARG;
 319
 320     if (query->desc.Query == D3D11_QUERY_PIPELINE_STATISTICS
 321             && data_size == sizeof(D3D10_QUERY_DATA_PIPELINE_STATISTICS))
 322     {
 323         data_size = sizeof(D3D11_QUERY_DATA_PIPELINE_STATISTICS);
 324         d3d10_data_pointer = data;
 325         data = &d3d11_data;
 326     }
 327
 328     wined3d_flags = wined3d_getdata_flags_from_d3d11_async_getdata_flags(flags);
 329
 330     wined3d_mutex_lock();
 331     if (!data_size || wined3d_query_get_data_size(query->wined3d_query) == data_size)
 332     {
 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;
 336     }
 337     else
 338     {
 339         WARN("Invalid data size %u.\n", data_size);
 340         hr = E_INVALIDARG;
 341     }
 342     wined3d_mutex_unlock();
 343
 344     if (d3d10_data_pointer && hr == S_OK)
 345         memcpy(d3d10_data_pointer, data, sizeof(D3D10_QUERY_DATA_PIPELINE_STATISTICS));
 346
 347     return hr;
 348 }
 349
 350 static UINT STDMETHODCALLTYPE d3d10_query_GetDataSize(ID3D10Query *iface)
 351 {
 352     struct d3d_query *query = impl_from_ID3D10Query(iface);
 353     unsigned int data_size;
 354
 355     TRACE("iface %p.\n", iface);
 356
 357     wined3d_mutex_lock();
 358     data_size = wined3d_query_get_data_size(query->wined3d_query);
 359     wined3d_mutex_unlock();
 360
 361     if (query->desc.Query == D3D11_QUERY_PIPELINE_STATISTICS)
 362         data_size = sizeof(D3D10_QUERY_DATA_PIPELINE_STATISTICS);
 363
 364     return data_size;
 365 }
 366
 367 /* ID3D10Query methods */
 368
 369 static void STDMETHODCALLTYPE d3d10_query_GetDesc(ID3D10Query *iface, D3D10_QUERY_DESC *desc)
 370 {
 371     struct d3d_query *query = impl_from_ID3D10Query(iface);
 372
 373     TRACE("iface %p, desc %p.\n", iface, desc);
 374
 375     memcpy(desc, &query->desc, sizeof(*desc));
 376 }
 377
 378 static const struct ID3D10QueryVtbl d3d10_query_vtbl =
 379 {
 380     /* IUnknown methods */
 381     d3d10_query_QueryInterface,
 382     d3d10_query_AddRef,
 383     d3d10_query_Release,
 384     /* ID3D10DeviceChild methods */
 385     d3d10_query_GetDevice,
 386     d3d10_query_GetPrivateData,
 387     d3d10_query_SetPrivateData,
 388     d3d10_query_SetPrivateDataInterface,
 389     /* ID3D10Asynchronous methods */
 390     d3d10_query_Begin,
 391     d3d10_query_End,
 392     d3d10_query_GetData,
 393     d3d10_query_GetDataSize,
 394     /* ID3D10Query methods */
 395     d3d10_query_GetDesc,
 396 };
 397
 398 struct d3d_query *unsafe_impl_from_ID3D10Query(ID3D10Query *iface)
 399 {
 400     if (!iface)
 401         return NULL;
 402     assert(iface->lpVtbl == &d3d10_query_vtbl);
 403     return CONTAINING_RECORD(iface, struct d3d_query, ID3D10Query_iface);
 404 }
 405
 406 static HRESULT d3d_query_init(struct d3d_query *query, struct d3d_device *device,
 407         const D3D11_QUERY_DESC *desc, BOOL predicate)
 408 {
 409     HRESULT hr;
 410
 411     static const enum wined3d_query_type query_type_map[] =
 412     {
 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,
 429     };
 430
 431     if (desc->Query >= ARRAY_SIZE(query_type_map))
 432     {
 433         FIXME("Unhandled query type %#x.\n", desc->Query);
 434         return E_INVALIDARG;
 435     }
 436
 437     if (desc->MiscFlags)
 438         FIXME("Ignoring MiscFlags %#x.\n", desc->MiscFlags);
 439
 440     query->ID3D11Query_iface.lpVtbl = &d3d11_query_vtbl;
 441     query->ID3D10Query_iface.lpVtbl = &d3d10_query_vtbl;
 442     query->refcount = 1;
 443
 444     query->desc = *desc;
 445
 446     wined3d_mutex_lock();
 447     wined3d_private_store_init(&query->private_store);
 448
 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)))
 451     {
 452         WARN("Failed to create wined3d query, hr %#x.\n", hr);
 453         wined3d_private_store_cleanup(&query->private_store);
 454         wined3d_mutex_unlock();
 455         return hr;
 456     }
 457     wined3d_mutex_unlock();
 458
 459     query->predicate = predicate;
 460     ID3D11Device2_AddRef(query->device = &device->ID3D11Device2_iface);
 461
 462     return S_OK;
 463 }
 464
 465 HRESULT d3d_query_create(struct d3d_device *device, const D3D11_QUERY_DESC *desc, BOOL predicate,
 466         struct d3d_query **query)
 467 {
 468     struct d3d_query *object;
 469     BOOL is_predicate_type;
 470     HRESULT hr;
 471
 472     if (!desc)
 473         return E_INVALIDARG;
 474
 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;
 481
 482     if (!is_predicate_type && predicate)
 483     {
 484         WARN("Query type %u is not a predicate.\n", desc->Query);
 485         return E_INVALIDARG;
 486     }
 487
 488     if (is_predicate_type)
 489         predicate = TRUE;
 490
 491     if (!(object = heap_alloc_zero(sizeof(*object))))
 492         return E_OUTOFMEMORY;
 493
 494     if (FAILED(hr = d3d_query_init(object, device, desc, predicate)))
 495     {
 496         WARN("Failed to initialize predicate, hr %#x.\n", hr);
 497         heap_free(object);
 498         return hr;
 499     }
 500
 501     TRACE("Created query %p.\n", object);
 502     *query = object;
 503
 504     return S_OK;
 505 }