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