Initial work

1 files changed, 101 insertions, 0 deletions

diff --git a/Ryujinx.Graphics.Shader/IntermediateRepresentation/Operation.cs b/Ryujinx.Graphics.Shader/IntermediateRepresentation/Operation.cs
new file mode 100644
index 00000000..fc01d47e
--- /dev/null
+++ b/Ryujinx.Graphics.Shader/IntermediateRepresentation/Operation.cs
@@ -0,0 +1,101 @@
+namespace Ryujinx.Graphics.Shader.IntermediateRepresentation
+{
+    class Operation : INode
+    {
+        public Instruction Inst { get; private set; }
+
+        private Operand _dest;
+
+        public Operand Dest
+        {
+            get => _dest;
+            set => _dest = AssignDest(value);
+        }
+
+        private Operand[] _sources;
+
+        public int SourcesCount => _sources.Length;
+
+        public int ComponentIndex { get; }
+
+        public Operation(Instruction inst, Operand dest, params Operand[] sources)
+        {
+            Inst = inst;
+            Dest = dest;
+
+            // The array may be modified externally, so we store a copy.
+            _sources = (Operand[])sources.Clone();
+
+            for (int index = 0; index < _sources.Length; index++)
+            {
+                Operand source = _sources[index];
+
+                if (source.Type == OperandType.LocalVariable)
+                {
+                    source.UseOps.Add(this);
+                }
+            }
+        }
+
+        public Operation(
+            Instruction      inst,
+            int              compIndex,
+            Operand          dest,
+            params Operand[] sources) : this(inst, dest, sources)
+        {
+            ComponentIndex = compIndex;
+        }
+
+        private Operand AssignDest(Operand dest)
+        {
+            if (dest != null && dest.Type == OperandType.LocalVariable)
+            {
+                dest.AsgOp = this;
+            }
+
+            return dest;
+        }
+
+        public Operand GetSource(int index)
+        {
+            return _sources[index];
+        }
+
+        public void SetSource(int index, Operand source)
+        {
+            Operand oldSrc = _sources[index];
+
+            if (oldSrc != null && oldSrc.Type == OperandType.LocalVariable)
+            {
+                oldSrc.UseOps.Remove(this);
+            }
+
+            if (source != null && source.Type == OperandType.LocalVariable)
+            {
+                source.UseOps.Add(this);
+            }
+
+            _sources[index] = source;
+        }
+
+        public void TurnIntoCopy(Operand source)
+        {
+            Inst = Instruction.Copy;
+
+            foreach (Operand oldSrc in _sources)
+            {
+                if (oldSrc != null && oldSrc.Type == OperandType.LocalVariable)
+                {
+                    oldSrc.UseOps.Remove(this);
+                }
+            }
+
+            if (source.Type == OperandType.LocalVariable)
+            {
+                source.UseOps.Add(this);
+            }
+
+            _sources = new Operand[] { source };
+        }
+    }
+}
\ No newline at end of file