metaforce/DataSpec/DNACommon/DNACommon.hpp

678 lines
23 KiB
C++
Raw Normal View History

2015-06-24 20:56:23 +00:00
#ifndef __DNA_COMMON_HPP__
#define __DNA_COMMON_HPP__
#include <stdio.h>
2015-08-01 00:38:35 +00:00
#include <Athena/DNAYaml.hpp>
#include <NOD/DiscBase.hpp>
2015-07-06 01:33:06 +00:00
#include "HECL/HECL.hpp"
2015-07-17 00:01:05 +00:00
#include "HECL/Database.hpp"
2015-08-05 21:46:07 +00:00
#include "../SpecBase.hpp"
2015-06-24 20:56:23 +00:00
2015-07-01 23:50:39 +00:00
namespace Retro
{
2015-07-16 01:57:34 +00:00
extern LogVisor::LogModule LogDNACommon;
2015-06-24 20:56:23 +00:00
/* This comes up a great deal */
typedef Athena::io::DNA<Athena::BigEndian> BigDNA;
2015-08-01 00:38:35 +00:00
typedef Athena::io::DNAYaml<Athena::BigEndian> BigYAML;
2015-06-24 20:56:23 +00:00
/* FourCC with DNA read/write */
2015-08-01 00:38:35 +00:00
class FourCC final : public BigYAML, public HECL::FourCC
2015-07-11 22:41:10 +00:00
{
public:
FourCC() : HECL::FourCC() {}
FourCC(const HECL::FourCC& other)
: HECL::FourCC() {num = other.toUint32();}
2015-07-11 22:41:10 +00:00
FourCC(const char* name)
: HECL::FourCC(name) {}
FourCC(uint32_t n)
: HECL::FourCC(n) {}
2015-07-11 22:41:10 +00:00
Delete expl;
inline void read(Athena::io::IStreamReader& reader)
{reader.readUBytesToBuf(fcc, 4);}
inline void write(Athena::io::IStreamWriter& writer) const
{writer.writeUBytes((atUint8*)fcc, 4);}
2015-08-01 00:38:35 +00:00
inline void fromYAML(Athena::io::YAMLDocReader& reader)
{std::string rs = reader.readString(nullptr); strncpy(fcc, rs.c_str(), 4);}
inline void toYAML(Athena::io::YAMLDocWriter& writer) const
{writer.writeString(nullptr, std::string(fcc, 4));}
2015-07-11 22:41:10 +00:00
};
2015-07-06 01:33:06 +00:00
/* PAK 32-bit Unique ID */
2015-08-01 00:38:35 +00:00
class UniqueID32 : public BigYAML
2015-06-24 20:56:23 +00:00
{
2015-08-11 23:32:02 +00:00
uint32_t m_id = 0xffffffff;
2015-06-24 20:56:23 +00:00
public:
Delete expl;
2015-08-11 23:32:02 +00:00
inline operator bool() const {return m_id != 0xffffffff;}
2015-07-06 01:33:06 +00:00
inline void read(Athena::io::IStreamReader& reader)
{m_id = reader.readUint32Big();}
2015-07-06 01:33:06 +00:00
inline void write(Athena::io::IStreamWriter& writer) const
{writer.writeUint32Big(m_id);}
2015-08-01 00:38:35 +00:00
inline void fromYAML(Athena::io::YAMLDocReader& reader)
{m_id = reader.readUint32(nullptr);}
inline void toYAML(Athena::io::YAMLDocWriter& writer) const
{writer.writeUint32(nullptr, m_id);}
2015-07-06 01:33:06 +00:00
inline bool operator!=(const UniqueID32& other) const {return m_id != other.m_id;}
inline bool operator==(const UniqueID32& other) const {return m_id == other.m_id;}
2015-07-06 02:07:57 +00:00
inline uint32_t toUint32() const {return m_id;}
2015-07-06 01:33:06 +00:00
inline std::string toString() const
{
char buf[9];
snprintf(buf, 9, "%08X", m_id);
return std::string(buf);
}
2015-06-24 20:56:23 +00:00
};
2015-07-06 01:33:06 +00:00
/* PAK 64-bit Unique ID */
class UniqueID64 : public BigDNA
2015-06-24 20:56:23 +00:00
{
2015-08-11 23:32:02 +00:00
uint64_t m_id = 0xffffffffffffffff;
2015-06-24 20:56:23 +00:00
public:
Delete expl;
2015-08-11 23:32:02 +00:00
inline operator bool() const {return m_id != 0xffffffffffffffff;}
2015-07-06 01:33:06 +00:00
inline void read(Athena::io::IStreamReader& reader)
{m_id = reader.readUint64Big();}
2015-07-06 01:33:06 +00:00
inline void write(Athena::io::IStreamWriter& writer) const
{writer.writeUint64Big(m_id);}
2015-07-06 01:33:06 +00:00
inline bool operator!=(const UniqueID64& other) const {return m_id != other.m_id;}
inline bool operator==(const UniqueID64& other) const {return m_id == other.m_id;}
2015-07-06 02:07:57 +00:00
inline uint64_t toUint64() const {return m_id;}
2015-07-06 01:33:06 +00:00
inline std::string toString() const
{
char buf[17];
2015-07-19 10:43:37 +00:00
snprintf(buf, 17, "%016lX", m_id);
2015-07-06 01:33:06 +00:00
return std::string(buf);
}
2015-06-24 20:56:23 +00:00
};
2015-07-06 01:33:06 +00:00
/* PAK 128-bit Unique ID */
class UniqueID128 : public BigDNA
2015-06-24 20:56:23 +00:00
{
2015-07-06 01:33:06 +00:00
union
{
uint64_t m_id[2];
#if __SSE__
__m128i m_id128;
#endif
};
2015-06-24 20:56:23 +00:00
public:
Delete expl;
2015-08-11 23:32:02 +00:00
UniqueID128() {m_id[0]=0xffffffffffffffff; m_id[1]=0xffffffffffffffff;}
inline operator bool() const
{return m_id[0] != 0xffffffffffffffff && m_id[1] != 0xffffffffffffffff;}
2015-07-06 01:33:06 +00:00
inline void read(Athena::io::IStreamReader& reader)
{
m_id[0] = reader.readUint64Big();
m_id[1] = reader.readUint64Big();
2015-07-06 01:33:06 +00:00
}
inline void write(Athena::io::IStreamWriter& writer) const
{
writer.writeUint64Big(m_id[0]);
writer.writeUint64Big(m_id[1]);
2015-07-06 01:33:06 +00:00
}
inline bool operator!=(const UniqueID128& other) const
{
#if __SSE__
2015-07-06 02:07:57 +00:00
__m128i vcmp = _mm_cmpeq_epi32(m_id128, other.m_id128);
int vmask = _mm_movemask_epi8(vcmp);
return vmask != 0xffff;
2015-07-06 01:33:06 +00:00
#else
return (m_id[0] != other.m_id[0]) || (m_id[1] != other.m_id[1]);
#endif
}
inline bool operator==(const UniqueID128& other) const
{
#if __SSE__
2015-07-06 02:07:57 +00:00
__m128i vcmp = _mm_cmpeq_epi32(m_id128, other.m_id128);
int vmask = _mm_movemask_epi8(vcmp);
return vmask == 0xffff;
2015-07-06 01:33:06 +00:00
#else
return (m_id[0] == other.m_id[0]) && (m_id[1] == other.m_id[1]);
#endif
}
2015-07-06 02:07:57 +00:00
inline uint64_t toHighUint64() const {return m_id[0];}
inline uint64_t toLowUint64() const {return m_id[1];}
2015-07-06 01:33:06 +00:00
inline std::string toString() const
{
char buf[33];
2015-07-19 10:43:37 +00:00
snprintf(buf, 33, "%016lX%016lX", m_id[0], m_id[1]);
2015-07-06 01:33:06 +00:00
return std::string(buf);
}
2015-06-24 20:56:23 +00:00
};
2015-07-14 00:38:48 +00:00
/* Case-insensitive comparator for std::map sorting */
2015-07-10 05:28:08 +00:00
struct CaseInsensitiveCompare
{
inline bool operator()(const std::string& lhs, const std::string& rhs) const
{
2015-07-14 00:38:48 +00:00
#if _WIN32
2015-07-22 19:05:18 +00:00
if (_stricmp(lhs.c_str(), rhs.c_str()) < 0)
2015-07-14 00:38:48 +00:00
#else
if (strcasecmp(lhs.c_str(), rhs.c_str()) < 0)
#endif
2015-07-10 05:28:08 +00:00
return true;
return false;
}
2015-07-22 19:05:18 +00:00
#if _WIN32
inline bool operator()(const std::wstring& lhs, const std::wstring& rhs) const
{
if (_wcsicmp(lhs.c_str(), rhs.c_str()) < 0)
return true;
return false;
}
#endif
2015-07-10 05:28:08 +00:00
};
2015-07-14 00:38:48 +00:00
2015-08-11 23:32:02 +00:00
/* Word Bitmap reader/writer */
struct WordBitmap
{
std::vector<atUint32> m_words;
size_t m_bitCount = 0;
void read(Athena::io::IStreamReader& reader, size_t bitCount)
{
m_bitCount = bitCount;
size_t wordCount = (bitCount + 31) / 32;
m_words.clear();
m_words.reserve(wordCount);
for (size_t w=0 ; w<wordCount ; ++w)
m_words.push_back(reader.readUint32Big());
2015-08-11 23:32:02 +00:00
}
void write(Athena::io::IStreamWriter& writer) const
{
for (atUint32 word : m_words)
writer.writeUint32(word);
}
size_t getBitCount() const {return m_bitCount;}
bool getBit(size_t idx) const
{
size_t wordIdx = idx / 32;
if (wordIdx >= m_words.size())
return false;
size_t wordCur = idx % 32;
return (m_words[wordIdx] >> wordCur) & 0x1;
}
void setBit(size_t idx)
{
size_t wordIdx = idx / 32;
while (wordIdx >= m_words.size())
m_words.push_back(0);
size_t wordCur = idx % 32;
m_words[wordIdx] |= (1 << wordCur);
}
void unsetBit(size_t idx)
{
size_t wordIdx = idx / 32;
while (wordIdx >= m_words.size())
m_words.push_back(0);
size_t wordCur = idx % 32;
m_words[wordIdx] &= ~(1 << wordCur);
}
void clear()
{
m_words.clear();
}
class Iterator : public std::iterator<std::forward_iterator_tag, bool>
{
friend class WordBitmap;
const WordBitmap& m_bmp;
size_t m_idx = 0;
Iterator(const WordBitmap& bmp, size_t idx) : m_bmp(bmp), m_idx(idx) {}
public:
Iterator& operator++() {++m_idx; return *this;}
bool operator*() {return m_bmp.getBit(m_idx);}
bool operator!=(const Iterator& other) const {return m_idx != other.m_idx;}
};
Iterator begin() const {return Iterator(*this, 0);}
Iterator end() const {return Iterator(*this, m_bitCount);}
};
2015-07-14 00:38:48 +00:00
/* PAK entry stream reader */
class PAKEntryReadStream : public Athena::io::IStreamReader
{
std::unique_ptr<atUint8[]> m_buf;
atUint64 m_sz;
atUint64 m_pos;
public:
PAKEntryReadStream() {}
operator bool() const {return m_buf.operator bool();}
PAKEntryReadStream(const PAKEntryReadStream& other) = delete;
PAKEntryReadStream(PAKEntryReadStream&& other) = default;
PAKEntryReadStream& operator=(const PAKEntryReadStream& other) = delete;
PAKEntryReadStream& operator=(PAKEntryReadStream&& other) = default;
2015-07-14 00:38:48 +00:00
PAKEntryReadStream(std::unique_ptr<atUint8[]>&& buf, atUint64 sz, atUint64 pos)
: m_buf(std::move(buf)), m_sz(sz), m_pos(pos)
{
if (m_pos >= m_sz)
2015-07-16 01:57:34 +00:00
LogDNACommon.report(LogVisor::FatalError, "PAK stream cursor overrun");
2015-07-14 00:38:48 +00:00
}
inline void seek(atInt64 pos, Athena::SeekOrigin origin)
{
if (origin == Athena::Begin)
m_pos = pos;
else if (origin == Athena::Current)
m_pos += pos;
else if (origin == Athena::End)
m_pos = m_sz + pos;
if (m_pos >= m_sz)
2015-07-16 01:57:34 +00:00
LogDNACommon.report(LogVisor::FatalError, "PAK stream cursor overrun");
2015-07-14 00:38:48 +00:00
}
inline atUint64 position() const {return m_pos;}
inline atUint64 length() const {return m_sz;}
inline const atUint8* data() const {return m_buf.get();}
2015-07-14 00:38:48 +00:00
inline atUint64 readUBytesToBuf(void* buf, atUint64 len)
{
atUint64 bufEnd = m_pos + len;
if (bufEnd > m_sz)
len -= bufEnd - m_sz;
memcpy(buf, m_buf.get() + m_pos, len);
m_pos += len;
return len;
}
};
2015-07-17 00:01:05 +00:00
struct UniqueResult
{
enum Type
{
UNIQUE_NOTFOUND,
UNIQUE_LEVEL,
UNIQUE_AREA,
UNIQUE_LAYER
} type = UNIQUE_NOTFOUND;
const HECL::SystemString* areaName = nullptr;
const HECL::SystemString* layerName = nullptr;
UniqueResult() = default;
UniqueResult(Type tp) : type(tp) {}
inline HECL::ProjectPath uniquePath(const HECL::ProjectPath& pakPath) const
{
if (type == UNIQUE_AREA)
{
HECL::ProjectPath areaDir(pakPath, *areaName);
areaDir.makeDir();
return areaDir;
}
else if (type == UNIQUE_LAYER)
{
HECL::ProjectPath areaDir(pakPath, *areaName);
areaDir.makeDir();
HECL::ProjectPath layerDir(areaDir, *layerName);
layerDir.makeDir();
return layerDir;
}
return pakPath;
}
};
template <class BRIDGETYPE>
class PAKRouter;
2015-07-17 00:01:05 +00:00
/* Resource extractor type */
template <class PAKBRIDGE>
struct ResExtractor
2015-07-17 00:01:05 +00:00
{
2015-08-05 21:46:07 +00:00
std::function<bool(const SpecBase&, PAKEntryReadStream&, const HECL::ProjectPath&)> func_a;
std::function<bool(const SpecBase&, PAKEntryReadStream&, const HECL::ProjectPath&, PAKRouter<PAKBRIDGE>&,
2015-08-11 23:32:02 +00:00
const typename PAKBRIDGE::PAKType::Entry&, bool)> func_b;
2015-08-15 04:12:15 +00:00
const char* fileExts[4];
unsigned weight;
};
2015-07-17 00:01:05 +00:00
2015-07-28 23:53:57 +00:00
/* PAKRouter (for detecting shared entry locations) */
template <class BRIDGETYPE>
class PAKRouter
{
public:
using PAKType = typename BRIDGETYPE::PAKType;
using IDType = typename PAKType::IDType;
using EntryType = typename PAKType::Entry;
private:
2015-08-05 21:46:07 +00:00
const SpecBase& m_dataSpec;
2015-08-11 23:32:02 +00:00
const std::vector<BRIDGETYPE>* m_bridges = nullptr;
2015-08-16 23:00:33 +00:00
std::vector<std::pair<HECL::ProjectPath,HECL::ProjectPath>> m_bridgePaths;
size_t m_curBridgeIdx = 0;
2015-07-28 23:53:57 +00:00
const HECL::ProjectPath& m_gameWorking;
const HECL::ProjectPath& m_gameCooked;
HECL::ProjectPath m_sharedWorking;
HECL::ProjectPath m_sharedCooked;
const PAKType* m_pak = nullptr;
const NOD::DiscBase::IPartition::Node* m_node = nullptr;
2015-08-16 23:00:33 +00:00
std::unordered_map<IDType, std::pair<size_t, EntryType*>> m_uniqueEntries;
std::unordered_map<IDType, std::pair<size_t, EntryType*>> m_sharedEntries;
2015-07-28 23:53:57 +00:00
public:
2015-08-05 21:46:07 +00:00
PAKRouter(const SpecBase& dataSpec, const HECL::ProjectPath& working, const HECL::ProjectPath& cooked)
: m_dataSpec(dataSpec),
m_gameWorking(working), m_gameCooked(cooked),
2015-07-28 23:53:57 +00:00
m_sharedWorking(working, "Shared"), m_sharedCooked(cooked, "Shared") {}
void build(std::vector<BRIDGETYPE>& bridges, std::function<void(float)> progress)
2015-07-28 23:53:57 +00:00
{
2015-08-11 23:32:02 +00:00
m_bridges = &bridges;
2015-08-16 23:00:33 +00:00
m_bridgePaths.clear();
2015-07-28 23:53:57 +00:00
m_uniqueEntries.clear();
m_sharedEntries.clear();
size_t count = 0;
float bridgesSz = bridges.size();
/* Route entries unique/shared per-pak */
2015-08-16 23:00:33 +00:00
size_t bridgeIdx = 0;
for (BRIDGETYPE& bridge : bridges)
2015-07-28 23:53:57 +00:00
{
2015-08-16 23:00:33 +00:00
const std::string& name = bridge.getName();
HECL::SystemStringView sysName(name);
HECL::SystemString::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)));
bridge.build();
2015-07-28 23:53:57 +00:00
const typename BRIDGETYPE::PAKType& pak = bridge.getPAK();
for (const auto& entry : pak.m_idMap)
{
2015-08-16 23:00:33 +00:00
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())
2015-07-28 23:53:57 +00:00
{
2015-08-16 23:00:33 +00:00
m_uniqueEntries.erase(uSearch);
m_sharedEntries[entry.first] = std::make_pair(bridgeIdx, entry.second);
2015-07-28 23:53:57 +00:00
}
else
2015-08-16 23:00:33 +00:00
m_uniqueEntries[entry.first] = std::make_pair(bridgeIdx, entry.second);
2015-07-28 23:53:57 +00:00
}
progress(++count / bridgesSz);
2015-08-16 23:00:33 +00:00
++bridgeIdx;
2015-07-28 23:53:57 +00:00
}
}
void enterPAKBridge(const BRIDGETYPE& pakBridge)
2015-08-16 23:00:33 +00:00
{
auto pit = m_bridgePaths.begin();
size_t bridgeIdx = 0;
for (const BRIDGETYPE& bridge : *m_bridges)
{
if (&bridge == &pakBridge)
{
pit->first.makeDir();
pit->second.makeDir();
m_pak = &pakBridge.getPAK();
m_node = &pakBridge.getNode();
m_curBridgeIdx = bridgeIdx;
return;
}
++pit;
++bridgeIdx;
}
LogDNACommon.report(LogVisor::FatalError, "PAKBridge provided to PAKRouter::enterPAKBridge() was not part of build()");
2015-07-28 23:53:57 +00:00
}
HECL::ProjectPath getWorking(const typename BRIDGETYPE::PAKType::Entry* entry,
const ResExtractor<BRIDGETYPE>& extractor) const
2015-07-28 23:53:57 +00:00
{
if (!m_pak)
LogDNACommon.report(LogVisor::FatalError,
"PAKRouter::enterPAKBridge() must be called before PAKRouter::getWorkingPath()");
auto uniqueSearch = m_uniqueEntries.find(entry->id);
2015-07-28 23:53:57 +00:00
if (uniqueSearch != m_uniqueEntries.end())
{
2015-08-16 23:00:33 +00:00
const HECL::ProjectPath& pakPath = m_bridgePaths[uniqueSearch->second.first].first;
pakPath.makeDir();
HECL::ProjectPath uniquePath = entry->unique.uniquePath(pakPath);
HECL::SystemString entName = m_pak->bestEntryName(*entry);
2015-08-15 04:12:15 +00:00
if (extractor.fileExts[0] && !extractor.fileExts[1])
entName += extractor.fileExts[0];
return HECL::ProjectPath(uniquePath, entName);
2015-07-28 23:53:57 +00:00
}
auto sharedSearch = m_sharedEntries.find(entry->id);
2015-07-28 23:53:57 +00:00
if (sharedSearch != m_sharedEntries.end())
{
2015-08-16 23:00:33 +00:00
const HECL::ProjectPath& pakPath = m_bridgePaths[m_curBridgeIdx].first;
HECL::ProjectPath uniquePathPre = entry->unique.uniquePath(pakPath);
2015-08-15 04:12:15 +00:00
HECL::SystemString entBase = m_pak->bestEntryName(*entry);
HECL::SystemString entName = entBase;
if (extractor.fileExts[0] && !extractor.fileExts[1])
entName += extractor.fileExts[0];
2015-07-28 23:53:57 +00:00
HECL::ProjectPath sharedPath(m_sharedWorking, entName);
2015-08-04 02:46:48 +00:00
HECL::ProjectPath uniquePath(uniquePathPre, entName);
if (extractor.func_a || extractor.func_b)
2015-08-15 04:12:15 +00:00
{
if (extractor.fileExts[0] && !extractor.fileExts[1])
uniquePath.makeLinkTo(sharedPath);
else
{
for (int e=0 ; e<4 ; ++e)
{
if (!extractor.fileExts[e])
break;
HECL::SystemString entName = entBase + extractor.fileExts[e];
HECL::ProjectPath sharedPath(m_sharedWorking, entName);
HECL::ProjectPath uniquePath(uniquePathPre, entName);
uniquePath.makeLinkTo(sharedPath);
}
}
}
2015-07-28 23:53:57 +00:00
m_sharedWorking.makeDir();
return sharedPath;
2015-07-28 23:53:57 +00:00
}
LogDNACommon.report(LogVisor::FatalError, "Unable to find entry %s", entry->id.toString().c_str());
return HECL::ProjectPath();
2015-07-28 23:53:57 +00:00
}
2015-08-11 23:32:02 +00:00
HECL::ProjectPath getWorking(const typename BRIDGETYPE::PAKType::Entry* entry) const
{
return getWorking(entry, BRIDGETYPE::LookupExtractor(*entry));
}
2015-08-16 23:00:33 +00:00
HECL::ProjectPath getWorking(const typename BRIDGETYPE::PAKType::IDType& id) const
{
return getWorking(lookupEntry(id));
}
HECL::ProjectPath getCooked(const typename BRIDGETYPE::PAKType::Entry* entry) const
2015-07-28 23:53:57 +00:00
{
if (!m_pak)
LogDNACommon.report(LogVisor::FatalError,
"PAKRouter::enterPAKBridge() must be called before PAKRouter::getCookedPath()");
auto uniqueSearch = m_uniqueEntries.find(entry->id);
2015-07-28 23:53:57 +00:00
if (uniqueSearch != m_uniqueEntries.end())
{
2015-08-16 23:00:33 +00:00
const HECL::ProjectPath& pakPath = m_bridgePaths[uniqueSearch->second.first].second;
pakPath.makeDir();
HECL::ProjectPath uniquePath = entry->unique.uniquePath(pakPath);
return HECL::ProjectPath(uniquePath, m_pak->bestEntryName(*entry));
2015-07-28 23:53:57 +00:00
}
auto sharedSearch = m_sharedEntries.find(entry->id);
2015-07-28 23:53:57 +00:00
if (sharedSearch != m_sharedEntries.end())
{
m_sharedCooked.makeDir();
return HECL::ProjectPath(m_sharedCooked, m_pak->bestEntryName(*entry));
2015-07-28 23:53:57 +00:00
}
LogDNACommon.report(LogVisor::FatalError, "Unable to find entry %s", entry->id.toString().c_str());
return HECL::ProjectPath();
}
2015-08-16 23:00:33 +00:00
HECL::ProjectPath getCooked(const typename BRIDGETYPE::PAKType::IDType& id) const
{
return getCooked(lookupEntry(id));
}
HECL::SystemString getResourceRelativePath(const typename BRIDGETYPE::PAKType::Entry& a,
const typename BRIDGETYPE::PAKType::IDType& b) const
{
if (!m_pak)
LogDNACommon.report(LogVisor::FatalError,
"PAKRouter::enterPAKBridge() must be called before PAKRouter::getResourceRelativePath()");
2015-08-16 23:00:33 +00:00
const typename BRIDGETYPE::PAKType::Entry* be = lookupEntry(b);
if (!be)
return HECL::SystemString();
HECL::ProjectPath aPath = getWorking(&a, BRIDGETYPE::LookupExtractor(a));
HECL::SystemString ret;
for (int i=0 ; i<aPath.levelCount() ; ++i)
ret += "../";
HECL::ProjectPath bPath = getWorking(be, BRIDGETYPE::LookupExtractor(*be));
ret += bPath.getRelativePath();
return ret;
}
std::string getBestEntryName(const typename BRIDGETYPE::PAKType::Entry& entry) const
{
if (!m_pak)
LogDNACommon.report(LogVisor::FatalError,
"PAKRouter::enterPAKBridge() must be called before PAKRouter::getBestEntryName()");
return m_pak->bestEntryName(entry);
}
std::string getBestEntryName(const typename BRIDGETYPE::PAKType::IDType& entry) const
{
if (!m_pak)
LogDNACommon.report(LogVisor::FatalError,
"PAKRouter::enterPAKBridge() must be called before PAKRouter::getBestEntryName()");
const typename BRIDGETYPE::PAKType::Entry* e = m_pak->lookupEntry(entry);
if (!e)
return entry.toString();
return m_pak->bestEntryName(*e);
}
bool extractResources(const BRIDGETYPE& pakBridge, bool force, std::function<void(float)> progress)
{
enterPAKBridge(pakBridge);
size_t count = 0;
size_t sz = m_pak->m_idMap.size();
float fsz = sz;
for (unsigned w=0 ; count<sz ; ++w)
{
for (const auto& item : m_pak->m_idMap)
{
ResExtractor<BRIDGETYPE> extractor = BRIDGETYPE::LookupExtractor(*item.second);
if (extractor.weight != w)
continue;
HECL::ProjectPath cooked = getCooked(item.second);
if (force || cooked.getPathType() == HECL::ProjectPath::PT_NONE)
{
PAKEntryReadStream s = item.second->beginReadStream(*m_node);
FILE* fout = HECL::Fopen(cooked.getAbsolutePath().c_str(), _S("wb"));
fwrite(s.data(), 1, s.length(), fout);
fclose(fout);
}
HECL::ProjectPath working = getWorking(item.second, extractor);
if (extractor.func_a) /* Doesn't need PAKRouter access */
{
if (force || working.getPathType() == HECL::ProjectPath::PT_NONE)
{
PAKEntryReadStream s = item.second->beginReadStream(*m_node);
2015-08-05 21:46:07 +00:00
extractor.func_a(m_dataSpec, s, working);
}
}
else if (extractor.func_b) /* Needs PAKRouter access */
{
if (force || working.getPathType() == HECL::ProjectPath::PT_NONE)
{
PAKEntryReadStream s = item.second->beginReadStream(*m_node);
2015-08-11 23:32:02 +00:00
extractor.func_b(m_dataSpec, s, working, *this, *item.second, force);
}
}
progress(++count / fsz);
}
}
return true;
2015-07-28 23:53:57 +00:00
}
2015-08-11 23:32:02 +00:00
const typename BRIDGETYPE::PAKType::Entry* lookupEntry(const typename BRIDGETYPE::PAKType::IDType& entry,
2015-08-16 23:00:33 +00:00
const NOD::DiscBase::IPartition::Node** nodeOut=nullptr) const
2015-08-11 23:32:02 +00:00
{
if (!m_bridges)
LogDNACommon.report(LogVisor::FatalError,
"PAKRouter::build() must be called before PAKRouter::lookupEntry()");
if (m_pak)
{
const typename BRIDGETYPE::PAKType::Entry* ent = m_pak->lookupEntry(entry);
if (ent)
{
if (nodeOut)
*nodeOut = m_node;
return ent;
}
}
for (const BRIDGETYPE& bridge : *m_bridges)
{
const typename BRIDGETYPE::PAKType& pak = bridge.getPAK();
const typename BRIDGETYPE::PAKType::Entry* ent = pak.lookupEntry(entry);
if (ent)
{
if (nodeOut)
*nodeOut = &bridge.getNode();
return ent;
}
}
2015-08-15 04:12:15 +00:00
LogDNACommon.report(LogVisor::Warning, "unable to find PAK entry %s", entry.toString().c_str());
2015-08-11 23:32:02 +00:00
if (nodeOut)
*nodeOut = nullptr;
return nullptr;
}
2015-08-13 07:29:00 +00:00
template <typename DNA>
bool lookupAndReadDNA(const typename BRIDGETYPE::PAKType::IDType& id, DNA& out)
{
const NOD::DiscBase::IPartition::Node* node;
const typename BRIDGETYPE::PAKType::Entry* entry = lookupEntry(id, &node);
if (!entry)
return false;
PAKEntryReadStream rs = entry->beginReadStream(*node);
out.read(rs);
return true;
}
2015-07-28 23:53:57 +00:00
};
2015-07-17 00:01:05 +00:00
/* Resource cooker function */
typedef std::function<bool(const HECL::ProjectPath&, const HECL::ProjectPath&)> ResCooker;
2015-07-10 05:28:08 +00:00
2015-07-06 01:33:06 +00:00
}
/* Hash template-specializations for UniqueID types */
namespace std
2015-06-24 20:56:23 +00:00
{
template<>
struct hash<Retro::FourCC>
{
inline size_t operator()(const Retro::FourCC& fcc) const
{return fcc.toUint32();}
};
2015-07-06 01:33:06 +00:00
template<>
struct hash<Retro::UniqueID32>
{
inline size_t operator()(const Retro::UniqueID32& id) const
2015-07-06 02:07:57 +00:00
{return id.toUint32();}
2015-07-06 01:33:06 +00:00
};
template<>
struct hash<Retro::UniqueID64>
{
inline size_t operator()(const Retro::UniqueID64& id) const
2015-07-06 02:07:57 +00:00
{return id.toUint64();}
2015-06-24 20:56:23 +00:00
};
2015-07-06 01:33:06 +00:00
template<>
struct hash<Retro::UniqueID128>
{
inline size_t operator()(const Retro::UniqueID128& id) const
2015-07-06 02:07:57 +00:00
{return id.toHighUint64() ^ id.toLowUint64();}
2015-07-06 01:33:06 +00:00
};
2015-07-01 23:50:39 +00:00
}
2015-06-24 20:56:23 +00:00
#endif // __DNA_COMMON_HPP__