1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
/*
SPDX-FileCopyrightText: 2007 David Nolden <david.nolden.kdevelop@art-master.de>
SPDX-License-Identifier: LGPL-2.0-or-later
*/
#include "expandingtree.h"
#include <QAbstractProxyModel>
#include <QAbstractTextDocumentLayout>
#include <QPainter>
#include "expandingwidgetmodel.h"
#include <util/path.h>
#include <util/widgetcolorizer.h>
using namespace KDevelop;
ExpandingTree::ExpandingTree(QWidget* parent) : QTreeView(parent)
{
m_drawText.documentLayout()->setPaintDevice(this);
setUniformRowHeights(false);
}
void ExpandingTree::setModel(QAbstractItemModel* model)
{
Q_ASSERT(!model || qobject_cast<const ExpandingWidgetModel*>(
qobject_cast<const QAbstractProxyModel*>(model)->sourceModel())
);
QTreeView::setModel(model);
}
void ExpandingTree::drawRow(QPainter* painter, const QStyleOptionViewItem& option, const QModelIndex& index) const
{
QTreeView::drawRow(painter, option, index);
const auto* eModel = qobject_cast<const ExpandingWidgetModel*>(qobject_cast<const QAbstractProxyModel*>(model())->sourceModel());
Q_ASSERT(eModel);
const QModelIndex sourceIndex = eModel->mapToSource(index);
if (eModel->isPartiallyExpanded(sourceIndex) != ExpandingWidgetModel::ExpansionType::NotExpanded) {
QRect rect = eModel->partialExpandRect(sourceIndex);
if (rect.isValid()) {
QStyleOption opt;
QAbstractTextDocumentLayout::PaintContext ctx;
opt.rect = rect;
style()->drawPrimitive(QStyle::PE_FrameLineEdit, &opt, painter);
ctx.clip = QRectF(0, 0, rect.width(), rect.height());
painter->setViewTransformEnabled(true);
painter->translate(rect.left(), rect.top());
m_drawText.setHtml(eModel->partialExpandText(sourceIndex));
WidgetColorizer::convertDocumentToDarkTheme(&m_drawText);
m_drawText.setPageSize(QSizeF(rect.width(), rect.height()));
m_drawText.documentLayout()->draw(painter, ctx);
painter->translate(-rect.left(), -rect.top());
}
}
}
int ExpandingTree::sizeHintForColumn(int column) const
{
return columnWidth(column);
}
void ExpandingTree::drawBranches(QPainter* painter, const QRect& rect, const QModelIndex& index) const
{
const auto& path = index.data(ProjectPathRole).value<Path>();
if (path.isValid()) {
const auto color = WidgetColorizer::colorForId(qHash(path), palette(), true);
WidgetColorizer::drawBranches(this, painter, rect, index, color);
}
QTreeView::drawBranches(painter, rect, index);
}
#include "moc_expandingtree.cpp"
|
