From 4aaa2192b993411f63d46a57b93e9e787b6a836d Mon Sep 17 00:00:00 2001
From: ReinUsesLisp <reinuseslisp@airmail.cc>
Date: Thu, 20 Dec 2018 22:36:17 -0300
Subject: [PATCH] shader_ir: Add immediate node constructors

---
 src/video_core/shader/shader_ir.cpp | 16 ++++++++++++++++
 src/video_core/shader/shader_ir.h   | 17 +++++++++++++++++
 2 files changed, 33 insertions(+)

diff --git a/src/video_core/shader/shader_ir.cpp b/src/video_core/shader/shader_ir.cpp
index db00c89028..c59ecf4573 100644
--- a/src/video_core/shader/shader_ir.cpp
+++ b/src/video_core/shader/shader_ir.cpp
@@ -35,6 +35,18 @@ Node ShaderIR::Comment(const std::string& text) {
     return StoreNode(CommentNode(text));
 }
 
+Node ShaderIR::Immediate(u32 value) {
+    return StoreNode(ImmediateNode(value));
+}
+
+Node ShaderIR::GetImmediate19(Instruction instr) {
+    return Immediate(instr.alu.GetImm20_19());
+}
+
+Node ShaderIR::GetImmediate32(Instruction instr) {
+    return Immediate(instr.alu.GetImm20_32());
+}
+
 Node ShaderIR::GetPredicate(u64 pred_, bool negated) {
     const auto pred = static_cast<Pred>(pred_);
     if (pred != Pred::UnusedIndex && pred != Pred::NeverExecute) {
@@ -44,6 +56,10 @@ Node ShaderIR::GetPredicate(u64 pred_, bool negated) {
     return StoreNode(PredicateNode(pred, negated));
 }
 
+Node ShaderIR::GetPredicate(bool immediate) {
+    return GetPredicate(static_cast<u64>(immediate ? Pred::UnusedIndex : Pred::NeverExecute));
+}
+
 /*static*/ OperationCode ShaderIR::SignedToUnsignedCode(OperationCode operation_code,
                                                         bool is_signed) {
     if (is_signed) {
diff --git a/src/video_core/shader/shader_ir.h b/src/video_core/shader/shader_ir.h
index 300cf1083a..db06d51cac 100644
--- a/src/video_core/shader/shader_ir.h
+++ b/src/video_core/shader/shader_ir.h
@@ -598,9 +598,26 @@ private:
     Node Conditional(Node condition, std::vector<Node>&& code);
     /// Creates a commentary
     Node Comment(const std::string& text);
+    /// Creates an u32 immediate
+    Node Immediate(u32 value);
+    /// Creates a s32 immediate
+    Node Immediate(s32 value) {
+        return Immediate(static_cast<u32>(value));
+    }
+    /// Creates a f32 immediate
+    Node Immediate(f32 value) {
+        // TODO(Rodrigo): Replace this with bit_cast when C++20 releases
+        return Immediate(*reinterpret_cast<const u32*>(&value));
+    }
 
+    /// Generates a node representing a 19-bit immediate value
+    Node GetImmediate19(Tegra::Shader::Instruction instr);
+    /// Generates a node representing a 32-bit immediate value
+    Node GetImmediate32(Tegra::Shader::Instruction instr);
     /// Generates a node for a passed predicate. It can be optionally negated
     Node GetPredicate(u64 pred, bool negated = false);
+    /// Generates a predicate node for an immediate true or false value
+    Node GetPredicate(bool immediate);
 
     template <typename... T>
     inline Node Operation(OperationCode code, const T*... operands) {