blob: d4f0c984c10e0797d6af20e4d0d654b5416b018a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
// Copyright (C) 2022 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0+ OR GPL-3.0 WITH Qt-GPL-exception-1.0
#pragma once
#include <QSortFilterProxyModel>
namespace Utils {
class SortFilterModel : public QSortFilterProxyModel
{
public:
using FilterColumnFunction = std::function<bool(int, const QModelIndex &source_parent)>;
using FilterRowFunction = std::function<bool(int, const QModelIndex &source_parent)>;
using LessThanFunction = std::function<bool(const QModelIndex &left, const QModelIndex &right)>;
using QSortFilterProxyModel::QSortFilterProxyModel;
bool filterAcceptsRow(int source_row, const QModelIndex &source_parent) const override
{
if (!m_filterRowFunction)
return true;
return m_filterRowFunction(source_row, source_parent);
}
bool filterAcceptsColumn(int source_column, const QModelIndex &source_parent) const override
{
if (!m_filterColumnFunction)
return true;
return m_filterColumnFunction(source_column, source_parent);
}
bool lessThan(const QModelIndex &left, const QModelIndex &right) const override
{
if (!m_lessThanFunction)
return QSortFilterProxyModel::lessThan(left, right);
return m_lessThanFunction(left, right);
}
void setFilterRowFunction(FilterRowFunction filterRowFunction)
{
m_filterRowFunction = filterRowFunction;
invalidate();
}
void setFilterColumnFunction(FilterColumnFunction filterColumnFunction)
{
m_filterColumnFunction = filterColumnFunction;
invalidate();
}
void setLessThanFunction(LessThanFunction lessThanFunction)
{
m_lessThanFunction = lessThanFunction;
invalidate();
}
private:
FilterRowFunction m_filterRowFunction;
FilterColumnFunction m_filterColumnFunction;
LessThanFunction m_lessThanFunction;
};
} // namespace Utils
|