144 lines
3.1 KiB
C++
144 lines
3.1 KiB
C++
#pragma once
|
|
/*
|
|
* Copyright (C) 2024 Brett Terpstra
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program 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
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef BLT_FS_STREAM_WRAPPERS_H
|
|
#define BLT_FS_STREAM_WRAPPERS_H
|
|
|
|
#include <iosfwd>
|
|
#include <sstream>
|
|
#include <blt/fs/fwddecl.h>
|
|
|
|
namespace blt::fs
|
|
{
|
|
/**
|
|
* reader_t wrapper for fstream
|
|
*/
|
|
class fstream_reader_t final : public reader_t
|
|
{
|
|
public:
|
|
explicit fstream_reader_t(std::istream& stream);
|
|
|
|
explicit fstream_reader_t(fstream_reader_t& copy) = delete;
|
|
|
|
fstream_reader_t& operator=(const fstream_reader_t& copy) = delete;
|
|
|
|
i64 read(char* buffer, size_t bytes) override;
|
|
|
|
private:
|
|
std::istream* m_stream;
|
|
};
|
|
|
|
class fstream_writer_t final : public writer_t
|
|
{
|
|
public:
|
|
explicit fstream_writer_t(std::ostream& stream);
|
|
|
|
explicit fstream_writer_t(fstream_writer_t& copy) = delete;
|
|
|
|
fstream_writer_t& operator=(const fstream_writer_t& copy) = delete;
|
|
|
|
i64 write(const char* buffer, size_t bytes) override;
|
|
|
|
void flush() override;
|
|
|
|
virtual ~fstream_writer_t() override // NOLINT
|
|
{
|
|
flush();
|
|
}
|
|
|
|
private:
|
|
std::ostream* m_stream;
|
|
};
|
|
|
|
class reader_wrapper_t
|
|
{
|
|
public:
|
|
explicit reader_wrapper_t(reader_t& reader): m_reader(&reader)
|
|
{}
|
|
|
|
template <typename T>
|
|
void read(T& out)
|
|
{
|
|
if (!m_reader->read(reinterpret_cast<char*>(&out), sizeof(T)))
|
|
throw std::runtime_error("Failed to read from reader");
|
|
}
|
|
|
|
template <typename T>
|
|
friend reader_wrapper_t& operator>>(reader_wrapper_t& reader, T& t)
|
|
{
|
|
reader.read(t);
|
|
return reader;
|
|
}
|
|
|
|
private:
|
|
reader_t* m_reader;
|
|
};
|
|
|
|
class writer_wrapper_t
|
|
{
|
|
public:
|
|
explicit writer_wrapper_t(writer_t& writer): m_writer(&writer)
|
|
{}
|
|
|
|
template <typename T>
|
|
void write(const T& t)
|
|
{
|
|
static_assert(std::is_trivially_copyable_v<T>);
|
|
m_writer->write(reinterpret_cast<char*>(&t), sizeof(T));
|
|
}
|
|
|
|
template <typename T>
|
|
friend writer_wrapper_t& operator<<(writer_wrapper_t& writer, const T& t)
|
|
{
|
|
writer.write(t);
|
|
return writer;
|
|
}
|
|
|
|
private:
|
|
writer_t* m_writer;
|
|
};
|
|
|
|
class writer_string_wrapper_t
|
|
{
|
|
public:
|
|
explicit writer_string_wrapper_t(writer_t& writer): m_writer(&writer)
|
|
{}
|
|
|
|
template <typename T>
|
|
void write(const T& t)
|
|
{
|
|
std::stringstream ss;
|
|
ss << t;
|
|
const auto str = ss.str();
|
|
m_writer->write(str.data(), str.size());
|
|
}
|
|
|
|
template <typename T>
|
|
friend writer_string_wrapper_t& operator<<(writer_string_wrapper_t& writer, const T& t)
|
|
{
|
|
writer.write(t);
|
|
return writer;
|
|
}
|
|
|
|
private:
|
|
writer_t* m_writer;
|
|
};
|
|
}
|
|
|
|
#endif //BLT_FS_STREAM_WRAPPERS_H
|