summaryrefslogtreecommitdiff
path: root/src/backend/bookshelfmodel/btcheckstatefilterproxymodel.cpp
blob: 184bd39de1800d505863b5ae5e36475371b1c5ef (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
66
67
68
69
70
/*********
*
* In the name of the Father, and of the Son, and of the Holy Spirit.
*
* This file is part of BibleTime's source code, http://www.bibletime.info/.
*
* Copyright 1999-2009 by the BibleTime developers.
* The BibleTime source code is licensed under the GNU General Public License
* version 2.0.
*
**********/

#include "backend/bookshelfmodel/btcheckstatefilterproxymodel.h"

BtCheckStateFilterProxyModel::BtCheckStateFilterProxyModel(QObject *parent)
        : QSortFilterProxyModel(parent), m_enabled(true), m_showChecked(true),
        m_showUnchecked(false), m_showPartiallyChecked(true) {
    setFilterRole(Qt::CheckStateRole);
}

BtCheckStateFilterProxyModel::~BtCheckStateFilterProxyModel() {
    // Intentionally empty
}

void BtCheckStateFilterProxyModel::setEnabled(bool enable) {
    if (enable == m_enabled) return;
    m_enabled = enable;
    invalidateFilter();
}

void BtCheckStateFilterProxyModel::setShowChecked(bool show) {
    if (m_showChecked == show) return;
    m_showChecked = show;
    invalidateFilter();
}

void BtCheckStateFilterProxyModel::setShowUnchecked(bool show) {
    if (m_showUnchecked == show) return;
    m_showUnchecked = show;
    invalidateFilter();
}

void BtCheckStateFilterProxyModel::setShowPartiallyChecked(bool show) {
    if (m_showPartiallyChecked == show) return;
    m_showPartiallyChecked = show;
    invalidateFilter();
}

bool BtCheckStateFilterProxyModel::filterAcceptsRow(int row,
        const QModelIndex &parent) const {
    typedef Qt::CheckState CS;

    if (!m_enabled) return true;

    QAbstractItemModel *m(sourceModel());

    QModelIndex i(m->index(row, filterKeyColumn(), parent));
    CS state((CS) m->data(i, filterRole()).toInt());
    Q_ASSERT(state == Qt::Checked || state == Qt::Unchecked ||
             state == Qt::PartiallyChecked);
    if (state == Qt::Unchecked) {
        return m_showUnchecked;
    }
    else if (state == Qt::Checked) {
        return m_showChecked;
    }
    else {
        return m_showPartiallyChecked;
    }
}