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
|
#include "FilesCollector.h"
#include "fileutils.h"
#include <algorithm>
#include <queue>
#include <wx/dir.h>
#include <wx/filefn.h>
#include <wx/filename.h>
#include <wx/tokenzr.h>
FilesCollector::FilesCollector(wxArrayString& filesAndFolders, const wxString& filespec, const wxString& excludeFolders,
wxProgressDialog* progress)
: m_filesAndFolders(filesAndFolders)
, m_progress(progress)
{
m_specArray = ::wxStringTokenize(filespec.Lower(), ";", wxTOKEN_STRTOK);
wxArrayString arrFolders = ::wxStringTokenize(excludeFolders, ";", wxTOKEN_STRTOK);
m_excludeFolders.insert(arrFolders.begin(), arrFolders.end());
}
FilesCollector::~FilesCollector() {}
bool FilesCollector::IsFileOK(const wxString& filename) const
{
if(FileUtils::WildMatch(m_specArray, filename)) { return true; }
return false;
}
void FilesCollector::Collect(const wxString& rootFolder)
{
if(!wxFileName::DirExists(rootFolder)) {
m_filesAndFolders.clear();
return;
}
std::queue<wxString> Q;
Q.push(rootFolder);
std::vector<wxString> V;
while(!Q.empty()) {
wxString dirpath = Q.front();
Q.pop();
wxDir dir(dirpath);
if(!dir.IsOpened()) { continue; }
wxString filename;
bool cont = dir.GetFirst(&filename);
while(cont) {
// Check to see if this is a folder
wxString fullpath;
fullpath << dir.GetNameWithSep() << filename;
bool isDirectory = wxFileName::DirExists(fullpath);
if(isDirectory && (m_excludeFolders.count(filename) == 0)) {
// A directory
Q.push(fullpath);
fullpath << wxFileName::GetPathSeparator() << FOLDER_MARKER;
V.push_back(fullpath);
} else if(!isDirectory && IsFileOK(filename)) {
// A file
V.push_back(fullpath);
}
cont = dir.GetNext(&filename);
}
}
m_filesAndFolders.Alloc(V.size());
std::for_each(V.begin(), V.end(), [&](const wxString& f) { m_filesAndFolders.push_back(f); });
}
|
