Francisco Casas : vkd3d-shader/hlsl: Absorb hlsl_ir_constant deref offsets into const_offset.

[Alexandre Julliard]

Module: vkd3d
Branch: master
Commit: 74767beaf658296950f3f2ca527192e471d39bf4
URL:    https://gitlab.winehq.org/wine/vkd3d/-/commit/74767beaf658296950f3f2ca527192...

Author: Francisco Casas <fcasas(a)codeweavers.com>
Date:   Thu May 11 15:18:02 2023 -0400

vkd3d-shader/hlsl: Absorb hlsl_ir_constant deref offsets into const_offset.

---

 libs/vkd3d-shader/hlsl_codegen.c | 29 +++++++++++++++++++++--------
 1 file changed, 21 insertions(+), 8 deletions(-)

diff --git a/libs/vkd3d-shader/hlsl_codegen.c b/libs/vkd3d-shader/hlsl_codegen.c
index 5c3b7eb6..0a9d3f67 100644
--- a/libs/vkd3d-shader/hlsl_codegen.c
+++ b/libs/vkd3d-shader/hlsl_codegen.c
@@ -174,6 +174,24 @@ static bool replace_deref_path_with_offset(struct hlsl_ctx *ctx, struct hlsl_der
     return true;
 }
 
+static bool clean_constant_deref_offset_srcs(struct hlsl_ctx *ctx, struct hlsl_deref *deref,
+        struct hlsl_ir_node *instr)
+{
+    if (deref->offset.node && deref->offset.node->type == HLSL_IR_CONSTANT)
+    {
+        enum hlsl_regset regset = hlsl_deref_get_regset(ctx, deref);
+
+        if (regset == HLSL_REGSET_NUMERIC)
+            deref->const_offset += 4 * hlsl_ir_constant(deref->offset.node)->value.u[0].u;
+        else
+            deref->const_offset += hlsl_ir_constant(deref->offset.node)->value.u[0].u;
+        hlsl_src_remove(&deref->offset);
+        return true;
+    }
+    return false;
+}
+
+
 /* Split uniforms into two variables representing the constant and temp
  * registers, and copy the former to the latter, so that writes to uniforms
  * work. */
@@ -4465,14 +4483,8 @@ bool hlsl_offset_from_deref(struct hlsl_ctx *ctx, const struct hlsl_deref *deref
         /* We should always have generated a cast to UINT. */
         assert(offset_node->data_type->class == HLSL_CLASS_SCALAR
                 && offset_node->data_type->base_type == HLSL_TYPE_UINT);
-
-        if (offset_node->type != HLSL_IR_CONSTANT)
-            return false;
-
-        if (regset == HLSL_REGSET_NUMERIC)
-            *offset += 4 * hlsl_ir_constant(offset_node)->value.u[0].u;
-        else
-            *offset += hlsl_ir_constant(offset_node)->value.u[0].u;
+        assert(offset_node->type != HLSL_IR_CONSTANT);
+        return false;
     }
 
     size = deref->var->data_type->reg_size[regset];
@@ -4789,6 +4801,7 @@ int hlsl_emit_bytecode(struct hlsl_ctx *ctx, struct hlsl_ir_function_decl *entry
     /* TODO: move forward, remove when no longer needed */
     transform_derefs(ctx, replace_deref_path_with_offset, body);
     while (hlsl_transform_ir(ctx, hlsl_fold_constant_exprs, body, NULL));
+    transform_derefs(ctx, clean_constant_deref_offset_srcs, body);
 
     do
         compute_liveness(ctx, entry_func);