QTreeView复选框颜色

#include <QApplication>
#include <QTreeView>
#include <QStandardItemModel>
#include <QStyleOptionButton>
#include <QPainter>

class MyStyle : public QStyle
{
public:
    void drawPrimitive(PrimitiveElement element, const QStyleOption* option, QPainter* painter, const QWidget* widget = nullptr) const override
    {
        if (element == PE_IndicatorCheckBox)
        {
            const QStyleOptionButton* buttonOption = qstyleoption_cast<const QStyleOptionButton*>(option);
            if (buttonOption)
            {
                QStyleOptionButton modifiedOption(*buttonOption);
                modifiedOption.palette.setColor(QPalette::Active, QPalette::ButtonText, Qt::red); // 修改复选框的颜色为红色
                QStyle::drawPrimitive(element, &modifiedOption, painter, widget);
                return;
            }
        }
        QStyle::drawPrimitive(element, option, painter, widget);
    }
};

int main(int argc, char *argv[])
{
    QApplication a(argc, argv);

    QTreeView treeView;
    QStandardItemModel model;
    treeView.setModel(&model);

    // 添加树形数据
    QStandardItem* rootItem = new QStandardItem("Root");
    model.appendRow(rootItem);
    for (int i = 0; i < 5; ++i)
    {
        QStandardItem* childItem = new QStandardItem(QString("Child %1").arg(i));
        childItem->setCheckable(true);
        rootItem->appendRow(childItem);
    }

    // 设置自定义样式
    MyStyle* myStyle = new MyStyle;
    treeView.setStyle(myStyle);

    treeView.show();

    return a.exec();
}