diff --git a/PGOHelper/PGOHelper.cpp b/PGOHelper/PGOHelper.cpp index ca190c1..50224ba 100644 --- a/PGOHelper/PGOHelper.cpp +++ b/PGOHelper/PGOHelper.cpp @@ -3,8 +3,8 @@ #include #include #if __has_include() - #include - namespace fs = std::filesystem; +#include +namespace fs = std::filesystem; #elif __has_include() #include namespace fs = std::experimental::filesystem; @@ -14,24 +14,28 @@ using std::string; using std::vector; extern "C" { - void __stdcall PgoRunTest(vector testRoms, bool enableDebugger); +void __stdcall PgoRunTest(vector testRoms, bool enableDebugger); } vector GetFilesInFolder(string rootFolder, std::unordered_set extensions) { vector files; - vector folders = { { rootFolder } }; + vector folders = {{rootFolder}}; std::error_code errorCode; - if(!fs::is_directory(fs::u8path(rootFolder), errorCode)) { + if (!fs::is_directory(fs::u8path(rootFolder), errorCode)) + { return files; } - for(string folder : folders) { - for(fs::directory_iterator i(fs::u8path(folder.c_str())), end; i != end; i++) { + for (string folder : folders) + { + for (fs::directory_iterator i(fs::u8path(folder.c_str())), end; i != end; i++) + { string extension = i->path().extension().u8string(); std::transform(extension.begin(), extension.end(), extension.begin(), ::tolower); - if(extensions.find(extension) != extensions.end()) { + if (extensions.find(extension) != extensions.end()) + { files.push_back(i->path().u8string()); } } @@ -43,12 +47,12 @@ vector GetFilesInFolder(string rootFolder, std::unordered_set ex int main(int argc, char* argv[]) { string romFolder = "../PGOGames"; - if(argc >= 2) { + if (argc >= 2) + { romFolder = argv[1]; } - vector testRoms = GetFilesInFolder(romFolder, { {".sfc", ".gb", ".gbc"} }); + vector testRoms = GetFilesInFolder(romFolder, {{".sfc", ".gb", ".gbc"}}); PgoRunTest(testRoms, true); return 0; } -