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
71
72
73
74
75
76
77
78
79
80
|
#include "ViewMarkingScrollBar.h"
#include <QPainter>
#include <QStyle>
#include <QStyleOptionSlider>
namespace GUI
{
static QModelIndexList visibleIndexImpl(QTreeView* view, int column,
const QModelIndex& parent)
{
if (parent.isValid() && !view->isExpanded(parent)) {
return {};
}
const auto* const model = view->model();
QModelIndexList index;
for (int i = 0; i < model->rowCount(parent); ++i) {
index.append(model->index(i, column, parent));
index.append(visibleIndexImpl(view, column, index.back()));
}
return index;
}
static QModelIndexList visibleIndex(QTreeView* view, int column)
{
return visibleIndexImpl(view, column, QModelIndex());
}
ViewMarkingScrollBar::ViewMarkingScrollBar(QTreeView* view, int role)
: QScrollBar(view), m_View{view}, m_Role{role}
{
// not implemented for horizontal sliders
Q_ASSERT(this->orientation() == Qt::Vertical);
}
QColor ViewMarkingScrollBar::color(const QModelIndex& index) const
{
const auto data = index.data(m_Role);
if (data.canConvert<QColor>()) {
return data.value<QColor>();
}
return QColor();
}
void ViewMarkingScrollBar::paintEvent(QPaintEvent* event)
{
if (m_View->model() == nullptr) {
return;
}
QScrollBar::paintEvent(event);
QStyleOptionSlider styleOption;
initStyleOption(&styleOption);
QPainter painter(this);
QRect handleRect = style()->subControlRect(QStyle::CC_ScrollBar, &styleOption,
QStyle::SC_ScrollBarSlider, this);
QRect innerRect = style()->subControlRect(QStyle::CC_ScrollBar, &styleOption,
QStyle::SC_ScrollBarGroove, this);
auto indices = visibleIndex(m_View, 0);
painter.translate(innerRect.topLeft() + QPoint(0, 3));
const qreal scale =
static_cast<qreal>(innerRect.height() - 3) / static_cast<qreal>(indices.size());
for (int i = 0; i < indices.size(); ++i) {
const QColor color = this->color(indices[i]);
if (color.isValid()) {
painter.setPen(color);
painter.setBrush(color);
painter.drawRect(QRect(2, i * scale - 2, handleRect.width() - 5, 3));
}
}
}
} // namespace GUI
|
