devicessortproxymodel.cpp 2.44 KB
Newer Older
1 2
/**
 * Copyright 2013 Albert Vaca <albertvaka@gmail.com>
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
 *
 * 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) 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 14 of version 3 of the license.
 *
 * 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
18
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
19 20 21
 */

#include "devicessortproxymodel.h"
22

Nicolas Fella's avatar
Nicolas Fella committed
23 24
#include "dbusinterfaces.h"
#include "devicesmodel.h"
25 26 27 28 29

DevicesSortProxyModel::DevicesSortProxyModel(DevicesModel* devicesModel)
    : QSortFilterProxyModel(devicesModel)
{
    setSourceModel(devicesModel);
30 31
}

32
void DevicesSortProxyModel::setSourceModel(QAbstractItemModel* devicesModel)
33 34 35 36
{
    QSortFilterProxyModel::setSourceModel(devicesModel);
    if (devicesModel) {
        setSortRole(DevicesModel::StatusModelRole);
37
        connect(devicesModel, &QAbstractItemModel::dataChanged, this, &DevicesSortProxyModel::sourceDataChanged);
38
    }
39 40 41
    sort(0);
}

Aleix Pol Gonzalez's avatar
Aleix Pol Gonzalez committed
42
void DevicesSortProxyModel::sourceDataChanged()
43 44 45 46 47 48
{
    sort(0);
}

bool DevicesSortProxyModel::lessThan(const QModelIndex& left, const QModelIndex& right) const
{
49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64
    QAbstractItemModel* model = sourceModel();
    Q_ASSERT(qobject_cast<DevicesModel*>(model));

    //Show connected devices first
    int statusLeft = model->data(left, DevicesModel::StatusModelRole).toInt();
    int statusRight = model->data(right, DevicesModel::StatusModelRole).toInt();

    if (statusLeft != statusRight) {
        return statusLeft > statusRight;
    }

    //Fallback to alphabetical order
    QString nameLeft = model->data(left, DevicesModel::NameModelRole).toString();
    QString nameRight = model->data(right, DevicesModel::NameModelRole).toString();

    return nameLeft > nameRight;
65 66 67 68 69 70 71 72 73 74

}

bool DevicesSortProxyModel::filterAcceptsRow(int source_row, const QModelIndex& source_parent) const
{
    Q_UNUSED(source_row);
    Q_UNUSED(source_parent);
    //Possible to-do: Implement filter
    return true;
}