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     switch(tag)
  32     {
  33         case TAG_ISGN:
  34             return shader_parse_signature(data, data_size, is);
  35
  36         default:
  37             FIXME("Unhandled chunk %s.\n", debugstr_an((const char *)&tag, 4));
  38             return S_OK;
  39     }
  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     if (FAILED(hr = parse_dxbc(shader_byte_code, shader_byte_code_length, isgn_handler, &is)))
  51     {
  52         ERR("Failed to parse input signature.\n");
  53         return E_FAIL;
  54     }
  55
  56     if (!(*wined3d_elements = d3d11_calloc(element_count, sizeof(**wined3d_elements))))
  57     {
  58         ERR("Failed to allocate wined3d vertex element array memory.\n");
  59         HeapFree(GetProcessHeap(), 0, is.elements);
  60         return E_OUTOFMEMORY;
  61     }
  62
  63     for (i = 0; i < element_count; ++i)
  64     {
  65         struct wined3d_vertex_element *e = &(*wined3d_elements)[i];
  66         const D3D11_INPUT_ELEMENT_DESC *f = &element_descs[i];
  67         unsigned int j;
  68
  69         e->format = wined3dformat_from_dxgi_format(f->Format);
  70         e->input_slot = f->InputSlot;
  71         e->offset = f->AlignedByteOffset;
  72         e->output_slot = WINED3D_OUTPUT_SLOT_UNUSED;
  73         e->input_slot_class = f->InputSlotClass;
  74         e->instance_data_step_rate = f->InstanceDataStepRate;
  75         e->method = WINED3D_DECL_METHOD_DEFAULT;
  76         e->usage = 0;
  77         e->usage_idx = 0;
  78
  79         for (j = 0; j < is.element_count; ++j)
  80         {
  81             if (!strcasecmp(element_descs[i].SemanticName, is.elements[j].semantic_name)
  82                     && element_descs[i].SemanticIndex == is.elements[j].semantic_idx)
  83             {
  84                 e->output_slot = is.elements[j].register_idx;
  85                 break;
  86             }
  87         }
  88
  89         if (e->output_slot == WINED3D_OUTPUT_SLOT_UNUSED)
  90             WARN("Unused input element %u.\n", i);
  91     }
  92
  93     shader_free_signature(&is);
  94
  95     return S_OK;
  96 }
  97
  98 /* ID3D11InputLayout methods */
  99
 100 static inline struct d3d_input_layout *impl_from_ID3D11InputLayout(ID3D11InputLayout *iface)
 101 {
 102     return CONTAINING_RECORD(iface, struct d3d_input_layout, ID3D11InputLayout_iface);
 103 }
 104
 105 static HRESULT STDMETHODCALLTYPE d3d11_input_layout_QueryInterface(ID3D11InputLayout *iface,
 106         REFIID riid, void **object)
 107 {
 108     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 109
 110     TRACE("iface %p, riid %s, object %p.\n", iface, debugstr_guid(riid), object);
 111
 112     if (IsEqualGUID(riid, &IID_ID3D11InputLayout)
 113             || IsEqualGUID(riid, &IID_ID3D11DeviceChild)
 114             || IsEqualGUID(riid, &IID_IUnknown))
 115     {
 116         ID3D11InputLayout_AddRef(iface);
 117         *object = iface;
 118         return S_OK;
 119     }
 120
 121     if (IsEqualGUID(riid, &IID_ID3D10InputLayout)
 122             || IsEqualGUID(riid, &IID_ID3D10DeviceChild))
 123     {
 124         ID3D10InputLayout_AddRef(&layout->ID3D10InputLayout_iface);
 125         *object = &layout->ID3D10InputLayout_iface;
 126         return S_OK;
 127     }
 128
 129     WARN("%s not implemented, returning E_NOINTERFACE.\n", debugstr_guid(riid));
 130
 131     *object = NULL;
 132     return E_NOINTERFACE;
 133 }
 134
 135 static ULONG STDMETHODCALLTYPE d3d11_input_layout_AddRef(ID3D11InputLayout *iface)
 136 {
 137     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 138     ULONG refcount = InterlockedIncrement(&layout->refcount);
 139
 140     TRACE("%p increasing refcount to %u.\n", layout, refcount);
 141
 142     if (refcount == 1)
 143     {
 144         wined3d_mutex_lock();
 145         wined3d_vertex_declaration_incref(layout->wined3d_decl);
 146         wined3d_mutex_unlock();
 147     }
 148
 149     return refcount;
 150 }
 151
 152 static ULONG STDMETHODCALLTYPE d3d11_input_layout_Release(ID3D11InputLayout *iface)
 153 {
 154     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 155     ULONG refcount = InterlockedDecrement(&layout->refcount);
 156
 157     TRACE("%p decreasing refcount to %u.\n", layout, refcount);
 158
 159     if (!refcount)
 160     {
 161         wined3d_mutex_lock();
 162         wined3d_vertex_declaration_decref(layout->wined3d_decl);
 163         wined3d_mutex_unlock();
 164     }
 165
 166     return refcount;
 167 }
 168
 169 static void STDMETHODCALLTYPE d3d11_input_layout_GetDevice(ID3D11InputLayout *iface,
 170         ID3D11Device **device)
 171 {
 172     FIXME("iface %p, device %p stub!\n", iface, device);
 173 }
 174
 175 static HRESULT STDMETHODCALLTYPE d3d11_input_layout_GetPrivateData(ID3D11InputLayout *iface,
 176         REFGUID guid, UINT *data_size, void *data)
 177 {
 178     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 179
 180     TRACE("iface %p, guid %s, data_size %p, data %p.\n", iface, debugstr_guid(guid), data_size, data);
 181
 182     return d3d_get_private_data(&layout->private_store, guid, data_size, data);
 183 }
 184
 185 static HRESULT STDMETHODCALLTYPE d3d11_input_layout_SetPrivateData(ID3D11InputLayout *iface,
 186         REFGUID guid, UINT data_size, const void *data)
 187 {
 188     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 189
 190     TRACE("iface %p, guid %s, data_size %u, data %p.\n", iface, debugstr_guid(guid), data_size, data);
 191
 192     return d3d_set_private_data(&layout->private_store, guid, data_size, data);
 193 }
 194
 195 static HRESULT STDMETHODCALLTYPE d3d11_input_layout_SetPrivateDataInterface(ID3D11InputLayout *iface,
 196         REFGUID guid, const IUnknown *data)
 197 {
 198     struct d3d_input_layout *layout = impl_from_ID3D11InputLayout(iface);
 199
 200     TRACE("iface %p, guid %s, data %p.\n", iface, debugstr_guid(guid), data);
 201
 202     return d3d_set_private_data_interface(&layout->private_store, guid, data);
 203 }
 204
 205 static const struct ID3D11InputLayoutVtbl d3d11_input_layout_vtbl =
 206 {
 207     /* IUnknown methods */
 208     d3d11_input_layout_QueryInterface,
 209     d3d11_input_layout_AddRef,
 210     d3d11_input_layout_Release,
 211     /* ID3D11DeviceChild methods */
 212     d3d11_input_layout_GetDevice,
 213     d3d11_input_layout_GetPrivateData,
 214     d3d11_input_layout_SetPrivateData,
 215     d3d11_input_layout_SetPrivateDataInterface,
 216 };
 217
 218 /* ID3D10InputLayout methods */
 219
 220 static inline struct d3d_input_layout *impl_from_ID3D10InputLayout(ID3D10InputLayout *iface)
 221 {
 222     return CONTAINING_RECORD(iface, struct d3d_input_layout, ID3D10InputLayout_iface);
 223 }
 224
 225 /* IUnknown methods */
 226
 227 static HRESULT STDMETHODCALLTYPE d3d10_input_layout_QueryInterface(ID3D10InputLayout *iface,
 228         REFIID riid, void **object)
 229 {
 230     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 231
 232     TRACE("iface %p, riid %s, object %p.\n", iface, debugstr_guid(riid), object);
 233
 234     return d3d11_input_layout_QueryInterface(&layout->ID3D11InputLayout_iface, riid, object);
 235 }
 236
 237 static ULONG STDMETHODCALLTYPE d3d10_input_layout_AddRef(ID3D10InputLayout *iface)
 238 {
 239     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 240
 241     TRACE("iface %p.\n", iface);
 242
 243     return d3d11_input_layout_AddRef(&layout->ID3D11InputLayout_iface);
 244 }
 245
 246 static ULONG STDMETHODCALLTYPE d3d10_input_layout_Release(ID3D10InputLayout *iface)
 247 {
 248     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 249
 250     TRACE("iface %p.\n", iface);
 251
 252     return d3d11_input_layout_Release(&layout->ID3D11InputLayout_iface);
 253 }
 254
 255 /* ID3D10DeviceChild methods */
 256
 257 static void STDMETHODCALLTYPE d3d10_input_layout_GetDevice(ID3D10InputLayout *iface, ID3D10Device **device)
 258 {
 259     FIXME("iface %p, device %p stub!\n", iface, device);
 260 }
 261
 262 static HRESULT STDMETHODCALLTYPE d3d10_input_layout_GetPrivateData(ID3D10InputLayout *iface,
 263         REFGUID guid, UINT *data_size, void *data)
 264 {
 265     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 266
 267     TRACE("iface %p, guid %s, data_size %p, data %p.\n",
 268             iface, debugstr_guid(guid), data_size, data);
 269
 270     return d3d_get_private_data(&layout->private_store, guid, data_size, data);
 271 }
 272
 273 static HRESULT STDMETHODCALLTYPE d3d10_input_layout_SetPrivateData(ID3D10InputLayout *iface,
 274         REFGUID guid, UINT data_size, const void *data)
 275 {
 276     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 277
 278     TRACE("iface %p, guid %s, data_size %u, data %p.\n",
 279             iface, debugstr_guid(guid), data_size, data);
 280
 281     return d3d_set_private_data(&layout->private_store, guid, data_size, data);
 282 }
 283
 284 static HRESULT STDMETHODCALLTYPE d3d10_input_layout_SetPrivateDataInterface(ID3D10InputLayout *iface,
 285         REFGUID guid, const IUnknown *data)
 286 {
 287     struct d3d_input_layout *layout = impl_from_ID3D10InputLayout(iface);
 288
 289     TRACE("iface %p, guid %s, data %p.\n", iface, debugstr_guid(guid), data);
 290
 291     return d3d_set_private_data_interface(&layout->private_store, guid, data);
 292 }
 293
 294 static const struct ID3D10InputLayoutVtbl d3d10_input_layout_vtbl =
 295 {
 296     /* IUnknown methods */
 297     d3d10_input_layout_QueryInterface,
 298     d3d10_input_layout_AddRef,
 299     d3d10_input_layout_Release,
 300     /* ID3D10DeviceChild methods */
 301     d3d10_input_layout_GetDevice,
 302     d3d10_input_layout_GetPrivateData,
 303     d3d10_input_layout_SetPrivateData,
 304     d3d10_input_layout_SetPrivateDataInterface,
 305 };
 306
 307 static void STDMETHODCALLTYPE d3d_input_layout_wined3d_object_destroyed(void *parent)
 308 {
 309     struct d3d_input_layout *layout = parent;
 310
 311     wined3d_private_store_cleanup(&layout->private_store);
 312     HeapFree(GetProcessHeap(), 0, parent);
 313 }
 314
 315 static const struct wined3d_parent_ops d3d_input_layout_wined3d_parent_ops =
 316 {
 317     d3d_input_layout_wined3d_object_destroyed,
 318 };
 319
 320 static HRESULT d3d_input_layout_init(struct d3d_input_layout *layout, struct d3d_device *device,
 321         const D3D11_INPUT_ELEMENT_DESC *element_descs, UINT element_count,
 322         const void *shader_byte_code, SIZE_T shader_byte_code_length)
 323 {
 324     struct wined3d_vertex_element *wined3d_elements;
 325     HRESULT hr;
 326
 327     layout->ID3D11InputLayout_iface.lpVtbl = &d3d11_input_layout_vtbl;
 328     layout->ID3D10InputLayout_iface.lpVtbl = &d3d10_input_layout_vtbl;
 329     layout->refcount = 1;
 330     wined3d_mutex_lock();
 331     wined3d_private_store_init(&layout->private_store);
 332
 333     if (FAILED(hr = d3d11_input_layout_to_wined3d_declaration(element_descs, element_count,
 334             shader_byte_code, shader_byte_code_length, &wined3d_elements)))
 335     {
 336         WARN("Failed to create wined3d vertex declaration elements, hr %#x.\n", hr);
 337         wined3d_private_store_cleanup(&layout->private_store);
 338         wined3d_mutex_unlock();
 339         return hr;
 340     }
 341
 342     hr = wined3d_vertex_declaration_create(device->wined3d_device, wined3d_elements, element_count,
 343             layout, &d3d_input_layout_wined3d_parent_ops, &layout->wined3d_decl);
 344     HeapFree(GetProcessHeap(), 0, wined3d_elements);
 345     if (FAILED(hr))
 346     {
 347         WARN("Failed to create wined3d vertex declaration, hr %#x.\n", hr);
 348         wined3d_private_store_cleanup(&layout->private_store);
 349         wined3d_mutex_unlock();
 350         return hr;
 351     }
 352     wined3d_mutex_unlock();
 353
 354     return S_OK;
 355 }
 356
 357 HRESULT d3d_input_layout_create(struct d3d_device *device,
 358         const D3D11_INPUT_ELEMENT_DESC *element_descs, UINT element_count,
 359         const void *shader_byte_code, SIZE_T shader_byte_code_length,
 360         struct d3d_input_layout **layout)
 361 {
 362     struct d3d_input_layout *object;
 363     HRESULT hr;
 364
 365     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*object));
 366     if (!object)
 367         return E_OUTOFMEMORY;
 368
 369     if (FAILED(hr = d3d_input_layout_init(object, device, element_descs, element_count,
 370             shader_byte_code, shader_byte_code_length)))
 371     {
 372         WARN("Failed to initialize input layout, hr %#x.\n", hr);
 373         HeapFree(GetProcessHeap(), 0, object);
 374         return hr;
 375     }
 376
 377     TRACE("Created input layout %p.\n", object);
 378     *layout = object;
 379
 380     return S_OK;
 381 }
 382
 383 struct d3d_input_layout *unsafe_impl_from_ID3D11InputLayout(ID3D11InputLayout *iface)
 384 {
 385     if (!iface)
 386         return NULL;
 387     assert(iface->lpVtbl == &d3d11_input_layout_vtbl);
 388
 389     return impl_from_ID3D11InputLayout(iface);
 390 }
 391
 392 struct d3d_input_layout *unsafe_impl_from_ID3D10InputLayout(ID3D10InputLayout *iface)
 393 {
 394     if (!iface)
 395         return NULL;
 396     assert(iface->lpVtbl == &d3d10_input_layout_vtbl);
 397
 398     return impl_from_ID3D10InputLayout(iface);
 399 }