2011-09-09 27 views
6

我想从文件流中读取N字节的数据并将它们附加到一个矢量。所以我们可以说,我们有一个从文件读取N个字节并将它们附加到std :: vector

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

但有了这个,我得到一个运行时错误(堆损坏)。这个解决方案有什么问题?或者有没有更好的方法来做到这一点呢?

回答

12

您的问题是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,当然)。

+0

现在它工作:)感谢您的帮助! –

+4

@Ajeet:'reserve()'不会改变容器的size()。我重视表现的正确性。 –

+0

是的。只有当他使用push_back()时才有用。删除我的评论。谢谢@R。 Martinho –

相关问题