2015-07-13 06:29:12 +00:00
|
|
|
#include <utility>
|
2015-07-14 00:38:48 +00:00
|
|
|
#include <set>
|
2015-07-13 06:29:12 +00:00
|
|
|
|
2020-04-11 07:53:04 +00:00
|
|
|
#include "DataSpec/SpecBase.hpp"
|
|
|
|
#include "DataSpec/DNAMP3/DNAMP3.hpp"
|
2015-07-13 06:29:12 +00:00
|
|
|
|
2020-04-11 07:53:04 +00:00
|
|
|
#include "DataSpec/DNAMP3/MLVL.hpp"
|
|
|
|
#include "DataSpec/DNAMP3/STRG.hpp"
|
|
|
|
#include "DataSpec/DNAMP3/MAPA.hpp"
|
|
|
|
#include "DataSpec/DNAMP2/STRG.hpp"
|
|
|
|
#include "DataSpec/DNACommon/TXTR.hpp"
|
2020-04-15 13:42:44 +00:00
|
|
|
#include "DataSpec/DNACommon/URDEVersionInfo.hpp"
|
2015-10-01 00:40:21 +00:00
|
|
|
|
2016-04-01 04:25:00 +00:00
|
|
|
#include "hecl/ClientProcess.hpp"
|
2017-12-29 08:08:12 +00:00
|
|
|
#include "hecl/Blender/Connection.hpp"
|
2018-03-23 21:56:17 +00:00
|
|
|
#include "hecl/MultiProgressPrinter.hpp"
|
2016-04-01 04:25:00 +00:00
|
|
|
|
2016-10-02 22:41:36 +00:00
|
|
|
#include "Runtime/RetroTypes.hpp"
|
2017-12-29 08:08:12 +00:00
|
|
|
#include "nod/nod.hpp"
|
2016-10-02 22:41:36 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
namespace DataSpec {
|
2015-07-13 06:29:12 +00:00
|
|
|
|
2017-11-13 06:19:18 +00:00
|
|
|
using namespace std::literals;
|
|
|
|
|
2016-03-04 23:04:53 +00:00
|
|
|
static logvisor::Module Log("urde::SpecMP3");
|
|
|
|
extern hecl::Database::DataSpecEntry SpecEntMP3;
|
2017-01-17 01:23:19 +00:00
|
|
|
extern hecl::Database::DataSpecEntry SpecEntMP3ORIG;
|
2015-07-16 01:57:34 +00:00
|
|
|
|
2020-04-11 07:53:04 +00:00
|
|
|
struct TextureCache {
|
|
|
|
static void Generate(PAKRouter<DNAMP3::PAKBridge>& pakRouter, hecl::Database::Project& project, const hecl::ProjectPath& pakPath) {
|
|
|
|
hecl::ProjectPath texturePath(pakPath, _SYS_STR("texture_cache.yaml"));
|
|
|
|
hecl::ProjectPath catalogPath(pakPath, _SYS_STR("!catalog.yaml"));
|
|
|
|
texturePath.makeDirChain(false);
|
|
|
|
|
|
|
|
if (const auto fp = hecl::FopenUnique(catalogPath.getAbsolutePath().data(), _SYS_STR("a"))) {
|
2020-04-11 22:51:39 +00:00
|
|
|
fmt::print(fp.get(), FMT_STRING("TextureCache: {}\n"), texturePath.getRelativePathUTF8());
|
2020-04-11 07:53:04 +00:00
|
|
|
}
|
|
|
|
|
2020-04-11 22:51:39 +00:00
|
|
|
Log.report(logvisor::Level::Info, FMT_STRING("Gathering Texture metadata (this can take up to 10 seconds)..."));
|
2020-04-11 07:53:04 +00:00
|
|
|
std::unordered_map<UniqueID64, TXTR::Meta> metaMap;
|
|
|
|
|
|
|
|
pakRouter.enumerateResources([&](const DNAMP3::PAK::Entry* ent) {
|
|
|
|
if (ent->type == FOURCC('TXTR') && metaMap.find(ent->id) == metaMap.end()) {
|
|
|
|
PAKEntryReadStream rs = pakRouter.beginReadStreamForId(ent->id);
|
|
|
|
metaMap[ent->id] = TXTR::GetMetaData(rs);
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
});
|
|
|
|
|
|
|
|
athena::io::YAMLDocWriter yamlW("MP3TextureCache");
|
|
|
|
for (const auto& pair : metaMap) {
|
|
|
|
hecl::ProjectPath path = pakRouter.getWorking(pair.first);
|
|
|
|
auto rec = yamlW.enterSubRecord(path.getRelativePathUTF8());
|
|
|
|
pair.second.write(yamlW);
|
|
|
|
}
|
|
|
|
|
|
|
|
athena::io::FileWriter fileW(texturePath.getAbsolutePath());
|
|
|
|
yamlW.finish(&fileW);
|
2020-04-11 22:51:39 +00:00
|
|
|
Log.report(logvisor::Level::Info, FMT_STRING("Done..."));
|
2020-04-11 07:53:04 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static void Cook(const hecl::ProjectPath& inPath, const hecl::ProjectPath& outPath) {
|
|
|
|
hecl::Database::Project& project = inPath.getProject();
|
|
|
|
athena::io::YAMLDocReader r;
|
|
|
|
athena::io::FileReader fr(inPath.getAbsolutePath());
|
|
|
|
if (!fr.isOpen() || !r.parse(&fr))
|
|
|
|
return;
|
|
|
|
|
|
|
|
std::vector<std::pair<UniqueID32, TXTR::Meta>> metaPairs;
|
|
|
|
metaPairs.reserve(r.getRootNode()->m_mapChildren.size());
|
|
|
|
for (const auto& node : r.getRootNode()->m_mapChildren) {
|
|
|
|
hecl::ProjectPath projectPath(project, node.first);
|
|
|
|
auto rec = r.enterSubRecord(node.first.c_str());
|
|
|
|
TXTR::Meta meta;
|
|
|
|
meta.read(r);
|
|
|
|
metaPairs.emplace_back(projectPath.parsedHash32(), meta);
|
|
|
|
}
|
|
|
|
|
|
|
|
std::sort(metaPairs.begin(), metaPairs.end(), [](const auto& a, const auto& b) -> bool {
|
|
|
|
return a.first < b.first;
|
|
|
|
});
|
|
|
|
|
|
|
|
athena::io::FileWriter w(outPath.getAbsolutePath());
|
|
|
|
w.writeUint32Big(metaPairs.size());
|
|
|
|
for (const auto& pair : metaPairs) {
|
|
|
|
pair.first.write(w);
|
|
|
|
pair.second.write(w);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
struct SpecMP3 : SpecBase {
|
2020-04-15 13:42:44 +00:00
|
|
|
bool checkStandaloneID(const char* id) const override { return memcmp(id, "RM3", 3) == 0; }
|
2018-12-08 05:30:43 +00:00
|
|
|
|
|
|
|
bool doMP3 = false;
|
|
|
|
std::vector<const nod::Node*> m_nonPaks;
|
|
|
|
std::vector<DNAMP3::PAKBridge> m_paks;
|
|
|
|
std::map<std::string, DNAMP3::PAKBridge*, hecl::CaseInsensitiveCompare> m_orderedPaks;
|
|
|
|
|
|
|
|
hecl::ProjectPath m_workPath;
|
|
|
|
hecl::ProjectPath m_cookPath;
|
2020-09-17 23:08:45 +00:00
|
|
|
hecl::ProjectPath m_outPath;
|
2018-12-08 05:30:43 +00:00
|
|
|
PAKRouter<DNAMP3::PAKBridge> m_pakRouter;
|
|
|
|
|
|
|
|
/* These are populated when extracting MPT's frontend (uses MP3's DataSpec) */
|
|
|
|
bool doMPTFE = false;
|
|
|
|
std::vector<const nod::Node*> m_feNonPaks;
|
|
|
|
std::vector<DNAMP3::PAKBridge> m_fePaks;
|
|
|
|
std::map<std::string, DNAMP3::PAKBridge*, hecl::CaseInsensitiveCompare> m_feOrderedPaks;
|
|
|
|
|
|
|
|
hecl::ProjectPath m_feWorkPath;
|
|
|
|
hecl::ProjectPath m_feCookPath;
|
2020-09-17 23:08:45 +00:00
|
|
|
hecl::ProjectPath m_feOutPath;
|
2018-12-08 05:30:43 +00:00
|
|
|
PAKRouter<DNAMP3::PAKBridge> m_fePakRouter;
|
|
|
|
|
|
|
|
SpecMP3(const hecl::Database::DataSpecEntry* specEntry, hecl::Database::Project& project, bool pc)
|
|
|
|
: SpecBase(specEntry, project, pc)
|
|
|
|
, m_workPath(project.getProjectWorkingPath(), _SYS_STR("MP3"))
|
|
|
|
, m_cookPath(project.getProjectCookedPath(SpecEntMP3), _SYS_STR("MP3"))
|
|
|
|
, m_pakRouter(*this, m_workPath, m_cookPath)
|
|
|
|
, m_feWorkPath(project.getProjectWorkingPath(), _SYS_STR("fe"))
|
|
|
|
, m_feCookPath(project.getProjectCookedPath(SpecEntMP3), _SYS_STR("fe"))
|
2019-10-01 07:38:03 +00:00
|
|
|
, m_fePakRouter(*this, m_feWorkPath, m_feCookPath) {
|
2020-04-15 13:42:44 +00:00
|
|
|
m_game = EGame::MetroidPrime3;
|
|
|
|
SpecBase::setThreadProject();
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void buildPaks(nod::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::Node& child : root) {
|
|
|
|
bool isPak = false;
|
|
|
|
auto 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;
|
2019-10-01 07:38:03 +00:00
|
|
|
if (std::string(extit, lowerName.end()) == ".pak") {
|
2018-12-08 05:30:43 +00:00
|
|
|
/* 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) {
|
|
|
|
std::string lowerArg(hecl::SystemUTF8Conv(arg).str());
|
|
|
|
std::transform(lowerArg.begin(), lowerArg.end(), lowerArg.begin(), tolower);
|
|
|
|
if (!lowerArg.compare(0, lowerBase.size(), lowerBase))
|
|
|
|
good = true;
|
|
|
|
}
|
2015-07-13 06:29:12 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
2015-07-18 04:33:38 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
if (fe)
|
|
|
|
m_fePaks.emplace_back(child, good);
|
|
|
|
else
|
|
|
|
m_paks.emplace_back(child, good);
|
2015-07-13 06:29:12 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
2015-07-13 06:29:12 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
if (!isPak) {
|
2015-07-18 04:33:38 +00:00
|
|
|
if (fe)
|
2018-12-08 05:30:43 +00:00
|
|
|
m_feNonPaks.push_back(&child);
|
2015-07-18 04:33:38 +00:00
|
|
|
else
|
2018-12-08 05:30:43 +00:00
|
|
|
m_nonPaks.push_back(&child);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Sort PAKs alphabetically */
|
|
|
|
if (fe) {
|
|
|
|
m_feOrderedPaks.clear();
|
|
|
|
for (DNAMP3::PAKBridge& dpak : m_fePaks)
|
|
|
|
m_feOrderedPaks[std::string(dpak.getName())] = &dpak;
|
|
|
|
} else {
|
|
|
|
m_orderedPaks.clear();
|
|
|
|
for (DNAMP3::PAKBridge& dpak : m_paks)
|
|
|
|
m_orderedPaks[std::string(dpak.getName())] = &dpak;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Assemble extract report */
|
2020-03-31 16:28:40 +00:00
|
|
|
for (const auto& item : fe ? m_feOrderedPaks : m_orderedPaks) {
|
2018-12-08 05:30:43 +00:00
|
|
|
if (!item.second->m_doExtract)
|
|
|
|
continue;
|
2020-03-31 16:32:01 +00:00
|
|
|
|
|
|
|
ExtractReport& childRep = rep.childOpts.emplace_back();
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::SystemStringConv nameView(item.first);
|
|
|
|
childRep.name = hecl::SystemString(nameView.sys_str());
|
2019-10-01 07:38:03 +00:00
|
|
|
if (item.first == "Worlds.pak")
|
2018-12-08 05:30:43 +00:00
|
|
|
continue;
|
2019-10-01 07:38:03 +00:00
|
|
|
else if (item.first == "Metroid6.pak") {
|
2018-12-08 05:30:43 +00:00
|
|
|
/* Phaaze doesn't have a world name D: */
|
|
|
|
childRep.desc = _SYS_STR("Phaaze");
|
|
|
|
continue;
|
2019-10-01 07:38:03 +00:00
|
|
|
} else if (item.first == "Metroid8.pak") {
|
2018-12-08 05:30:43 +00:00
|
|
|
/* Space world is misnamed */
|
|
|
|
childRep.desc = _SYS_STR("Space");
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
childRep.desc = item.second->getLevelString();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
bool checkFromStandaloneDisc(nod::DiscBase& disc, const hecl::SystemString& regstr,
|
2019-08-10 02:32:48 +00:00
|
|
|
const std::vector<hecl::SystemString>& args, std::vector<ExtractReport>& reps) override {
|
2018-12-08 05:30:43 +00:00
|
|
|
doMP3 = true;
|
|
|
|
nod::IPartition* partition = disc.getDataPartition();
|
|
|
|
std::unique_ptr<uint8_t[]> dolBuf = partition->getDOLBuf();
|
2020-04-15 13:42:44 +00:00
|
|
|
const char* buildInfo = static_cast<char*>(memmem(dolBuf.get(), partition->getDOLSize(), "MetroidBuildInfo", 16)) + 19;
|
|
|
|
if (buildInfo == nullptr) {
|
2018-12-08 05:30:43 +00:00
|
|
|
return false;
|
2020-04-15 13:42:44 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
|
|
|
|
/* We don't want no stinking demo dammit */
|
2020-04-15 13:42:44 +00:00
|
|
|
if (strcmp(buildInfo, "Build v3.068 3/2/2006 14:55:13") == 0) {
|
2018-12-08 05:30:43 +00:00
|
|
|
return false;
|
2020-04-15 13:42:44 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
|
2020-04-15 07:56:59 +00:00
|
|
|
m_version = std::string(buildInfo);
|
2018-12-08 05:30:43 +00:00
|
|
|
/* Root Report */
|
2020-03-31 16:32:01 +00:00
|
|
|
ExtractReport& rep = reps.emplace_back();
|
2018-12-08 05:30:43 +00:00
|
|
|
rep.name = _SYS_STR("MP3");
|
|
|
|
rep.desc = _SYS_STR("Metroid Prime 3 ") + regstr;
|
2020-04-15 07:56:59 +00:00
|
|
|
hecl::SystemStringConv buildView(m_version);
|
2018-12-08 05:30:43 +00:00
|
|
|
rep.desc += _SYS_STR(" (") + buildView + _SYS_STR(")");
|
|
|
|
|
|
|
|
/* Iterate PAKs and build level options */
|
|
|
|
nod::Node& root = partition->getFSTRoot();
|
|
|
|
buildPaks(root, args, rep, false);
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool checkFromTrilogyDisc(nod::DiscBase& disc, const hecl::SystemString& regstr,
|
2019-08-10 02:32:48 +00:00
|
|
|
const std::vector<hecl::SystemString>& args, std::vector<ExtractReport>& reps) override {
|
2018-12-08 05:30:43 +00:00
|
|
|
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, _SYS_STR("mp3"))) {
|
|
|
|
doMP3 = true;
|
|
|
|
mp3args.reserve(args.size());
|
|
|
|
size_t slashPos = arg.find(_SYS_STR('/'));
|
|
|
|
if (slashPos == hecl::SystemString::npos)
|
|
|
|
slashPos = arg.find(_SYS_STR('\\'));
|
|
|
|
if (slashPos != hecl::SystemString::npos)
|
|
|
|
mp3args.emplace_back(hecl::SystemString(arg.begin() + slashPos + 1, arg.end()));
|
2015-07-18 04:33:38 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
for (const hecl::SystemString& arg : args) {
|
|
|
|
hecl::SystemString lowerArg = arg;
|
|
|
|
hecl::ToLower(lowerArg);
|
|
|
|
if (!lowerArg.compare(0, 2, _SYS_STR("fe"))) {
|
|
|
|
doMPTFE = true;
|
|
|
|
feargs.reserve(args.size());
|
|
|
|
size_t slashPos = arg.find(_SYS_STR('/'));
|
|
|
|
if (slashPos == hecl::SystemString::npos)
|
|
|
|
slashPos = arg.find(_SYS_STR('\\'));
|
|
|
|
if (slashPos != hecl::SystemString::npos)
|
|
|
|
feargs.emplace_back(hecl::SystemString(arg.begin() + slashPos + 1, arg.end()));
|
2015-07-13 06:29:12 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
doMP3 = true;
|
|
|
|
doMPTFE = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!doMP3 && !doMPTFE)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
nod::IPartition* partition = disc.getDataPartition();
|
|
|
|
nod::Node& root = partition->getFSTRoot();
|
|
|
|
|
|
|
|
/* MP3 extract */
|
|
|
|
while (doMP3) {
|
|
|
|
nod::Node::DirectoryIterator dolIt = root.find("rs5mp3_p.dol");
|
|
|
|
if (dolIt == root.end()) {
|
|
|
|
doMP3 = false;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::unique_ptr<uint8_t[]> dolBuf = dolIt->getBuf();
|
|
|
|
const char* buildInfo = (char*)memmem(dolBuf.get(), dolIt->size(), "MetroidBuildInfo", 16) + 19;
|
|
|
|
|
|
|
|
if (!buildInfo) {
|
|
|
|
doMP3 = false;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* We don't want no stinking demo dammit */
|
|
|
|
if (!strcmp(buildInfo, "Build v3.068 3/2/2006 14:55:13")) {
|
|
|
|
doMP3 = false;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Root Report */
|
2020-03-31 16:32:01 +00:00
|
|
|
ExtractReport& rep = reps.emplace_back();
|
2018-12-08 05:30:43 +00:00
|
|
|
rep.name = _SYS_STR("MP3");
|
|
|
|
rep.desc = _SYS_STR("Metroid Prime 3 ") + regstr;
|
|
|
|
|
2020-04-15 13:42:44 +00:00
|
|
|
m_version = std::string(buildInfo);
|
|
|
|
hecl::SystemStringConv buildView(m_version);
|
2018-12-08 05:30:43 +00:00
|
|
|
rep.desc += _SYS_STR(" (") + buildView + _SYS_STR(")");
|
|
|
|
|
|
|
|
/* Iterate PAKs and build level options */
|
|
|
|
nod::Node::DirectoryIterator mp3It = root.find("MP3");
|
|
|
|
if (mp3It == root.end()) {
|
|
|
|
doMP3 = false;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
buildPaks(*mp3It, mp3args, rep, false);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* MPT Frontend extract */
|
|
|
|
while (doMPTFE) {
|
|
|
|
nod::Node::DirectoryIterator dolIt = root.find("rs5fe_p.dol");
|
|
|
|
if (dolIt == root.end()) {
|
|
|
|
doMPTFE = false;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::unique_ptr<uint8_t[]> dolBuf = dolIt->getBuf();
|
|
|
|
const char* buildInfo = (char*)memmem(dolBuf.get(), dolIt->size(), "MetroidBuildInfo", 16) + 19;
|
|
|
|
|
|
|
|
/* Root Report */
|
2020-03-31 16:32:01 +00:00
|
|
|
ExtractReport& rep = reps.emplace_back();
|
2018-12-08 05:30:43 +00:00
|
|
|
rep.name = _SYS_STR("fe");
|
|
|
|
rep.desc = _SYS_STR("Metroid Prime Trilogy Frontend ") + regstr;
|
|
|
|
if (buildInfo) {
|
2015-09-24 07:47:30 +00:00
|
|
|
std::string buildStr(buildInfo);
|
2017-11-13 06:19:18 +00:00
|
|
|
hecl::SystemStringConv buildView(buildStr);
|
2018-10-14 20:16:21 +00:00
|
|
|
rep.desc += _SYS_STR(" (") + buildView + _SYS_STR(")");
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Iterate PAKs and build level options */
|
|
|
|
nod::Node::DirectoryIterator feIt = root.find("fe");
|
|
|
|
if (feIt == root.end()) {
|
|
|
|
doMPTFE = false;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
buildPaks(*feIt, feargs, rep, true);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
return doMP3 || doMPTFE;
|
|
|
|
}
|
|
|
|
|
2019-08-10 02:32:48 +00:00
|
|
|
bool extractFromDisc(nod::DiscBase& disc, bool force, const hecl::MultiProgressPrinter& progress) override {
|
2020-03-31 16:30:00 +00:00
|
|
|
hecl::SystemString currentTarget;
|
2018-12-08 05:30:43 +00:00
|
|
|
size_t nodeCount = 0;
|
|
|
|
int prog = 0;
|
|
|
|
nod::ExtractionContext ctx = {force, [&](std::string_view name, float) {
|
|
|
|
hecl::SystemStringConv nameView(name);
|
|
|
|
progress.print(currentTarget.c_str(), nameView.c_str(), prog / (float)nodeCount);
|
|
|
|
}};
|
|
|
|
if (doMP3) {
|
|
|
|
m_workPath.makeDir();
|
|
|
|
|
|
|
|
progress.startNewLine();
|
|
|
|
progress.print(_SYS_STR("Indexing PAKs"), _SYS_STR(""), 0.0);
|
|
|
|
m_pakRouter.build(m_paks,
|
|
|
|
[&progress](float factor) { progress.print(_SYS_STR("Indexing PAKs"), _SYS_STR(""), factor); });
|
|
|
|
progress.print(_SYS_STR("Indexing PAKs"), _SYS_STR(""), 1.0);
|
|
|
|
progress.startNewLine();
|
|
|
|
|
|
|
|
hecl::ProjectPath outPath(m_project.getProjectWorkingPath(), _SYS_STR("out"));
|
|
|
|
outPath.makeDir();
|
|
|
|
disc.getDataPartition()->extractSysFiles(outPath.getAbsolutePath(), ctx);
|
2020-09-17 23:08:45 +00:00
|
|
|
m_outPath = {outPath, m_standalone ? _SYS_STR("files") : _SYS_STR("files/MP3")};
|
|
|
|
m_outPath.makeDirChain(true);
|
2018-12-08 05:30:43 +00:00
|
|
|
|
|
|
|
currentTarget = _SYS_STR("MP3 Root");
|
|
|
|
progress.print(currentTarget.c_str(), _SYS_STR(""), 0.0);
|
|
|
|
prog = 0;
|
|
|
|
|
|
|
|
nodeCount = m_nonPaks.size();
|
|
|
|
// TODO: Make this more granular
|
|
|
|
for (const nod::Node* node : m_nonPaks) {
|
2020-09-17 23:08:45 +00:00
|
|
|
node->extractToDirectory(m_outPath.getAbsolutePath(), ctx);
|
2018-12-08 05:30:43 +00:00
|
|
|
prog++;
|
|
|
|
}
|
|
|
|
ctx.progressCB = nullptr;
|
|
|
|
|
|
|
|
progress.print(currentTarget.c_str(), _SYS_STR(""), 1.0);
|
|
|
|
progress.startNewLine();
|
|
|
|
|
|
|
|
hecl::ClientProcess process;
|
|
|
|
for (std::pair<const std::string, DNAMP3::PAKBridge*>& pair : m_orderedPaks) {
|
|
|
|
DNAMP3::PAKBridge& pak = *pair.second;
|
|
|
|
if (!pak.m_doExtract)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
auto name = pak.getName();
|
|
|
|
hecl::SystemStringConv sysName(name);
|
|
|
|
|
|
|
|
auto pakName = hecl::SystemString(sysName.sys_str());
|
|
|
|
process.addLambdaTransaction([this, &progress, &pak, pakName, force](hecl::blender::Token& btok) {
|
|
|
|
m_pakRouter.extractResources(pak, force, btok,
|
|
|
|
[&progress, &pakName](const hecl::SystemChar* substr, float factor) {
|
|
|
|
progress.print(pakName.c_str(), substr, factor);
|
|
|
|
});
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
process.waitUntilComplete();
|
|
|
|
}
|
|
|
|
|
|
|
|
if (doMPTFE) {
|
|
|
|
m_feWorkPath.makeDir();
|
|
|
|
|
|
|
|
progress.startNewLine();
|
|
|
|
progress.print(_SYS_STR("Indexing PAKs"), _SYS_STR(""), 0.0);
|
|
|
|
m_fePakRouter.build(
|
|
|
|
m_fePaks, [&progress](float factor) { progress.print(_SYS_STR("Indexing PAKs"), _SYS_STR(""), factor); });
|
|
|
|
progress.print(_SYS_STR("Indexing PAKs"), _SYS_STR(""), 1.0);
|
|
|
|
progress.startNewLine();
|
|
|
|
|
|
|
|
hecl::ProjectPath outPath(m_project.getProjectWorkingPath(), _SYS_STR("out"));
|
|
|
|
outPath.makeDir();
|
|
|
|
disc.getDataPartition()->extractSysFiles(outPath.getAbsolutePath(), ctx);
|
2020-09-17 23:08:45 +00:00
|
|
|
m_feOutPath = {outPath, _SYS_STR("files/fe")};
|
|
|
|
m_feOutPath.makeDirChain(true);
|
2018-12-08 05:30:43 +00:00
|
|
|
|
|
|
|
currentTarget = _SYS_STR("fe Root");
|
|
|
|
progress.print(currentTarget.c_str(), _SYS_STR(""), 0.0);
|
|
|
|
prog = 0;
|
|
|
|
nodeCount = m_feNonPaks.size();
|
|
|
|
|
|
|
|
// TODO: Make this more granular
|
|
|
|
for (const nod::Node* node : m_feNonPaks) {
|
2020-09-17 23:08:45 +00:00
|
|
|
node->extractToDirectory(m_feOutPath.getAbsolutePath(), ctx);
|
2018-12-08 05:30:43 +00:00
|
|
|
prog++;
|
|
|
|
}
|
|
|
|
progress.print(currentTarget.c_str(), _SYS_STR(""), 1.0);
|
|
|
|
progress.startNewLine();
|
|
|
|
|
|
|
|
hecl::ClientProcess process;
|
2020-03-31 16:28:40 +00:00
|
|
|
for (auto& pair : m_feOrderedPaks) {
|
2018-12-08 05:30:43 +00:00
|
|
|
DNAMP3::PAKBridge& pak = *pair.second;
|
|
|
|
if (!pak.m_doExtract)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
auto name = pak.getName();
|
|
|
|
hecl::SystemStringConv sysName(name);
|
|
|
|
|
|
|
|
hecl::SystemString pakName(sysName.sys_str());
|
|
|
|
process.addLambdaTransaction([this, &progress, &pak, pakName, force](hecl::blender::Token& btok) {
|
|
|
|
m_fePakRouter.extractResources(pak, force, btok,
|
|
|
|
[&progress, &pakName](const hecl::SystemChar* substr, float factor) {
|
|
|
|
progress.print(pakName.c_str(), substr, factor);
|
|
|
|
});
|
|
|
|
});
|
|
|
|
}
|
2015-07-13 06:29:12 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
process.waitUntilComplete();
|
2020-04-15 07:56:59 +00:00
|
|
|
}
|
2020-04-11 07:53:04 +00:00
|
|
|
|
2020-04-15 07:56:59 +00:00
|
|
|
/* Generate Texture Cache containing meta data for every texture file */
|
2020-09-17 23:08:45 +00:00
|
|
|
if (doMP3) {
|
|
|
|
hecl::ProjectPath noAramPath(m_workPath, _SYS_STR("URDE"));
|
|
|
|
TextureCache::Generate(m_pakRouter, m_project, noAramPath);
|
|
|
|
}
|
|
|
|
if (doMPTFE) {
|
|
|
|
hecl::ProjectPath noAramPath(m_feWorkPath, _SYS_STR("URDE"));
|
|
|
|
TextureCache::Generate(m_fePakRouter, m_project, noAramPath);
|
|
|
|
}
|
2020-04-15 07:56:59 +00:00
|
|
|
/* Write version data */
|
2020-09-17 23:08:45 +00:00
|
|
|
if (doMP3) {
|
|
|
|
WriteVersionInfo(m_project, m_outPath);
|
|
|
|
}
|
|
|
|
if (doMPTFE) {
|
|
|
|
WriteVersionInfo(m_project, m_feOutPath);
|
2015-11-10 02:07:15 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
return true;
|
|
|
|
}
|
2015-11-10 02:07:15 +00:00
|
|
|
|
2019-08-10 02:32:48 +00:00
|
|
|
const hecl::Database::DataSpecEntry& getOriginalSpec() const override { return SpecEntMP3; }
|
2015-10-04 05:08:56 +00:00
|
|
|
|
2019-08-10 02:32:48 +00:00
|
|
|
const hecl::Database::DataSpecEntry& getUnmodifiedSpec() const override { return SpecEntMP3ORIG; }
|
2015-10-04 05:08:56 +00:00
|
|
|
|
2019-08-10 02:32:48 +00:00
|
|
|
hecl::ProjectPath getWorking(class UniqueID64& id) override { return m_pakRouter.getWorking(id); }
|
2016-10-02 22:41:36 +00:00
|
|
|
|
2019-08-10 02:32:48 +00:00
|
|
|
bool checkPathPrefix(const hecl::ProjectPath& path) const override {
|
2018-12-08 05:30:43 +00:00
|
|
|
return path.getRelativePath().compare(0, 4, _SYS_STR("MP3/")) == 0;
|
|
|
|
}
|
2015-10-04 05:08:56 +00:00
|
|
|
|
2019-08-10 02:32:48 +00:00
|
|
|
bool validateYAMLDNAType(athena::io::IStreamReader& fp) const override {
|
2018-12-08 05:30:43 +00:00
|
|
|
if (athena::io::ValidateFromYAMLStream<DNAMP3::MLVL>(fp))
|
|
|
|
return true;
|
|
|
|
if (athena::io::ValidateFromYAMLStream<DNAMP3::STRG>(fp))
|
|
|
|
return true;
|
|
|
|
if (athena::io::ValidateFromYAMLStream<DNAMP2::STRG>(fp))
|
|
|
|
return true;
|
|
|
|
return false;
|
|
|
|
}
|
2017-10-17 05:51:53 +00:00
|
|
|
|
2019-08-10 02:32:48 +00:00
|
|
|
urde::SObjectTag buildTagFromPath(const hecl::ProjectPath& path) const override { return {}; }
|
2018-02-25 08:23:27 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
void cookMesh(const hecl::ProjectPath& out, const hecl::ProjectPath& in, BlendStream& ds, bool fast,
|
2019-08-10 02:32:48 +00:00
|
|
|
hecl::blender::Token& btok, FCookProgress progress) override {}
|
2015-10-04 05:08:56 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
void cookColMesh(const hecl::ProjectPath& out, const hecl::ProjectPath& in, BlendStream& ds, bool fast,
|
2019-08-10 02:32:48 +00:00
|
|
|
hecl::blender::Token& btok, FCookProgress progress) override {}
|
2015-10-04 05:08:56 +00:00
|
|
|
|
2019-10-01 07:38:03 +00:00
|
|
|
void cookArmature(const hecl::ProjectPath& out, const hecl::ProjectPath& in, BlendStream& ds, bool fast,
|
|
|
|
hecl::blender::Token& btok, FCookProgress progress) override {}
|
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
void cookPathMesh(const hecl::ProjectPath& out, const hecl::ProjectPath& in, BlendStream& ds, bool fast,
|
2019-08-10 02:32:48 +00:00
|
|
|
hecl::blender::Token& btok, FCookProgress progress) override {}
|
2016-09-30 22:43:19 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
void cookActor(const hecl::ProjectPath& out, const hecl::ProjectPath& in, BlendStream& ds, bool fast,
|
2019-08-10 02:32:48 +00:00
|
|
|
hecl::blender::Token& btok, FCookProgress progress) override {}
|
2017-01-17 01:23:19 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
void cookArea(const hecl::ProjectPath& out, const hecl::ProjectPath& in, BlendStream& ds, bool fast,
|
2019-08-10 02:32:48 +00:00
|
|
|
hecl::blender::Token& btok, FCookProgress progress) override {}
|
2016-09-18 23:47:48 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
void cookWorld(const hecl::ProjectPath& out, const hecl::ProjectPath& in, BlendStream& ds, bool fast,
|
2019-08-10 02:32:48 +00:00
|
|
|
hecl::blender::Token& btok, FCookProgress progress) override {}
|
2016-10-02 22:41:36 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
void cookGuiFrame(const hecl::ProjectPath& out, const hecl::ProjectPath& in, BlendStream& ds,
|
2019-08-10 02:32:48 +00:00
|
|
|
hecl::blender::Token& btok, FCookProgress progress) override {}
|
2017-11-15 04:12:13 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
void cookYAML(const hecl::ProjectPath& out, const hecl::ProjectPath& in, athena::io::IStreamReader& fin,
|
2019-10-01 07:38:03 +00:00
|
|
|
hecl::blender::Token& btok, FCookProgress progress) override {}
|
2016-09-18 23:47:48 +00:00
|
|
|
|
2019-08-10 02:32:48 +00:00
|
|
|
void flattenDependenciesYAML(athena::io::IStreamReader& fin, std::vector<hecl::ProjectPath>& pathsOut) override {}
|
2017-03-20 05:09:53 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
void flattenDependenciesANCSYAML(athena::io::IStreamReader& fin, std::vector<hecl::ProjectPath>& pathsOut,
|
2019-08-10 02:32:48 +00:00
|
|
|
int charIdx) override {}
|
2017-03-20 05:09:53 +00:00
|
|
|
|
2019-08-10 02:32:48 +00:00
|
|
|
void cookAudioGroup(const hecl::ProjectPath& out, const hecl::ProjectPath& in, FCookProgress progress) override {}
|
2018-07-08 23:03:35 +00:00
|
|
|
|
2019-08-10 02:32:48 +00:00
|
|
|
void cookSong(const hecl::ProjectPath& out, const hecl::ProjectPath& in, FCookProgress progress) override {}
|
2018-07-08 23:03:35 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
void cookMapArea(const hecl::ProjectPath& out, const hecl::ProjectPath& in, BlendStream& ds,
|
2019-08-10 02:32:48 +00:00
|
|
|
hecl::blender::Token& btok, FCookProgress progress) override {
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::blender::MapArea mapa = ds.compileMapArea();
|
|
|
|
ds.close();
|
|
|
|
DNAMP3::MAPA::Cook(mapa, out);
|
|
|
|
progress(_SYS_STR("Done"));
|
|
|
|
}
|
2018-07-08 23:03:35 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
void cookMapUniverse(const hecl::ProjectPath& out, const hecl::ProjectPath& in, BlendStream& ds,
|
2019-08-10 02:32:48 +00:00
|
|
|
hecl::blender::Token& btok, FCookProgress progress) override {}
|
2017-01-17 01:23:19 +00:00
|
|
|
};
|
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::Database::DataSpecEntry SpecEntMP3(
|
2019-05-08 03:50:21 +00:00
|
|
|
_SYS_STR("MP3"sv), _SYS_STR("Data specification for original Metroid Prime 3 engine"sv), _SYS_STR(".pak"sv),
|
2018-12-08 05:30:43 +00:00
|
|
|
[](hecl::Database::Project& project, hecl::Database::DataSpecTool) -> std::unique_ptr<hecl::Database::IDataSpec> {
|
|
|
|
return std::make_unique<SpecMP3>(&SpecEntMP3, project, false);
|
|
|
|
});
|
|
|
|
|
|
|
|
hecl::Database::DataSpecEntry SpecEntMP3PC = {
|
|
|
|
_SYS_STR("MP3-PC"sv), _SYS_STR("Data specification for PC-optimized Metroid Prime 3 engine"sv), _SYS_STR(".upak"sv),
|
|
|
|
[](hecl::Database::Project& project,
|
|
|
|
hecl::Database::DataSpecTool tool) -> std::unique_ptr<hecl::Database::IDataSpec> {
|
|
|
|
if (tool != hecl::Database::DataSpecTool::Extract)
|
|
|
|
return std::make_unique<SpecMP3>(&SpecEntMP3PC, project, true);
|
|
|
|
return nullptr;
|
|
|
|
}};
|
|
|
|
|
|
|
|
hecl::Database::DataSpecEntry SpecEntMP3ORIG = {
|
2019-05-08 03:50:21 +00:00
|
|
|
_SYS_STR("MP3-ORIG"sv), _SYS_STR("Data specification for unmodified Metroid Prime 3 resources"sv), {}, {}};
|
2018-12-08 05:30:43 +00:00
|
|
|
|
|
|
|
} // namespace DataSpec
|