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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
|
/*
widget3dwindowmodel.cpp
This file is part of GammaRay, the Qt application inspection and manipulation tool.
SPDX-FileCopyrightText: 2011 Klarälvdalens Datakonsult AB, a KDAB Group company <info@kdab.com>
Author: Daniel Vrátil <daniel.vratil@kdab.com>
SPDX-License-Identifier: GPL-2.0-or-later
Contact KDAB at <info@kdab.com> for commercial licensing options.
*/
#include "widget3dwindowmodel.h"
#include "widget3dmodel.h"
#include <QDebug>
namespace GammaRay {
class Widget3DWindowModel::WindowNode
{
public:
explicit WindowNode(const QModelIndex &idx)
: sourceIdx(idx)
{
}
QPersistentModelIndex sourceIdx;
};
}
using namespace GammaRay;
Widget3DWindowModel::Widget3DWindowModel(QObject *parent)
: QAbstractProxyModel(parent)
{
}
Widget3DWindowModel::~Widget3DWindowModel()
{
}
void Widget3DWindowModel::setSourceModel(QAbstractItemModel *sourceModel_)
{
if (sourceModel()) {
disconnect(sourceModel(), nullptr, this, nullptr);
}
QAbstractProxyModel::setSourceModel(sourceModel_);
connect(sourceModel_, &QAbstractItemModel::rowsInserted,
this, &Widget3DWindowModel::sourceModelRowsInserted);
connect(sourceModel_, &QAbstractItemModel::rowsRemoved,
this, &Widget3DWindowModel::sourceModelRowsRemoved);
connect(sourceModel_, &QAbstractItemModel::modelReset,
this, &Widget3DWindowModel::sourceModelReset);
beginResetModel();
qDeleteAll(mNodes);
mNodes.clear();
populate();
endResetModel();
}
QModelIndex Widget3DWindowModel::index(int row, int column, const QModelIndex &parent) const
{
if (parent.isValid()) {
return QModelIndex();
}
if (row < 0 || row >= mNodes.size() || column != 0) {
return QModelIndex();
}
return createIndex(row, column, mNodes.at(row));
}
QModelIndex Widget3DWindowModel::parent(const QModelIndex &) const
{
return QModelIndex(); // list
}
int Widget3DWindowModel::rowCount(const QModelIndex &parent) const
{
return parent.isValid() ? 0 : mNodes.size();
}
int Widget3DWindowModel::columnCount(const QModelIndex &) const
{
return 1;
}
QModelIndex Widget3DWindowModel::indexForNode(WindowNode *node) const
{
int row = mNodes.indexOf(node);
Q_ASSERT(row > -1);
return createIndex(row, 0, node);
}
QModelIndex Widget3DWindowModel::mapFromSource(const QModelIndex &sourceIndex) const
{
for (WindowNode *node : mNodes) {
if (node->sourceIdx == sourceIndex) {
return indexForNode(node);
}
}
return QModelIndex();
}
QModelIndex Widget3DWindowModel::mapToSource(const QModelIndex &proxyIndex) const
{
if (!proxyIndex.isValid()) {
return QModelIndex();
}
auto node = static_cast<WindowNode *>(proxyIndex.internalPointer());
if (!node) {
return QModelIndex();
}
return node->sourceIdx;
}
void Widget3DWindowModel::sourceModelRowsInserted(const QModelIndex &parent, int first, int last)
{
for (int i = first; i <= last; ++i) {
const QModelIndex srcIdx = sourceModel()->index(i, 0, parent);
if (srcIdx.data(Widget3DModel::IsWindowRole).toBool()) {
const int pos = mNodes.size();
beginInsertRows(QModelIndex(), pos, pos);
mNodes.push_back(new WindowNode(srcIdx));
endInsertRows();
}
}
}
void Widget3DWindowModel::sourceModelRowsRemoved()
{
// Iterate over all windows we know and check if any of the indexes has
// become invalid. We won't ever have that many windows for this to become
// too slow and we don't have to have a complex code to traverse the entire
// removed subtree just to check it happened to contain one of our windows.
for (int i = mNodes.size() - 1; i >= 0; --i) {
if (!mNodes.at(i)->sourceIdx.isValid()) {
beginRemoveRows(QModelIndex(), i, i);
delete mNodes.takeAt(i);
endRemoveRows();
}
}
}
void Widget3DWindowModel::sourceModelReset()
{
beginResetModel();
qDeleteAll(mNodes);
mNodes.clear();
populate();
endResetModel();
}
void Widget3DWindowModel::populate()
{
QVector<QModelIndex> toVisit = { QModelIndex() };
while (!toVisit.isEmpty()) {
const auto idx = toVisit.takeFirst();
if (idx.data(Widget3DModel::IsWindowRole).toBool()) {
mNodes.push_back(new WindowNode(idx));
}
for (int i = 0, cnt = sourceModel()->rowCount(idx); i < cnt; ++i) {
toVisit.push_front(sourceModel()->index(i, 0, idx));
}
}
}
|
