将文件加载到向量<char> </char>中

时间:2011-08-30 10:33:55

标签: c++ iostream

我想将文本文件的内容加载到vector<char>(或者加入任何char输入迭代器,如果可能的话)。目前我的代码如下所示:

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 ++。有更直接的方法吗?

5 个答案:

答案 0 :(得分:16)

如果你想避免通过char读取char:

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

答案 1 :(得分:10)

我认为它是这样的,但没有环境可以测试它:

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

你可能需要使用io操纵器来处理换行/空白等事情。

编辑:如评论中所述,可能会受到影响。

答案 2 :(得分:6)

使用迭代器:

#include <iterator>

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

答案 3 :(得分:6)

另一种方法,首先使用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));
}

答案 4 :(得分:2)

有很多好的回应。谢谢大家!我决定使用的代码是:

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

显然,这不适用于超大型文件或性能关键代码,但它足以用于通用目的。