1 files changed, 43 insertions, 0 deletions
diff --git a/Ryujinx.Graphics.Nvdec/H264Decoder.cs b/Ryujinx.Graphics.Nvdec/H264Decoder.cs
new file mode 100644
index 00000000..57ce12d0
--- /dev/null
+++ b/Ryujinx.Graphics.Nvdec/H264Decoder.cs
@@ -0,0 +1,43 @@
+using Ryujinx.Graphics.Nvdec.H264;
+using Ryujinx.Graphics.Nvdec.Image;
+using Ryujinx.Graphics.Nvdec.Types.H264;
+using Ryujinx.Graphics.Video;
+using System;
+
+namespace Ryujinx.Graphics.Nvdec
+{
+    static class H264Decoder
+    {
+        private const int MbSizeInPixels = 16;
+
+        private static readonly Decoder _decoder = new Decoder();
+
+        public unsafe static void Decode(NvdecDevice device, ResourceManager rm, ref NvdecRegisters state)
+        {
+            PictureInfo pictureInfo = rm.Gmm.DeviceRead<PictureInfo>(state.SetPictureInfoOffset);
+            H264PictureInfo info = pictureInfo.Convert();
+
+            ReadOnlySpan<byte> bitstream = rm.Gmm.DeviceGetSpan(state.SetBitstreamOffset, (int)pictureInfo.BitstreamSize);
+
+            int width  = (int)pictureInfo.PicWidthInMbs * MbSizeInPixels;
+            int height = (int)pictureInfo.PicHeightInMbs * MbSizeInPixels;
+
+            ISurface outputSurface = rm.Cache.Get(_decoder, CodecId.H264, 0, 0, width, height);
+
+            if (_decoder.Decode(ref info, outputSurface, bitstream))
+            {
+                int li = (int)pictureInfo.LumaOutputSurfaceIndex;
+                int ci = (int)pictureInfo.ChromaOutputSurfaceIndex;
+
+                uint lumaOffset   = state.SetSurfaceLumaOffset[li];
+                uint chromaOffset = state.SetSurfaceChromaOffset[ci];
+
+                SurfaceWriter.Write(rm.Gmm, outputSurface, lumaOffset, chromaOffset);
+
+                device.OnFrameDecoded(CodecId.H264, lumaOffset, chromaOffset);
+            }
+
+            rm.Cache.Put(outputSurface);
+        }
+    }
+}