1 files changed, 83 insertions, 0 deletions
diff --git a/src/Ryujinx.Graphics.Nvdec/NvdecDevice.cs b/src/Ryujinx.Graphics.Nvdec/NvdecDevice.cs
new file mode 100644
index 00000000..ef8185f4
--- /dev/null
+++ b/src/Ryujinx.Graphics.Nvdec/NvdecDevice.cs
@@ -0,0 +1,83 @@
+using Ryujinx.Common.Logging;
+using Ryujinx.Graphics.Device;
+using Ryujinx.Graphics.Gpu.Memory;
+using Ryujinx.Graphics.Nvdec.Image;
+using System.Collections.Concurrent;
+using System.Collections.Generic;
+using System.Threading;
+
+namespace Ryujinx.Graphics.Nvdec
+{
+    public class NvdecDevice : IDeviceStateWithContext
+    {
+        private readonly ResourceManager _rm;
+        private readonly DeviceState<NvdecRegisters> _state;
+
+        private long _currentId;
+        private ConcurrentDictionary<long, NvdecDecoderContext> _contexts;
+        private NvdecDecoderContext _currentContext;
+
+        public NvdecDevice(MemoryManager gmm)
+        {
+            _rm = new ResourceManager(gmm, new SurfaceCache(gmm));
+            _state = new DeviceState<NvdecRegisters>(new Dictionary<string, RwCallback>
+            {
+                { nameof(NvdecRegisters.Execute), new RwCallback(Execute, null) }
+            });
+            _contexts = new ConcurrentDictionary<long, NvdecDecoderContext>();
+        }
+
+        public long CreateContext()
+        {
+            long id = Interlocked.Increment(ref _currentId);
+            _contexts.TryAdd(id, new NvdecDecoderContext());
+
+            return id;
+        }
+
+        public void DestroyContext(long id)
+        {
+            if (_contexts.TryRemove(id, out var context))
+            {
+                context.Dispose();
+            }
+
+            _rm.Cache.Trim();
+        }
+
+        public void BindContext(long id)
+        {
+            if (_contexts.TryGetValue(id, out var context))
+            {
+                _currentContext = context;
+            }
+        }
+
+        public int Read(int offset) => _state.Read(offset);
+        public void Write(int offset, int data) => _state.Write(offset, data);
+
+        private void Execute(int data)
+        {
+            Decode((ApplicationId)_state.State.SetApplicationId);
+        }
+
+        private void Decode(ApplicationId applicationId)
+        {
+            switch (applicationId)
+            {
+                case ApplicationId.H264:
+                    H264Decoder.Decode(_currentContext, _rm, ref _state.State);
+                    break;
+                case ApplicationId.Vp8:
+                    Vp8Decoder.Decode(_currentContext, _rm, ref _state.State);
+                    break;
+                case ApplicationId.Vp9:
+                    Vp9Decoder.Decode(_rm, ref _state.State);
+                    break;
+                default:
+                    Logger.Error?.Print(LogClass.Nvdec, $"Unsupported codec \"{applicationId}\".");
+                    break;
+            }
+        }
+    }
+}