SomeBufferReader.hpp

Go to the documentation of this file.

#ifndef BYTEME_SOME_BUFFER_READER_HPP
#define BYTEME_SOME_BUFFER_READER_HPP
 
#include <cstddef>
#include <memory>
#include <cstdio>
 
#include "Reader.hpp"
#include "RawBufferReader.hpp"
#include "ZlibBufferReader.hpp"
#include "magic_numbers.hpp"
 
namespace byteme {
 
struct SomeBufferReaderOptions {
    std::size_t buffer_size = 65536;
};
 
class SomeBufferReader final : public Reader {
public:
    SomeBufferReader(const unsigned char* buffer, std::size_t length, const SomeBufferReaderOptions& options) {
        if (is_zlib(buffer, length) || is_gzip(buffer, length)) {
            ZlibBufferReaderOptions zopt;
            zopt.buffer_size = options.buffer_size;
            my_source.reset(new ZlibBufferReader(buffer, length, zopt));
        } else {
            my_source.reset(new RawBufferReader(buffer, length));
        }
    }
 
public:
    bool load() {
        return my_source->load();
    }
 
    const unsigned char* buffer() const {
        return my_source->buffer();
    }
 
    std::size_t available() const {
        return my_source->available();
    }
 
private:
    std::unique_ptr<Reader> my_source;
};
 
}
 
#endif