41

为了提高从文件读取的性能,我试图将一个大(几 MB)文件的全部内容读入内存,然后使用 istringstream 访问信息。

我的问题是,读取这些信息并将其“导入”到字符串流中的最佳方式是什么?这种方法的一个问题(见下文)是,在创建字符串流时,缓冲区会被复制,内存使用量会加倍。

#include <fstream>
#include <sstream>

using namespace std;

int main() {
  ifstream is;
  is.open (sFilename.c_str(), ios::binary );

  // get length of file:
  is.seekg (0, std::ios::end);
  long length = is.tellg();
  is.seekg (0, std::ios::beg);

  // allocate memory:
  char *buffer = new char [length];

  // read data as a block:
  is.read (buffer,length);

  // create string stream of memory contents
  // NOTE: this ends up copying the buffer!!!
  istringstream iss( string( buffer ) );

  // delete temporary buffer
  delete [] buffer;

  // close filestream
  is.close();

  /* ==================================
   * Use iss to access data
   */

}
4

3 回答 3

53

std::ifstream有一个方法rdbuf(),它返回一个指向 a 的指针filebuf。然后,您可以将其“推送”filebuf到您的stringstream

#include <fstream>
#include <sstream>

int main()
{
    std::ifstream file( "myFile" );

    if ( file )
    {
        std::stringstream buffer;

        buffer << file.rdbuf();

        file.close();

        // operations on the buffer...
    }
}

编辑:正如 Martin York 在评论中所说,这可能不是最快的解决方案,因为stringstream'soperator<<将逐个字符地读取 filebuf。您可能想检查他的答案,他像以前一样使用ifstream'read方法,然后将stringstream缓冲区设置为指向先前分配的内存。

于 2008-09-25T09:52:08.000 回答
43

好的。我并不是说这会比从文件中读取更快

但这是一种创建缓冲区的方法,在将数据读入缓冲区后,直接将其用作字符串流的源。

注意值得一提的是 std::ifstream 是缓冲的。它以(相对较大的)块的形式从文件中读取数据。对缓冲区执行流操作,仅在需要更多数据时才返回文件进行另一次读取。因此,在将所有数据吸入内存之前,请确认这是一个瓶颈。

#include <fstream>
#include <sstream>
#include <vector>

int main()
{
    std::ifstream       file("Plop");
    if (file)
    {
        /*
         * Get the size of the file
         */
        file.seekg(0,std::ios::end);
        std::streampos          length = file.tellg();
        file.seekg(0,std::ios::beg);

        /*
         * Use a vector as the buffer.
         * It is exception safe and will be tidied up correctly.
         * This constructor creates a buffer of the correct length.
         *
         * Then read the whole file into the buffer.
         */
        std::vector<char>       buffer(length);
        file.read(&buffer[0],length);

        /*
         * Create your string stream.
         * Get the stringbuffer from the stream and set the vector as it source.
         */
        std::stringstream       localStream;
        localStream.rdbuf()->pubsetbuf(&buffer[0],length);

        /*
         * Note the buffer is NOT copied, if it goes out of scope
         * the stream will be reading from released memory.
         */
    }
}
于 2008-09-26T10:29:20.103 回答
1

这对我来说似乎是过早的优化。处理过程中完成了多少工作。假设一个现代化的桌面/服务器,而不是嵌入式系统,在初始化期间复制几 MB 的数据相当便宜,尤其是与首先从磁盘读取文件相比。我会坚持你所拥有的,在系统完成时对其进行测量,并确定潜在的性能提升是否值得。当然,如果内存紧张,这是在一个内部循环中,或者是一个经常被调用的程序(比如每秒一次),这会改变平衡。

于 2008-09-25T12:52:24.977 回答