将文件加载到矢量中；

Question

我想将文本文件的内容加载到 vector （或任何字符输入迭代器，如果可能的话）。目前我的代码如下所示：

std::vector<char> vec;
std::ifstream file("test.txt");
assert(file.is_open());
while (!(file.eof() || file.fail())) {
    char buffer[100];
    file.read(buffer, 100);
    vec.insert(vec.end(), buffer, buffer + file.gcount());
}

我不喜欢手动使用缓冲区（为什么是 100 个字符？为什么不是 200、25 或其他什么？），也不喜欢这需要大量的行。该代码看起来非常丑陋且非 C++。有更直接的方法吗？

Answer 1

如果你想避免逐个字符地读取：

if (!file.eof() && !file.fail())
{
    file.seekg(0, std::ios_base::end);
    std::streampos fileSize = file.tellg();
    vec.resize(fileSize);

    file.seekg(0, std::ios_base::beg);
    file.read(&vec[0], fileSize);
}

Answer 2

另一种方法是首先使用 rdbuf() 来将整个文件读取到 std::stringstream ：

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

// for check:
#include <algorithm>
#include <iterator>
#include <iostream>

int main() {
   std::ifstream file("test.cc");
   std::ostringstream ss;
   ss << file.rdbuf();
   const std::string& s = ss.str();
   std::vector<char> vec(s.begin(), s.end());

   // check:
   std::copy(vec.begin(), vec.end(), std::ostream_iterator<char>(std::cout));
}

Answer 3

我认为是这样的，但没有环境来测试它：

std::copy(std::istream_iterator<char>(file), std::istream_iterator<char>(), std::back_inserter(vec));

可能你必须使用 io 操纵器来处理诸如换行符/空格之类的事情。

编辑：正如评论中所述，可能会对性能造成影响。

Answer 4

有很多好的回应。谢谢大家！我决定使用的代码是这样的：

std::vector<char> vec;
std::ifstream file;
file.exceptions(
    std::ifstream::badbit
  | std::ifstream::failbit
  | std::ifstream::eofbit);
//Need to use binary mode; otherwise CRLF line endings count as 2 for
//`length` calculation but only 1 for `file.read` (on some platforms),
//and we get undefined  behaviour when trying to read `length` characters.
file.open("test.txt", std::ifstream::in | std::ifstream::binary);
file.seekg(0, std::ios::end);
std::streampos length(file.tellg());
if (length) {
    file.seekg(0, std::ios::beg);
    vec.resize(static_cast<std::size_t>(length));
    file.read(&vec.front(), static_cast<std::size_t>(length));
}

显然，这不适合非常大的文件或性能关键的代码，但对于一般用途来说已经足够了。

Answer 5

使用迭代器：

#include <iterator>

istream_iterator<char> data( file );
istream_iterator<char> end;
vec.insert( std::back_inserter(vec), data, end );