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
|
#include <Windows.h>
#include <iostream>
#include <string>
#include <vector>
#include <unordered_map>
const std::string currentDirectory(".") ;
const std::string previousDirectory("..") ;
class FileIterator
{
public:
struct FileInfo
{
std::string fileName ;
std::string path ;
};
FileIterator( const std::string& path, const std::string& wildcard, const std::vector<std::string>& filters )
: _filters(filters), _directories(1,path), _wildcard(wildcard), _directoryIndex(0), _startingDirectory(true)
{
}
bool getNext( FileInfo & f )
{
WIN32_FIND_DATA ffd ;
for ( ; ; )
{
if ( _startingDirectory )
{
while (_directoryIndex < _directories.size() &&
(_findHandle = FindFirstFile((_directories[_directoryIndex]+_wildcard).c_str(), &ffd)) == INVALID_HANDLE_VALUE )
{
++_directoryIndex ;
}
if ( _findHandle == INVALID_HANDLE_VALUE || _directoryIndex == _directories.size() )
return false ;
_startingDirectory = false ;
if ( _shouldReturnFile( ffd ) )
{
f.fileName = ffd.cFileName ;
f.path = _directories[_directoryIndex] ;
return true ;
}
}
else
{
if ( !FindNextFile(_findHandle, &ffd) )
{
_startingDirectory = true ;
++_directoryIndex ;
}
else if ( _shouldReturnFile(ffd) )
{
f.fileName = ffd.cFileName ;
f.path = _directories[_directoryIndex] ;
return true ;
}
}
}
}
private:
bool _shouldReturnFile( const WIN32_FIND_DATA& f )
{
if ( f.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY )
{
if ( !(f.cFileName == currentDirectory || f.cFileName == previousDirectory) )
_directories.push_back( _directories[_directoryIndex] + f.cFileName + '/' ) ;
return false ;
}
return _matchesFilters(f.cFileName) ;
}
bool _matchesFilters( const std::string & s )
{
for ( auto& filter : _filters )
{
if ( std::string::npos != s.find(filter) )
return true ;
}
return false ;
}
std::vector<std::string> _filters ;
std::vector<std::string> _directories ;
std::string _wildcard ;
HANDLE _findHandle ;
unsigned _directoryIndex ;
bool _startingDirectory ;
};
int main()
{
char const* path = "C:/Projects/TestDir/" ;
char const* wildcard = "*" ;
char const* filterArray[] = { "+", "%20", "%2b" } ;
const unsigned filterArraySize = sizeof(filterArray) / sizeof(filterArray[0]) ;
std::vector<std::string> filters(filterArray, filterArray + filterArraySize);
FileIterator fileIterator( path, wildcard, filters) ;
std::unordered_map<std::string, std::vector<std::string>> fileMap ;
FileIterator::FileInfo fi ;
while ( fileIterator.getNext(fi) )
fileMap[fi.path].push_back(fi.fileName) ;
for ( auto& dir : fileMap )
{
std::cout << "Directory: \"" << dir.first << "\"\n" ;
for ( auto & file : dir.second )
std::cout << "\t\"" << file << "\"\n" ;
}
}
|