如何将“虚拟”值替换为std :: cin?

时间:2016-06-01 16:41:18

标签: c++ iostream cin

鉴于以下简单程序:

int main() {
    int v;
    std::vector<int> values;
    while(std::cin >> v) {
        values.emplace_back(v);
    }
    std::cout << "The Sum is " << std::accumulate(values.begin(), values.end(), 0) << std::endl;
    return 0;
}

我想在此代码执行之前以编程方式填写std::cin,方式与以下内容类似:

int main() {
    int v;
    std::vector<int> values;
    for(int i = 0; i < 10; i++) {
        std::cin << i << " "; //Doesn't compile, obviously
    }
    /*
    The Rest of the Code.
    */
    return 0;
}

但当然,该代码不起作用。我能做些什么可以让我将数据“管道”到std::cin ,而无需手动管道来自不同程序或命令行shell 的数据,如echo "1 2 3 4 5 6 7 8 9 10" | myprogram.exe

1 个答案:

答案 0 :(得分:3)

您可以操纵与rdbuf相关联的std::cin将其关闭。

#include <iostream>
#include <sstream>
#include <vector>
#include <algorithm>

int main() {
   int v;
   std::vector<int> values;

   // Create a istringstream using a hard coded string.
   std::string data = "10 15 20";
   std::istringstream str(data);

   // Use the rdbuf of the istringstream as the rdbuf of std::cin.
   auto old = std::cin.rdbuf(str.rdbuf());

   while(std::cin >> v) {
      values.emplace_back(v);
   }
   std::cout << "The Sum is " << std::accumulate(values.begin(), values.end(), 0) << std::endl;

   // Restore the rdbuf of std::cin.
   std::cin.rdbuf(old);

   return 0;
}

http://ideone.com/ZF02op处查看它。