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
+2 -2
View File
@@ -1195,7 +1195,7 @@ void Companion::Process() {
vWriter.Write((uint32_t) 0);
}
for (const auto & entry : fs::recursive_directory_iterator(this->gAssetPath)){
for (const auto & entry : Torch::getRecursiveEntries(this->gAssetPath)){
if(entry.is_directory()) {
continue;
}
@@ -1258,7 +1258,7 @@ void Companion::Pack(const std::string& folder, const std::string& output, const
auto start = duration_cast<milliseconds>(system_clock::now().time_since_epoch());
std::unordered_map<std::string, std::vector<char>> files;
for (const auto & entry : fs::recursive_directory_iterator(folder)){
for (const auto & entry : Torch::getRecursiveEntries(folder)){
if(entry.is_directory()) {
continue;
}
+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;
}
+4 -1
View File
@@ -4,8 +4,9 @@
#include <fstream>
#include <sstream>
#include <iomanip>
#include <algorithm>
#include <cstdint>
#include <algorithm>
#include <filesystem>
namespace Torch {
template< typename T >
@@ -23,4 +24,6 @@ std::string to_hex(T number, const bool append0x = true) {
}
uint32_t translate(uint32_t offset);
std::vector<std::filesystem::directory_entry> getRecursiveEntries(const std::filesystem::path baseDir);
};