romfs: Add SingleDiscard extraction type

Needed for manual RomFS extraction, as Full generates an extra directory and Truncated generates variable results.
master
Zach Hilman 2018-12-24 16:18:28 +07:00
parent 94db6e5f3f
commit abbcc8e61e
2 changed files with 6 additions and 2 deletions

@ -119,6 +119,9 @@ VirtualDir ExtractRomFS(VirtualFile file, RomFSExtractionType type) {
VirtualDir out = std::move(root); VirtualDir out = std::move(root);
if (type == RomFSExtractionType::SingleDiscard)
return out->GetSubdirectories().front();
while (out->GetSubdirectories().size() == 1 && out->GetFiles().empty()) { while (out->GetSubdirectories().size() == 1 && out->GetFiles().empty()) {
if (out->GetSubdirectories().front()->GetName() == "data" && if (out->GetSubdirectories().front()->GetName() == "data" &&
type == RomFSExtractionType::Truncated) type == RomFSExtractionType::Truncated)

@ -35,6 +35,7 @@ static_assert(sizeof(IVFCHeader) == 0xE0, "IVFCHeader has incorrect size.");
enum class RomFSExtractionType { enum class RomFSExtractionType {
Full, // Includes data directory Full, // Includes data directory
Truncated, // Traverses into data directory Truncated, // Traverses into data directory
SingleDiscard, // Traverses into the first subdirectory of root
}; };
// Converts a RomFS binary blob to VFS Filesystem // Converts a RomFS binary blob to VFS Filesystem