Henri Verbeet : vkd3d-shader: Handle non-contiguous destination write masks in vkd3d_dxbc_compiler_emit_store().

[Alexandre Julliard]

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

Author: Henri Verbeet <hverbeet at codeweavers.com>
Date:   Thu Aug 20 23:57:45 2020 +0430

vkd3d-shader: Handle non-contiguous destination write masks in vkd3d_dxbc_compiler_emit_store().

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

---

 libs/vkd3d-shader/spirv.c | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/libs/vkd3d-shader/spirv.c b/libs/vkd3d-shader/spirv.c
index c677963..14ff7f4 100644
--- a/libs/vkd3d-shader/spirv.c
+++ b/libs/vkd3d-shader/spirv.c
@@ -3379,7 +3379,7 @@ static void vkd3d_dxbc_compiler_emit_store(struct vkd3d_dxbc_compiler *compiler,
     struct vkd3d_spirv_builder *builder = &compiler->spirv_builder;
     unsigned int component_count, dst_component_count;
     uint32_t components[VKD3D_VEC4_SIZE];
-    unsigned int i, component_idx;
+    unsigned int i, src_idx, dst_idx;
     uint32_t type_id, dst_val_id;
 
     assert(write_mask);
@@ -3409,12 +3409,15 @@ static void vkd3d_dxbc_compiler_emit_store(struct vkd3d_dxbc_compiler *compiler,
 
         assert(component_count <= ARRAY_SIZE(components));
 
-        for (i = 0, component_idx = 0; i < dst_component_count; ++i)
+        for (i = 0, src_idx = 0, dst_idx = 0; dst_idx < VKD3D_VEC4_SIZE; ++dst_idx)
         {
-            if (write_mask & (VKD3DSP_WRITEMASK_0 << i))
-                components[i] = dst_component_count + component_idx++;
+            if (write_mask & (VKD3DSP_WRITEMASK_0 << dst_idx))
+                components[i] = dst_component_count + src_idx++;
             else
                 components[i] = i;
+
+            if (dst_write_mask & (VKD3DSP_WRITEMASK_0 << dst_idx))
+                ++i;
         }
 
         val_id = vkd3d_spirv_build_op_vector_shuffle(builder,