diff --git a/ui/mirror/mirrormodel.cpp b/ui/mirror/mirrormodel.cpp index c3bf7b52..17316176 100644 --- a/ui/mirror/mirrormodel.cpp +++ b/ui/mirror/mirrormodel.cpp @@ -1,517 +1,518 @@ /*************************************************************************** * Copyright (C) 2009 Matthias Fuchs * * * * 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 2 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, write to the * * Free Software Foundation, Inc., * * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA . * ***************************************************************************/ #include "mirrormodel.h" #include #include #include #include #include +#include MirrorDelegate::MirrorDelegate(QObject *parent) : QStyledItemDelegate(parent), m_countrySort(nullptr) { } MirrorDelegate::MirrorDelegate(QSortFilterProxyModel *countrySort, QObject *parent) : QStyledItemDelegate(parent), m_countrySort(countrySort) { } QWidget *MirrorDelegate::createEditor(QWidget *parent, const QStyleOptionViewItem &option, const QModelIndex &index) const { Q_UNUSED(option) if (index.isValid()) { if (index.column() == MirrorItem::Url) { KLineEdit *line = new KLineEdit(parent); return line; } else if (index.column() == MirrorItem::Connections) { QSpinBox *numConnections = new QSpinBox(parent); numConnections->setRange(0, 20); return numConnections; } else if (index.column() == MirrorItem::Priority) { QSpinBox *priority = new QSpinBox(parent); priority->setRange(0, 999999); return priority; } else if (index.column() == MirrorItem::Country) { if (m_countrySort) { QComboBox *countrySort = new QComboBox(parent); countrySort->setModel(m_countrySort); return countrySort; } } } return nullptr; } void MirrorDelegate::setEditorData(QWidget *editor, const QModelIndex &index) const { if (index.isValid() && editor) { if (index.column() == MirrorItem::Url) { KLineEdit *line = static_cast(editor); const QUrl url = index.data(Qt::EditRole).toUrl(); line->setUrl(url); } else if (index.column() == MirrorItem::Connections) { QSpinBox *numConnections = static_cast(editor); const int num = index.data(Qt::EditRole).toInt(); numConnections->setValue(num); } else if (index.column() == MirrorItem::Priority) { QSpinBox *priority = static_cast(editor); const int num = index.data(Qt::EditRole).toInt(); priority->setValue(num); } else if (index.column() == MirrorItem::Country) { QComboBox *countrySort = static_cast(editor); const QString countryCode = index.data(Qt::EditRole).toString(); const int indexCountrySort = countrySort->findData(countryCode); countrySort->setCurrentIndex(indexCountrySort); } } } void MirrorDelegate::setModelData(QWidget *editor, QAbstractItemModel *model, const QModelIndex &index) const { if (index.isValid() && editor && model) { if (index.column() == MirrorItem::Url) { KLineEdit *line = static_cast(editor); if (!line->text().isEmpty()) { model->setData(index, line->text()); } } else if (index.column() == MirrorItem::Connections) { QSpinBox *numConnections = static_cast(editor); model->setData(index, numConnections->value()); } else if (index.column() == MirrorItem::Priority) { QSpinBox *priority = static_cast(editor); model->setData(index, priority->value()); } else if (index.column() == MirrorItem::Country) { QComboBox *countrySort = static_cast(editor); const QString countryCode = countrySort->itemData(countrySort->currentIndex()).toString(); model->setData(index, countryCode); } } } void MirrorDelegate::updateEditorGeometry(QWidget *editor, const QStyleOptionViewItem &option, const QModelIndex &index) const { Q_UNUSED(index) editor->setGeometry(option.rect); } QSize MirrorDelegate::sizeHint(const QStyleOptionViewItem &option, const QModelIndex &index) const { //make the sizeHint a little bit nicer to have more beautiful editors QSize hint; hint.setWidth(QStyledItemDelegate::sizeHint(option, index).width()); hint.setHeight(option.fontMetrics.height() + 7); return hint; } MirrorProxyModel::MirrorProxyModel(QObject *parent) : QSortFilterProxyModel(parent) { } bool MirrorProxyModel::lessThan(const QModelIndex &left, const QModelIndex &right) const { if (left.column() == MirrorItem::Used) { const int leftData = sourceModel()->data(left, Qt::CheckStateRole).toInt(); const int rightData = sourceModel()->data(right, Qt::CheckStateRole).toInt(); return leftData < rightData; } else if (left.column() == MirrorItem::Priority) { const int leftData = sourceModel()->data(left, Qt::UserRole).toInt(); const int rightData = sourceModel()->data(right, Qt::UserRole).toInt(); return (!leftData ? true : (leftData > rightData) && rightData);//0 is always smallest, otherwise larger is smaller } return QSortFilterProxyModel::lessThan(left, right); } MirrorItem::MirrorItem() : m_checked(Qt::Unchecked), m_numConnections(0), m_priority(0) { } QVariant MirrorItem::data(int column, int role) const { if (column == MirrorItem::Used) { if (role == Qt::CheckStateRole) { return m_checked; } } else if (column == MirrorItem::Url) { if (role == Qt::DisplayRole) { return m_url.toString(); } else if ((role == Qt::UserRole) || (role == Qt::EditRole)) { return QVariant(m_url); } } else if (column == MirrorItem::Connections) { if (role == Qt::DisplayRole) { if (m_numConnections) { return m_numConnections; } else { return i18n("not specified"); } } else if ((role == Qt::EditRole) || (role == Qt::UserRole)) { return m_numConnections; } } else if (column == MirrorItem::Priority) { if (role == Qt::DisplayRole) { if (m_priority) { return m_priority; } else { return i18n("not specified"); } } else if ((role == Qt::EditRole) || (role == Qt::UserRole)) { return m_priority; } } else if (column == MirrorItem::Country) { if (role == Qt::DisplayRole) { return m_countryName; } else if (role == Qt::DecorationRole) { return m_countryFlag; } else if ((role == Qt::UserRole) || (role == Qt::EditRole)) { return m_countryCode; } } return QVariant(); } Qt::ItemFlags MirrorItem::flags(int column) const { Qt::ItemFlags flags = Qt::ItemIsEnabled | Qt::ItemIsSelectable; if (column == MirrorItem::Used) { flags |= Qt::ItemIsUserCheckable; } else if (column == MirrorItem::Url) { flags |= Qt::ItemIsEditable; } else if (column == MirrorItem::Connections) { flags |= Qt::ItemIsEditable; } else if (column == MirrorItem::Priority) { flags |= Qt::ItemIsEditable; } else if (column == MirrorItem::Country) { flags |= Qt::ItemIsEditable; } return flags; } bool MirrorItem::setData(int column, const QVariant &value, int role) { if ((column == MirrorItem::Used) && (role == Qt::CheckStateRole)) { m_checked = static_cast(value.toInt()); return true; } else if ((column == MirrorItem::Url) && (role == Qt::EditRole)) { QUrl url; if (value.type() == QVariant::Url) { url = QUrl(value.toUrl()); } else if (value.type() == QVariant::String) { url = QUrl(value.toString()); } if (!url.isEmpty()) { m_url = url; return true; } } else if ((column == MirrorItem::Connections) && (role == Qt::EditRole)) { m_numConnections = value.toInt(); return true; } else if ((column == MirrorItem::Priority) && (role == Qt::EditRole)) { m_priority = value.toInt(); return true; } else if ((column == MirrorItem::Country) && (role == Qt::EditRole)) { m_countryCode = value.toString(); m_countryName = KLocale::global()->countryCodeToName(m_countryCode); if (!m_countryName.isEmpty()) { QString path = QStandardPaths::locate(QStandardPaths::GenericDataLocation, QLatin1String("locale/") + QString::fromLatin1("l10n/%1/flag.png").arg(m_countryCode)); if (path.isEmpty()) { m_countryFlag = QIcon(); } else { m_countryFlag = QIcon::fromTheme(path); } } else { m_countryFlag = QIcon(); } return true; } return false; } MirrorModel::MirrorModel(QObject *parent) : QAbstractTableModel(parent) { } MirrorModel::~MirrorModel() { qDeleteAll(m_data); } int MirrorModel::rowCount(const QModelIndex &index) const { if (!index.isValid()) { return m_data.count(); } else { return 0; } } int MirrorModel::columnCount(const QModelIndex &index) const { if (index.isValid()) { return 0; } return 5; } QVariant MirrorModel::headerData(int section, Qt::Orientation orientation, int role) const { if (orientation == Qt::Vertical) { return QVariant(); } if ((section == MirrorItem::Url) && (role == Qt::DisplayRole)) { return i18nc("Mirror as in server, in url", "Mirror"); } else if (section == MirrorItem::Priority) { if (role == Qt::DisplayRole) { return i18nc("The priority of the mirror", "Priority"); } else if (role == Qt::DecorationRole) { return QIcon::fromTheme("games-highscores"); } } else if ((section == MirrorItem::Connections) && (role == Qt::DisplayRole)) { return i18nc("Number of paralell connections to the mirror", "Connections"); } else if ((section == MirrorItem::Country) && (role == Qt::DisplayRole)) { return i18nc("Location = country", "Location"); } return QVariant(); } QVariant MirrorModel::data(const QModelIndex &index, int role) const { if (!index.isValid()) { return QVariant(); } return m_data.at(index.row())->data(index.column(), role); } Qt::ItemFlags MirrorModel::flags(const QModelIndex &index) const { if (!index.isValid()) { return 0; } return m_data.at(index.row())->flags(index.column()); } bool MirrorModel::setData(const QModelIndex &index, const QVariant &value, int role) { if (!index.isValid()) { return false; } const bool changed = m_data.at(index.row())->setData(index.column(), value, role); if (changed) { emit dataChanged(index, index); } return changed; } bool MirrorModel::removeRows(int row, int count, const QModelIndex &parent) { if (parent.isValid() || (row < 0) || (count < 1) || (row + count > rowCount())) { return false; } beginRemoveRows(parent, row, row + count - 1); while (count) { MirrorItem *item = m_data[row]; m_data.removeAt(row); delete item; --count; } endRemoveRows(); return true; } void MirrorModel::addMirror(const QUrl &url, int numConnections, int priority, const QString &countryCode) { if (!url.isValid()) { return; } for (int i = 0; i < rowCount(); ++i) { //exists already, so remove the row if (QUrl(m_data.at(i)->data(MirrorItem::Url).toString()) == url) { removeRow(i); break; } } int index = rowCount(); emit beginInsertRows(QModelIndex(), index, index); MirrorItem *item = new MirrorItem; m_data.append(item); item->setData(MirrorItem::Used, Qt::Checked, Qt::CheckStateRole);//every newly added mirror is set to checked automatically item->setData(MirrorItem::Url, QVariant(url)); item->setData(MirrorItem::Connections, numConnections); item->setData(MirrorItem::Priority, priority); item->setData(MirrorItem::Country, countryCode); emit endInsertRows(); } void MirrorModel::setMirrors(const QHash > &mirrors) { removeRows(0, rowCount()); QHash >::const_iterator it; QHash >::const_iterator itEnd = mirrors.constEnd(); for (it = mirrors.constBegin(); it != itEnd; ++it) { MirrorItem *item = new MirrorItem; item->setData(MirrorItem::Url, QVariant(it.key())); Qt::CheckState state = (*it).first ? Qt::Checked : Qt::Unchecked; item->setData(MirrorItem::Used, state, Qt::CheckStateRole); item->setData(MirrorItem::Connections, (*it).second); m_data.append(item); } emit reset(); } QHash > MirrorModel::availableMirrors() const { QHash > mirrors; foreach (MirrorItem *item, m_data) { bool used = (item->data(MirrorItem::Used, Qt::CheckStateRole).toInt() == Qt::Checked) ? true : false; const QUrl url = QUrl(item->data(MirrorItem::Url).toString()); mirrors[url] = QPair(used, item->data(MirrorItem::Connections, Qt::UserRole).toInt()); } return mirrors; }