Ziqing Hui : d2d1: Implement LoadVertexShader().

[Alexandre Julliard]

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

Author: Ziqing Hui <zhui at codeweavers.com>
Date:   Tue May 10 12:48:40 2022 +0800

d2d1: Implement LoadVertexShader().

Signed-off-by: Ziqing Hui <zhui at codeweavers.com>
Signed-off-by: Henri Verbeet <hverbeet at codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 dlls/d2d1/d2d1_private.h | 10 ++++++++++
 dlls/d2d1/effect.c       | 40 ++++++++++++++++++++++++++++++++++++++--
 2 files changed, 48 insertions(+), 2 deletions(-)

diff --git a/dlls/d2d1/d2d1_private.h b/dlls/d2d1/d2d1_private.h
index cb9ee6d5b80..b7e1a12b3b8 100644
--- a/dlls/d2d1/d2d1_private.h
+++ b/dlls/d2d1/d2d1_private.h
@@ -568,12 +568,22 @@ struct d2d_device
 
 void d2d_device_init(struct d2d_device *device, ID2D1Factory1 *factory, IDXGIDevice *dxgi_device) DECLSPEC_HIDDEN;
 
+struct d2d_shader
+{
+    const GUID *id;
+    IUnknown *shader;
+};
+
 struct d2d_effect_context
 {
     ID2D1EffectContext ID2D1EffectContext_iface;
     LONG refcount;
 
     struct d2d_device_context *device_context;
+
+    struct d2d_shader *shaders;
+    size_t shaders_size;
+    size_t shader_count;
 };
 
 void d2d_effect_context_init(struct d2d_effect_context *effect_context,
diff --git a/dlls/d2d1/effect.c b/dlls/d2d1/effect.c
index 90e07c0810b..b87ae6bb77d 100644
--- a/dlls/d2d1/effect.c
+++ b/dlls/d2d1/effect.c
@@ -37,6 +37,15 @@ static inline struct d2d_effect_context *impl_from_ID2D1EffectContext(ID2D1Effec
 
 static void d2d_effect_context_cleanup(struct d2d_effect_context *effect_context)
 {
+    unsigned int i;
+
+    for (i = 0; i < effect_context->shader_count; ++i)
+    {
+        if (effect_context->shaders[i].shader)
+            IUnknown_Release(effect_context->shaders[i].shader);
+    }
+    heap_free(effect_context->shaders);
+
     ID2D1DeviceContext_Release(&effect_context->device_context->ID2D1DeviceContext_iface);
 }
 
@@ -178,10 +187,37 @@ static HRESULT STDMETHODCALLTYPE d2d_effect_context_LoadPixelShader(ID2D1EffectC
 static HRESULT STDMETHODCALLTYPE d2d_effect_context_LoadVertexShader(ID2D1EffectContext *iface,
         REFGUID shader_id, const BYTE *buffer, UINT32 buffer_size)
 {
-    FIXME("iface %p, shader_id %s, buffer %p, buffer_size %u stub!\n",
+    struct d2d_effect_context *effect_context = impl_from_ID2D1EffectContext(iface);
+    ID3D11VertexShader *vertex_shader;
+    struct d2d_shader *shader;
+    HRESULT hr;
+
+    TRACE("iface %p, shader_id %s, buffer %p, buffer_size %u.\n",
             iface, debugstr_guid(shader_id), buffer, buffer_size);
 
-    return E_NOTIMPL;
+    if (FAILED(hr = ID3D11Device1_CreateVertexShader(effect_context->device_context->d3d_device,
+            buffer, buffer_size, NULL, &vertex_shader)))
+    {
+        WARN("Failed to create vertex shader, hr %#lx.\n", hr);
+        return hr;
+    }
+
+    if (!d2d_array_reserve((void **)&effect_context->shaders, &effect_context->shaders_size,
+            effect_context->shader_count + 1, sizeof(*effect_context->shaders)))
+    {
+        ERR("Failed to resize shaders array.\n");
+        ID3D11VertexShader_Release(vertex_shader);
+        return E_OUTOFMEMORY;
+    }
+    memset(effect_context->shaders + effect_context->shader_count, 0,
+            sizeof(*effect_context->shaders) * (effect_context->shaders_size - effect_context->shader_count));
+
+    effect_context->shader_count++;
+    shader = &effect_context->shaders[effect_context->shader_count - 1];
+    shader->id = shader_id;
+    shader->shader = (IUnknown *)vertex_shader;
+
+    return S_OK;
 }
 
 static HRESULT STDMETHODCALLTYPE d2d_effect_context_LoadComputeShader(ID2D1EffectContext *iface,