Henri Verbeet : d3d10core: Implement d3d10_pixel_shader_SetPrivateData().

[Alexandre Julliard]

Module: wine
Branch: master
Commit: cae40a165361e2fc57eb7891cc4c463fc501ff4d
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=cae40a165361e2fc57eb7891cc4c463fc501ff4d

Author: Henri Verbeet <hverbeet at codeweavers.com>
Date:   Fri Feb 27 12:29:43 2015 +0100

d3d10core: Implement d3d10_pixel_shader_SetPrivateData().

---

 dlls/d3d10core/d3d10core_private.h |  1 +
 dlls/d3d10core/shader.c            | 12 ++++++++++--
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/dlls/d3d10core/d3d10core_private.h b/dlls/d3d10core/d3d10core_private.h
index 8526cf2..1756afc 100644
--- a/dlls/d3d10core/d3d10core_private.h
+++ b/dlls/d3d10core/d3d10core_private.h
@@ -229,6 +229,7 @@ struct d3d10_pixel_shader
     ID3D10PixelShader ID3D10PixelShader_iface;
     LONG refcount;
 
+    struct wined3d_private_store private_store;
     struct wined3d_shader *wined3d_shader;
     ID3D10Device1 *device;
 };
diff --git a/dlls/d3d10core/shader.c b/dlls/d3d10core/shader.c
index dbd6132..cdd0608 100644
--- a/dlls/d3d10core/shader.c
+++ b/dlls/d3d10core/shader.c
@@ -560,10 +560,12 @@ static HRESULT STDMETHODCALLTYPE d3d10_pixel_shader_GetPrivateData(ID3D10PixelSh
 static HRESULT STDMETHODCALLTYPE d3d10_pixel_shader_SetPrivateData(ID3D10PixelShader *iface,
         REFGUID guid, UINT data_size, const void *data)
 {
-    FIXME("iface %p, guid %s, data_size %u, data %p stub!\n",
+    struct d3d10_pixel_shader *shader = impl_from_ID3D10PixelShader(iface);
+
+    TRACE("iface %p, guid %s, data_size %u, data %p.\n",
             iface, debugstr_guid(guid), data_size, data);
 
-    return E_NOTIMPL;
+    return d3d10_set_private_data(&shader->private_store, guid, data_size, data);
 }
 
 static HRESULT STDMETHODCALLTYPE d3d10_pixel_shader_SetPrivateDataInterface(ID3D10PixelShader *iface,
@@ -589,6 +591,9 @@ static const struct ID3D10PixelShaderVtbl d3d10_pixel_shader_vtbl =
 
 static void STDMETHODCALLTYPE d3d10_pixel_shader_wined3d_object_destroyed(void *parent)
 {
+    struct d3d10_pixel_shader *shader = parent;
+
+    wined3d_private_store_cleanup(&shader->private_store);
     HeapFree(GetProcessHeap(), 0, parent);
 }
 
@@ -608,12 +613,14 @@ HRESULT d3d10_pixel_shader_init(struct d3d10_pixel_shader *shader, struct d3d10_
 
     shader->ID3D10PixelShader_iface.lpVtbl = &d3d10_pixel_shader_vtbl;
     shader->refcount = 1;
+    wined3d_private_store_init(&shader->private_store);
 
     shader_info.input_signature = &input_signature;
     shader_info.output_signature = &output_signature;
     if (FAILED(hr = shader_extract_from_dxbc(byte_code, byte_code_length, &shader_info)))
     {
         ERR("Failed to extract shader, hr %#x.\n", hr);
+        wined3d_private_store_cleanup(&shader->private_store);
         return hr;
     }
 
@@ -629,6 +636,7 @@ HRESULT d3d10_pixel_shader_init(struct d3d10_pixel_shader *shader, struct d3d10_
     if (FAILED(hr))
     {
         WARN("Failed to create wined3d pixel shader, hr %#x.\n", hr);
+        wined3d_private_store_cleanup(&shader->private_store);
         return E_INVALIDARG;
     }