C/C++ Qt TreeWidget 巢狀節點操作技巧

在上一篇博文《C/C++ Qt TreeWidget 單層樹形元件應用》中給大家演示瞭如何使用TreeWidget元件建立單層樹形結構，並給這個樹形元件增加了右鍵選單功能，接下來將繼續延申樹形元件的使用，並實現對樹形框多節點的各種操作。

常用樹形框節點間的操作方法如下:

• 節點遍歷
• 初始化節點
• 單擊雙擊節點
• 新增根節點
• 新增子節點
• 修改選中節點
• 刪除選中節點
• 列舉全部節點
• 列舉選中節點
• 獲取節點子節點

簡單的節點遍歷: 首先我們還是使用TreeView元件實現一個簡單的多層巢狀樹結構，程式碼執行後，首先迴圈設定3個外層節點，接著迴圈內層節點，並將內層中的QStandardItem追加到外層上面。

#include "mainwindow.h"
#include "ui_mainwindow.h"

#include <QTreeView>
#include <QStandardItemModel>

// By: LyShark
// https://www.cnblogs.com/lyshark
MainWindow::MainWindow(QWidget *parent) :QMainWindow(parent),ui(new Ui::MainWindow)
{
    ui->setupUi(this);

    QStandardItemModel *tree = new QStandardItemModel(0,3,this);

    ui->treeView->setColumnWidth(0,50);      // 設定第1列長度
    ui->treeView->setColumnWidth(1,200);     // 設定第2列長度
    ui->treeView->setColumnWidth(2,200);     // 設定第3列長度

    tree->setHeaderData(0, Qt::Horizontal, tr("序號"));
    tree->setHeaderData(1, Qt::Horizontal, tr("姓名"));
    tree->setHeaderData(2, Qt::Horizontal, tr("年齡"));

    ui->treeView->setModel(tree);

    for (int i = 0; i < 4; ++i)
    {
        // 設定3個外層節點
        QList<QStandardItem *> items;
        for (int i = 0; i < 3; ++i)
        {
            QStandardItem *item = new QStandardItem(QString("%0").arg(i));

            if (0 == i)
                item->setCheckable(true);

            items.push_back(item);
        }
        tree->appendRow(items);

        // 設定內層
        for (int i = 0; i < 2; ++i)
        {
            QList<QStandardItem *> childItems;
            for (int i = 0; i < 3; ++i)
            {
             QStandardItem *item = new QStandardItem(QString("lyshark"));
             if (0 == i)
                 item->setCheckable(true);
             childItems.push_back(item);
            }
            items.at(0)->appendRow(childItems);
        }
    }
}

MainWindow::~MainWindow()
{
    delete ui;
}

程式碼執行效果如下:

初始化樹形節點: 首先在開始操作元素之前,我們可以在MainWindow::MainWindow中對樹形節點進行簡單的初始化,插入幾個測試節點.

#include "mainwindow.h"
#include "ui_mainwindow.h"

#include <iostream>
#include <QTreeWidgetItem>
#include <QString>

// By: LyShark
// https://www.cnblogs.com/lyshark
MainWindow::MainWindow(QWidget *parent): QMainWindow(parent), ui(new Ui::MainWindow)
{
    ui->setupUi(this);
    ui->treeWidget->clear();

    // 設定QTreeWidget的列數
    ui->treeWidget->setColumnCount(1);
    // 設定QTreeWidget標題隱藏
    ui->treeWidget->setHeaderHidden(true);

    // 建立QTreeWidget的朋友節點,父節點是tree
    QTreeWidgetItem *Friend = new QTreeWidgetItem(ui->treeWidget,QStringList(QString("朋友")));
    Friend->setIcon(0,QIcon(":/image/4.ico"));  // 新增一個圖示
    Friend->setFlags(Qt::ItemIsSelectable | Qt::ItemIsUserCheckable
                     | Qt::ItemIsEnabled | Qt::ItemIsAutoTristate);
    Friend->setCheckState(0,Qt::Checked);

    // 給Friend新增一個子節點frd
    QTreeWidgetItem *frd = new QTreeWidgetItem(Friend);
    frd->setText(0,"www.lyshark.com");
    frd->setIcon(0,QIcon(tr(":/image/1.ico")));
    frd->setCheckState(0,Qt::Checked);               // 預設選中狀態

    QTreeWidgetItem *frs = new QTreeWidgetItem(Friend);
    frs->setText(0,"cdn.lyshark.com");
    frs->setIcon(0,QIcon(tr(":/image/1.ico")));
    frs->setCheckState(0,Qt::Unchecked);            // 預設未選中

    // ----------------------------------------------------------
    // 建立名叫同學節點,父節點同樣是tree
    QTreeWidgetItem * ClassMate = new QTreeWidgetItem(ui->treeWidget,QStringList(QString("同學")));
    ClassMate->setIcon(0,QIcon(":/image/5.ico"));  // 新增一個圖示
    ClassMate->setCheckState(0,Qt::Checked);       // 預設選中

    //Fly是ClassMate的子節點
    QTreeWidgetItem *Fly = new QTreeWidgetItem(QStringList(QString("nas.lyshark.com")));
    Fly->setIcon(0,QIcon(tr(":/image/2.ico")));
    //建立子節點的另一種方法
    ClassMate->addChild(Fly);
    Fly->setCheckState(0,Qt::Checked);       // 設定為選中

    QTreeWidgetItem *Fls = new QTreeWidgetItem(QStringList(QString("lyshark.cnblogs.com")));
    Fls->setIcon(0,QIcon(tr(":/image/2.ico")));
    ClassMate->addChild(Fls);
    Fls->setCheckState(0,Qt::Checked);       // 設定為選中

    // ----------------------------------------------------------
    // 陌生人單獨一欄
    QTreeWidgetItem  *Strange = new QTreeWidgetItem(true);
    Strange->setText(0,"陌生人");
    Strange->setIcon(0,QIcon(":/image/6.ico"));  // 新增一個圖示

    ui->treeWidget->addTopLevelItem(ClassMate);
    ui->treeWidget->addTopLevelItem(Strange);

    // 增加文字到編輯框
    ui->plainTextEdit->appendPlainText("hello lyshark");

    //展開QTreeWidget的所有節點
    //ui->treeWidget->expandAll();
    //ui->treeWidget->resize(271,401);
}

MainWindow::~MainWindow()
{
    delete ui;
}

程式碼執行效果如下:

單擊雙擊節點反饋: 當我們將滑鼠停靠在指定節點內並點選時,我們需要觸發treeWidget_itemDoubleClicked屬性讓其反饋該行標題等基本屬性.

// 當我們雙擊指定的成員時獲取到該成員的名字
void MainWindow::on_treeWidget_itemDoubleClicked(QTreeWidgetItem *item, int column)
{
    QString str = item->text(column);
    std::cout << str.toStdString().data() << std::endl;
    ui->plainTextEdit->appendPlainText(str.toStdString().data());
}

// 當我們單擊指定成員時獲取資料
void MainWindow::on_treeWidget_itemClicked(QTreeWidgetItem *item, int column)
{
    QString str = item->text(column);
    std::cout << str.toStdString().data() << std::endl;
    ui->plainTextEdit->appendPlainText(str.toStdString().data());
}

程式碼執行效果如下:

新增 父節點/子節點: 通過程式碼的方式當點選on_pushButton_clicked時分別實現增加一個父節點和一個子節點的功能。

// 單擊按鈕新增新的父節點
void MainWindow::on_pushButton_clicked()
{
    QString NodeText = "新的父節點";
    QTreeWidgetItem  *item = new QTreeWidgetItem(true);
    item->setText(0,NodeText);
    item->setIcon(0,QIcon(":/image/7.ico"));
    ui->treeWidget->addTopLevelItem(item);
}

// 單擊按鈕新增子節點
void MainWindow::on_pushButton_4_clicked()
{
    QTreeWidgetItem * item= ui->treeWidget->currentItem();
        if(item!=NULL)
            AddTreeNode(item,"新子節點","新子節點");
        else
            AddTreeRoot("新子節點","新子節點");
}

程式碼執行效果如下:

刪除選中節點: 首先選中要刪除的指定節點，然後可以對該節點進行刪除操作，刪除子節點直接移除即可，刪除父節點需要連同內部子節點一併刪掉。

// 刪除選中的節點
void MainWindow::on_pushButton_3_clicked()
{
    QTreeWidgetItem *currentItem = ui->treeWidget->currentItem();
    if(currentItem == NULL)
        return;

    // 如果沒有父節點則直接刪除
    if(currentItem->parent() == NULL)
    {
        delete ui->treeWidget->takeTopLevelItem(ui->treeWidget->currentIndex().row());
        std::cout << ui->treeWidget->currentIndex().row() << std::endl;
    }
    else
    {
        // 如果有父節點就要用父節點的takeChild刪除節點
        delete currentItem->parent()->takeChild(ui->treeWidget->currentIndex().row());
    }
}

程式碼執行效果如下:

修改指定節點名稱: 單擊後將指定節點修改為Modify並將圖示設定為新的

// 修改節點
// By: LyShark
// https://www.cnblogs.com/lyshark
void MainWindow::on_pushButton_2_clicked()
{
    // 得到當前節點
    QTreeWidgetItem *currentItem = ui->treeWidget->currentItem();
    if(currentItem == NULL)
        return;
    // 修改選中項
    for(int x=0;x<currentItem->columnCount();x++)
    {
        currentItem->setText(x,tr("Modify") + QString::number(x));
        currentItem->setIcon(x,QIcon(":/image/1.ico"));
    }
}

程式碼執行效果如下:

列舉所有節點元素: 列舉當前Tree中的所有節點元素，並將結果輸出到右側編輯框內。

// 列舉所有節點
// By: LyShark
// https://www.cnblogs.com/lyshark
// 列舉所有節點
void MainWindow::on_pushButton_5_clicked()
{
    // 獲取到全部的根節點數量
    int size = ui->treeWidget->topLevelItemCount();
    QTreeWidgetItem *child;
    for(int x=0;x<size;x++)
    {
        // 輸出所有父節點
        child = ui->treeWidget->topLevelItem(x);
        std::cout << "all root = "<< child->text(0).toStdString().data() << std::endl;
        ui->plainTextEdit->appendPlainText(child->text(0).toStdString().data());

        // 得到所有子節點計數
        int childCount = child->childCount();
        // std::cout << "all child count = " << childCount << std::endl;

        // 輸出根節點下面的子節點
        for(int y=0;y<childCount;++y)
        {
            QTreeWidgetItem *grandson = child->child(y);
            std::cout << "--> sub child = "<< grandson->text(0).toStdString().data() << std::endl;
            ui->plainTextEdit->appendPlainText(grandson->text(0).toStdString().data());
        }
    }
}

程式碼執行效果如下:

列舉選中節點元素: 列舉當前Tree中選中節點的元素，並將結果輸出到右側編輯框內。

// 列舉所有的 【選中】節點
// https://www.cnblogs.com/lyshark
void MainWindow::on_pushButton_7_clicked()
{
    // 獲取到全部的根節點數量
    int size = ui->treeWidget->topLevelItemCount();
    QTreeWidgetItem *child;
    for(int x=0;x<size;x++)
    {
        // 輸出所有父節點
        child = ui->treeWidget->topLevelItem(x);

        // 得到所有子節點計數
        int childCount = child->childCount();

        // 輸出根節點下面的子節點
        for(int y=0;y<childCount;++y)
        {
            QTreeWidgetItem *grandson = child->child(y);
            // 判斷是否選中,如果選中輸出父節點與子節點
            if(Qt::Checked == grandson->checkState(0))
            {
                std::cout << "root -> " << child->text(0).toStdString().data()
                          << "--> sub child = "<< grandson->text(0).toStdString().data() << std::endl;

                ui->plainTextEdit->appendPlainText(grandson->text(0).toStdString().data());
            }
        }
    }
}

程式碼執行效果如下:

獲取選中子節點的父節點: 獲取子節點的父節點ID,然後根據ID得到子節點名字。

void MainWindow::on_pushButton_6_clicked()
{
    // 取所有的父節點
    QTreeWidgetItem *currentItem = ui->treeWidget->currentItem()->parent();
    int root_count = ui->treeWidget->indexOfTopLevelItem(currentItem);
    std::cout << "root Count = " <<  root_count << std::endl;
    if(root_count != -1)
    {
        // 指定序號對應的父節點名字
        QTreeWidgetItem *child;

        child = ui->treeWidget->topLevelItem(root_count);
        std::cout << "root name= "<< child->text(0).toStdString().data() << std::endl;
        
        ui->plainTextEdit->appendPlainText(child->text(0).toStdString().data());
    }
}

程式碼執行效果如下:

補充一下節點插入函式的定義,AddTreeRoot/AddTreeNode兩個函式定義如下所示.

// mainwindow.h 中增加頭部宣告
    QTreeWidgetItem * AddTreeRoot(QString name,QString desc);
    QTreeWidgetItem * AddTreeNode(QTreeWidgetItem *parent,QString name,QString desc);

// mainwindow.cpp 中增加實現部分
QTreeWidgetItem * MainWindow::AddTreeRoot(QString name,QString desc)
{
    QTreeWidgetItem * item=new QTreeWidgetItem(QStringList()<<name<<desc);
    ui->treeWidget->addTopLevelItem(item);
    return item;
}
QTreeWidgetItem * MainWindow::AddTreeNode(QTreeWidgetItem *parent,QString name,QString desc)
{
    QTreeWidgetItem * item=new QTreeWidgetItem(QStringList()<<name<<desc);
    parent->addChild(item);
    return item;
}
``