QTreeWidget树的三态操作

在工作中我们会经常使用到树状结构，例如Windows的资源管理器中的目录结构就用到了类似的控件。在Qt中可以使用QTreeWidget来实现类似的效果。

Qt中提供的QTreeWidget功能相对比较少，在实际中经常需要扩展它的功能，比如本文将要谈到的三态树的问题。

首先我们先创建一个QTreeWidget的对象，并添加一些节点：

void WidgetTreeTEST::createDirectoryStruct()
{
QTreeWidgetItem *topLevelDirectory = new QTreeWidgetItem();
topLevelDirectory->setText(0, "DIR");
topLevelDirectory->setCheckState(0, Qt::Checked);
topLevelDirectory->setIcon(0, QIcon(":/Resources/folder.png"));

//Direcoty C
QTreeWidgetItem *directoryC = new QTreeWidgetItem();
directoryC->setText(0, "LocalDrive C");
directoryC->setCheckState(0, Qt::Checked);
directoryC->setIcon(0, QIcon(":/Resources/folder.png"));

QTreeWidgetItem *file1 = new QTreeWidgetItem();
file1->setText(0, "file1");
file1->setCheckState(0, Qt::Checked);
file1->setIcon(0, QIcon(":/Resources/file.png"));

QTreeWidgetItem *file2 = new QTreeWidgetItem();
file2->setText(0, "file2");
file2->setCheckState(0, Qt::Checked);
file2->setIcon(0, QIcon(":/Resources/file.png"));

directoryC->addChild(file1);
directoryC->addChild(file2);

//Directory D
QTreeWidgetItem *directoryD = new QTreeWidgetItem();
directoryD->setText(0, "LocalDrive D");
directoryD->setCheckState(0, Qt::Checked);
directoryD->setIcon(0, QIcon(":/Resources/folder.png"));

QTreeWidgetItem *file3 = new QTreeWidgetItem();
file3->setText(0, "file3");
file3->setCheckState(0, Qt::Checked);
file3->setIcon(0, QIcon(":/Resources/file.png"));

directoryD->addChild(file3);

//Direcoty E
QTreeWidgetItem *directoryE = new QTreeWidgetItem();
directoryE->setText(0, "LocalDrive E");
directoryE->setCheckState(0, Qt::Checked);
directoryE->setIcon(0, QIcon(":/Resources/folder.png"));

QTreeWidgetItem *file4 = new QTreeWidgetItem();
file4->setText(0, "file4");
file4->setCheckState(0, Qt::Checked);
file4->setIcon(0, QIcon(":/Resources/file.png"));

QTreeWidgetItem *file5 = new QTreeWidgetItem();
file5->setText(0, "file5");
file5->setCheckState(0, Qt::Checked);
file5->setIcon(0, QIcon(":/Resources/file.png"));

QTreeWidgetItem *direcotryE1 = new QTreeWidgetItem();
direcotryE1->setText(0, "file6");
direcotryE1->setCheckState(0, Qt::Checked);
direcotryE1->setIcon(0, QIcon(":/Resources/folder.png"));

QTreeWidgetItem *file6 = new QTreeWidgetItem();
file6->setText(0, "file6");
file6->setCheckState(0, Qt::Checked);
file6->setIcon(0, QIcon(":/Resources/file.png"));
direcotryE1->addChild(file6);

directoryE->addChild(file4);
directoryE->addChild(file5);
directoryE->addChild(direcotryE1);

QList<QTreeWidgetItem*> topLevelItemList;
topLevelItemList << directoryC << directoryD << directoryE;
topLevelDirectory->addChildren(topLevelItemList);
addTopLevelItem(topLevelDirectory);
}创建的场景结构如下图所示：



在QTreeWidget中树结构的每一项都是一个QTreeWidgetItem，通过对它的设置可以修改树的显示效果，对于树的三态切换需要处理QTreeWidget中的一个信号：



当某一个节点被选中或者取消选中的时候需要处理以下情况：

1. 处理该节点的子节点（如果它有子节点），它的子节点的状态（Check或者Uncheck）和它一样 （如果它的子节点中有目录，那么还需要递归处理子节点的子节点）

2. 处理该节点的父节点，父节点会根据当前它子节点的状态来调整自身的状态（如果该父节点还有父节点，那么还需要递归处理父节点的父节点）

在QTreeWidget中的itemChanged事件会一直递归的调用，也就是说如果我们设置了子节点的状态（使用程序设置，或者手动点击），那么被设置的节点会继续调用itemChanged信号，根据这个特点，我们在编写代码的过程中不需要考虑递归的问题，只需要设置一个层级的处理即可。

具体实现如下：

设置子节点和父节点的状态

void WidgetTreeTEST::setChildCheckState(QTreeWidgetItem *item, Qt::CheckState cs)
{
if(!item) return;
for (int i=0;i<item->childCount();i++)
{
QTreeWidgetItem* child=item->child(i);
if(child->checkState(0)!=cs)
{
child->setCheckState(0, cs);
}
}
setParentCheckState(item->parent());
}

void WidgetTreeTEST::setParentCheckState(QTreeWidgetItem *item)
{
if(!item) return;
int selectedCount=0;
int childCount = item->childCount();
for (int i=0;i<childCount;i++)
{
QTreeWidgetItem* child= item->child(i);
if(child->checkState(0)==Qt::Checked)
{
selectedCount++;
}
}

if(selectedCount == 0) {
item->setCheckState(0,Qt::Unchecked);
} else if (selectedCount == childCount) {
item->setCheckState(0,Qt::Checked);
} else {
item->setCheckState(0,Qt::PartiallyChecked);
}
}在信号的响应槽函数中：
void WidgetTreeTEST::itemChangedSlot(QTreeWidgetItem* item, int column)
{
if(Qt::PartiallyChecked!=item->checkState(0))
setChildCheckState(item,item->checkState(0));

if(Qt::PartiallyChecked==item->checkState(0))
if(!isTopItem(item))
item->parent()->setCheckState(0,Qt::PartiallyChecked);
}

bool WidgetTreeTEST::isTopItem(QTreeWidgetItem* item)
{
if(!item) return false;
if(!item->parent()) return true;
return false;
}槽函数中根据PartiallyChecked进行分类：因为只有组节点（包含子节点的节点）才可能有partiallyChecked的状态，当组节点被设置为partiallyChecked的时候它的父节点也会被设置为这种类型的状态，并且当组节点设置为PartiallyChecked的时候对子节点的状态没有任何影响。
最后结果如下图所示：