metaforce/DataSpec/DNAMP1/STRG.hpp

78 lines
2.3 KiB
C++

#pragma once
#include <unordered_map>
#include "DataSpec/DNACommon/DNACommon.hpp"
#include "DataSpec/DNACommon/STRG.hpp"
#include "DNAMP1.hpp"
namespace DataSpec::DNAMP1 {
struct STRG : ISTRG {
AT_DECL_EXPLICIT_DNA_YAML
AT_DECL_DNAV
void _read(athena::io::IStreamReader& reader);
std::vector<std::pair<FourCC, std::vector<std::u16string>>> langs;
std::unordered_map<FourCC, std::vector<std::u16string>*> langMap;
int32_t lookupIdx(std::string_view name) const override { return -1; }
size_t count() const override {
size_t retval = 0;
for (const auto& item : langs) {
size_t sz = item.second.size();
if (sz > retval)
retval = sz;
}
return retval;
}
std::string getUTF8(const FourCC& lang, size_t idx) const override {
auto search = langMap.find(lang);
if (search != langMap.end())
return hecl::Char16ToUTF8(search->second->at(idx));
return std::string();
}
std::u16string getUTF16(const FourCC& lang, size_t idx) const override {
auto search = langMap.find(lang);
if (search != langMap.end())
return search->second->at(idx);
return std::u16string();
}
hecl::SystemString getSystemString(const FourCC& lang, size_t idx) const override {
auto search = langMap.find(lang);
if (search != langMap.end())
#if HECL_UCS2
return hecl::Char16ToWide(search->second->at(idx));
#else
return hecl::Char16ToUTF8(search->second->at(idx));
#endif
return hecl::SystemString();
}
static bool Extract(PAKEntryReadStream& rs, const hecl::ProjectPath& outPath) {
STRG strg;
strg.read(rs);
athena::io::TransactionalFileWriter writer(outPath.getAbsolutePath());
athena::io::ToYAMLStream(strg, writer);
return true;
}
static bool Cook(const hecl::ProjectPath& inPath, const hecl::ProjectPath& outPath) {
STRG strg;
athena::io::FileReader reader(inPath.getAbsolutePath());
athena::io::FromYAMLStream(strg, reader);
athena::io::TransactionalFileWriter ws(outPath.getAbsolutePath());
strg.write(ws);
return true;
}
static bool Cook(const STRG& strg, const hecl::ProjectPath& outPath) {
athena::io::TransactionalFileWriter ws(outPath.getAbsolutePath());
strg.write(ws);
return true;
}
void gatherDependencies(std::vector<hecl::ProjectPath>& pathsOut) const override;
};
} // namespace DataSpec::DNAMP1