dlls/d3d11/shader.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 shdr_handler(const char *data, DWORD data_size, DWORD tag, void *ctx)
  28 {
  29     struct d3d10_shader_info *shader_info = ctx;
  30     HRESULT hr;
  31
  32     switch (tag)
  33     {
  34         case TAG_ISGN:
  35             if (FAILED(hr = shader_parse_signature(data, data_size, shader_info->input_signature)))
  36                 return hr;
  37             break;
  38
  39         case TAG_OSGN:
  40             if (FAILED(hr = shader_parse_signature(data, data_size, shader_info->output_signature)))
  41                 return hr;
  42             break;
  43
  44         case TAG_SHDR:
  45             shader_info->shader_code = (const DWORD *)data;
  46             break;
  47
  48         default:
  49             FIXME("Unhandled chunk %s\n", debugstr_an((const char *)&tag, 4));
  50             break;
  51     }
  52
  53     return S_OK;
  54 }
  55
  56 static HRESULT shader_extract_from_dxbc(const void *dxbc, SIZE_T dxbc_length, struct d3d10_shader_info *shader_info)
  57 {
  58     HRESULT hr;
  59
  60     shader_info->shader_code = NULL;
  61     memset(shader_info->input_signature, 0, sizeof(*shader_info->input_signature));
  62     memset(shader_info->output_signature, 0, sizeof(*shader_info->output_signature));
  63
  64     hr = parse_dxbc(dxbc, dxbc_length, shdr_handler, shader_info);
  65     if (!shader_info->shader_code) hr = E_INVALIDARG;
  66
  67     if (FAILED(hr))
  68     {
  69         ERR("Failed to parse shader, hr %#x\n", hr);
  70         shader_free_signature(shader_info->input_signature);
  71         shader_free_signature(shader_info->output_signature);
  72     }
  73
  74     return hr;
  75 }
  76
  77 HRESULT shader_parse_signature(const char *data, DWORD data_size, struct wined3d_shader_signature *s)
  78 {
  79     struct wined3d_shader_signature_element *e;
  80     const char *ptr = data;
  81     unsigned int i;
  82     DWORD count;
  83
  84     read_dword(&ptr, &count);
  85     TRACE("%u elements\n", count);
  86
  87     skip_dword_unknown(&ptr, 1);
  88
  89     e = HeapAlloc(GetProcessHeap(), 0, count * sizeof(*e));
  90     if (!e)
  91     {
  92         ERR("Failed to allocate input signature memory.\n");
  93         return E_OUTOFMEMORY;
  94     }
  95
  96     for (i = 0; i < count; ++i)
  97     {
  98         UINT name_offset;
  99
 100         read_dword(&ptr, &name_offset);
 101         e[i].semantic_name = data + name_offset;
 102         read_dword(&ptr, &e[i].semantic_idx);
 103         read_dword(&ptr, &e[i].sysval_semantic);
 104         read_dword(&ptr, &e[i].component_type);
 105         read_dword(&ptr, &e[i].register_idx);
 106         read_dword(&ptr, &e[i].mask);
 107
 108         TRACE("semantic: %s, semantic idx: %u, sysval_semantic %#x, "
 109                 "type %u, register idx: %u, use_mask %#x, input_mask %#x\n",
 110                 debugstr_a(e[i].semantic_name), e[i].semantic_idx, e[i].sysval_semantic,
 111                 e[i].component_type, e[i].register_idx, (e[i].mask >> 8) & 0xff, e[i].mask & 0xff);
 112     }
 113
 114     s->elements = e;
 115     s->element_count = count;
 116
 117     return S_OK;
 118 }
 119
 120 void shader_free_signature(struct wined3d_shader_signature *s)
 121 {
 122     HeapFree(GetProcessHeap(), 0, s->elements);
 123 }
 124
 125 static inline struct d3d10_vertex_shader *impl_from_ID3D10VertexShader(ID3D10VertexShader *iface)
 126 {
 127     return CONTAINING_RECORD(iface, struct d3d10_vertex_shader, ID3D10VertexShader_iface);
 128 }
 129
 130 /* IUnknown methods */
 131
 132 static HRESULT STDMETHODCALLTYPE d3d10_vertex_shader_QueryInterface(ID3D10VertexShader *iface,
 133         REFIID riid, void **object)
 134 {
 135     TRACE("iface %p, riid %s, object %p\n", iface, debugstr_guid(riid), object);
 136
 137     if (IsEqualGUID(riid, &IID_ID3D10VertexShader)
 138             || IsEqualGUID(riid, &IID_ID3D10DeviceChild)
 139             || IsEqualGUID(riid, &IID_IUnknown))
 140     {
 141         IUnknown_AddRef(iface);
 142         *object = iface;
 143         return S_OK;
 144     }
 145
 146     WARN("%s not implemented, returning E_NOINTERFACE\n", debugstr_guid(riid));
 147
 148     *object = NULL;
 149     return E_NOINTERFACE;
 150 }
 151
 152 static ULONG STDMETHODCALLTYPE d3d10_vertex_shader_AddRef(ID3D10VertexShader *iface)
 153 {
 154     struct d3d10_vertex_shader *This = impl_from_ID3D10VertexShader(iface);
 155     ULONG refcount = InterlockedIncrement(&This->refcount);
 156
 157     TRACE("%p increasing refcount to %u\n", This, refcount);
 158
 159     if (refcount == 1)
 160     {
 161         ID3D10Device1_AddRef(This->device);
 162         wined3d_mutex_lock();
 163         wined3d_shader_incref(This->wined3d_shader);
 164         wined3d_mutex_unlock();
 165     }
 166
 167     return refcount;
 168 }
 169
 170 static ULONG STDMETHODCALLTYPE d3d10_vertex_shader_Release(ID3D10VertexShader *iface)
 171 {
 172     struct d3d10_vertex_shader *This = impl_from_ID3D10VertexShader(iface);
 173     ULONG refcount = InterlockedDecrement(&This->refcount);
 174
 175     TRACE("%p decreasing refcount to %u\n", This, refcount);
 176
 177     if (!refcount)
 178     {
 179         ID3D10Device1 *device = This->device;
 180
 181         wined3d_mutex_lock();
 182         wined3d_shader_decref(This->wined3d_shader);
 183         wined3d_mutex_unlock();
 184         /* Release the device last, it may cause the wined3d device to be
 185          * destroyed. */
 186         ID3D10Device1_Release(device);
 187     }
 188
 189     return refcount;
 190 }
 191
 192 /* ID3D10DeviceChild methods */
 193
 194 static void STDMETHODCALLTYPE d3d10_vertex_shader_GetDevice(ID3D10VertexShader *iface, ID3D10Device **device)
 195 {
 196     struct d3d10_vertex_shader *shader = impl_from_ID3D10VertexShader(iface);
 197
 198     TRACE("iface %p, device %p.\n", iface, device);
 199
 200     *device = (ID3D10Device *)shader->device;
 201     ID3D10Device_AddRef(*device);
 202 }
 203
 204 static HRESULT STDMETHODCALLTYPE d3d10_vertex_shader_GetPrivateData(ID3D10VertexShader *iface,
 205         REFGUID guid, UINT *data_size, void *data)
 206 {
 207     struct d3d10_vertex_shader *shader = impl_from_ID3D10VertexShader(iface);
 208
 209     TRACE("iface %p, guid %s, data_size %p, data %p.\n",
 210             iface, debugstr_guid(guid), data_size, data);
 211
 212     return d3d10_get_private_data(&shader->private_store, guid, data_size, data);
 213 }
 214
 215 static HRESULT STDMETHODCALLTYPE d3d10_vertex_shader_SetPrivateData(ID3D10VertexShader *iface,
 216         REFGUID guid, UINT data_size, const void *data)
 217 {
 218     struct d3d10_vertex_shader *shader = impl_from_ID3D10VertexShader(iface);
 219
 220     TRACE("iface %p, guid %s, data_size %u, data %p.\n",
 221             iface, debugstr_guid(guid), data_size, data);
 222
 223     return d3d10_set_private_data(&shader->private_store, guid, data_size, data);
 224 }
 225
 226 static HRESULT STDMETHODCALLTYPE d3d10_vertex_shader_SetPrivateDataInterface(ID3D10VertexShader *iface,
 227         REFGUID guid, const IUnknown *data)
 228 {
 229     struct d3d10_vertex_shader *shader = impl_from_ID3D10VertexShader(iface);
 230
 231     TRACE("iface %p, guid %s, data %p.\n", iface, debugstr_guid(guid), data);
 232
 233     return d3d10_set_private_data_interface(&shader->private_store, guid, data);
 234 }
 235
 236 static const struct ID3D10VertexShaderVtbl d3d10_vertex_shader_vtbl =
 237 {
 238     /* IUnknown methods */
 239     d3d10_vertex_shader_QueryInterface,
 240     d3d10_vertex_shader_AddRef,
 241     d3d10_vertex_shader_Release,
 242     /* ID3D10DeviceChild methods */
 243     d3d10_vertex_shader_GetDevice,
 244     d3d10_vertex_shader_GetPrivateData,
 245     d3d10_vertex_shader_SetPrivateData,
 246     d3d10_vertex_shader_SetPrivateDataInterface,
 247 };
 248
 249 static void STDMETHODCALLTYPE d3d10_vertex_shader_wined3d_object_destroyed(void *parent)
 250 {
 251     struct d3d10_vertex_shader *shader = parent;
 252
 253     wined3d_private_store_cleanup(&shader->private_store);
 254     HeapFree(GetProcessHeap(), 0, parent);
 255 }
 256
 257 static const struct wined3d_parent_ops d3d10_vertex_shader_wined3d_parent_ops =
 258 {
 259     d3d10_vertex_shader_wined3d_object_destroyed,
 260 };
 261
 262 HRESULT d3d10_vertex_shader_init(struct d3d10_vertex_shader *shader, struct d3d10_device *device,
 263         const void *byte_code, SIZE_T byte_code_length)
 264 {
 265     struct wined3d_shader_signature output_signature;
 266     struct wined3d_shader_signature input_signature;
 267     struct d3d10_shader_info shader_info;
 268     struct wined3d_shader_desc desc;
 269     HRESULT hr;
 270
 271     shader->ID3D10VertexShader_iface.lpVtbl = &d3d10_vertex_shader_vtbl;
 272     shader->refcount = 1;
 273     wined3d_mutex_lock();
 274     wined3d_private_store_init(&shader->private_store);
 275
 276     shader_info.input_signature = &input_signature;
 277     shader_info.output_signature = &output_signature;
 278     if (FAILED(hr = shader_extract_from_dxbc(byte_code, byte_code_length, &shader_info)))
 279     {
 280         ERR("Failed to extract shader, hr %#x.\n", hr);
 281         wined3d_private_store_cleanup(&shader->private_store);
 282         wined3d_mutex_unlock();
 283         return hr;
 284     }
 285
 286     desc.byte_code = shader_info.shader_code;
 287     desc.input_signature = &input_signature;
 288     desc.output_signature = &output_signature;
 289     desc.max_version = 4;
 290
 291     hr = wined3d_shader_create_vs(device->wined3d_device, &desc, shader,
 292             &d3d10_vertex_shader_wined3d_parent_ops, &shader->wined3d_shader);
 293     shader_free_signature(&input_signature);
 294     shader_free_signature(&output_signature);
 295     if (FAILED(hr))
 296     {
 297         WARN("Failed to create wined3d vertex shader, hr %#x.\n", hr);
 298         wined3d_private_store_cleanup(&shader->private_store);
 299         wined3d_mutex_unlock();
 300         return E_INVALIDARG;
 301     }
 302     wined3d_mutex_unlock();
 303
 304     shader->device = &device->ID3D10Device1_iface;
 305     ID3D10Device1_AddRef(shader->device);
 306
 307     return S_OK;
 308 }
 309
 310 struct d3d10_vertex_shader *unsafe_impl_from_ID3D10VertexShader(ID3D10VertexShader *iface)
 311 {
 312     if (!iface)
 313         return NULL;
 314     assert(iface->lpVtbl == &d3d10_vertex_shader_vtbl);
 315
 316     return impl_from_ID3D10VertexShader(iface);
 317 }
 318
 319 static inline struct d3d10_geometry_shader *impl_from_ID3D10GeometryShader(ID3D10GeometryShader *iface)
 320 {
 321     return CONTAINING_RECORD(iface, struct d3d10_geometry_shader, ID3D10GeometryShader_iface);
 322 }
 323
 324 /* IUnknown methods */
 325
 326 static HRESULT STDMETHODCALLTYPE d3d10_geometry_shader_QueryInterface(ID3D10GeometryShader *iface,
 327         REFIID riid, void **object)
 328 {
 329     TRACE("iface %p, riid %s, object %p\n", iface, debugstr_guid(riid), object);
 330
 331     if (IsEqualGUID(riid, &IID_ID3D10GeometryShader)
 332             || IsEqualGUID(riid, &IID_ID3D10DeviceChild)
 333             || IsEqualGUID(riid, &IID_IUnknown))
 334     {
 335         IUnknown_AddRef(iface);
 336         *object = iface;
 337         return S_OK;
 338     }
 339
 340     WARN("%s not implemented, returning E_NOINTERFACE\n", debugstr_guid(riid));
 341
 342     *object = NULL;
 343     return E_NOINTERFACE;
 344 }
 345
 346 static ULONG STDMETHODCALLTYPE d3d10_geometry_shader_AddRef(ID3D10GeometryShader *iface)
 347 {
 348     struct d3d10_geometry_shader *This = impl_from_ID3D10GeometryShader(iface);
 349     ULONG refcount = InterlockedIncrement(&This->refcount);
 350
 351     TRACE("%p increasing refcount to %u\n", This, refcount);
 352
 353     return refcount;
 354 }
 355
 356 static ULONG STDMETHODCALLTYPE d3d10_geometry_shader_Release(ID3D10GeometryShader *iface)
 357 {
 358     struct d3d10_geometry_shader *This = impl_from_ID3D10GeometryShader(iface);
 359     ULONG refcount = InterlockedDecrement(&This->refcount);
 360
 361     TRACE("%p decreasing refcount to %u\n", This, refcount);
 362
 363     if (!refcount)
 364     {
 365         wined3d_mutex_lock();
 366         wined3d_shader_decref(This->wined3d_shader);
 367         wined3d_mutex_unlock();
 368     }
 369
 370     return refcount;
 371 }
 372
 373 /* ID3D10DeviceChild methods */
 374
 375 static void STDMETHODCALLTYPE d3d10_geometry_shader_GetDevice(ID3D10GeometryShader *iface, ID3D10Device **device)
 376 {
 377     FIXME("iface %p, device %p stub!\n", iface, device);
 378 }
 379
 380 static HRESULT STDMETHODCALLTYPE d3d10_geometry_shader_GetPrivateData(ID3D10GeometryShader *iface,
 381         REFGUID guid, UINT *data_size, void *data)
 382 {
 383     struct d3d10_geometry_shader *shader = impl_from_ID3D10GeometryShader(iface);
 384
 385     TRACE("iface %p, guid %s, data_size %p, data %p.\n",
 386             iface, debugstr_guid(guid), data_size, data);
 387
 388     return d3d10_get_private_data(&shader->private_store, guid, data_size, data);
 389 }
 390
 391 static HRESULT STDMETHODCALLTYPE d3d10_geometry_shader_SetPrivateData(ID3D10GeometryShader *iface,
 392         REFGUID guid, UINT data_size, const void *data)
 393 {
 394     struct d3d10_geometry_shader *shader = impl_from_ID3D10GeometryShader(iface);
 395
 396     TRACE("iface %p, guid %s, data_size %u, data %p.\n",
 397             iface, debugstr_guid(guid), data_size, data);
 398
 399     return d3d10_set_private_data(&shader->private_store, guid, data_size, data);
 400 }
 401
 402 static HRESULT STDMETHODCALLTYPE d3d10_geometry_shader_SetPrivateDataInterface(ID3D10GeometryShader *iface,
 403         REFGUID guid, const IUnknown *data)
 404 {
 405     struct d3d10_geometry_shader *shader = impl_from_ID3D10GeometryShader(iface);
 406
 407     TRACE("iface %p, guid %s, data %p.\n", iface, debugstr_guid(guid), data);
 408
 409     return d3d10_set_private_data_interface(&shader->private_store, guid, data);
 410 }
 411
 412 static const struct ID3D10GeometryShaderVtbl d3d10_geometry_shader_vtbl =
 413 {
 414     /* IUnknown methods */
 415     d3d10_geometry_shader_QueryInterface,
 416     d3d10_geometry_shader_AddRef,
 417     d3d10_geometry_shader_Release,
 418     /* ID3D10DeviceChild methods */
 419     d3d10_geometry_shader_GetDevice,
 420     d3d10_geometry_shader_GetPrivateData,
 421     d3d10_geometry_shader_SetPrivateData,
 422     d3d10_geometry_shader_SetPrivateDataInterface,
 423 };
 424
 425 static void STDMETHODCALLTYPE d3d10_geometry_shader_wined3d_object_destroyed(void *parent)
 426 {
 427     struct d3d10_geometry_shader *shader = parent;
 428
 429     wined3d_private_store_cleanup(&shader->private_store);
 430     HeapFree(GetProcessHeap(), 0, parent);
 431 }
 432
 433 static const struct wined3d_parent_ops d3d10_geometry_shader_wined3d_parent_ops =
 434 {
 435     d3d10_geometry_shader_wined3d_object_destroyed,
 436 };
 437
 438 HRESULT d3d10_geometry_shader_init(struct d3d10_geometry_shader *shader, struct d3d10_device *device,
 439         const void *byte_code, SIZE_T byte_code_length)
 440 {
 441     struct wined3d_shader_signature output_signature;
 442     struct wined3d_shader_signature input_signature;
 443     struct d3d10_shader_info shader_info;
 444     struct wined3d_shader_desc desc;
 445     HRESULT hr;
 446
 447     shader->ID3D10GeometryShader_iface.lpVtbl = &d3d10_geometry_shader_vtbl;
 448     shader->refcount = 1;
 449     wined3d_mutex_lock();
 450     wined3d_private_store_init(&shader->private_store);
 451
 452     shader_info.input_signature = &input_signature;
 453     shader_info.output_signature = &output_signature;
 454     if (FAILED(hr = shader_extract_from_dxbc(byte_code, byte_code_length, &shader_info)))
 455     {
 456         ERR("Failed to extract shader, hr %#x.\n", hr);
 457         wined3d_private_store_cleanup(&shader->private_store);
 458         wined3d_mutex_unlock();
 459         return hr;
 460     }
 461
 462     desc.byte_code = shader_info.shader_code;
 463     desc.input_signature = &input_signature;
 464     desc.output_signature = &output_signature;
 465     desc.max_version = 4;
 466
 467     hr = wined3d_shader_create_gs(device->wined3d_device, &desc, shader,
 468             &d3d10_geometry_shader_wined3d_parent_ops, &shader->wined3d_shader);
 469     shader_free_signature(&input_signature);
 470     shader_free_signature(&output_signature);
 471     if (FAILED(hr))
 472     {
 473         WARN("Failed to create wined3d geometry shader, hr %#x.\n", hr);
 474         wined3d_private_store_cleanup(&shader->private_store);
 475         wined3d_mutex_unlock();
 476         return E_INVALIDARG;
 477     }
 478     wined3d_mutex_unlock();
 479
 480     return S_OK;
 481 }
 482
 483 struct d3d10_geometry_shader *unsafe_impl_from_ID3D10GeometryShader(ID3D10GeometryShader *iface)
 484 {
 485     if (!iface)
 486         return NULL;
 487     assert(iface->lpVtbl == &d3d10_geometry_shader_vtbl);
 488
 489     return impl_from_ID3D10GeometryShader(iface);
 490 }
 491
 492 static inline struct d3d10_pixel_shader *impl_from_ID3D10PixelShader(ID3D10PixelShader *iface)
 493 {
 494     return CONTAINING_RECORD(iface, struct d3d10_pixel_shader, ID3D10PixelShader_iface);
 495 }
 496
 497 /* IUnknown methods */
 498
 499 static HRESULT STDMETHODCALLTYPE d3d10_pixel_shader_QueryInterface(ID3D10PixelShader *iface,
 500         REFIID riid, void **object)
 501 {
 502     TRACE("iface %p, riid %s, object %p\n", iface, debugstr_guid(riid), object);
 503
 504     if (IsEqualGUID(riid, &IID_ID3D10PixelShader)
 505             || IsEqualGUID(riid, &IID_ID3D10DeviceChild)
 506             || IsEqualGUID(riid, &IID_IUnknown))
 507     {
 508         IUnknown_AddRef(iface);
 509         *object = iface;
 510         return S_OK;
 511     }
 512
 513     WARN("%s not implemented, returning E_NOINTERFACE\n", debugstr_guid(riid));
 514
 515     *object = NULL;
 516     return E_NOINTERFACE;
 517 }
 518
 519 static ULONG STDMETHODCALLTYPE d3d10_pixel_shader_AddRef(ID3D10PixelShader *iface)
 520 {
 521     struct d3d10_pixel_shader *This = impl_from_ID3D10PixelShader(iface);
 522     ULONG refcount = InterlockedIncrement(&This->refcount);
 523
 524     TRACE("%p increasing refcount to %u\n", This, refcount);
 525
 526     if (refcount == 1)
 527     {
 528         ID3D10Device1_AddRef(This->device);
 529         wined3d_mutex_lock();
 530         wined3d_shader_incref(This->wined3d_shader);
 531         wined3d_mutex_unlock();
 532     }
 533
 534     return refcount;
 535 }
 536
 537 static ULONG STDMETHODCALLTYPE d3d10_pixel_shader_Release(ID3D10PixelShader *iface)
 538 {
 539     struct d3d10_pixel_shader *This = impl_from_ID3D10PixelShader(iface);
 540     ULONG refcount = InterlockedDecrement(&This->refcount);
 541
 542     TRACE("%p decreasing refcount to %u\n", This, refcount);
 543
 544     if (!refcount)
 545     {
 546         ID3D10Device1 *device = This->device;
 547
 548         wined3d_mutex_lock();
 549         wined3d_shader_decref(This->wined3d_shader);
 550         wined3d_mutex_unlock();
 551         /* Release the device last, it may cause the wined3d device to be
 552          * destroyed. */
 553         ID3D10Device1_Release(device);
 554     }
 555
 556     return refcount;
 557 }
 558
 559 /* ID3D10DeviceChild methods */
 560
 561 static void STDMETHODCALLTYPE d3d10_pixel_shader_GetDevice(ID3D10PixelShader *iface, ID3D10Device **device)
 562 {
 563     struct d3d10_pixel_shader *shader = impl_from_ID3D10PixelShader(iface);
 564
 565     TRACE("iface %p, device %p.\n", iface, device);
 566
 567     *device = (ID3D10Device *)shader->device;
 568     ID3D10Device_AddRef(*device);
 569 }
 570
 571 static HRESULT STDMETHODCALLTYPE d3d10_pixel_shader_GetPrivateData(ID3D10PixelShader *iface,
 572         REFGUID guid, UINT *data_size, void *data)
 573 {
 574     struct d3d10_pixel_shader *shader = impl_from_ID3D10PixelShader(iface);
 575
 576     TRACE("iface %p, guid %s, data_size %p, data %p.\n",
 577             iface, debugstr_guid(guid), data_size, data);
 578
 579     return d3d10_get_private_data(&shader->private_store, guid, data_size, data);
 580 }
 581
 582 static HRESULT STDMETHODCALLTYPE d3d10_pixel_shader_SetPrivateData(ID3D10PixelShader *iface,
 583         REFGUID guid, UINT data_size, const void *data)
 584 {
 585     struct d3d10_pixel_shader *shader = impl_from_ID3D10PixelShader(iface);
 586
 587     TRACE("iface %p, guid %s, data_size %u, data %p.\n",
 588             iface, debugstr_guid(guid), data_size, data);
 589
 590     return d3d10_set_private_data(&shader->private_store, guid, data_size, data);
 591 }
 592
 593 static HRESULT STDMETHODCALLTYPE d3d10_pixel_shader_SetPrivateDataInterface(ID3D10PixelShader *iface,
 594         REFGUID guid, const IUnknown *data)
 595 {
 596     struct d3d10_pixel_shader *shader = impl_from_ID3D10PixelShader(iface);
 597
 598     TRACE("iface %p, guid %s, data %p.\n", iface, debugstr_guid(guid), data);
 599
 600     return d3d10_set_private_data_interface(&shader->private_store, guid, data);
 601 }
 602
 603 static const struct ID3D10PixelShaderVtbl d3d10_pixel_shader_vtbl =
 604 {
 605     /* IUnknown methods */
 606     d3d10_pixel_shader_QueryInterface,
 607     d3d10_pixel_shader_AddRef,
 608     d3d10_pixel_shader_Release,
 609     /* ID3D10DeviceChild methods */
 610     d3d10_pixel_shader_GetDevice,
 611     d3d10_pixel_shader_GetPrivateData,
 612     d3d10_pixel_shader_SetPrivateData,
 613     d3d10_pixel_shader_SetPrivateDataInterface,
 614 };
 615
 616 static void STDMETHODCALLTYPE d3d10_pixel_shader_wined3d_object_destroyed(void *parent)
 617 {
 618     struct d3d10_pixel_shader *shader = parent;
 619
 620     wined3d_private_store_cleanup(&shader->private_store);
 621     HeapFree(GetProcessHeap(), 0, parent);
 622 }
 623
 624 static const struct wined3d_parent_ops d3d10_pixel_shader_wined3d_parent_ops =
 625 {
 626     d3d10_pixel_shader_wined3d_object_destroyed,
 627 };
 628
 629 HRESULT d3d10_pixel_shader_init(struct d3d10_pixel_shader *shader, struct d3d10_device *device,
 630         const void *byte_code, SIZE_T byte_code_length)
 631 {
 632     struct wined3d_shader_signature output_signature;
 633     struct wined3d_shader_signature input_signature;
 634     struct d3d10_shader_info shader_info;
 635     struct wined3d_shader_desc desc;
 636     HRESULT hr;
 637
 638     shader->ID3D10PixelShader_iface.lpVtbl = &d3d10_pixel_shader_vtbl;
 639     shader->refcount = 1;
 640     wined3d_mutex_lock();
 641     wined3d_private_store_init(&shader->private_store);
 642
 643     shader_info.input_signature = &input_signature;
 644     shader_info.output_signature = &output_signature;
 645     if (FAILED(hr = shader_extract_from_dxbc(byte_code, byte_code_length, &shader_info)))
 646     {
 647         ERR("Failed to extract shader, hr %#x.\n", hr);
 648         wined3d_private_store_cleanup(&shader->private_store);
 649         wined3d_mutex_unlock();
 650         return hr;
 651     }
 652
 653     desc.byte_code = shader_info.shader_code;
 654     desc.input_signature = &input_signature;
 655     desc.output_signature = &output_signature;
 656     desc.max_version = 4;
 657
 658     hr = wined3d_shader_create_ps(device->wined3d_device, &desc, shader,
 659             &d3d10_pixel_shader_wined3d_parent_ops, &shader->wined3d_shader);
 660     shader_free_signature(&input_signature);
 661     shader_free_signature(&output_signature);
 662     if (FAILED(hr))
 663     {
 664         WARN("Failed to create wined3d pixel shader, hr %#x.\n", hr);
 665         wined3d_private_store_cleanup(&shader->private_store);
 666         wined3d_mutex_unlock();
 667         return E_INVALIDARG;
 668     }
 669     wined3d_mutex_unlock();
 670
 671     shader->device = &device->ID3D10Device1_iface;
 672     ID3D10Device1_AddRef(shader->device);
 673
 674     return S_OK;
 675 }
 676
 677 struct d3d10_pixel_shader *unsafe_impl_from_ID3D10PixelShader(ID3D10PixelShader *iface)
 678 {
 679     if (!iface)
 680         return NULL;
 681     assert(iface->lpVtbl == &d3d10_pixel_shader_vtbl);
 682
 683     return impl_from_ID3D10PixelShader(iface);
 684 }