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

Question

我想N从文件流中读取数据字节并将它们附加到矢量.所以我们说我们有一个

basic_ifstream<uint8_t> myFileStream;

和a

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可能需要重新分配整个向量,从而使您之前的无效ptr.你需要在之后使用指针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当然),此实现也将起作用.