2016-03-02 07:29:19 +00:00
|
|
|
#include "PAK.hpp"
|
|
|
|
#include "../DNAMP1/DNAMP1.hpp"
|
|
|
|
#include "../DNAMP2/DNAMP2.hpp"
|
|
|
|
#include "../DNAMP3/DNAMP3.hpp"
|
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
namespace DataSpec {
|
2016-03-02 07:29:19 +00:00
|
|
|
|
|
|
|
template <class PAKBRIDGE>
|
2018-12-08 05:30:43 +00:00
|
|
|
void UniqueResult::checkEntry(const PAKBRIDGE& pakBridge, const typename PAKBRIDGE::PAKType::Entry& entry) {
|
|
|
|
UniqueResult::Type resultType = UniqueResult::Type::NotFound;
|
|
|
|
bool foundOneLayer = false;
|
|
|
|
const hecl::SystemString* levelName = nullptr;
|
|
|
|
typename PAKBRIDGE::PAKType::IDType levelId;
|
|
|
|
typename PAKBRIDGE::PAKType::IDType areaId;
|
|
|
|
unsigned layerIdx;
|
|
|
|
for (const auto& lpair : pakBridge.m_levelDeps) {
|
|
|
|
if (entry.id == lpair.first) {
|
|
|
|
levelName = &lpair.second.name;
|
|
|
|
resultType = UniqueResult::Type::Level;
|
|
|
|
break;
|
|
|
|
}
|
2016-03-02 07:29:19 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
for (const auto& pair : lpair.second.areas) {
|
|
|
|
unsigned l = 0;
|
|
|
|
for (const auto& layer : pair.second.layers) {
|
|
|
|
if (layer.resources.find(entry.id) != layer.resources.end()) {
|
|
|
|
if (foundOneLayer) {
|
|
|
|
if (areaId == pair.first) {
|
|
|
|
resultType = UniqueResult::Type::Area;
|
|
|
|
} else if (levelId == lpair.first) {
|
|
|
|
resultType = UniqueResult::Type::Level;
|
|
|
|
break;
|
|
|
|
} else {
|
|
|
|
m_type = UniqueResult::Type::Pak;
|
|
|
|
return;
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
continue;
|
|
|
|
} else
|
|
|
|
resultType = UniqueResult::Type::Layer;
|
|
|
|
levelName = &lpair.second.name;
|
|
|
|
levelId = lpair.first;
|
|
|
|
areaId = pair.first;
|
|
|
|
layerIdx = l;
|
|
|
|
foundOneLayer = true;
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
++l;
|
|
|
|
}
|
|
|
|
if (pair.second.resources.find(entry.id) != pair.second.resources.end()) {
|
|
|
|
if (foundOneLayer) {
|
|
|
|
if (areaId == pair.first) {
|
|
|
|
resultType = UniqueResult::Type::Area;
|
|
|
|
} else if (levelId == lpair.first) {
|
|
|
|
resultType = UniqueResult::Type::Level;
|
|
|
|
break;
|
|
|
|
} else {
|
|
|
|
m_type = UniqueResult::Type::Pak;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
continue;
|
|
|
|
} else
|
|
|
|
resultType = UniqueResult::Type::Area;
|
|
|
|
levelName = &lpair.second.name;
|
|
|
|
levelId = lpair.first;
|
|
|
|
areaId = pair.first;
|
|
|
|
foundOneLayer = true;
|
|
|
|
}
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
|
|
|
m_type = resultType;
|
|
|
|
m_levelName = levelName;
|
|
|
|
if (resultType == UniqueResult::Type::Layer || resultType == UniqueResult::Type::Area) {
|
|
|
|
const typename PAKBRIDGE::Level::Area& area = pakBridge.m_levelDeps.at(levelId).areas.at(areaId);
|
|
|
|
m_areaName = &area.name;
|
|
|
|
if (resultType == UniqueResult::Type::Layer) {
|
|
|
|
const typename PAKBRIDGE::Level::Area::Layer& layer = area.layers[layerIdx];
|
|
|
|
m_layerName = &layer.name;
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template void UniqueResult::checkEntry(const DNAMP1::PAKBridge& pakBridge,
|
|
|
|
const DNAMP1::PAKBridge::PAKType::Entry& entry);
|
|
|
|
template void UniqueResult::checkEntry(const DNAMP2::PAKBridge& pakBridge,
|
|
|
|
const DNAMP2::PAKBridge::PAKType::Entry& entry);
|
|
|
|
template void UniqueResult::checkEntry(const DNAMP3::PAKBridge& pakBridge,
|
|
|
|
const DNAMP3::PAKBridge::PAKType::Entry& entry);
|
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::ProjectPath UniqueResult::uniquePath(const hecl::ProjectPath& pakPath) const {
|
|
|
|
if (m_type == Type::Pak)
|
|
|
|
return pakPath;
|
|
|
|
|
|
|
|
hecl::ProjectPath levelDir;
|
|
|
|
if (m_levelName)
|
|
|
|
levelDir.assign(pakPath, *m_levelName);
|
|
|
|
else
|
|
|
|
levelDir = pakPath;
|
|
|
|
|
|
|
|
if (m_type == Type::Area) {
|
|
|
|
hecl::ProjectPath areaDir(levelDir, *m_areaName);
|
|
|
|
return areaDir;
|
|
|
|
} else if (m_type == Type::Layer) {
|
|
|
|
hecl::ProjectPath areaDir(levelDir, *m_areaName);
|
|
|
|
hecl::ProjectPath layerDir(areaDir, *m_layerName);
|
|
|
|
return layerDir;
|
|
|
|
}
|
|
|
|
|
|
|
|
return levelDir;
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
void PAKRouter<BRIDGETYPE>::build(std::vector<BRIDGETYPE>& bridges, std::function<void(float)> progress) {
|
|
|
|
m_bridges = &bridges;
|
|
|
|
m_bridgePaths.clear();
|
|
|
|
|
|
|
|
m_uniqueEntries.clear();
|
|
|
|
m_sharedEntries.clear();
|
|
|
|
m_charAssoc.m_cmdlRigs.clear();
|
|
|
|
size_t count = 0;
|
|
|
|
float bridgesSz = bridges.size();
|
|
|
|
|
|
|
|
/* Route entries unique/shared per-pak */
|
|
|
|
size_t bridgeIdx = 0;
|
|
|
|
for (BRIDGETYPE& bridge : bridges) {
|
|
|
|
const auto& name = bridge.getName();
|
|
|
|
hecl::SystemStringConv sysName(name);
|
|
|
|
|
|
|
|
hecl::SystemStringView::const_iterator extit = sysName.sys_str().end() - 4;
|
|
|
|
hecl::SystemString baseName(sysName.sys_str().begin(), extit);
|
|
|
|
|
|
|
|
m_bridgePaths.emplace_back(
|
|
|
|
std::make_pair(hecl::ProjectPath(m_gameWorking, baseName), hecl::ProjectPath(m_gameCooked, baseName)));
|
|
|
|
|
|
|
|
/* Index this PAK */
|
|
|
|
bridge.build();
|
|
|
|
|
|
|
|
/* Add to global entry lookup */
|
|
|
|
const typename BRIDGETYPE::PAKType& pak = bridge.getPAK();
|
|
|
|
for (const auto& entry : pak.m_entries) {
|
|
|
|
if (!pak.m_noShare) {
|
|
|
|
auto sSearch = m_sharedEntries.find(entry.first);
|
|
|
|
if (sSearch != m_sharedEntries.end())
|
|
|
|
continue;
|
|
|
|
auto uSearch = m_uniqueEntries.find(entry.first);
|
|
|
|
if (uSearch != m_uniqueEntries.end()) {
|
|
|
|
m_uniqueEntries.erase(uSearch);
|
|
|
|
m_sharedEntries[entry.first] = std::make_pair(bridgeIdx, &entry.second);
|
|
|
|
} else
|
|
|
|
m_uniqueEntries[entry.first] = std::make_pair(bridgeIdx, &entry.second);
|
|
|
|
} else
|
|
|
|
m_uniqueEntries[entry.first] = std::make_pair(bridgeIdx, &entry.second);
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
/* Add RigPairs to global map */
|
|
|
|
bridge.addCMDLRigPairs(*this, m_charAssoc);
|
|
|
|
|
|
|
|
progress(++count / bridgesSz);
|
|
|
|
++bridgeIdx;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Add named resources to catalog YAML files */
|
|
|
|
for (BRIDGETYPE& bridge : bridges) {
|
|
|
|
athena::io::YAMLDocWriter catalogWriter(nullptr);
|
|
|
|
|
|
|
|
enterPAKBridge(bridge);
|
|
|
|
|
|
|
|
/* Add MAPA transforms to global map */
|
|
|
|
bridge.addMAPATransforms(*this, m_mapaTransforms, m_overrideEntries);
|
|
|
|
|
|
|
|
const typename BRIDGETYPE::PAKType& pak = bridge.getPAK();
|
|
|
|
for (const auto& namedEntry : pak.m_nameEntries) {
|
|
|
|
if (namedEntry.name == "holo_cinf")
|
|
|
|
continue; /* Problematic corner case */
|
|
|
|
if (auto rec = catalogWriter.enterSubRecord(namedEntry.name.c_str())) {
|
|
|
|
hecl::ProjectPath working = getWorking(namedEntry.id);
|
|
|
|
if (working.getAuxInfoUTF8().size()) {
|
|
|
|
if (auto v = catalogWriter.enterSubVector(nullptr)) {
|
|
|
|
catalogWriter.writeString(nullptr, working.getRelativePathUTF8());
|
|
|
|
catalogWriter.writeString(nullptr, working.getAuxInfoUTF8());
|
|
|
|
}
|
|
|
|
} else
|
|
|
|
catalogWriter.writeString(nullptr, working.getRelativePathUTF8());
|
|
|
|
}
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
|
|
|
|
/* Write catalog */
|
|
|
|
intptr_t curBridgeIdx = reinterpret_cast<intptr_t>(m_curBridgeIdx.get());
|
|
|
|
const hecl::ProjectPath& pakPath = m_bridgePaths[curBridgeIdx].first;
|
|
|
|
pakPath.makeDirChain(true);
|
|
|
|
athena::io::FileWriter writer(hecl::ProjectPath(pakPath, "!catalog.yaml").getAbsolutePath());
|
|
|
|
catalogWriter.finish(&writer);
|
|
|
|
}
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
void PAKRouter<BRIDGETYPE>::enterPAKBridge(const BRIDGETYPE& pakBridge) {
|
|
|
|
g_PakRouter.reset(this);
|
|
|
|
auto pit = m_bridgePaths.begin();
|
|
|
|
size_t bridgeIdx = 0;
|
|
|
|
for (const BRIDGETYPE& bridge : *m_bridges) {
|
|
|
|
if (&bridge == &pakBridge) {
|
|
|
|
m_pak.reset(&pakBridge.getPAK());
|
|
|
|
m_node.reset(&pakBridge.getNode());
|
|
|
|
m_curBridgeIdx.reset(reinterpret_cast<void*>(bridgeIdx));
|
|
|
|
return;
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
++pit;
|
|
|
|
++bridgeIdx;
|
|
|
|
}
|
|
|
|
LogDNACommon.report(logvisor::Fatal, "PAKBridge provided to PAKRouter::enterPAKBridge() was not part of build()");
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
2016-09-19 05:29:05 +00:00
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::ProjectPath PAKRouter<BRIDGETYPE>::getCharacterWorking(const EntryType* entry) const {
|
|
|
|
auto characterSearch = m_charAssoc.m_cskrCinfToCharacter.find(entry->id);
|
|
|
|
if (characterSearch != m_charAssoc.m_cskrCinfToCharacter.cend()) {
|
|
|
|
hecl::ProjectPath characterPath = getWorking(characterSearch->second.first);
|
|
|
|
if (entry->type == FOURCC('EVNT')) {
|
|
|
|
hecl::SystemStringConv wideStr(characterSearch->second.second);
|
|
|
|
return characterPath.getWithExtension((hecl::SystemString(_SYS_STR(".")) + wideStr.c_str()).c_str(), true);
|
2016-09-19 05:29:05 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
return characterPath.ensureAuxInfo(characterSearch->second.second);
|
|
|
|
}
|
|
|
|
return {};
|
2016-09-19 05:29:05 +00:00
|
|
|
}
|
|
|
|
|
2016-03-02 07:29:19 +00:00
|
|
|
template <class BRIDGETYPE>
|
2016-03-05 00:03:41 +00:00
|
|
|
hecl::ProjectPath PAKRouter<BRIDGETYPE>::getWorking(const EntryType* entry,
|
2018-12-08 05:30:43 +00:00
|
|
|
const ResExtractor<BRIDGETYPE>& extractor) const {
|
|
|
|
if (!entry)
|
|
|
|
return hecl::ProjectPath();
|
2016-10-01 23:20:20 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
auto overrideSearch = m_overrideEntries.find(entry->id);
|
|
|
|
if (overrideSearch != m_overrideEntries.end())
|
|
|
|
return overrideSearch->second;
|
2016-10-01 23:20:20 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
const PAKType* pak = m_pak.get();
|
|
|
|
intptr_t curBridgeIdx = reinterpret_cast<intptr_t>(m_curBridgeIdx.get());
|
|
|
|
if (pak && pak->m_noShare) {
|
|
|
|
const EntryType* singleSearch = pak->lookupEntry(entry->id);
|
|
|
|
if (singleSearch) {
|
|
|
|
const hecl::ProjectPath& pakPath = m_bridgePaths[curBridgeIdx].first;
|
2016-03-02 07:29:19 +00:00
|
|
|
#if HECL_UCS2
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::SystemString entName = hecl::UTF8ToWide(getBestEntryName(*entry));
|
2016-03-02 07:29:19 +00:00
|
|
|
#else
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::SystemString entName = getBestEntryName(*entry);
|
2016-03-02 07:29:19 +00:00
|
|
|
#endif
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::SystemString auxInfo;
|
|
|
|
if (extractor.fileExts[0] && !extractor.fileExts[1])
|
|
|
|
entName += extractor.fileExts[0];
|
|
|
|
else if (extractor.fileExts[0])
|
|
|
|
entName += _SYS_STR(".*");
|
|
|
|
else if (hecl::ProjectPath chWork = getCharacterWorking(entry))
|
|
|
|
return chWork;
|
|
|
|
return hecl::ProjectPath(pakPath, entName).ensureAuxInfo(auxInfo);
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
2016-03-02 07:29:19 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
auto uniqueSearch = m_uniqueEntries.find(entry->id);
|
|
|
|
if (uniqueSearch != m_uniqueEntries.end()) {
|
|
|
|
const BRIDGETYPE& bridge = m_bridges->at(uniqueSearch->second.first);
|
|
|
|
const hecl::ProjectPath& pakPath = m_bridgePaths[uniqueSearch->second.first].first;
|
2016-03-02 07:29:19 +00:00
|
|
|
#if HECL_UCS2
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::SystemString entName = hecl::UTF8ToWide(getBestEntryName(*entry));
|
2016-03-02 07:29:19 +00:00
|
|
|
#else
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::SystemString entName = getBestEntryName(*entry);
|
2016-03-02 07:29:19 +00:00
|
|
|
#endif
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::SystemString auxInfo;
|
|
|
|
if (extractor.fileExts[0] && !extractor.fileExts[1])
|
|
|
|
entName += extractor.fileExts[0];
|
|
|
|
else if (extractor.fileExts[0])
|
|
|
|
entName += _SYS_STR(".*");
|
|
|
|
else if (hecl::ProjectPath chWork = getCharacterWorking(entry))
|
|
|
|
return chWork;
|
|
|
|
if (bridge.getPAK().m_noShare) {
|
|
|
|
return hecl::ProjectPath(pakPath, entName).ensureAuxInfo(auxInfo);
|
|
|
|
} else {
|
|
|
|
hecl::ProjectPath uniquePath = entry->unique.uniquePath(pakPath);
|
|
|
|
return hecl::ProjectPath(uniquePath, entName).ensureAuxInfo(auxInfo);
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
2016-03-02 07:29:19 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
auto sharedSearch = m_sharedEntries.find(entry->id);
|
|
|
|
if (sharedSearch != m_sharedEntries.end()) {
|
2016-03-02 07:29:19 +00:00
|
|
|
#if HECL_UCS2
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::SystemString entBase = hecl::UTF8ToWide(getBestEntryName(*entry));
|
2016-03-02 07:29:19 +00:00
|
|
|
#else
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::SystemString entBase = getBestEntryName(*entry);
|
2016-03-02 07:29:19 +00:00
|
|
|
#endif
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::SystemString auxInfo;
|
|
|
|
hecl::SystemString entName = entBase;
|
|
|
|
if (extractor.fileExts[0] && !extractor.fileExts[1])
|
|
|
|
entName += extractor.fileExts[0];
|
|
|
|
else if (extractor.fileExts[0])
|
|
|
|
entName += _SYS_STR(".*");
|
|
|
|
else if (hecl::ProjectPath chWork = getCharacterWorking(entry))
|
|
|
|
return chWork;
|
|
|
|
hecl::ProjectPath sharedPath(m_sharedWorking, entName);
|
|
|
|
return sharedPath.ensureAuxInfo(auxInfo);
|
|
|
|
}
|
|
|
|
|
|
|
|
LogDNACommon.report(logvisor::Fatal, "Unable to find entry %s", entry->id.toString().c_str());
|
|
|
|
return hecl::ProjectPath();
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::ProjectPath PAKRouter<BRIDGETYPE>::getWorking(const EntryType* entry) const {
|
|
|
|
if (!entry)
|
|
|
|
return hecl::ProjectPath();
|
|
|
|
return getWorking(entry, BRIDGETYPE::LookupExtractor(*m_node.get(), *m_pak.get(), *entry));
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::ProjectPath PAKRouter<BRIDGETYPE>::getWorking(const IDType& id, bool silenceWarnings) const {
|
|
|
|
return getWorking(lookupEntry(id, nullptr, silenceWarnings, false));
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::ProjectPath PAKRouter<BRIDGETYPE>::getCooked(const EntryType* entry) const {
|
|
|
|
if (!entry)
|
|
|
|
return hecl::ProjectPath();
|
2016-10-01 23:20:20 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
auto overrideSearch = m_overrideEntries.find(entry->id);
|
|
|
|
if (overrideSearch != m_overrideEntries.end()) {
|
|
|
|
return overrideSearch->second.getCookedPath(*m_dataSpec.overrideDataSpec(
|
2019-05-08 03:50:21 +00:00
|
|
|
overrideSearch->second, m_dataSpec.getDataSpecEntry()));
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
const PAKType* pak = m_pak.get();
|
|
|
|
intptr_t curBridgeIdx = reinterpret_cast<intptr_t>(m_curBridgeIdx.get());
|
|
|
|
if (pak && pak->m_noShare) {
|
|
|
|
const EntryType* singleSearch = pak->lookupEntry(entry->id);
|
|
|
|
if (singleSearch) {
|
|
|
|
const hecl::ProjectPath& pakPath = m_bridgePaths[curBridgeIdx].second;
|
|
|
|
return hecl::ProjectPath(pakPath, getBestEntryName(*entry));
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
|
|
|
auto uniqueSearch = m_uniqueEntries.find(entry->id);
|
|
|
|
if (uniqueSearch != m_uniqueEntries.end()) {
|
|
|
|
const BRIDGETYPE& bridge = m_bridges->at(uniqueSearch->second.first);
|
|
|
|
const hecl::ProjectPath& pakPath = m_bridgePaths[uniqueSearch->second.first].second;
|
|
|
|
if (bridge.getPAK().m_noShare) {
|
|
|
|
return hecl::ProjectPath(pakPath, getBestEntryName(*entry));
|
|
|
|
} else {
|
|
|
|
hecl::ProjectPath uniquePath = entry->unique.uniquePath(pakPath);
|
|
|
|
return hecl::ProjectPath(uniquePath, getBestEntryName(*entry));
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
|
|
|
auto sharedSearch = m_sharedEntries.find(entry->id);
|
|
|
|
if (sharedSearch != m_sharedEntries.end()) {
|
|
|
|
return hecl::ProjectPath(m_sharedCooked, getBestEntryName(*entry));
|
|
|
|
}
|
|
|
|
LogDNACommon.report(logvisor::Fatal, "Unable to find entry %s", entry->id.toString().c_str());
|
|
|
|
return hecl::ProjectPath();
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::ProjectPath PAKRouter<BRIDGETYPE>::getCooked(const IDType& id, bool silenceWarnings) const {
|
|
|
|
return getCooked(lookupEntry(id, nullptr, silenceWarnings, false));
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::SystemString PAKRouter<BRIDGETYPE>::getResourceRelativePath(const EntryType& a, const IDType& b) const {
|
|
|
|
const nod::Node* node = m_node.get();
|
|
|
|
const PAKType* pak = m_pak.get();
|
|
|
|
if (!pak)
|
|
|
|
LogDNACommon.report(logvisor::Fatal,
|
|
|
|
"PAKRouter::enterPAKBridge() must be called before PAKRouter::getResourceRelativePath()");
|
|
|
|
const typename BRIDGETYPE::PAKType::Entry* be = lookupEntry(b);
|
|
|
|
if (!be)
|
|
|
|
return hecl::SystemString();
|
|
|
|
hecl::ProjectPath aPath = getWorking(&a, BRIDGETYPE::LookupExtractor(*node, *pak, a));
|
|
|
|
hecl::SystemString ret;
|
|
|
|
for (int i = 0; i < aPath.levelCount(); ++i)
|
|
|
|
ret += _SYS_STR("../");
|
|
|
|
hecl::ProjectPath bPath = getWorking(be, BRIDGETYPE::LookupExtractor(*node, *pak, *be));
|
|
|
|
ret += bPath.getRelativePath();
|
|
|
|
return ret;
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
std::string PAKRouter<BRIDGETYPE>::getBestEntryName(const EntryType& entry, bool stdOverride) const {
|
|
|
|
std::string name;
|
|
|
|
for (const BRIDGETYPE& bridge : *m_bridges) {
|
|
|
|
const typename BRIDGETYPE::PAKType& pak = bridge.getPAK();
|
2018-12-31 05:01:42 +00:00
|
|
|
const typename BRIDGETYPE::PAKType::Entry* e = pak.lookupEntry(entry.id);
|
|
|
|
if (!e)
|
|
|
|
continue;
|
2018-12-08 05:30:43 +00:00
|
|
|
|
2018-12-31 05:01:42 +00:00
|
|
|
if (stdOverride && !pak.m_noShare) {
|
2018-12-08 05:30:43 +00:00
|
|
|
if (entry.type == FOURCC('MLVL'))
|
|
|
|
return "!world";
|
|
|
|
else if (entry.type == FOURCC('MREA'))
|
|
|
|
return "!area";
|
|
|
|
else if (entry.type == FOURCC('MAPA'))
|
|
|
|
return "!map";
|
|
|
|
else if (entry.type == FOURCC('PATH'))
|
|
|
|
return "!path";
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
|
|
|
|
bool named;
|
|
|
|
name = pak.bestEntryName(bridge.getNode(), entry, named);
|
|
|
|
if (named)
|
|
|
|
return name;
|
|
|
|
}
|
|
|
|
return name;
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
std::string PAKRouter<BRIDGETYPE>::getBestEntryName(const IDType& entry, bool stdOverride) const {
|
|
|
|
std::string name;
|
|
|
|
for (const BRIDGETYPE& bridge : *m_bridges) {
|
|
|
|
const typename BRIDGETYPE::PAKType& pak = bridge.getPAK();
|
|
|
|
const typename BRIDGETYPE::PAKType::Entry* e = pak.lookupEntry(entry);
|
|
|
|
if (!e)
|
|
|
|
continue;
|
|
|
|
|
2018-12-31 05:01:42 +00:00
|
|
|
if (stdOverride && !pak.m_noShare) {
|
2018-12-08 05:30:43 +00:00
|
|
|
if (e->type == FOURCC('MLVL'))
|
|
|
|
return "!world";
|
|
|
|
else if (e->type == FOURCC('MREA'))
|
|
|
|
return "!area";
|
|
|
|
else if (e->type == FOURCC('MAPA'))
|
|
|
|
return "!map";
|
|
|
|
else if (e->type == FOURCC('PATH'))
|
|
|
|
return "!path";
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
|
|
|
|
bool named;
|
|
|
|
name = pak.bestEntryName(bridge.getNode(), *e, named);
|
|
|
|
if (named)
|
|
|
|
return name;
|
|
|
|
}
|
|
|
|
return name;
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template <class BRIDGETYPE>
|
2017-12-29 08:08:12 +00:00
|
|
|
bool PAKRouter<BRIDGETYPE>::extractResources(const BRIDGETYPE& pakBridge, bool force, hecl::blender::Token& btok,
|
2018-12-08 05:30:43 +00:00
|
|
|
std::function<void(const hecl::SystemChar*, float)> progress) {
|
|
|
|
enterPAKBridge(pakBridge);
|
|
|
|
size_t count = 0;
|
|
|
|
size_t sz = m_pak->m_entries.size();
|
|
|
|
float fsz = sz;
|
|
|
|
for (unsigned w = 0; count < sz; ++w) {
|
|
|
|
for (const auto& item : m_pak->m_firstEntries) {
|
|
|
|
const auto* entryPtr = m_pak->lookupEntry(item);
|
|
|
|
ResExtractor<BRIDGETYPE> extractor = BRIDGETYPE::LookupExtractor(*m_node.get(), *m_pak.get(), *entryPtr);
|
|
|
|
if (extractor.weight != w)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
std::string bestName = getBestEntryName(*entryPtr, false);
|
|
|
|
hecl::SystemStringConv bestNameView(bestName);
|
|
|
|
float thisFac = ++count / fsz;
|
|
|
|
progress(bestNameView.c_str(), thisFac);
|
|
|
|
|
|
|
|
const nod::Node* node = m_node.get();
|
|
|
|
|
|
|
|
hecl::ProjectPath working = getWorking(entryPtr, extractor);
|
|
|
|
working.makeDirChain(false);
|
|
|
|
hecl::ResourceLock resLk(working);
|
|
|
|
if (!resLk)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
/* Extract to unmodified directory */
|
|
|
|
hecl::ProjectPath cooked = working.getCookedPath(m_dataSpec.getUnmodifiedSpec());
|
|
|
|
if (force || cooked.isNone()) {
|
|
|
|
cooked.makeDirChain(false);
|
|
|
|
PAKEntryReadStream s = entryPtr->beginReadStream(*node);
|
|
|
|
FILE* fout = hecl::Fopen(cooked.getAbsolutePath().data(), _SYS_STR("wb"));
|
|
|
|
fwrite(s.data(), 1, s.length(), fout);
|
|
|
|
fclose(fout);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (extractor.func_a) /* Doesn't need PAKRouter access */
|
|
|
|
{
|
|
|
|
if (force || !extractor.IsFullyExtracted(working)) {
|
|
|
|
PAKEntryReadStream s = entryPtr->beginReadStream(*node);
|
|
|
|
extractor.func_a(s, working);
|
|
|
|
}
|
|
|
|
} else if (extractor.func_b) /* Needs PAKRouter access */
|
|
|
|
{
|
|
|
|
if (force || !extractor.IsFullyExtracted(working)) {
|
|
|
|
PAKEntryReadStream s = entryPtr->beginReadStream(*node);
|
|
|
|
extractor.func_b(m_dataSpec, s, working, *this, *entryPtr, force, btok,
|
|
|
|
[&progress, thisFac](const hecl::SystemChar* update) { progress(update, thisFac); });
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
2016-03-02 07:29:19 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
return true;
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template <class BRIDGETYPE>
|
|
|
|
const typename BRIDGETYPE::PAKType::Entry* PAKRouter<BRIDGETYPE>::lookupEntry(const IDType& entry,
|
2016-03-05 00:03:41 +00:00
|
|
|
const nod::Node** nodeOut,
|
2016-03-02 07:29:19 +00:00
|
|
|
bool silenceWarnings,
|
2018-12-08 05:30:43 +00:00
|
|
|
bool currentPAK) const {
|
|
|
|
if (!entry)
|
|
|
|
return nullptr;
|
|
|
|
|
|
|
|
if (!m_bridges)
|
|
|
|
LogDNACommon.report(logvisor::Fatal, "PAKRouter::build() must be called before PAKRouter::lookupEntry()");
|
|
|
|
|
|
|
|
const PAKType* pak = m_pak.get();
|
|
|
|
const nod::Node* node = m_node.get();
|
|
|
|
if (pak) {
|
|
|
|
const EntryType* ent = pak->lookupEntry(entry);
|
|
|
|
if (ent) {
|
|
|
|
if (nodeOut)
|
|
|
|
*nodeOut = node;
|
|
|
|
return ent;
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
2016-03-02 07:29:19 +00:00
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
if (currentPAK) {
|
2016-03-04 01:01:37 +00:00
|
|
|
#ifndef NDEBUG
|
2018-12-08 05:30:43 +00:00
|
|
|
if (!silenceWarnings)
|
|
|
|
LogDNACommon.report(logvisor::Warning, "unable to find PAK entry %s in current PAK", entry.toString().c_str());
|
2016-03-04 01:01:37 +00:00
|
|
|
#endif
|
2018-12-08 05:30:43 +00:00
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (const BRIDGETYPE& bridge : *m_bridges) {
|
|
|
|
const PAKType& pak = bridge.getPAK();
|
|
|
|
const EntryType* ent = pak.lookupEntry(entry);
|
|
|
|
if (ent) {
|
|
|
|
if (nodeOut)
|
|
|
|
*nodeOut = &bridge.getNode();
|
|
|
|
return ent;
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
2018-12-08 05:30:43 +00:00
|
|
|
}
|
2016-03-02 07:29:19 +00:00
|
|
|
|
2016-03-04 01:01:37 +00:00
|
|
|
#ifndef NDEBUG
|
2018-12-08 05:30:43 +00:00
|
|
|
if (!silenceWarnings)
|
|
|
|
LogDNACommon.report(logvisor::Warning, "unable to find PAK entry %s", entry.toString().c_str());
|
2016-03-04 01:01:37 +00:00
|
|
|
#endif
|
2018-12-08 05:30:43 +00:00
|
|
|
if (nodeOut)
|
|
|
|
*nodeOut = nullptr;
|
|
|
|
return nullptr;
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template <class BRIDGETYPE>
|
2018-10-11 20:50:05 +00:00
|
|
|
const typename CharacterAssociations<typename PAKRouter<BRIDGETYPE>::IDType>::RigPair*
|
2018-12-08 05:30:43 +00:00
|
|
|
PAKRouter<BRIDGETYPE>::lookupCMDLRigPair(const IDType& id) const {
|
|
|
|
auto search = m_charAssoc.m_cmdlRigs.find(id);
|
|
|
|
if (search == m_charAssoc.m_cmdlRigs.end())
|
|
|
|
return nullptr;
|
|
|
|
return &search->second;
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
2018-10-11 20:50:05 +00:00
|
|
|
template <class BRIDGETYPE>
|
|
|
|
const typename CharacterAssociations<typename PAKRouter<BRIDGETYPE>::IDType>::MultimapIteratorPair
|
2018-12-08 05:30:43 +00:00
|
|
|
PAKRouter<BRIDGETYPE>::lookupCharacterAttachmentRigs(const IDType& id) const {
|
|
|
|
return m_charAssoc.m_characterToAttachmentRigs.equal_range(id);
|
2018-10-11 20:50:05 +00:00
|
|
|
}
|
|
|
|
|
2016-10-01 23:20:20 +00:00
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
const zeus::CMatrix4f* PAKRouter<BRIDGETYPE>::lookupMAPATransform(const IDType& id) const {
|
|
|
|
auto search = m_mapaTransforms.find(id);
|
|
|
|
if (search == m_mapaTransforms.end())
|
|
|
|
return nullptr;
|
|
|
|
return &search->second;
|
2016-10-01 23:20:20 +00:00
|
|
|
}
|
|
|
|
|
2016-03-02 07:29:19 +00:00
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::ProjectPath PAKRouter<BRIDGETYPE>::getAreaLayerWorking(const IDType& areaId, int layerIdx) const {
|
|
|
|
if (!m_bridges)
|
|
|
|
LogDNACommon.report(logvisor::Fatal, "PAKRouter::build() must be called before PAKRouter::getAreaLayerWorking()");
|
|
|
|
auto bridgePathIt = m_bridgePaths.cbegin();
|
|
|
|
for (const BRIDGETYPE& bridge : *m_bridges) {
|
|
|
|
for (const auto& level : bridge.m_levelDeps)
|
|
|
|
for (const auto& area : level.second.areas)
|
|
|
|
if (area.first == areaId) {
|
|
|
|
hecl::ProjectPath levelPath(bridgePathIt->first, level.second.name);
|
|
|
|
hecl::ProjectPath areaPath(levelPath, area.second.name);
|
|
|
|
if (layerIdx < 0)
|
|
|
|
return areaPath;
|
|
|
|
return hecl::ProjectPath(areaPath, area.second.layers.at(layerIdx).name);
|
|
|
|
}
|
|
|
|
++bridgePathIt;
|
|
|
|
}
|
|
|
|
return hecl::ProjectPath();
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
2016-08-10 02:52:00 +00:00
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::ProjectPath PAKRouter<BRIDGETYPE>::getAreaLayerWorking(const IDType& areaId, int layerIdx,
|
|
|
|
bool& activeOut) const {
|
|
|
|
activeOut = false;
|
|
|
|
if (!m_bridges)
|
|
|
|
LogDNACommon.report(logvisor::Fatal, "PAKRouter::build() must be called before PAKRouter::getAreaLayerWorking()");
|
|
|
|
auto bridgePathIt = m_bridgePaths.cbegin();
|
|
|
|
for (const BRIDGETYPE& bridge : *m_bridges) {
|
|
|
|
for (const auto& level : bridge.m_levelDeps)
|
|
|
|
for (const auto& area : level.second.areas)
|
|
|
|
if (area.first == areaId) {
|
|
|
|
hecl::ProjectPath levelPath(bridgePathIt->first, level.second.name);
|
|
|
|
hecl::ProjectPath areaPath(levelPath, area.second.name);
|
|
|
|
if (layerIdx < 0)
|
|
|
|
return areaPath;
|
|
|
|
const typename Level<IDType>::Area::Layer& layer = area.second.layers.at(layerIdx);
|
|
|
|
activeOut = layer.active;
|
|
|
|
return hecl::ProjectPath(areaPath, layer.name);
|
|
|
|
}
|
|
|
|
++bridgePathIt;
|
|
|
|
}
|
|
|
|
return hecl::ProjectPath();
|
2016-08-10 02:52:00 +00:00
|
|
|
}
|
|
|
|
|
2016-03-02 07:29:19 +00:00
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::ProjectPath PAKRouter<BRIDGETYPE>::getAreaLayerCooked(const IDType& areaId, int layerIdx) const {
|
|
|
|
if (!m_bridges)
|
|
|
|
LogDNACommon.report(logvisor::Fatal, "PAKRouter::build() must be called before PAKRouter::getAreaLayerCooked()");
|
|
|
|
auto bridgePathIt = m_bridgePaths.cbegin();
|
|
|
|
for (const BRIDGETYPE& bridge : *m_bridges) {
|
|
|
|
for (const auto& level : bridge.m_levelDeps)
|
|
|
|
for (const auto& area : level.second.areas)
|
|
|
|
if (area.first == areaId) {
|
|
|
|
hecl::ProjectPath levelPath(bridgePathIt->second, level.second.name);
|
|
|
|
hecl::ProjectPath areaPath(levelPath, area.second.name);
|
|
|
|
if (layerIdx < 0)
|
|
|
|
return areaPath;
|
|
|
|
return hecl::ProjectPath(areaPath, area.second.layers.at(layerIdx).name);
|
|
|
|
}
|
|
|
|
++bridgePathIt;
|
|
|
|
}
|
|
|
|
return hecl::ProjectPath();
|
2016-03-02 07:29:19 +00:00
|
|
|
}
|
|
|
|
|
2016-08-10 02:52:00 +00:00
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
hecl::ProjectPath PAKRouter<BRIDGETYPE>::getAreaLayerCooked(const IDType& areaId, int layerIdx, bool& activeOut) const {
|
|
|
|
activeOut = false;
|
|
|
|
if (!m_bridges)
|
|
|
|
LogDNACommon.report(logvisor::Fatal, "PAKRouter::build() must be called before PAKRouter::getAreaLayerCooked()");
|
|
|
|
auto bridgePathIt = m_bridgePaths.cbegin();
|
|
|
|
for (const BRIDGETYPE& bridge : *m_bridges) {
|
|
|
|
for (const auto& level : bridge.m_levelDeps)
|
|
|
|
for (const auto& area : level.second.areas)
|
|
|
|
if (area.first == areaId) {
|
|
|
|
hecl::ProjectPath levelPath(bridgePathIt->second, level.second.name);
|
|
|
|
hecl::ProjectPath areaPath(levelPath, area.second.name);
|
|
|
|
if (layerIdx < 0)
|
|
|
|
return areaPath;
|
|
|
|
const typename Level<IDType>::Area::Layer& layer = area.second.layers.at(layerIdx);
|
|
|
|
activeOut = layer.active;
|
|
|
|
return hecl::ProjectPath(areaPath, layer.name);
|
|
|
|
}
|
|
|
|
++bridgePathIt;
|
|
|
|
}
|
|
|
|
return hecl::ProjectPath();
|
2016-08-10 02:52:00 +00:00
|
|
|
}
|
|
|
|
|
2017-01-30 23:22:26 +00:00
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
void PAKRouter<BRIDGETYPE>::enumerateResources(const std::function<bool(const EntryType*)>& func) {
|
|
|
|
if (!m_bridges)
|
|
|
|
LogDNACommon.report(logvisor::Fatal, "PAKRouter::build() must be called before PAKRouter::enumerateResources()");
|
|
|
|
for (const auto& entryPair : m_uniqueEntries)
|
|
|
|
if (!func(entryPair.second.second))
|
|
|
|
return;
|
|
|
|
for (const auto& entryPair : m_sharedEntries)
|
|
|
|
if (!func(entryPair.second.second))
|
|
|
|
return;
|
2017-01-30 23:22:26 +00:00
|
|
|
}
|
|
|
|
|
2018-04-07 20:55:57 +00:00
|
|
|
template <class BRIDGETYPE>
|
2018-12-08 05:30:43 +00:00
|
|
|
bool PAKRouter<BRIDGETYPE>::mreaHasDupeResources(const IDType& id) const {
|
|
|
|
const PAKType* pak = m_pak.get();
|
|
|
|
if (!pak)
|
|
|
|
LogDNACommon.report(logvisor::Fatal,
|
|
|
|
"PAKRouter::enterPAKBridge() must be called before PAKRouter::mreaHasDupeResources()");
|
|
|
|
return pak->mreaHasDupeResources(id);
|
2018-04-07 20:55:57 +00:00
|
|
|
}
|
|
|
|
|
2016-03-02 07:29:19 +00:00
|
|
|
template class PAKRouter<DNAMP1::PAKBridge>;
|
|
|
|
template class PAKRouter<DNAMP2::PAKBridge>;
|
|
|
|
template class PAKRouter<DNAMP3::PAKBridge>;
|
|
|
|
|
2018-12-08 05:30:43 +00:00
|
|
|
} // namespace DataSpec
|