[PATCH v7 6/8] vkd3d-shader/ir: Keep track of loops by header block.

[Giovanni Mascellani]

From: Giovanni Mascellani

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

diff --git a/libs/vkd3d-shader/ir.c b/libs/vkd3d-shader/ir.c
index 1fd92dd8e..b0fb10ca5 100644
--- a/libs/vkd3d-shader/ir.c
+++ b/libs/vkd3d-shader/ir.c
@@ -3107,6 +3107,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)
@@ -3121,6 +3122,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);
@@ -3391,6 +3393,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];
@@ -3416,6 +3421,16 @@ 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);
+                vkd3d_shader_parser_error(cfg->parser, VKD3D_SHADER_ERROR_VSIR_NOT_IMPLEMENTED,
+                        "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;
         }
     }