utility.cpp
1.62 KB
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
#include <openbr/core/qtutils.h>
#include "utility.h"
QStringList br::getFiles(QDir dir, bool recursive)
{
dir = QDir(dir.canonicalPath());
QStringList files;
foreach (const QString &file, QtUtils::naturalSort(dir.entryList(QDir::Files)))
files.append(dir.absoluteFilePath(file));
if (!recursive) return files;
foreach (const QString &folder, QtUtils::naturalSort(dir.entryList(QDir::Dirs | QDir::NoDotAndDotDot))) {
QDir subdir(dir);
bool success = subdir.cd(folder); if (!success) qFatal("cd failure.");
files.append(getFiles(subdir, true));
}
return files;
}
QList<br::FilesWithLabel> br::getFilesWithLabels(QDir dir)
{
dir = QDir(dir.canonicalPath());
QStringList files;
foreach (const QString &file, dir.entryList(QDir::Files))
files.append(dir.absoluteFilePath(file));
QList<br::FilesWithLabel> filesWithLabels;
filesWithLabels.append(br::FilesWithLabel(dir.dirName(),files));
foreach (const QString &folder, dir.entryList(QDir::Dirs | QDir::NoDotAndDotDot)) {
QDir subdir(dir);
bool success = subdir.cd(folder); if (!success) qFatal("cd failure.");
filesWithLabels.append(getFilesWithLabels(subdir));
}
return filesWithLabels;
}
QStringList br::getFiles(const QString ®exp)
{
QFileInfo fileInfo(regexp);
QDir dir(fileInfo.dir());
QRegExp re(fileInfo.fileName());
re.setPatternSyntax(QRegExp::Wildcard);
QStringList files;
foreach (const QString &fileName, dir.entryList(QDir::Files))
if (re.exactMatch(fileName))
files.append(dir.filePath(fileName));
return files;
}