From 9bc2bdd78f9e73aae5b1ce610d5d61bda5d35f09 Mon Sep 17 00:00:00 2001
From: Faith Ekstrand <faith.ekstrand@collabora.com>
Date: Tue, 25 Apr 2023 16:36:07 -0500
Subject: [PATCH] nak: Add a Src::supports_src_type() helper

Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/24998>
---
 src/nouveau/compiler/nak_ir.rs | 73 ++++++++++++++++++++++++++++++++++
 1 file changed, 73 insertions(+)

diff --git a/src/nouveau/compiler/nak_ir.rs b/src/nouveau/compiler/nak_ir.rs
index d21df588f72..8cfd5ed931d 100644
--- a/src/nouveau/compiler/nak_ir.rs
+++ b/src/nouveau/compiler/nak_ir.rs
@@ -28,6 +28,13 @@ impl RegFile {
         }
     }
 
+    pub fn is_gpr(&self) -> bool {
+        match self {
+            RegFile::GPR | RegFile::UGPR => true,
+            RegFile::Pred | RegFile::UPred => false,
+        }
+    }
+
     pub fn is_predicate(&self) -> bool {
         match self {
             RegFile::GPR | RegFile::UGPR => false,
@@ -100,6 +107,10 @@ pub trait HasRegFile {
         self.file().is_uniform()
     }
 
+    fn is_gpr(&self) -> bool {
+        self.file().is_gpr()
+    }
+
     fn is_predicate(&self) -> bool {
         self.file().is_predicate()
     }
@@ -425,6 +436,15 @@ pub enum SrcRef {
 }
 
 impl SrcRef {
+    pub fn is_alu(&self) -> bool {
+        match self {
+            SrcRef::Zero | SrcRef::Imm32(_) | SrcRef::CBuf(_) => true,
+            SrcRef::SSA(ssa) => ssa.is_gpr(),
+            SrcRef::Reg(reg) => reg.is_gpr(),
+            SrcRef::True | SrcRef::False => false,
+        }
+    }
+
     pub fn is_predicate(&self) -> bool {
         match self {
             SrcRef::Zero | SrcRef::Imm32(_) | SrcRef::CBuf(_) => false,
@@ -727,6 +747,59 @@ impl Src {
             | SrcRef::CBuf(_) => false,
         }
     }
+
+    pub fn supports_type(&self, src_type: &SrcType) -> bool {
+        match src_type {
+            SrcType::SSA => {
+                if !self.src_mod.is_none() {
+                    return false;
+                }
+
+                match self.src_ref {
+                    SrcRef::SSA(_) | SrcRef::Reg(_) => true,
+                    _ => false,
+                }
+            }
+            SrcType::GPR => {
+                if !self.src_mod.is_none() {
+                    return false;
+                }
+
+                match self.src_ref {
+                    SrcRef::Zero | SrcRef::SSA(_) | SrcRef::Reg(_) => true,
+                    _ => false,
+                }
+            }
+            SrcType::ALU => self.src_mod.is_none() && self.src_ref.is_alu(),
+            SrcType::F32 | SrcType::F64 => {
+                match self.src_mod {
+                    SrcMod::None
+                    | SrcMod::FAbs
+                    | SrcMod::FNeg
+                    | SrcMod::FNegAbs => (),
+                    _ => return false,
+                }
+
+                self.src_ref.is_alu()
+            }
+            SrcType::I32 => {
+                match self.src_mod {
+                    SrcMod::None | SrcMod::INeg => (),
+                    _ => return false,
+                }
+
+                self.src_ref.is_alu()
+            }
+            SrcType::Pred => {
+                match self.src_mod {
+                    SrcMod::None | SrcMod::BNot => (),
+                    _ => return false,
+                }
+
+                self.src_ref.is_predicate()
+            }
+        }
+    }
 }
 
 impl<T: Into<SrcRef>> From<T> for Src {