src/log.h

   1 /* Copyright (C) 2007-2011 Vincent Ollivier
   2  *
   3  * Purple Haze is free software: you can redistribute it and/or modify
   4  * it under the terms of the GNU General Public License as published by
   5  * the Free Software Foundation, either version 3 of the License, or
   6  * (at your option) any later version.
   7  *
   8  * Purple Haze is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  11  * GNU General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License
  14  * along with this program. If not, see <http://www.gnu.org/licenses/>.
  15  */
  16
  17 #ifndef LOG_H
  18 #define LOG_H
  19
  20 #include <iostream>
  21 #include <fstream>
  22
  23 class Log : public std::ostream
  24 {
  25     private:
  26         template <class T>
  27         struct LogStream {
  28             T stream;
  29             bool state;
  30
  31             LogStream() :
  32                 stream(),
  33                 state(false)
  34                 {}
  35             LogStream(std::streambuf *sb) :
  36                 stream(sb),
  37                 state(true)
  38                 {}
  39             LogStream(std::string filename) :
  40                 stream(filename, std::ios::app),
  41                 state(true)
  42                 {}
  43         };
  44         LogStream<std::ostream> cout;
  45         LogStream<std::ofstream> file;
  46
  47     public:
  48         enum Stream { COUT, FILE, BOTH };
  49         enum LogDirection { IN, OUT, DEBUG };
  50
  51         Log() :
  52             std::ostream(std::cout.rdbuf()),
  53             cout(std::cout.rdbuf()),
  54             file()
  55             {}
  56
  57         Log(std::string filename) :
  58             std::ostream(std::cout.rdbuf()),
  59             cout(std::cout.rdbuf()),
  60             file(filename)
  61             {}
  62
  63         void open(std::string filename) {
  64             file.stream.open(filename, std::ios::app);
  65             file.state = true;
  66         };
  67
  68         template <typename T>
  69         Log& operator<<(const T &val) {
  70             if (cout.state) {
  71                 cout.stream << val;
  72             }
  73             if (file.state && file.stream.is_open()) {
  74                 file.stream << val;
  75             }
  76             return *this;
  77         }
  78
  79         Log& operator<<(std::ostream& (*manip)(std::ostream&)) {
  80             if (cout.state) {
  81                 cout.stream << manip;
  82             }
  83             if (file.state && file.stream.is_open()) {
  84                 file.stream << manip;
  85             }
  86             return *this;
  87         }
  88
  89         Log& operator<<(const LogDirection &dir);
  90         Log& to(Stream s);
  91 };
  92 #endif /* !LOG_H*/