[PATCH v2 5/6] d3d9: Handle stateblocks in d3d9_device_SetVertexShaderConstantF().

[Zebediah Figura]

Signed-off-by: Zebediah Figura 
---
 dlls/d3d9/device.c | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/dlls/d3d9/device.c b/dlls/d3d9/device.c
index 3e3d5c70500..8436967ddd3 100644
--- a/dlls/d3d9/device.c
+++ b/dlls/d3d9/device.c
@@ -3471,8 +3471,13 @@ static HRESULT WINAPI d3d9_device_SetVertexShaderConstantF(IDirect3DDevice9Ex *i
     }

wined3d_mutex_lock();
-    hr = wined3d_device_set_vs_consts_f(device->wined3d_device,
-            reg_idx, count, (const struct wined3d_vec4 *)data);
+    hr = wined3d_stateblock_set_vs_consts_f(device->update_state, reg_idx,
+            count, (const struct wined3d_vec4 *)data);
+    if (SUCCEEDED(hr) && !device->recording)
+    {
+        hr = wined3d_device_set_vs_consts_f(device->wined3d_device,
+                reg_idx, count, (const struct wined3d_vec4 *)data);
+    }
     wined3d_mutex_unlock();

return hr;