mirror of
https://github.com/OpenMW/openmw.git
synced 2025-11-30 11:34:32 +00:00
Only reserve capacity when reading collections
But do not initialize. If the meta information is invalid and has a big value, initialization will take significant amount of time but there might be no actual data in the file because it's too small.
This commit is contained in:
parent
434f450778
commit
246759ecd2
16 changed files with 252 additions and 193 deletions
|
|
@ -8,6 +8,13 @@
|
|||
|
||||
namespace Nif
|
||||
{
|
||||
namespace
|
||||
{
|
||||
void readSkinnedShapeGroup(NIFStream& stream, std::vector<NiBoneLODController::SkinInfo>& value)
|
||||
{
|
||||
stream.readVectorOfRecords<uint32_t>(value);
|
||||
}
|
||||
}
|
||||
|
||||
void NiTimeController::read(NIFStream* nif)
|
||||
{
|
||||
|
|
@ -84,11 +91,10 @@ namespace Nif
|
|||
nif->read(mAccumRootName);
|
||||
mTextKeys.read(nif);
|
||||
}
|
||||
mControlledBlocks.resize(nif->get<uint32_t>());
|
||||
const uint32_t size = nif->get<uint32_t>();
|
||||
if (nif->getVersion() >= NIFStream::generateVersion(10, 1, 0, 106))
|
||||
nif->read(mArrayGrowBy);
|
||||
for (ControlledBlock& block : mControlledBlocks)
|
||||
block.read(nif);
|
||||
nif->readVectorOfRecords(size, mControlledBlocks);
|
||||
}
|
||||
|
||||
void NiSequence::post(Reader& nif)
|
||||
|
|
@ -122,13 +128,8 @@ namespace Nif
|
|||
mStringPalette.read(nif);
|
||||
else if (nif->getVersion() >= NIFFile::NIFVersion::VER_BGS && nif->getBethVersion() >= 24)
|
||||
{
|
||||
if (nif->getBethVersion() >= 29)
|
||||
mAnimNotesList.resize(nif->get<uint16_t>());
|
||||
else
|
||||
mAnimNotesList.resize(1);
|
||||
|
||||
for (auto& notes : mAnimNotesList)
|
||||
notes.read(nif);
|
||||
const uint16_t size = nif->getBethVersion() >= 29 ? nif->get<uint16_t>() : 1;
|
||||
nif->readVectorOfRecords(size, mAnimNotesList);
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -482,29 +483,27 @@ namespace Nif
|
|||
mData.post(nif);
|
||||
}
|
||||
|
||||
void NiBoneLODController::SkinInfo::read(NIFStream* nif)
|
||||
{
|
||||
mShape.read(nif);
|
||||
mSkin.read(nif);
|
||||
}
|
||||
|
||||
void NiBoneLODController::read(NIFStream* nif)
|
||||
{
|
||||
NiTimeController::read(nif);
|
||||
|
||||
nif->read(mLOD);
|
||||
mNodeGroups.resize(nif->get<uint32_t>());
|
||||
const uint32_t nodeGroupsCount = nif->get<uint32_t>();
|
||||
mNodeGroups.reserve(nodeGroupsCount);
|
||||
nif->read(mNumNodeGroups);
|
||||
for (NiAVObjectList& group : mNodeGroups)
|
||||
readRecordList(nif, group);
|
||||
for (uint32_t i = 0; i < nodeGroupsCount; ++i)
|
||||
readRecordList(nif, mNodeGroups.emplace_back());
|
||||
|
||||
if (nif->getBethVersion() != 0 || nif->getVersion() < NIFStream::generateVersion(4, 2, 2, 0))
|
||||
return;
|
||||
|
||||
mSkinnedShapeGroups.resize(nif->get<uint32_t>());
|
||||
for (std::vector<SkinInfo>& group : mSkinnedShapeGroups)
|
||||
{
|
||||
group.resize(nif->get<uint32_t>());
|
||||
for (SkinInfo& info : group)
|
||||
{
|
||||
info.mShape.read(nif);
|
||||
info.mSkin.read(nif);
|
||||
}
|
||||
}
|
||||
nif->readVectorOfRecords<uint32_t>(readSkinnedShapeGroup, mSkinnedShapeGroups);
|
||||
readRecordList(nif, mShapeGroups);
|
||||
}
|
||||
|
||||
|
|
@ -892,9 +891,7 @@ namespace Nif
|
|||
|
||||
void BSTreadTransfInterpolator::read(NIFStream* nif)
|
||||
{
|
||||
mTransforms.resize(nif->get<uint32_t>());
|
||||
for (BSTreadTransform& transform : mTransforms)
|
||||
transform.read(nif);
|
||||
nif->readVectorOfRecords<uint32_t>(mTransforms);
|
||||
mData.read(nif);
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -322,6 +322,8 @@ namespace Nif
|
|||
{
|
||||
NiTriBasedGeomPtr mShape;
|
||||
NiSkinInstancePtr mSkin;
|
||||
|
||||
void read(NIFStream* nif);
|
||||
};
|
||||
|
||||
uint32_t mLOD;
|
||||
|
|
|
|||
|
|
@ -8,6 +8,46 @@
|
|||
|
||||
namespace Nif
|
||||
{
|
||||
namespace
|
||||
{
|
||||
void readNiSkinDataVertWeight(NIFStream& stream, NiSkinData::VertWeight& value)
|
||||
{
|
||||
auto& [vertex, weight] = value;
|
||||
stream.read(vertex);
|
||||
stream.read(weight);
|
||||
}
|
||||
|
||||
struct ReadNiSkinDataBoneInfo
|
||||
{
|
||||
bool mHasVertexWeights;
|
||||
|
||||
void operator()(NIFStream& stream, NiSkinData::BoneInfo& value) const
|
||||
{
|
||||
stream.read(value.mTransform);
|
||||
stream.read(value.mBoundSphere);
|
||||
|
||||
const uint16_t numVertices = stream.get<uint16_t>();
|
||||
|
||||
if (!mHasVertexWeights)
|
||||
return;
|
||||
|
||||
stream.readVectorOfRecords(numVertices, readNiSkinDataVertWeight, value.mWeights);
|
||||
}
|
||||
};
|
||||
|
||||
struct ReadNiMorphDataMorphData
|
||||
{
|
||||
uint32_t mNumVerts;
|
||||
|
||||
void operator()(NIFStream& stream, NiMorphData::MorphData& value) const
|
||||
{
|
||||
value.mKeyFrames = std::make_shared<FloatKeyMap>();
|
||||
value.mKeyFrames->read(&stream, /*morph*/ true);
|
||||
stream.readVector(value.mVertices, mNumVerts);
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
void NiGeometryData::read(NIFStream* nif)
|
||||
{
|
||||
if (nif->getVersion() >= NIFStream::generateVersion(10, 1, 0, 114))
|
||||
|
|
@ -215,20 +255,21 @@ namespace Nif
|
|||
nif->read(mSigned);
|
||||
}
|
||||
|
||||
void NiPixelData::Mipmap::read(NIFStream* nif)
|
||||
{
|
||||
nif->read(mWidth);
|
||||
nif->read(mHeight);
|
||||
nif->read(mOffset);
|
||||
}
|
||||
|
||||
void NiPixelData::read(NIFStream* nif)
|
||||
{
|
||||
mPixelFormat.read(nif);
|
||||
mPalette.read(nif);
|
||||
mMipmaps.resize(nif->get<uint32_t>());
|
||||
const uint32_t mipmapsCount = nif->get<uint32_t>();
|
||||
nif->read(mBytesPerPixel);
|
||||
for (Mipmap& mip : mMipmaps)
|
||||
{
|
||||
nif->read(mip.mWidth);
|
||||
nif->read(mip.mHeight);
|
||||
nif->read(mip.mOffset);
|
||||
}
|
||||
uint32_t numPixels;
|
||||
nif->read(numPixels);
|
||||
nif->readVectorOfRecords(mipmapsCount, mMipmaps);
|
||||
const uint32_t numPixels = nif->get<uint32_t>();
|
||||
if (nif->getVersion() >= NIFStream::generateVersion(10, 4, 0, 2))
|
||||
nif->read(mNumFaces);
|
||||
nif->readVector(mData, numPixels * mNumFaces);
|
||||
|
|
@ -247,12 +288,14 @@ namespace Nif
|
|||
|
||||
void NiVisData::read(NIFStream* nif)
|
||||
{
|
||||
mKeys = std::make_shared<std::vector<std::pair<float, bool>>>(nif->get<uint32_t>());
|
||||
for (auto& [time, value] : *mKeys)
|
||||
{
|
||||
nif->read(time);
|
||||
value = nif->get<uint8_t>() != 0;
|
||||
}
|
||||
const auto readPair = [](NIFStream& stream, std::pair<float, bool>& value) {
|
||||
stream.read(value.first);
|
||||
value.second = stream.get<uint8_t>() != 0;
|
||||
};
|
||||
|
||||
std::vector<std::pair<float, bool>> keys;
|
||||
nif->readVectorOfRecords<uint32_t>(readPair, keys);
|
||||
mKeys = std::make_shared<std::vector<std::pair<float, bool>>>(std::move(keys));
|
||||
}
|
||||
|
||||
void NiSkinInstance::read(NIFStream* nif)
|
||||
|
|
@ -296,16 +339,17 @@ namespace Nif
|
|||
return partitions;
|
||||
}
|
||||
|
||||
void BSDismemberSkinInstance::BodyPart::read(NIFStream* nif)
|
||||
{
|
||||
nif->read(mFlags);
|
||||
nif->read(mType);
|
||||
}
|
||||
|
||||
void BSDismemberSkinInstance::read(NIFStream* nif)
|
||||
{
|
||||
NiSkinInstance::read(nif);
|
||||
|
||||
mParts.resize(nif->get<uint32_t>());
|
||||
for (BodyPart& part : mParts)
|
||||
{
|
||||
nif->read(part.mFlags);
|
||||
nif->read(part.mType);
|
||||
}
|
||||
nif->readVectorOfRecords<uint32_t>(mParts);
|
||||
}
|
||||
|
||||
void BSSkinInstance::read(NIFStream* nif)
|
||||
|
|
@ -339,8 +383,7 @@ namespace Nif
|
|||
{
|
||||
nif->read(mTransform);
|
||||
|
||||
uint32_t numBones;
|
||||
nif->read(numBones);
|
||||
const uint32_t numBones = nif->get<uint32_t>();
|
||||
bool hasVertexWeights = true;
|
||||
if (nif->getVersion() >= NIFFile::NIFVersion::VER_MW)
|
||||
{
|
||||
|
|
@ -351,25 +394,8 @@ namespace Nif
|
|||
nif->read(hasVertexWeights);
|
||||
}
|
||||
|
||||
mBones.resize(numBones);
|
||||
for (BoneInfo& bi : mBones)
|
||||
{
|
||||
nif->read(bi.mTransform);
|
||||
nif->read(bi.mBoundSphere);
|
||||
|
||||
uint16_t numVertices;
|
||||
nif->read(numVertices);
|
||||
|
||||
if (!hasVertexWeights)
|
||||
continue;
|
||||
|
||||
bi.mWeights.resize(numVertices);
|
||||
for (auto& [vertex, weight] : bi.mWeights)
|
||||
{
|
||||
nif->read(vertex);
|
||||
nif->read(weight);
|
||||
}
|
||||
}
|
||||
const ReadNiSkinDataBoneInfo readBoneInfo{ .mHasVertexWeights = hasVertexWeights };
|
||||
nif->readVectorOfRecords(numBones, readBoneInfo, mBones);
|
||||
}
|
||||
|
||||
void NiSkinData::post(Reader& nif)
|
||||
|
|
@ -467,18 +493,12 @@ namespace Nif
|
|||
|
||||
void NiMorphData::read(NIFStream* nif)
|
||||
{
|
||||
uint32_t numMorphs, numVerts;
|
||||
nif->read(numMorphs);
|
||||
nif->read(numVerts);
|
||||
const uint32_t numMorphs = nif->get<uint32_t>();
|
||||
const uint32_t numVerts = nif->get<uint32_t>();
|
||||
nif->read(mRelativeTargets);
|
||||
|
||||
mMorphs.resize(numMorphs);
|
||||
for (MorphData& morph : mMorphs)
|
||||
{
|
||||
morph.mKeyFrames = std::make_shared<FloatKeyMap>();
|
||||
morph.mKeyFrames->read(nif, /*morph*/ true);
|
||||
nif->readVector(morph.mVertices, numVerts);
|
||||
}
|
||||
const ReadNiMorphDataMorphData readMorph{ .mNumVerts = numVerts };
|
||||
nif->readVectorOfRecords(numMorphs, readMorph, mMorphs);
|
||||
}
|
||||
|
||||
void NiKeyframeData::read(NIFStream* nif)
|
||||
|
|
|
|||
|
|
@ -169,8 +169,11 @@ namespace Nif
|
|||
{
|
||||
struct Mipmap
|
||||
{
|
||||
uint32_t mWidth, mHeight;
|
||||
uint32_t mWidth;
|
||||
uint32_t mHeight;
|
||||
uint32_t mOffset;
|
||||
|
||||
void read(NIFStream* nif);
|
||||
};
|
||||
|
||||
NiPixelFormat mPixelFormat;
|
||||
|
|
@ -218,6 +221,8 @@ namespace Nif
|
|||
{
|
||||
uint16_t mFlags;
|
||||
uint16_t mType;
|
||||
|
||||
void read(NIFStream* nif);
|
||||
};
|
||||
|
||||
std::vector<BodyPart> mParts;
|
||||
|
|
|
|||
|
|
@ -2,7 +2,6 @@
|
|||
|
||||
namespace Nif
|
||||
{
|
||||
|
||||
void NiExtraData::read(NIFStream* nif)
|
||||
{
|
||||
Extra::read(nif);
|
||||
|
|
@ -17,18 +16,17 @@ namespace Nif
|
|||
nif->getSizedStrings(mData, nif->get<uint32_t>());
|
||||
}
|
||||
|
||||
void NiTextKeyExtraData::TextKey::read(NIFStream* nif)
|
||||
{
|
||||
nif->read(mTime);
|
||||
nif->read(mText);
|
||||
}
|
||||
|
||||
void NiTextKeyExtraData::read(NIFStream* nif)
|
||||
{
|
||||
Extra::read(nif);
|
||||
|
||||
uint32_t numKeys;
|
||||
nif->read(numKeys);
|
||||
mList.resize(numKeys);
|
||||
for (TextKey& key : mList)
|
||||
{
|
||||
nif->read(key.mTime);
|
||||
nif->read(key.mText);
|
||||
}
|
||||
nif->readVectorOfRecords<uint32_t>(mList);
|
||||
}
|
||||
|
||||
void NiVertWeightsExtraData::read(NIFStream* nif)
|
||||
|
|
@ -66,20 +64,11 @@ namespace Nif
|
|||
{
|
||||
Extra::read(nif);
|
||||
|
||||
uint32_t num;
|
||||
nif->read(num);
|
||||
const uint32_t num = nif->get<uint32_t>();
|
||||
if (nif->getBethVersion() <= NIFFile::BethVersion::BETHVER_FO3)
|
||||
{
|
||||
mLegacyMarkers.resize(num);
|
||||
for (auto& marker : mLegacyMarkers)
|
||||
marker.read(nif);
|
||||
}
|
||||
nif->readVectorOfRecords(num, mLegacyMarkers);
|
||||
else
|
||||
{
|
||||
mMarkers.resize(num);
|
||||
for (auto& marker : mMarkers)
|
||||
marker.read(nif);
|
||||
}
|
||||
nif->readVectorOfRecords(num, mMarkers);
|
||||
}
|
||||
|
||||
void BSInvMarker::read(NIFStream* nif)
|
||||
|
|
|
|||
|
|
@ -77,6 +77,8 @@ namespace Nif
|
|||
{
|
||||
float mTime;
|
||||
std::string mText;
|
||||
|
||||
void read(NIFStream* nif);
|
||||
};
|
||||
std::vector<TextKey> mList;
|
||||
|
||||
|
|
|
|||
|
|
@ -600,7 +600,9 @@ namespace Nif
|
|||
if (hasUserVersion)
|
||||
nif.read(mUserVersion);
|
||||
|
||||
mRecords.resize(nif.get<std::uint32_t>());
|
||||
const std::uint32_t recordsCount = nif.get<std::uint32_t>();
|
||||
|
||||
mRecords.reserve(recordsCount);
|
||||
|
||||
// Bethesda stream header
|
||||
{
|
||||
|
|
@ -646,14 +648,14 @@ namespace Nif
|
|||
else
|
||||
{
|
||||
nif.getSizedStrings(recTypes, nif.get<std::uint16_t>());
|
||||
nif.readVector(recTypeIndices, mRecords.size());
|
||||
nif.readVector(recTypeIndices, recordsCount);
|
||||
}
|
||||
}
|
||||
|
||||
if (hasRecordSizes) // Record sizes
|
||||
{
|
||||
std::vector<std::uint32_t> recSizes; // Currently unused
|
||||
nif.readVector(recSizes, mRecords.size());
|
||||
nif.readVector(recSizes, recordsCount);
|
||||
}
|
||||
|
||||
if (hasStringTable)
|
||||
|
|
@ -670,7 +672,7 @@ namespace Nif
|
|||
nif.readVector(groups, nif.get<std::uint32_t>());
|
||||
}
|
||||
|
||||
for (std::size_t i = 0; i < mRecords.size(); i++)
|
||||
for (std::size_t i = 0; i < recordsCount; i++)
|
||||
{
|
||||
std::unique_ptr<Record> r;
|
||||
|
||||
|
|
@ -701,22 +703,23 @@ namespace Nif
|
|||
r->recName = std::move(rec);
|
||||
r->recIndex = static_cast<unsigned>(i);
|
||||
r->read(&nif);
|
||||
mRecords[i] = std::move(r);
|
||||
mRecords.push_back(std::move(r));
|
||||
}
|
||||
|
||||
// Determine which records are roots
|
||||
mRoots.resize(nif.get<uint32_t>());
|
||||
for (std::size_t i = 0; i < mRoots.size(); i++)
|
||||
const std::uint32_t rootsCount = nif.get<uint32_t>();
|
||||
mRoots.reserve(rootsCount);
|
||||
for (std::size_t i = 0; i < rootsCount; i++)
|
||||
{
|
||||
std::int32_t idx;
|
||||
nif.read(idx);
|
||||
if (idx >= 0 && static_cast<std::size_t>(idx) < mRecords.size())
|
||||
{
|
||||
mRoots[i] = mRecords[idx].get();
|
||||
mRoots.push_back(mRecords[idx].get());
|
||||
}
|
||||
else
|
||||
{
|
||||
mRoots[i] = nullptr;
|
||||
mRoots.push_back(nullptr);
|
||||
Log(Debug::Warning) << "NIFFile Warning: Root " << i + 1 << " does not point to a record: index " << idx
|
||||
<< ". File: " << mFilename;
|
||||
}
|
||||
|
|
|
|||
|
|
@ -76,8 +76,7 @@ namespace Nif
|
|||
}
|
||||
}
|
||||
|
||||
uint32_t count;
|
||||
nif->read(count);
|
||||
const uint32_t count = nif->get<uint32_t>();
|
||||
|
||||
if (count == 0 && !morph)
|
||||
return;
|
||||
|
|
@ -86,42 +85,21 @@ namespace Nif
|
|||
|
||||
mKeys.reserve(count);
|
||||
|
||||
KeyType key = {};
|
||||
|
||||
if (mInterpolationType == InterpolationType_Linear || mInterpolationType == InterpolationType_Constant)
|
||||
{
|
||||
for (size_t i = 0; i < count; i++)
|
||||
{
|
||||
float time;
|
||||
nif->read(time);
|
||||
readValue(*nif, key);
|
||||
mKeys.emplace_back(time, key);
|
||||
}
|
||||
nif->readVectorOfRecords(count, readValuePair, mKeys);
|
||||
}
|
||||
else if (mInterpolationType == InterpolationType_Quadratic)
|
||||
{
|
||||
for (size_t i = 0; i < count; i++)
|
||||
{
|
||||
float time;
|
||||
nif->read(time);
|
||||
readQuadratic(*nif, key);
|
||||
mKeys.emplace_back(time, key);
|
||||
}
|
||||
nif->readVectorOfRecords(count, readQuadraticPair, mKeys);
|
||||
}
|
||||
else if (mInterpolationType == InterpolationType_TCB)
|
||||
{
|
||||
std::vector<TCBKey<T>> tcbKeys(count);
|
||||
for (TCBKey<T>& tcbKey : tcbKeys)
|
||||
{
|
||||
nif->read(tcbKey.mTime);
|
||||
tcbKey.mValue = ((*nif).*getValue)();
|
||||
nif->read(tcbKey.mTension);
|
||||
nif->read(tcbKey.mContinuity);
|
||||
nif->read(tcbKey.mBias);
|
||||
}
|
||||
std::vector<TCBKey<T>> tcbKeys;
|
||||
nif->readVectorOfRecords(count, readTCBKey, tcbKeys);
|
||||
generateTCBTangents(tcbKeys);
|
||||
for (TCBKey<T>& tcbKey : tcbKeys)
|
||||
mKeys.emplace_back(std::move(tcbKey.mTime),
|
||||
mKeys.emplace_back(tcbKey.mTime,
|
||||
KeyType{ std::move(tcbKey.mValue), std::move(tcbKey.mInTan), std::move(tcbKey.mOutTan) });
|
||||
}
|
||||
else if (mInterpolationType == InterpolationType_XYZ)
|
||||
|
|
@ -143,17 +121,42 @@ namespace Nif
|
|||
}
|
||||
|
||||
private:
|
||||
static void readValue(NIFStream& nif, KeyT<T>& key) { key.mValue = (nif.*getValue)(); }
|
||||
static void readValue(NIFStream& nif, KeyType& key) { key.mValue = (nif.*getValue)(); }
|
||||
|
||||
template <typename U>
|
||||
static void readQuadratic(NIFStream& nif, KeyT<U>& key)
|
||||
static void readValuePair(NIFStream& nif, std::pair<float, KeyType>& value)
|
||||
{
|
||||
readValue(nif, key);
|
||||
key.mInTan = (nif.*getValue)();
|
||||
key.mOutTan = (nif.*getValue)();
|
||||
nif.read(value.first);
|
||||
readValue(nif, value.second);
|
||||
}
|
||||
|
||||
static void readQuadratic(NIFStream& nif, KeyT<osg::Quat>& key) { readValue(nif, key); }
|
||||
static void readQuadratic(NIFStream& nif, KeyType& key)
|
||||
{
|
||||
if constexpr (std::is_same_v<T, osg::Quat>)
|
||||
{
|
||||
readValue(nif, key);
|
||||
}
|
||||
else
|
||||
{
|
||||
readValue(nif, key);
|
||||
key.mInTan = (nif.*getValue)();
|
||||
key.mOutTan = (nif.*getValue)();
|
||||
}
|
||||
}
|
||||
|
||||
static void readQuadraticPair(NIFStream& nif, std::pair<float, KeyType>& value)
|
||||
{
|
||||
nif.read(value.first);
|
||||
readQuadratic(nif, value.second);
|
||||
}
|
||||
|
||||
static void readTCBKey(NIFStream& nif, TCBKey<T>& value)
|
||||
{
|
||||
nif.read(value.mTime);
|
||||
value.mValue = (nif.*getValue)();
|
||||
nif.read(value.mTension);
|
||||
nif.read(value.mContinuity);
|
||||
nif.read(value.mBias);
|
||||
}
|
||||
|
||||
template <typename U>
|
||||
static void generateTCBTangents(std::vector<TCBKey<U>>& keys)
|
||||
|
|
@ -192,6 +195,16 @@ namespace Nif
|
|||
}
|
||||
};
|
||||
|
||||
template <class Key, class Value>
|
||||
void readKeyMapPair(NIFStream& stream, std::pair<Key, KeyT<Value>>& value);
|
||||
|
||||
template <>
|
||||
inline void readKeyMapPair(NIFStream& stream, std::pair<float, KeyT<bool>>& value)
|
||||
{
|
||||
stream.read(value.first);
|
||||
value.second.mValue = stream.get<uint8_t>() != 0;
|
||||
}
|
||||
|
||||
using FloatKeyMap = KeyMapT<float, &NIFStream::get<float>>;
|
||||
using Vector3KeyMap = KeyMapT<osg::Vec3f, &NIFStream::get<osg::Vec3f>>;
|
||||
using Vector4KeyMap = KeyMapT<osg::Vec4f, &NIFStream::get<osg::Vec4f>>;
|
||||
|
|
|
|||
|
|
@ -73,9 +73,10 @@ namespace Nif
|
|||
|
||||
void NIFStream::getSizedStrings(std::vector<std::string>& vec, size_t size)
|
||||
{
|
||||
vec.resize(size);
|
||||
for (size_t i = 0; i < vec.size(); i++)
|
||||
vec[i] = getSizedString();
|
||||
vec.clear();
|
||||
vec.reserve(size);
|
||||
for (size_t i = 0; i < size; i++)
|
||||
vec.push_back(getSizedString());
|
||||
}
|
||||
|
||||
std::string NIFStream::getVersionString()
|
||||
|
|
|
|||
|
|
@ -73,6 +73,11 @@ namespace Nif
|
|||
Misc::swapEndiannessInplace(dest[i]);
|
||||
}
|
||||
|
||||
class NIFStream;
|
||||
|
||||
template <class T>
|
||||
void readRecord(NIFStream& stream, T& value);
|
||||
|
||||
class NIFStream
|
||||
{
|
||||
const Reader& mReader;
|
||||
|
|
@ -166,10 +171,47 @@ namespace Nif
|
|||
/// Read a sequence of null-terminated strings
|
||||
std::string getStringPalette();
|
||||
|
||||
template <class Count, class T, class Read>
|
||||
void readVectorOfRecords(Count count, Read&& read, std::vector<T>& values)
|
||||
{
|
||||
values.clear();
|
||||
values.reserve(count);
|
||||
for (Count i = 0; i < count; ++i)
|
||||
{
|
||||
T value;
|
||||
read(*this, value);
|
||||
values.push_back(std::move(value));
|
||||
}
|
||||
}
|
||||
|
||||
template <class Count, class T, class Read>
|
||||
void readVectorOfRecords(Read&& read, std::vector<T>& values)
|
||||
{
|
||||
readVectorOfRecords(get<Count>(), std::forward<Read>(read), values);
|
||||
}
|
||||
|
||||
template <class Count, class T>
|
||||
void readVectorOfRecords(Count count, std::vector<T>& values)
|
||||
{
|
||||
readVectorOfRecords(count, readRecord<T>, values);
|
||||
}
|
||||
|
||||
template <class Count, class T>
|
||||
void readVectorOfRecords(std::vector<T>& values)
|
||||
{
|
||||
readVectorOfRecords<Count>(readRecord<T>, values);
|
||||
}
|
||||
|
||||
private:
|
||||
void checkStreamSize(std::size_t size);
|
||||
};
|
||||
|
||||
template <class T>
|
||||
void readRecord(NIFStream& stream, T& value)
|
||||
{
|
||||
value.read(&stream);
|
||||
}
|
||||
|
||||
template <>
|
||||
void NIFStream::read<osg::Vec2f>(osg::Vec2f& vec);
|
||||
template <>
|
||||
|
|
|
|||
|
|
@ -105,9 +105,7 @@ namespace Nif
|
|||
}
|
||||
case UNION_BV:
|
||||
{
|
||||
mChildren.resize(nif->get<uint32_t>());
|
||||
for (BoundingVolume& child : mChildren)
|
||||
child.read(nif);
|
||||
nif->readVectorOfRecords<uint32_t>(mChildren);
|
||||
break;
|
||||
}
|
||||
case HALFSPACE_BV:
|
||||
|
|
@ -426,6 +424,12 @@ namespace Nif
|
|||
nif->read(mInitialIndex);
|
||||
}
|
||||
|
||||
void NiLODNode::LODRange::read(NIFStream* nif)
|
||||
{
|
||||
nif->read(mMinRange);
|
||||
nif->read(mMaxRange);
|
||||
}
|
||||
|
||||
void NiLODNode::read(NIFStream* nif)
|
||||
{
|
||||
NiSwitchNode::read(nif);
|
||||
|
|
@ -439,12 +443,7 @@ namespace Nif
|
|||
if (nif->getVersion() >= NIFFile::NIFVersion::VER_MW)
|
||||
nif->read(mLODCenter);
|
||||
|
||||
mLODLevels.resize(nif->get<uint32_t>());
|
||||
for (LODRange& level : mLODLevels)
|
||||
{
|
||||
nif->read(level.mMinRange);
|
||||
nif->read(level.mMaxRange);
|
||||
}
|
||||
nif->readVectorOfRecords<uint32_t>(mLODLevels);
|
||||
}
|
||||
|
||||
void NiFltAnimationNode::read(NIFStream* nif)
|
||||
|
|
|
|||
|
|
@ -232,6 +232,8 @@ namespace Nif
|
|||
{
|
||||
float mMinRange;
|
||||
float mMaxRange;
|
||||
|
||||
void read(NIFStream* nif);
|
||||
};
|
||||
|
||||
osg::Vec3f mLODCenter;
|
||||
|
|
|
|||
|
|
@ -225,11 +225,7 @@ namespace Nif
|
|||
|
||||
bool hasData = nif->getBethVersion() < NIFFile::BethVersion::BETHVER_FO3;
|
||||
if (hasData)
|
||||
{
|
||||
mParticles.resize(mNumVertices);
|
||||
for (NiParticleInfo& info : mParticles)
|
||||
info.read(nif);
|
||||
}
|
||||
nif->readVectorOfRecords(mNumVertices, mParticles);
|
||||
|
||||
if (nif->getBethVersion() > NIFFile::BethVersion::BETHVER_FO4)
|
||||
nif->skip(12); // Unknown
|
||||
|
|
@ -680,12 +676,7 @@ namespace Nif
|
|||
mFloatKeyList = std::make_shared<FloatKeyMap>();
|
||||
mFloatKeyList->read(nif);
|
||||
mVisKeyList = std::make_shared<BoolKeyMap>();
|
||||
mVisKeyList->mKeys.resize(nif->get<uint32_t>());
|
||||
for (auto& [time, key] : mVisKeyList->mKeys)
|
||||
{
|
||||
nif->read(time);
|
||||
key.mValue = nif->get<uint8_t>() != 0;
|
||||
}
|
||||
nif->readVectorOfRecords<uint32_t>(readKeyMapPair<float, bool>, mVisKeyList->mKeys);
|
||||
}
|
||||
|
||||
void NiPSysCollider::read(NIFStream* nif)
|
||||
|
|
|
|||
|
|
@ -555,11 +555,7 @@ namespace Nif
|
|||
if (nif->getVersion() >= NIFStream::generateVersion(10, 1, 0, 0))
|
||||
nif->read(mScale);
|
||||
readRecordList(nif, mData);
|
||||
uint32_t numFilters;
|
||||
nif->read(numFilters);
|
||||
mHavokFilters.resize(numFilters);
|
||||
for (HavokFilter& filter : mHavokFilters)
|
||||
filter.read(nif);
|
||||
nif->readVectorOfRecords<uint32_t>(mHavokFilters);
|
||||
}
|
||||
|
||||
void bhkNiTriStripsShape::post(Reader& nif)
|
||||
|
|
|
|||
|
|
@ -73,10 +73,12 @@ namespace Nif
|
|||
if (nif->getVersion() <= NIFStream::generateVersion(20, 1, 0, 1))
|
||||
mApplyMode = static_cast<ApplyMode>(nif->get<uint32_t>());
|
||||
|
||||
mTextures.resize(nif->get<uint32_t>());
|
||||
for (size_t i = 0; i < mTextures.size(); i++)
|
||||
const uint32_t texturesSize = nif->get<uint32_t>();
|
||||
|
||||
mTextures.reserve(texturesSize);
|
||||
for (size_t i = 0; i < texturesSize; i++)
|
||||
{
|
||||
mTextures[i].read(nif);
|
||||
mTextures.emplace_back().read(nif);
|
||||
|
||||
if (i == 5 && mTextures[5].mEnabled)
|
||||
{
|
||||
|
|
@ -89,13 +91,16 @@ namespace Nif
|
|||
|
||||
if (nif->getVersion() >= NIFStream::generateVersion(10, 0, 1, 0))
|
||||
{
|
||||
mShaderTextures.resize(nif->get<uint32_t>());
|
||||
mShaderIds.resize(mShaderTextures.size());
|
||||
for (size_t i = 0; i < mShaderTextures.size(); i++)
|
||||
const uint32_t sharedTexturesSize = nif->get<uint32_t>();
|
||||
mShaderTextures.reserve(sharedTexturesSize);
|
||||
mShaderIds.reserve(sharedTexturesSize);
|
||||
for (size_t i = 0; i < sharedTexturesSize; i++)
|
||||
{
|
||||
mShaderTextures[i].read(nif);
|
||||
mShaderTextures.emplace_back().read(nif);
|
||||
uint32_t id = 0;
|
||||
if (mShaderTextures[i].mEnabled)
|
||||
nif->read(mShaderIds[i]);
|
||||
nif->read(id);
|
||||
mShaderIds.push_back(id);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
@ -346,9 +351,10 @@ namespace Nif
|
|||
mTranslucency.read(nif);
|
||||
if (nif->get<uint8_t>() != 0)
|
||||
{
|
||||
mTextureArrays.resize(nif->get<uint32_t>());
|
||||
for (std::vector<std::string>& textureArray : mTextureArrays)
|
||||
nif->getSizedStrings(textureArray, nif->get<uint32_t>());
|
||||
const uint32_t size = nif->get<uint32_t>();
|
||||
mTextureArrays.reserve(size);
|
||||
for (uint32_t i = 0; i < size; ++i)
|
||||
nif->getSizedStrings(mTextureArrays.emplace_back(), nif->get<uint32_t>());
|
||||
}
|
||||
}
|
||||
if (nif->getBethVersion() >= NIFFile::BethVersion::BETHVER_STF)
|
||||
|
|
|
|||
|
|
@ -160,16 +160,7 @@ namespace Nif
|
|||
template <class T>
|
||||
void readRecordList(NIFStream* nif, RecordListT<T>& list)
|
||||
{
|
||||
const std::uint32_t length = nif->get<std::uint32_t>();
|
||||
|
||||
// No reasonable list can hit this generous limit
|
||||
if (length >= (1 << 24))
|
||||
throw std::runtime_error("Record list too long: " + std::to_string(length));
|
||||
|
||||
list.resize(length);
|
||||
|
||||
for (auto& value : list)
|
||||
value.read(nif);
|
||||
nif->readVectorOfRecords<uint32_t>(list);
|
||||
}
|
||||
|
||||
template <class T>
|
||||
|
|
|
|||
Loading…
Reference in a new issue