将 cout 分配给变量名

Question

在 ANSI C++ 中，如何将 cout 流分配给变量名？ 我想要做的是，如果用户指定了输出文件名，我将输出发送到那里，否则，将其发送到屏幕。 所以类似：

ofstream outFile;
if (outFileRequested) 
    outFile.open("foo.txt", ios::out);
else
    outFile = cout;  // Will not compile because outFile does not have an 
                     // assignment operator

outFile << "whatever" << endl;

我也尝试将其作为宏函数执行：

#define OUTPUT outFileRequested?outFile:cout

OUTPUT << "whatever" << endl;

但这也给了我一个编译器错误。

我想我可以为每个输出使用 IF-THEN 块，但如果可以的话我想避免这种情况。 有任何想法吗？

Answer 1

使用参考。 请注意，引用的类型必须是 std::ostream，而不是 std::ofstream，因为 std::cout 是一个 std::ostream，因此您必须使用最小公分母。

std::ofstream realOutFile;

if(outFileRequested)
    realOutFile.open("foo.txt", std::ios::out);

std::ostream & outFile = (outFileRequested ? realOutFile : std::cout);

Answer 2

我假设你的程序的行为类似于标准的unix工具，当没有给定一个文件时，它将写入标准输出，当给定一个文件时，它将写入该文件。 您可以重定向cout以写入另一个流缓冲区。 只要您的重定向有效，写入 cout 的所有内容都会透明地写入您指定的目的地。 一旦重定向对象超出范围，原始流就会被放入，输出将再次写入屏幕：

struct CoutRedirect { 
    std::streambuf * old; 
    CoutRedirect():old(0) {
        // empty
    }

    ~CoutRedirect() {
        if(old != 0) {
            std::cout.rdbuf(old);
        }
    }

    void redirect(std::streambuf * to) {
        old = std::cout.rdbuf(to);
    }
}

int main() {
    std::filebuf file;
    CoutRedirect pipe;
    if(outFileRequested) {
        file.open("foo.txt", std::ios_base::out);
        pipe.redirect(&file);
    }
}

现在，只要管道在 main 中处于活动状态，cout 就会重定向到文件。 您可以通过使其不可复制来使其更加“生产就绪”，因为它还没有准备好被复制：如果副本超出范围，它将已经恢复原始流。

Answer 3

您可以在这里找到有关如何执行此操作的非常详细的说明：
http://groups.google.com/group/ comp.lang.c++/msg/1d941c0f26ea0d81?pli=1

希望有人能更清楚地写出这一点，以便堆栈溢出能够得分......

Answer 4

遵循 Adam Rosenfield 的轨迹，但修复了使用三元和逗号运算符的引用初始化问题：（

bool outFileRequested = false;

std::ofstream realOutFile;
std::ostream & outFile = outFileRequested
    ? realOutFile.open("foo.txt", std::ios::out), realOutFile
    : std::cout;

outFile << "some witty remark";

在 VS 中测试）

Answer 5

我认为 Adam 走在正确的轨道上，但我不认为你可以分配引用 - 你需要使用指针：然后

std::ofstream realOutFile;
std::ostream * poutFile;

if(outFileRequested)
{
    realOutFile.open("foo.txt", std::ios::out);
    poutFile = &realOutFile;
}
else
    poutFile = &std::cout;

你可以将引用定义为指针的值，但它不会是全局的

std::ostream & outFile = *poutFile;

Answer 6

我不确定您是否可以将 cout 分配给 ofstream 类型的变量。 cout 是 ostream 类型的对象（而 cin 是 istream 类型），我不确定其中一个是从另一个继承的。 因此，也许检查文件是否已给定/存在并创建适当的流类型将是更好的方法。

Answer 7

这花了大约两个小时才得到。 基本上，我有一个运行测试套件的外部类。 我发送委托来运行测试，因此为了访问输出，我需要发送输出流。 我想我可以为每个测试做不同的流。 无论如何，我想传入ofstream以供稍后使用。

// Main code to create Test Suite Object
ofstream debugFile("debug.txt");
TestSuiteObject* myTestSuite = new TestSuiteObject(&debugFile);

// Test Suite Object
class TestSuiteObject: public Test::Suite
{
public:
 TestSuiteObject(std::ofstream* debug) : m_debug(*debug)
 {
  m_debug << "some witty remark" << std::endl;
  TEST_ADD(TestSuiteObject::test1);
  TEST_ADD(TestSuiteObject::test2);
  TEST_ADD(TestSuiteObject::test3);

 }

 void test1();
 void test2();
 void test3();

private:
 std::ofstream& m_debug;
};