nir: Add a function for rewriting the condition of an if statement

Reviewed-by: Connor Abbott <cwabbott0@gmail.com>
This commit is contained in:
Jason Ekstrand 2015-04-24 10:34:30 -07:00
parent 300d729436
commit f72a8d1cf0
2 changed files with 23 additions and 0 deletions

View file

@ -1894,6 +1894,28 @@ nir_instr_rewrite_src(nir_instr *instr, nir_src *src, nir_src new_src)
}
}
void
nir_if_rewrite_condition(nir_if *if_stmt, nir_src new_src)
{
for (nir_src *src = &if_stmt->condition; src;
src = src->is_ssa ? NULL : src->reg.indirect) {
struct set *uses = src->is_ssa ? src->ssa->if_uses
: src->reg.reg->if_uses;
struct set_entry *entry = _mesa_set_search(uses, if_stmt);
assert(entry);
_mesa_set_remove(uses, entry);
}
if_stmt->condition = new_src;
for (nir_src *src = &if_stmt->condition; src;
src = src->is_ssa ? NULL : src->reg.indirect) {
struct set *uses = src->is_ssa ? src->ssa->if_uses
: src->reg.reg->if_uses;
_mesa_set_add(uses, if_stmt);
}
}
void
nir_ssa_def_init(nir_instr *instr, nir_ssa_def *def,
unsigned num_components, const char *name)

View file

@ -1549,6 +1549,7 @@ bool nir_foreach_src(nir_instr *instr, nir_foreach_src_cb cb, void *state);
nir_const_value *nir_src_as_const_value(nir_src src);
bool nir_srcs_equal(nir_src src1, nir_src src2);
void nir_instr_rewrite_src(nir_instr *instr, nir_src *src, nir_src new_src);
void nir_if_rewrite_condition(nir_if *if_stmt, nir_src new_src);
void nir_ssa_dest_init(nir_instr *instr, nir_dest *dest,
unsigned num_components, const char *name);