吃完 EOF 后重用 std::cin

Question

unix 命令wc 有这个功能：

$ wc - - -
aaa bbb ccc<EOF>
0 3 11 -
aaa bbb ccc<EOF>
0 3 11 -
aaa bbb ccc<EOF>
0 3 11 -
0 9 33 total

每个<EOF> 表示一个<C-d> 键序列，它将EOF 输入到标准输入中。然后 wc 就可以接收到这个EOF。

我正在尝试在 C++ 中实现它。一个常见的建议是clear() 和ignore() 的组合。

char c;
while (std::cin >> c) { ... }

std::cin.clear();
std::cin.ignore();

while (std::cin >> c) { /* never executed */ }

我也试过std::cin.peekg(std::cin.beg)，也不管用。

Answer 1

在 Davis 的耐心帮助下，我了解了以 Ctrl+D 完成的键入输入与我不知道的 here 文档之间的区别。 （我向戴维斯·赫林致敬。）

有一次，我得到了它，C++ 中的其余部分非常简单，就像我的MCVE 中所示。

line-count.cc:

#include <fstream>
#include <iostream>
#include <string>

unsigned process(const std::string &fileName, std::istream &in)
{
  unsigned nLines = 0;
  if (in.bad()) {
    std::cerr << "ERROR: Cannot open '" << fileName << "'!\n";
    return 0;
  }
  for (std::string buffer; std::getline(in, buffer); ++nLines);
  std::cout << "File: '" << fileName << "', " << nLines << " counted.\n";
  return nLines;
}

int main(int argc, char **argv)
{
  unsigned nLines = 0;
  for (int i = 1; i < argc; ++i) {
    const std::string arg = argv[i];
    if (arg == "-") {
      nLines += process(arg, std::cin);
      std::cin.clear();
    } else {
      std::ifstream fIn(arg.c_str());
      nLines += process(arg, fIn);
    }
  }
  std::cout << "Total: " << nLines << " counted.\n";
  return 0;
}

在cygwin64中编译测试：

$ g++ -std=c++11 -o line-count line-count.cc

$ ./line-count line-count.cc - line-count.cc -
File: 'line-count.cc', 32 counted.
1
2
3
File: '-', 3 counted.
File: 'line-count.cc', 32 counted.
1
2
3
File: '-', 3 counted.
Total: 70 counted.

$

所以，std::cin.clear() 的诀窍确实是重置输入流中的 EOF 标志，并可以再次从 /dev/stdin 读取。

在 OP 的情况下，std::cin.clear() 之后的 std::cin.ignore() 恕我直言是错误的。它会丢弃重新启用的标准输入的第一个字符，这会导致以下处理错误（不计算第一个字符）。

戴维斯（再次）提供了一个简短但显而易见的解释，我用我的话给出了：

使用 CtrlD，标准输入会在下一次尝试读取时收到 EOF；并将其存储在其内部标志中。不过，可以重置标志。如果没有任何进一步的输入，则下一次读取尝试将失败，否则，可以继续输入。

可能，值得强调std::ios 中的内部EOF 标志。如果没有std::cin.clear()，即使有更多输入可用，读取尝试也会失败。只要内部std::stream 标志没有导致良好的状态，即使可能成功，也不会在较低级别执行读取尝试。