IstreamReader.hpp

Go to the documentation of this file.

#ifndef BYTEME_ISTREAM_READER_HPP
#define BYTEME_ISTREAM_READER_HPP
 
#include <vector>
#include <stdexcept>
#include <memory>
#include <cstddef>
 
#include "Reader.hpp"
#include "utils.hpp"
 
namespace byteme {
 
template<class Pointer_>
class IstreamReader final : public Reader {
public:
    IstreamReader(Pointer_ input) : my_input(std::move(input)) {}
 
public:
    std::size_t read(unsigned char* buffer, std::size_t n) {
        return safe_read<std::streamsize>(buffer, n, [this](unsigned char* buffer, std::streamsize n) -> std::streamsize {
            my_input->read(reinterpret_cast<char*>(buffer), n); 
            const auto my_read = my_input->gcount();
            if (my_read < n) {
                if (!(my_input->eof())) {
                    throw std::runtime_error("failed to finish reading the input stream");
                }
            }
            return my_read;
        });
    }
 
private:
    Pointer_ my_input;
};
 
}
 
#endif