[PATCH 5/7] vkd3d-shader/ir: Compute loop as intervals of the block order.

[Giovanni Mascellani]

From: Giovanni Mascellani

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

diff --git a/libs/vkd3d-shader/ir.c b/libs/vkd3d-shader/ir.c
index 3c862f33e..05b40cafd 100644
--- a/libs/vkd3d-shader/ir.c
+++ b/libs/vkd3d-shader/ir.c
@@ -3140,6 +3140,16 @@ struct vsir_cfg
     size_t *loops_by_header;

struct vsir_block_list order;
+    struct cfg_loop_interval
+    {
+        /* `begin' is the position of the first block of the loop in
+         * the topological sort; `end' is the position of the first
+         * block after the loop. In other words, `begin' is where a
+         * `continue' instruction would jump and `end' is where a
+         * `break' instruction would jump. */
+        unsigned int begin, end;
+    } *loop_intervals;
+    size_t loop_interval_count, loop_interval_capacity;
 };

static void vsir_cfg_cleanup(struct vsir_cfg *cfg)
@@ -3157,11 +3167,29 @@ static void vsir_cfg_cleanup(struct vsir_cfg *cfg)
     vkd3d_free(cfg->blocks);
     vkd3d_free(cfg->loops);
     vkd3d_free(cfg->loops_by_header);
+    vkd3d_free(cfg->loop_intervals);

if (TRACE_ON())
         vkd3d_string_buffer_cleanup(&cfg->debug_buffer);
 }

+static enum vkd3d_result vsir_cfg_add_loop_interval(struct vsir_cfg *cfg, unsigned int begin,
+        unsigned int end)
+{
+    struct cfg_loop_interval *interval;
+
+    if (!vkd3d_array_reserve((void **)&cfg->loop_intervals, &cfg->loop_interval_capacity,
+            cfg->loop_interval_count + 1, sizeof(*cfg->loop_intervals)))
+        return VKD3D_ERROR_OUT_OF_MEMORY;
+
+    interval = &cfg->loop_intervals[cfg->loop_interval_count++];
+
+    interval->begin = begin;
+    interval->end = end;
+
+    return VKD3D_OK;
+}
+
 static bool vsir_block_dominates(struct vsir_block *b1, struct vsir_block *b2)
 {
     return bitmap_is_set(b1->dominates, b2->label - 1);
@@ -3496,6 +3524,7 @@ struct vsir_cfg_node_sorter
     {
         struct vsir_block_list *loop;
         unsigned int seen_count;
+        unsigned int open_time;
     } *stack;
     size_t stack_count, stack_capacity;
     struct vsir_block_list available_blocks;
@@ -3522,6 +3551,7 @@ static enum vkd3d_result vsir_cfg_node_sorter_make_node_available(struct vsir_cf
     item = &sorter->stack[sorter->stack_count++];
     item->loop = loop;
     item->seen_count = 0;
+    item->open_time = sorter->cfg->order.count;

return VKD3D_OK;
 }
@@ -3646,6 +3676,10 @@ static enum vkd3d_result vsir_cfg_sort_nodes(struct vsir_cfg *cfg)
             if (inner_stack_item->seen_count != inner_stack_item->loop->count)
                 break;

+            if ((ret = vsir_cfg_add_loop_interval(cfg, inner_stack_item->open_time,
+                    cfg->order.count)) < 0)
+                goto fail;
+
             new_seen_count = inner_stack_item->loop->count;
             --sorter.stack_count;
         }
@@ -3694,6 +3728,9 @@ static enum vkd3d_result vsir_cfg_sort_nodes(struct vsir_cfg *cfg)

TRACE("%s\n", cfg->debug_buffer.buffer);
         vkd3d_string_buffer_clear(&cfg->debug_buffer);
+
+        for (i = 0; i < cfg->loop_interval_count; ++i)
+            TRACE("Loop interval %u - %u\n", cfg->loop_intervals[i].begin, cfg->loop_intervals[i].end);
     }

return VKD3D_OK;