[PATCH 7/7] vkd3d-shader/hlsl: Support matrix indexing in the lhs.

[Francisco Casas]

From: Francisco Casas

---
 libs/vkd3d-shader/hlsl.y               | 38 ++++++++++++++++++++++++--
 tests/hlsl-matrix-indexing.shader_test |  6 ++--
 2 files changed, 39 insertions(+), 5 deletions(-)

diff --git a/libs/vkd3d-shader/hlsl.y b/libs/vkd3d-shader/hlsl.y
index a7ffe951..719f4b35 100644
--- a/libs/vkd3d-shader/hlsl.y
+++ b/libs/vkd3d-shader/hlsl.y
@@ -1719,8 +1719,42 @@ static struct hlsl_ir_node *add_assignment(struct hlsl_ctx *ctx, struct list *in
     }
     else if (lhs->type == HLSL_IR_INDEX && hlsl_index_is_crooked_matrix_indexing(hlsl_ir_index(lhs)))
     {
-        hlsl_fixme(ctx, &lhs->loc, "Column-major matrix index store.");
-        return NULL;
+        struct hlsl_ir_node *mat = hlsl_ir_index(lhs)->val.node, *idx = hlsl_ir_index(lhs)->idx.node;
+        unsigned int i, k = 0;
+
+        for (i = 0; i < mat->data_type->dimx; ++i)
+        {
+            struct hlsl_ir_store *store;
+            struct hlsl_ir_constant *c;
+            struct hlsl_ir_load *load;
+            struct hlsl_ir_index *col;
+            struct hlsl_deref deref;
+
+            if (!(writemask & (1 << i)))
+                continue;
+
+            if (!(c = hlsl_new_uint_constant(ctx, i, &lhs->loc)))
+                return NULL;
+            list_add_tail(instrs, &c->node.entry);
+
+            if (!(col = hlsl_new_index(ctx, mat, &c->node, true, &lhs->loc)))
+                return NULL;
+            list_add_tail(instrs, &col->node.entry);
+
+            if (!(load = add_load_component(ctx, instrs, rhs, k++, &rhs->loc)))
+                return NULL;
+
+            if (!hlsl_init_deref_from_index_chain(ctx, &deref, &col->node))
+                return NULL;
+
+            if (!(store = hlsl_new_store_index(ctx, &deref, idx, &load->node, 0, &rhs->loc)))
+            {
+                hlsl_cleanup_deref(&deref);
+                return NULL;
+            }
+            list_add_tail(instrs, &store->node.entry);
+            hlsl_cleanup_deref(&deref);
+        }
     }
     else
     {
diff --git a/tests/hlsl-matrix-indexing.shader_test b/tests/hlsl-matrix-indexing.shader_test
index 4ca0e871..1dc1ede0 100644
--- a/tests/hlsl-matrix-indexing.shader_test
+++ b/tests/hlsl-matrix-indexing.shader_test
@@ -78,7 +78,7 @@ draw quad
 probe all rgba (1.0, 5.0, 7.0, 12.0)

-[pixel shader todo]
+[pixel shader]
 float4 main() : SV_TARGET
 {
     float3x2 m = {1, 2, 3, 4, 5, 6};
@@ -89,8 +89,8 @@ float4 main() : SV_TARGET
 }

[test]
-todo draw quad
-todo probe all rgba (30.0, 40.0, 5.0, 6.0)
+draw quad
+probe all rgba (30.0, 40.0, 5.0, 6.0)

[pixel shader]