c++ - 尝试在 std::cout 和 std::ifstream 之间切换

Question

我想从命令行确定的 std::cin 或 std::ifstream 读取输入。该命令看起来像 ./run 1 或 ./run 2。眼下，我必须写两个基于读取模式的几乎相似的函数。

void read1()
{
  int a, b;
  while (std::cin >> a >> b) {
    // do something
  }
}

或

void read2()
{
  int a, b;
  std::ifstream fin("file.txt");
  while (fin >> a >> b) {
    // do something
  }
}

对于大循环来说，很难维护这两个函数，因为循环部分是通用的，唯一的区别是输入源。

如何整合这两个功能？

Answer 1

std::cin 和 std::ifstream 都是 std::istream，所以你可以通过使用对 std::istream 的引用进行操作的函数。这适用于 std::cin、std::ifstream 实例和任何其他 std::istreams:

void read(std::istream& input)
{
  while (input >> a >> b) { .... }
}

然后切换到来电端。

if (something)
{
  read(std::cin);
} else
{
  isfream input(....);
  read(input);
}