[PATCH 05/11] vkd3d-shader/ir: Keep track of loops by header block.

[Giovanni Mascellani]

From: Giovanni Mascellani

---
 libs/vkd3d-shader/ir.c | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/libs/vkd3d-shader/ir.c b/libs/vkd3d-shader/ir.c
index 1eb534b24..2d879846e 100644
--- a/libs/vkd3d-shader/ir.c
+++ b/libs/vkd3d-shader/ir.c
@@ -3106,6 +3106,7 @@ struct vsir_cfg

struct vsir_block_list *loops;
     size_t loops_count, loops_capacity;
+    struct vsir_block_list **loops_by_header;
 };

static void vsir_cfg_cleanup(struct vsir_cfg *cfg)
@@ -3120,6 +3121,7 @@ static void vsir_cfg_cleanup(struct vsir_cfg *cfg)

vkd3d_free(cfg->blocks);
     vkd3d_free(cfg->loops);
+    vkd3d_free(cfg->loops_by_header);

if (TRACE_ON())
         vkd3d_string_buffer_cleanup(&cfg->debug_buffer);
@@ -3399,6 +3401,9 @@ static enum vkd3d_result vsir_cfg_compute_loops(struct vsir_cfg *cfg)
 {
     size_t i, j;

+    if (!(cfg->loops_by_header = vkd3d_calloc(cfg->block_count, sizeof(*cfg->loops_by_header))))
+        return VKD3D_ERROR_OUT_OF_MEMORY;
+
     for (i = 0; i < cfg->block_count; ++i)
     {
         struct vsir_block *block = &cfg->blocks[i];
@@ -3423,6 +3428,14 @@ static enum vkd3d_result vsir_cfg_compute_loops(struct vsir_cfg *cfg)

if ((ret = vsir_cfg_scan_loop(loop, block, header)) < 0)
                 return ret;
+
+            if (cfg->loops_by_header[header->label - 1])
+            {
+                FIXME("Block %u is header to more than one loop, this is not implemented.", header->label);
+                return VKD3D_ERROR_NOT_IMPLEMENTED;
+            }
+
+            cfg->loops_by_header[header->label - 1] = loop;
         }
     }