src/VerboseInfo.h

   1 #ifndef AVR_VERBOSEINFO_H
   2 #define AVR_VERBOSEINFO_H
   3
   4 #include <iostream>
   5
   6 namespace avr {
   7
   8         enum Verbosity { NONE = 0, WARN = 1, INFO = 2, DBG = 3 };
   9
  10         class VerboseInfo;
  11         class VerboseInfo {
  12                 public:
  13                         VerboseInfo() : verbosity(0) {}
  14
  15                 public:
  16                         void setVerbosity(Verbosity verbosity);
  17                         void setVerbosity(int verbosity);
  18                         void increaseVerbosity();
  19
  20                 public:
  21                         bool shouldOutput(Verbosity vb) const {
  22                                 return ((int)vb > verbosity);
  23                         }
  24
  25                 public:
  26                         class Printer {
  27                                 public:
  28                                         Printer(const VerboseInfo & vi, Verbosity vb,
  29                                                         std::ostream & ostr)
  30                                                 : vi(vi), vb(vb), ostr(ostr) {}
  31
  32                                 public:
  33                                         template <class T>
  34                                         Printer & operator <<(const T & c) {
  35                                                 if( vi.shouldOutput(vb) )
  36                                                         ostr << c;
  37                                                 return *this;
  38                                         }
  39
  40                                     Printer & operator<<(std::ostream & (*pf)(std::ostream &)) {
  41                                                 if( vi.shouldOutput(vb) )
  42                                                         ostr << pf;
  43                                                 return *this;
  44                                         }
  45
  46                                 private:
  47                                         const VerboseInfo & vi;
  48                                         Verbosity vb;
  49                                         std::ostream & ostr;
  50                         };
  51
  52                         Printer operator ()(Verbosity vb,
  53                                         std::ostream & ostr = std::cout) const {
  54                                 return Printer(*this, vb, ostr);
  55                         }
  56
  57                 private:
  58                         int verbosity;
  59         };
  60
  61         inline void VerboseInfo::setVerbosity(Verbosity verbosity) {
  62                 this->verbosity = (int)verbosity;
  63         }
  64
  65         inline void VerboseInfo::setVerbosity(int verbosity) {
  66                 this->verbosity = verbosity;
  67         }
  68
  69         inline void VerboseInfo::increaseVerbosity() {
  70                 this->verbosity++;
  71         }
  72
  73         extern VerboseInfo info;
  74 }
  75
  76 #endif /* AVR_VERBOSEINFO_H */