mirror of https://github.com/AxioDL/metaforce.git
445 lines
15 KiB
C++
445 lines
15 KiB
C++
#include <utility>
|
|
#include <set>
|
|
|
|
#include "SpecBase.hpp"
|
|
#include "DNAMP3/DNAMP3.hpp"
|
|
|
|
namespace Retro
|
|
{
|
|
|
|
static LogVisor::LogModule Log("Retro::SpecMP3");
|
|
extern HECL::Database::DataSpecEntry SpecEntMP3;
|
|
|
|
struct SpecMP3 : SpecBase
|
|
{
|
|
bool checkStandaloneID(const char* id) const
|
|
{
|
|
if (!memcmp(id, "RM3", 3))
|
|
return true;
|
|
return false;
|
|
}
|
|
|
|
bool doMP3 = false;
|
|
std::vector<const NOD::DiscBase::IPartition::Node*> m_nonPaks;
|
|
std::vector<DNAMP3::PAKBridge> m_paks;
|
|
std::map<std::string, DNAMP3::PAKBridge*, CaseInsensitiveCompare> m_orderedPaks;
|
|
|
|
HECL::ProjectPath m_workPath;
|
|
HECL::ProjectPath m_cookPath;
|
|
PAKRouter<DNAMP3::PAKBridge> m_pakRouter;
|
|
|
|
/* These are populated when extracting MPT's frontend (uses MP3's DataSpec) */
|
|
bool doMPTFE = false;
|
|
std::vector<const NOD::DiscBase::IPartition::Node*> m_feNonPaks;
|
|
std::vector<DNAMP3::PAKBridge> m_fePaks;
|
|
std::map<std::string, DNAMP3::PAKBridge*, CaseInsensitiveCompare> m_feOrderedPaks;
|
|
|
|
HECL::ProjectPath m_feWorkPath;
|
|
HECL::ProjectPath m_feCookPath;
|
|
PAKRouter<DNAMP3::PAKBridge> m_fePakRouter;
|
|
|
|
SpecMP3(HECL::Database::Project& project)
|
|
: SpecBase(project),
|
|
m_workPath(project.getProjectRootPath(), _S("MP3")),
|
|
m_cookPath(project.getProjectCookedPath(SpecEntMP3), _S("MP3")),
|
|
m_pakRouter(*this, m_workPath, m_cookPath),
|
|
m_feWorkPath(project.getProjectRootPath(), _S("fe")),
|
|
m_feCookPath(project.getProjectCookedPath(SpecEntMP3), _S("fe")),
|
|
m_fePakRouter(*this, m_feWorkPath, m_feCookPath) {}
|
|
|
|
void buildPaks(NOD::DiscBase::IPartition::Node& root,
|
|
const std::vector<HECL::SystemString>& args,
|
|
ExtractReport& rep,
|
|
bool fe)
|
|
{
|
|
if (fe)
|
|
{
|
|
m_feNonPaks.clear();
|
|
m_fePaks.clear();
|
|
}
|
|
else
|
|
{
|
|
m_nonPaks.clear();
|
|
m_paks.clear();
|
|
}
|
|
for (const NOD::DiscBase::IPartition::Node& child : root)
|
|
{
|
|
bool isPak = false;
|
|
const std::string& name = child.getName();
|
|
std::string lowerName = name;
|
|
std::transform(lowerName.begin(), lowerName.end(), lowerName.begin(), tolower);
|
|
if (name.size() > 4)
|
|
{
|
|
std::string::iterator extit = lowerName.end() - 4;
|
|
if (!std::string(extit, lowerName.end()).compare(".pak"))
|
|
{
|
|
/* This is a pak */
|
|
isPak = true;
|
|
std::string lowerBase(lowerName.begin(), extit);
|
|
|
|
/* Needs filter */
|
|
bool good = true;
|
|
if (args.size())
|
|
{
|
|
good = false;
|
|
if (!lowerName.compare(0, 7, "metroid"))
|
|
{
|
|
HECL::SystemChar idxChar = lowerName[7];
|
|
for (const HECL::SystemString& arg : args)
|
|
{
|
|
if (arg.size() == 1 && iswdigit(arg[0]))
|
|
if (arg[0] == idxChar)
|
|
good = true;
|
|
}
|
|
}
|
|
else
|
|
good = true;
|
|
|
|
if (!good)
|
|
{
|
|
for (const HECL::SystemString& arg : args)
|
|
{
|
|
#if HECL_UCS2
|
|
std::string lowerArg = HECL::WideToUTF8(arg);
|
|
#else
|
|
std::string lowerArg = arg;
|
|
#endif
|
|
std::transform(lowerArg.begin(), lowerArg.end(), lowerArg.begin(), tolower);
|
|
if (!lowerArg.compare(0, lowerBase.size(), lowerBase))
|
|
good = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (good)
|
|
{
|
|
if (fe)
|
|
m_fePaks.emplace_back(m_project, child);
|
|
else
|
|
m_paks.emplace_back(m_project, child);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!isPak)
|
|
{
|
|
if (fe)
|
|
m_feNonPaks.push_back(&child);
|
|
else
|
|
m_nonPaks.push_back(&child);
|
|
}
|
|
}
|
|
|
|
/* Sort PAKs alphabetically */
|
|
if (fe)
|
|
{
|
|
m_feOrderedPaks.clear();
|
|
for (DNAMP3::PAKBridge& dpak : m_fePaks)
|
|
m_feOrderedPaks[dpak.getName()] = &dpak;
|
|
}
|
|
else
|
|
{
|
|
m_orderedPaks.clear();
|
|
for (DNAMP3::PAKBridge& dpak : m_paks)
|
|
m_orderedPaks[dpak.getName()] = &dpak;
|
|
}
|
|
|
|
/* Assemble extract report */
|
|
for (const std::pair<std::string, DNAMP3::PAKBridge*>& item : fe ? m_feOrderedPaks : m_orderedPaks)
|
|
{
|
|
rep.childOpts.emplace_back();
|
|
ExtractReport& childRep = rep.childOpts.back();
|
|
HECL::SystemStringView nameView(item.first);
|
|
childRep.name = nameView;
|
|
if (!item.first.compare("Worlds.pak"))
|
|
continue;
|
|
else if (!item.first.compare("Metroid6.pak"))
|
|
{
|
|
/* Phaaze doesn't have a world name D: */
|
|
childRep.desc = _S("Phaaze");
|
|
continue;
|
|
}
|
|
childRep.desc = item.second->getLevelString();
|
|
}
|
|
}
|
|
|
|
bool checkFromStandaloneDisc(NOD::DiscBase& disc,
|
|
const HECL::SystemString& regstr,
|
|
const std::vector<HECL::SystemString>& args,
|
|
std::vector<ExtractReport>& reps)
|
|
{
|
|
doMP3 = true;
|
|
NOD::DiscGCN::IPartition* partition = disc.getDataPartition();
|
|
std::unique_ptr<uint8_t[]> dolBuf = partition->getDOLBuf();
|
|
const char* buildInfo = (char*)memmem(dolBuf.get(), partition->getDOLSize(), "MetroidBuildInfo", 16) + 19;
|
|
|
|
/* Root Report */
|
|
reps.emplace_back();
|
|
ExtractReport& rep = reps.back();
|
|
rep.name = _S("MP3");
|
|
rep.desc = _S("Metroid Prime 3 ") + regstr;
|
|
if (buildInfo)
|
|
{
|
|
std::string buildStr(buildInfo);
|
|
HECL::SystemStringView buildView(buildStr);
|
|
rep.desc += _S(" (") + buildView + _S(")");
|
|
}
|
|
|
|
/* Iterate PAKs and build level options */
|
|
NOD::DiscBase::IPartition::Node& root = partition->getFSTRoot();
|
|
buildPaks(root, args, rep, false);
|
|
|
|
return true;
|
|
}
|
|
|
|
bool checkFromTrilogyDisc(NOD::DiscBase& disc,
|
|
const HECL::SystemString& regstr,
|
|
const std::vector<HECL::SystemString>& args,
|
|
std::vector<ExtractReport>& reps)
|
|
{
|
|
std::vector<HECL::SystemString> mp3args;
|
|
std::vector<HECL::SystemString> feargs;
|
|
if (args.size())
|
|
{
|
|
/* Needs filter */
|
|
for (const HECL::SystemString& arg : args)
|
|
{
|
|
HECL::SystemString lowerArg = arg;
|
|
HECL::ToLower(lowerArg);
|
|
if (!lowerArg.compare(0, 3, _S("mp3")))
|
|
{
|
|
doMP3 = true;
|
|
size_t slashPos = arg.find(_S('/'));
|
|
if (slashPos == HECL::SystemString::npos)
|
|
slashPos = arg.find(_S('\\'));
|
|
if (slashPos != HECL::SystemString::npos)
|
|
mp3args.emplace_back(HECL::SystemString(arg.begin() + slashPos + 1, arg.end()));
|
|
}
|
|
}
|
|
|
|
for (const HECL::SystemString& arg : args)
|
|
{
|
|
HECL::SystemString lowerArg = arg;
|
|
HECL::ToLower(lowerArg);
|
|
if (!lowerArg.compare(0, 2, _S("fe")))
|
|
{
|
|
doMPTFE = true;
|
|
size_t slashPos = arg.find(_S('/'));
|
|
if (slashPos == HECL::SystemString::npos)
|
|
slashPos = arg.find(_S('\\'));
|
|
if (slashPos != HECL::SystemString::npos)
|
|
feargs.emplace_back(HECL::SystemString(arg.begin() + slashPos + 1, arg.end()));
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
doMP3 = true;
|
|
doMPTFE = true;
|
|
}
|
|
|
|
if (!doMP3 && !doMPTFE)
|
|
return false;
|
|
|
|
NOD::DiscGCN::IPartition* partition = disc.getDataPartition();
|
|
NOD::DiscBase::IPartition::Node& root = partition->getFSTRoot();
|
|
|
|
/* MP3 extract */
|
|
if (doMP3)
|
|
{
|
|
NOD::DiscBase::IPartition::Node::DirectoryIterator dolIt = root.find("rs5mp3_p.dol");
|
|
if (dolIt == root.end())
|
|
return false;
|
|
|
|
std::unique_ptr<uint8_t[]> dolBuf = dolIt->getBuf();
|
|
const char* buildInfo = (char*)memmem(dolBuf.get(), dolIt->size(), "MetroidBuildInfo", 16) + 19;
|
|
|
|
/* Root Report */
|
|
reps.emplace_back();
|
|
ExtractReport& rep = reps.back();
|
|
rep.name = _S("MP3");
|
|
rep.desc = _S("Metroid Prime 3 ") + regstr;
|
|
if (buildInfo)
|
|
{
|
|
std::string buildStr(buildInfo);
|
|
HECL::SystemStringView buildView(buildStr);
|
|
rep.desc += _S(" (") + buildView + _S(")");
|
|
}
|
|
|
|
/* Iterate PAKs and build level options */
|
|
NOD::DiscBase::IPartition::Node::DirectoryIterator mp3It = root.find("MP3");
|
|
if (mp3It == root.end())
|
|
return false;
|
|
buildPaks(*mp3It, mp3args, rep, false);
|
|
}
|
|
|
|
/* MPT Frontend extract */
|
|
if (doMPTFE)
|
|
{
|
|
NOD::DiscBase::IPartition::Node::DirectoryIterator dolIt = root.find("rs5fe_p.dol");
|
|
if (dolIt == root.end())
|
|
return false;
|
|
|
|
std::unique_ptr<uint8_t[]> dolBuf = dolIt->getBuf();
|
|
const char* buildInfo = (char*)memmem(dolBuf.get(), dolIt->size(), "MetroidBuildInfo", 16) + 19;
|
|
|
|
/* Root Report */
|
|
reps.emplace_back();
|
|
ExtractReport& rep = reps.back();
|
|
rep.name = _S("fe");
|
|
rep.desc = _S("Metroid Prime Trilogy Frontend ") + regstr;
|
|
if (buildInfo)
|
|
{
|
|
std::string buildStr(buildInfo);
|
|
HECL::SystemStringView buildView(buildStr);
|
|
rep.desc += _S(" (") + buildView + _S(")");
|
|
}
|
|
|
|
/* Iterate PAKs and build level options */
|
|
NOD::DiscBase::IPartition::Node::DirectoryIterator feIt = root.find("fe");
|
|
if (feIt == root.end())
|
|
return false;
|
|
buildPaks(*feIt, feargs, rep, true);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool extractFromDisc(NOD::DiscBase&, bool force, FExtractProgress progress)
|
|
{
|
|
int compIdx = 2;
|
|
int prog;
|
|
if (doMP3)
|
|
{
|
|
progress(_S("Indexing PAKs"), compIdx, 0.0);
|
|
m_pakRouter.build(m_paks, [&progress, &compIdx](float factor)
|
|
{
|
|
progress(_S("Indexing PAKs"), compIdx, factor);
|
|
});
|
|
progress(_S("Indexing PAKs"), compIdx++, 1.0);
|
|
|
|
HECL::ProjectPath mp3WorkPath(m_project.getProjectRootPath(), "MP3");
|
|
mp3WorkPath.makeDir();
|
|
progress(_S("MP3 Root"), compIdx, 0.0);
|
|
prog = 0;
|
|
for (const NOD::DiscBase::IPartition::Node* node : m_nonPaks)
|
|
{
|
|
node->extractToDirectory(mp3WorkPath.getAbsolutePath(), force);
|
|
progress(_S("MP3 Root"), compIdx, prog++ / (float)m_nonPaks.size());
|
|
}
|
|
progress(_S("MP3 Root"), compIdx++, 1.0);
|
|
|
|
const HECL::ProjectPath& cookPath = m_project.getProjectCookedPath(SpecEntMP3);
|
|
cookPath.makeDir();
|
|
HECL::ProjectPath mp3CookPath(cookPath, "MP3");
|
|
mp3CookPath.makeDir();
|
|
|
|
prog = 0;
|
|
for (std::pair<std::string, DNAMP3::PAKBridge*> pair : m_orderedPaks)
|
|
{
|
|
DNAMP3::PAKBridge& pak = *pair.second;
|
|
m_pakRouter.enterPAKBridge(pak);
|
|
|
|
const std::string& name = pak.getName();
|
|
HECL::SystemStringView sysName(name);
|
|
|
|
progress(sysName.sys_str().c_str(), compIdx, 0.0);
|
|
m_pakRouter.extractResources(pak, force,
|
|
[&progress, &sysName, &compIdx](float factor)
|
|
{
|
|
progress(sysName.sys_str().c_str(), compIdx, factor);
|
|
});
|
|
progress(sysName.sys_str().c_str(), compIdx++, 1.0);
|
|
}
|
|
}
|
|
|
|
if (doMPTFE)
|
|
{
|
|
progress(_S("Indexing PAKs"), compIdx, 0.0);
|
|
m_fePakRouter.build(m_fePaks, [&progress, &compIdx](float factor)
|
|
{
|
|
progress(_S("Indexing PAKs"), compIdx, factor);
|
|
});
|
|
progress(_S("Indexing PAKs"), compIdx++, 1.0);
|
|
|
|
m_feWorkPath.makeDir();
|
|
progress(_S("fe Root"), compIdx, 0.0);
|
|
int prog = 0;
|
|
for (const NOD::DiscBase::IPartition::Node* node : m_feNonPaks)
|
|
{
|
|
node->extractToDirectory(m_feWorkPath.getAbsolutePath(), force);
|
|
progress(_S("fe Root"), compIdx, prog++ / (float)m_feNonPaks.size());
|
|
}
|
|
progress(_S("fe Root"), compIdx++, 1.0);
|
|
|
|
const HECL::ProjectPath& cookPath = m_project.getProjectCookedPath(SpecEntMP3);
|
|
cookPath.makeDir();
|
|
m_feCookPath.makeDir();
|
|
|
|
prog = 0;
|
|
for (std::pair<std::string, DNAMP3::PAKBridge*> pair : m_feOrderedPaks)
|
|
{
|
|
DNAMP3::PAKBridge& pak = *pair.second;
|
|
const std::string& name = pak.getName();
|
|
HECL::SystemStringView sysName(name);
|
|
|
|
progress(sysName.sys_str().c_str(), compIdx, 0.0);
|
|
m_fePakRouter.extractResources(pak, force,
|
|
[&progress, &sysName, &compIdx](float factor)
|
|
{
|
|
progress(sysName.sys_str().c_str(), compIdx, factor);
|
|
});
|
|
progress(sysName.sys_str().c_str(), compIdx++, 1.0);
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool checkFromProject()
|
|
{
|
|
return false;
|
|
}
|
|
bool readFromProject()
|
|
{
|
|
return false;
|
|
}
|
|
|
|
bool visitGameObjects(std::function<bool(const HECL::Database::ObjectBase&)>)
|
|
{
|
|
return false;
|
|
}
|
|
struct LevelSpec : public ILevelSpec
|
|
{
|
|
bool visitLevelObjects(std::function<bool(const HECL::Database::ObjectBase&)>)
|
|
{
|
|
return false;
|
|
}
|
|
struct AreaSpec : public IAreaSpec
|
|
{
|
|
bool visitAreaObjects(std::function<bool(const HECL::Database::ObjectBase&)>)
|
|
{
|
|
return false;
|
|
}
|
|
};
|
|
bool visitAreas(std::function<bool(const IAreaSpec&)>)
|
|
{
|
|
return false;
|
|
}
|
|
};
|
|
bool visitLevels(std::function<bool(const ILevelSpec&)>)
|
|
{
|
|
return false;
|
|
}
|
|
};
|
|
|
|
HECL::Database::DataSpecEntry SpecEntMP3
|
|
(
|
|
_S("MP3"),
|
|
_S("Data specification for original Metroid Prime 3 engine"),
|
|
[](HECL::Database::Project& project, HECL::Database::DataSpecTool)
|
|
-> HECL::Database::IDataSpec* {return new struct SpecMP3(project);}
|
|
);
|
|
|
|
}
|