一、开心一刻
一程序员第一次上女朋友家她妈板着脸问 ：你想娶我女儿，有多少存款？
程序员低了下头：五百！
她妈更鄙视了：才五百块，买个厕所都不够！
程序员忙说：不是人民币！
她妈：就算是美元，还是不够买厕所！
程序员：其实是比特币！
她妈：哇，贤婿，我给你买只大龙虾去
二、自绘树节点？
自绘树节点？听起来都挺复杂的，可是为什么还要自绘树节点呢？这充分说明产品的脑子是什么东西都能想出来的。
有一天产品说我们的软件里缺少一个美丽的树控件，然后就要求开发去实现这个功能。
对于有一定开发经验的同学可能直接会去百度，或者上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类的帮助文档，查找这个类都有哪些可供重写的接口，然后就发现了这么几个函数
看名字大概都知道是什么意思，不过这里还是做简要说明
函数名 | 含义 |
drawBranches | 绘制branch |
drawRow | 绘制行 |
drawTree | 绘制树 |
indexRowSizeHint | 默认行高 |
rowHeight | 获取行高 |
前边提到我们要自己绘制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学习资料+视频教程~ 「链接」
需要注意的点
- 除根节点外，每个节点都需要绘制文字前边的水平线
- 有父亲的节点需要绘制垂直线。绘制的竖线是否绘制到底，取决于是否有向下的兄弟
- 有爷爷的节点可能需要额外绘制向下的竖线。是否绘制取决于自己的父亲是否有向下的兄弟
- 规则3其实是一个循环的处理，也就是说爷爷如果有爸爸，也就是说节点如果有祖爷爷，那么可能还需要绘制更多的向下竖线。是否绘制取决于节点的爷爷是否有向下的兄弟
代码这里就不细说了，有兴趣的可以自己研究研究。绘制规则就是上述4点
//绘制branch
{
DataNode * node = static_cast(index.internalPointer());
bool hasChild = node->children().size() != 0;//是否有孩子
QList & 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 & 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 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(index.internalPointer());
if (nullptr == node)
return;
collapsed_p(node);
VHeaderView::instance->UpdateCache();
//要对水平头的最后一列进行重设大小，引起水平头自己的更新操作，从而使整个界面显示正确
HHeaderView::instance->resizeLastSection(true);
}
树控件展开时显示垂直表头指定行
void expanded_p(DataNode * node)
{
QList 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(index.internalPointer());
if (nullptr == node)
return;
VHeaderView::instance->blockSignals(true);
expanded_p(node);
VHeaderView::instance->UpdateCache();
VHeaderView::instance->blockSignals(false);
//要对水平头的最后一列进行重设大小，引起水平头自己的更新操作，从而使整个界面显示正确
HHeaderView::instance->resizeLastSection(false);
}