从文件中读取 N 个字节并将它们附加到 std::vector

Question

我想从文件流中读取 N 字节的数据并将它们附加到向量中。所以假设我们有一个

basic_ifstream<uint8_t> myFileStream;

还有一个

vector<uint8_t> myBuffer;

目前我正在做这样的事情：

myBuffer.reserve(N);
for (int i=0; i<N; ++i)
{
    uint8_t tmpByte;
    myFileStream.read(&tmpByte, 1);
    myBuffer.push_back(tmpByte);
}

但这非常慢。

现在我尝试让myFileStream.read 直接将数据复制到向量中。由于向量将其元素存储在一个连续的存储位置，我认为这样的事情应该是可能的：

uint8_t* ptr = &myBuffer.back(); // there is already some elements in the buffer (I know)
ptr++; // first element after existing data
myBuffer.resize(myBuffer.size() + N);
myFileStream.read(ptr, N);

但是这样我得到一个运行时错误（堆损坏）。这个解决方案有什么问题？还是有更好的方法来做到这一点？

Answer 1

您的问题是resize 可能需要重新分配整个向量，从而使您之前的@​​987654322@ 无效。只需在resize 之后取指针。

std::size_t oldSize = myBuffer.size();
// resize first
myBuffer.resize(oldSize + N);
uint8_t* ptr = &myBuffer[oldSize]; // already first element after existing data
myFileStream.read(ptr, N);

请注意，即使原始向量为空（当然，对于 N != 0），此实现也会起作用。