Nikolay Sivov : vkd3d-shader/tpf: Fix used temp registers accounting for dcl_temps.

[Alexandre Julliard]

Module: vkd3d
Branch: master
Commit: 31346e2cba466272eccc19ba982c1549b0378c19
URL:    https://gitlab.winehq.org/wine/vkd3d/-/commit/31346e2cba466272eccc19ba982c15...

Author: Nikolay Sivov 
Date:   Sat Oct 28 00:06:51 2023 +0200

vkd3d-shader/tpf: Fix used temp registers accounting for dcl_temps.

Otherwise we always output "dcl_temps 1" even when no temp registers were used.

---

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

diff --git a/libs/vkd3d-shader/hlsl_codegen.c b/libs/vkd3d-shader/hlsl_codegen.c
index 51c3d68f..6eac5d49 100644
--- a/libs/vkd3d-shader/hlsl_codegen.c
+++ b/libs/vkd3d-shader/hlsl_codegen.c
@@ -3461,22 +3461,20 @@ static void compute_liveness(struct hlsl_ctx *ctx, struct hlsl_ir_function_decl

struct register_allocator
 {
-    size_t count, capacity;
-
-    /* Highest register index that has been allocated.
-     * Used to declare sm4 temp count. */
-    uint32_t max_reg;
-
     struct allocation
     {
         uint32_t reg;
         unsigned int writemask;
         unsigned int first_write, last_read;
     } *allocations;
+    size_t count, capacity;

/* Indexable temps are allocated separately and always keep their index regardless of their
      * lifetime. */
     size_t indexable_count;
+
+    /* Total number of registers allocated so far. Used to declare sm4 temp count. */
+    uint32_t reg_count;
 };

static unsigned int get_available_writemask(const struct register_allocator *allocator,
@@ -3519,7 +3517,7 @@ static void record_allocation(struct hlsl_ctx *ctx, struct register_allocator *a
     allocation->first_write = first_write;
     allocation->last_read = last_read;

-    allocator->max_reg = max(allocator->max_reg, reg_idx);
+    allocator->reg_count = max(allocator->reg_count, reg_idx + 1);
 }

/* reg_size is the number of register components to be reserved, while component_count is the number
@@ -3986,7 +3984,7 @@ static void allocate_temp_registers(struct hlsl_ctx *ctx, struct hlsl_ir_functio
     }

allocate_temp_registers_recurse(ctx, &entry_func->body, &allocator);
-    ctx->temp_count = allocator.max_reg + 1;
+    ctx->temp_count = allocator.reg_count;
     vkd3d_free(allocator.allocations);
 }