diff options
Diffstat (limited to 'src/mesh/assimp-master/include/assimp/IOStreamBuffer.h')
-rw-r--r-- | src/mesh/assimp-master/include/assimp/IOStreamBuffer.h | 352 |
1 files changed, 352 insertions, 0 deletions
diff --git a/src/mesh/assimp-master/include/assimp/IOStreamBuffer.h b/src/mesh/assimp-master/include/assimp/IOStreamBuffer.h new file mode 100644 index 0000000..09ca1c9 --- /dev/null +++ b/src/mesh/assimp-master/include/assimp/IOStreamBuffer.h @@ -0,0 +1,352 @@ +/* +Open Asset Import Library (assimp) +---------------------------------------------------------------------- + +Copyright (c) 2006-2022, assimp team + +All rights reserved. + +Redistribution and use of this software in source and binary forms, +with or without modification, are permitted provided that the +following conditions are met: + +* Redistributions of source code must retain the above +copyright notice, this list of conditions and the +following disclaimer. + +* Redistributions in binary form must reproduce the above +copyright notice, this list of conditions and the +following disclaimer in the documentation and/or other +materials provided with the distribution. + +* Neither the name of the assimp team, nor the names of its +contributors may be used to endorse or promote products +derived from this software without specific prior +written permission of the assimp team. + +THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS +"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT +LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR +A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT +OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, +SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT +LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, +DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY +THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE +OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +---------------------------------------------------------------------- +*/ + +#pragma once +#ifndef AI_IOSTREAMBUFFER_H_INC +#define AI_IOSTREAMBUFFER_H_INC + +#ifdef __GNUC__ +#pragma GCC system_header +#endif + +#include <assimp/ParsingUtils.h> +#include <assimp/types.h> +#include <assimp/IOStream.hpp> + +#include <vector> + +namespace Assimp { + +// --------------------------------------------------------------------------- +/** + * Implementation of a cached stream buffer. + */ +template <class T> +class IOStreamBuffer { +public: + /// @brief The class constructor. + IOStreamBuffer(size_t cache = 4096 * 4096); + + /// @brief The class destructor. + ~IOStreamBuffer(); + + /// @brief Will open the cached access for a given stream. + /// @param stream The stream to cache. + /// @return true if successful. + bool open(IOStream *stream); + + /// @brief Will close the cached access. + /// @return true if successful. + bool close(); + + /// @brief Returns the file-size. + /// @return The file-size. + size_t size() const; + + /// @brief Returns the cache size. + /// @return The cache size. + size_t cacheSize() const; + + /// @brief Will read the next block. + /// @return true if successful. + bool readNextBlock(); + + /// @brief Returns the number of blocks to read. + /// @return The number of blocks. + size_t getNumBlocks() const; + + /// @brief Returns the current block index. + /// @return The current block index. + size_t getCurrentBlockIndex() const; + + /// @brief Returns the current file pos. + /// @return The current file pos. + size_t getFilePos() const; + + /// @brief Will read the next line. + /// @param buffer The buffer for the next line. + /// @return true if successful. + bool getNextDataLine(std::vector<T> &buffer, T continuationToken); + + /// @brief Will read the next line ascii or binary end line char. + /// @param buffer The buffer for the next line. + /// @return true if successful. + bool getNextLine(std::vector<T> &buffer); + + /// @brief Will read the next block. + /// @param buffer The buffer for the next block. + /// @return true if successful. + bool getNextBlock(std::vector<T> &buffer); + +private: + IOStream *m_stream; + size_t m_filesize; + size_t m_cacheSize; + size_t m_numBlocks; + size_t m_blockIdx; + std::vector<T> m_cache; + size_t m_cachePos; + size_t m_filePos; +}; + +template <class T> +AI_FORCE_INLINE IOStreamBuffer<T>::IOStreamBuffer(size_t cache) : + m_stream(nullptr), + m_filesize(0), + m_cacheSize(cache), + m_numBlocks(0), + m_blockIdx(0), + m_cachePos(0), + m_filePos(0) { + m_cache.resize(cache); + std::fill(m_cache.begin(), m_cache.end(), '\n'); +} + +template <class T> +AI_FORCE_INLINE IOStreamBuffer<T>::~IOStreamBuffer() { + // empty +} + +template <class T> +AI_FORCE_INLINE bool IOStreamBuffer<T>::open(IOStream *stream) { + // file still opened! + if (nullptr != m_stream) { + return false; + } + + // Invalid stream pointer + if (nullptr == stream) { + return false; + } + + m_stream = stream; + m_filesize = m_stream->FileSize(); + if (m_filesize == 0) { + return false; + } + if (m_filesize < m_cacheSize) { + m_cacheSize = m_filesize; + } + + m_numBlocks = m_filesize / m_cacheSize; + if ((m_filesize % m_cacheSize) > 0) { + m_numBlocks++; + } + + return true; +} + +template <class T> +AI_FORCE_INLINE bool IOStreamBuffer<T>::close() { + if (nullptr == m_stream) { + return false; + } + + // init counters and state vars + m_stream = nullptr; + m_filesize = 0; + m_numBlocks = 0; + m_blockIdx = 0; + m_cachePos = 0; + m_filePos = 0; + + return true; +} + +template <class T> +AI_FORCE_INLINE + size_t + IOStreamBuffer<T>::size() const { + return m_filesize; +} + +template <class T> +AI_FORCE_INLINE + size_t + IOStreamBuffer<T>::cacheSize() const { + return m_cacheSize; +} + +template <class T> +AI_FORCE_INLINE bool IOStreamBuffer<T>::readNextBlock() { + m_stream->Seek(m_filePos, aiOrigin_SET); + size_t readLen = m_stream->Read(&m_cache[0], sizeof(T), m_cacheSize); + if (readLen == 0) { + return false; + } + if (readLen < m_cacheSize) { + m_cacheSize = readLen; + } + m_filePos += m_cacheSize; + m_cachePos = 0; + m_blockIdx++; + + return true; +} + +template <class T> +AI_FORCE_INLINE size_t IOStreamBuffer<T>::getNumBlocks() const { + return m_numBlocks; +} + +template <class T> +AI_FORCE_INLINE size_t IOStreamBuffer<T>::getCurrentBlockIndex() const { + return m_blockIdx; +} + +template <class T> +AI_FORCE_INLINE size_t IOStreamBuffer<T>::getFilePos() const { + return m_filePos; +} + +template <class T> +AI_FORCE_INLINE bool IOStreamBuffer<T>::getNextDataLine(std::vector<T> &buffer, T continuationToken) { + buffer.resize(m_cacheSize); + if (m_cachePos >= m_cacheSize || 0 == m_filePos) { + if (!readNextBlock()) { + return false; + } + } + + size_t i = 0; + for (;;) { + if (continuationToken == m_cache[m_cachePos] && IsLineEnd(m_cache[m_cachePos + 1])) { + ++m_cachePos; + while (m_cache[m_cachePos] != '\n') { + ++m_cachePos; + } + ++m_cachePos; + } else if (IsLineEnd(m_cache[m_cachePos])) { + break; + } + + buffer[i] = m_cache[m_cachePos]; + ++m_cachePos; + ++i; + + if(i == buffer.size()) { + buffer.resize(buffer.size() * 2); + } + + if (m_cachePos >= size()) { + break; + } + if (m_cachePos >= m_cacheSize) { + if (!readNextBlock()) { + return false; + } + } + } + + buffer[i] = '\n'; + ++m_cachePos; + + return true; +} + +static AI_FORCE_INLINE bool isEndOfCache(size_t pos, size_t cacheSize) { + return (pos == cacheSize); +} + +template <class T> +AI_FORCE_INLINE bool IOStreamBuffer<T>::getNextLine(std::vector<T> &buffer) { + buffer.resize(m_cacheSize); + if (isEndOfCache(m_cachePos, m_cacheSize) || 0 == m_filePos) { + if (!readNextBlock()) { + return false; + } + } + + if (IsLineEnd(m_cache[m_cachePos])) { + // skip line end + while (m_cache[m_cachePos] != '\n') { + ++m_cachePos; + } + ++m_cachePos; + if (isEndOfCache(m_cachePos, m_cacheSize)) { + if (!readNextBlock()) { + return false; + } + } + } + + size_t i(0); + while (!IsLineEnd(m_cache[m_cachePos])) { + buffer[i] = m_cache[m_cachePos]; + ++m_cachePos; + ++i; + + if(i == buffer.size()) { + buffer.resize(buffer.size() * 2); + } + + if (m_cachePos >= m_cacheSize) { + if (!readNextBlock()) { + return false; + } + } + } + buffer[i] = '\n'; + ++m_cachePos; + + return true; +} + +template <class T> +AI_FORCE_INLINE bool IOStreamBuffer<T>::getNextBlock(std::vector<T> &buffer) { + // Return the last block-value if getNextLine was used before + if (0 != m_cachePos) { + buffer = std::vector<T>(m_cache.begin() + m_cachePos, m_cache.end()); + m_cachePos = 0; + } else { + if (!readNextBlock()) { + return false; + } + + buffer = std::vector<T>(m_cache.begin(), m_cache.end()); + } + + return true; +} + +} // namespace Assimp + +#endif // AI_IOSTREAMBUFFER_H_INC |