Initial work

1 files changed, 73 insertions, 0 deletions

diff --git a/Ryujinx.Graphics.Shader/StructuredIr/AstHelper.cs b/Ryujinx.Graphics.Shader/StructuredIr/AstHelper.cs
new file mode 100644
index 00000000..9d3148e1
--- /dev/null
+++ b/Ryujinx.Graphics.Shader/StructuredIr/AstHelper.cs
@@ -0,0 +1,73 @@
+using Ryujinx.Graphics.Shader.IntermediateRepresentation;
+
+namespace Ryujinx.Graphics.Shader.StructuredIr
+{
+    static class AstHelper
+    {
+        public static void AddUse(IAstNode node, IAstNode parent)
+        {
+            if (node is AstOperand operand && operand.Type == OperandType.LocalVariable)
+            {
+                operand.Uses.Add(parent);
+            }
+        }
+
+        public static void AddDef(IAstNode node, IAstNode parent)
+        {
+            if (node is AstOperand operand && operand.Type == OperandType.LocalVariable)
+            {
+                operand.Defs.Add(parent);
+            }
+        }
+
+        public static void RemoveUse(IAstNode node, IAstNode parent)
+        {
+            if (node is AstOperand operand && operand.Type == OperandType.LocalVariable)
+            {
+                operand.Uses.Remove(parent);
+            }
+        }
+
+        public static void RemoveDef(IAstNode node, IAstNode parent)
+        {
+            if (node is AstOperand operand && operand.Type == OperandType.LocalVariable)
+            {
+                operand.Defs.Remove(parent);
+            }
+        }
+
+        public static AstAssignment Assign(IAstNode destination, IAstNode source)
+        {
+            return new AstAssignment(destination, source);
+        }
+
+        public static AstOperand Const(int value)
+        {
+            return new AstOperand(OperandType.Constant, value);
+        }
+
+        public static AstOperand Local(VariableType type)
+        {
+            AstOperand local = new AstOperand(OperandType.LocalVariable);
+
+            local.VarType = type;
+
+            return local;
+        }
+
+        public static IAstNode InverseCond(IAstNode cond)
+        {
+            return new AstOperation(Instruction.LogicalNot, cond);
+        }
+
+        public static IAstNode Next(IAstNode node)
+        {
+            return node.LLNode.Next?.Value;
+        }
+
+        public static IAstNode Previous(IAstNode node)
+        {
+            return node.LLNode.Previous?.Value;
+        }
+    }
+}
\ No newline at end of file