dlls/d3d11/inputlayout.c

   1 /*
   2  * Copyright 2009 Henri Verbeet for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  *
  18  */
  19
  20 #include "config.h"
  21 #include "wine/port.h"
  22
  23 #include "d3d11_private.h"
  24
  25 WINE_DEFAULT_DEBUG_CHANNEL(d3d11);
  26
  27 static HRESULT isgn_handler(const char *data, DWORD data_size, DWORD tag, void *ctx)
  28 {
  29     struct wined3d_shader_signature *is = ctx;
  30
  31     if (tag != TAG_ISGN)
  32         return S_OK;
  33
  34     if (is->elements)
  35     {
  36         FIXME("Multiple input signatures.\n");
  37         shader_free_signature(is);
  38     }
  39     return shader_parse_signature(tag, data, data_size, is);
  40 }
  41
  42 static HRESULT d3d11_input_layout_to_wined3d_declaration(const D3D11_INPUT_ELEMENT_DESC *element_descs,
  43         UINT element_count, const void *shader_byte_code, SIZE_T shader_byte_code_length,
  44         struct wined3d_vertex_element **wined3d_elements)
  45 {
  46     struct wined3d_shader_signature is;
  47     unsigned int i;
  48     HRESULT hr;
  49
  50     memset(&is, 0, sizeof(is));
  51     if (FAILED(hr = parse_dxbc(shader_byte_code, shader_byte_code_length, isgn_handler, &is)))
  52     {
  53         ERR("Failed to parse input signature.\n");
  54         return E_FAIL;
  55     }
  56
  57     if (!(*wined3d_elements = d3d11_calloc(element_count, sizeof(**wined3d_elements))))
  58     {
  59         ERR("Failed to allocate wined3d vertex element array memory.\n");
  60         shader_free_signature(&is);
  61         return E_OUTOFMEMORY;
  62     }
  63
  64     for (i = 0; i < element_count; ++i)
  65     {
  66         struct wined3d_vertex_element *e = &(*wined3d_elements)[i];
  67         const D3D11_INPUT_ELEMENT_DESC *f = &element_descs[i];
  68         struct wined3d_shader_signature_element *element;
  69
  70         e->format = wined3dformat_from_dxgi_format(f->Format);
  71         e->input_slot = f->InputSlot;
  72         e->offset = f->AlignedByteOffset;
  73         e->output_slot = WINED3D_OUTPUT_SLOT_UNUSED;
  74         e->input_slot_class = f->InputSlotClass;
  75         e->instance_data_step_rate = f->InstanceDataStepRate;
  76         e->method = WINED3D_DECL_METHOD_DEFAULT;
  77         e->usage = 0;
  78         e->usage_idx = 0;
  79
  80         if ((element = shader_find_signature_element(&is, f->SemanticName, f->SemanticIndex, 0)))
  81             e->output_slot = element->register_idx;
  82         else
  83             WARN("Unused input element %u.\n", i);
  84     }
  85
  86     shader_free_signature(&is);
  87
  88     return S_OK;
  89 }
  90
  91 /* ID3D11InputLayout methods */
  92
  93 static inline struct d3d_input_layout *impl_from_ID3D11InputLayout(ID3D11InputLayout *iface)
  94 {
  95     return CONTAINING_RECORD(iface, struct d3d_input_layout, ID3D11InputLayout_iface);
  96 }
  97
  98 static HRESULT STDMETHODCALLTYPE d3d11_input_layout_QueryInterface(ID3D11InputLayout *iface,
  99         REFIID riid, void **object)
 100 {
 101     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 102
 103     TRACE("iface %p, riid %s, object %p.\n", iface, debugstr_guid(riid), object);
 104
 105     if (IsEqualGUID(riid, &IID_ID3D11InputLayout)
 106             || IsEqualGUID(riid, &IID_ID3D11DeviceChild)
 107             || IsEqualGUID(riid, &IID_IUnknown))
 108     {
 109         ID3D11InputLayout_AddRef(iface);
 110         *object = iface;
 111         return S_OK;
 112     }
 113
 114     if (IsEqualGUID(riid, &IID_ID3D10InputLayout)
 115             || IsEqualGUID(riid, &IID_ID3D10DeviceChild))
 116     {
 117         ID3D10InputLayout_AddRef(&layout->ID3D10InputLayout_iface);
 118         *object = &layout->ID3D10InputLayout_iface;
 119         return S_OK;
 120     }
 121
 122     WARN("%s not implemented, returning E_NOINTERFACE.\n", debugstr_guid(riid));
 123
 124     *object = NULL;
 125     return E_NOINTERFACE;
 126 }
 127
 128 static ULONG STDMETHODCALLTYPE d3d11_input_layout_AddRef(ID3D11InputLayout *iface)
 129 {
 130     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 131     ULONG refcount = InterlockedIncrement(&layout->refcount);
 132
 133     TRACE("%p increasing refcount to %u.\n", layout, refcount);
 134
 135     if (refcount == 1)
 136     {
 137         ID3D11Device_AddRef(layout->device);
 138         wined3d_mutex_lock();
 139         wined3d_vertex_declaration_incref(layout->wined3d_decl);
 140         wined3d_mutex_unlock();
 141     }
 142
 143     return refcount;
 144 }
 145
 146 static ULONG STDMETHODCALLTYPE d3d11_input_layout_Release(ID3D11InputLayout *iface)
 147 {
 148     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 149     ULONG refcount = InterlockedDecrement(&layout->refcount);
 150
 151     TRACE("%p decreasing refcount to %u.\n", layout, refcount);
 152
 153     if (!refcount)
 154     {
 155         ID3D11Device *device = layout->device;
 156
 157         wined3d_mutex_lock();
 158         wined3d_vertex_declaration_decref(layout->wined3d_decl);
 159         wined3d_mutex_unlock();
 160
 161         ID3D11Device_Release(device);
 162     }
 163
 164     return refcount;
 165 }
 166
 167 static void STDMETHODCALLTYPE d3d11_input_layout_GetDevice(ID3D11InputLayout *iface,
 168         ID3D11Device **device)
 169 {
 170     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 171
 172     TRACE("iface %p, device %p.\n", iface, device);
 173
 174     ID3D11Device_AddRef(*device = layout->device);
 175 }
 176
 177 static HRESULT STDMETHODCALLTYPE d3d11_input_layout_GetPrivateData(ID3D11InputLayout *iface,
 178         REFGUID guid, UINT *data_size, void *data)
 179 {
 180     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 181
 182     TRACE("iface %p, guid %s, data_size %p, data %p.\n", iface, debugstr_guid(guid), data_size, data);
 183
 184     return d3d_get_private_data(&layout->private_store, guid, data_size, data);
 185 }
 186
 187 static HRESULT STDMETHODCALLTYPE d3d11_input_layout_SetPrivateData(ID3D11InputLayout *iface,
 188         REFGUID guid, UINT data_size, const void *data)
 189 {
 190     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 191
 192     TRACE("iface %p, guid %s, data_size %u, data %p.\n", iface, debugstr_guid(guid), data_size, data);
 193
 194     return d3d_set_private_data(&layout->private_store, guid, data_size, data);
 195 }
 196
 197 static HRESULT STDMETHODCALLTYPE d3d11_input_layout_SetPrivateDataInterface(ID3D11InputLayout *iface,
 198         REFGUID guid, const IUnknown *data)
 199 {
 200     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 201
 202     TRACE("iface %p, guid %s, data %p.\n", iface, debugstr_guid(guid), data);
 203
 204     return d3d_set_private_data_interface(&layout->private_store, guid, data);
 205 }
 206
 207 static const struct ID3D11InputLayoutVtbl d3d11_input_layout_vtbl =
 208 {
 209     /* IUnknown methods */
 210     d3d11_input_layout_QueryInterface,
 211     d3d11_input_layout_AddRef,
 212     d3d11_input_layout_Release,
 213     /* ID3D11DeviceChild methods */
 214     d3d11_input_layout_GetDevice,
 215     d3d11_input_layout_GetPrivateData,
 216     d3d11_input_layout_SetPrivateData,
 217     d3d11_input_layout_SetPrivateDataInterface,
 218 };
 219
 220 /* ID3D10InputLayout methods */
 221
 222 static inline struct d3d_input_layout *impl_from_ID3D10InputLayout(ID3D10InputLayout *iface)
 223 {
 224     return CONTAINING_RECORD(iface, struct d3d_input_layout, ID3D10InputLayout_iface);
 225 }
 226
 227 /* IUnknown methods */
 228
 229 static HRESULT STDMETHODCALLTYPE d3d10_input_layout_QueryInterface(ID3D10InputLayout *iface,
 230         REFIID riid, void **object)
 231 {
 232     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 233
 234     TRACE("iface %p, riid %s, object %p.\n", iface, debugstr_guid(riid), object);
 235
 236     return d3d11_input_layout_QueryInterface(&layout->ID3D11InputLayout_iface, riid, object);
 237 }
 238
 239 static ULONG STDMETHODCALLTYPE d3d10_input_layout_AddRef(ID3D10InputLayout *iface)
 240 {
 241     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 242
 243     TRACE("iface %p.\n", iface);
 244
 245     return d3d11_input_layout_AddRef(&layout->ID3D11InputLayout_iface);
 246 }
 247
 248 static ULONG STDMETHODCALLTYPE d3d10_input_layout_Release(ID3D10InputLayout *iface)
 249 {
 250     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 251
 252     TRACE("iface %p.\n", iface);
 253
 254     return d3d11_input_layout_Release(&layout->ID3D11InputLayout_iface);
 255 }
 256
 257 /* ID3D10DeviceChild methods */
 258
 259 static void STDMETHODCALLTYPE d3d10_input_layout_GetDevice(ID3D10InputLayout *iface, ID3D10Device **device)
 260 {
 261     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 262
 263     TRACE("iface %p, device %p.\n", iface, device);
 264
 265     ID3D11Device_QueryInterface(layout->device, &IID_ID3D10Device, (void **)device);
 266 }
 267
 268 static HRESULT STDMETHODCALLTYPE d3d10_input_layout_GetPrivateData(ID3D10InputLayout *iface,
 269         REFGUID guid, UINT *data_size, void *data)
 270 {
 271     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 272
 273     TRACE("iface %p, guid %s, data_size %p, data %p.\n",
 274             iface, debugstr_guid(guid), data_size, data);
 275
 276     return d3d_get_private_data(&layout->private_store, guid, data_size, data);
 277 }
 278
 279 static HRESULT STDMETHODCALLTYPE d3d10_input_layout_SetPrivateData(ID3D10InputLayout *iface,
 280         REFGUID guid, UINT data_size, const void *data)
 281 {
 282     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 283
 284     TRACE("iface %p, guid %s, data_size %u, data %p.\n",
 285             iface, debugstr_guid(guid), data_size, data);
 286
 287     return d3d_set_private_data(&layout->private_store, guid, data_size, data);
 288 }
 289
 290 static HRESULT STDMETHODCALLTYPE d3d10_input_layout_SetPrivateDataInterface(ID3D10InputLayout *iface,
 291         REFGUID guid, const IUnknown *data)
 292 {
 293     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 294
 295     TRACE("iface %p, guid %s, data %p.\n", iface, debugstr_guid(guid), data);
 296
 297     return d3d_set_private_data_interface(&layout->private_store, guid, data);
 298 }
 299
 300 static const struct ID3D10InputLayoutVtbl d3d10_input_layout_vtbl =
 301 {
 302     /* IUnknown methods */
 303     d3d10_input_layout_QueryInterface,
 304     d3d10_input_layout_AddRef,
 305     d3d10_input_layout_Release,
 306     /* ID3D10DeviceChild methods */
 307     d3d10_input_layout_GetDevice,
 308     d3d10_input_layout_GetPrivateData,
 309     d3d10_input_layout_SetPrivateData,
 310     d3d10_input_layout_SetPrivateDataInterface,
 311 };
 312
 313 static void STDMETHODCALLTYPE d3d_input_layout_wined3d_object_destroyed(void *parent)
 314 {
 315     struct d3d_input_layout *layout = parent;
 316
 317     wined3d_private_store_cleanup(&layout->private_store);
 318     HeapFree(GetProcessHeap(), 0, parent);
 319 }
 320
 321 static const struct wined3d_parent_ops d3d_input_layout_wined3d_parent_ops =
 322 {
 323     d3d_input_layout_wined3d_object_destroyed,
 324 };
 325
 326 static HRESULT d3d_input_layout_init(struct d3d_input_layout *layout, struct d3d_device *device,
 327         const D3D11_INPUT_ELEMENT_DESC *element_descs, UINT element_count,
 328         const void *shader_byte_code, SIZE_T shader_byte_code_length)
 329 {
 330     struct wined3d_vertex_element *wined3d_elements;
 331     HRESULT hr;
 332
 333     layout->ID3D11InputLayout_iface.lpVtbl = &d3d11_input_layout_vtbl;
 334     layout->ID3D10InputLayout_iface.lpVtbl = &d3d10_input_layout_vtbl;
 335     layout->refcount = 1;
 336     wined3d_mutex_lock();
 337     wined3d_private_store_init(&layout->private_store);
 338
 339     if (FAILED(hr = d3d11_input_layout_to_wined3d_declaration(element_descs, element_count,
 340             shader_byte_code, shader_byte_code_length, &wined3d_elements)))
 341     {
 342         WARN("Failed to create wined3d vertex declaration elements, hr %#x.\n", hr);
 343         wined3d_private_store_cleanup(&layout->private_store);
 344         wined3d_mutex_unlock();
 345         return hr;
 346     }
 347
 348     hr = wined3d_vertex_declaration_create(device->wined3d_device, wined3d_elements, element_count,
 349             layout, &d3d_input_layout_wined3d_parent_ops, &layout->wined3d_decl);
 350     HeapFree(GetProcessHeap(), 0, wined3d_elements);
 351     if (FAILED(hr))
 352     {
 353         WARN("Failed to create wined3d vertex declaration, hr %#x.\n", hr);
 354         wined3d_private_store_cleanup(&layout->private_store);
 355         wined3d_mutex_unlock();
 356         return hr;
 357     }
 358     wined3d_mutex_unlock();
 359
 360     ID3D11Device_AddRef(layout->device = &device->ID3D11Device_iface);
 361
 362     return S_OK;
 363 }
 364
 365 HRESULT d3d_input_layout_create(struct d3d_device *device,
 366         const D3D11_INPUT_ELEMENT_DESC *element_descs, UINT element_count,
 367         const void *shader_byte_code, SIZE_T shader_byte_code_length,
 368         struct d3d_input_layout **layout)
 369 {
 370     struct d3d_input_layout *object;
 371     HRESULT hr;
 372
 373     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*object));
 374     if (!object)
 375         return E_OUTOFMEMORY;
 376
 377     if (FAILED(hr = d3d_input_layout_init(object, device, element_descs, element_count,
 378             shader_byte_code, shader_byte_code_length)))
 379     {
 380         WARN("Failed to initialize input layout, hr %#x.\n", hr);
 381         HeapFree(GetProcessHeap(), 0, object);
 382         return hr;
 383     }
 384
 385     TRACE("Created input layout %p.\n", object);
 386     *layout = object;
 387
 388     return S_OK;
 389 }
 390
 391 struct d3d_input_layout *unsafe_impl_from_ID3D11InputLayout(ID3D11InputLayout *iface)
 392 {
 393     if (!iface)
 394         return NULL;
 395     assert(iface->lpVtbl == &d3d11_input_layout_vtbl);
 396
 397     return impl_from_ID3D11InputLayout(iface);
 398 }
 399
 400 struct d3d_input_layout *unsafe_impl_from_ID3D10InputLayout(ID3D10InputLayout *iface)
 401 {
 402     if (!iface)
 403         return NULL;
 404     assert(iface->lpVtbl == &d3d10_input_layout_vtbl);
 405
 406     return impl_from_ID3D10InputLayout(iface);
 407 }