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
+5 -549
View File
@@ -15,230 +15,14 @@
along with this program. If not, see <http://www.gnu.org/licenses/>.
******************************************************************************/
#include "moc_window-remux.cpp"
#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 <QDir>
#include <QStyle>
#include "window-basic-main.hpp"
#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.
**********************************************************/
RemuxEntryPathItemDelegate::RemuxEntryPathItemDelegate(bool isOutput, const QString &defaultPath)
: QStyledItemDelegate(),
isOutput(isOutput),
defaultPath(defaultPath)
{
}
QWidget *RemuxEntryPathItemDelegate::createEditor(QWidget *parent, const QStyleOptionViewItem & /* option */,
const QModelIndex &index) const
{
RemuxEntryState state = index.model()
->index(index.row(), RemuxEntryColumn::State)
.data(RemuxEntryRole::EntryStateRole)
.value<RemuxEntryState>();
if (state == RemuxEntryState::Pending || state == RemuxEntryState::InProgress) {
// Never allow modification of rows that are
// in progress.
return Q_NULLPTR;
} else if (isOutput && state != RemuxEntryState::Ready) {
// Do not allow modification of output rows
// that aren't associated with a valid input.
return Q_NULLPTR;
} else if (!isOutput && state == RemuxEntryState::Complete) {
// Don't allow modification of rows that are
// already complete.
return Q_NULLPTR;
} else {
QSizePolicy buttonSizePolicy(QSizePolicy::Policy::Minimum, QSizePolicy::Policy::Expanding,
QSizePolicy::ControlType::PushButton);
QWidget *container = new QWidget(parent);
auto browseCallback = [this, container]() {
const_cast<RemuxEntryPathItemDelegate *>(this)->handleBrowse(container);
};
auto clearCallback = [this, container]() {
const_cast<RemuxEntryPathItemDelegate *>(this)->handleClear(container);
};
QHBoxLayout *layout = new QHBoxLayout();
layout->setContentsMargins(0, 0, 0, 0);
layout->setSpacing(0);
QLineEdit *text = new QLineEdit();
text->setObjectName(QStringLiteral("text"));
text->setSizePolicy(QSizePolicy(QSizePolicy::Policy::Expanding, QSizePolicy::Policy::Expanding,
QSizePolicy::ControlType::LineEdit));
layout->addWidget(text);
QObject::connect(text, &QLineEdit::editingFinished, this, &RemuxEntryPathItemDelegate::updateText);
QToolButton *browseButton = new QToolButton();
browseButton->setText("...");
browseButton->setSizePolicy(buttonSizePolicy);
layout->addWidget(browseButton);
container->connect(browseButton, &QToolButton::clicked, browseCallback);
// The "clear" button is not shown in output cells
// or the insertion point's input cell.
if (!isOutput && state != RemuxEntryState::Empty) {
QToolButton *clearButton = new QToolButton();
clearButton->setText("X");
clearButton->setSizePolicy(buttonSizePolicy);
layout->addWidget(clearButton);
container->connect(clearButton, &QToolButton::clicked, clearCallback);
}
container->setLayout(layout);
container->setFocusProxy(text);
return container;
}
}
void RemuxEntryPathItemDelegate::setEditorData(QWidget *editor, const QModelIndex &index) const
{
QLineEdit *text = editor->findChild<QLineEdit *>();
text->setText(index.data().toString());
editor->setProperty(PATH_LIST_PROP, QVariant());
}
void RemuxEntryPathItemDelegate::setModelData(QWidget *editor, QAbstractItemModel *model,
const QModelIndex &index) const
{
// We use the PATH_LIST_PROP property to pass a list of
// path strings from the editor widget into the model's
// NewPathsToProcessRole. This is only used when paths
// are selected through the "browse" or "delete" buttons
// in the editor. If the user enters new text in the
// text box, we simply pass that text on to the model
// as normal text data in the default role.
QVariant pathListProp = editor->property(PATH_LIST_PROP);
if (pathListProp.isValid()) {
QStringList list = editor->property(PATH_LIST_PROP).toStringList();
if (isOutput) {
if (list.size() > 0)
model->setData(index, list);
} else
model->setData(index, list, RemuxEntryRole::NewPathsToProcessRole);
} else {
QLineEdit *lineEdit = editor->findChild<QLineEdit *>();
model->setData(index, lineEdit->text());
}
}
void RemuxEntryPathItemDelegate::paint(QPainter *painter, const QStyleOptionViewItem &option,
const QModelIndex &index) const
{
RemuxEntryState state = index.model()
->index(index.row(), RemuxEntryColumn::State)
.data(RemuxEntryRole::EntryStateRole)
.value<RemuxEntryState>();
QStyleOptionViewItem localOption = option;
initStyleOption(&localOption, index);
if (isOutput) {
if (state != Ready) {
QColor background =
localOption.palette.color(QPalette::ColorGroup::Disabled, QPalette::ColorRole::Window);
localOption.backgroundBrush = QBrush(background);
}
}
QApplication::style()->drawControl(QStyle::CE_ItemViewItem, &localOption, painter);
}
void RemuxEntryPathItemDelegate::handleBrowse(QWidget *container)
{
QString ExtensionPattern = "(*.mp4 *.flv *.mov *.mkv *.ts *.m3u8)";
QLineEdit *text = container->findChild<QLineEdit *>();
QString currentPath = text->text();
if (currentPath.isEmpty())
currentPath = defaultPath;
bool isSet = false;
if (isOutput) {
QString newPath = SaveFile(container, QTStr("Remux.SelectTarget"), currentPath, ExtensionPattern);
if (!newPath.isEmpty()) {
container->setProperty(PATH_LIST_PROP, QStringList() << newPath);
isSet = true;
}
} else {
QStringList paths = OpenFiles(container, QTStr("Remux.SelectRecording"), currentPath,
QTStr("Remux.OBSRecording") + QString(" ") + ExtensionPattern);
if (!paths.empty()) {
container->setProperty(PATH_LIST_PROP, paths);
isSet = true;
}
#ifdef __APPLE__
// TODO: Revisit when QTBUG-42661 is fixed
container->window()->raise();
#endif
}
if (isSet)
emit commitData(container);
}
void RemuxEntryPathItemDelegate::handleClear(QWidget *container)
{
// An empty string list will indicate that the entry is being
// blanked and should be deleted.
container->setProperty(PATH_LIST_PROP, QStringList());
emit commitData(container);
}
void RemuxEntryPathItemDelegate::updateText()
{
QLineEdit *lineEdit = dynamic_cast<QLineEdit *>(sender());
QWidget *editor = lineEdit->parentWidget();
emit commitData(editor);
}
/**********************************************************
Model - Manages the queue's data
**********************************************************/
#include "moc_RemuxQueueModel.cpp"
int RemuxQueueModel::rowCount(const QModelIndex &) const
{
@@ -594,331 +378,3 @@ void RemuxQueueModel::finishEntry(bool success)
}
}
}
/**********************************************************
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);
}