如何使用迭代器从文件中读取 n 个字节并将它们放入 vector<uint8_t> 中？

Question

基于这个问题：

How to read a binary file into a vector of unsigned chars

他们的答案是：

std::vector<BYTE> readFile(const char* filename)
{
    // open the file:
    std::basic_ifstream<BYTE> file(filename, std::ios::binary);

    // read the data:
    return std::vector<BYTE>((std::istreambuf_iterator<BYTE>(file)),
                              std::istreambuf_iterator<BYTE>());
}

将整个文件读入向量。

我想做的是在向量中一次读取（例如）100 个字节，然后做一些事情，然后将下一个 100 个字节读入向量（清除向量之间的向量）。我不知道如何指定要读取的文件量（即如何设置迭代器）。这可能吗？

我试图避免必须编写自己的代码循环来一次复制每个字节。

Answer 1

您可以使用 ifstream::read。

std::vector<BYTE> v(100);
while ( file.read(reinterpret_cast<char*>(v.data()), 100) )
{
   // Find out how many characters were actually read.
   auto count = file.gcount();

   // Use v up to count BTYEs.
}

Answer 2

您可以编写一个函数来：

void readFile( const std::string &fileName, size_t chunk, std::function<void(const std::vector<BYTE>&)> proc )
{
    std::ifstream f( fileName );
    std::vector<BYTE> v(chunk);
    while( f.read( v.data(), v.size() ) ) {
        v.resize( f.gcount() );
        proc( v );
        v.resize( chunk );
    }
}

那么用法很简单：

void process( const std::vector<BYTE> &v ) { ... }

readFile( "foobar", 100, process ); // call process for every 100 bytes of data

或者您可以使用 lambda 等进行回调。

Answer 3

或者您可以为此编写自己的函数：

template<typename Data>
std::istreambuf_iterator<Data> readChunk(std::istreambuf_iterator<Data>& curr, std::vector<Data>& vec, size_t chunk = 100) {
    for (int i = 0; curr != std::istreambuf_iterator<Data>() && i < chunk; ++i, ++curr) {
        vec.emplace_back(*curr);
    }
    return curr;
}

并将其用作：

std::ifstream file("test.cpp");
std::vector<BYTE> v;
std::istreambuf_iterator<BYTE> curr(file);
readChunk<BYTE>(curr, v);

您可以再次调用此函数。

如何使用迭代器从文件中读取 n 个字节并将它们放入 vector<uint8_t> 中？

How do you read n bytes from a file and put them into a vector<uint8_t> using iterators?

c++

vector

ifstream

readfile