如何扩展QFileSystemModel添加IsSelectedRole并实现data/setData？

我来帮你搞定这个问题——扩展QFileSystemModel添加自定义选中状态角色其实不用折腾复杂的布尔树，用QHash来存选中状态就够简单高效了，下面一步步给你讲怎么实现：

一、核心思路：用文件路径映射选中状态 #

QFileSystemModel里每个项都对应唯一的文件路径，所以我们完全可以用QHash<QString, bool>来存储路径和选中状态的映射，不用自己构建和文件系统对应的树结构，既省事儿又不容易出错。

二、派生类的具体实现 #

1. 头文件定义 #

首先创建派生类，定义自定义角色，加上存储选中状态的成员变量：

#include <QFileSystemModel>
#include <QHash>

class CustomFileSystemModel : public QFileSystemModel
{
    Q_OBJECT
public:
    explicit CustomFileSystemModel(QObject *parent = nullptr) : QFileSystemModel(parent) {}

    // 定义自定义角色
    static const int IsSelectedRole = Qt::UserRole + 1;

    // 重写data和setData方法
    QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
    bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole) override;

private:
    // 存储路径到选中状态的映射表
    QHash<QString, bool> m_selectedStates;
};

2. 实现data()函数 #

当视图请求IsSelectedRole的数据时，我们从hash表中取出对应路径的值，默认返回false（未选中）；其他角色的请求直接交给父类处理：

QVariant CustomFileSystemModel::data(const QModelIndex &index, int role) const
{
    if (role == IsSelectedRole) {
        QString filePath = this->filePath(index);
        return m_selectedStates.value(filePath, false);
    }
    // 其他角色的逻辑交给父类QFileSystemModel处理
    return QFileSystemModel::data(index, role);
}

3. 实现setData()函数 #

当视图设置IsSelectedRole的值时，我们把路径和布尔值存入hash表，然后发送dataChanged信号通知视图更新；其他情况同样交给父类处理：

bool CustomFileSystemModel::setData(const QModelIndex &index, const QVariant &value, int role)
{
    if (role == IsSelectedRole) {
        QString filePath = this->filePath(index);
        bool isSelected = value.toBool();
        m_selectedStates[filePath] = isSelected;
        // 仅通知该索引的IsSelectedRole数据更新，提升效率
        emit dataChanged(index, index, {IsSelectedRole});
        return true;
    }
    return QFileSystemModel::setData(index, value, role);
}

三、QML中的使用示例 #

在QML的TreeView里，你可以直接绑定这个自定义角色来控制选中状态，比如用CheckBox来切换：

TreeView {
    id: fileTree
    model: customFileSystemModel // 这里是你实例化的CustomFileSystemModel对象

    delegate: Item {
        CheckBox {
            checked: model.IsSelectedRole
            onCheckedChanged: {
                model.setData(index, checked, CustomFileSystemModel.IsSelectedRole)
            }
        }
        // 其他显示文件名、图标等的组件...
        Text {
            text: model.fileName
            anchors.left: checkBox.right
            anchors.verticalCenter: checkBox.verticalCenter
        }
    }
}

四、额外注意事项 #

• 如果模型会频繁刷新（比如目录内容变化），可以根据需求决定是否保留之前的选中状态：比如监听模型的rowsRemoved信号，移除对应路径的hash条目；或者保留，这样文件重新出现时选中状态还能恢复。
• 如果需要持久化选中状态，可以把m_selectedStates的内容保存到配置文件（比如用QSettings），程序启动时再加载回来。

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