oak-editor/app/codec/conformmanager.cpp

#include "conformmanager.h"

#include <QDir>

#include "task/taskmanager.h"

namespace olive
{

ConformManager *ConformManager::instance_ = nullptr;

ConformManager::Conform ConformManager::GetConformState(
	const QString &decoder_id, const QString &cache_path,
	const Decoder::CodecStream &stream, const AudioParams &params, bool wait)
{
	// Mutex because we'll need to check the status of a conform task
	QMutexLocker locker(&mutex_);

	// Return existing conform if exists
	QVector<QString> filenames =
		GetConformedFilename(cache_path, stream, params);
	if (AllConformsExist(filenames)) {
		return { kConformExists, filenames, nullptr };
	}

	ConformTask *conforming_task = nullptr;

	foreach (const ConformData &data, conforming_) {
		if (data.stream == stream && data.params == params) {
			// Already creating conform in a task
			conforming_task = data.task;
			break;
		}
	}

	if (!conforming_task) {
		// Not conforming yet, create a task to do so

		// We conform to a different filename until it's done to make it clear even across sessions
		// whether this conform is ready or not
		QVector<QString> working_filenames = filenames;
		for (int i = 0; i < working_filenames.size(); i++) {
			working_filenames[i].append(QStringLiteral(".working"));
		}

		conforming_task =
			new ConformTask(decoder_id, stream, params, working_filenames);
		connect(conforming_task, &ConformTask::Finished, this,
				&ConformManager::ConformTaskFinished);
		conforming_task->moveToThread(TaskManager::instance()->thread());
		QMetaObject::invokeMethod(TaskManager::instance(), "AddTask",
								  Qt::QueuedConnection,
								  Q_ARG(Task *, conforming_task));

		conforming_.append(
			{ stream, params, conforming_task, working_filenames, filenames });
	}

	if (wait) {
		do {
			conform_done_condition_.wait(&mutex_);
		} while (!AllConformsExist(filenames));
		return { kConformExists, filenames, nullptr };
	}

	return { kConformGenerating, QVector<QString>(), conforming_task };
}

QVector<QString>
ConformManager::GetConformedFilename(const QString &cache_path,
									 const Decoder::CodecStream &stream,
									 const AudioParams &params)
{
	QVector<QString> filenames(params.channel_count());

	for (int i = 0; i < filenames.size(); i++) {
		QString index_fn =
			QStringLiteral("%1-%2.%3.%4.%5.%6.pcm")
				.arg(FileFunctions::GetUniqueFileIdentifier(stream.filename()),
					 QString::number(stream.stream()),
					 QString::number(params.sample_rate()),
					 QString::number(params.format()),
					 QString::number(params.channel_layout().u.mask),
					 QString::number(i));

		filenames[i] = QDir(cache_path).filePath(index_fn);
	}

	return filenames;
}

bool ConformManager::AllConformsExist(const QVector<QString> &filenames)
{
	foreach (const QString &fn, filenames) {
		if (!QFileInfo::exists(fn)) {
			return false;
		}
	}

	return true;
}

void ConformManager::ConformTaskFinished(Task *task, bool succeeded)
{
	QMutexLocker locker(&mutex_);

	ConformData data;

	// Remove conform data from list
	for (int i = 0; i < conforming_.size(); i++) {
		const ConformData &c = conforming_.at(i);
		if (c.task == task) {
			data = c;
			conforming_.removeAt(i);
			break;
		}
	}

	if (succeeded) {
		// Move file to standard conform name, making it clear this conform is ready for use
		for (int i = 0; i < data.finished_filename.size(); i++) {
			const QString &finished = data.finished_filename.at(i);
			const QString &working = data.working_filename.at(i);

			QFile::remove(finished);
			QFile::rename(working, finished);
		}

		conform_done_condition_.wakeAll();
		locker.unlock();
		emit ConformReady();
	} else {
		// Failed, just delete the working filename if exists
		for (int i = 0; i < data.working_filename.size(); i++) {
			QFile::remove(data.working_filename.at(i));
		}
	}
}

}