Rewrite optional mod dialog

Signed-off-by: TheKodeToad <TheKodeToad@proton.me>
This commit is contained in:
TheKodeToad
2023-08-27 00:49:21 +01:00
parent f897b14e3e
commit 479335dfe0
3 changed files with 113 additions and 222 deletions

View File

@ -19,125 +19,45 @@
#include "OptionalModDialog.h"
#include "ui_OptionalModDialog.h"
OptionalModListModel::OptionalModListModel(QWidget* parent, QStringList mods) : QAbstractListModel(parent), m_mods(mods) {}
QStringList OptionalModListModel::getResult()
{
QStringList result;
for (const auto& mod : m_mods) {
if (m_selected.value(mod, false)) {
result << mod;
}
}
return result;
}
int OptionalModListModel::rowCount(const QModelIndex& parent) const
{
return parent.isValid() ? 0 : m_mods.size();
}
int OptionalModListModel::columnCount(const QModelIndex& parent) const
{
// Enabled, Name
return parent.isValid() ? 0 : 2;
}
QVariant OptionalModListModel::data(const QModelIndex& index, int role) const
{
auto row = index.row();
auto mod = m_mods.at(row);
if (role == Qt::DisplayRole && index.column() == NameColumn) {
return mod;
} else if (role == Qt::CheckStateRole && index.column() == EnabledColumn) {
return m_selected.value(mod, false) ? Qt::Checked : Qt::Unchecked;
}
return {};
}
bool OptionalModListModel::setData(const QModelIndex& index, [[maybe_unused]] const QVariant& value, int role)
{
if (role == Qt::CheckStateRole) {
auto row = index.row();
auto mod = m_mods.at(row);
toggleMod(mod, row);
return true;
}
return false;
}
QVariant OptionalModListModel::headerData(int section, Qt::Orientation orientation, int role) const
{
if (role == Qt::DisplayRole && orientation == Qt::Horizontal) {
switch (section) {
case EnabledColumn:
return QString();
case NameColumn:
return QString("Name");
}
}
return {};
}
Qt::ItemFlags OptionalModListModel::flags(const QModelIndex& index) const
{
auto flags = QAbstractListModel::flags(index);
if (index.isValid() && index.column() == EnabledColumn) {
flags |= Qt::ItemIsUserCheckable;
}
return flags;
}
void OptionalModListModel::toggleAll(bool enabled)
{
for (const auto& mod : m_mods) {
m_selected[mod] = enabled;
}
emit dataChanged(OptionalModListModel::index(0, EnabledColumn), OptionalModListModel::index(m_mods.size() - 1, EnabledColumn));
}
void OptionalModListModel::toggleMod(QString mod, int index)
{
auto enable = !m_selected.value(mod, false);
setMod(mod, index, enable);
}
void OptionalModListModel::setMod(QString mod, int index, bool enable, bool shouldEmit)
{
if (m_selected.value(mod, false) == enable)
return;
m_selected[mod] = enable;
if (shouldEmit) {
emit dataChanged(OptionalModListModel::index(index, EnabledColumn), OptionalModListModel::index(index, EnabledColumn));
}
}
OptionalModDialog::OptionalModDialog(QWidget* parent, QStringList mods) : QDialog(parent), ui(new Ui::OptionalModDialog)
OptionalModDialog::OptionalModDialog(QWidget* parent, const QStringList& mods) : QDialog(parent), ui(new Ui::OptionalModDialog)
{
ui->setupUi(this);
for (const QString& mod : mods) {
auto item = new QListWidgetItem(mod, ui->list);
item->setFlags(item->flags() | Qt::ItemIsUserCheckable);
item->setCheckState(Qt::Unchecked);
item->setData(Qt::UserRole, mod);
}
listModel = new OptionalModListModel(this, mods);
ui->treeView->setModel(listModel);
ui->treeView->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
ui->treeView->header()->setSectionResizeMode(OptionalModListModel::NameColumn, QHeaderView::Stretch);
connect(ui->selectAllButton, &QPushButton::clicked, listModel, &OptionalModListModel::selectAll);
connect(ui->clearAllButton, &QPushButton::clicked, listModel, &OptionalModListModel::clearAll);
connect(ui->installButton, &QPushButton::clicked, this, &QDialog::accept);
connect(ui->cancelButton, &QPushButton::clicked, this, &QDialog::reject);
connect(ui->selectAllButton, &QPushButton::clicked, ui->list, [this] {
for (int i = 0; i < ui->list->count(); i++)
ui->list->item(i)->setCheckState(Qt::Checked);
});
connect(ui->clearAllButton, &QPushButton::clicked, ui->list, [this] {
for (int i = 0; i < ui->list->count(); i++)
ui->list->item(i)->setCheckState(Qt::Unchecked);
});
connect(ui->list, &QListWidget::itemActivated, [](QListWidgetItem* item) {
if (item->checkState() == Qt::Checked)
item->setCheckState(Qt::Unchecked);
else
item->setCheckState(Qt::Checked);
});
}
OptionalModDialog::~OptionalModDialog()
{
delete ui;
}
QStringList OptionalModDialog::getResult()
{
QStringList result;
result.reserve(ui->list->count());
for (int i = 0; i < ui->list->count(); i++) {
auto item = ui->list->item(i);
if (item->checkState() == Qt::Checked)
result.append(item->data(Qt::UserRole).toString());
}
return result;
}