定制cout

时间:2009-03-23 16:22:16

标签: c++ stl iostream

如何从cout派生一个类,以便例如写入

new_cout << "message";

等同于

cout << __FUNCTION__ << "message" << "end of message" << endl;

6 个答案:

答案 0 :(得分:29)

class Log
{
public:
    Log(const std::string &funcName)
    {
        std::cout << funcName << ": ";
    }

    template <class T>
    Log &operator<<(const T &v)
    {
        std::cout << v;
        return *this;
    }

    ~Log()
    {
        std::cout << " [end of message]" << std::endl;
    }
};

#define MAGIC_LOG Log(__FUNCTION__)

因此:

MAGIC_LOG << "here's a message";
MAGIC_LOG << "here's one with a number: " << 5;

答案 1 :(得分:2)

#define debug_print(message) (std::cout << __FUNCTION__ << (message) << std::endl)

这样做的好处是,您可以在完成后立即禁用所有调试消息

#define debug_print(message) ()

答案 2 :(得分:1)

根据Mykola的回复,我的代码中有以下实现。 用法是

         LOG_DEBUG("print 3 " << 3); 

打印

         DEBUG (f.cpp, 101): print 3 3 

您可以修改它以使用 FUNCTION 代替 LINE FILE

/// Implements a simple logging facility. 
class Logger
{
        std::ostringstream os_;
        static Logger* instance_;
        Logger();
public:
        static Logger* getLogger();
        bool isDebugEnabled() const;
        void log(LogLevelEnum l, std::ostringstream& os, const char* filename, int lineno) const;
        std::ostringstream& getStream()
        { return os_; }
};

void Logger::log(LogLevelEnum l, std::ostringstream& os, const char* filename, int lineno) const
{
        std::cout << logLevelEnumToString(l) << "\t(" << fileName << ": " << lineno << ")\t- " << os.str();
        os.str("");
}

#define LOG_common(level, cptext) do {\
        utility::Logger::getLogger()->getStream() << cptext; \
        utility::Logger::getLogger()->log(utility::level, utility::Logger::getLogger()->getStream(), __FILE__, __LINE__);  \
} while(0); 

enum LogLevelEnum {
        DEBUG_LOG_LEVEL,
        INFO_LOG_LEVEL,
        WARN_LOG_LEVEL,
        ERROR_LOG_LEVEL,
        NOTICE_LOG_LEVEL,
        FATAL_LOG_LEVEL
};

#define LOG_DEBUG(cptext)    LOG_common(DEBUG_LOG_LEVEL, cptext)
#define LOG_INFO(cptext)     LOG_common(INFO_LOG_LEVEL , cptext)
#define LOG_WARN(cptext)     LOG_common(WARN_LOG_LEVEL , cptext)
#define LOG_ERROR(cptext)    LOG_common(ERROR_LOG_LEVEL, cptext)
#define LOG_NOTICE(cptext)   LOG_common(NOTICE_LOG_LEVEL, cptext)
#define LOG_FATAL(cptext)    LOG_common(FATAL_LOG_LEVEL, cptext)

const char* logLevelEnumToString(LogLevelEnum m)
{
        switch(m)
        {
                case DEBUG_LOG_LEVEL:
                        return "DEBUG";
                case INFO_LOG_LEVEL:
                        return "INFO";
                case WARN_LOG_LEVEL:
                        return "WARN";
                case NOTICE_LOG_LEVEL:
                        return "NOTICE";
                case ERROR_LOG_LEVEL:
                        return "ERROR";
                case FATAL_LOG_LEVEL:
                        return "FATAL";
                default:
                        CP_MSG_ASSERT(false, CP_TEXT("invalid value of LogLevelEnum"));
                        return 0;
        }
}

答案 3 :(得分:0)

您必须覆盖operator&lt;&lt;(),但您甚至不必继承std :: cout的子类。您也可以创建一个新对象或使用这样的现有对象。

答案 4 :(得分:0)

出于记录目的,我使用

之类的东西
#define LOG(x) \
  cout << __FUNCTION__ << x << endl

// ...
LOG("My message with number " << number << " and some more");

你的方法的问题是(正如Mykola Golybyew所解释的) FUNCTION 在编译时被处理,因此总是使用非预处理器解决方案打印相同的名称。

如果仅用于在邮件中添加endl,您可以尝试以下内容:

class MyLine {
public:
  bool written;
  std::ostream& stream;
  MyLine(const MyLine& _line) : stream(_line.stream), written(false) { }
  MyLine(std::ostream& _stream) : stream(_stream), written(false) { }
  ~MyLine() { if (!written) stream << "End of Message" << std::endl; }
};

template <class T> MyLine operator<<(MyLine& line, const T& _val) {
  line.stream << _val;
  line.written = true;
  return line;
}

class MyStream {
public:
  std::ostream& parentStream;
  MyStream(std::ostream& _parentStream) : parentStream(_parentStream) { }
  MyLine getLine() { return MyLine(parentStream); }
};

template <class T> MyLine operator<<(MyStream& stream, const T& _val) {
  return (stream.getLine() << _val);
}

int main()
{
      MyStream stream(std::cout);
      stream << "Hello " << 13 << " some more data";
      stream << "This is in the next line " << " 1 ";
    return 0;
}

请注意,重要的是不要从操作员函数返回引用。由于MyLine应仅作为临时存在(因为其析构函数触发endl的写入),因此第一个对象(由getLine()中的MyStream函数返回)将在第二个operator<<被调用之前被破坏。因此,MyLine对象将复制到每个operator<<中,从而创建一个新对象。最后一个对象被破坏而不被写入并在其析构函数中写入消息的末尾。

只需在调试器中试一试,了解最新情况......

答案 5 :(得分:-1)

你也可以override the operator。它允许你调用另一个函数或前缀/后缀任何你想要留下输出缓冲区的东西:在你的情况下,你会输出一个特定的字符串。