1 files changed, 137 insertions, 0 deletions
diff --git a/src/Ryujinx.Graphics.Host1x/ThiDevice.cs b/src/Ryujinx.Graphics.Host1x/ThiDevice.cs
new file mode 100644
index 00000000..259c8836
--- /dev/null
+++ b/src/Ryujinx.Graphics.Host1x/ThiDevice.cs
@@ -0,0 +1,137 @@
+using Ryujinx.Common;
+using Ryujinx.Graphics.Device;
+using System;
+using System.Collections.Generic;
+
+namespace Ryujinx.Graphics.Host1x
+{
+    class ThiDevice : IDeviceStateWithContext, IDisposable
+    {
+        private readonly ClassId _classId;
+        private readonly IDeviceState _device;
+
+        private readonly SyncptIncrManager _syncptIncrMgr;
+
+        private long _currentContextId;
+        private long _previousContextId;
+
+        private class CommandAction
+        {
+            public long ContextId { get; }
+            public int Data { get; }
+
+            public CommandAction(long contextId, int data)
+            {
+                ContextId = contextId;
+                Data = data;
+            }
+        }
+
+        private class MethodCallAction : CommandAction
+        {
+            public int Method { get; }
+
+            public MethodCallAction(long contextId, int method, int data) : base(contextId, data)
+            {
+                Method = method;
+            }
+        }
+
+        private class SyncptIncrAction : CommandAction
+        {
+            public SyncptIncrAction(long contextId, uint syncptIncrHandle) : base(contextId, (int)syncptIncrHandle)
+            {
+            }
+        }
+
+        private readonly AsyncWorkQueue<CommandAction> _commandQueue;
+
+        private readonly DeviceState<ThiRegisters> _state;
+
+        public ThiDevice(ClassId classId, IDeviceState device, SyncptIncrManager syncptIncrMgr)
+        {
+            _classId = classId;
+            _device = device;
+            _syncptIncrMgr = syncptIncrMgr;
+            _commandQueue = new AsyncWorkQueue<CommandAction>(Process, $"Ryujinx.{classId}Processor");
+            _state = new DeviceState<ThiRegisters>(new Dictionary<string, RwCallback>
+            {
+                { nameof(ThiRegisters.IncrSyncpt), new RwCallback(IncrSyncpt, null) },
+                { nameof(ThiRegisters.Method1), new RwCallback(Method1, null) }
+            });
+
+            _previousContextId = -1;
+        }
+
+        public long CreateContext()
+        {
+            if (_device is IDeviceStateWithContext deviceWithContext)
+            {
+                return deviceWithContext.CreateContext();
+            }
+
+            return -1;
+        }
+
+        public void DestroyContext(long id)
+        {
+            if (_device is IDeviceStateWithContext deviceWithContext)
+            {
+                deviceWithContext.DestroyContext(id);
+            }
+        }
+
+        public void BindContext(long id)
+        {
+            _currentContextId = id;
+        }
+
+        public int Read(int offset) => _state.Read(offset);
+        public void Write(int offset, int data) => _state.Write(offset, data);
+
+        private void IncrSyncpt(int data)
+        {
+            uint syncpointId = (uint)(data & 0xFF);
+            uint cond = (uint)((data >> 8) & 0xFF); // 0 = Immediate, 1 = Done
+
+            if (cond == 0)
+            {
+                _syncptIncrMgr.Increment(syncpointId);
+            }
+            else
+            {
+                _commandQueue.Add(new SyncptIncrAction(_currentContextId, _syncptIncrMgr.IncrementWhenDone(_classId, syncpointId)));
+            }
+        }
+
+        private void Method1(int data)
+        {
+            _commandQueue.Add(new MethodCallAction(_currentContextId, (int)_state.State.Method0 * sizeof(uint), data));
+        }
+
+        private void Process(CommandAction cmdAction)
+        {
+            long contextId = cmdAction.ContextId;
+            if (contextId != _previousContextId)
+            {
+                _previousContextId = contextId;
+
+                if (_device is IDeviceStateWithContext deviceWithContext)
+                {
+                    deviceWithContext.BindContext(contextId);
+                }
+            }
+
+            if (cmdAction is SyncptIncrAction syncptIncrAction)
+            {
+                _syncptIncrMgr.SignalDone((uint)syncptIncrAction.Data);
+            }
+            else if (cmdAction is MethodCallAction methodCallAction)
+            {
+                _device.Write(methodCallAction.Method, methodCallAction.Data);
+            }
+        }
+
+        public void Dispose() => _commandQueue.Dispose();
+    }
+}