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         wined3d_mutex_lock();
 138         wined3d_vertex_declaration_incref(layout->wined3d_decl);
 139         wined3d_mutex_unlock();
 140     }
 141
 142     return refcount;
 143 }
 144
 145 static ULONG STDMETHODCALLTYPE d3d11_input_layout_Release(ID3D11InputLayout *iface)
 146 {
 147     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 148     ULONG refcount = InterlockedDecrement(&layout->refcount);
 149
 150     TRACE("%p decreasing refcount to %u.\n", layout, refcount);
 151
 152     if (!refcount)
 153     {
 154         ID3D11Device *device = layout->device;
 155
 156         wined3d_mutex_lock();
 157         wined3d_vertex_declaration_decref(layout->wined3d_decl);
 158         wined3d_mutex_unlock();
 159
 160         ID3D11Device_Release(device);
 161     }
 162
 163     return refcount;
 164 }
 165
 166 static void STDMETHODCALLTYPE d3d11_input_layout_GetDevice(ID3D11InputLayout *iface,
 167         ID3D11Device **device)
 168 {
 169     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 170
 171     TRACE("iface %p, device %p.\n", iface, device);
 172
 173     ID3D11Device_AddRef(*device = layout->device);
 174 }
 175
 176 static HRESULT STDMETHODCALLTYPE d3d11_input_layout_GetPrivateData(ID3D11InputLayout *iface,
 177         REFGUID guid, UINT *data_size, void *data)
 178 {
 179     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 180
 181     TRACE("iface %p, guid %s, data_size %p, data %p.\n", iface, debugstr_guid(guid), data_size, data);
 182
 183     return d3d_get_private_data(&layout->private_store, guid, data_size, data);
 184 }
 185
 186 static HRESULT STDMETHODCALLTYPE d3d11_input_layout_SetPrivateData(ID3D11InputLayout *iface,
 187         REFGUID guid, UINT data_size, const void *data)
 188 {
 189     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 190
 191     TRACE("iface %p, guid %s, data_size %u, data %p.\n", iface, debugstr_guid(guid), data_size, data);
 192
 193     return d3d_set_private_data(&layout->private_store, guid, data_size, data);
 194 }
 195
 196 static HRESULT STDMETHODCALLTYPE d3d11_input_layout_SetPrivateDataInterface(ID3D11InputLayout *iface,
 197         REFGUID guid, const IUnknown *data)
 198 {
 199     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 200
 201     TRACE("iface %p, guid %s, data %p.\n", iface, debugstr_guid(guid), data);
 202
 203     return d3d_set_private_data_interface(&layout->private_store, guid, data);
 204 }
 205
 206 static const struct ID3D11InputLayoutVtbl d3d11_input_layout_vtbl =
 207 {
 208     /* IUnknown methods */
 209     d3d11_input_layout_QueryInterface,
 210     d3d11_input_layout_AddRef,
 211     d3d11_input_layout_Release,
 212     /* ID3D11DeviceChild methods */
 213     d3d11_input_layout_GetDevice,
 214     d3d11_input_layout_GetPrivateData,
 215     d3d11_input_layout_SetPrivateData,
 216     d3d11_input_layout_SetPrivateDataInterface,
 217 };
 218
 219 /* ID3D10InputLayout methods */
 220
 221 static inline struct d3d_input_layout *impl_from_ID3D10InputLayout(ID3D10InputLayout *iface)
 222 {
 223     return CONTAINING_RECORD(iface, struct d3d_input_layout, ID3D10InputLayout_iface);
 224 }
 225
 226 /* IUnknown methods */
 227
 228 static HRESULT STDMETHODCALLTYPE d3d10_input_layout_QueryInterface(ID3D10InputLayout *iface,
 229         REFIID riid, void **object)
 230 {
 231     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 232
 233     TRACE("iface %p, riid %s, object %p.\n", iface, debugstr_guid(riid), object);
 234
 235     return d3d11_input_layout_QueryInterface(&layout->ID3D11InputLayout_iface, riid, object);
 236 }
 237
 238 static ULONG STDMETHODCALLTYPE d3d10_input_layout_AddRef(ID3D10InputLayout *iface)
 239 {
 240     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 241
 242     TRACE("iface %p.\n", iface);
 243
 244     return d3d11_input_layout_AddRef(&layout->ID3D11InputLayout_iface);
 245 }
 246
 247 static ULONG STDMETHODCALLTYPE d3d10_input_layout_Release(ID3D10InputLayout *iface)
 248 {
 249     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 250
 251     TRACE("iface %p.\n", iface);
 252
 253     return d3d11_input_layout_Release(&layout->ID3D11InputLayout_iface);
 254 }
 255
 256 /* ID3D10DeviceChild methods */
 257
 258 static void STDMETHODCALLTYPE d3d10_input_layout_GetDevice(ID3D10InputLayout *iface, ID3D10Device **device)
 259 {
 260     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 261
 262     TRACE("iface %p, device %p.\n", iface, device);
 263
 264     ID3D11Device_QueryInterface(layout->device, &IID_ID3D10Device, (void **)device);
 265 }
 266
 267 static HRESULT STDMETHODCALLTYPE d3d10_input_layout_GetPrivateData(ID3D10InputLayout *iface,
 268         REFGUID guid, UINT *data_size, void *data)
 269 {
 270     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 271
 272     TRACE("iface %p, guid %s, data_size %p, data %p.\n",
 273             iface, debugstr_guid(guid), data_size, data);
 274
 275     return d3d_get_private_data(&layout->private_store, guid, data_size, data);
 276 }
 277
 278 static HRESULT STDMETHODCALLTYPE d3d10_input_layout_SetPrivateData(ID3D10InputLayout *iface,
 279         REFGUID guid, UINT data_size, const void *data)
 280 {
 281     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 282
 283     TRACE("iface %p, guid %s, data_size %u, data %p.\n",
 284             iface, debugstr_guid(guid), data_size, data);
 285
 286     return d3d_set_private_data(&layout->private_store, guid, data_size, data);
 287 }
 288
 289 static HRESULT STDMETHODCALLTYPE d3d10_input_layout_SetPrivateDataInterface(ID3D10InputLayout *iface,
 290         REFGUID guid, const IUnknown *data)
 291 {
 292     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 293
 294     TRACE("iface %p, guid %s, data %p.\n", iface, debugstr_guid(guid), data);
 295
 296     return d3d_set_private_data_interface(&layout->private_store, guid, data);
 297 }
 298
 299 static const struct ID3D10InputLayoutVtbl d3d10_input_layout_vtbl =
 300 {
 301     /* IUnknown methods */
 302     d3d10_input_layout_QueryInterface,
 303     d3d10_input_layout_AddRef,
 304     d3d10_input_layout_Release,
 305     /* ID3D10DeviceChild methods */
 306     d3d10_input_layout_GetDevice,
 307     d3d10_input_layout_GetPrivateData,
 308     d3d10_input_layout_SetPrivateData,
 309     d3d10_input_layout_SetPrivateDataInterface,
 310 };
 311
 312 static void STDMETHODCALLTYPE d3d_input_layout_wined3d_object_destroyed(void *parent)
 313 {
 314     struct d3d_input_layout *layout = parent;
 315
 316     wined3d_private_store_cleanup(&layout->private_store);
 317     HeapFree(GetProcessHeap(), 0, parent);
 318 }
 319
 320 static const struct wined3d_parent_ops d3d_input_layout_wined3d_parent_ops =
 321 {
 322     d3d_input_layout_wined3d_object_destroyed,
 323 };
 324
 325 static HRESULT d3d_input_layout_init(struct d3d_input_layout *layout, struct d3d_device *device,
 326         const D3D11_INPUT_ELEMENT_DESC *element_descs, UINT element_count,
 327         const void *shader_byte_code, SIZE_T shader_byte_code_length)
 328 {
 329     struct wined3d_vertex_element *wined3d_elements;
 330     HRESULT hr;
 331
 332     layout->ID3D11InputLayout_iface.lpVtbl = &d3d11_input_layout_vtbl;
 333     layout->ID3D10InputLayout_iface.lpVtbl = &d3d10_input_layout_vtbl;
 334     layout->refcount = 1;
 335     wined3d_mutex_lock();
 336     wined3d_private_store_init(&layout->private_store);
 337
 338     if (FAILED(hr = d3d11_input_layout_to_wined3d_declaration(element_descs, element_count,
 339             shader_byte_code, shader_byte_code_length, &wined3d_elements)))
 340     {
 341         WARN("Failed to create wined3d vertex declaration elements, hr %#x.\n", hr);
 342         wined3d_private_store_cleanup(&layout->private_store);
 343         wined3d_mutex_unlock();
 344         return hr;
 345     }
 346
 347     hr = wined3d_vertex_declaration_create(device->wined3d_device, wined3d_elements, element_count,
 348             layout, &d3d_input_layout_wined3d_parent_ops, &layout->wined3d_decl);
 349     HeapFree(GetProcessHeap(), 0, wined3d_elements);
 350     if (FAILED(hr))
 351     {
 352         WARN("Failed to create wined3d vertex declaration, hr %#x.\n", hr);
 353         wined3d_private_store_cleanup(&layout->private_store);
 354         wined3d_mutex_unlock();
 355         return hr;
 356     }
 357     wined3d_mutex_unlock();
 358
 359     ID3D11Device_AddRef(layout->device = &device->ID3D11Device_iface);
 360
 361     return S_OK;
 362 }
 363
 364 HRESULT d3d_input_layout_create(struct d3d_device *device,
 365         const D3D11_INPUT_ELEMENT_DESC *element_descs, UINT element_count,
 366         const void *shader_byte_code, SIZE_T shader_byte_code_length,
 367         struct d3d_input_layout **layout)
 368 {
 369     struct d3d_input_layout *object;
 370     HRESULT hr;
 371
 372     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*object));
 373     if (!object)
 374         return E_OUTOFMEMORY;
 375
 376     if (FAILED(hr = d3d_input_layout_init(object, device, element_descs, element_count,
 377             shader_byte_code, shader_byte_code_length)))
 378     {
 379         WARN("Failed to initialize input layout, hr %#x.\n", hr);
 380         HeapFree(GetProcessHeap(), 0, object);
 381         return hr;
 382     }
 383
 384     TRACE("Created input layout %p.\n", object);
 385     *layout = object;
 386
 387     return S_OK;
 388 }
 389
 390 struct d3d_input_layout *unsafe_impl_from_ID3D11InputLayout(ID3D11InputLayout *iface)
 391 {
 392     if (!iface)
 393         return NULL;
 394     assert(iface->lpVtbl == &d3d11_input_layout_vtbl);
 395
 396     return impl_from_ID3D11InputLayout(iface);
 397 }
 398
 399 struct d3d_input_layout *unsafe_impl_from_ID3D10InputLayout(ID3D10InputLayout *iface)
 400 {
 401     if (!iface)
 402         return NULL;
 403     assert(iface->lpVtbl == &d3d10_input_layout_vtbl);
 404
 405     return impl_from_ID3D10InputLayout(iface);
 406 }