QTreeWidget的添加删除、勾选、导入以及拖拽操作示例

由来

因为在工作中对QTreeWidget的操作基本上都是这几个操作。然后平时没做过备忘。就写了很多次重复代码。今天特此备忘一下。以便于后续的使用

功能描述

此TreeWidget通过右键菜单的方式来操作。目前支持添加、删除、勾选QTreeWidgetItem。同时可以拖拽QTreeWidgetItem用来形成新的层级关系。同时支持从另一个TreeWidget导入到本TreeWidget上面。且维持同样的导入层级关系。

添加实现思路

根据右键选择的QTreeWIdgetItem作为父节点。直接添加新的子节点。
实现代码

void FCustomTreeWidget::onAddItem()
{
	auto selectItemList = this->selectedItems();
	QTreeWidgetItem *item = new QTreeWidgetItem;//构造函数给this的话 就直接添加为了顶层节点
	item->setText(0, QStringLiteral("新建节点"));
	item->setCheckState(0, Qt::CheckState::Unchecked);
	item->setFlags(item->flags() | Qt::ItemFlag::ItemIsEditable);
	if (selectItemList.size() == 0)
	{
		this->addTopLevelItem(item);
	}
	else
	{
		auto selectItem = selectItemList.first();
		selectItem->addChild(item);
	}
}

删除实现思路

根据右键选择的QTreeWIdgetItem作为父节点。然后可以选择是否删除它的子节点。
实现代码

void FCustomTreeWidget::onRemoveItem()
{
	auto selectItemList = this->selectedItems();
	if (selectItemList.size() == 0)
	{
		qDebug() << "Select Items Is Empty";
		return;
	}
	auto selectItem = selectItemList.first();
	QMessageBox msgBox;
	msgBox.setText(QStringLiteral("是否删除此节点的子节点"));
	msgBox.setStandardButtons(QMessageBox::Yes | QMessageBox::No);
	msgBox.setDefaultButton(QMessageBox::Yes);
	int ret = msgBox.exec();
	if (ret == QMessageBox::Yes)
	{
		removeItemWithChild(selectItem);//递归删除子节点
	}
	else
	{
		removeItemNotWithChild(selectItem);//递归删除子节点
	}
}

节点可编辑

直接通过设置itemFlags可以实现
实现代码

QTreeWidgetItem *item = new QTreeWidgetItem;
	item->setText(0, QStringLiteral("新建节点"));
	item->setFlags(item->flags() | Qt::ItemFlag::ItemIsEditable);

右键弹出菜单

此实现直接调用setContextMenuPolicy(Qt::ContextMenuPolicy::CustomContextMenu);然后再实现一个槽函数。绑定customContextMenuRequested信号就行了。不过有一个注意的地方就是。菜单弹出的位置使用QCursor::pos()直接获取鼠标光标的位置。这样在单屏幕或者是多屏幕都不会存在问题。
实现代码

void FCustomTreeWidget::onCustomContextMenuRequested(const QPoint &pos)
{
	QMenu  menu(this);
	menu.addAction(_actionAddItem);
	menu.addAction(_actionAddItem);
	menu.addAction(_actionRemoveItem);
	menu.addAction(_actionSelectAll);
	menu.addAction(_actionUnselectAll);
	menu.addAction(_actionImport);
	menu.exec(QCursor::pos());
}

节点的勾选与取消勾选

此操作首先设置QTreeWidgetItem可以被勾选。然后可以勾选自身节点以以及子节点。
实现代码

void FCustomTreeWidget::setItemWithChildItemCheckState(QTreeWidgetItem* item, bool isChecked)
{
	if (item == nullptr)
	{
		return;
	}
	if (isChecked)
	{
		item->setCheckState(0, Qt::CheckState::Checked);
	}
	else
	{
		item->setCheckState(0, Qt::CheckState::Unchecked);
	}
	int childCount = item->childCount();
	for (auto iter = 0;iter<childCount;++iter)
	{
		auto childItem = item->child(iter);
		setItemWithChildItemCheckState(childItem, isChecked);
	}
}

节点的拖拽改变父子关系

首先设置QTreeWidget可以被拖拽。然后重写dropEvent处理节点放下事件。
实现代码

void FCustomTreeWidget::dropEvent(QDropEvent *event)
{
	auto selItemList = this->selectedItems();
	if (selItemList.size() <1)
	{
		return;
	}
	auto curItem = selItemList.first();
	if (curItem == nullptr)
	{
		return;
	}
	auto parentItem = curItem->parent();
	if (parentItem  == nullptr)
	{
		this->takeTopLevelItem(this->indexOfTopLevelItem(curItem));

	}
	else
	{
		parentItem->removeChild(curItem);
	}
	auto item = this->itemAt(event->pos());
	if (item  == nullptr)
	{
		this->addTopLevelItem(curItem);
	}
	else
	{
		item->addChild(curItem);
	}
}

导入另一个TreeWIdget

导入的时候主要是需要维持父子关系。例如存在A,B,C三层节点。若是只勾选了A，C节点。那么此时构建父子关系的时候C的父节点就是A了。
实现代码

void FImportDialog::onOk(bool val)
{
	//根据勾选的层级构建导入关系
	auto topItemCount = _treeWidget->topLevelItemCount();
	_importItemList.clear();
	for (int index = 0;index<topItemCount;++index)
	{
		auto item = _treeWidget->topLevelItem(index);
		getItemChildItem(item,nullptr, _importItemList);
	}
	accept();
}

关于QTreeWidget的勾选

本例中只显示了勾选和未勾选等。关于半勾选状态可以参见博文。
https://blog.csdn.net/weixin_39308337/article/details/106353102

代码链接

因为很多人上GitHub的网速都很慢。所以就将代码托管在了Gitee上面。
注意
代码不是一个完整的工程。只有.h文件以及.cpp文件。文件名是FCustomTreeWidget.h和FCustomTreeWidget.cpp
git地址
https://gitee.com/zmf199785/csdncode.git
代码在FCustomTreeWidget 目录中