diff options
Diffstat (limited to 'Source/Core/VideoCommon/AVIDump.cpp')
| -rw-r--r-- | Source/Core/VideoCommon/AVIDump.cpp | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/Source/Core/VideoCommon/AVIDump.cpp b/Source/Core/VideoCommon/AVIDump.cpp index f6f7783b16..28292ea9c5 100644 --- a/Source/Core/VideoCommon/AVIDump.cpp +++ b/Source/Core/VideoCommon/AVIDump.cpp @@ -35,6 +35,7 @@ static AVFormatContext* s_format_context = nullptr; static AVStream* s_stream = nullptr; static AVFrame* s_src_frame = nullptr; static AVFrame* s_scaled_frame = nullptr; +AVPixelFormat s_pix_fmt = AV_PIX_FMT_BGR24; static uint8_t* s_yuv_buffer = nullptr; static SwsContext* s_sws_context = nullptr; static int s_width; @@ -54,8 +55,13 @@ static void InitAVCodec() } } -bool AVIDump::Start(int w, int h) +bool AVIDump::Start(int w, int h, DumpFormat format) { + if (format == DumpFormat::FORMAT_BGR) + s_pix_fmt = AV_PIX_FMT_BGR24; + else + s_pix_fmt = AV_PIX_FMT_RGBA; + s_width = w; s_height = h; @@ -147,12 +153,12 @@ static void PreparePacket(AVPacket* pkt) void AVIDump::AddFrame(const u8* data, int width, int height) { - avpicture_fill((AVPicture*)s_src_frame, const_cast<u8*>(data), AV_PIX_FMT_BGR24, width, height); + avpicture_fill((AVPicture*)s_src_frame, const_cast<u8*>(data), s_pix_fmt, width, height); - // Convert image from BGR24 to desired pixel format, and scale to initial + // Convert image from {BGR24, RGBA} to desired pixel format, and scale to initial // width and height if ((s_sws_context = sws_getCachedContext(s_sws_context, - width, height, AV_PIX_FMT_BGR24, + width, height, s_pix_fmt, s_width, s_height, s_stream->codec->pix_fmt, SWS_BICUBIC, nullptr, nullptr, nullptr))) { |