frontend: Split Qt UI dialogs into single files per C++ class

This commit is contained in:
PatTheMav
2025-01-08 15:36:55 +01:00
parent e4a43f6555
commit 085c6245b0
27 changed files with 114 additions and 7119 deletions
+7 -720
View File
@@ -15,44 +15,18 @@
along with this program. If not, see <http://www.gnu.org/licenses/>.
******************************************************************************/
#include "moc_window-remux.cpp"
#include "RemuxEntryPathItemDelegate.hpp"
#include "RemuxQueueModel.hpp"
#include "obs-app.hpp"
#include <OBSApp.hpp>
#include <QCloseEvent>
#include <QDirIterator>
#include <QItemDelegate>
#include <QLineEdit>
#include <QMessageBox>
#include <QMimeData>
#include <QPainter>
#include <QPushButton>
#include <QStandardItemModel>
#include <QStyledItemDelegate>
#include <QToolButton>
#include <QTimer>
#include <qt-wrappers.hpp>
#include "window-basic-main.hpp"
#include <QHBoxLayout>
#include <QLineEdit>
#include <QToolButton>
#include <memory>
#include <cmath>
using namespace std;
enum RemuxEntryColumn {
State,
InputPath,
OutputPath,
Count
};
enum RemuxEntryRole { EntryStateRole = Qt::UserRole, NewPathsToProcessRole };
/**********************************************************
Delegate - Presents cells in the grid.
**********************************************************/
#include "moc_RemuxEntryPathItemDelegate.cpp"
RemuxEntryPathItemDelegate::RemuxEntryPathItemDelegate(bool isOutput, const QString &defaultPath)
: QStyledItemDelegate(),
@@ -235,690 +209,3 @@ void RemuxEntryPathItemDelegate::updateText()
QWidget *editor = lineEdit->parentWidget();
emit commitData(editor);
}
/**********************************************************
Model - Manages the queue's data
**********************************************************/
int RemuxQueueModel::rowCount(const QModelIndex &) const
{
return queue.length() + (isProcessing ? 0 : 1);
}
int RemuxQueueModel::columnCount(const QModelIndex &) const
{
return RemuxEntryColumn::Count;
}
QVariant RemuxQueueModel::data(const QModelIndex &index, int role) const
{
QVariant result = QVariant();
if (index.row() >= queue.length()) {
return QVariant();
} else if (role == Qt::DisplayRole) {
switch (index.column()) {
case RemuxEntryColumn::InputPath:
result = queue[index.row()].sourcePath;
break;
case RemuxEntryColumn::OutputPath:
result = queue[index.row()].targetPath;
break;
}
} else if (role == Qt::DecorationRole && index.column() == RemuxEntryColumn::State) {
result = getIcon(queue[index.row()].state);
} else if (role == RemuxEntryRole::EntryStateRole) {
result = queue[index.row()].state;
}
return result;
}
QVariant RemuxQueueModel::headerData(int section, Qt::Orientation orientation, int role) const
{
QVariant result = QVariant();
if (role == Qt::DisplayRole && orientation == Qt::Orientation::Horizontal) {
switch (section) {
case RemuxEntryColumn::State:
result = QString();
break;
case RemuxEntryColumn::InputPath:
result = QTStr("Remux.SourceFile");
break;
case RemuxEntryColumn::OutputPath:
result = QTStr("Remux.TargetFile");
break;
}
}
return result;
}
Qt::ItemFlags RemuxQueueModel::flags(const QModelIndex &index) const
{
Qt::ItemFlags flags = QAbstractTableModel::flags(index);
if (index.column() == RemuxEntryColumn::InputPath) {
flags |= Qt::ItemIsEditable;
} else if (index.column() == RemuxEntryColumn::OutputPath && index.row() != queue.length()) {
flags |= Qt::ItemIsEditable;
}
return flags;
}
bool RemuxQueueModel::setData(const QModelIndex &index, const QVariant &value, int role)
{
bool success = false;
if (role == RemuxEntryRole::NewPathsToProcessRole) {
QStringList pathList = value.toStringList();
if (pathList.size() == 0) {
if (index.row() < queue.size()) {
beginRemoveRows(QModelIndex(), index.row(), index.row());
queue.removeAt(index.row());
endRemoveRows();
}
} else {
if (pathList.size() >= 1 && index.row() < queue.length()) {
queue[index.row()].sourcePath = pathList[0];
checkInputPath(index.row());
pathList.removeAt(0);
success = true;
}
if (pathList.size() > 0) {
int row = index.row();
int lastRow = row + pathList.size() - 1;
beginInsertRows(QModelIndex(), row, lastRow);
for (QString path : pathList) {
RemuxQueueEntry entry;
entry.sourcePath = path;
entry.state = RemuxEntryState::Empty;
queue.insert(row, entry);
row++;
}
endInsertRows();
for (row = index.row(); row <= lastRow; row++) {
checkInputPath(row);
}
success = true;
}
}
} else if (index.row() == queue.length()) {
QString path = value.toString();
if (!path.isEmpty()) {
RemuxQueueEntry entry;
entry.sourcePath = path;
entry.state = RemuxEntryState::Empty;
beginInsertRows(QModelIndex(), queue.length() + 1, queue.length() + 1);
queue.append(entry);
endInsertRows();
checkInputPath(index.row());
success = true;
}
} else {
QString path = value.toString();
if (path.isEmpty()) {
if (index.column() == RemuxEntryColumn::InputPath) {
beginRemoveRows(QModelIndex(), index.row(), index.row());
queue.removeAt(index.row());
endRemoveRows();
}
} else {
switch (index.column()) {
case RemuxEntryColumn::InputPath:
queue[index.row()].sourcePath = value.toString();
checkInputPath(index.row());
success = true;
break;
case RemuxEntryColumn::OutputPath:
queue[index.row()].targetPath = value.toString();
emit dataChanged(index, index);
success = true;
break;
}
}
}
return success;
}
QVariant RemuxQueueModel::getIcon(RemuxEntryState state)
{
QVariant icon;
QStyle *style = QApplication::style();
switch (state) {
case RemuxEntryState::Complete:
icon = style->standardIcon(QStyle::SP_DialogApplyButton);
break;
case RemuxEntryState::InProgress:
icon = style->standardIcon(QStyle::SP_ArrowRight);
break;
case RemuxEntryState::Error:
icon = style->standardIcon(QStyle::SP_DialogCancelButton);
break;
case RemuxEntryState::InvalidPath:
icon = style->standardIcon(QStyle::SP_MessageBoxWarning);
break;
default:
break;
}
return icon;
}
void RemuxQueueModel::checkInputPath(int row)
{
RemuxQueueEntry &entry = queue[row];
if (entry.sourcePath.isEmpty()) {
entry.state = RemuxEntryState::Empty;
} else {
entry.sourcePath = QDir::toNativeSeparators(entry.sourcePath);
QFileInfo fileInfo(entry.sourcePath);
if (fileInfo.exists())
entry.state = RemuxEntryState::Ready;
else
entry.state = RemuxEntryState::InvalidPath;
QString newExt = ".mp4";
QString suffix = fileInfo.suffix();
if (suffix.contains("mov", Qt::CaseInsensitive) || suffix.contains("mp4", Qt::CaseInsensitive)) {
newExt = ".remuxed." + suffix;
}
if (entry.state == RemuxEntryState::Ready)
entry.targetPath = QDir::toNativeSeparators(fileInfo.path() + QDir::separator() +
fileInfo.completeBaseName() + newExt);
}
if (entry.state == RemuxEntryState::Ready && isProcessing)
entry.state = RemuxEntryState::Pending;
emit dataChanged(index(row, 0), index(row, RemuxEntryColumn::Count));
}
QFileInfoList RemuxQueueModel::checkForOverwrites() const
{
QFileInfoList list;
for (const RemuxQueueEntry &entry : queue) {
if (entry.state == RemuxEntryState::Ready) {
QFileInfo fileInfo(entry.targetPath);
if (fileInfo.exists()) {
list.append(fileInfo);
}
}
}
return list;
}
bool RemuxQueueModel::checkForErrors() const
{
bool hasErrors = false;
for (const RemuxQueueEntry &entry : queue) {
if (entry.state == RemuxEntryState::Error) {
hasErrors = true;
break;
}
}
return hasErrors;
}
void RemuxQueueModel::clearAll()
{
beginRemoveRows(QModelIndex(), 0, queue.size() - 1);
queue.clear();
endRemoveRows();
}
void RemuxQueueModel::clearFinished()
{
int index = 0;
for (index = 0; index < queue.size(); index++) {
const RemuxQueueEntry &entry = queue[index];
if (entry.state == RemuxEntryState::Complete) {
beginRemoveRows(QModelIndex(), index, index);
queue.removeAt(index);
endRemoveRows();
index--;
}
}
}
bool RemuxQueueModel::canClearFinished() const
{
bool canClearFinished = false;
for (const RemuxQueueEntry &entry : queue)
if (entry.state == RemuxEntryState::Complete) {
canClearFinished = true;
break;
}
return canClearFinished;
}
void RemuxQueueModel::beginProcessing()
{
for (RemuxQueueEntry &entry : queue)
if (entry.state == RemuxEntryState::Ready)
entry.state = RemuxEntryState::Pending;
// Signal that the insertion point no longer exists.
beginRemoveRows(QModelIndex(), queue.length(), queue.length());
endRemoveRows();
isProcessing = true;
emit dataChanged(index(0, RemuxEntryColumn::State), index(queue.length(), RemuxEntryColumn::State));
}
void RemuxQueueModel::endProcessing()
{
for (RemuxQueueEntry &entry : queue) {
if (entry.state == RemuxEntryState::Pending) {
entry.state = RemuxEntryState::Ready;
}
}
// Signal that the insertion point exists again.
isProcessing = false;
if (!autoRemux) {
beginInsertRows(QModelIndex(), queue.length(), queue.length());
endInsertRows();
}
emit dataChanged(index(0, RemuxEntryColumn::State), index(queue.length(), RemuxEntryColumn::State));
}
bool RemuxQueueModel::beginNextEntry(QString &inputPath, QString &outputPath)
{
bool anyStarted = false;
for (int row = 0; row < queue.length(); row++) {
RemuxQueueEntry &entry = queue[row];
if (entry.state == RemuxEntryState::Pending) {
entry.state = RemuxEntryState::InProgress;
inputPath = entry.sourcePath;
outputPath = entry.targetPath;
QModelIndex index = this->index(row, RemuxEntryColumn::State);
emit dataChanged(index, index);
anyStarted = true;
break;
}
}
return anyStarted;
}
void RemuxQueueModel::finishEntry(bool success)
{
for (int row = 0; row < queue.length(); row++) {
RemuxQueueEntry &entry = queue[row];
if (entry.state == RemuxEntryState::InProgress) {
if (success)
entry.state = RemuxEntryState::Complete;
else
entry.state = RemuxEntryState::Error;
QModelIndex index = this->index(row, RemuxEntryColumn::State);
emit dataChanged(index, index);
break;
}
}
}
/**********************************************************
The actual remux window implementation
**********************************************************/
OBSRemux::OBSRemux(const char *path, QWidget *parent, bool autoRemux_)
: QDialog(parent),
queueModel(new RemuxQueueModel),
worker(new RemuxWorker()),
ui(new Ui::OBSRemux),
recPath(path),
autoRemux(autoRemux_)
{
setAcceptDrops(true);
setWindowFlags(windowFlags() & ~Qt::WindowContextHelpButtonHint);
ui->setupUi(this);
ui->progressBar->setVisible(false);
ui->buttonBox->button(QDialogButtonBox::Ok)->setEnabled(false);
ui->buttonBox->button(QDialogButtonBox::RestoreDefaults)->setEnabled(false);
if (autoRemux) {
resize(280, 40);
ui->tableView->hide();
ui->buttonBox->hide();
ui->label->hide();
}
ui->progressBar->setMinimum(0);
ui->progressBar->setMaximum(1000);
ui->progressBar->setValue(0);
ui->tableView->setModel(queueModel);
ui->tableView->setItemDelegateForColumn(RemuxEntryColumn::InputPath,
new RemuxEntryPathItemDelegate(false, recPath));
ui->tableView->setItemDelegateForColumn(RemuxEntryColumn::OutputPath,
new RemuxEntryPathItemDelegate(true, recPath));
ui->tableView->horizontalHeader()->setSectionResizeMode(QHeaderView::ResizeMode::Stretch);
ui->tableView->horizontalHeader()->setSectionResizeMode(RemuxEntryColumn::State,
QHeaderView::ResizeMode::Fixed);
ui->tableView->setEditTriggers(QAbstractItemView::EditTrigger::CurrentChanged);
ui->tableView->setTextElideMode(Qt::ElideMiddle);
ui->tableView->setWordWrap(false);
installEventFilter(CreateShortcutFilter());
ui->buttonBox->button(QDialogButtonBox::Ok)->setText(QTStr("Remux.Remux"));
ui->buttonBox->button(QDialogButtonBox::Reset)->setText(QTStr("Remux.ClearFinished"));
ui->buttonBox->button(QDialogButtonBox::RestoreDefaults)->setText(QTStr("Remux.ClearAll"));
ui->buttonBox->button(QDialogButtonBox::Reset)->setDisabled(true);
connect(ui->buttonBox->button(QDialogButtonBox::Ok), &QPushButton::clicked, this, &OBSRemux::beginRemux);
connect(ui->buttonBox->button(QDialogButtonBox::Reset), &QPushButton::clicked, this, &OBSRemux::clearFinished);
connect(ui->buttonBox->button(QDialogButtonBox::RestoreDefaults), &QPushButton::clicked, this,
&OBSRemux::clearAll);
connect(ui->buttonBox->button(QDialogButtonBox::Close), &QPushButton::clicked, this, &OBSRemux::close);
worker->moveToThread(&remuxer);
remuxer.start();
connect(worker.data(), &RemuxWorker::updateProgress, this, &OBSRemux::updateProgress);
connect(&remuxer, &QThread::finished, worker.data(), &QObject::deleteLater);
connect(worker.data(), &RemuxWorker::remuxFinished, this, &OBSRemux::remuxFinished);
connect(this, &OBSRemux::remux, worker.data(), &RemuxWorker::remux);
connect(queueModel.data(), &RemuxQueueModel::rowsInserted, this, &OBSRemux::rowCountChanged);
connect(queueModel.data(), &RemuxQueueModel::rowsRemoved, this, &OBSRemux::rowCountChanged);
QModelIndex index = queueModel->createIndex(0, 1);
QMetaObject::invokeMethod(ui->tableView, "setCurrentIndex", Qt::QueuedConnection,
Q_ARG(const QModelIndex &, index));
}
bool OBSRemux::stopRemux()
{
if (!worker->isWorking)
return true;
// By locking the worker thread's mutex, we ensure that its
// update poll will be blocked as long as we're in here with
// the popup open.
QMutexLocker lock(&worker->updateMutex);
bool exit = false;
if (QMessageBox::critical(nullptr, QTStr("Remux.ExitUnfinishedTitle"), QTStr("Remux.ExitUnfinished"),
QMessageBox::Yes | QMessageBox::No, QMessageBox::No) == QMessageBox::Yes) {
exit = true;
}
if (exit) {
// Inform the worker it should no longer be
// working. It will interrupt accordingly in
// its next update callback.
worker->isWorking = false;
}
return exit;
}
OBSRemux::~OBSRemux()
{
stopRemux();
remuxer.quit();
remuxer.wait();
}
void OBSRemux::rowCountChanged(const QModelIndex &, int, int)
{
// See if there are still any rows ready to remux. Change
// the state of the "go" button accordingly.
// There must be more than one row, since there will always be
// at least one row for the empty insertion point.
if (queueModel->rowCount() > 1) {
ui->buttonBox->button(QDialogButtonBox::Ok)->setEnabled(true);
ui->buttonBox->button(QDialogButtonBox::RestoreDefaults)->setEnabled(true);
ui->buttonBox->button(QDialogButtonBox::Reset)->setEnabled(queueModel->canClearFinished());
} else {
ui->buttonBox->button(QDialogButtonBox::Ok)->setEnabled(false);
ui->buttonBox->button(QDialogButtonBox::RestoreDefaults)->setEnabled(false);
ui->buttonBox->button(QDialogButtonBox::Reset)->setEnabled(false);
}
}
void OBSRemux::dropEvent(QDropEvent *ev)
{
QStringList urlList;
for (QUrl url : ev->mimeData()->urls()) {
QFileInfo fileInfo(url.toLocalFile());
if (fileInfo.isDir()) {
QStringList directoryFilter;
directoryFilter << "*.flv"
<< "*.mp4"
<< "*.mov"
<< "*.mkv"
<< "*.ts"
<< "*.m3u8";
QDirIterator dirIter(fileInfo.absoluteFilePath(), directoryFilter, QDir::Files,
QDirIterator::Subdirectories);
while (dirIter.hasNext()) {
urlList.append(dirIter.next());
}
} else {
urlList.append(fileInfo.canonicalFilePath());
}
}
if (urlList.empty()) {
QMessageBox::information(nullptr, QTStr("Remux.NoFilesAddedTitle"), QTStr("Remux.NoFilesAdded"),
QMessageBox::Ok);
} else if (!autoRemux) {
QModelIndex insertIndex = queueModel->index(queueModel->rowCount() - 1, RemuxEntryColumn::InputPath);
queueModel->setData(insertIndex, urlList, RemuxEntryRole::NewPathsToProcessRole);
}
}
void OBSRemux::dragEnterEvent(QDragEnterEvent *ev)
{
if (ev->mimeData()->hasUrls() && !worker->isWorking)
ev->accept();
}
void OBSRemux::beginRemux()
{
if (worker->isWorking) {
stopRemux();
return;
}
bool proceedWithRemux = true;
QFileInfoList overwriteFiles = queueModel->checkForOverwrites();
if (!overwriteFiles.empty()) {
QString message = QTStr("Remux.FileExists");
message += "\n\n";
for (QFileInfo fileInfo : overwriteFiles)
message += fileInfo.canonicalFilePath() + "\n";
if (OBSMessageBox::question(this, QTStr("Remux.FileExistsTitle"), message) != QMessageBox::Yes)
proceedWithRemux = false;
}
if (!proceedWithRemux)
return;
// Set all jobs to "pending" first.
queueModel->beginProcessing();
ui->progressBar->setVisible(true);
ui->buttonBox->button(QDialogButtonBox::Ok)->setText(QTStr("Remux.Stop"));
setAcceptDrops(false);
remuxNextEntry();
}
void OBSRemux::AutoRemux(QString inFile, QString outFile)
{
if (inFile != "" && outFile != "" && autoRemux) {
ui->progressBar->setVisible(true);
emit remux(inFile, outFile);
autoRemuxFile = outFile;
}
}
void OBSRemux::remuxNextEntry()
{
worker->lastProgress = 0.f;
QString inputPath, outputPath;
if (queueModel->beginNextEntry(inputPath, outputPath)) {
emit remux(inputPath, outputPath);
} else {
queueModel->autoRemux = autoRemux;
queueModel->endProcessing();
if (!autoRemux) {
OBSMessageBox::information(this, QTStr("Remux.FinishedTitle"),
queueModel->checkForErrors() ? QTStr("Remux.FinishedError")
: QTStr("Remux.Finished"));
}
ui->progressBar->setVisible(autoRemux);
ui->buttonBox->button(QDialogButtonBox::Ok)->setText(QTStr("Remux.Remux"));
ui->buttonBox->button(QDialogButtonBox::RestoreDefaults)->setEnabled(true);
ui->buttonBox->button(QDialogButtonBox::Reset)->setEnabled(queueModel->canClearFinished());
setAcceptDrops(true);
}
}
void OBSRemux::closeEvent(QCloseEvent *event)
{
if (!stopRemux())
event->ignore();
else
QDialog::closeEvent(event);
}
void OBSRemux::reject()
{
if (!stopRemux())
return;
QDialog::reject();
}
void OBSRemux::updateProgress(float percent)
{
ui->progressBar->setValue(percent * 10);
}
void OBSRemux::remuxFinished(bool success)
{
ui->buttonBox->button(QDialogButtonBox::Ok)->setEnabled(true);
queueModel->finishEntry(success);
if (autoRemux && autoRemuxFile != "") {
QTimer::singleShot(3000, this, &OBSRemux::close);
OBSBasic *main = OBSBasic::Get();
main->ShowStatusBarMessage(QTStr("Basic.StatusBar.AutoRemuxedTo").arg(autoRemuxFile));
}
remuxNextEntry();
}
void OBSRemux::clearFinished()
{
queueModel->clearFinished();
}
void OBSRemux::clearAll()
{
queueModel->clearAll();
}
/**********************************************************
Worker thread - Executes the libobs remux operation as a
background process.
**********************************************************/
void RemuxWorker::UpdateProgress(float percent)
{
if (abs(lastProgress - percent) < 0.1f)
return;
emit updateProgress(percent);
lastProgress = percent;
}
void RemuxWorker::remux(const QString &source, const QString &target)
{
isWorking = true;
auto callback = [](void *data, float percent) {
RemuxWorker *rw = static_cast<RemuxWorker *>(data);
QMutexLocker lock(&rw->updateMutex);
rw->UpdateProgress(percent);
return rw->isWorking;
};
bool stopped = false;
bool success = false;
media_remux_job_t mr_job = nullptr;
if (media_remux_job_create(&mr_job, QT_TO_UTF8(source), QT_TO_UTF8(target))) {
success = media_remux_job_process(mr_job, callback, this);
media_remux_job_destroy(mr_job);
stopped = !isWorking;
}
isWorking = false;
emit remuxFinished(!stopped && success);
}