From 7115f60b91b5717d90f643fd692010905c7004db Mon Sep 17 00:00:00 2001 From: Bryan Galdrikian Date: Thu, 31 May 2018 11:36:08 -0700 Subject: Blast 1.1.3. See docs/release_notes.txt. --- .../source/NvBlastExtSerialization.cpp | 802 ++++++++++----------- 1 file changed, 401 insertions(+), 401 deletions(-) mode change 100644 => 100755 sdk/extensions/serialization/source/NvBlastExtSerialization.cpp (limited to 'sdk/extensions/serialization/source/NvBlastExtSerialization.cpp') diff --git a/sdk/extensions/serialization/source/NvBlastExtSerialization.cpp b/sdk/extensions/serialization/source/NvBlastExtSerialization.cpp old mode 100644 new mode 100755 index cae8f46..6d86358 --- a/sdk/extensions/serialization/source/NvBlastExtSerialization.cpp +++ b/sdk/extensions/serialization/source/NvBlastExtSerialization.cpp @@ -1,401 +1,401 @@ -// This code contains NVIDIA Confidential Information and is disclosed to you -// under a form of NVIDIA software license agreement provided separately to you. -// -// Notice -// NVIDIA Corporation and its licensors retain all intellectual property and -// proprietary rights in and to this software and related documentation and -// any modifications thereto. Any use, reproduction, disclosure, or -// distribution of this software and related documentation without an express -// license agreement from NVIDIA Corporation is strictly prohibited. -// -// ALL NVIDIA DESIGN SPECIFICATIONS, CODE ARE PROVIDED "AS IS.". NVIDIA MAKES -// NO WARRANTIES, EXPRESSED, IMPLIED, STATUTORY, OR OTHERWISE WITH RESPECT TO -// THE MATERIALS, AND EXPRESSLY DISCLAIMS ALL IMPLIED WARRANTIES OF NONINFRINGEMENT, -// MERCHANTABILITY, AND FITNESS FOR A PARTICULAR PURPOSE. -// -// Information and code furnished is believed to be accurate and reliable. -// However, NVIDIA Corporation assumes no responsibility for the consequences of use of such -// information or for any infringement of patents or other rights of third parties that may -// result from its use. No license is granted by implication or otherwise under any patent -// or patent rights of NVIDIA Corporation. Details are subject to change without notice. -// This code supersedes and replaces all information previously supplied. -// NVIDIA Corporation products are not authorized for use as critical -// components in life support devices or systems without express written approval of -// NVIDIA Corporation. -// -// Copyright (c) 2018 NVIDIA Corporation. All rights reserved. - - -#include "NvBlastExtSerialization.h" -#include "NvBlastExtLlSerialization.h" -#include "NvBlastHashMap.h" -#include "NvBlastExtSerializationInternal.h" - - -namespace Nv -{ -namespace Blast -{ - -class ExtSerializationImpl : public ExtSerializationInternal -{ -public: - // Default buffer provider - class AllocBufferProvider : public ExtSerialization::BufferProvider - { - public: - virtual void* requestBuffer(size_t size) override; - }; - - - ExtSerializationImpl(); - ~ExtSerializationImpl(); - - // ExtSerialization interface begin - virtual bool setSerializationEncoding(uint32_t encodingID) override; - virtual uint32_t getSerializationEncoding() const override; - - virtual void setBufferProvider(BufferProvider* bufferProvider) override; - - virtual bool peekHeader(uint32_t* objectTypeID, uint32_t* encodingID, uint64_t* dataSize, const void* buffer, uint64_t bufferSize) override; - virtual const void* skipObject(uint64_t& bufferSize, const void* buffer) override; - - virtual void* deserializeFromBuffer(const void* buffer, uint64_t size, uint32_t* objectTypeIDPtr = nullptr) override; - virtual uint64_t serializeIntoBuffer(void*& buffer, const void* object, uint32_t objectTypeID) override; - - virtual void release() override; - // ExtSerialization interface end - - // ExtSerializationInternal interface begin - virtual bool registerSerializer(ExtSerializer& serializer) override; - virtual bool unregisterSerializer(ExtSerializer& serializer) override; - - virtual ExtSerializer* findSerializer(uint32_t objectTypeID, uint32_t encodingID) override; - // ExtSerializationInternal interface end - -private: - char* writeHeaderIntoBuffer(char* buffer, uint64_t bufferSize, uint32_t objectTypeID, uint32_t encodingID, uint64_t dataSize) const; - const char* readHeaderFromBuffer(uint32_t* objectTypeID, uint32_t* encodingID, uint64_t* dataSize, const char* buffer, uint64_t bufferSize) const; - - //// Static data //// - static const char* s_identifier; - static const char* s_version; - static AllocBufferProvider s_defaultBufferProvider; - - //// Member data //// - HashMap::type m_serializers; - uint32_t m_serializationEncoding; - BufferProvider* m_bufferProvider; -}; - - -//////// ExtSerializationImpl static member variables //////// - -/** Module identifying header. This should never change. */ -const char* ExtSerializationImpl::s_identifier = "NVidia(r) GameWorks Blast(tm) v."; - -const char* ExtSerializationImpl::s_version = "1"; - -ExtSerializationImpl::AllocBufferProvider ExtSerializationImpl::s_defaultBufferProvider; - - -//////// Local utility functions //////// - -static NV_INLINE uint64_t generateKey(uint32_t objectTypeID, uint32_t encodingID) -{ - return static_cast(encodingID) << 32 | static_cast(objectTypeID); -} - - -static NV_INLINE uint64_t generateKey(const ExtSerializer& serializer) -{ - return generateKey(serializer.getObjectTypeID(), serializer.getEncodingID()); -} - - -static NV_INLINE void writeIDToBuffer(char* buffer, uint32_t id) -{ - for (int i = 0; i < 4; ++i, id >>= 8) - { - *buffer++ = static_cast(id & 0xFF); - } -} - - -static NV_INLINE uint32_t readIDFromBuffer(const char* buffer) -{ - return NVBLAST_FOURCC(buffer[0], buffer[1], buffer[2], buffer[3]); -} - - -static NV_INLINE void writeU64InHexToBuffer(char* buffer, uint64_t val) -{ - for (char* curr = buffer + 16; curr-- > buffer; val >>= 4) - { - *curr = "0123456789ABCDEF"[val & 0xF]; - } -} - - -static NV_INLINE uint64_t readU64InHexFromBuffer(const char* buffer) -{ - uint64_t val = 0; - for (const char* curr = buffer; curr < buffer + 16; ++curr) - { - const char c = *curr; - const char msn = c >> 4; - const char mask = ((88 >> msn) & 1) - 1; - const unsigned char digit = "\x0\x1\x2\x3\x4\x5\x6\x7\x8\x9\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xA\xB\xC\xD\xE\xF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF"[((msn - 3) & 1) << 4 | (c & 0xF)] | mask; - if (digit == 0xFF) - { - return 0; // Not a hexidecimal digit - } - val = val << 4 | digit; - } - return val; -} - - -//////// ExtSerialization member functions //////// - -ExtSerializationImpl::ExtSerializationImpl() : m_serializationEncoding(EncodingID::CapnProtoBinary), m_bufferProvider(&s_defaultBufferProvider) -{ -} - - -ExtSerializationImpl::~ExtSerializationImpl() -{ - // Release and remove all registered serializers - auto it = m_serializers.getEraseIterator(); - while (auto entry = it.eraseCurrentGetNext(true)) - { - entry->second->release(); - } -} - - -char* ExtSerializationImpl::writeHeaderIntoBuffer(char* buffer, uint64_t bufferSize, uint32_t objectTypeID, uint32_t encodingID, uint64_t dataSize) const -{ - if (bufferSize < HeaderSize) - { - return nullptr; - } - - char* stop = buffer + HeaderSize; - - size_t versionLen = strlen(s_version); - if (versionLen > 63) - { - versionLen = 63; - } - - memset(buffer, ' ', HeaderSize); - memcpy(buffer, s_identifier, 32); buffer += 32; - memcpy(buffer, s_version, versionLen); buffer += 64; - writeIDToBuffer(buffer, objectTypeID); buffer += 5; - writeIDToBuffer(buffer, encodingID); buffer += 5; - writeU64InHexToBuffer(buffer, dataSize); buffer += 16; - *(stop - 1) = '\n'; - - return stop; -} - - -const char* ExtSerializationImpl::readHeaderFromBuffer(uint32_t* objectTypeID, uint32_t* encodingID, uint64_t* dataSize, const char* buffer, uint64_t bufferSize) const -{ - if (bufferSize < HeaderSize) - { - NVBLAST_LOG_ERROR("ExtSerializationImpl::readHeaderFromBuffer: header terminator not found."); - return nullptr; - } - - const char* stop = buffer + HeaderSize; - - if (memcmp(buffer, s_identifier, 32)) - { - NVBLAST_LOG_ERROR("ExtSerializationImpl::readHeaderFromBuffer: file identifier does not match expected value."); - return nullptr; - } - buffer += 32; - - const char* s = strchr(buffer, ' '); - if (s == nullptr) - { - NVBLAST_LOG_ERROR("ExtSerializationImpl::readHeaderFromBuffer: file format error reading serializer library version."); - } - if (memcmp(buffer, s_version, s - buffer)) - { - NVBLAST_LOG_ERROR("ExtSerializationImpl::readHeaderFromBuffer: file version does not match serializer library version."); - return nullptr; - } - buffer += 64; - - if (objectTypeID != nullptr) - { - *objectTypeID = readIDFromBuffer(buffer); - } - buffer += 5; - - if (encodingID != nullptr) - { - *encodingID = readIDFromBuffer(buffer); - } - buffer += 5; - - if (dataSize != nullptr) - { - *dataSize = readU64InHexFromBuffer(buffer); - } - buffer += 16; - - return stop; -} - - -bool ExtSerializationImpl::registerSerializer(ExtSerializer& serializer) -{ - return m_serializers.insert(generateKey(serializer), &serializer); -} - - -bool ExtSerializationImpl::unregisterSerializer(ExtSerializer& serializer) -{ - const uint64_t key = generateKey(serializer); - const auto entry = m_serializers.find(key); - if (entry == nullptr) - { - return false; - } - entry->second->release(); - return m_serializers.erase(key); -} - - -ExtSerializer* ExtSerializationImpl::findSerializer(uint32_t objectTypeID, uint32_t encodingID) -{ - auto entry = m_serializers.find(generateKey(objectTypeID, encodingID)); - return entry != nullptr ? entry->second : nullptr; -} - - -bool ExtSerializationImpl::setSerializationEncoding(uint32_t encodingID) -{ - m_serializationEncoding = encodingID; - - return true; -} - - -uint32_t ExtSerializationImpl::getSerializationEncoding() const -{ - return m_serializationEncoding; -} - - -void ExtSerializationImpl::setBufferProvider(BufferProvider* bufferProvider) -{ - m_bufferProvider = bufferProvider != nullptr ? bufferProvider : &s_defaultBufferProvider; -} - - -bool ExtSerializationImpl::peekHeader(uint32_t* objectTypeID, uint32_t* encodingID, uint64_t* dataSize, const void* buffer, uint64_t bufferSize) -{ - return nullptr != readHeaderFromBuffer(objectTypeID, encodingID, dataSize, reinterpret_cast(buffer), bufferSize); -} - - -const void* ExtSerializationImpl::skipObject(uint64_t& bufferSize, const void* buffer) -{ - uint64_t dataSize; - const char* next = readHeaderFromBuffer(nullptr, nullptr, &dataSize, static_cast(buffer), bufferSize); - if (next == nullptr) - { - return nullptr; - } - next += dataSize; - const uint64_t skipSize = next - static_cast(buffer); - NVBLAST_CHECK_ERROR(skipSize <= bufferSize, "Object size in buffer is too large for given buffer size.", return nullptr); - bufferSize -= skipSize; - return next; -} - - -void* ExtSerializationImpl::deserializeFromBuffer(const void* buffer, uint64_t bufferSize, uint32_t* objectTypeIDPtr) -{ - uint32_t objectTypeID; - uint32_t encodingID; - uint64_t dataSize; - void* result = nullptr; - - buffer = readHeaderFromBuffer(&objectTypeID, &encodingID, &dataSize, reinterpret_cast(buffer), bufferSize); - if (buffer != nullptr) - { - auto entry = m_serializers.find(generateKey(objectTypeID, encodingID)); - if (entry != nullptr && entry->second != nullptr) - { - result = entry->second->deserializeFromBuffer(buffer, dataSize); - } - } - - if (objectTypeIDPtr != nullptr) - { - *objectTypeIDPtr = result != nullptr ? objectTypeID : 0; - } - - return result; -} - - -uint64_t ExtSerializationImpl::serializeIntoBuffer(void*& buffer, const void* object, uint32_t objectTypeID) -{ - if (!m_serializationEncoding) - { - NVBLAST_LOG_ERROR("ExtSerializationImpl::serializeIntoBuffer: no serialization encoding has been set."); - return false; // No encoding available - } - - auto entry = m_serializers.find(generateKey(objectTypeID, m_serializationEncoding)); - if (entry == nullptr || entry->second == nullptr) - { - return false; - } - - const uint64_t size = entry->second->serializeIntoBuffer(buffer, *m_bufferProvider, object, HeaderSize); - if (size < HeaderSize) - { - NVBLAST_LOG_ERROR("ExtSerializationImpl::serializeIntoBuffer: failed to write data to buffer."); - return 0; - } - - writeHeaderIntoBuffer(reinterpret_cast(buffer), HeaderSize, objectTypeID, m_serializationEncoding, size - HeaderSize); - - return size; -} - - -void ExtSerializationImpl::release() -{ - NVBLAST_DELETE(this, ExtSerializationImpl); -} - - -//////// ExtSerializationImpl::AllocBufferProvider member functions //////// - -void* ExtSerializationImpl::AllocBufferProvider::requestBuffer(size_t size) -{ - return NVBLAST_ALLOC(size); -} - -} // namespace Blast -} // namespace Nv - - -Nv::Blast::ExtSerialization* NvBlastExtSerializationCreate() -{ - Nv::Blast::ExtSerializationImpl* serialization = NVBLAST_NEW(Nv::Blast::ExtSerializationImpl) (); - - // Automatically load LL serializers - NvBlastExtLlSerializerLoadSet(*serialization); - - return serialization; -} +// This code contains NVIDIA Confidential Information and is disclosed to you +// under a form of NVIDIA software license agreement provided separately to you. +// +// Notice +// NVIDIA Corporation and its licensors retain all intellectual property and +// proprietary rights in and to this software and related documentation and +// any modifications thereto. Any use, reproduction, disclosure, or +// distribution of this software and related documentation without an express +// license agreement from NVIDIA Corporation is strictly prohibited. +// +// ALL NVIDIA DESIGN SPECIFICATIONS, CODE ARE PROVIDED "AS IS.". NVIDIA MAKES +// NO WARRANTIES, EXPRESSED, IMPLIED, STATUTORY, OR OTHERWISE WITH RESPECT TO +// THE MATERIALS, AND EXPRESSLY DISCLAIMS ALL IMPLIED WARRANTIES OF NONINFRINGEMENT, +// MERCHANTABILITY, AND FITNESS FOR A PARTICULAR PURPOSE. +// +// Information and code furnished is believed to be accurate and reliable. +// However, NVIDIA Corporation assumes no responsibility for the consequences of use of such +// information or for any infringement of patents or other rights of third parties that may +// result from its use. No license is granted by implication or otherwise under any patent +// or patent rights of NVIDIA Corporation. Details are subject to change without notice. +// This code supersedes and replaces all information previously supplied. +// NVIDIA Corporation products are not authorized for use as critical +// components in life support devices or systems without express written approval of +// NVIDIA Corporation. +// +// Copyright (c) 2018 NVIDIA Corporation. All rights reserved. + + +#include "NvBlastExtSerialization.h" +#include "NvBlastExtLlSerialization.h" +#include "NvBlastHashMap.h" +#include "NvBlastExtSerializationInternal.h" + + +namespace Nv +{ +namespace Blast +{ + +class ExtSerializationImpl : public ExtSerializationInternal +{ +public: + // Default buffer provider + class AllocBufferProvider : public ExtSerialization::BufferProvider + { + public: + virtual void* requestBuffer(size_t size) override; + }; + + + ExtSerializationImpl(); + ~ExtSerializationImpl(); + + // ExtSerialization interface begin + virtual bool setSerializationEncoding(uint32_t encodingID) override; + virtual uint32_t getSerializationEncoding() const override; + + virtual void setBufferProvider(BufferProvider* bufferProvider) override; + + virtual bool peekHeader(uint32_t* objectTypeID, uint32_t* encodingID, uint64_t* dataSize, const void* buffer, uint64_t bufferSize) override; + virtual const void* skipObject(uint64_t& bufferSize, const void* buffer) override; + + virtual void* deserializeFromBuffer(const void* buffer, uint64_t size, uint32_t* objectTypeIDPtr = nullptr) override; + virtual uint64_t serializeIntoBuffer(void*& buffer, const void* object, uint32_t objectTypeID) override; + + virtual void release() override; + // ExtSerialization interface end + + // ExtSerializationInternal interface begin + virtual bool registerSerializer(ExtSerializer& serializer) override; + virtual bool unregisterSerializer(ExtSerializer& serializer) override; + + virtual ExtSerializer* findSerializer(uint32_t objectTypeID, uint32_t encodingID) override; + // ExtSerializationInternal interface end + +private: + char* writeHeaderIntoBuffer(char* buffer, uint64_t bufferSize, uint32_t objectTypeID, uint32_t encodingID, uint64_t dataSize) const; + const char* readHeaderFromBuffer(uint32_t* objectTypeID, uint32_t* encodingID, uint64_t* dataSize, const char* buffer, uint64_t bufferSize) const; + + //// Static data //// + static const char* s_identifier; + static const char* s_version; + static AllocBufferProvider s_defaultBufferProvider; + + //// Member data //// + HashMap::type m_serializers; + uint32_t m_serializationEncoding; + BufferProvider* m_bufferProvider; +}; + + +//////// ExtSerializationImpl static member variables //////// + +/** Module identifying header. This should never change. */ +const char* ExtSerializationImpl::s_identifier = "NVidia(r) GameWorks Blast(tm) v."; + +const char* ExtSerializationImpl::s_version = "1"; + +ExtSerializationImpl::AllocBufferProvider ExtSerializationImpl::s_defaultBufferProvider; + + +//////// Local utility functions //////// + +static NV_INLINE uint64_t generateKey(uint32_t objectTypeID, uint32_t encodingID) +{ + return static_cast(encodingID) << 32 | static_cast(objectTypeID); +} + + +static NV_INLINE uint64_t generateKey(const ExtSerializer& serializer) +{ + return generateKey(serializer.getObjectTypeID(), serializer.getEncodingID()); +} + + +static NV_INLINE void writeIDToBuffer(char* buffer, uint32_t id) +{ + for (int i = 0; i < 4; ++i, id >>= 8) + { + *buffer++ = static_cast(id & 0xFF); + } +} + + +static NV_INLINE uint32_t readIDFromBuffer(const char* buffer) +{ + return NVBLAST_FOURCC(buffer[0], buffer[1], buffer[2], buffer[3]); +} + + +static NV_INLINE void writeU64InHexToBuffer(char* buffer, uint64_t val) +{ + for (char* curr = buffer + 16; curr-- > buffer; val >>= 4) + { + *curr = "0123456789ABCDEF"[val & 0xF]; + } +} + + +static NV_INLINE uint64_t readU64InHexFromBuffer(const char* buffer) +{ + uint64_t val = 0; + for (const char* curr = buffer; curr < buffer + 16; ++curr) + { + const char c = *curr; + const char msn = c >> 4; + const char mask = ((88 >> msn) & 1) - 1; + const unsigned char digit = "\x0\x1\x2\x3\x4\x5\x6\x7\x8\x9\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xA\xB\xC\xD\xE\xF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF"[((msn - 3) & 1) << 4 | (c & 0xF)] | mask; + if (digit == 0xFF) + { + return 0; // Not a hexidecimal digit + } + val = val << 4 | digit; + } + return val; +} + + +//////// ExtSerialization member functions //////// + +ExtSerializationImpl::ExtSerializationImpl() : m_serializationEncoding(EncodingID::CapnProtoBinary), m_bufferProvider(&s_defaultBufferProvider) +{ +} + + +ExtSerializationImpl::~ExtSerializationImpl() +{ + // Release and remove all registered serializers + auto it = m_serializers.getEraseIterator(); + while (auto entry = it.eraseCurrentGetNext(true)) + { + entry->second->release(); + } +} + + +char* ExtSerializationImpl::writeHeaderIntoBuffer(char* buffer, uint64_t bufferSize, uint32_t objectTypeID, uint32_t encodingID, uint64_t dataSize) const +{ + if (bufferSize < HeaderSize) + { + return nullptr; + } + + char* stop = buffer + HeaderSize; + + size_t versionLen = strlen(s_version); + if (versionLen > 63) + { + versionLen = 63; + } + + memset(buffer, ' ', HeaderSize); + memcpy(buffer, s_identifier, 32); buffer += 32; + memcpy(buffer, s_version, versionLen); buffer += 64; + writeIDToBuffer(buffer, objectTypeID); buffer += 5; + writeIDToBuffer(buffer, encodingID); buffer += 5; + writeU64InHexToBuffer(buffer, dataSize); buffer += 16; + *(stop - 1) = '\n'; + + return stop; +} + + +const char* ExtSerializationImpl::readHeaderFromBuffer(uint32_t* objectTypeID, uint32_t* encodingID, uint64_t* dataSize, const char* buffer, uint64_t bufferSize) const +{ + if (bufferSize < HeaderSize) + { + NVBLAST_LOG_ERROR("ExtSerializationImpl::readHeaderFromBuffer: header terminator not found."); + return nullptr; + } + + const char* stop = buffer + HeaderSize; + + if (memcmp(buffer, s_identifier, 32)) + { + NVBLAST_LOG_ERROR("ExtSerializationImpl::readHeaderFromBuffer: file identifier does not match expected value."); + return nullptr; + } + buffer += 32; + + const char* s = strchr(buffer, ' '); + if (s == nullptr) + { + NVBLAST_LOG_ERROR("ExtSerializationImpl::readHeaderFromBuffer: file format error reading serializer library version."); + } + if (memcmp(buffer, s_version, s - buffer)) + { + NVBLAST_LOG_ERROR("ExtSerializationImpl::readHeaderFromBuffer: file version does not match serializer library version."); + return nullptr; + } + buffer += 64; + + if (objectTypeID != nullptr) + { + *objectTypeID = readIDFromBuffer(buffer); + } + buffer += 5; + + if (encodingID != nullptr) + { + *encodingID = readIDFromBuffer(buffer); + } + buffer += 5; + + if (dataSize != nullptr) + { + *dataSize = readU64InHexFromBuffer(buffer); + } + buffer += 16; + + return stop; +} + + +bool ExtSerializationImpl::registerSerializer(ExtSerializer& serializer) +{ + return m_serializers.insert(generateKey(serializer), &serializer); +} + + +bool ExtSerializationImpl::unregisterSerializer(ExtSerializer& serializer) +{ + const uint64_t key = generateKey(serializer); + const auto entry = m_serializers.find(key); + if (entry == nullptr) + { + return false; + } + entry->second->release(); + return m_serializers.erase(key); +} + + +ExtSerializer* ExtSerializationImpl::findSerializer(uint32_t objectTypeID, uint32_t encodingID) +{ + auto entry = m_serializers.find(generateKey(objectTypeID, encodingID)); + return entry != nullptr ? entry->second : nullptr; +} + + +bool ExtSerializationImpl::setSerializationEncoding(uint32_t encodingID) +{ + m_serializationEncoding = encodingID; + + return true; +} + + +uint32_t ExtSerializationImpl::getSerializationEncoding() const +{ + return m_serializationEncoding; +} + + +void ExtSerializationImpl::setBufferProvider(BufferProvider* bufferProvider) +{ + m_bufferProvider = bufferProvider != nullptr ? bufferProvider : &s_defaultBufferProvider; +} + + +bool ExtSerializationImpl::peekHeader(uint32_t* objectTypeID, uint32_t* encodingID, uint64_t* dataSize, const void* buffer, uint64_t bufferSize) +{ + return nullptr != readHeaderFromBuffer(objectTypeID, encodingID, dataSize, reinterpret_cast(buffer), bufferSize); +} + + +const void* ExtSerializationImpl::skipObject(uint64_t& bufferSize, const void* buffer) +{ + uint64_t dataSize; + const char* next = readHeaderFromBuffer(nullptr, nullptr, &dataSize, static_cast(buffer), bufferSize); + if (next == nullptr) + { + return nullptr; + } + next += dataSize; + const uint64_t skipSize = next - static_cast(buffer); + NVBLAST_CHECK_ERROR(skipSize <= bufferSize, "Object size in buffer is too large for given buffer size.", return nullptr); + bufferSize -= skipSize; + return next; +} + + +void* ExtSerializationImpl::deserializeFromBuffer(const void* buffer, uint64_t bufferSize, uint32_t* objectTypeIDPtr) +{ + uint32_t objectTypeID; + uint32_t encodingID; + uint64_t dataSize; + void* result = nullptr; + + buffer = readHeaderFromBuffer(&objectTypeID, &encodingID, &dataSize, reinterpret_cast(buffer), bufferSize); + if (buffer != nullptr) + { + auto entry = m_serializers.find(generateKey(objectTypeID, encodingID)); + if (entry != nullptr && entry->second != nullptr) + { + result = entry->second->deserializeFromBuffer(buffer, dataSize); + } + } + + if (objectTypeIDPtr != nullptr) + { + *objectTypeIDPtr = result != nullptr ? objectTypeID : 0; + } + + return result; +} + + +uint64_t ExtSerializationImpl::serializeIntoBuffer(void*& buffer, const void* object, uint32_t objectTypeID) +{ + if (!m_serializationEncoding) + { + NVBLAST_LOG_ERROR("ExtSerializationImpl::serializeIntoBuffer: no serialization encoding has been set."); + return false; // No encoding available + } + + auto entry = m_serializers.find(generateKey(objectTypeID, m_serializationEncoding)); + if (entry == nullptr || entry->second == nullptr) + { + return false; + } + + const uint64_t size = entry->second->serializeIntoBuffer(buffer, *m_bufferProvider, object, HeaderSize); + if (size < HeaderSize) + { + NVBLAST_LOG_ERROR("ExtSerializationImpl::serializeIntoBuffer: failed to write data to buffer."); + return 0; + } + + writeHeaderIntoBuffer(reinterpret_cast(buffer), HeaderSize, objectTypeID, m_serializationEncoding, size - HeaderSize); + + return size; +} + + +void ExtSerializationImpl::release() +{ + NVBLAST_DELETE(this, ExtSerializationImpl); +} + + +//////// ExtSerializationImpl::AllocBufferProvider member functions //////// + +void* ExtSerializationImpl::AllocBufferProvider::requestBuffer(size_t size) +{ + return NVBLAST_ALLOC(size); +} + +} // namespace Blast +} // namespace Nv + + +Nv::Blast::ExtSerialization* NvBlastExtSerializationCreate() +{ + Nv::Blast::ExtSerializationImpl* serialization = NVBLAST_NEW(Nv::Blast::ExtSerializationImpl) (); + + // Automatically load LL serializers + NvBlastExtLlSerializerLoadSet(*serialization); + + return serialization; +} -- cgit v1.2.3