summaryrefslogtreecommitdiff
path: root/src/viewmarkingscrollbar.cpp
blob: 596af533c5965cdc04aae6adef3af130f982cd70 (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
#include "viewmarkingscrollbar.h"
#include <QPainter>
#include <QStyle>
#include <QStyleOptionSlider>

ViewMarkingScrollBar::ViewMarkingScrollBar(QAbstractItemModel* model, QWidget* parent, int role)
    : QScrollBar(parent), m_Model(model), m_Role(role) {
    // not implemented for horizontal sliders
    Q_ASSERT(this->orientation() == Qt::Vertical);
}

void ViewMarkingScrollBar::paintEvent(QPaintEvent* event) {
    if (m_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);

    painter.translate(innerRect.topLeft() + QPoint(0, 3));
    qreal scale = static_cast<qreal>(innerRect.height() - 3) / static_cast<qreal>(m_Model->rowCount());

    for (int i = 0; i < m_Model->rowCount(); ++i) {
        QVariant data = m_Model->data(m_Model->index(i, 0), m_Role);
        if (data.isValid()) {
            QColor col = data.value<QColor>();
            painter.setPen(col);
            painter.setBrush(col);
            painter.drawRect(QRect(2, i * scale - 2, handleRect.width() - 5, 3));
        }
    }
}