Qt QTreeWidget 树形结构实现

Qt QTreeWidget 树形结构实现

转自

http://www.cnblogs.com/Romi/archive/2012/04/16/2452709.html

Qt中实现树形结构可以使用QTreeWidget类，也可以使用QTreeView类，QTreeWidget继承自QTreeView类。树形效果如下图所示：

这是怎么实现的呢？还有点击节点时会有相应的事件响应。

1. 树形结构实现

QT GUI中有treeWidget部件，将该控件在Gui中布局好，假设其对象名为treeWidget。

QTreeWidget类官方文档：http://qt-project.org/doc/qt-4.8/qtreewidget.html

树形结构通过QTreeWidget类和QTreeWidgetItem类实现，QTreeWidgetItem类实现结点的添加。上图代码实现如下：

 

ui->treeWidget->setColumnCount(1); //设置列数
ui->treeWidget->setHeaderLabel(tr("图像选择")); //设置头的标题

QTreeWidgetItem *imageItem1 = new QTreeWidgetItem(ui->treeWidget,QStringList(QString("图像1")));
imageItem1->setIcon(0,QIcon("xxx.png"));
QTreeWidgetItem *imageItem1_1 = new QTreeWidgetItem(imageItem1,QStringList(QString("Band1"))); //子节点1
imageItem1->addChild(imageItem1_1); //添加子节点

QTreeWidgetItem *imageItem2 = new QTreeWidgetItem(ui->treeWidget,QStringList(QString("图像2")));
QTreeWidgetItem *imageItem2_1 = new QTreeWidgetItem(imageItem2,QStringList(QString("Band1"))); //子节点1
QTreeWidgetItem *imageItem2_2 = new QTreeWidgetItem(imageItem2,QStringList(QString("Band2"))); //子节点2
imageItem2->addChild(imageItem2_1);  //添加子节点
imageItem2->addChild(imageItem2_2);

ui->treeWidget->expandAll(); //结点全部展开

 

当然，还有其他的一些方法用于设置，具体需要时查查帮助文档学习。

除了使用上面这种方法之外，还可以使用QList & items实现结点的添加。QT中对C++中的STL库中的容器使用进行了封装，使用其封装的类可以很方便的解决很多类似很复杂数据结构的问题。实现如下：

//只写结点的实现
QList rootList;

QTreeWidgetItem *imageItem1 = new QTreeWidgetItem;   //添加第一个父节点
imageItem1->setText(0,tr("图像1"));
rootList.append(imageItem1);

QTreeWidgetItem *imageItem1_1 = new QTreeWidgetItem(imageItem1,QStringList(QString("Band1"))); //添加子节点
imageItem1->addChild(imageItem1_1);

QTreeWidgetItem *imageItem2 = new QTreeWidgetItem;   //添加第二个父节点
imageItem2->setText(0,tr("图像2"));
rootList.append(imageItem2);

QTreeWidgetItem *imageItem2_1 = new QTreeWidgetItem(imageItem2,QStringList(QString("Band1")));  //添加子节点
QTreeWidgetItem *imageItem2_2 = new QTreeWidgetItem(imageItem2,QStringList(QString("Band2")));
imageItem2->addChild(imageItem2_1);
imageItem2->addChild(imageItem2_2);

ui->treeWidget->insertTopLevelItems(0,rootList);  //将结点插入部件中

ui->treeWidget->expandAll(); //全部展开

2. 点击节点的事件响应

首先想到有没有点击某个节点的信号，查看文档，有一个void itemClicked ( QTreeWidgetItem * item, int column )信号，是双击某个节点的信号，将该信号与某个自定义槽相连，当双击节点时触发槽函数。

看一下这个信号，第一个参数为点击的QTreeWidgetItem类对象，第二个参数为节点所在列号。

思路：根据点击的QTreeWidgetItem类对象可以通过parent()函数得到父节点，如果QTreeWidgetItem类对象就是最最顶端的节点时，parent()函数返回的就是NULL。通过insertChildren ( int index, const QList & children )函数可以得到该节点在父节点中的索引值。

目前只能解决只有一个最顶端父节点时的事件响应，当最顶端的父节点有多个(比如本文开头有2个)，这时点击子节点时，无法判断子节点的父节点是哪一个(本人愚笨啊！)，因此没法为其槽函数执行相应的操作。

这里就以一个分支为例。

 1 private slots:
 2     void showSelectedImage(QTreeWidgetItem * item, int column); //点击树节点事件
 3 
 4 connect(ui->treeWidget,SIGNAL(itemDoubleClicked(QTreeWidgetItem*,int)),this,SLOT(showSelectedImage(QTreeWidgetItem*,int)));
 5 
 6 void MainWindow::showSelectedImage(QTreeWidgetItem *item, int column)
 7 {
 8     QTreeWidgetItem *parent = item->parent();
 9     if(NULL==parent) //注意：最顶端项是没有父节点的，双击这些项时注意(陷阱)
10         return;
11     int col = parent->indexOfChild(item); //item在父项中的节点行号(从0开始)
12 
13     if(0==col) //Band1
14     {
15         //执行对应操作
16     }
17     if(1==col) //Band2
18     {
19         //执行对应操作
20     }
21 }

如果不加父节点是否为空的判断，当节点有父节点时，不会出错，当节点没有父节点时，程序会出错(运行错误)，判断之后，双击没有父节点的节点就会是树的收缩操作。

QTreeWidget是实现树形结构的类，在很多软件中都可以看到类似树形结构的界面。

我做的一个示例如下图，用来处理图像，最顶层节点是图像的路径名，子节点是图像的各个波段，双击各个波段会显示图像各波段的灰度图像，同时还有删除指定节点(父节点和子节点同时删除)的功能。效果如下所示

要完成这样的功能需要注意一下几点：

①.在内存中保存各个节点，当然要在堆上分配内存，删除节点时，除了去除QtreeWidget控件上的节点外，还要讲存储在内存中的节点也要删除，否则会出现内存泄露的问题。

②.节点双击的事件响应，准确定位到是哪个图像的哪个波段。

下面详细叙述。

1.变量

需要有个变量记录图像的路径名，这里定义一个容器，数据类型为QString

QVector imgFile;

2.定义信号和槽

需要三个槽，

打开菜单：每使用打开菜单打开一幅图像就将该图像的路径名和波段数设计成父节点和子节点添加到QTreeWidget控件中。

删除节点的按钮：删除指定节点（该节点处于高亮状态，即选中状态）

双击某波段：显示该波段的灰度图像。

这里着重讲解与QTreeWidget相关的，因此有关显示图像的内容概不论述。

信号与槽链接如下：

打开菜单      connect(ui.Open,SIGNAL(triggered()),this,SLOT(menu_Open()));

双击某波段   connect(ui.treeWidget,SIGNAL(itemDoubleClicked(QTreeWidgetItem*,int)),this,SLOT(showSelectedImg(QTreeWidgetItem*,int)));

删除节点      connect(ui.DeleteNode,SIGNAL(clicked()),this,SLOT(btn_DeleteNode()));

3.打开图像(添加节点)

    imgFile.append(fileName);//影像路径添加进容器
    QTreeWidgetItem *item=new QTreeWidgetItem(ui.treeWidget,QStringList(QString(fileName)));//添加节点
    //添加子节点
    for (int i=0;i)
    {
        QTreeWidgetItem *item1=new QTreeWidgetItem(item,QStringList(QString("Band")+QString::number(i+1)));
        item->addChild(item1);
    }

其中rasterNum为路径名为fileName的图像的波段数。

这样在QTreewidget部件对象treeWidget中就增加了树节点及其子节点。每次打开都会在树形节点最后面添加。

这里需要提醒一点：这里的QTreewidgetItem指针对象指向的地址都分配在堆上，会不会造成内存泄露呢？因为函数结束后作为局部变量的指针当然是消失了，但保存QTreeWidgetItem节点的内存地址还在，有没有办法在需要的时候将其内存地址释放掉呢，答案是肯定的，将在后面删除节点时论述。

4.双击某波段显示波段图像

        QTreeWidgetItem *parent=item->parent();//获得父节点
    if(NULL==parent)
        return;
    progessBar->setValue(0);//进度条置0
    int row=parent->indexOfChild(item);//获得节点在父节点中的行号(从0开始)
    QString fileName=parent->text(0);//获得父节点的文本字符(即影像路径)
    /* QString->const char* */
    QByteArray ba=fileName.toLocal8Bit();
    const char* filePath=ba.data();

这里贴出的是找到图像路径和子节点波段的方法，至于显示出图像就是根据图像路径和波段号显示出波段响应灰度图像，该问题不在讨论范畴，略去。

5.删除节点

这里的删除节点不是删除所有节点，而是删除与该节点有关的图像的所有节点，比如，鼠标指向了最开始图中第二个父节点的任意一个子节点，则就将该父节点和所有子节点删除。

    QTreeWidgetItem* item=ui.treeWidget->currentItem();//获得当前节点
    if(NULL==item)//没有选择节点
        return;
    QTreeWidgetItem* parent=item->parent();//获得当前节点的父节点
    int index;//top节点的索引号
    if(NULL==parent)//item就是top节点
    {
        QString fileName=item->text(0);//获得top节点的文本字符(即影像路径)
        for (int i=0;i)
        {
            if (fileName==imgFile.at(i))
            {
                index=i;
                break;
            }
        }
        ui.treeWidget->takeTopLevelItem(index);//去除节点 Removes the top-level item at the given index in the tree and returns it
        imgFile.remove(index);//移除容器index处内容

        //释放掉存放节点的内存空间
        int childCount=item->childCount();//子节点数
        for (int i=0;i)
        {
            QTreeWidgetItem* childItem=item->child(0);
            delete childItem;
            childItem=NULL;
        }
        delete item;
        item=NULL;
    }
    else//parent才是top节点
    {
        QString fileName=parent->text(0);//获得top节点的文本字符(即影像路径)
        for (int i=0;i)
        {
            if (fileName==imgFile.at(i))
            {
                index=i;
                break;
            }
        }
        ui.treeWidget->takeTopLevelItem(index);//去除节点 Removes the top-level item at the given index in the tree and returns it
        imgFile.remove(index);//移除容器index处内容

        //释放掉存放节点的内存空间
        int childCount=parent->childCount();//子节点数
        for (int i=0;i)
        {
            item=parent->child(0);
            delete item;
            item=NULL;
        }
        delete parent;
        parent=NULL;
    }

主要方法就是根据图像路径名获得图像在top节点中的索引号，然后将top节点及其子节点删除，采用takeTopLevelItem(index)方法可以把treeWidget中的所因为index的节点去除掉，注意这里只是将节点从treeWidget中去除掉，其节点(父节点+子节点)仍然存在内存中，所以还要讲内存中的地址也要释放掉，否则就会出现所谓的内存泄露的问题。所采用的方法时使用QTreeWidgetItem指针找到存放节点的内存地址，然后将其delete掉，如此就释放掉内存空间了(有疑问的可以跟踪调试下)，一定注意将要删除的节点全部delete掉，先释放子节点内存，再释放父节点内存。

还需要注意的是，每清除掉一个子节点的内存空间，对应父节点就会失去该子节点，因此每次都是清理掉父节点索引为0的子节点，即获得child(0)的子节点。此外不要忘了保存图像路径的向量imgFile中也要去除对应的索引内容。

总结

以上给出的是实现功能的核心实现部分，并非完整的程序，其他未论述的都是与QTreeWidget该控件无关的了。上述方法可移植性强，可以用到有类似需求的地方。