Initial work

1 files changed, 49 insertions, 0 deletions

diff --git a/Ryujinx.Graphics.Shader/StructuredIr/AstOperation.cs b/Ryujinx.Graphics.Shader/StructuredIr/AstOperation.cs
new file mode 100644
index 00000000..1607ffec
--- /dev/null
+++ b/Ryujinx.Graphics.Shader/StructuredIr/AstOperation.cs
@@ -0,0 +1,49 @@
+using Ryujinx.Graphics.Shader.IntermediateRepresentation;
+
+using static Ryujinx.Graphics.Shader.StructuredIr.AstHelper;
+
+namespace Ryujinx.Graphics.Shader.StructuredIr
+{
+    class AstOperation : AstNode
+    {
+        public Instruction Inst { get; }
+
+        public int ComponentMask { get; }
+
+        private IAstNode[] _sources;
+
+        public int SourcesCount => _sources.Length;
+
+        public AstOperation(Instruction inst, params IAstNode[] sources)
+        {
+            Inst     = inst;
+            _sources = sources;
+
+            foreach (IAstNode source in sources)
+            {
+                AddUse(source, this);
+            }
+
+            ComponentMask = 1;
+        }
+
+        public AstOperation(Instruction inst, int compMask, params IAstNode[] sources) : this(inst, sources)
+        {
+            ComponentMask = compMask;
+        }
+
+        public IAstNode GetSource(int index)
+        {
+            return _sources[index];
+        }
+
+        public void SetSource(int index, IAstNode source)
+        {
+            RemoveUse(_sources[index], this);
+
+            AddUse(source, this);
+
+            _sources[index] = source;
+        }
+    }
+}
\ No newline at end of file