trinity/include/roomlistsortmodel.h

#pragma once

#include <QSortFilterProxyModel>

class RoomListSortModel : public QSortFilterProxyModel
{
    Q_OBJECT
public:
    bool lessThan(const QModelIndex &source_left, const QModelIndex &source_right) const;

    Q_INVOKABLE unsigned int getOriginalIndex(const unsigned int i) const {
        auto const proxyIndex = index(i, 0);
        auto const sourceIndex = mapToSource(proxyIndex);

        if(!sourceIndex.isValid())
            return 0;
        else
            return sourceIndex.row();
    }
};

class MemberListSortModel : public QSortFilterProxyModel
{
    Q_OBJECT
public:
    bool lessThan(const QModelIndex &source_left, const QModelIndex &source_right) const;

    Q_INVOKABLE unsigned int getOriginalIndex(const unsigned int i) const {
        auto const proxyIndex = index(i, 0);
        auto const sourceIndex = mapToSource(proxyIndex);

        if(!sourceIndex.isValid())
            return 0;
        else
            return sourceIndex.row();
    }
};