SomeFileReader.hpp

Go to the documentation of this file.

#ifndef BYTEME_SOME_FILE_READER_HPP
#define BYTEME_SOME_FILE_READER_HPP
 
#include "Reader.hpp"
#include "RawFileReader.hpp"
#include "GzipFileReader.hpp"
#include "magic_numbers.hpp"
#include <memory>
#include <cstdio>
 
namespace byteme {
 
class SomeFileReader : public Reader {
public:
    SomeFileReader(const char* path, size_t buffer_size = 65536) { 
        unsigned char header[3];
        size_t read;
        {
            SelfClosingFILE file(path, "rb");
            read = std::fread(header, sizeof(unsigned char), 3, file.handle);
        }
 
        if (is_gzip(header, read)) {
            my_source.reset(new GzipFileReader(path, buffer_size));
        } else {
            my_source.reset(new RawFileReader(path, buffer_size));
        }
    }
 
    SomeFileReader(const std::string& path, size_t buffer_size = 65536) : SomeFileReader(path.c_str(), buffer_size) {}
 
public:
    bool load() {
        return my_source->load();
    }
 
    const unsigned char* buffer() const {
        return my_source->buffer();
    }
 
    size_t available() const {
        return my_source->available();
    }
 
private:
    std::unique_ptr<Reader> my_source;
};
 
}
 
#endif