OstreamWriter.hpp

Go to the documentation of this file.

#ifndef BYTEME_OSTREAM_WRITER_HPP
#define BYTEME_OSTREAM_WRITER_HPP
 
#include <ostream>
#include <stdexcept>
#include "Writer.hpp"
 
namespace byteme {
 
template<class Pointer_ = std::ostream*>
class OstreamWriter : public Writer {
public:
    OstreamWriter(Pointer_ output) : my_output(std::move(output)) {}
 
    using Writer::write;
 
    void write(const unsigned char* buffer, size_t n) {
        my_output->write(reinterpret_cast<const char*>(buffer), n);
        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