QTreeView树形控件查找搜索

【实例简介】

1.自定义树形model

TreeModel::TreeModel(QObject *parent) :
    QAbstractItemModel(parent)
{
    initTreeData();
    m_pRoot = new TreeItem("根节点", 0);
    setupModelData();
}

TreeModel::~TreeModel()
{
    delete m_pRoot;
}

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());
        return item->name;
    }
    else if(role == Qt::DecorationRole)
    {
        return QVariant();
    }

    return QVariant();
}

Qt::ItemFlags TreeModel::flags(const QModelIndex &index) const
{
    if(!index.isValid())
        return 0;

    if(index.column() == 0)
        return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsUserCheckable;

    return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
}

QVariant TreeModel::headerData(int section, Qt::Orientation orientation, int role) const
{
    Q_UNUSED(section)

    if(orientation == Qt::Horizontal  && role == Qt::DisplayRole)
        return m_pRoot->name;

    return QVariant();
}

QModelIndex TreeModel::index(int row, int column, const QModelIndex &parent) const
{
    if(!hasIndex(row, column, parent))
        return QModelIndex();

    if(!parent.isValid())//ROOT下挂装置NodeMac
    {
        TreeItem *pNodeMac = m_pRoot->getChild(row);
        QModelIndex modelIndex = createIndex(row, column, pNodeMac);

        return modelIndex;
    }

    TreeItem *pParent = static_cast<TreeItem*>(parent.internalPointer());
    TreeItem *pChild = pParent->getChild(row);
    return createIndex(row,column,pChild);

    return QModelIndex();
}

QModelIndex TreeModel::parent(const QModelIndex &index) const
{
    if(!index.isValid())
    {
        return QModelIndex();
    }

    TreeItem *childItem = static_cast<TreeItem*>(index.internalPointer());
    TreeItem *parentItem = childItem->getParent();

    if(m_pRoot == parentItem)
    {
        return QModelIndex();
    }

    return createIndex(0, 0, parentItem);

    return QModelIndex();
}

int TreeModel::rowCount(const QModelIndex &parent) const
{
    TreeItem *parentItem;

    if(parent.column() > 0)
    {
        return 0;
    }

    if(!parent.isValid())
    {
        return m_pRoot->getChildCount();
    }

    parentItem = static_cast<TreeItem*>(parent.internalPointer());
    return parentItem->getChildCount();

    return 0;
}

int TreeModel::columnCount(const QModelIndex &parent) const
{
    Q_UNUSED(parent)
    return 1;
}

void TreeModel::setupModelData()
{
    QHash<QString,TreeItemData>::iterator it;

    for(it = m_treeDataList.begin(); it != m_treeDataList.end(); it  )
    {
        TreeItem *pParent = new TreeItem(it.key(), m_pRoot);
        QHash<QString,TreeItemData>::iterator it2;

        for(it2 = it.value().childList.begin(); it2 != it.value().childList.end(); it2  )
        {
            TreeItem *pChild = new TreeItem(it2.key(), pParent);
            pParent->childList.insert(it2.key(), pChild);
            QHash<QString,TreeItemData>::iterator it3;

            for(it3 = it2.value().childList.begin(); it3 != it2.value().childList.end(); it3  )
            {
                TreeItem *pChild3 = new TreeItem(it3.key(), pChild);
                pChild->childList.insert(it3.key(), pChild3);
            }
        }

        m_pRoot->childList.insert(it.key(), pParent);
    }
}

void TreeModel::initTreeData()
{
    for(int i = 0; i < 10; i  )
    {
        TreeItemData child1;
        child1.name = tr("一级节点%1").arg(i 1);

        for(int j = 0; j < 10; j  )
        {
            TreeItemData child2;
            child2.name = tr("二级节点%1%2").arg(i 1).arg(j 1);

            for(int k = 0; k < 10; k  )
            {
                TreeItemData child3;
                child3.name = tr("三级节点%1%2%3").arg(i 1).arg(j 1).arg(k 1);
                child2.childList.insert(child3.name, child3);
            }

            child1.childList.insert(child2.name, child2);
        }

        m_treeDataList.insert(child1.name, child1);
    }
}