From 3611c1313d560ab45ae77e70154e945fab1aecfa Mon Sep 17 00:00:00 2001
From: Simon Perretta <simon.perretta@imgtec.com>
Date: Sun, 29 Dec 2024 09:34:31 +0000
Subject: [PATCH] pco: support re-indexing loops and ifs

Signed-off-by: Simon Perretta <simon.perretta@imgtec.com>
Acked-by: Erik Faye-Lund <erik.faye-lund@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/36412>
---
 src/imagination/pco/pco_index.c | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/imagination/pco/pco_index.c b/src/imagination/pco/pco_index.c
index eb26342b00c..d1425c78eb1 100644
--- a/src/imagination/pco/pco_index.c
+++ b/src/imagination/pco/pco_index.c
@@ -82,9 +82,15 @@ bool pco_index(pco_shader *shader, bool skip_ssa)
          }
       }
 
-      /* TODO: */
-      /* pco_foreach_if_in_func */
-      /* pco_foreach_loop_in_func */
+      func->next_if = 0;
+      pco_foreach_if_in_func (pif, func) {
+         pif->index = func->next_if++;
+      }
+
+      func->next_loop = 0;
+      pco_foreach_loop_in_func (loop, func) {
+         loop->index = func->next_loop++;
+      }
 
       if (!skip_ssa) {
          _mesa_hash_table_u64_destroy(func->vec_infos);