summaryrefslogtreecommitdiff
path: root/src/mesh/assimp-master/include/assimp/IOStreamBuffer.h
diff options
context:
space:
mode:
authorsanine <sanine.not@pm.me>2022-03-04 10:47:15 -0600
committersanine <sanine.not@pm.me>2022-03-04 10:47:15 -0600
commit058f98a63658dc1a2579826ba167fd61bed1e21f (patch)
treebcba07a1615a14d943f3af3f815a42f3be86b2f3 /src/mesh/assimp-master/include/assimp/IOStreamBuffer.h
parent2f8028ac9e0812cb6f3cbb08f0f419e4e717bd22 (diff)
add assimp submodule
Diffstat (limited to 'src/mesh/assimp-master/include/assimp/IOStreamBuffer.h')
-rw-r--r--src/mesh/assimp-master/include/assimp/IOStreamBuffer.h352
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