一、开心一刻

一程序员第一次上女朋友家她妈板着脸问 ：你想娶我女儿，有多少存款？

程序员低了下头：五百！

她妈更鄙视了：才五百块，买个厕所都不够！

程序员忙说：不是人民币！

她妈：就算是美元，还是不够买厕所！

程序员：其实是比特币！

她妈：哇，贤婿，我给你买只大龙虾去

二、自绘树节点？

自绘树节点？听起来都挺复杂的，可是为什么还要自绘树节点呢？这充分说明产品的脑子是什么东西都能想出来的。

有一天产品说我们的软件里缺少一个美丽的树控件，然后就要求开发去实现这个功能。

对于有一定开发经验的同学可能直接会去百度，或者上Qt帮助文档上查找资料，然后发现直接设置qss就能达到我们需要的效果，于是一顿操作后，发现效果还是不错滴。

setStyleSheet(""
	"QTreeView {outline:none;show-decoration-selected: 1;}"
	"QTreeView {outline:none;border:0px;}"
	"QTreeView::branch{		background-color: transparent;	}"
	"QTreeView::item:hover, QTreeView::branch:hover { background-color: transparent;border-color: rgb(255, 0, 0);}"
	"QTreeView::item:selected, QTreeView::branch:selected { background-color: #C5E0F7;}"
	"QTreeView::branch:open:has-children{image: url(:/branch-expand.png);}"
	"QTreeView::branch:closed:has-children{image: url(:/branch-collapse.png);}"
	"QTreeView::branch:has-siblings:!adjoins-item{border-image:url(:/branch-line.png) 0;}"
	"QTreeView::branch:has-siblings:adjoins-item{border-image:url(:/branch-more.png) 0;}"
	"QTreeView::branch:!has-children:!has-siblings:adjoins-item{border-image:url(:/branch-end.png) 0;	}"
	"QTreeView::branch:has-children:!has-siblings:closed,QTreeView::branch:closed:has-children:has-siblings{border-image:none;image: url(:/branch-collapse.png);	}"
	"QTreeView::branch:open:has-children:!has-siblings,QTreeView::branch:open:has-children:has-siblings{border-image:none;image: url(:/branch-expand.png);	}"
	);

遂找来产品验证，当产品看到这个效果后，脸直接都绿了。

产品：我不是说要一个树形控件吗？行高需要能动态调整那种！

开发：。。。

开发：行高调整了，那branch上贴的图拉伸后不是模糊了么。。。

产品：。。。

产品：我不管，这个行高可拖拽功能很重要，怎么实现我不管，但是功能必须要有。

开发：卧槽，看来只有出终极大法了，直接自绘吧

三、效果展示

如下图所示，是一个简单的树branch自绘效果。

此处主要是展示一个demo效果，如果需要美化需要专业设计师出图来做。

四、实现思路

既然要自己绘制树形节点，那必然要去研究Qt的源码。

1、可扩展接口

首先我们打开QTreeView类的帮助文档，查找这个类都有哪些可供重写的接口，然后就发现了这么几个函数

看名字大概都知道是什么意思，不过这里还是做简要说明

前边提到我们要自己绘制branch线条，但是其余的东西还是要走Qt默认的绘制风格，因此在重写绘制函数时，千万不要忘记了调用原有的绘制方法。

表格中前3个函数就是绘制树控件的具体方法，这3个函数搭配起来完成了树控件内容格子的绘制。下面我们来重写这3个函数，分别完成我们的需求

2、函数重写

a、绘制行drawRow

drawRow顾名思义就是绘制一行的意思，这里也确实如此。为什么要重写这个函数呢？答案也很简单。

树控件本身是不具有垂直分割线的，既然我们要模拟表格的样式，那么垂直分割线必然是需要的。

实现代码可能像下面这样，是不是很简单。

void FrozenTreeView::drawRow(QPainter * painter, const QStyleOptionViewItem & options, const QModelIndex & index) const
{
	QTreeView::drawRow(painter, options, index);

	//绘制网格线
	QPen pen;
	pen.setWidth(m_iWidth);
	pen.setColor(m_gridLineColor);

	painter->save();
	painter->setPen(pen);
	painter->drawRect(options.rect);
	painter->restore();
}

b、绘制branch

绘制行函数主要是添加了单元格边框绘制，接下来就是第一列的branch绘制。

绘制branch时一定不要忘记调用原有的绘制函数，否则界面显示会异常。

{
	painter->save();
	QTreeView::drawBranches(painter, rect, index);
	painter->restore();
}

绘制branch时主要是根据当前节点是否展开、是否有孩子节点、是否有兄弟节点等状态来联合判断并进行绘制

如下是绘制代码，可能有些长，但是应该比较好理解。

点击领取Qt学习资料+视频教程~ 「链接」

需要注意的点

1. 除根节点外，每个节点都需要绘制文字前边的水平线
2. 有父亲的节点需要绘制垂直线。绘制的竖线是否绘制到底，取决于是否有向下的兄弟
3. 有爷爷的节点可能需要额外绘制向下的竖线。是否绘制取决于自己的父亲是否有向下的兄弟
4. 规则3其实是一个循环的处理，也就是说爷爷如果有爸爸，也就是说节点如果有祖爷爷，那么可能还需要绘制更多的向下竖线。是否绘制取决于节点的爷爷是否有向下的兄弟

代码这里就不细说了，有兴趣的可以自己研究研究。绘制规则就是上述4点

//绘制branch
{
	DataNode * node = static_cast<DataNode *>(index.internalPointer());
	bool hasChild = node->children().size() != 0;//是否有孩子

	QList<DataNode *> & children = node->parent()->children();
	bool has_next_siblings = children.indexOf(node) != (children.size() - 1);//是否有向后的兄弟
	bool has_pre_siblings = children.indexOf(node) != 0;//是否有向前的兄弟

	int level = node->level();
	int indentaion = indentation();//缩进
	int indentaions = indentaion * (level - 1);//缩进距离

	QRect r = rect;
	r.setLeft(r.left() + indentaions);//图标绘制位置

	painter->save();
	painter->setPen(m_branchLine);

	bool expaned = isExpanded(index);//节点是否展开

	QLine line(r.center() + QPoint(0, r.top() - r.center().y()), r.center() + QPoint(0, r.bottom() - r.center().y()));
	line.translate(-indentaion, 0);
	//QLine line(r.topLeft(), r.bottomLeft());
	//循环绘制(具有兄弟节点的)父节点向下的竖线
	DataNode * parent_node = node->parent();
	DataNode * sub_node = node;
	bool isNeed = node->children().size() == 0;
	for (int i = level - 1; i >= 0; --i)
	{
		QList<DataNode *> & children = parent_node->children();
		bool has_next_siblings = children.indexOf(sub_node) != (children.size() - 1);//父节点是否有(向后的)兄弟

		if (has_next_siblings)
		{
			painter->drawLine(line);
		}
		

		if (level - 1 == i)
		{
			QPoint pos = (line.p1() + line.p2()) / 2;
			QPoint pos2 = pos + QPoint(indentaion / 2, 0);

			painter->drawLine(pos, pos2);

			if (!has_next_siblings)
			{
				painter->drawLine(line.p1(), (line.p1() + line.p2()) / 2);
			}
		}

		sub_node = parent_node;
		parent_node = parent_node->parent();
		line.translate(-indentaion, 0);
	}

	QPixmap pix;
	if (expaned)
	{
		if (hasChild)
		{
			pix = QPixmap(":/branch-expand.png");
		}
	}
	else
	{
		if (hasChild)
		{
			pix = QPixmap(":/branch-collapse.png");
		}
	}
	if (pix.isNull() == false)
	{
		QRect pixRect = QRect(QPoint(0, 0), pix.size());
		pixRect.moveCenter(r.center());

		if (expaned)
		{
			QLine line(r.center(), r.center() + QPoint(0, r.bottom() - r.center().y()));
			painter->drawLine(line);
		}

		painter->drawPixmap(pixRect, pix);
	}

	painter->restore();
}

3、同步左侧表头

上一篇文章Qt实现表格树控件-支持多级表头 中已经说了，我们的表格控件是使用QTableView+QTreeView来实现的，那么我们操作树控件时必然要对表格中的表头进行同步操作了。

点击领取Qt学习资料+视频教程~ 「链接」

树控件折叠时隐藏垂直表头指定行

void collapsed_p(DataNode * node)
{
	QList<DataNode *> childNodeList = node->children();
	//DataManager::getInstance()->allChildNode(node, childNodeList);

	int size = childNodeList.size();
	for (int i = 0; i < size; ++i)
	{
		int serial = DataManager::getInstance()->serialNoOfNode(childNodeList.at(i));
		VHeaderView::instance->SetRowHide(serial, true);

		QModelIndex subIndex = FrozenTreeView::instance->rowIndex(serial);

		collapsed_p(childNodeList.at(i));
	}
}

void FrozenTreeView::onCollapsed(const QModelIndex & index)
{
	if (!index.isValid())
		return;

	DataNode * node = static_cast<DataNode*>(index.internalPointer());
	if (nullptr == node)
		return;

	collapsed_p(node);
	VHeaderView::instance->UpdateCache();

	//要对水平头的最后一列进行重设大小，引起水平头自己的更新操作，从而使整个界面显示正确
	HHeaderView::instance->resizeLastSection(true);
}

树控件展开时显示垂直表头指定行

void expanded_p(DataNode * node)
{
	QList<DataNode *> childNodeList = node->children();

	int size = childNodeList.size();
	for (int i = 0; i < size; ++i)
	{
		int serial = DataManager::getInstance()->serialNoOfNode(childNodeList.at(i));
		VHeaderView::instance->SetRowHide(serial, false);

		QModelIndex subIndex = FrozenTreeView::instance->rowIndex(serial);

		if (FrozenTreeView::instance->isExpanded(subIndex))
		{
			expanded_p(childNodeList.at(i));
		}
	}
}

void FrozenTreeView::onExpanded(const QModelIndex & index)
{
	DataNode * node = static_cast<DataNode *>(index.internalPointer());
	if (nullptr == node)
		return;

	VHeaderView::instance->blockSignals(true);
	expanded_p(node);
	VHeaderView::instance->UpdateCache();
	VHeaderView::instance->blockSignals(false);

	//要对水平头的最后一列进行重设大小，引起水平头自己的更新操作，从而使整个界面显示正确
	HHeaderView::instance->resizeLastSection(false);
}