Zebediah Figura : d3d9: Handle stateblocks in d3d9_device_SetPixelShaderConstantI().

[Alexandre Julliard]

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

Author: Zebediah Figura <z.figura12 at gmail.com>
Date:   Wed Oct  2 19:39:57 2019 -0500

d3d9: Handle stateblocks in d3d9_device_SetPixelShaderConstantI().

Signed-off-by: Zebediah Figura <z.figura12 at gmail.com>
Signed-off-by: Henri Verbeet <hverbeet at codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 dlls/d3d9/device.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/dlls/d3d9/device.c b/dlls/d3d9/device.c
index 539af97fbb..daa0b1b502 100644
--- a/dlls/d3d9/device.c
+++ b/dlls/d3d9/device.c
@@ -3850,8 +3850,11 @@ static HRESULT WINAPI d3d9_device_SetPixelShaderConstantI(IDirect3DDevice9Ex *if
     TRACE("iface %p, reg_idx %u, data %p, count %u.\n", iface, reg_idx, data, count);
 
     wined3d_mutex_lock();
-    hr = wined3d_device_set_ps_consts_i(device->wined3d_device,
+    hr = wined3d_stateblock_set_ps_consts_i(device->update_state,
             reg_idx, count, (const struct wined3d_ivec4 *)data);
+    if (SUCCEEDED(hr) && !device->recording)
+        hr = wined3d_device_set_ps_consts_i(device->wined3d_device,
+                reg_idx, count, (const struct wined3d_ivec4 *)data);
     wined3d_mutex_unlock();
 
     return hr;