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
|
/*
simplefiletreemodel.cpp
Provides a simple tree model for files and folders
*/
#include "simplefiletreemodel.h"
#include "simplefiletreeitem.h"
#include <QDir>
#include <QFileIconProvider>
#include <QStringList>
SimpleFileTreeModel::SimpleFileTreeModel(const QStringList& data, QObject* parent)
: QAbstractItemModel(parent)
{
QFileIconProvider* provider = new QFileIconProvider();
m_FolderIcon = provider->icon(QFileIconProvider::Folder);
m_FileIcon = provider->icon(QFileIconProvider::File);
delete provider;
m_RootItem = new SimpleFileTreeItem({tr("File Name")});
setupModelData(data, m_RootItem);
}
SimpleFileTreeModel::~SimpleFileTreeModel()
{
delete m_RootItem;
}
int SimpleFileTreeModel::columnCount(const QModelIndex& parent) const
{
if (parent.isValid())
return static_cast<SimpleFileTreeItem*>(parent.internalPointer())->columnCount();
return m_RootItem->columnCount();
}
QVariant SimpleFileTreeModel::data(const QModelIndex& index, int role) const
{
if (!index.isValid())
return QVariant();
SimpleFileTreeItem* item = static_cast<SimpleFileTreeItem*>(index.internalPointer());
if (role == Qt::DecorationRole) {
if (item->childCount() > 0) {
return m_FolderIcon;
} else {
return m_FileIcon;
}
}
if (role != Qt::DisplayRole)
return QVariant();
return item->data(index.column());
}
Qt::ItemFlags SimpleFileTreeModel::flags(const QModelIndex& index) const
{
if (!index.isValid())
return Qt::NoItemFlags;
return QAbstractItemModel::flags(index);
}
QVariant SimpleFileTreeModel::headerData(int section, Qt::Orientation orientation,
int role) const
{
if (orientation == Qt::Horizontal && role == Qt::DisplayRole)
return m_RootItem->data(section);
return QVariant();
}
QModelIndex SimpleFileTreeModel::index(int row, int column,
const QModelIndex& parent) const
{
if (!hasIndex(row, column, parent))
return QModelIndex();
SimpleFileTreeItem* parentItem;
if (!parent.isValid())
parentItem = m_RootItem;
else
parentItem = static_cast<SimpleFileTreeItem*>(parent.internalPointer());
SimpleFileTreeItem* childItem = parentItem->child(row);
if (childItem)
return createIndex(row, column, childItem);
return QModelIndex();
}
QModelIndex SimpleFileTreeModel::parent(const QModelIndex& index) const
{
if (!index.isValid())
return QModelIndex();
SimpleFileTreeItem* childItem =
static_cast<SimpleFileTreeItem*>(index.internalPointer());
SimpleFileTreeItem* parentItem = childItem->parentItem();
if (parentItem == m_RootItem)
return QModelIndex();
return createIndex(parentItem->row(), 0, parentItem);
}
int SimpleFileTreeModel::rowCount(const QModelIndex& parent) const
{
SimpleFileTreeItem* parentItem;
if (parent.column() > 0)
return 0;
if (!parent.isValid())
parentItem = m_RootItem;
else
parentItem = static_cast<SimpleFileTreeItem*>(parent.internalPointer());
return parentItem->childCount();
}
void SimpleFileTreeModel::setupModelData(const QStringList& lines, SimpleFileTreeItem*)
{
for (QString line : lines) {
auto fullPath = QDir::cleanPath(line);
QStringList lineEntries = fullPath.split("/");
auto currentParent = m_RootItem;
for (int i = 0; i < lineEntries.count(); i++) {
QString currentEntryName = lineEntries[i];
// check if item was already added
SimpleFileTreeItem* currentEntry = currentParent->childByName(currentEntryName);
// add tree item if not found
if (currentEntry == nullptr) {
QVector<QVariant> columnData;
columnData.reserve(m_ColumnCount);
columnData << currentEntryName;
currentEntry = new SimpleFileTreeItem(columnData, currentParent);
currentParent->appendChild(currentEntryName, currentEntry);
}
// as we go deeper into the path
currentParent = currentEntry;
}
}
}
|
