1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
#ifndef BYTEME_OSTREAM_WRITER_HPP
#define BYTEME_OSTREAM_WRITER_HPP
#include <ostream>
#include <stdexcept>
#include "Writer.hpp"
/**
* @file OstreamWriter.hpp
*
* @brief Write bytes to an arbitrary output stream.
*/
namespace byteme {
/**
* @brief Read bytes from a `std::ostream`.
*
* @tparam Pointer_ A (possibly smart) pointer to an `std::ostream` object.
*
* This is just a wrapper around `std::ostream::write` for compatibility.
*/
template<class Pointer_ = std::ostream*>
class OstreamWriter : public Writer {
public:
/**
* @param output Pointer to an output stream.
* This is assumed to live until `finish()` is called.
*/
OstreamWriter(Pointer_ output) : ptr(std::move(output)) {}
using Writer::write;
void write(const unsigned char* buffer, size_t n) {
ptr->write(reinterpret_cast<const char*>(buffer), n);
if (!(ptr->good())) {
throw std::runtime_error("failed to write to arbitrary output stream");
}
}
void finish() {
ptr->flush();
if (ptr->fail() || ptr->bad()) {
throw std::runtime_error("failed to flush to arbitrary output stream");
}
}
private:
Pointer_ ptr;
};
}
#endif
|
