include/Log.hpp

   1 #ifndef SCX_LOG_HPP
   2 #define SCX_LOG_HPP
   3
   4 #include <iostream>
   5 #include <fstream>
   6 #include <sstream>
   7 #include <string>
   8 #include "Mutex.hpp"
   9 #include "ColStr.hpp"
  10
  11 namespace scx
  12 {
  13 class Log
  14 {
  15 public:
  16     Log()
  17     {
  18         EnableStdOut(true);
  19         EnableFileOut(false);
  20     }
  21
  22     ~Log()
  23     {
  24         Close();
  25     }
  26
  27     bool Open(const std::string& file = "log.txt")
  28     {
  29         m_FileStream.open(file.c_str(), std::ios::out);
  30         if (m_FileStream.is_open())
  31             return true;
  32         else
  33             return false;
  34     }
  35
  36     void Close()
  37     {
  38         if (m_FileStream.is_open())
  39             m_FileStream.close();
  40     }
  41
  42     stringstream& Info()
  43     {
  44         return m_StrStream;
  45     }
  46
  47     stringstream& Warning()
  48     {
  49         return m_StrStream;
  50     }
  51
  52     stringstream& Error()
  53     {
  54         return m_StrStream;
  55     }
  56
  57     template <typename T>
  58     ostream& operator<<(T& val)
  59     {
  60         m_FileStream << val;
  61         cout << val;
  62         return m_FileStream;
  63     }
  64
  65     ostream& operator<<(ostream& (*fn)(ostream&))
  66     {
  67         m_FileStream << fn;
  68         cout << fn;
  69         return m_FileStream;
  70     }
  71
  72     ostream& operator<<(ios& (*fn)(ios&))
  73     {
  74         return m_FileStream;
  75     }
  76
  77 private:
  78     Mutex m_Mutex;
  79     bool m_EnableStdOut;
  80     bool m_EnableFileOut;
  81     std::ofstream m_FileStream;
  82     std::stringstream m_StrStream;
  83 };
  84
  85 class ColorStream
  86 {
  87 public:
  88     template <typename T>
  89     ostream& operator<<(T&)
  90     {
  91
  92     }
  93
  94 private:
  95
  96 }
  97 }
  98
  99 #endif