OstreamWriter.hpp

Go to the documentation of this file.

#ifndef BYTEME_OSTREAM_WRITER_HPP
#define BYTEME_OSTREAM_WRITER_HPP
 
#include <memory>
#include <stdexcept>
#include <cstddef>
 
#include "Writer.hpp"
 
namespace byteme {
 
template<class Pointer_>
class OstreamWriter final : public Writer {
public:
    OstreamWriter(Pointer_ output) : my_output(std::move(output)) {}
 
public:
    using Writer::write;
 
    void write(const unsigned char* buffer, std::size_t n) {
        safe_write<std::streamsize, false>(
            reinterpret_cast<const char*>(buffer),
            n,
            [&](const char* ptr0, std::streamsize n0) -> void {
                my_output->write(ptr0, n0);
                if (!(my_output->good())) {
                    throw std::runtime_error("failed to write to arbitrary output stream");
                }
            }
        );
    }
 
    void finish() {
        my_output->flush();
        if (my_output->fail() || my_output->bad()) {
            throw std::runtime_error("failed to flush to arbitrary output stream");
        }
    }
 
private:
    Pointer_ my_output;
};
 
}
 
#endif