如何获取QAbstractItemModel树形项的根项及判断根项

针对你在QT5 TreeModel/TreeView中需要的两个功能，对应实现方式如下：

1. 验证当前项是否为根项 #

在QAbstractItemModel体系中，顶级根项的父索引是无效索引，因此可以通过判断index.parent().isValid()来确定当前项是否为根项，对应你之前用TreeWidgetItem::parent()是否为null的逻辑。

示例代码：

// 在TreeModel中添加成员函数，或直接在需要的地方判断
bool TreeModel::isRootItem(const QModelIndex &index) const
{
    if (!index.isValid())
        return false;
    // 父索引无效则为根项
    return !index.parent().isValid();
}

// 调用示例（比如在data方法中）
QVariant TreeModel::data(const QModelIndex &index, int role) const
{
    if (!index.isValid())
        return QVariant();

    if (role == Qt::DisplayRole) {
        TreeItem *item = static_cast<TreeItem*>(index.internalPointer());
        if (isRootItem(index)) {
            qDebug() << "当前是根项：" << item->data(0);
        } else {
            qDebug() << "当前不是根项";
        }
        return item->data(index.column());
    }

    return QVariant();
}

2. 获取指定项对应的根项 #

和TreeWidgetItem的循环向上查找逻辑一致，通过不断获取当前索引的父索引，直到父索引无效，此时的索引就是根项的索引，再通过internalPointer()拿到对应的TreeItem。

示例代码：

// 获取根项的QModelIndex
QModelIndex TreeModel::getRootIndex(const QModelIndex &index) const
{
    QModelIndex currentIndex = index;
    // 循环向上找父索引，直到父索引无效
    while (currentIndex.parent().isValid()) {
        currentIndex = currentIndex.parent();
    }
    return currentIndex;
}

// 获取根项对应的TreeItem
TreeItem* TreeModel::getRootItem(const QModelIndex &index) const
{
    if (!index.isValid())
        return nullptr;
    QModelIndex rootIndex = getRootIndex(index);
    return static_cast<TreeItem*>(rootIndex.internalPointer());
}

// 调用示例
// 假设你有目标项的index（比如B11的index）
TreeItem* rootItem = getRootItem(targetIndex);
if (rootItem) {
    qDebug() << "根项文本：" << rootItem->data(0);
}

逻辑对应说明 #

• 之前TreeWidgetItem::parent()返回nullptr → 对应QModelIndex::parent().isValid()返回false
• 循环向上查找父项的逻辑完全一致，只是操作对象从TreeWidgetItem变成了QModelIndex

内容的提问来源于stack exchange，提问作者BadRobot