[PATCH vkd3d v2 3/6] vkd3d-shader/hlsl: Support complex numeric initializers.

[Francisco Casas]

Signed-off-by: Francisco Casas <fcasas(a)codeweavers.com>
Signed-off-by: Giovanni Mascellani <gmascellani(a)codeweavers.com>

---
v2:
- Use new name of hlsl_type_is_single_reg(), hlsl_type_uses_writemask().

Signed-off-by: Francisco Casas <fcasas(a)codeweavers.com>
---
 Makefile.am              |  1 -
 libs/vkd3d-shader/hlsl.y | 93 +++++++++++++++++++++++++++++++---------
 2 files changed, 73 insertions(+), 21 deletions(-)

diff --git a/Makefile.am b/Makefile.am
index 8617a09d..3fc803c3 100644
--- a/Makefile.am
+++ b/Makefile.am
@@ -310,7 +310,6 @@ XFAIL_TESTS = \
 	tests/hlsl-initializer-invalid-arg-count.shader_test \
 	tests/hlsl-initializer-local-array.shader_test \
 	tests/hlsl-initializer-nested.shader_test \
-	tests/hlsl-initializer-numeric.shader_test \
 	tests/hlsl-initializer-static-array.shader_test \
 	tests/hlsl-initializer-struct.shader_test \
 	tests/hlsl-bool-cast.shader_test \
diff --git a/libs/vkd3d-shader/hlsl.y b/libs/vkd3d-shader/hlsl.y
index 5e142914..725cbad3 100644
--- a/libs/vkd3d-shader/hlsl.y
+++ b/libs/vkd3d-shader/hlsl.y
@@ -1321,6 +1321,51 @@ static bool add_increment(struct hlsl_ctx *ctx, struct list *instrs, bool decrem
     return true;
 }
 
+static void initialize_numeric_var(struct hlsl_ctx *ctx, struct hlsl_ir_var *var,
+        struct parse_initializer *initializer, unsigned int reg_offset, struct hlsl_type *type,
+        unsigned int *initializer_offset)
+{
+    unsigned int i;
+
+    if (type->type == HLSL_CLASS_MATRIX)
+        hlsl_fixme(ctx, &var->loc, "Matrix initializer.");
+
+    for (i = 0; i < type->dimx; i++)
+    {
+        struct hlsl_ir_store *store;
+        struct hlsl_ir_constant *c;
+        struct hlsl_ir_node *node;
+
+        node = initializer->args[*initializer_offset];
+        *initializer_offset += 1;
+
+        if (!(node = add_implicit_conversion(ctx, initializer->instrs, node,
+                hlsl_get_scalar_type(ctx, type->base_type), &node->loc)))
+            return;
+
+        if (hlsl_type_uses_writemask(var->data_type))
+        {
+            if (!(c = hlsl_new_uint_constant(ctx, reg_offset, node->loc)))
+                return;
+            list_add_tail(initializer->instrs, &c->node.entry);
+
+            if (!(store = hlsl_new_store(ctx, var, &c->node, node, (1u << i), node->loc)))
+                return;
+        }
+        else
+        {
+            if (!(c = hlsl_new_uint_constant(ctx, reg_offset + i, node->loc)))
+                return;
+            list_add_tail(initializer->instrs, &c->node.entry);
+
+            if (!(store = hlsl_new_store(ctx, var, &c->node, node, 0, node->loc)))
+                return;
+        }
+
+        list_add_tail(initializer->instrs, &store->node.entry);
+    }
+}
+
 static void struct_var_initializer(struct hlsl_ctx *ctx, struct list *list, struct hlsl_ir_var *var,
         struct parse_initializer *initializer)
 {
@@ -1490,20 +1535,16 @@ static struct list *declare_vars(struct hlsl_ctx *ctx, struct hlsl_type *basic_t
         if (v->initializer.args_count)
         {
             unsigned int size = initializer_size(&v->initializer);
-            struct hlsl_ir_load *load;
 
-            if (type->type <= HLSL_CLASS_LAST_NUMERIC
-                    && type->dimx * type->dimy != size && size != 1)
+            if (type->type <= HLSL_CLASS_LAST_NUMERIC && v->initializer.args_count != 1
+                    && type->dimx * type->dimy != size)
             {
-                if (size < type->dimx * type->dimy)
-                {
-                    hlsl_error(ctx, &v->loc, VKD3D_SHADER_ERROR_HLSL_WRONG_PARAMETER_COUNT,
-                            "Expected %u components in numeric initializer, but got %u.",
-                            type->dimx * type->dimy, size);
-                    free_parse_initializer(&v->initializer);
-                    vkd3d_free(v);
-                    continue;
-                }
+                hlsl_error(ctx, &v->loc, VKD3D_SHADER_ERROR_HLSL_WRONG_PARAMETER_COUNT,
+                        "Expected %u components in numeric initializer, but got %u.",
+                        type->dimx * type->dimy, v->initializer.args_count);
+                free_parse_initializer(&v->initializer);
+                vkd3d_free(v);
+                continue;
             }
             if ((type->type == HLSL_CLASS_STRUCT || type->type == HLSL_CLASS_ARRAY)
                     && hlsl_type_component_count(type) != size)
@@ -1537,23 +1578,35 @@ static struct list *declare_vars(struct hlsl_ctx *ctx, struct hlsl_type *basic_t
                 vkd3d_free(v);
                 continue;
             }
+
             if (v->initializer.args_count > 1)
             {
-                hlsl_fixme(ctx, &v->loc, "Complex initializer.");
-                free_parse_initializer(&v->initializer);
-                vkd3d_free(v);
-                continue;
+                unsigned int initializer_offset = 0;
+
+                if (v->initializer.args_count != size)
+                {
+                    hlsl_fixme(ctx, &v->loc, "Flatten initializer.");
+                    free_parse_initializer(&v->initializer);
+                    vkd3d_free(v);
+                    continue;
+                }
+
+                initialize_numeric_var(ctx, var, &v->initializer, 0, type, &initializer_offset);
             }
+            else
+            {
+                struct hlsl_ir_load *load = hlsl_new_var_load(ctx, var, var->loc);
 
-            load = hlsl_new_var_load(ctx, var, var->loc);
-            list_add_tail(v->initializer.instrs, &load->node.entry);
-            add_assignment(ctx, v->initializer.instrs, &load->node, ASSIGN_OP_ASSIGN, v->initializer.args[0]);
-            vkd3d_free(v->initializer.args);
+                list_add_tail(v->initializer.instrs, &load->node.entry);
+                add_assignment(ctx, v->initializer.instrs, &load->node, ASSIGN_OP_ASSIGN, v->initializer.args[0]);
+            }
 
             if (modifiers & HLSL_STORAGE_STATIC)
                 list_move_tail(&ctx->static_initializers, v->initializer.instrs);
             else
                 list_move_tail(statements_list, v->initializer.instrs);
+
+            vkd3d_free(v->initializer.args);
             vkd3d_free(v->initializer.instrs);
         }
 
-- 
2.25.1

[Zebediah Figura (she/her)]

On 2/22/22 16:13, Francisco Casas wrote:
> +static void initialize_numeric_var(struct hlsl_ctx *ctx, struct hlsl_ir_var *var,
> +        struct parse_initializer *initializer, unsigned int reg_offset, struct hlsl_type *type,
> +        unsigned int *initializer_offset)
> +{
> +    unsigned int i;
> +
> +    if (type->type == HLSL_CLASS_MATRIX)
> +        hlsl_fixme(ctx, &var->loc, "Matrix initializer.");
> +
> +    for (i = 0; i < type->dimx; i++)
> +    {
> +        struct hlsl_ir_store *store;
> +        struct hlsl_ir_constant *c;
> +        struct hlsl_ir_node *node;
> +
> +        node = initializer->args[*initializer_offset];
> +        *initializer_offset += 1;
> +
> +        if (!(node = add_implicit_conversion(ctx, initializer->instrs, node,
> +                hlsl_get_scalar_type(ctx, type->base_type), &node->loc)))
> +            return;
> +
> +        if (hlsl_type_uses_writemask(var->data_type))
> +        {
> +            if (!(c = hlsl_new_uint_constant(ctx, reg_offset, node->loc)))
> +                return;
> +            list_add_tail(initializer->instrs, &c->node.entry);
> +
> +            if (!(store = hlsl_new_store(ctx, var, &c->node, node, (1u << i), node->loc)))
> +                return;
> +        }
> +        else
> +        {
> +            if (!(c = hlsl_new_uint_constant(ctx, reg_offset + i, node->loc)))
> +                return;
> +            list_add_tail(initializer->instrs, &c->node.entry);
> +
> +            if (!(store = hlsl_new_store(ctx, var, &c->node, node, 0, node->loc)))
> +                return;
> +        }

I'm inclined to say we don't want to be concerned with the writemask 
here. That is, we should just deal in offsets in hlsl.y, in a unified 
manner, and let a later pass lower that to a swizzle or writemask. That 
way we don't have to care in add_array_load() either.

I.e. this should basically just look like the "else" branch.
> +
> +        list_add_tail(initializer->instrs, &store->node.entry);
> +    }
> +}
> +
>   static void struct_var_initializer(struct hlsl_ctx *ctx, struct list *list, struct hlsl_ir_var *var,
>           struct parse_initializer *initializer)
>   {
> @@ -1490,20 +1535,16 @@ static struct list *declare_vars(struct hlsl_ctx *ctx, struct hlsl_type *basic_t
>           if (v->initializer.args_count)
>           {
>               unsigned int size = initializer_size(&v->initializer);
> -            struct hlsl_ir_load *load;
>
> -            if (type->type <= HLSL_CLASS_LAST_NUMERIC
> -                    && type->dimx * type->dimy != size && size != 1)
> +            if (type->type <= HLSL_CLASS_LAST_NUMERIC && v->initializer.args_count != 1
> +                    && type->dimx * type->dimy != size)
>               {
> -                if (size < type->dimx * type->dimy)
> -                {
> -                    hlsl_error(ctx, &v->loc, VKD3D_SHADER_ERROR_HLSL_WRONG_PARAMETER_COUNT,
> -                            "Expected %u components in numeric initializer, but got %u.",
> -                            type->dimx * type->dimy, size);
> -                    free_parse_initializer(&v->initializer);
> -                    vkd3d_free(v);
> -                    continue;
> -                }
> +                hlsl_error(ctx, &v->loc, VKD3D_SHADER_ERROR_HLSL_WRONG_PARAMETER_COUNT,
> +                        "Expected %u components in numeric initializer, but got %u.",
> +                        type->dimx * type->dimy, v->initializer.args_count);
> +                free_parse_initializer(&v->initializer);
> +                vkd3d_free(v);
> +                continue;
>               }
>               if ((type->type == HLSL_CLASS_STRUCT || type->type == HLSL_CLASS_ARRAY)
>                       && hlsl_type_component_count(type) != size)

This seems like a separate change, no?

Changing "size != 1" to "v->initializer.args_count != 1" seems like yet 
another separate change. (Which I guess is technically more correct, 
although still not perfectly fixed yet...)