[PATCH 4/8] vkd3d-shader/hlsl: Express deref->offset in whole registers.

[Francisco Casas]

From: Francisco Casas

This is required to use SM4 relative addressing, because it is limited
to whole-register granularity.
---
 libs/vkd3d-shader/hlsl.c         |  2 +-
 libs/vkd3d-shader/hlsl.h         |  2 +-
 libs/vkd3d-shader/hlsl_codegen.c | 19 +++++++++----------
 3 files changed, 11 insertions(+), 12 deletions(-)

diff --git a/libs/vkd3d-shader/hlsl.c b/libs/vkd3d-shader/hlsl.c
index 9e3ee8174..cb5b23db3 100644
--- a/libs/vkd3d-shader/hlsl.c
+++ b/libs/vkd3d-shader/hlsl.c
@@ -2338,7 +2338,7 @@ static void dump_deref(struct vkd3d_string_buffer *buffer, const struct hlsl_der
             if (deref->offset.node)
             {
                 dump_src(buffer, &deref->offset);
-                vkd3d_string_buffer_printf(buffer, "c + ");
+                vkd3d_string_buffer_printf(buffer, " + ");
             }
             vkd3d_string_buffer_printf(buffer, "%uc]", deref->offset_const);
         }
diff --git a/libs/vkd3d-shader/hlsl.h b/libs/vkd3d-shader/hlsl.h
index 9a333844c..d801da31c 100644
--- a/libs/vkd3d-shader/hlsl.h
+++ b/libs/vkd3d-shader/hlsl.h
@@ -619,7 +619,7 @@ struct hlsl_deref
      *   regset) from the start of the variable, to the part of the variable that is referenced.
      * This offset is stored using two fields, one for a variable part and other for a constant
      *   part, which are added together:
-     *   - offset: An offset given by an instruction node, in number of register components.
+     *   - offset: An offset given by an instruction node, in whole registers.
      *   - offset_const: A constant number of register components.
      * Since the type information cannot longer be retrieved from the offset alone, the type is
      *   stored in the data_type field. */
diff --git a/libs/vkd3d-shader/hlsl_codegen.c b/libs/vkd3d-shader/hlsl_codegen.c
index c519ac4ec..9f4e367bd 100644
--- a/libs/vkd3d-shader/hlsl_codegen.c
+++ b/libs/vkd3d-shader/hlsl_codegen.c
@@ -39,14 +39,7 @@ static struct hlsl_ir_node *new_offset_from_path_index(struct hlsl_ctx *ctx, str

case HLSL_CLASS_MATRIX:
         {
-            if (!(c = hlsl_new_uint_constant(ctx, 4, loc)))
-                return NULL;
-            hlsl_block_add_instr(block, c);
-
-            if (!(idx_offset = hlsl_new_binary_expr(ctx, HLSL_OP2_MUL, c, idx)))
-                return NULL;
-            hlsl_block_add_instr(block, idx_offset);
-
+            idx_offset = idx;
             break;
         }

@@ -54,6 +47,9 @@ static struct hlsl_ir_node *new_offset_from_path_index(struct hlsl_ctx *ctx, str
         {
             unsigned int size = hlsl_type_get_array_element_reg_size(type->e.array.type, regset);

+            if (regset == HLSL_REGSET_NUMERIC)
+                size /= 4;
+
             if (!(c = hlsl_new_uint_constant(ctx, size, loc)))
                 return NULL;
             hlsl_block_add_instr(block, c);
@@ -75,7 +71,7 @@ static struct hlsl_ir_node *new_offset_from_path_index(struct hlsl_ctx *ctx, str
             {
                 assert(*offset_component == 0);
                 *offset_component = field_offset % 4;
-                field_offset -= *offset_component;
+                field_offset /= 4;
             }

if (!(c = hlsl_new_uint_constant(ctx, field_offset, loc)))
@@ -4229,7 +4225,10 @@ bool hlsl_offset_from_deref(struct hlsl_ctx *ctx, const struct hlsl_deref *deref
         if (offset_node->type != HLSL_IR_CONSTANT)
             return false;

-        *offset += hlsl_ir_constant(offset_node)->value.u[0].u;
+        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;
     }

size = deref->var->data_type->reg_size[regset];