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

时间:2018-05-23 15:08:33

标签: c++ vector readfile ifstream

基于这个问题:

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个字节读入向量(清除之间的向量)。我不知道如何指定要读取的文件数量(即如何设置迭代器)。这甚至可能吗?

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

3 个答案:

答案 0 :(得分:3)

您可以使用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.
}

答案 1 :(得分:1)

你可以写一个函数:

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等进行回调。

答案 2 :(得分:0)

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

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);

您可以再次调用此功能。