RawFileReader.hpp

Go to the documentation of this file.

#ifndef BYTEME_RAW_FILE_READER_HPP
#define BYTEME_RAW_FILE_READER_HPP
 
#include <vector>
#include <stdexcept>
#include <cstdio>
#include <cstddef>
#include <optional>
 
#include "Reader.hpp"
#include "SelfClosingFILE.hpp"
 
namespace byteme {
 
struct RawFileReaderOptions {
    std::optional<std::size_t> buffer_size;
};
 
class RawFileReader final : public Reader {
public:
    RawFileReader(const char* path, const RawFileReaderOptions& options) : my_file(path, "rb") {
        set_optional_bufsiz(my_file, options.buffer_size);
    }
 
public:
    std::size_t read(unsigned char* buffer, std::size_t n) {
        auto& handle = my_file.handle;
        auto read = std::fread(buffer, sizeof(unsigned char), n, handle);
        if (read < n && std::ferror(handle)) {
            throw std::runtime_error("file read failed");
        }
        return read;
    }
 
private:
    SelfClosingFILE my_file;
};
 
}
 
#endif