Add support for framedumping to OpenGL ES.

This commit is contained in:
Ryan Houdek 2016-01-01 08:50:15 -06:00
parent af9beecc10
commit 3f15aa4b57
4 changed files with 66 additions and 56 deletions

View File

@ -840,7 +840,7 @@ void Renderer::SwapImpl(u32 xfbAddr, u32 fbWidth, u32 fbStride, u32 fbHeight, co
{ {
s_recordWidth = source_width; s_recordWidth = source_width;
s_recordHeight = source_height; s_recordHeight = source_height;
bAVIDumping = AVIDump::Start(s_recordWidth, s_recordHeight); bAVIDumping = AVIDump::Start(s_recordWidth, s_recordHeight, AVIDump::DumpFormat::FORMAT_BGR);
if (!bAVIDumping) if (!bAVIDumping)
{ {
PanicAlert("Error dumping frames to AVI."); PanicAlert("Error dumping frames to AVI.");

View File

@ -1336,27 +1336,26 @@ void Renderer::SwapImpl(u32 xfbAddr, u32 fbWidth, u32 fbStride, u32 fbHeight, co
} }
// Frame dumps are handled a little differently in Windows // Frame dumps are handled a little differently in Windows
// Frame dumping disabled entirely on GLES3
if (GLInterface->GetMode() == GLInterfaceMode::MODE_OPENGL)
{
#if defined(HAVE_LIBAV) || defined (_WIN32) #if defined(HAVE_LIBAV) || defined (_WIN32)
if (SConfig::GetInstance().m_DumpFrames) if (SConfig::GetInstance().m_DumpFrames)
{ {
std::lock_guard<std::mutex> lk(s_criticalScreenshot); std::lock_guard<std::mutex> lk(s_criticalScreenshot);
if (frame_data.empty() || w != flipped_trc.GetWidth() ||
h != flipped_trc.GetHeight()) if (frame_data.empty() ||
w != flipped_trc.GetWidth() || h != flipped_trc.GetHeight())
{ {
w = flipped_trc.GetWidth(); w = flipped_trc.GetWidth();
h = flipped_trc.GetHeight(); h = flipped_trc.GetHeight();
frame_data.resize(3 * w * h); frame_data.resize(4 * w * h);
} }
glPixelStorei(GL_PACK_ALIGNMENT, 1); glPixelStorei(GL_PACK_ALIGNMENT, 1);
glReadPixels(flipped_trc.left, flipped_trc.bottom, w, h, GL_BGR, GL_UNSIGNED_BYTE, &frame_data[0]); glReadPixels(flipped_trc.left, flipped_trc.bottom, w, h, GL_RGBA, GL_UNSIGNED_BYTE, &frame_data[0]);
if (w > 0 && h > 0) if (w > 0 && h > 0)
{ {
if (!bLastFrameDumped) if (!bLastFrameDumped)
{ {
bAVIDumping = AVIDump::Start(w, h); bAVIDumping = AVIDump::Start(w, h, AVIDump::DumpFormat::FORMAT_RGBA);
if (!bAVIDumping) if (!bAVIDumping)
{ {
OSD::AddMessage("AVIDump Start failed", 2000); OSD::AddMessage("AVIDump Start failed", 2000);
@ -1370,7 +1369,7 @@ void Renderer::SwapImpl(u32 xfbAddr, u32 fbWidth, u32 fbStride, u32 fbHeight, co
} }
if (bAVIDumping) if (bAVIDumping)
{ {
FlipImageData(&frame_data[0], w, h); FlipImageData(&frame_data[0], w, h, 4);
AVIDump::AddFrame(&frame_data[0], w, h); AVIDump::AddFrame(&frame_data[0], w, h);
} }
@ -1394,7 +1393,6 @@ void Renderer::SwapImpl(u32 xfbAddr, u32 fbWidth, u32 fbStride, u32 fbHeight, co
bLastFrameDumped = false; bLastFrameDumped = false;
} }
#endif #endif
}
// Finish up the current frame, print some stats // Finish up the current frame, print some stats
SetWindowSize(fbStride, fbHeight); SetWindowSize(fbStride, fbHeight);

View File

@ -36,6 +36,7 @@ static AVFormatContext* s_format_context = nullptr;
static AVStream* s_stream = nullptr; static AVStream* s_stream = nullptr;
static AVFrame* s_src_frame = nullptr; static AVFrame* s_src_frame = nullptr;
static AVFrame* s_scaled_frame = nullptr; static AVFrame* s_scaled_frame = nullptr;
AVPixelFormat s_pix_fmt = AV_PIX_FMT_BGR24;
static uint8_t* s_yuv_buffer = nullptr; static uint8_t* s_yuv_buffer = nullptr;
static SwsContext* s_sws_context = nullptr; static SwsContext* s_sws_context = nullptr;
static int s_width; static int s_width;
@ -55,8 +56,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_width = w;
s_height = h; s_height = h;
@ -148,12 +154,12 @@ static void PreparePacket(AVPacket* pkt)
void AVIDump::AddFrame(const u8* data, int width, int height) 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 // width and height
if ((s_sws_context = sws_getCachedContext(s_sws_context, 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, s_width, s_height, s_stream->codec->pix_fmt,
SWS_BICUBIC, nullptr, nullptr, nullptr))) SWS_BICUBIC, nullptr, nullptr, nullptr)))
{ {

View File

@ -13,7 +13,13 @@ private:
static void CloseFile(); static void CloseFile();
public: public:
static bool Start(int w, int h); enum class DumpFormat
{
FORMAT_BGR,
FORMAT_RGBA
};
static bool Start(int w, int h, DumpFormat format);
static void AddFrame(const u8* data, int width, int height); static void AddFrame(const u8* data, int width, int height);
static void Stop(); static void Stop();
static void DoState(); static void DoState();