nir: Add a deref_instr_has_indirect helper

Reviewed-by: Timothy Arceri <tarceri@itsqueeze.com>
Reviewed-by: Iago Toral Quiroga <itoral@igalia.com>
Reviewed-by: Kenneth Graunke <kenneth@whitecape.org>

diff --git a/src/compiler/nir/nir.h b/src/compiler/nir/nir.h
index c16ce547642b54c53a8b565b5fc7a8f017a9a9d2..e35bef612df8842fa86b88a433aa117850a30598 100644 (file)
--- a/src/compiler/nir/nir.h
+++ b/src/compiler/nir/nir.h
@@ -987,6 +987,8 @@ nir_deref_instr_get_variable(const nir_deref_instr *instr)
    return instr->var;
 }
 
+bool nir_deref_instr_has_indirect(nir_deref_instr *instr);
+
 bool nir_deref_instr_remove_if_unused(nir_deref_instr *instr);
 
 typedef struct {

diff --git a/src/compiler/nir/nir_deref.c b/src/compiler/nir/nir_deref.c
index 1a00157c2fcd5ff300fdd526bbf168ef2162c867..22ecde4ecca9e8e2295537472bee744f7b1d97a8 100644 (file)
--- a/src/compiler/nir/nir_deref.c
+++ b/src/compiler/nir/nir_deref.c
@@ -102,6 +102,24 @@ nir_deref_instr_remove_if_unused(nir_deref_instr *instr)
    return progress;
 }
 
+bool
+nir_deref_instr_has_indirect(nir_deref_instr *instr)
+{
+   while (instr->deref_type != nir_deref_type_var) {
+      /* Consider casts to be indirects */
+      if (instr->deref_type == nir_deref_type_cast)
+         return true;
+
+      if (instr->deref_type == nir_deref_type_array &&
+          !nir_src_as_const_value(instr->arr.index))
+         return true;
+
+      instr = nir_deref_instr_parent(instr);
+   }
+
+   return false;
+}
+
 bool
 nir_remove_dead_derefs_impl(nir_function_impl *impl)
 {