mirror of
https://github.com/Ryujinx/Ryujinx.git
synced 2024-12-25 00:46:02 +00:00
168 lines
4.5 KiB
C#
168 lines
4.5 KiB
C#
|
using FFmpeg.AutoGen;
|
||
|
using System;
|
||
|
using System.Runtime.InteropServices;
|
||
|
|
||
|
namespace Ryujinx.Graphics.VDec
|
||
|
{
|
||
|
unsafe static class FFmpegWrapper
|
||
|
{
|
||
|
private static AVCodec* Codec;
|
||
|
private static AVCodecContext* Context;
|
||
|
private static AVFrame* Frame;
|
||
|
private static SwsContext* ScalerCtx;
|
||
|
|
||
|
private static int ScalerWidth;
|
||
|
private static int ScalerHeight;
|
||
|
|
||
|
public static bool IsInitialized { get; private set; }
|
||
|
|
||
|
public static void H264Initialize()
|
||
|
{
|
||
|
EnsureCodecInitialized(AVCodecID.AV_CODEC_ID_H264);
|
||
|
}
|
||
|
|
||
|
public static void Vp9Initialize()
|
||
|
{
|
||
|
EnsureCodecInitialized(AVCodecID.AV_CODEC_ID_VP9);
|
||
|
}
|
||
|
|
||
|
private static void EnsureCodecInitialized(AVCodecID CodecId)
|
||
|
{
|
||
|
if (IsInitialized)
|
||
|
{
|
||
|
Uninitialize();
|
||
|
}
|
||
|
|
||
|
Codec = ffmpeg.avcodec_find_decoder(CodecId);
|
||
|
Context = ffmpeg.avcodec_alloc_context3(Codec);
|
||
|
Frame = ffmpeg.av_frame_alloc();
|
||
|
|
||
|
ffmpeg.avcodec_open2(Context, Codec, null);
|
||
|
|
||
|
IsInitialized = true;
|
||
|
}
|
||
|
|
||
|
public static int DecodeFrame(byte[] Data)
|
||
|
{
|
||
|
if (!IsInitialized)
|
||
|
{
|
||
|
throw new InvalidOperationException("Tried to use uninitialized codec!");
|
||
|
}
|
||
|
|
||
|
AVPacket Packet;
|
||
|
|
||
|
ffmpeg.av_init_packet(&Packet);
|
||
|
|
||
|
fixed (byte* Ptr = Data)
|
||
|
{
|
||
|
Packet.data = Ptr;
|
||
|
Packet.size = Data.Length;
|
||
|
|
||
|
ffmpeg.avcodec_send_packet(Context, &Packet);
|
||
|
}
|
||
|
|
||
|
return ffmpeg.avcodec_receive_frame(Context, Frame);
|
||
|
}
|
||
|
|
||
|
public static FFmpegFrame GetFrame()
|
||
|
{
|
||
|
if (!IsInitialized)
|
||
|
{
|
||
|
throw new InvalidOperationException("Tried to use uninitialized codec!");
|
||
|
}
|
||
|
|
||
|
AVFrame ManagedFrame = Marshal.PtrToStructure<AVFrame>((IntPtr)Frame);
|
||
|
|
||
|
byte*[] Data = ManagedFrame.data.ToArray();
|
||
|
|
||
|
return new FFmpegFrame()
|
||
|
{
|
||
|
Width = ManagedFrame.width,
|
||
|
Height = ManagedFrame.height,
|
||
|
|
||
|
LumaPtr = Data[0],
|
||
|
ChromaBPtr = Data[1],
|
||
|
ChromaRPtr = Data[2]
|
||
|
};
|
||
|
}
|
||
|
|
||
|
public static FFmpegFrame GetFrameRgba()
|
||
|
{
|
||
|
if (!IsInitialized)
|
||
|
{
|
||
|
throw new InvalidOperationException("Tried to use uninitialized codec!");
|
||
|
}
|
||
|
|
||
|
AVFrame ManagedFrame = Marshal.PtrToStructure<AVFrame>((IntPtr)Frame);
|
||
|
|
||
|
EnsureScalerSetup(ManagedFrame.width, ManagedFrame.height);
|
||
|
|
||
|
byte*[] Data = ManagedFrame.data.ToArray();
|
||
|
|
||
|
int[] LineSizes = ManagedFrame.linesize.ToArray();
|
||
|
|
||
|
byte[] Dst = new byte[ManagedFrame.width * ManagedFrame.height * 4];
|
||
|
|
||
|
fixed (byte* Ptr = Dst)
|
||
|
{
|
||
|
byte*[] DstData = new byte*[] { Ptr };
|
||
|
|
||
|
int[] DstLineSizes = new int[] { ManagedFrame.width * 4 };
|
||
|
|
||
|
ffmpeg.sws_scale(ScalerCtx, Data, LineSizes, 0, ManagedFrame.height, DstData, DstLineSizes);
|
||
|
}
|
||
|
|
||
|
return new FFmpegFrame()
|
||
|
{
|
||
|
Width = ManagedFrame.width,
|
||
|
Height = ManagedFrame.height,
|
||
|
|
||
|
Data = Dst
|
||
|
};
|
||
|
}
|
||
|
|
||
|
private static void EnsureScalerSetup(int Width, int Height)
|
||
|
{
|
||
|
if (Width == 0 || Height == 0)
|
||
|
{
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if (ScalerCtx == null || ScalerWidth != Width || ScalerHeight != Height)
|
||
|
{
|
||
|
FreeScaler();
|
||
|
|
||
|
ScalerCtx = ffmpeg.sws_getContext(
|
||
|
Width, Height, AVPixelFormat.AV_PIX_FMT_YUV420P,
|
||
|
Width, Height, AVPixelFormat.AV_PIX_FMT_RGBA, 0, null, null, null);
|
||
|
|
||
|
ScalerWidth = Width;
|
||
|
ScalerHeight = Height;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public static void Uninitialize()
|
||
|
{
|
||
|
if (IsInitialized)
|
||
|
{
|
||
|
ffmpeg.av_frame_unref(Frame);
|
||
|
ffmpeg.av_free(Frame);
|
||
|
ffmpeg.avcodec_close(Context);
|
||
|
|
||
|
FreeScaler();
|
||
|
|
||
|
IsInitialized = false;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
private static void FreeScaler()
|
||
|
{
|
||
|
if (ScalerCtx != null)
|
||
|
{
|
||
|
ffmpeg.sws_freeContext(ScalerCtx);
|
||
|
|
||
|
ScalerCtx = null;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|