|
|
|
/*
|
|
|
|
Copyright (C) 2015-2021 cc9cii
|
|
|
|
|
|
|
|
This software is provided 'as-is', without any express or implied
|
|
|
|
warranty. In no event will the authors be held liable for any damages
|
|
|
|
arising from the use of this software.
|
|
|
|
|
|
|
|
Permission is granted to anyone to use this software for any purpose,
|
|
|
|
including commercial applications, and to alter it and redistribute it
|
|
|
|
freely, subject to the following restrictions:
|
|
|
|
|
|
|
|
1. The origin of this software must not be misrepresented; you must not
|
|
|
|
claim that you wrote the original software. If you use this software
|
|
|
|
in a product, an acknowledgment in the product documentation would be
|
|
|
|
appreciated but is not required.
|
|
|
|
2. Altered source versions must be plainly marked as such, and must not be
|
|
|
|
misrepresented as being the original software.
|
|
|
|
3. This notice may not be removed or altered from any source distribution.
|
|
|
|
|
|
|
|
cc9cii cc9c@iinet.net.au
|
|
|
|
|
|
|
|
*/
|
|
|
|
#include "reader.hpp"
|
|
|
|
|
|
|
|
#ifdef NDEBUG // FIXME: debugging only
|
|
|
|
#undef NDEBUG
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#undef DEBUG_GROUPSTACK
|
|
|
|
|
|
|
|
#include <cassert>
|
|
|
|
#include <iomanip> // for debugging
|
|
|
|
#include <iostream> // for debugging
|
|
|
|
#include <sstream> // for debugging
|
|
|
|
#include <stdexcept>
|
|
|
|
#include <unordered_map>
|
|
|
|
|
|
|
|
#if defined(_MSC_VER)
|
|
|
|
#pragma warning(push)
|
|
|
|
#pragma warning(disable : 4706)
|
|
|
|
#include <boost/iostreams/filter/zlib.hpp>
|
|
|
|
#pragma warning(pop)
|
|
|
|
#else
|
|
|
|
#include <boost/iostreams/filter/zlib.hpp>
|
|
|
|
#endif
|
|
|
|
#include <boost/iostreams/copy.hpp>
|
|
|
|
#include <boost/iostreams/filtering_streambuf.hpp>
|
|
|
|
|
|
|
|
#include <components/bsa/memorystream.hpp>
|
|
|
|
#include <components/files/constrainedfilestream.hpp>
|
|
|
|
#include <components/files/conversion.hpp>
|
|
|
|
#include <components/misc/strings/lower.hpp>
|
|
|
|
#include <components/to_utf8/to_utf8.hpp>
|
|
|
|
|
|
|
|
#include "formid.hpp"
|
|
|
|
#include "grouptype.hpp"
|
|
|
|
|
|
|
|
namespace ESM4
|
|
|
|
{
|
|
|
|
|
|
|
|
ReaderContext::ReaderContext()
|
|
|
|
: modIndex(0)
|
|
|
|
, recHeaderSize(sizeof(RecordHeader))
|
|
|
|
, filePos(0)
|
|
|
|
, fileRead(0)
|
|
|
|
, recordRead(0)
|
|
|
|
, currWorld({ 0, 0 })
|
|
|
|
, currCell({ 0, 0 })
|
|
|
|
, currCellGrid(FormId{ 0, 0 })
|
|
|
|
, cellGridValid(false)
|
|
|
|
{
|
|
|
|
subRecordHeader.typeId = 0;
|
|
|
|
subRecordHeader.dataSize = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
Reader::Reader(Files::IStreamPtr&& esmStream, const std::filesystem::path& filename, VFS::Manager const* vfs)
|
|
|
|
: mVFS(vfs)
|
|
|
|
, mEncoder(nullptr)
|
|
|
|
, mFileSize(0)
|
|
|
|
, mStream(std::move(esmStream))
|
|
|
|
{
|
|
|
|
// used by ESMReader only?
|
|
|
|
mCtx.filename = filename;
|
|
|
|
|
|
|
|
mCtx.fileRead = 0;
|
|
|
|
mStream->seekg(0, mStream->end);
|
|
|
|
mFileSize = mStream->tellg();
|
|
|
|
mStream->seekg(20); // go to the start but skip the "TES4" record header
|
|
|
|
|
|
|
|
mSavedStream.reset();
|
|
|
|
|
|
|
|
// determine header size
|
|
|
|
std::uint32_t subRecName = 0;
|
|
|
|
mStream->read((char*)&subRecName, sizeof(subRecName));
|
|
|
|
if (subRecName == 0x52444548) // "HEDR"
|
|
|
|
mCtx.recHeaderSize = sizeof(RecordHeader) - 4; // TES4 header size is 4 bytes smaller than TES5 header
|
|
|
|
else
|
|
|
|
mCtx.recHeaderSize = sizeof(RecordHeader);
|
|
|
|
|
|
|
|
// restart from the beginning (i.e. "TES4" record header)
|
|
|
|
mStream->seekg(0, mStream->beg);
|
|
|
|
#if 0
|
|
|
|
unsigned int esmVer = mHeader.mData.version.ui;
|
|
|
|
bool isTes4 = esmVer == ESM::VER_080 || esmVer == ESM::VER_100;
|
|
|
|
//bool isTes5 = esmVer == ESM::VER_094 || esmVer == ESM::VER_170;
|
|
|
|
//bool isFONV = esmVer == ESM::VER_132 || esmVer == ESM::VER_133 || esmVer == ESM::VER_134;
|
|
|
|
|
|
|
|
// TES4 header size is 4 bytes smaller than TES5 header
|
|
|
|
mCtx.recHeaderSize = isTes4 ? sizeof(ESM4::RecordHeader) - 4 : sizeof(ESM4::RecordHeader);
|
|
|
|
#endif
|
|
|
|
getRecordHeader();
|
|
|
|
if (mCtx.recordHeader.record.typeId == REC_TES4)
|
|
|
|
{
|
|
|
|
mHeader.load(*this);
|
|
|
|
mCtx.fileRead += mCtx.recordHeader.record.dataSize;
|
|
|
|
|
|
|
|
buildLStringIndex(); // for localised strings in Skyrim
|
|
|
|
}
|
|
|
|
else
|
|
|
|
fail("Unknown file format");
|
|
|
|
}
|
|
|
|
|
|
|
|
Reader::~Reader()
|
|
|
|
{
|
|
|
|
close();
|
|
|
|
}
|
|
|
|
|
|
|
|
// Since the record data may have been compressed, it is not always possible to use seek() to
|
|
|
|
// go to a position of a sub record.
|
|
|
|
//
|
|
|
|
// The record header needs to be saved in the context or the header needs to be re-loaded after
|
|
|
|
// restoring the context. The latter option was chosen.
|
|
|
|
ReaderContext Reader::getContext()
|
|
|
|
{
|
|
|
|
mCtx.filePos = mStream->tellg();
|
|
|
|
if (mCtx.filePos == std::streampos(-1))
|
|
|
|
return mCtx;
|
|
|
|
mCtx.filePos -= mCtx.recHeaderSize; // update file position
|
|
|
|
return mCtx;
|
|
|
|
}
|
|
|
|
|
|
|
|
// NOTE: Assumes that the caller has reopened the file if necessary
|
|
|
|
bool Reader::restoreContext(const ReaderContext& ctx)
|
|
|
|
{
|
|
|
|
if (mSavedStream) // TODO: doesn't seem to ever happen
|
|
|
|
{
|
|
|
|
mStream = std::move(mSavedStream);
|
|
|
|
}
|
|
|
|
|
|
|
|
mCtx.groupStack.clear(); // probably not necessary since it will be overwritten
|
|
|
|
mCtx = ctx;
|
|
|
|
mStream->seekg(ctx.filePos); // update file position
|
|
|
|
|
|
|
|
return getRecordHeader();
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::close()
|
|
|
|
{
|
|
|
|
mStream.reset();
|
|
|
|
// clearCtx();
|
|
|
|
// mHeader.blank();
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::openRaw(Files::IStreamPtr&& stream, const std::filesystem::path& filename)
|
|
|
|
{
|
|
|
|
close();
|
|
|
|
|
|
|
|
mStream = std::move(stream);
|
|
|
|
mCtx.filename = filename;
|
|
|
|
mCtx.fileRead = 0;
|
|
|
|
mStream->seekg(0, mStream->end);
|
|
|
|
mFileSize = mStream->tellg();
|
|
|
|
mStream->seekg(0, mStream->beg);
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::open(Files::IStreamPtr&& stream, const std::filesystem::path& filename)
|
|
|
|
{
|
|
|
|
openRaw(std::move(stream), filename);
|
|
|
|
|
|
|
|
// should at least have the size of ESM3 record header (20 or 24 bytes for ESM4)
|
|
|
|
assert(mFileSize >= 16);
|
|
|
|
std::uint32_t modVer = 0;
|
|
|
|
if (getExact(modVer)) // get the first 4 bytes of the record header only
|
|
|
|
{
|
|
|
|
// FIXME: need to setup header/context
|
|
|
|
if (modVer == REC_TES4)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
throw std::runtime_error("Unknown file format"); // can't yet use fail() as mCtx is not setup
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::open(const std::filesystem::path& filename)
|
|
|
|
{
|
|
|
|
open(Files::openConstrainedFileStream(filename), filename);
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::setRecHeaderSize(const std::size_t size)
|
|
|
|
{
|
|
|
|
mCtx.recHeaderSize = size;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::buildLStringIndex()
|
|
|
|
{
|
|
|
|
if ((mHeader.mFlags & Rec_ESM) == 0 || (mHeader.mFlags & Rec_Localized) == 0)
|
|
|
|
return;
|
|
|
|
|
|
|
|
const auto filename = mCtx.filename.stem().filename().u8string();
|
|
|
|
|
|
|
|
static const std::filesystem::path s("Strings");
|
|
|
|
buildLStringIndex(s / (filename + u8"_English.STRINGS"), Type_Strings);
|
|
|
|
buildLStringIndex(s / (filename + u8"_English.ILSTRINGS"), Type_ILStrings);
|
|
|
|
buildLStringIndex(s / (filename + u8"_English.DLSTRINGS"), Type_DLStrings);
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::buildLStringIndex(const std::filesystem::path& stringFile, LocalizedStringType stringType)
|
|
|
|
{
|
|
|
|
std::uint32_t numEntries;
|
|
|
|
std::uint32_t dataSize;
|
|
|
|
std::uint32_t stringId;
|
|
|
|
LStringOffset sp;
|
|
|
|
sp.type = stringType;
|
|
|
|
|
|
|
|
// TODO: possibly check if the resource exists?
|
|
|
|
Files::IStreamPtr filestream = mVFS
|
|
|
|
? mVFS->get(stringFile.string())
|
|
|
|
: Files::openConstrainedFileStream(mCtx.filename.parent_path() / stringFile);
|
|
|
|
|
|
|
|
filestream->seekg(0, std::ios::end);
|
|
|
|
std::size_t fileSize = filestream->tellg();
|
|
|
|
filestream->seekg(0, std::ios::beg);
|
|
|
|
|
|
|
|
std::istream* stream = filestream.get();
|
|
|
|
switch (stringType)
|
|
|
|
{
|
|
|
|
case Type_Strings:
|
|
|
|
mStrings = std::move(filestream);
|
|
|
|
break;
|
|
|
|
case Type_ILStrings:
|
|
|
|
mILStrings = std::move(filestream);
|
|
|
|
break;
|
|
|
|
case Type_DLStrings:
|
|
|
|
mDLStrings = std::move(filestream);
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
throw std::runtime_error("ESM4::Reader::unknown localised string type");
|
|
|
|
}
|
|
|
|
|
|
|
|
stream->read((char*)&numEntries, sizeof(numEntries));
|
|
|
|
stream->read((char*)&dataSize, sizeof(dataSize));
|
|
|
|
std::size_t dataStart = fileSize - dataSize;
|
|
|
|
for (unsigned int i = 0; i < numEntries; ++i)
|
|
|
|
{
|
|
|
|
stream->read((char*)&stringId, sizeof(stringId));
|
|
|
|
stream->read((char*)&sp.offset, sizeof(sp.offset));
|
|
|
|
sp.offset += (std::uint32_t)dataStart;
|
|
|
|
mLStringIndex[FormId::fromUint32(stringId)] = sp;
|
|
|
|
}
|
|
|
|
// assert (dataStart - stream->tell() == 0 && "String file start of data section mismatch");
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::getLocalizedString(std::string& str)
|
|
|
|
{
|
|
|
|
if (!hasLocalizedStrings())
|
|
|
|
return (void)getZString(str);
|
|
|
|
|
|
|
|
std::uint32_t stringId; // FormId
|
|
|
|
get(stringId);
|
|
|
|
if (stringId) // TES5 FoxRace, BOOK
|
|
|
|
getLocalizedStringImpl(FormId::fromUint32(stringId), str);
|
|
|
|
}
|
|
|
|
|
|
|
|
// FIXME: very messy and probably slow/inefficient
|
|
|
|
void Reader::getLocalizedStringImpl(const FormId stringId, std::string& str)
|
|
|
|
{
|
|
|
|
const std::map<FormId, LStringOffset>::const_iterator it = mLStringIndex.find(stringId);
|
|
|
|
|
|
|
|
if (it != mLStringIndex.end())
|
|
|
|
{
|
|
|
|
std::istream* filestream = nullptr;
|
|
|
|
|
|
|
|
switch (it->second.type)
|
|
|
|
{
|
|
|
|
case Type_Strings: // no string size provided
|
|
|
|
{
|
|
|
|
filestream = mStrings.get();
|
|
|
|
filestream->seekg(it->second.offset);
|
|
|
|
|
|
|
|
char ch;
|
|
|
|
std::vector<char> data;
|
|
|
|
do
|
|
|
|
{
|
|
|
|
filestream->read(&ch, sizeof(ch));
|
|
|
|
data.push_back(ch);
|
|
|
|
} while (ch != 0);
|
|
|
|
|
|
|
|
str = std::string(data.data());
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
case Type_ILStrings:
|
|
|
|
filestream = mILStrings.get();
|
|
|
|
break;
|
|
|
|
case Type_DLStrings:
|
|
|
|
filestream = mDLStrings.get();
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
throw std::runtime_error("ESM4::Reader::getLocalizedString unknown string type");
|
|
|
|
}
|
|
|
|
|
|
|
|
// get ILStrings or DLStrings (they provide string size)
|
|
|
|
filestream->seekg(it->second.offset);
|
|
|
|
std::uint32_t size = 0;
|
|
|
|
filestream->read((char*)&size, sizeof(size));
|
|
|
|
getStringImpl(str, size, *filestream, mEncoder, true); // expect null terminated string
|
|
|
|
}
|
|
|
|
else
|
|
|
|
throw std::runtime_error("ESM4::Reader::getLocalizedString localized string not found");
|
|
|
|
}
|
|
|
|
|
|
|
|
bool Reader::getRecordHeader()
|
|
|
|
{
|
|
|
|
// FIXME: this seems very hacky but we may have skipped subrecords from within an inflated data block
|
|
|
|
if (/*mStream->eof() && */ mSavedStream)
|
|
|
|
{
|
|
|
|
mStream = std::move(mSavedStream);
|
|
|
|
}
|
|
|
|
|
|
|
|
mStream->read((char*)&mCtx.recordHeader, mCtx.recHeaderSize);
|
|
|
|
std::size_t bytesRead = (std::size_t)mStream->gcount();
|
|
|
|
|
|
|
|
// keep track of data left to read from the file
|
|
|
|
mCtx.fileRead += mCtx.recHeaderSize;
|
|
|
|
|
|
|
|
mCtx.recordRead = 0; // for keeping track of sub records
|
|
|
|
|
|
|
|
// After reading the record header we can cache a WRLD or CELL formId for convenient access later.
|
|
|
|
// FIXME: currently currWorld and currCell are set manually when loading the WRLD and CELL records
|
|
|
|
|
|
|
|
// HACK: mCtx.groupStack.back() is updated before the record data are read/skipped
|
|
|
|
// N.B. the data must be fully read/skipped for this to work
|
|
|
|
if (mCtx.recordHeader.record.typeId != REC_GRUP && !mCtx.groupStack.empty())
|
|
|
|
{
|
|
|
|
mCtx.groupStack.back().second += (std::uint32_t)mCtx.recHeaderSize + mCtx.recordHeader.record.dataSize;
|
|
|
|
|
|
|
|
// keep track of data left to read from the file
|
|
|
|
mCtx.fileRead += mCtx.recordHeader.record.dataSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
return bytesRead == mCtx.recHeaderSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::getRecordData(bool dump)
|
|
|
|
{
|
|
|
|
std::uint32_t uncompressedSize = 0;
|
|
|
|
|
|
|
|
if ((mCtx.recordHeader.record.flags & Rec_Compressed) != 0)
|
|
|
|
{
|
|
|
|
mStream->read(reinterpret_cast<char*>(&uncompressedSize), sizeof(std::uint32_t));
|
|
|
|
|
|
|
|
std::size_t recordSize = mCtx.recordHeader.record.dataSize - sizeof(std::uint32_t);
|
|
|
|
Bsa::MemoryInputStream compressedRecord(recordSize);
|
|
|
|
mStream->read(compressedRecord.getRawData(), recordSize);
|
|
|
|
std::istream* fileStream = (std::istream*)&compressedRecord;
|
|
|
|
mSavedStream = std::move(mStream);
|
|
|
|
|
|
|
|
mCtx.recordHeader.record.dataSize = uncompressedSize - sizeof(uncompressedSize);
|
|
|
|
|
|
|
|
auto memoryStreamPtr = std::make_unique<Bsa::MemoryInputStream>(uncompressedSize);
|
|
|
|
|
|
|
|
boost::iostreams::filtering_streambuf<boost::iostreams::input> inputStreamBuf;
|
|
|
|
inputStreamBuf.push(boost::iostreams::zlib_decompressor());
|
|
|
|
inputStreamBuf.push(*fileStream);
|
|
|
|
|
|
|
|
boost::iostreams::basic_array_sink<char> sr(memoryStreamPtr->getRawData(), uncompressedSize);
|
|
|
|
boost::iostreams::copy(inputStreamBuf, sr);
|
|
|
|
|
|
|
|
// For debugging only
|
|
|
|
// #if 0
|
|
|
|
if (dump)
|
|
|
|
{
|
|
|
|
std::ostringstream ss;
|
|
|
|
char* data = memoryStreamPtr->getRawData();
|
|
|
|
for (unsigned int i = 0; i < uncompressedSize; ++i)
|
|
|
|
{
|
|
|
|
if (data[i] > 64 && data[i] < 91)
|
|
|
|
ss << (char)(data[i]) << " ";
|
|
|
|
else
|
|
|
|
ss << std::setfill('0') << std::setw(2) << std::hex << (int)(data[i]);
|
|
|
|
if ((i & 0x000f) == 0xf)
|
|
|
|
ss << "\n";
|
|
|
|
else if (i < uncompressedSize - 1)
|
|
|
|
ss << " ";
|
|
|
|
}
|
|
|
|
std::cout << ss.str() << std::endl;
|
|
|
|
}
|
|
|
|
// #endif
|
|
|
|
mStream = std::make_unique<Files::StreamWithBuffer<Bsa::MemoryInputStream>>(std::move(memoryStreamPtr));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::skipRecordData()
|
|
|
|
{
|
|
|
|
assert(mCtx.recordRead <= mCtx.recordHeader.record.dataSize && "Skipping after reading more than available");
|
|
|
|
mStream->ignore(mCtx.recordHeader.record.dataSize - mCtx.recordRead);
|
|
|
|
mCtx.recordRead = mCtx.recordHeader.record.dataSize; // for getSubRecordHeader()
|
|
|
|
}
|
|
|
|
|
|
|
|
bool Reader::getSubRecordHeader()
|
|
|
|
{
|
|
|
|
bool result = false;
|
|
|
|
// NOTE: some SubRecords have 0 dataSize (e.g. SUB_RDSD in one of REC_REGN records in Oblivion.esm).
|
|
|
|
// Also SUB_XXXX has zero dataSize and the following 4 bytes represent the actual dataSize
|
|
|
|
// - hence it require manual updtes to mCtx.recordRead via updateRecordRead()
|
|
|
|
// See ESM4::NavMesh and ESM4::World.
|
|
|
|
if (mCtx.recordHeader.record.dataSize - mCtx.recordRead >= sizeof(mCtx.subRecordHeader))
|
|
|
|
{
|
|
|
|
result = getExact(mCtx.subRecordHeader);
|
|
|
|
// HACK: below assumes sub-record data will be read or skipped in full;
|
|
|
|
// this hack aims to avoid updating mCtx.recordRead each time anything is read
|
|
|
|
mCtx.recordRead += (sizeof(mCtx.subRecordHeader) + mCtx.subRecordHeader.dataSize);
|
|
|
|
}
|
|
|
|
else if (mCtx.recordRead > mCtx.recordHeader.record.dataSize)
|
|
|
|
{
|
|
|
|
// try to correct any overshoot, seek to the end of the expected data
|
|
|
|
// this will only work if mCtx.subRecordHeader.dataSize was fully read or skipped
|
|
|
|
// (i.e. it will only correct mCtx.subRecordHeader.dataSize being incorrect)
|
|
|
|
// TODO: not tested
|
|
|
|
std::uint32_t overshoot = (std::uint32_t)mCtx.recordRead - mCtx.recordHeader.record.dataSize;
|
|
|
|
|
|
|
|
std::size_t pos = mStream->tellg();
|
|
|
|
mStream->seekg(pos - overshoot);
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::skipSubRecordData()
|
|
|
|
{
|
|
|
|
mStream->ignore(mCtx.subRecordHeader.dataSize);
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::skipSubRecordData(std::uint32_t size)
|
|
|
|
{
|
|
|
|
mStream->ignore(size);
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::enterGroup()
|
|
|
|
{
|
|
|
|
#ifdef DEBUG_GROUPSTACK
|
|
|
|
std::string padding; // FIXME: debugging only
|
|
|
|
padding.insert(0, mCtx.groupStack.size() * 2, ' ');
|
|
|
|
std::cout << padding << "Starting record group "
|
|
|
|
<< printLabel(mCtx.recordHeader.group.label, mCtx.recordHeader.group.type) << std::endl;
|
|
|
|
#endif
|
|
|
|
// empty group if the group size is same as the header size
|
|
|
|
if (mCtx.recordHeader.group.groupSize == (std::uint32_t)mCtx.recHeaderSize)
|
|
|
|
{
|
|
|
|
#ifdef DEBUG_GROUPSTACK
|
|
|
|
std::cout << padding << "Ignoring record group " // FIXME: debugging only
|
|
|
|
<< printLabel(mCtx.recordHeader.group.label, mCtx.recordHeader.group.type) << " (empty)"
|
|
|
|
<< std::endl;
|
|
|
|
#endif
|
|
|
|
if (!mCtx.groupStack.empty()) // top group may be empty (e.g. HAIR in Skyrim)
|
|
|
|
{
|
|
|
|
// don't put on the stack, exitGroupCheck() may not get called before recursing into this method
|
|
|
|
mCtx.groupStack.back().second += mCtx.recordHeader.group.groupSize;
|
|
|
|
exitGroupCheck();
|
|
|
|
}
|
|
|
|
|
|
|
|
return; // don't push an empty group, just return
|
|
|
|
}
|
|
|
|
|
|
|
|
// push group
|
|
|
|
mCtx.groupStack.push_back(std::make_pair(mCtx.recordHeader.group, (std::uint32_t)mCtx.recHeaderSize));
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::exitGroupCheck()
|
|
|
|
{
|
|
|
|
if (mCtx.groupStack.empty())
|
|
|
|
return;
|
|
|
|
|
|
|
|
// pop finished groups (note reading too much is allowed here)
|
|
|
|
std::uint32_t lastGroupSize = mCtx.groupStack.back().first.groupSize;
|
|
|
|
while (mCtx.groupStack.back().second >= lastGroupSize)
|
|
|
|
{
|
|
|
|
#ifdef DEBUG_GROUPSTACK
|
|
|
|
GroupTypeHeader grp = mCtx.groupStack.back().first; // FIXME: grp is for debugging only
|
|
|
|
#endif
|
|
|
|
// try to correct any overshoot
|
|
|
|
// TODO: not tested
|
|
|
|
std::uint32_t overshoot = mCtx.groupStack.back().second - lastGroupSize;
|
|
|
|
if (overshoot > 0)
|
|
|
|
{
|
|
|
|
std::size_t pos = mStream->tellg();
|
|
|
|
mStream->seekg(pos - overshoot);
|
|
|
|
}
|
|
|
|
|
|
|
|
mCtx.groupStack.pop_back();
|
|
|
|
#ifdef DEBUG_GROUPSTACK
|
|
|
|
std::string padding; // FIXME: debugging only
|
|
|
|
padding.insert(0, mCtx.groupStack.size() * 2, ' ');
|
|
|
|
std::cout << padding << "Finished record group " << printLabel(grp.label, grp.type) << std::endl;
|
|
|
|
#endif
|
|
|
|
// if the previous group was the final one no need to do below
|
|
|
|
if (mCtx.groupStack.empty())
|
|
|
|
return;
|
|
|
|
|
|
|
|
mCtx.groupStack.back().second += lastGroupSize;
|
|
|
|
lastGroupSize = mCtx.groupStack.back().first.groupSize;
|
|
|
|
|
|
|
|
assert(lastGroupSize >= mCtx.groupStack.back().second && "Read more records than available");
|
|
|
|
// #if 0
|
|
|
|
if (mCtx.groupStack.back().second > lastGroupSize) // FIXME: debugging only
|
|
|
|
std::cerr << printLabel(mCtx.groupStack.back().first.label, mCtx.groupStack.back().first.type)
|
|
|
|
<< " read more records than available" << std::endl;
|
|
|
|
// #endif
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// WARNING: this method should be used after first calling enterGroup()
|
|
|
|
// else the method may try to dereference an element that does not exist
|
|
|
|
const GroupTypeHeader& Reader::grp(std::size_t pos) const
|
|
|
|
{
|
|
|
|
assert(pos <= mCtx.groupStack.size() - 1 && "ESM4::Reader::grp - exceeded stack depth");
|
|
|
|
|
|
|
|
return (*(mCtx.groupStack.end() - pos - 1)).first;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::skipGroupData()
|
|
|
|
{
|
|
|
|
assert(!mCtx.groupStack.empty() && "Skipping group with an empty stack");
|
|
|
|
|
|
|
|
// subtract what was already read/skipped
|
|
|
|
std::uint32_t skipSize = mCtx.groupStack.back().first.groupSize - mCtx.groupStack.back().second;
|
|
|
|
|
|
|
|
mStream->ignore(skipSize);
|
|
|
|
|
|
|
|
// keep track of data left to read from the file
|
|
|
|
mCtx.fileRead += skipSize;
|
|
|
|
|
|
|
|
mCtx.groupStack.back().second = mCtx.groupStack.back().first.groupSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::skipGroup()
|
|
|
|
{
|
|
|
|
#ifdef DEBUG_GROUPSTACK
|
|
|
|
std::string padding; // FIXME: debugging only
|
|
|
|
padding.insert(0, mCtx.groupStack.size() * 2, ' ');
|
|
|
|
std::cout << padding << "Skipping record group "
|
|
|
|
<< printLabel(mCtx.recordHeader.group.label, mCtx.recordHeader.group.type) << std::endl;
|
|
|
|
#endif
|
|
|
|
// subtract the size of header already read before skipping
|
|
|
|
std::uint32_t skipSize = mCtx.recordHeader.group.groupSize - (std::uint32_t)mCtx.recHeaderSize;
|
|
|
|
mStream->ignore(skipSize);
|
|
|
|
|
|
|
|
// keep track of data left to read from the file
|
|
|
|
mCtx.fileRead += skipSize;
|
|
|
|
|
|
|
|
// NOTE: mCtx.groupStack.back().second already has mCtx.recHeaderSize from enterGroup()
|
|
|
|
if (!mCtx.groupStack.empty())
|
|
|
|
mCtx.groupStack.back().second += mCtx.recordHeader.group.groupSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
const CellGrid& Reader::currCellGrid() const
|
|
|
|
{
|
|
|
|
// Maybe should throw an exception instead?
|
|
|
|
assert(mCtx.cellGridValid && "Attempt to use an invalid cell grid");
|
|
|
|
|
|
|
|
return mCtx.currCellGrid;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::updateModIndices(const std::map<std::string, int>& fileToModIndex)
|
|
|
|
{
|
|
|
|
mCtx.parentFileIndices.resize(mHeader.mMaster.size());
|
|
|
|
for (unsigned int i = 0; i < mHeader.mMaster.size(); ++i)
|
|
|
|
{
|
|
|
|
// locate the position of the dependency in already loaded files
|
|
|
|
auto it = fileToModIndex.find(Misc::StringUtils::lowerCase(mHeader.mMaster[i].name));
|
|
|
|
if (it != fileToModIndex.end())
|
|
|
|
mCtx.parentFileIndices[i] = it->second;
|
|
|
|
else
|
|
|
|
throw std::runtime_error(
|
|
|
|
"ESM4::Reader::updateModIndices required dependency '" + mHeader.mMaster[i].name + "' not found");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// ModIndex adjusted formId according to master file dependencies
|
|
|
|
// (see http://www.uesp.net/wiki/Tes4Mod:FormID_Fixup)
|
|
|
|
// NOTE: need to update modindex to parentFileIndices.size() before saving
|
|
|
|
//
|
|
|
|
// FIXME: probably should add a parameter to check for mCtx.header::mOverrides
|
|
|
|
// (ACHR, LAND, NAVM, PGRE, PHZD, REFR), but not sure what exactly overrides mean
|
|
|
|
// i.e. use the modindx of its master?
|
|
|
|
// FIXME: Apparently ModIndex '00' in an ESP means the object is defined in one of its masters.
|
|
|
|
// This means we may need to search multiple times to get the correct id.
|
|
|
|
// (see https://www.uesp.net/wiki/Tes4Mod:Formid#ModIndex_Zero)
|
|
|
|
void Reader::adjustFormId(FormId& id) const
|
|
|
|
{
|
|
|
|
if (id.hasContentFile() && id.mContentFile < static_cast<int>(mCtx.parentFileIndices.size()))
|
|
|
|
id.mContentFile = mCtx.parentFileIndices[id.mContentFile];
|
|
|
|
else
|
|
|
|
id.mContentFile = mCtx.modIndex;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::adjustFormId(FormId32& id) const
|
|
|
|
{
|
|
|
|
FormId formId = FormId::fromUint32(id);
|
|
|
|
adjustFormId(formId);
|
|
|
|
id = formId.toUint32();
|
|
|
|
}
|
|
|
|
|
|
|
|
bool Reader::getFormId(FormId& id)
|
|
|
|
{
|
|
|
|
FormId32 v;
|
|
|
|
if (!getExact(v))
|
|
|
|
return false;
|
|
|
|
id = FormId::fromUint32(v);
|
|
|
|
|
|
|
|
adjustFormId(id);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
ESM::FormIdRefId Reader::getRefIdFromHeader() const
|
|
|
|
{
|
|
|
|
FormId formId = hdr().record.getFormId();
|
|
|
|
adjustFormId(formId);
|
|
|
|
return ESM::FormIdRefId(formId);
|
|
|
|
}
|
|
|
|
|
|
|
|
void Reader::adjustGRUPFormId()
|
|
|
|
{
|
|
|
|
adjustFormId(mCtx.recordHeader.group.label.value);
|
|
|
|
}
|
|
|
|
|
|
|
|
[[noreturn]] void Reader::fail(const std::string& msg)
|
|
|
|
{
|
|
|
|
std::stringstream ss;
|
|
|
|
|
|
|
|
ss << "ESM Error: " << msg;
|
|
|
|
ss << "\n File: " << Files::pathToUnicodeString(mCtx.filename);
|
|
|
|
ss << "\n Record: " << ESM::printName(mCtx.recordHeader.record.typeId);
|
|
|
|
ss << "\n Subrecord: " << ESM::printName(mCtx.subRecordHeader.typeId);
|
|
|
|
if (mStream.get())
|
|
|
|
ss << "\n Offset: 0x" << std::hex << mStream->tellg();
|
|
|
|
|
|
|
|
throw std::runtime_error(ss.str());
|
|
|
|
}
|
|
|
|
|
|
|
|
bool Reader::getStringImpl(std::string& str, std::size_t size, std::istream& stream,
|
|
|
|
const ToUTF8::StatelessUtf8Encoder* encoder, bool hasNull)
|
|
|
|
{
|
|
|
|
std::size_t newSize = size;
|
|
|
|
|
|
|
|
if (encoder)
|
|
|
|
{
|
|
|
|
std::string input(size, '\0');
|
|
|
|
stream.read(input.data(), size);
|
|
|
|
if (stream.gcount() == static_cast<std::streamsize>(size))
|
|
|
|
{
|
|
|
|
const std::string_view result
|
|
|
|
= encoder->getUtf8(input, ToUTF8::BufferAllocationPolicy::FitToRequiredSize, str);
|
|
|
|
if (str.empty() && !result.empty())
|
|
|
|
{
|
|
|
|
str = std::move(input);
|
|
|
|
str.resize(result.size());
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
if (hasNull)
|
|
|
|
newSize -= 1; // don't read the null terminator yet
|
|
|
|
|
|
|
|
str.resize(newSize); // assumed C++11
|
|
|
|
stream.read(str.data(), newSize);
|
|
|
|
if (static_cast<std::size_t>(stream.gcount()) == newSize)
|
|
|
|
{
|
|
|
|
if (hasNull)
|
|
|
|
{
|
|
|
|
char ch;
|
|
|
|
stream.read(&ch, 1); // read the null terminator
|
|
|
|
if (ch != '\0')
|
|
|
|
throw std::runtime_error("ESM4::Reader::getString string is not terminated with a null");
|
|
|
|
}
|
|
|
|
#if 0
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// NOTE: normal ESMs don't but omwsave has locals or spells with null terminator
|
|
|
|
assert (str[newSize - 1] != '\0'
|
|
|
|
&& "ESM4::Reader::getString string is unexpectedly terminated with a null");
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
str.clear();
|
|
|
|
return false; // FIXME: throw instead?
|
|
|
|
}
|
|
|
|
|
|
|
|
bool Reader::getZeroTerminatedStringArray(std::vector<std::string>& values)
|
|
|
|
{
|
|
|
|
const std::size_t size = mCtx.subRecordHeader.dataSize;
|
|
|
|
std::string input(size, '\0');
|
|
|
|
mStream->read(input.data(), size);
|
|
|
|
|
|
|
|
if (mStream->gcount() != static_cast<std::streamsize>(size))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
std::string_view inputView(input.data(), input.size());
|
|
|
|
std::string buffer;
|
|
|
|
while (true)
|
|
|
|
{
|
|
|
|
std::string_view value(inputView.data());
|
|
|
|
const std::size_t next = inputView.find_first_not_of('\0', value.size());
|
|
|
|
if (mEncoder != nullptr)
|
|
|
|
value = mEncoder->getUtf8(value, ToUTF8::BufferAllocationPolicy::UseGrowFactor, buffer);
|
|
|
|
values.emplace_back(value);
|
|
|
|
if (next == std::string_view::npos)
|
|
|
|
break;
|
|
|
|
inputView = inputView.substr(next);
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
namespace
|
|
|
|
{
|
|
|
|
constexpr std::string_view sGroupType[] = {
|
|
|
|
"Record Type",
|
|
|
|
"World Child",
|
|
|
|
"Interior Cell",
|
|
|
|
"Interior Sub Cell",
|
|
|
|
"Exterior Cell",
|
|
|
|
"Exterior Sub Cell",
|
|
|
|
"Cell Child",
|
|
|
|
"Topic Child",
|
|
|
|
"Cell Persistent Child",
|
|
|
|
"Cell Temporary Child",
|
|
|
|
"Cell Visible Dist Child",
|
|
|
|
"Unknown",
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string printLabel(const GroupLabel& label, const std::uint32_t type)
|
|
|
|
{
|
|
|
|
std::ostringstream ss;
|
|
|
|
ss << sGroupType[std::min<std::size_t>(type, std::size(sGroupType))]; // avoid out of range
|
|
|
|
|
|
|
|
switch (type)
|
|
|
|
{
|
|
|
|
case ESM4::Grp_RecordType:
|
|
|
|
{
|
|
|
|
ss << ": " << std::string((char*)label.recordType, 4);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case ESM4::Grp_ExteriorCell:
|
|
|
|
case ESM4::Grp_ExteriorSubCell:
|
|
|
|
{
|
|
|
|
// short x, y;
|
|
|
|
// y = label & 0xff;
|
|
|
|
// x = (label >> 16) & 0xff;
|
|
|
|
ss << ": grid (x, y) " << std::dec << label.grid[1] << ", " << label.grid[0];
|
|
|
|
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case ESM4::Grp_InteriorCell:
|
|
|
|
case ESM4::Grp_InteriorSubCell:
|
|
|
|
{
|
|
|
|
ss << ": block 0x" << std::hex << label.value;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case ESM4::Grp_WorldChild:
|
|
|
|
case ESM4::Grp_CellChild:
|
|
|
|
case ESM4::Grp_TopicChild:
|
|
|
|
case ESM4::Grp_CellPersistentChild:
|
|
|
|
case ESM4::Grp_CellTemporaryChild:
|
|
|
|
case ESM4::Grp_CellVisibleDistChild:
|
|
|
|
{
|
|
|
|
ss << ": FormId 0x" << formIdToString(FormId::fromUint32(label.value));
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
return ss.str();
|
|
|
|
}
|
|
|
|
}
|