From 1b36e340958e6f61c2c86962d7b2deeb764860b9 Mon Sep 17 00:00:00 2001
From: Simon Perretta <simon.perretta@imgtec.com>
Date: Fri, 24 May 2024 16:45:40 +0100
Subject: [PATCH] pco, pygen: support querying ref mods, if op/ref mods have
 been set

Signed-off-by: Simon Perretta <simon.perretta@imgtec.com>
Acked-by: Frank Binns <frank.binns@imgtec.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/32258>
---
 src/imagination/pco/pco_builder_ops.h.py | 44 ++++++++++++++++++++++--
 src/imagination/pco/pco_internal.h       | 12 +++++++
 2 files changed, 54 insertions(+), 2 deletions(-)

diff --git a/src/imagination/pco/pco_builder_ops.h.py b/src/imagination/pco/pco_builder_ops.h.py
index 46e4d6d1cf9..45e1419d69d 100644
--- a/src/imagination/pco/pco_builder_ops.h.py
+++ b/src/imagination/pco/pco_builder_ops.h.py
@@ -27,6 +27,39 @@ template = """/*
 #include <assert.h>
 #include <stdbool.h>
 
+/** Ref mod info. */
+static inline
+bool pco_instr_dest_has_mod(const pco_instr *instr, unsigned dest, enum pco_ref_mod mod)
+{
+   const struct pco_op_info *info = &pco_op_info[instr->op];
+   assert(dest < info->num_dests);
+   assert(mod < _PCO_REF_MOD_COUNT);
+   return (info->dest_mods[dest] & (1ULL << mod)) != 0;
+}
+
+static inline
+bool pco_instr_src_has_mod(const pco_instr *instr, unsigned src, enum pco_ref_mod mod)
+{
+   const struct pco_op_info *info = &pco_op_info[instr->op];
+   assert(src < info->num_srcs);
+   assert(mod < _PCO_REF_MOD_COUNT);
+   return (info->src_mods[src] & (1ULL << mod)) != 0;
+}
+
+% for ref_mod in ref_mods.values():
+static inline
+bool pco_instr_dest_has_${ref_mod.t.tname}(const pco_instr *instr, unsigned dest)
+{
+   return pco_instr_dest_has_mod(instr, dest, ${ref_mod.cname});
+}
+
+static inline
+bool pco_instr_src_has_${ref_mod.t.tname}(const pco_instr *instr, unsigned src)
+{
+   return pco_instr_src_has_mod(instr, src, ${ref_mod.cname});
+}
+
+% endfor
 /** Op mod getting/setting. */
 static inline
 bool pco_instr_has_mod(const pco_instr *instr, enum pco_op_mod mod)
@@ -45,7 +78,7 @@ void pco_instr_set_mod(pco_instr *instr, enum pco_op_mod mod, uint32_t val)
 }
 
 static inline
-uint32_t pco_instr_get_mod(pco_instr *instr, enum pco_op_mod mod)
+uint32_t pco_instr_get_mod(const pco_instr *instr, enum pco_op_mod mod)
 {
    assert(mod < _PCO_OP_MOD_COUNT);
    unsigned mod_index = pco_op_info[instr->op].mod_map[mod];
@@ -53,6 +86,13 @@ uint32_t pco_instr_get_mod(pco_instr *instr, enum pco_op_mod mod)
    return instr->mod[mod_index - 1];
 }
 
+static inline
+bool pco_instr_mod_is_set(const pco_instr *instr, enum pco_op_mod mod)
+{
+   const struct pco_op_mod_info *info = &pco_op_mod_info[mod];
+   return pco_instr_get_mod(instr, mod) != (info->nzdefault ? info->nzdefault : 0);
+}
+
 % for op_mod in op_mods.values():
 static inline
 bool pco_instr_has_${op_mod.t.tname}(const pco_instr *instr)
@@ -67,7 +107,7 @@ void pco_instr_set_${op_mod.t.tname}(pco_instr *instr, ${op_mod.t.name} val)
 }
 
 static inline
-${op_mod.t.name} pco_instr_get_${op_mod.t.tname}(pco_instr *instr)
+${op_mod.t.name} pco_instr_get_${op_mod.t.tname}(const pco_instr *instr)
 {
    return pco_instr_get_mod(instr, ${op_mod.cname});
 }
diff --git a/src/imagination/pco/pco_internal.h b/src/imagination/pco/pco_internal.h
index d77c7175799..e800d0c922c 100644
--- a/src/imagination/pco/pco_internal.h
+++ b/src/imagination/pco/pco_internal.h
@@ -1354,6 +1354,18 @@ static inline enum pco_drc pco_ref_get_drc(pco_ref ref)
    return ref.val;
 }
 
+/**
+ * \brief Returns whether the reference has any mods set.
+ *
+ * \param[in] ref Reference.
+ * \return True if any mods are set.
+ */
+static inline bool pco_ref_has_mods_set(pco_ref ref)
+{
+   return ref.oneminus || ref.clamp || ref.abs || ref.neg || ref.flr ||
+          (ref.elem != 0);
+}
+
 /* PCO ref builders. */
 /**
  * \brief Builds and returns a null reference.