mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-11 10:06:47 +00:00
181 lines
6.5 KiB
C++
181 lines
6.5 KiB
C++
|
|
/********************************************************************************
|
||
|
|
* *
|
||
|
|
* This file is part of IfcOpenShell. *
|
||
|
|
* *
|
||
|
|
* IfcOpenShell is free software: you can redistribute it and/or modify *
|
||
|
|
* it under the terms of the Lesser GNU General Public License as published by *
|
||
|
|
* the Free Software Foundation, either version 3.0 of the License, or *
|
||
|
|
* (at your option) any later version. *
|
||
|
|
* *
|
||
|
|
* IfcOpenShell is distributed in the hope that it will be useful, *
|
||
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
||
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
||
|
|
* Lesser GNU General Public License for more details. *
|
||
|
|
* *
|
||
|
|
* You should have received a copy of the Lesser GNU General Public License *
|
||
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>. *
|
||
|
|
* *
|
||
|
|
********************************************************************************/
|
||
|
|
|
||
|
|
/*********************************************************************************
|
||
|
|
* *
|
||
|
|
* Reads a file and provides functions to access its *
|
||
|
|
* contents randomly and character by character *
|
||
|
|
* *
|
||
|
|
********************************************************************************/
|
||
|
|
|
||
|
|
#ifndef IFCSPFSTREAM_H
|
||
|
|
#define IFCSPFSTREAM_H
|
||
|
|
|
||
|
|
#include "ifc_parse_api.h"
|
||
|
|
|
||
|
|
/*
|
||
|
|
#include <string>
|
||
|
|
|
||
|
|
#ifdef USE_MMAP
|
||
|
|
#include <boost/iostreams/device/mapped_file.hpp>
|
||
|
|
#endif
|
||
|
|
|
||
|
|
namespace IfcParse {
|
||
|
|
/// The FileReader class represents a ISO 10303-21 IFC-SPF file in memory.
|
||
|
|
/// The file is interpreted as a sequence of tokens which are lazily
|
||
|
|
/// interpreted only when requested.
|
||
|
|
class IFC_PARSE_API FileReader {
|
||
|
|
private:
|
||
|
|
#ifdef USE_MMAP
|
||
|
|
boost::iostreams::mapped_file_source mfs;
|
||
|
|
#endif
|
||
|
|
FILE* stream_;
|
||
|
|
const char* buffer_;
|
||
|
|
size_t ptr_;
|
||
|
|
size_t len_;
|
||
|
|
size_t buf_size_;
|
||
|
|
size_t ptr_offset_ = 0;
|
||
|
|
|
||
|
|
public:
|
||
|
|
bool valid;
|
||
|
|
bool eof;
|
||
|
|
size_t size;
|
||
|
|
|
||
|
|
FileReader(const std::string& path, bool mmap = false, size_t buf_size = 0);
|
||
|
|
FileReader(std::istream& stream, int length);
|
||
|
|
FileReader(void* data, int length);
|
||
|
|
~FileReader();
|
||
|
|
/// Returns the character at the cursor
|
||
|
|
char peek();
|
||
|
|
/// Returns the character at specified offset
|
||
|
|
char Read(size_t offset);
|
||
|
|
/// Increment the file cursor and reads new page if necessary
|
||
|
|
void increment();
|
||
|
|
void Close();
|
||
|
|
/// Moves the file cursor to an arbitrary offset in the file
|
||
|
|
void seek(size_t offset);
|
||
|
|
/// Returns the cursor position
|
||
|
|
size_t Tell() const;
|
||
|
|
|
||
|
|
bool is_eof_at(size_t) const;
|
||
|
|
void increment_at(size_t&);
|
||
|
|
char peek_at(size_t);
|
||
|
|
|
||
|
|
operator bool() const { return valid && !eof; }
|
||
|
|
};
|
||
|
|
} // namespace IfcParse
|
||
|
|
|
||
|
|
#endif
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <cstddef>
|
||
|
|
#include <memory>
|
||
|
|
#include <optional>
|
||
|
|
#include <string>
|
||
|
|
#include <variant>
|
||
|
|
#include <stdexcept>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
namespace IfcParse {
|
||
|
|
|
||
|
|
/// \brief Read-only file accessor that supports four backends:
|
||
|
|
/// - full-buffer in RAM
|
||
|
|
/// - paged (LRU-cached with capacity)
|
||
|
|
/// - memory-mapped via boost::iostreams
|
||
|
|
/// - user-pushed sequential pages (caller feeds pages intended for streaming reads in WASM)
|
||
|
|
class IFC_PARSE_API FileReader {
|
||
|
|
public:
|
||
|
|
struct Page { std::vector<char> data; };
|
||
|
|
|
||
|
|
/// \brief Tag to choose memory-mapped backend via boost::iostreams::mapped_file_source.
|
||
|
|
struct mmap_tag {};
|
||
|
|
/// \brief Tag to choose user-pushed sequential pages.
|
||
|
|
struct caller_fed_tag {};
|
||
|
|
|
||
|
|
/// \brief Construct a FileReader.
|
||
|
|
/// \param fn File path.
|
||
|
|
/// \param maybe_mmaped_or_chunked
|
||
|
|
/// - std::nullopt: full-buffer mode (entire file loaded into memory)
|
||
|
|
/// - size_t: paged mode with this page size in bytes (e.g., 4096)
|
||
|
|
/// - mmap_tag: memory-mapped mode using boost::iostreams::mapped_file_source
|
||
|
|
/// \param page_cache_capacity Number of pages to keep in the LRU when in paged mode.
|
||
|
|
FileReader(const std::string& fn);
|
||
|
|
|
||
|
|
FileReader(const std::string& fn, const mmap_tag&);
|
||
|
|
|
||
|
|
FileReader(const caller_fed_tag&);
|
||
|
|
FileReader(const std::string& content, const caller_fed_tag&);
|
||
|
|
|
||
|
|
FileReader(const std::string& fn, size_t page_size, size_t page_capacity);
|
||
|
|
|
||
|
|
/// \brief Copy-construct a new reader sharing the underlying storage but with its own cursor.
|
||
|
|
FileReader clone() const;
|
||
|
|
|
||
|
|
/// \brief Seek to an absolute byte position.
|
||
|
|
/// \throws std::out_of_range if pos > size().
|
||
|
|
void seek(size_t pos);
|
||
|
|
|
||
|
|
/// \brief Return the current cursor position.
|
||
|
|
size_t tell() const;
|
||
|
|
|
||
|
|
/// \brief Total file size in bytes.
|
||
|
|
size_t size() const;
|
||
|
|
|
||
|
|
/// \brief Peek the byte at the current cursor.
|
||
|
|
/// \throws std::out_of_range at EOF.
|
||
|
|
char peek() const;
|
||
|
|
|
||
|
|
/// \brief Advance the cursor by n bytes (default 1).
|
||
|
|
/// \throws std::out_of_range if advancing crosses EOF.
|
||
|
|
void increment(size_t n = 1);
|
||
|
|
|
||
|
|
/// \brief Push the next sequential page (pushed backend only).
|
||
|
|
/// \param data Contents of the page.
|
||
|
|
/// \throws std::logic_error if the current backend is not pushed mode, or if a next page is already queued.
|
||
|
|
void push_next_page(const std::string& data);
|
||
|
|
|
||
|
|
/// \brief Returns true if the cursor is at or beyond the end of available data.
|
||
|
|
/// For the pushed backend, EOF means all pushed bytes have been consumed.
|
||
|
|
bool eof() const;
|
||
|
|
|
||
|
|
/// \brief Equivalent of peek() followed by increment(1)
|
||
|
|
char read();
|
||
|
|
|
||
|
|
/// \brief Equivalent of peek() followed by increment(1)
|
||
|
|
char get(size_t offset) const;
|
||
|
|
|
||
|
|
struct Impl {
|
||
|
|
virtual ~Impl() = default;
|
||
|
|
virtual size_t size() const = 0;
|
||
|
|
virtual char get(size_t pos) const = 0;
|
||
|
|
/// \brief Backend may support pushing pages; default throws.
|
||
|
|
virtual void push_next_page(const std::string&) {
|
||
|
|
throw std::logic_error("push_next_page: backend does not support pushed mode");
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
private:
|
||
|
|
std::shared_ptr<Impl> impl_;
|
||
|
|
size_t cursor_ = 0;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace IfcParse
|
||
|
|
|
||
|
|
#endif
|