CheckBoxHeader.h

// Author: Kang Lin <kl222@126.com>
 
#pragma once
#include <QHeaderView>
#include <QSet>

class CCheckBoxHeader : public QHeaderView {
    Q_OBJECT
 
public:
    explicit CCheckBoxHeader(Qt::Orientation orientation, QWidget *parent = nullptr);

    void SetCheckable(int index, bool checkable, Qt::CheckState state = Qt::Unchecked);
    bool isCheckable(int index) const;
    void SetCheckState(int index, Qt::CheckState state);
    Qt::CheckState GetCheckState(int index) const;
 
signals:
    void sigCheckStateChanged(int index, Qt::CheckState state);
 
protected:
    void paintSection(QPainter *painter, const QRect &rect, int logicalIndex) const override;
    void mousePressEvent(QMouseEvent *event) override;
 
private:
    QRect CheckboxRect(const QRect &sectionRect) const;
    void DrawText(QPainter* painter, const QRect& rect, 
                  const QString& text, Qt::Alignment alignment) const;
    void DrawCheckBox(QPainter* painter, const QRect& rect,
                      Qt::CheckState state, bool enabled) const;
 
    QHash<int, Qt::CheckState> m_CheckableSections;
};