RawBufferWriter.hpp

Go to the documentation of this file.

#ifndef BYTEME_RAW_BUFFER_WRITER_HPP
#define BYTEME_RAW_BUFFER_WRITER_HPP
 
#include <cstddef>
#include <vector>
 
#include "Writer.hpp"
#include "utils.hpp"
 
namespace byteme {
 
struct RawBufferWriterOptions {
    std::size_t reserve = 0;
};
 
class RawBufferWriter final : public Writer {
public:
    RawBufferWriter(const RawBufferWriterOptions& options) {
        output.reserve(options.reserve);
    }
 
public:
    using Writer::write;
 
    void write(const unsigned char* buffer, std::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