diff --git a/libtaskmanager/abstracttasksmodel.h b/libtaskmanager/abstracttasksmodel.h index b0b10dbd..82d140d5 100644 --- a/libtaskmanager/abstracttasksmodel.h +++ b/libtaskmanager/abstracttasksmodel.h @@ -1,279 +1,280 @@ /******************************************************************** Copyright 2016 Eike Hein This library is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 2.1 of the License, or (at your option) version 3, or any later version accepted by the membership of KDE e.V. (or its successor approved by the membership of KDE e.V.), which shall act as a proxy defined in Section 6 of version 3 of the license. This library 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 Lesser General Public License for more details. You should have received a copy of the GNU Lesser General Public License along with this library. If not, see . *********************************************************************/ #ifndef ABSTRACTTASKSMODEL_H #define ABSTRACTTASKSMODEL_H #include "abstracttasksmodeliface.h" #include #include "taskmanager_export.h" namespace TaskManager { /** * @short An abstract base class for (flat) tasks models. * * This class serves as abstract base class for flat tasks model implementations. * It provides data roles and no-op default implementations of methods in the * AbstractTasksModelIface interface. * * @author Eike Hein **/ class TASKMANAGER_EXPORT AbstractTasksModel : public QAbstractListModel, public AbstractTasksModelIface { Q_OBJECT public: enum AdditionalRoles { AppId = Qt::UserRole + 1, /**< KService storage id (.desktop name sans extension). */ AppName, /**< Application name. */ GenericName, /**< Generic application name. */ LauncherUrl, /**< URL that can be used to launch this application (.desktop or executable). */ LauncherUrlWithoutIcon, /**< Special path to get a launcher URL while skipping fallback icon encoding. Used as speed optimization. */ LegacyWinIdList, /**< X11 window ids. Stopgap until we have something better. */ MimeType, /**< MIME type for this task (window, window group), needed for DND. */ MimeData, /**< Data for MimeType. */ IsWindow, /**< This is a window task. */ IsStartup, /**< This is a startup task. */ IsLauncher, /**< This is a launcher task. */ IsGroupParent, /**< This is a parent item for a group of child tasks. */ + ChildCount, /**< The number of tasks in this group. */ IsGroupable, /**< Whether this task is being ignored by grouping or not. */ IsActive, /**< This is the currently active task. */ IsClosable, /**< requestClose (see below) available. */ IsMovable, /**< requestMove (see below) available. */ IsResizable, /**< requestResize (see below) available. */ IsMaximizable, /**< requestToggleMaximize (see below) available. */ IsMaximized, /**< Task (i.e. window) is maximized. */ IsMinimizable, /**< requestToggleMinimize (see below) available. */ IsMinimized, /**< Task (i.e. window) is minimized. */ IsKeepAbove, /**< Task (i.e. window) is keep-above. */ IsKeepBelow, /**< Task (i.e. window) is keep-below. */ IsFullScreenable, /**< requestToggleFullScreen (see below) available. */ IsFullScreen, /**< Task (i.e. window) is fullscreen. */ IsShadeable, /**< requestToggleShade (see below) available. */ IsShaded, /**< Task (i.e. window) is shaded. */ IsVirtualDesktopChangeable, /**< requestVirtualDesktop (see below) available. */ VirtualDesktop, /**< Virtual desktop for the task (i.e. window). */ IsOnAllVirtualDesktops, /**< Task is on all virtual desktops. */ Geometry, /**< The task's geometry (i.e. the window's). */ ScreenGeometry, /**< Screen geometry for the task (i.e. the window's screen). */ Activities, /**< Activities for the task (i.e. window). */ IsDemandingAttention, /**< Task is demanding attention. */ SkipTaskbar, /**< Task should not be shown in a 'task bar' user interface. */ SkipPager, /**< Task should not to be shown in a 'pager' user interface. */ }; Q_ENUM(AdditionalRoles) explicit AbstractTasksModel(QObject *parent = nullptr); virtual ~AbstractTasksModel(); QHash roleNames() const override; virtual QModelIndex index(int row, int column = 0, const QModelIndex &parent = QModelIndex()) const override; /** * Request activation of the task at the given index. Derived classes are * free to interpret the meaning of "activate" themselves depending on * the nature and state of the task, e.g. launch or raise a window task. * * This base implementation does nothing. * * @param index An index in this tasks model. **/ virtual void requestActivate(const QModelIndex &index) override; /** * Request an additional instance of the application backing the task * at the given index. * * This base implementation does nothing. * * @param index An index in this tasks model. **/ virtual void requestNewInstance(const QModelIndex &index) override; /** * Requests to open the given URLs with the application backing the task * at the given index. * * This base implementation does nothing. * * @param index An index in this tasks model. * @param urls The URLs to be passed to the application. **/ virtual void requestOpenUrls(const QModelIndex &index, const QList &urls) override; /** * Request the task at the given index be closed. * * This base implementation does nothing. * * @param index An index in this tasks model. **/ virtual void requestClose(const QModelIndex &index) override; /** * Request starting an interactive move for the task at the given index. * * This is meant for tasks that have an associated window, and may be * a no-op when there is no window. * * This base implementation does nothing. * * @param index An index in this tasks model. **/ virtual void requestMove(const QModelIndex &index) override; /** * Request starting an interactive resize for the task at the given index. * * This is meant for tasks that have an associated window, and may be a * no-op when there is no window. * * This base implementation does nothing. * * @param index An index in this tasks model. **/ virtual void requestResize(const QModelIndex &index) override; /** * Request toggling the minimized state of the task at the given index. * * This is meant for tasks that have an associated window, and may be * a no-op when there is no window. * * This base implementation does nothing. * * @param index An index in this tasks model. **/ virtual void requestToggleMinimized(const QModelIndex &index) override; /** * Request toggling the maximized state of the task at the given index. * * This is meant for tasks that have an associated window, and may be * a no-op when there is no window. * * This base implementation does nothing. * * @param index An index in this tasks model. **/ virtual void requestToggleMaximized(const QModelIndex &index) override; /** * Request toggling the keep-above state of the task at the given index. * * This is meant for tasks that have an associated window, and may be * a no-op when there is no window. * * This base implementation does nothing. * * @param index An index in this tasks model. **/ virtual void requestToggleKeepAbove(const QModelIndex &index) override; /** * Request toggling the keep-below state of the task at the given index. * * This is meant for tasks that have an associated window, and may be * a no-op when there is no window. * * This base implementation does nothing. * * @param index An index in this tasks model. **/ virtual void requestToggleKeepBelow(const QModelIndex &index) override; /** * Request toggling the fullscreen state of the task at the given index. * * This is meant for tasks that have an associated window, and may be * a no-op when there is no window. * * This base implementation does nothing. * * @param index An index in this tasks model. **/ virtual void requestToggleFullScreen(const QModelIndex &index) override; /** * Request toggling the shaded state of the task at the given index. * * This is meant for tasks that have an associated window, and may be * a no-op when there is no window. * * This base implementation does nothing. * * @param index An index in this tasks model. **/ virtual void requestToggleShaded(const QModelIndex &index) override; /** * Request moving the task at the given index to the specified virtual * desktop. * * This is meant for tasks that have an associated window, and may be * a no-op when there is no window. * * This base implementation does nothing. * * @param index An index in this tasks model. * @param desktop A virtual desktop number. **/ virtual void requestVirtualDesktop(const QModelIndex &index, qint32 desktop) override; /** * Request moving the task at the given index to the specified activities. * * This is meant for tasks that have an associated window, and may be * a no-op when there is no window. * * This base implementation does nothing. * * @param index An index in this tasks model. * @param activities The new list of activities. **/ virtual void requestActivities(const QModelIndex &index, const QStringList &activities) override; /** * Request informing the window manager of new geometry for a visual * delegate for the task at the given index. The geometry should be in * screen coordinates. * * This base implementation does nothing. * * @param index An index in this tasks model. * @param geometry Visual delegate geometry in screen coordinates. * @param delegate The delegate. Implementations are on their own with * regard to extracting information from this, and should take care to * reject invalid objects. **/ virtual void requestPublishDelegateGeometry(const QModelIndex &index, const QRect &geometry, QObject *delegate = nullptr) override; }; } #endif diff --git a/libtaskmanager/taskgroupingproxymodel.cpp b/libtaskmanager/taskgroupingproxymodel.cpp index 85e94e4a..ae1d6cfb 100644 --- a/libtaskmanager/taskgroupingproxymodel.cpp +++ b/libtaskmanager/taskgroupingproxymodel.cpp @@ -1,1229 +1,1222 @@ /******************************************************************** Copyright 2016 Eike Hein This library is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 2.1 of the License, or (at your option) version 3, or any later version accepted by the membership of KDE e.V. (or its successor approved by the membership of KDE e.V.), which shall act as a proxy defined in Section 6 of version 3 of the license. This library 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 Lesser General Public License for more details. You should have received a copy of the GNU Lesser General Public License along with this library. If not, see . *********************************************************************/ #include "taskgroupingproxymodel.h" #include "abstracttasksmodel.h" #include "tasktools.h" #include namespace TaskManager { class TaskGroupingProxyModel::Private { public: Private(TaskGroupingProxyModel *q); AbstractTasksModelIface *abstractTasksSourceModel = nullptr; TasksModel::GroupMode groupMode = TasksModel::GroupApplications; bool groupDemandingAttention = false; int windowTasksThreshold = -1; QVector> rowMap; QSet blacklistedAppIds; QSet blacklistedLauncherUrls; bool isGroup(int row); bool any(const QModelIndex &parent, int role); bool all(const QModelIndex &parent, int role); void sourceRowsAboutToBeInserted(const QModelIndex &parent, int first, int last); void sourceRowsInserted(const QModelIndex &parent, int start, int end); void sourceRowsAboutToBeRemoved(const QModelIndex &parent, int first, int last); void sourceRowsRemoved(const QModelIndex &parent, int start, int end); void sourceModelAboutToBeReset(); void sourceModelReset(); void sourceDataChanged(QModelIndex topLeft, QModelIndex bottomRight, const QVector &roles = QVector()); void adjustMap(int anchor, int delta); void rebuildMap(); bool shouldGroupTasks(); void checkGrouping(bool silent = false); bool isBlacklisted(const QModelIndex &sourceIndex); bool tryToGroup(const QModelIndex &sourceIndex, bool silent = false); void formGroupFor(const QModelIndex &index); void breakGroupFor(const QModelIndex &index, bool silent = false); private: TaskGroupingProxyModel *q; }; TaskGroupingProxyModel::Private::Private(TaskGroupingProxyModel *q) : q(q) { } bool TaskGroupingProxyModel::Private::isGroup(int row) { if (row < 0 || row >= rowMap.count()) { return false; } return (rowMap.at(row).count() > 1); } bool TaskGroupingProxyModel::Private::any(const QModelIndex &parent, int role) { bool is = false; for (int i = 0; i < q->rowCount(parent); ++i) { if (parent.child(i, 0).data(role).toBool()) { return true; } } return is; } bool TaskGroupingProxyModel::Private::all(const QModelIndex &parent, int role) { bool is = true; for (int i = 0; i < q->rowCount(parent); ++i) { if (!parent.child(i, 0).data(role).toBool()) { return false; } } return is; } void TaskGroupingProxyModel::Private::sourceRowsAboutToBeInserted(const QModelIndex &parent, int first, int last) { Q_UNUSED(parent) Q_UNUSED(first) Q_UNUSED(last) } void TaskGroupingProxyModel::Private::sourceRowsInserted(const QModelIndex &parent, int start, int end) { // We only support flat source models. if (parent.isValid()) { return; } adjustMap(start, (end - start) + 1); bool shouldGroup = shouldGroupTasks(); // Can be slightly expensive; cache return value. for (int i = start; i <= end; ++i) { if (!shouldGroup || !tryToGroup(q->sourceModel()->index(i, 0))) { q->beginInsertRows(QModelIndex(), rowMap.count(), rowMap.count()); rowMap.append(QVector{i}); q->endInsertRows(); } } checkGrouping(); } void TaskGroupingProxyModel::Private::sourceRowsAboutToBeRemoved(const QModelIndex &parent, int first, int last) { // We only support flat source models. if (parent.isValid()) { return; } for (int i = first; i <= last; ++i) { for (int j = 0; j < rowMap.count(); ++j) { const QVector &sourceRows = rowMap.at(j); const int mapIndex = sourceRows.indexOf(i); if (mapIndex != -1) { // Remove top-level item. if (sourceRows.count() == 1) { q->beginRemoveRows(QModelIndex(), j, j); rowMap.remove(j); // index() encodes parent row numbers in indices returned for group // members. endRemoveRows() is not aware of this scheme, and so won't // update any persistent indices with new row values. We're now going // to do it ourselves. foreach (const QModelIndex &idx, q->persistentIndexList()) { // internalId() for group members is parent row + 1, so the first // id after j is (j + 2). if (idx.internalId() > ((uint)j + 1)) { q->changePersistentIndex(idx, q->createIndex(idx.row(), 0, idx.internalId() - 1)); } } q->endRemoveRows(); // Dissolve group. } else if (sourceRows.count() == 2) { const QModelIndex parent = q->index(j, 0); q->beginRemoveRows(parent, 0, 1); rowMap[j].remove(mapIndex); // index() encodes parent row numbers in indices returned for group // members. endRemoveRows() is not aware of this scheme, and so won't // invalidate persistent indices for the members of the group we're // dissolving. We're now going to do it ourselves. foreach (const QModelIndex &idx, q->persistentIndexList()) { if (idx.internalId() == ((uint)j + 1)) { q->changePersistentIndex(idx, QModelIndex()); } } q->endRemoveRows(); // We're no longer a group parent. q->dataChanged(parent, parent); // Remove group member. } else { const QModelIndex parent = q->index(j, 0); q->beginRemoveRows(parent, mapIndex, mapIndex); rowMap[j].remove(mapIndex); q->endRemoveRows(); // Various roles of the parent evaluate child data, and the // child list has changed. q->dataChanged(parent, parent); } break; } } } } void TaskGroupingProxyModel::Private::sourceRowsRemoved(const QModelIndex &parent, int start, int end) { // We only support flat source models. if (parent.isValid()) { return; } adjustMap(start + 1, -((end - start) + 1)); checkGrouping(); } void TaskGroupingProxyModel::Private::sourceModelAboutToBeReset() { q->beginResetModel(); } void TaskGroupingProxyModel::Private::sourceModelReset() { rebuildMap(); q->endResetModel(); } void TaskGroupingProxyModel::Private::sourceDataChanged(QModelIndex topLeft, QModelIndex bottomRight, const QVector &roles) { for (int i = topLeft.row(); i <= bottomRight.row(); ++i) { const QModelIndex &sourceIndex = q->sourceModel()->index(i, 0); QModelIndex proxyIndex = q->mapFromSource(sourceIndex); if (!proxyIndex.isValid()) { return; } const QModelIndex parent = proxyIndex.parent(); // If a child item changes, its parent may need an update as well as many of // the data roles evaluate child data. See data(). // TODO: Some roles do not need to bubble up as they fall through to the first // child in data(); it _might_ be worth adding constraints here later. if (parent.isValid()) { q->dataChanged(parent, parent, roles); } // When Private::groupDemandingAttention is false, tryToGroup() exempts tasks // which demand attention from being grouped. Therefore if this task is no longer // demanding attention, we need to try grouping it now. if (!parent.isValid() && !groupDemandingAttention && roles.contains(AbstractTasksModel::IsDemandingAttention) && !sourceIndex.data(AbstractTasksModel::IsDemandingAttention).toBool()) { if (shouldGroupTasks() && tryToGroup(sourceIndex)) { q->beginRemoveRows(QModelIndex(), proxyIndex.row(), proxyIndex.row()); rowMap.remove(proxyIndex.row()); q->endRemoveRows(); } else { q->dataChanged(proxyIndex, proxyIndex, roles); } } else { q->dataChanged(proxyIndex, proxyIndex, roles); } } } void TaskGroupingProxyModel::Private::adjustMap(int anchor, int delta) { for (int i = 0; i < rowMap.count(); ++i) { bool changed = false; QVector sourceRows = rowMap.at(i); QMutableVectorIterator it(sourceRows); while (it.hasNext()) { it.next(); if (it.value() >= anchor) { it.setValue(it.value() + delta); changed = true; } } if (changed) { rowMap.replace(i, sourceRows); } } } void TaskGroupingProxyModel::Private::rebuildMap() { rowMap.clear(); const int rows = q->sourceModel()->rowCount(); rowMap.reserve(rows); for (int i = 0; i < rows; ++i) { rowMap.append(QVector{i}); } checkGrouping(true /* silent */); } bool TaskGroupingProxyModel::Private::shouldGroupTasks() { if (groupMode == TasksModel::GroupDisabled) { return false; } if (windowTasksThreshold != -1) { // We're going to check the number of window tasks in the source model // against the grouping threshold. In practice that means we're ignoring // launcher and startup tasks. Startup tasks because they're very short- // lived (i.e. forming/breaking groups as they come and go would be very // noisy) and launcher tasks because we expect consumers to budget for // them in the threshold they set. int windowTasksCount = 0; for (int i = 0; i < q->sourceModel()->rowCount(); ++i) { const QModelIndex &idx = q->sourceModel()->index(i, 0); if (idx.data(AbstractTasksModel::IsWindow).toBool()) { ++windowTasksCount; } } return (windowTasksCount > windowTasksThreshold); } return true; } void TaskGroupingProxyModel::Private::checkGrouping(bool silent) { if (shouldGroupTasks()) { for (int i = (rowMap.count()) - 1; i >= 0; --i) { if (isGroup(i)) { continue; } if (tryToGroup(q->sourceModel()->index(rowMap.at(i).constFirst(), 0), silent)) { q->beginRemoveRows(QModelIndex(), i, i); rowMap.remove(i); // Safe since we're iterating backwards. q->endRemoveRows(); } } } else { for (int i = (rowMap.count()) - 1; i >= 0; --i) { breakGroupFor(q->index(i, 0), silent); } } } bool TaskGroupingProxyModel::Private::isBlacklisted(const QModelIndex &sourceIndex) { // Check app id against blacklist. if (blacklistedAppIds.count() && blacklistedAppIds.contains(sourceIndex.data(AbstractTasksModel::AppId).toString())) { return true; } // Check launcher URL (sans query items) against blacklist. if (blacklistedLauncherUrls.count()) { const QUrl &launcherUrl = sourceIndex.data(AbstractTasksModel::LauncherUrl).toUrl(); const QString &launcherUrlString = launcherUrl.toString(QUrl::PrettyDecoded | QUrl::RemoveQuery); if (blacklistedLauncherUrls.contains(launcherUrlString)) { return true; } } return false; } bool TaskGroupingProxyModel::Private::tryToGroup(const QModelIndex &sourceIndex, bool silent) { // NOTE: We only group window tasks at this time. If this ever changes, the // implementation of data() will have to be adjusted significantly, as for // many roles it currently falls through to the first child item when dealing // with requests for the parent (e.g. IsWindow). if (!sourceIndex.data(AbstractTasksModel::IsWindow).toBool()) { return false; } // If Private::groupDemandingAttention is false and this task is demanding // attention, don't group it at this time. We'll instead try to group it once // it no longer demands attention (see sourceDataChanged()). if (!groupDemandingAttention && sourceIndex.data(AbstractTasksModel::IsDemandingAttention).toBool()) { return false; } // Blacklist checks. if (isBlacklisted(sourceIndex)) { return false; } // Meat of the matter: Try to add this source row to a sub-list with source rows // associated with the same application. for (int i = 0; i < rowMap.count(); ++i) { const QModelIndex &groupRep = q->sourceModel()->index(rowMap.at(i).constFirst(), 0); // Don't match a row with itself. if (sourceIndex == groupRep) { continue; } // Don't group windows with anything other than windows. if (!groupRep.data(AbstractTasksModel::IsWindow).toBool()) { continue; } if (appsMatch(sourceIndex, groupRep)) { const QModelIndex parent = q->index(i, 0); - bool groupFormed = false; if (!silent) { const int newIndex = rowMap.at(i).count(); if (newIndex == 1) { - groupFormed = true; q->beginInsertRows(parent, 0, 1); } else { q->beginInsertRows(parent, newIndex, newIndex); } } rowMap[i].append(sourceIndex.row()); if (!silent) { q->endInsertRows(); - // If we turned a top-level item into a group parent, we need - // to let consumers know. - // TODO: It _might_ be worth optimizing this at some point by - // adding a roles constaint (e.g. IsGroupParent, MimeType, ...); - // see data(). - if (groupFormed) { - q->dataChanged(parent, parent); - } + q->dataChanged(parent, parent); } return true; } } return false; } void TaskGroupingProxyModel::Private::formGroupFor(const QModelIndex &index) { // Already in group or a group. if (index.parent().isValid() || isGroup(index.row())) { return; } // We need to grab a source index as we may invalidate the index passed // in through grouping. const QModelIndex &sourceTarget = q->mapToSource(index); for (int i = (rowMap.count() - 1); i >= 0; --i) { const QModelIndex &sourceIndex = q->sourceModel()->index(rowMap.at(i).constFirst(), 0); if (!appsMatch(sourceTarget, sourceIndex)) { continue; } if (tryToGroup(sourceIndex)) { q->beginRemoveRows(QModelIndex(), i, i); rowMap.remove(i); // Safe since we're iterating backwards. q->endRemoveRows(); } } } void TaskGroupingProxyModel::Private::breakGroupFor(const QModelIndex &index, bool silent) { const int row = index.row(); if (!isGroup(row)) { return; } // The first child will move up to the top level. QVector extraChildren = rowMap.at(row).mid(1); // NOTE: We're going to do remove+insert transactions instead of a // single reparenting move transaction to save on complexity in the // proxies above us. // TODO: This could technically be optimized, though it's very // unlikely to be ever worth it. if (!silent) { q->beginRemoveRows(index, 0, extraChildren.count()); } rowMap[row].resize(1); // index() encodes parent row numbers in indices returned for group // members. endRemoveRows() is not aware of this scheme, and so won't // invalidate persistent indices for the members of the group we're // dissolving. We're now going to do it ourselves. foreach (const QModelIndex &idx, q->persistentIndexList()) { if (idx.internalId() == ((uint)row + 1)) { q->changePersistentIndex(idx, QModelIndex()); } } if (!silent) { q->endRemoveRows(); // We're no longer a group parent. q->dataChanged(index, index); q->beginInsertRows(QModelIndex(), rowMap.count(), rowMap.count() + (extraChildren.count() - 1)); } for (int i = 0; i < extraChildren.count(); ++i) { rowMap.append(QVector{extraChildren.at(i)}); } if (!silent) { q->endInsertRows(); } } TaskGroupingProxyModel::TaskGroupingProxyModel(QObject *parent) : QAbstractProxyModel(parent) , d(new Private(this)) { } TaskGroupingProxyModel::~TaskGroupingProxyModel() { } QModelIndex TaskGroupingProxyModel::index(int row, int column, const QModelIndex &parent) const { if (row < 0 || column != 0) { return QModelIndex(); } if (parent.isValid() && row < d->rowMap.at(parent.row()).count()) { return createIndex(row, column, parent.row() + 1); } if (row < d->rowMap.count()) { return createIndex(row, column, (quintptr)0); } return QModelIndex(); } QModelIndex TaskGroupingProxyModel::parent(const QModelIndex &child) const { return (child.internalId() == 0) ? QModelIndex() : index(child.internalId() - 1, 0); } QModelIndex TaskGroupingProxyModel::mapFromSource(const QModelIndex &sourceIndex) const { if (!sourceIndex.isValid() || sourceIndex.model() != sourceModel()) { return QModelIndex(); } for (int i = 0; i < d->rowMap.count(); ++i) { const QVector &sourceRows = d->rowMap.at(i); const int childIndex = sourceRows.indexOf(sourceIndex.row()); const QModelIndex parent = index(i, 0); if (childIndex == 0) { // If the sub-list we found the source row in is larger than 1 (i.e. part // of a group, map to the logical child item instead of the parent item // the source row also stands in for. The parent is therefore unreachable // from mapToSource(). if (d->isGroup(i)) { return index(0, 0, parent); // Otherwise map to the top-level item. } else { return parent; } } else if (childIndex != -1) { return index(childIndex, 0, parent); } } return QModelIndex(); } QModelIndex TaskGroupingProxyModel::mapToSource(const QModelIndex &proxyIndex) const { if (!proxyIndex.isValid() || proxyIndex.model() != this || !sourceModel()) { return QModelIndex(); } const QModelIndex &parent = proxyIndex.parent(); if (parent.isValid()) { if (parent.row() < 0 || parent.row() >= d->rowMap.count()) { return QModelIndex(); } return sourceModel()->index(d->rowMap.at(parent.row()).at(proxyIndex.row()), 0); } else { // Group parents items therefore equate to the first child item; the source // row logically appears twice in the proxy. // mapFromSource() is not required to handle this well (consider proxies can // filter out rows, too) and opts to map to the child item, as the group parent // has its Qt::DisplayRole mangled by data(), and it's more useful for trans- // lating dataChanged() from the source model. return sourceModel()->index(d->rowMap.at(proxyIndex.row()).at(0), 0); } return QModelIndex(); } int TaskGroupingProxyModel::rowCount(const QModelIndex &parent) const { if (!sourceModel()) { return 0; } if (parent.isValid() && parent.model() == this) { // Don't return row count for top-level item at child row: Group members // never have further children of their own. if (parent.parent().isValid()) { return 0; } if (parent.row() < 0 || parent.row() >= d->rowMap.count()) { return 0; } const uint rowCount = d->rowMap.at(parent.row()).count(); // If this sub-list in the map only has one entry, it's a plain item, not // parent to a group. if (rowCount == 1) { return 0; } else { return rowCount; } } return d->rowMap.count(); } bool TaskGroupingProxyModel::hasChildren(const QModelIndex &parent) const { if ((parent.model() && parent.model() != this) || !sourceModel()) { return false; } return rowCount(parent); } int TaskGroupingProxyModel::columnCount(const QModelIndex &parent) const { Q_UNUSED(parent) return 1; } QVariant TaskGroupingProxyModel::data(const QModelIndex &proxyIndex, int role) const { if (!proxyIndex.isValid() || proxyIndex.model() != this || !sourceModel()) { return QVariant(); } const QModelIndex &parent = proxyIndex.parent(); const bool isGroup = (!parent.isValid() && d->isGroup(proxyIndex.row())); // For group parent items, this will map to the first child task. const QModelIndex &sourceIndex = mapToSource(proxyIndex); if (!sourceIndex.isValid()) { return QVariant(); } if (role == AbstractTasksModel::IsGroupable) { return !d->isBlacklisted(sourceIndex); } if (isGroup) { // For group parent items, DisplayRole is mapped to AppName of the first child. if (role == Qt::DisplayRole) { const QString &appName = sourceIndex.data(AbstractTasksModel::AppName).toString(); // Groups are formed by app id or launcher URL; neither requires // AppName to be available. If it's not, fall back to the app id /// rather than an empty string. if (appName.isEmpty()) { return sourceIndex.data(AbstractTasksModel::AppId); } return appName; } else if (role == AbstractTasksModel::LegacyWinIdList) { QVariantList winIds; for (int i = 0; i < rowCount(proxyIndex); ++i) { winIds.append(proxyIndex.child(i, 0).data(AbstractTasksModel::LegacyWinIdList).toList()); } return winIds; } else if (role == AbstractTasksModel::MimeType) { // FIXME: Legacy X11 stuff, but it's what we have for now. return QStringLiteral("windowsystem/multiple-winids"); } else if (role == AbstractTasksModel::MimeData) { // FIXME TODO: Implement. return QVariant(); } else if (role == AbstractTasksModel::IsGroupParent) { return true; + } else if (role == AbstractTasksModel::ChildCount) { + return rowCount(proxyIndex); } else if (role == AbstractTasksModel::IsActive) { return d->any(proxyIndex, AbstractTasksModel::IsActive); } else if (role == AbstractTasksModel::IsClosable) { return d->all(proxyIndex, AbstractTasksModel::IsClosable); } else if (role == AbstractTasksModel::IsMovable) { // Moving groups makes no sense. return false; } else if (role == AbstractTasksModel::IsResizable) { // Resizing groups makes no sense. return false; } else if (role == AbstractTasksModel::IsMaximizable) { return d->all(proxyIndex, AbstractTasksModel::IsMaximizable); } else if (role == AbstractTasksModel::IsMaximized) { return d->all(proxyIndex, AbstractTasksModel::IsMaximized); } else if (role == AbstractTasksModel::IsMinimizable) { return d->all(proxyIndex, AbstractTasksModel::IsMinimizable); } else if (role == AbstractTasksModel::IsMinimized) { return d->all(proxyIndex, AbstractTasksModel::IsMinimized); } else if (role == AbstractTasksModel::IsKeepAbove) { return d->all(proxyIndex, AbstractTasksModel::IsKeepAbove); } else if (role == AbstractTasksModel::IsKeepBelow) { return d->all(proxyIndex, AbstractTasksModel::IsKeepBelow); } else if (role == AbstractTasksModel::IsFullScreenable) { return d->all(proxyIndex, AbstractTasksModel::IsFullScreenable); } else if (role == AbstractTasksModel::IsFullScreen) { return d->all(proxyIndex, AbstractTasksModel::IsFullScreen); } else if (role == AbstractTasksModel::IsShadeable) { return d->all(proxyIndex, AbstractTasksModel::IsShadeable); } else if (role == AbstractTasksModel::IsShaded) { return d->all(proxyIndex, AbstractTasksModel::IsShaded); } else if (role == AbstractTasksModel::IsVirtualDesktopChangeable) { return d->all(proxyIndex, AbstractTasksModel::IsVirtualDesktopChangeable); } else if (role == AbstractTasksModel::VirtualDesktop) { // TODO: Nothing needs this for now and it would add complexity to // make it a list; skip it until needed. return QVariant(); } else if (role == AbstractTasksModel::ScreenGeometry) { // TODO: Nothing needs this for now and it would add complexity to // make it a list; skip it until needed. return QVariant(); } else if (role == AbstractTasksModel::Activities) { QStringList activities; for (int i = 0; i < rowCount(proxyIndex); ++i) { activities.append(proxyIndex.child(i, 0).data(AbstractTasksModel::Activities).toStringList()); } activities.removeDuplicates(); return activities; } else if (role == AbstractTasksModel::IsDemandingAttention) { return d->any(proxyIndex, AbstractTasksModel::IsDemandingAttention); } else if (role == AbstractTasksModel::SkipTaskbar) { return d->all(proxyIndex, AbstractTasksModel::SkipTaskbar); } } return sourceIndex.data(role); } void TaskGroupingProxyModel::setSourceModel(QAbstractItemModel *sourceModel) { if (sourceModel == QAbstractProxyModel::sourceModel()) { return; } beginResetModel(); if (QAbstractProxyModel::sourceModel()) { QAbstractProxyModel::sourceModel()->disconnect(this); } QAbstractProxyModel::setSourceModel(sourceModel); d->abstractTasksSourceModel = dynamic_cast(sourceModel); if (sourceModel) { d->rebuildMap(); connect(sourceModel, SIGNAL(rowsAboutToBeInserted(QModelIndex,int,int)), this, SLOT(sourceRowsAboutToBeInserted(QModelIndex,int,int)), Qt::UniqueConnection); connect(sourceModel, SIGNAL(rowsInserted(QModelIndex,int,int)), this, SLOT(sourceRowsInserted(QModelIndex,int,int)), Qt::UniqueConnection); connect(sourceModel, SIGNAL(rowsAboutToBeRemoved(QModelIndex,int,int)), this, SLOT(sourceRowsAboutToBeRemoved(QModelIndex,int,int)), Qt::UniqueConnection); connect(sourceModel, SIGNAL(rowsRemoved(QModelIndex,int,int)), this, SLOT(sourceRowsRemoved(QModelIndex,int,int)), Qt::UniqueConnection); connect(sourceModel, SIGNAL(modelAboutToBeReset()), this, SLOT(sourceModelAboutToBeReset()), Qt::UniqueConnection); connect(sourceModel, SIGNAL(modelReset()), this, SLOT(sourceModelReset()), Qt::UniqueConnection); connect(sourceModel, SIGNAL(dataChanged(QModelIndex,QModelIndex,QVector)), this, SLOT(sourceDataChanged(QModelIndex,QModelIndex,QVector)), Qt::UniqueConnection); } else { d->rowMap.clear(); } endResetModel(); } TasksModel::GroupMode TaskGroupingProxyModel::groupMode() const { return d->groupMode; } void TaskGroupingProxyModel::setGroupMode(TasksModel::GroupMode mode) { if (d->groupMode != mode) { d->groupMode = mode; d->checkGrouping(); emit groupModeChanged(); } } bool TaskGroupingProxyModel::groupDemandingAttention() const { return d->groupDemandingAttention; } void TaskGroupingProxyModel::setGroupDemandingAttention(bool group) { if (d->groupDemandingAttention != group) { d->groupDemandingAttention = group; d->checkGrouping(); emit groupDemandingAttentionChanged(); } } int TaskGroupingProxyModel::windowTasksThreshold() const { return d->windowTasksThreshold; } void TaskGroupingProxyModel::setWindowTasksThreshold(int threshold) { if (d->windowTasksThreshold != threshold) { d->windowTasksThreshold = threshold; d->checkGrouping(); emit windowTasksThresholdChanged(); } } QStringList TaskGroupingProxyModel::blacklistedAppIds() const { return d->blacklistedAppIds.toList(); } void TaskGroupingProxyModel::setBlacklistedAppIds(const QStringList &list) { const QSet &set = QSet::fromList(list); if (d->blacklistedAppIds != set) { d->blacklistedAppIds = set; // checkGrouping() will gather and group up what's newly-allowed under the changed // blacklist. d->checkGrouping(); // Now break apart what we need to. for (int i = (d->rowMap.count() - 1); i >= 0; --i) { if (d->isGroup(i)) { const QModelIndex &groupRep = index(i, 0); if (set.contains(groupRep.data(AbstractTasksModel::AppId).toString())) { d->breakGroupFor(groupRep); // Safe since we're iterating backwards. } } } emit blacklistedAppIdsChanged(); } } QStringList TaskGroupingProxyModel::blacklistedLauncherUrls() const { return d->blacklistedLauncherUrls.toList(); } void TaskGroupingProxyModel::setBlacklistedLauncherUrls(const QStringList &list) { const QSet &set = QSet::fromList(list); if (d->blacklistedLauncherUrls != set) { d->blacklistedLauncherUrls = set; // checkGrouping() will gather and group up what's newly-allowed under the changed // blacklist. d->checkGrouping(); // Now break apart what we need to. for (int i = (d->rowMap.count() - 1); i >= 0; --i) { if (d->isGroup(i)) { const QModelIndex &groupRep = index(i, 0); const QUrl &launcherUrl = groupRep.data(AbstractTasksModel::LauncherUrl).toUrl(); const QString &launcherUrlString = launcherUrl.toString(QUrl::RemoveQuery | QUrl::RemoveQuery); if (set.contains(launcherUrlString)) { d->breakGroupFor(groupRep); // Safe since we're iterating backwards. } } } emit blacklistedLauncherUrlsChanged(); } } void TaskGroupingProxyModel::requestActivate(const QModelIndex &index) { if (!d->abstractTasksSourceModel || !index.isValid() || index.model() != this) { return; } if (index.parent().isValid() || !d->isGroup(index.row())) { d->abstractTasksSourceModel->requestActivate(mapToSource(index)); } } void TaskGroupingProxyModel::requestNewInstance(const QModelIndex &index) { if (!d->abstractTasksSourceModel || !index.isValid() || index.model() != this) { return; } d->abstractTasksSourceModel->requestNewInstance(mapToSource(index)); } void TaskGroupingProxyModel::requestOpenUrls(const QModelIndex &index, const QList &urls) { if (!d->abstractTasksSourceModel || !index.isValid() || index.model() != this) { return; } d->abstractTasksSourceModel->requestOpenUrls(mapToSource(index), urls); } void TaskGroupingProxyModel::requestClose(const QModelIndex &index) { if (!d->abstractTasksSourceModel || !index.isValid() || index.model() != this) { return; } if (index.parent().isValid() || !d->isGroup(index.row())) { d->abstractTasksSourceModel->requestClose(mapToSource(index)); } else { const int row = index.row(); for (int i = (rowCount(index) - 1); i >= 1; --i) { const QModelIndex &sourceChild = mapToSource(index.child(i, 0)); d->abstractTasksSourceModel->requestClose(sourceChild); } d->abstractTasksSourceModel->requestClose(mapToSource(TaskGroupingProxyModel::index(row, 0))); } } void TaskGroupingProxyModel::requestMove(const QModelIndex &index) { if (!d->abstractTasksSourceModel || !index.isValid() || index.model() != this) { return; } if (index.parent().isValid() || !d->isGroup(index.row())) { d->abstractTasksSourceModel->requestMove(mapToSource(index)); } } void TaskGroupingProxyModel::requestResize(const QModelIndex &index) { if (!d->abstractTasksSourceModel || !index.isValid() || index.model() != this) { return; } if (index.parent().isValid() || !d->isGroup(index.row())) { d->abstractTasksSourceModel->requestResize(mapToSource(index)); } } void TaskGroupingProxyModel::requestToggleMinimized(const QModelIndex &index) { if (!d->abstractTasksSourceModel || !index.isValid() || index.model() != this) { return; } if (index.parent().isValid() || !d->isGroup(index.row())) { d->abstractTasksSourceModel->requestToggleMinimized(mapToSource(index)); } else { const bool goalState = !index.data(AbstractTasksModel::IsMinimized).toBool(); for (int i = 0; i < rowCount(index); ++i) { const QModelIndex &child = index.child(i, 0); if (child.data(AbstractTasksModel::IsMinimized).toBool() != goalState) { d->abstractTasksSourceModel->requestToggleMinimized(mapToSource(child)); } } } } void TaskGroupingProxyModel::requestToggleMaximized(const QModelIndex &index) { if (!d->abstractTasksSourceModel || !index.isValid() || index.model() != this) { return; } if (index.parent().isValid() || !d->isGroup(index.row())) { d->abstractTasksSourceModel->requestToggleMaximized(mapToSource(index)); } else { const bool goalState = !index.data(AbstractTasksModel::IsMaximized).toBool(); for (int i = 0; i < rowCount(index); ++i) { const QModelIndex &child = index.child(i, 0); if (child.data(AbstractTasksModel::IsMaximized).toBool() != goalState) { d->abstractTasksSourceModel->requestToggleMaximized(mapToSource(child)); } } } } void TaskGroupingProxyModel::requestToggleKeepAbove(const QModelIndex &index) { if (!d->abstractTasksSourceModel || !index.isValid() || index.model() != this) { return; } if (index.parent().isValid() || !d->isGroup(index.row())) { d->abstractTasksSourceModel->requestToggleKeepAbove(mapToSource(index)); } else { const bool goalState = !index.data(AbstractTasksModel::IsKeepAbove).toBool(); for (int i = 0; i < rowCount(index); ++i) { const QModelIndex &child = index.child(i, 0); if (child.data(AbstractTasksModel::IsKeepAbove).toBool() != goalState) { d->abstractTasksSourceModel->requestToggleKeepAbove(mapToSource(child)); } } } } void TaskGroupingProxyModel::requestToggleKeepBelow(const QModelIndex &index) { if (!d->abstractTasksSourceModel || !index.isValid() || index.model() != this) { return; } if (index.parent().isValid() || !d->isGroup(index.row())) { d->abstractTasksSourceModel->requestToggleKeepBelow(mapToSource(index)); } else { const bool goalState = !index.data(AbstractTasksModel::IsKeepBelow).toBool(); for (int i = 0; i < rowCount(index); ++i) { const QModelIndex &child = index.child(i, 0); if (child.data(AbstractTasksModel::IsKeepBelow).toBool() != goalState) { d->abstractTasksSourceModel->requestToggleKeepBelow(mapToSource(child)); } } } } void TaskGroupingProxyModel::requestToggleFullScreen(const QModelIndex &index) { if (!d->abstractTasksSourceModel || !index.isValid() || index.model() != this) { return; } if (index.parent().isValid() || !d->isGroup(index.row())) { d->abstractTasksSourceModel->requestToggleFullScreen(mapToSource(index)); } else { const bool goalState = !index.data(AbstractTasksModel::IsFullScreen).toBool(); for (int i = 0; i < rowCount(index); ++i) { const QModelIndex &child = index.child(i, 0); if (child.data(AbstractTasksModel::IsFullScreen).toBool() != goalState) { d->abstractTasksSourceModel->requestToggleFullScreen(mapToSource(child)); } } } } void TaskGroupingProxyModel::requestToggleShaded(const QModelIndex &index) { if (!d->abstractTasksSourceModel || !index.isValid() || index.model() != this) { return; } if (index.parent().isValid() || !d->isGroup(index.row())) { d->abstractTasksSourceModel->requestToggleShaded(mapToSource(index)); } else { const bool goalState = !index.data(AbstractTasksModel::IsShaded).toBool(); for (int i = 0; i < rowCount(index); ++i) { const QModelIndex &child = index.child(i, 0); if (child.data(AbstractTasksModel::IsShaded).toBool() != goalState) { d->abstractTasksSourceModel->requestToggleShaded(mapToSource(child)); } } } } void TaskGroupingProxyModel::requestVirtualDesktop(const QModelIndex &index, qint32 desktop) { if (!d->abstractTasksSourceModel || !index.isValid() || index.model() != this) { return; } if (index.parent().isValid() || !d->isGroup(index.row())) { d->abstractTasksSourceModel->requestVirtualDesktop(mapToSource(index), desktop); } else { const int row = index.row(); for (int i = (rowCount(index) - 1); i >= 1; --i) { const QModelIndex &sourceChild = mapToSource(index.child(i, 0)); d->abstractTasksSourceModel->requestVirtualDesktop(sourceChild, desktop); } d->abstractTasksSourceModel->requestVirtualDesktop(mapToSource(TaskGroupingProxyModel::index(row, 0)), desktop); } } void TaskGroupingProxyModel::requestActivities(const QModelIndex &index, const QStringList &activities) { if (!d->abstractTasksSourceModel || !index.isValid() || index.model() != this) { return; } if (index.parent().isValid() || !d->isGroup(index.row())) { d->abstractTasksSourceModel->requestActivities(mapToSource(index), activities); } else { const int row = index.row(); for (int i = (rowCount(index) - 1); i >= 1; --i) { const QModelIndex &sourceChild = mapToSource(index.child(i, 0)); d->abstractTasksSourceModel->requestActivities(sourceChild, activities); } d->abstractTasksSourceModel->requestActivities(mapToSource(TaskGroupingProxyModel::index(row, 0)), activities); } } void TaskGroupingProxyModel::requestPublishDelegateGeometry(const QModelIndex &index, const QRect &geometry, QObject *delegate) { if (!d->abstractTasksSourceModel || !index.isValid() || index.model() != this) { return; } if (index.parent().isValid() || !d->isGroup(index.row())) { d->abstractTasksSourceModel->requestPublishDelegateGeometry(mapToSource(index), geometry, delegate); } else { for (int i = 0; i < rowCount(index); ++i) { d->abstractTasksSourceModel->requestPublishDelegateGeometry(mapToSource(index.child(i, 0)), geometry, delegate); } } } void TaskGroupingProxyModel::requestToggleGrouping(const QModelIndex &index) { const QString &appId = index.data(AbstractTasksModel::AppId).toString(); const QUrl &launcherUrl = index.data(AbstractTasksModel::LauncherUrl).toUrl(); const QString &launcherUrlString = launcherUrl.toString(QUrl::RemoveQuery | QUrl::RemoveQuery); if (d->blacklistedAppIds.contains(appId) || d->blacklistedLauncherUrls.contains(launcherUrlString)) { d->blacklistedAppIds.remove(appId); d->blacklistedLauncherUrls.remove(launcherUrlString); if (d->groupMode != TasksModel::GroupDisabled) { d->formGroupFor(index.parent().isValid() ? index.parent() : index); } } else { d->blacklistedAppIds.insert(appId); d->blacklistedLauncherUrls.insert(launcherUrlString); if (d->groupMode != TasksModel::GroupDisabled) { d->breakGroupFor(index.parent().isValid() ? index.parent() : index); } } // Update IsGroupable data role for all relevant top-level items. We don't need to update // for group members since they've just been inserted -- it's logically impossible to // toggle grouping _on_ from a group member. for (int i = 0; i < d->rowMap.count(); ++i) { if (!d->isGroup(i)) { const QModelIndex &idx = TaskGroupingProxyModel::index(i, 0); if (idx.data(AbstractTasksModel::AppId).toString() == appId || launcherUrlsMatch(idx.data(AbstractTasksModel::LauncherUrl).toUrl(), launcherUrl, IgnoreQueryItems)) { dataChanged(idx, idx, QVector{AbstractTasksModel::IsGroupable}); } } } emit blacklistedAppIdsChanged(); emit blacklistedLauncherUrlsChanged(); } } #include "moc_taskgroupingproxymodel.cpp"