1 files changed, 51 insertions, 0 deletions
diff --git a/Ryujinx.Graphics.Nvdec.H264/FFmpegContext.cs b/Ryujinx.Graphics.Nvdec.H264/FFmpegContext.cs
new file mode 100644
index 00000000..b4f9206b
--- /dev/null
+++ b/Ryujinx.Graphics.Nvdec.H264/FFmpegContext.cs
@@ -0,0 +1,51 @@
+using FFmpeg.AutoGen;
+using System;
+
+namespace Ryujinx.Graphics.Nvdec.H264
+{
+    unsafe class FFmpegContext : IDisposable
+    {
+        private readonly AVCodec* _codec;
+        private AVCodecContext* _context;
+
+        public FFmpegContext()
+        {
+            _codec = ffmpeg.avcodec_find_decoder(AVCodecID.AV_CODEC_ID_H264);
+            _context = ffmpeg.avcodec_alloc_context3(_codec);
+
+            ffmpeg.avcodec_open2(_context, _codec, null);
+        }
+
+        public int DecodeFrame(Surface output, ReadOnlySpan<byte> bitstream)
+        {
+            AVPacket packet;
+
+            ffmpeg.av_init_packet(&packet);
+
+            fixed (byte* ptr = bitstream)
+            {
+                packet.data = ptr;
+                packet.size = bitstream.Length;
+
+                int rc = ffmpeg.avcodec_send_packet(_context, &packet);
+
+                if (rc != 0)
+                {
+                    return rc;
+                }
+            }
+
+            return ffmpeg.avcodec_receive_frame(_context, output.Frame);
+        }
+
+        public void Dispose()
+        {
+            ffmpeg.avcodec_close(_context);
+
+            fixed (AVCodecContext** ppContext = &_context)
+            {
+                ffmpeg.avcodec_free_context(ppContext);
+            }
+        }
+    }
+}