QTreeView::branch 设置子目录没有子项也有图标-QT开发中文网

可以通过在QStandardItemModel中添加一个特殊的标志来实现子目录即使没有子项也有图标的效果。具体步骤如下：

定义一个枚举类型，用于表示特殊标志：

enum MyItemFlags {
    HasIcon = Qt::UserRole + 1, // 自定义标志：是否显示图标
};

在创建模型时，设置每个节点默认不显示图标：

QStandardItemModel *model = new QStandardItemModel(this);
model->setItemRoleNames({{MyItemFlags::HasIcon, "hasIcon"}});
// 设置每个节点都不显示图标
for (int i = 0; i < model->rowCount(); ++i) {
    for (int j = 0; j < model->columnCount(); ++j) {
        QModelIndex index = model->index(i, j);
        QStandardItem *item = model->itemFromIndex(index);
        item->setData(false, MyItemFlags::HasIcon);
    }
}

在需要显示图标的节点上设置特殊标志：

// 设置某些节点需要显示图标
QModelIndex index1 = model->index(0, 0);
QStandardItem *item1 = model->itemFromIndex(index1);
item1->setData(true, MyItemFlags::HasIcon);

QModelIndex index2 = model->index(1, 0);
QStandardItem *item2 = model->itemFromIndex(index2);
item2->setData(true, MyItemFlags::HasIcon);

// ...

重写drawBranches函数，在绘制每个节点时判断是否需要显示图标：

void MyTreeView::drawBranches(QPainter *painter, const QRect &rect, const QModelIndex &index) const
{
    QTreeView::drawBranches(painter, rect, index);

    if (!isExpanded(index)) {
        return;
    }

    // 如果当前节点没有子项，则绘制一个图标
    QStandardItemModel *model = qobject_cast<QStandardItemModel *>(const_cast<QAbstractItemModel *>(this->model()));
    if (model && model->data(index, MyItemFlags::HasIcon).toBool()) {
        QPixmap pixmap(":/icons/folder.png");
        QPoint center = rect.center() - QPoint(pixmap.width() / 2, pixmap.height() / 2);
        painter->drawPixmap(center, pixmap);
    }
}

上述代码中，我们首先调用了父类的drawBranches函数来绘制原本的箭头图标。然后判断当前节点是否已展开，如果已展开则进一步判断该节点是否有子项。如果没有子项并且设置了特殊标志，则加载一个图标并在节点中心位置绘制即可。这样就实现了子目录即使没有子项也有图标的效果。