如何以 STL 方式将文件中一定数量的字符复制到向量？

Question

如果我想将文件的内容复制到向量中，我可以这样做：

std::ifstream file("path_to_file");
std::vector<char> buffer(std::istream_iterator<char>(file), 
                         std::istream_iterator<char>());

我的问题是，如果我只想复制前 n 个字符，我该怎么做？

编辑我可以编写自己的copy，但是有没有办法仅使用现有组件来做到这一点？

Answer 1

正如 Alexander 指出的，最快的方法是

std::vector<char> buffer(n);
file.read(&buffer[0], n);

在 C++0x 中，您可以使用 buffer.data() 而不是 &buffer[0];如果n == 0，后者具有未定义的行为。

Answer 2

正如 史蒂夫，这需要< code>copy_n()，由于疏忽，它不在当前的标准库中，但将在 C++1x 中。您可以轻松地自己实现一个，这是我认为正确的一个：

template<class InIt, class OutIt> 
OutIt copy_n(InIt src, OutIt dest, size_t n) 
{
  if (!n) return dest;
  *dest = *src;
  while (--n)
    *++dest = *++src;
  return ++dest; 
} 

请注意，std::copy_n() 假定输入迭代器能够传递 n 对象。从文件读取时，这可能会出现问题。

---

如果没有 std::copy_n()，您可以使用 std::generate_n。

template< typename InIt >
struct input_generator {
  typedef std::iterator_traits<InIt>::value_type value_type;

  input_generator(InIt begin, InIt end) begin_(begin), end_(end) {}

  value_type operator()()
  {
    assert(it_ != end);
    return *it_++;
  }

  Init begin_;
  Init end_;
};


std::vector<char> buffer;
buffer.reserve(42);

std::generate_n( std::back_inserter(buffer)
               , 42
               , input_generator(std::istream_iterator<char>(file))
               , input_generator(std::istream_iterator<char>()) );

---

但是，我不认为这比直接从文件中读取 avakar 显示。

Answer 3

“STL方式”是使用copy_n，它是STL中的，但不是C++标准中的。

Answer 4

char *buffer = new char[n];
file.read(buffer,n);
std::vector<char> myVector
for (int i=0;i<n;i++) myVector.push_back(buffer[i]);
delete [] buffer;

// myVector now holds the first n bytes of file.

这可能不是最漂亮或最快的方法，但这是我会做的。

Answer 5

感谢@sbi，我忘记了 generate_n。

file 内的 streambuf 有一个函数 snextc，一旦其隐式 this<，它就会返回下一个字符，将其限定为生成器函数。 /code> 参数已绑定。

generate_n( back_inserter( buffer ), 42,
            tr1::bind( tr1::mem_fn( &streambuf::snextc ), file.rdbuf() ) );

对于vector，直接读取即可。这对于 deque 或其他什么都有好处。