diff --git a/dlls/d3d11/shader.c b/dlls/d3d11/shader.c index 6266a34d198..b8c50d9e96c 100644 --- a/dlls/d3d11/shader.c +++ b/dlls/d3d11/shader.c @@ -635,6 +635,14 @@ static ULONG STDMETHODCALLTYPE d3d11_hull_shader_AddRef(ID3D11HullShader *iface) TRACE("%p increasing refcount to %u.\n", shader, refcount); + if (refcount == 1) + { + ID3D11Device_AddRef(shader->device); + wined3d_mutex_lock(); + wined3d_shader_incref(shader->wined3d_shader); + wined3d_mutex_unlock(); + } + return refcount; } @@ -831,6 +839,14 @@ static ULONG STDMETHODCALLTYPE d3d11_domain_shader_AddRef(ID3D11DomainShader *if TRACE("%p increasing refcount to %u.\n", shader, refcount); + if (refcount == 1) + { + ID3D11Device_AddRef(shader->device); + wined3d_mutex_lock(); + wined3d_shader_incref(shader->wined3d_shader); + wined3d_mutex_unlock(); + } + return refcount; } @@ -1037,6 +1053,14 @@ static ULONG STDMETHODCALLTYPE d3d11_geometry_shader_AddRef(ID3D11GeometryShader TRACE("%p increasing refcount to %u.\n", shader, refcount); + if (refcount == 1) + { + ID3D11Device_AddRef(shader->device); + wined3d_mutex_lock(); + wined3d_shader_incref(shader->wined3d_shader); + wined3d_mutex_unlock(); + } + return refcount; } @@ -1890,6 +1914,14 @@ static ULONG STDMETHODCALLTYPE d3d11_compute_shader_AddRef(ID3D11ComputeShader * TRACE("%p increasing refcount to %u.\n", shader, refcount); + if (refcount == 1) + { + ID3D11Device_AddRef(shader->device); + wined3d_mutex_lock(); + wined3d_shader_incref(shader->wined3d_shader); + wined3d_mutex_unlock(); + } + return refcount; }