RawBufferWriter.hpp

Go to the documentation of this file.

#ifndef BYTEME_RAW_BUFFER_WRITER_HPP
#define BYTEME_RAW_BUFFER_WRITER_HPP
 
#include "Writer.hpp"
#include <vector>
 
namespace byteme {
 
class RawBufferWriter : public Writer {
public:
    RawBufferWriter(size_t reserve = 0) {
        output.reserve(reserve);
    }
 
public:
    using Writer::write;
 
    void write(const unsigned char* buffer, size_t n) {
        output.insert(output.end(), buffer, buffer + n);
    }
 
    void finish() {}
 
public:
    // Exposed for back-compatibility only.
    std::vector<unsigned char> output;
    std::vector<unsigned char>& get_output() {
        return output;
    }
};
 
}
 
#endif