從文件讀取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可能需要重新分配整個向量，並因此使之前的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，當然）。