如何在 C++ 中覆盖 cout？

Question

我有一个需求，我还需要使用printf 和cout 将数据显示到console and file 中。 对于printf我已经做到了，但是对于cout我很挣扎，怎么办？

   #ifdef _MSC_VER
     #define GWEN_FNULL "NUL"
     #define va_copy(d,s) ((d) = (s))
         #else
         #define GWEN_FNULL "/dev/null"
        #endif
        #include <iostream>
        #include <fstream>

        using namespace std;
        void printf (FILE *  outfile, const char * format, ...) 
        {

            va_list ap1, ap2;
            int i = 5;
            va_start(ap1, format);
            va_copy(ap2, ap1);
            vprintf(format, ap1);
            vfprintf(outfile, format, ap2);
            va_end(ap2);
            va_end(ap1);
        }
    /*    void COUT(const char* fmt, ...)
        {
            ofstream out("output-file.txt");
            std::cout << "Cout to file";
            out << "Cout to file";
        }*/
        int main (int argc, char *argv[]) {

            FILE *outfile;
            char *mode = "a+";
            char outputFilename[] = "PRINT.log";
            outfile = fopen(outputFilename, mode);

            char bigfoot[] = "Hello 

World!\n";
        int howbad = 10;

        printf(outfile, "\n--------\n");
        //myout();

        /* then i realized that i can't send the arguments to fn:PRINTs */
        printf(outfile, "%s %i",bigfoot, howbad); /* error here! I can't send bigfoot and howbad*/

        system("pause");
        return 0;
    }

我已经在COUT(caps，上面代码的注释部分) 中完成了它。但是我想使用普通的std::cout，那么我该如何覆盖它。它应该适用于sting and variables 之类的

int i = 5;
cout << "Hello world" << i <<endl;

或者无论如何要捕获stdout数据，以便它们也可以轻松写入file and console。

Answer 1

如果你有另一个流缓冲区，你可以替换std::cout的：

std::cout.rdbuf(some_other_rdbuf);

见http://en.cppreference.com/w/cpp/io/basic_ios/rdbuf。

Answer 2

您可以交换底层缓冲区。这是通过 RAII 完成的。

#include <streambuf>

class buffer_restore
{
    std::ostream&   os;
    std::streambuf* buf;
public:
    buffer_restore(std::ostream& os) : os(os), buf(os.rdbuf())
    { }

    ~buffer_restore()
    {
        os.rdbuf(buf);
    }
};

int main()
{
    buffer_restore b(std::cout);
    std::ofstream file("file.txt");

    std::cout.rdbuf(file.rdbuf());
    // ...
}

Answer 3

覆盖std::cout 的行为是一个非常糟糕的主意，因为其他开发人员将很难理解std::cout 的使用行为与往常不同。

用一个简单的类来明确你的意图

#include <fstream>
#include <iostream>

class DualStream
{
   std::ofstream file_stream;
   bool valid_state;
   public:
      DualStream(const char* filename) // the ofstream needs a path
      :
         file_stream(filename),  // open the file stream
         valid_state(file_stream) // set the state of the DualStream according to the state of the ofstream
      {
      }
      explicit operator bool() const
      {
         return valid_state;
      }
      template <typename T>
      DualStream& operator<<(T&& t) // provide a generic operator<<
      {
         if ( !valid_state ) // if it previously was in a bad state, don't try anything
         {
            return *this;
         }
         if ( !(std::cout << t) ) // to console!
         {
            valid_state = false;
            return *this;
         }
         if ( !(file_stream << t) ) // to file!
         {
            valid_state = false;
            return *this;
         }
         return *this;
      }
};
// let's test it:
int main()
{
   DualStream ds("testfile");
   if ( (ds << 1 << "\n" << 2 << "\n") )
   {
      std::cerr << "all went fine\n";
   }
   else
   {
      std::cerr << "bad bad stream\n";
   }
}

这提供了一个干净的界面，并为控制台和文件输出相同的内容。 您可能需要添加刷新方法或以追加模式打开文件。

Answer 4

我假设您有一些使用 std::cout 和 printf 的代码，您无法修改，否则解决问题的最简单方法是从 cout 写入不同的流并使用 fprintf 而不是或与printf结合使用。

通过采用这种方法，您可以定义一个新的流类，该流类实际上既可以写入标准输出，也可以写入给定文件，还可以定义一个函数，该函数结合了对printf 和fprintf 的调用。

然而，更简单的方法是使用最初来自 UNIX 的 tee 程序，它将输入复制到输出和给定文件。有了它，您可以简单地以这种方式调用您的程序：

your_program | tee your_log_file

对this question 的回答导致了一些可用于Windows 的替代实现。就我个人而言，我总是在我的 PC 上安装 cygwin 以提供 UNIX/Linux 实用程序。

Answer 5

如果我猜对了，您希望将输出的所有内容也记录到文件中。

你想要的是observer pattern。

将代码中的所有直接登录替换为对新中继的调用。 日志中继将您的消息发送给观察者。 您的一位观察者将消息记录到屏幕上。 另一个记录到文件中。 尽可能避免让你的中继成为单例。

此建议仅在您可以编辑所有源文件时才有效。

Answer 6

std::cout 写入stdout 文件，您可以在 Linux 和 Windows 上执行以下操作

#include <stdio.h>
#include <iostream>



int main()
{
    freopen("test.txt", "w", stdout);
    std::cout << "Hello strange stdout\n";
}

将其改回使用取自here的以下内容

#include <stdio.h>
#include <stdlib.h>

void main(void)
{
   FILE *stream ;
   if((stream = freopen("file.txt", "w", stdout)) == NULL)
      exit(-1);

   printf("this is stdout output\n");

   stream = freopen("CON", "w", stdout);

   printf("And now back to the console once again\n");
}

注意：后者仅适用于窗口

Answer 7

cout 通常实现为对象实例，因此您不能像重载/覆盖函数或类那样覆盖它。

您最好不要与之抗争 - 是的，您可以构建 my_cout 和 #define cout my_cout，但这会使您的代码变得迟钝。

为了便于阅读，我将保留 cout 原样。这是一个标准，每个人都知道它能做什么，不能做什么。

Answer 8

尝试使用宏 - 类似这样（您需要添加包含）：

#define MY_COUT(theos,printThis) { cout << printThis ; theos <<  printThis; }

void test()
{
     ofstream myos;
    myos.open("testfile", ios::trunc|ios::out);
    int i = 7;
    MY_COUT(myos, "try this numbers" << i << i + 1 << endl);
    myos.close()
}

Answer 9

为此已经有一个 Boost 类：tee