Mesen-SX/Utilities/AviRecorder.cpp

#include "stdafx.h"
#include "AviRecorder.h"

AviRecorder::AviRecorder(VideoCodec codec, uint32_t compressionLevel)
{
	_recording = false;
	_stopFlag = false;
	_frameBuffer = nullptr;
	_frameBufferLength = 0;
	_sampleRate = 0;
	_codec = codec;
	_compressionLevel = compressionLevel;
}

AviRecorder::~AviRecorder()
{
	if (_recording)
	{
		StopRecording();
	}

	if (_frameBuffer)
	{
		delete[] _frameBuffer;
		_frameBuffer = nullptr;
	}
}

bool AviRecorder::StartRecording(string filename, uint32_t width, uint32_t height, uint32_t bpp,
                                 uint32_t audioSampleRate, double fps)
{
	if (!_recording)
	{
		_outputFile = filename;
		_sampleRate = audioSampleRate;
		_width = width;
		_height = height;
		_fps = fps;
		_frameBufferLength = height * width * bpp;
		_frameBuffer = new uint8_t[_frameBufferLength];

		_aviWriter.reset(new AviWriter());
		if (!_aviWriter->StartWrite(filename, _codec, width, height, bpp, (uint32_t)(_fps * 1000000), audioSampleRate,
		                            _compressionLevel))
		{
			_aviWriter.reset();
			return false;
		}

		_aviWriterThread = std::thread([=]()
		{
			while (!_stopFlag)
			{
				_waitFrame.Wait();
				if (_stopFlag)
				{
					break;
				}

				auto lock = _lock.AcquireSafe();
				_aviWriter->AddFrame(_frameBuffer);
			}
		});

		_recording = true;
	}
	return true;
}

void AviRecorder::StopRecording()
{
	if (_recording)
	{
		_recording = false;

		_stopFlag = true;
		_waitFrame.Signal();
		_aviWriterThread.join();

		_aviWriter->EndWrite();
		_aviWriter.reset();
	}
}

void AviRecorder::AddFrame(void* frameBuffer, uint32_t width, uint32_t height, double fps)
{
	if (_recording)
	{
		if (_width != width || _height != height || _fps != fps)
		{
			StopRecording();
		}
		else
		{
			auto lock = _lock.AcquireSafe();
			memcpy(_frameBuffer, frameBuffer, _frameBufferLength);
			_waitFrame.Signal();
		}
	}
}

void AviRecorder::AddSound(int16_t* soundBuffer, uint32_t sampleCount, uint32_t sampleRate)
{
	if (_recording)
	{
		if (_sampleRate != sampleRate)
		{
			auto lock = _lock.AcquireSafe();
			StopRecording();
		}
		else
		{
			_aviWriter->AddSound(soundBuffer, sampleCount);
		}
	}
}

bool AviRecorder::IsRecording()
{
	return _recording;
}

string AviRecorder::GetOutputFile()
{
	return _outputFile;
}