Implement proxy media workflow

2026-07-13 10:19:29 +08:00
parent 9c1fba4932
commit 8ae8e3a159
18 changed files with 1671 additions and 5 deletions
@@ -0,0 +1,114 @@
+/*
+ * Oak Video Editor - Non-Linear Video Editor
+ * Copyright (C) 2026 Oak Team
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "proxy.h"
+
+#include <QDir>
+#include <QFile>
+#include <QFileInfo>
+#include <QProcess>
+#include <QStandardPaths>
+
+namespace olive
+{
+
+ProxyTask::ProxyTask(const QString &source_filename,
+					 int stream_index,
+					 const ProxyManager::ProxyParams &params,
+					 const QString &output_filename)
+	: source_filename_(source_filename)
+	, stream_index_(stream_index)
+	, params_(params)
+	, output_filename_(output_filename)
+{
+	SetTitle(tr("Generating Proxy %1:%2")
+				 .arg(source_filename_, QString::number(stream_index_)));
+}
+
+bool ProxyTask::Run()
+{
+	const QString ffmpeg = QStandardPaths::findExecutable(QStringLiteral("ffmpeg"));
+	if (ffmpeg.isEmpty()) {
+		SetError(tr("Failed to generate proxy: ffmpeg executable was not found"));
+		return false;
+	}
+
+	QDir output_dir = QFileInfo(output_filename_).dir();
+	if (!output_dir.exists() && !output_dir.mkpath(QStringLiteral("."))) {
+		SetError(tr("Failed to create proxy output directory"));
+		return false;
+	}
+
+	QFile::remove(output_filename_);
+
+	const QString scale_filter = QStringLiteral(
+		"scale=w=%1:h=%2:force_original_aspect_ratio=decrease")
+							.arg(QString::number(params_.width),
+								 QString::number(params_.height));
+
+	QStringList args;
+	args << QStringLiteral("-y")
+		 << QStringLiteral("-i") << source_filename_
+		 << QStringLiteral("-map") << QStringLiteral("0:%1").arg(stream_index_)
+		 << QStringLiteral("-an")
+		 << QStringLiteral("-vf") << scale_filter
+		 << QStringLiteral("-c:v") << QStringLiteral("libx264")
+		 << QStringLiteral("-preset") << QStringLiteral("veryfast")
+		 << QStringLiteral("-crf") << QStringLiteral("23")
+		 << QStringLiteral("-pix_fmt") << QStringLiteral("yuv420p")
+		 << QStringLiteral("-movflags") << QStringLiteral("+faststart")
+		 << output_filename_;
+
+	QProcess process;
+	process.setProgram(ffmpeg);
+	process.setArguments(args);
+	process.setProcessChannelMode(QProcess::MergedChannels);
+	process.start();
+
+	if (!process.waitForStarted()) {
+		SetError(tr("Failed to start ffmpeg for proxy generation"));
+		return false;
+	}
+
+	while (!process.waitForFinished(100)) {
+		if (IsCancelled()) {
+			process.kill();
+			process.waitForFinished();
+			QFile::remove(output_filename_);
+			SetError(tr("Proxy generation was cancelled"));
+			return false;
+		}
+	}
+
+	if (process.exitStatus() != QProcess::NormalExit || process.exitCode() != 0) {
+		const QString output = QString::fromUtf8(process.readAll()).trimmed();
+		QFile::remove(output_filename_);
+		SetError(tr("ffmpeg failed to generate proxy: %1").arg(output));
+		return false;
+	}
+
+	if (!QFileInfo::exists(output_filename_)) {
+		SetError(tr("ffmpeg finished but proxy file was not created"));
+		return false;
+	}
+
+	emit ProgressChanged(1.0);
+	return true;
+}
+
+}