1 files changed, 75 insertions, 0 deletions
diff --git a/Ryujinx.HLE/HOS/Tamper/CompilationContext.cs b/Ryujinx.HLE/HOS/Tamper/CompilationContext.cs
new file mode 100644
index 00000000..71e64bb8
--- /dev/null
+++ b/Ryujinx.HLE/HOS/Tamper/CompilationContext.cs
@@ -0,0 +1,75 @@
+using Ryujinx.HLE.HOS.Tamper.Operations;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Ryujinx.HLE.HOS.Tamper
+{
+    class CompilationContext
+    {
+        public OperationBlock CurrentBlock => BlockStack.Peek();
+        public List<IOperation> CurrentOperations => CurrentBlock.Operations;
+
+        public ITamperedProcess Process { get; }
+        public Parameter<long> PressedKeys { get; }
+        public Stack<OperationBlock> BlockStack { get; }
+        public Dictionary<byte, Register> Registers { get; }
+        public Dictionary<byte, Register> SavedRegisters { get; }
+        public Dictionary<byte, Register> StaticRegisters { get; }
+        public ulong ExeAddress { get; }
+        public ulong HeapAddress { get; }
+
+        public CompilationContext(ulong exeAddress, ulong heapAddress, ITamperedProcess process)
+        {
+            Process = process;
+            PressedKeys = new Parameter<long>(0);
+            BlockStack = new Stack<OperationBlock>();
+            Registers = new Dictionary<byte, Register>();
+            SavedRegisters = new Dictionary<byte, Register>();
+            StaticRegisters = new Dictionary<byte, Register>();
+            ExeAddress = exeAddress;
+            HeapAddress = heapAddress;
+        }
+
+        public Register GetRegister(byte index)
+        {
+            if (Registers.TryGetValue(index, out Register register))
+            {
+                return register;
+            }
+
+            register = new Register($"R_{index:X2}");
+            Registers.Add(index, register);
+
+            return register;
+        }
+
+        public Register GetSavedRegister(byte index)
+        {
+            if (SavedRegisters.TryGetValue(index, out Register register))
+            {
+                return register;
+            }
+
+            register = new Register($"S_{index:X2}");
+            SavedRegisters.Add(index, register);
+
+            return register;
+        }
+
+        public Register GetStaticRegister(byte index)
+        {
+            if (SavedRegisters.TryGetValue(index, out Register register))
+            {
+                return register;
+            }
+
+            register = new Register($"T_{index:X2}");
+            SavedRegisters.Add(index, register);
+
+            return register;
+        }
+    }
+}