Replaced recursive iteration to be sorted by name and depth

This commit is contained in:
KiritoDv
2025-01-30 10:35:05 -06:00
committed by Lywx
parent 18b85b04ee
commit 043fc76ebc
3 changed files with 34 additions and 3 deletions
+28
View File
@@ -1,9 +1,12 @@
#include "TorchUtils.h"
#include <stack>
#include <Companion.h>
#include <factories/BaseFactory.h>
#include "spdlog/spdlog.h"
namespace fs = std::filesystem;
uint32_t Torch::translate(const uint32_t offset) {
if(SEGMENT_NUMBER(offset) > 0x01) {
auto segment = SEGMENT_NUMBER(offset);
@@ -18,3 +21,28 @@ uint32_t Torch::translate(const uint32_t offset) {
return offset;
}
int getFileDepth(const fs::path& base, const fs::path& p) {
return std::distance(base.begin(), p.begin());
}
std::vector<fs::directory_entry> Torch::getRecursiveEntries(const fs::path baseDir) {
std::vector<fs::directory_entry> result;
for (const auto& entry : fs::recursive_directory_iterator(baseDir)) {
result.push_back(entry);
}
std::sort(result.begin(), result.end(), [](const fs::directory_entry& a, const fs::directory_entry& b) {
std::string nameA = a.path().parent_path().string();
std::string nameB = b.path().parent_path().string();
int depthA = std::distance(a.path().begin(), a.path().end());
int depthB = std::distance(b.path().begin(), b.path().end());
if (nameA != nameB)
return nameA < nameB;
return depthA < depthB;
});
return result;
}