一、开心一刻
一程序员第一次上女朋友家她妈板着脸问 :你想娶我女儿,有多少存款?
程序员低了下头:五百!
她妈更鄙视了:才五百块,买个厕所都不够!
程序员忙说:不是人民币!
她妈:就算是美元,还是不够买厕所!
程序员:其实是比特币!
她妈:哇,贤婿,我给你买只大龙虾去
二、自绘树节点?
自绘树节点?听起来都挺复杂的,可是为什么还要自绘树节点呢?这充分说明产品的脑子是什么东西都能想出来的。
有一天产品说我们的软件里缺少一个美丽的树控件,然后就要求开发去实现这个功能。
对于有一定开发经验的同学可能直接会去百度,或者上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时主要是根据当前节点是否展开、是否有孩子节点、是否有兄弟节点等状态来联合判断并进行绘制
如下是绘制代码,可能有些长,但是应该比较好理解。
需要注意的点
- 除根节点外,每个节点都需要绘制文字前边的水平线
- 有父亲的节点需要绘制垂直线。绘制的竖线是否绘制到底,取决于是否有向下的兄弟
- 有爷爷的节点可能需要额外绘制向下的竖线。是否绘制取决于自己的父亲是否有向下的兄弟
- 规则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来实现的,那么我们操作树控件时必然要对表格中的表头进行同步操作了。
树控件折叠时隐藏垂直表头指定行
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);
}