src/console/console.cpp

   1 /*
   2 Copyright © 2007-2008 Kővágó Zoltán <DirtY.iCE.hu@gmail.com>
   3
   4 This program is free software; you can redistribute it and/or
   5 modify it under the terms of the GNU General Public License
   6 as published by the Free Software Foundation; either version 2
   7 of the License, or (at your option) any later version.
   8
   9 This program is distributed in the hope that it will be useful,
  10 but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  12
  13 See the GNU General Public License for more details.
  14
  15 You should have received a copy of the GNU General Public License
  16 along with this program; if not, write to the Free Software
  17 Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  18
  19 */
  20
  21 /** @file console/console.cpp
  22  * Console routines
  23  * Contains methods to write to console
  24  */
  25
  26 #include <cstdio>
  27 #include <iomanip>
  28 #include <sstream>
  29
  30 #include "console.hpp"
  31
  32 /**
  33  * Initializes DebugStream and set the message to
  34  * @param message Set message to
  35  */
  36 DebugStream::DebugStream(const std::string& message) : message(message), file(""), func(""), line(0), putline(true)
  37 {
  38         setp(buf, buf + sizeof(buf));
  39 }
  40
  41 /// If the buffer is out of space
  42 int DebugStream::overflow(int c)
  43 {
  44         size_t size = pptr() - pbase();
  45         if (!size) return 0;
  46
  47         for (char * pos = pbase(); pos < pptr(); pos++)
  48         {
  49                 if (putline)
  50                         PutLineBeg();
  51
  52                 putline = (*pos == '\n');
  53
  54                 putchar(*pos);
  55         }
  56
  57         if (c != traits_type::eof())
  58         {
  59                 if (putline)
  60                         PutLineBeg();
  61
  62                 putline = (char(c) == '\n');
  63
  64                 putchar(char(c));
  65         }
  66
  67         setp(buf, buf + sizeof(buf));
  68         return 0;
  69 }
  70
  71 /// Flush the cache
  72 int DebugStream::sync()
  73 {
  74         return overflow(traits_type::eof());
  75 }
  76
  77 /// Writes out the beginning of the line (message, func, file, line)
  78 void DebugStream::PutLineBeg()
  79 {
  80         std::stringstream str;
  81
  82 #ifdef DEBUG
  83         if (file.size() > file_maxw) file_maxw = file.size();
  84         str << line;
  85         if (str.str().size() > line_maxw) line_maxw = str.str().size();
  86         str.str("");
  87         if (func.size() > func_maxw) func_maxw = func.size();
  88
  89         str << "[" << message << "] { ";
  90
  91         if (show_func)
  92                 str << std::setw(func_maxw) << func << " (";
  93
  94         str << std::setw(file_maxw) << file << ":" << std::setw(line_maxw) << line;
  95
  96         if (show_func)
  97                 str << ")";
  98         str << " } : ";
  99
 100 #else
 101         str << "[" << message << "] ";
 102 #endif
 103
 104         fputs(str.str().c_str(), stdout);
 105 }
 106
 107 unsigned short DebugStream::file_maxw = 16;
 108 unsigned short DebugStream::line_maxw = 3;
 109 unsigned short DebugStream::func_maxw = 74;
 110 bool DebugStream::show_func = false;
 111
 112 #ifdef DEBUG
 113
 114 #include <boost/lexical_cast.hpp>
 115 #include <boost/tokenizer.hpp>
 116
 117 /* static */ void DebugLevelManager::ParseOpts(const std::string& opts)
 118 {
 119         boost::char_separator<char> sep(",");
 120         boost::tokenizer<boost::char_separator<char> > tokens(opts, sep);
 121
 122         for (boost::tokenizer<boost::char_separator<char> >::iterator iter = tokens.begin(); iter != tokens.end(); iter++)
 123         {
 124                 size_t pos;
 125                 if ((pos = iter->find_first_of("=")) == std::string::npos)
 126                 {
 127                         try
 128                         {
 129                                 global_level = boost::lexical_cast<unsigned short>(*iter);
 130                         }
 131                         catch (const boost::bad_lexical_cast& e)
 132                         {
 133                                 error << "Failed to set global level to '" << *iter << "'!" << std::endl
 134                                           << "The error was: " << e.what() << " Parameter ignored." << std::endl;
 135                         }
 136                 }
 137                 else
 138                 {
 139                         try
 140                         {
 141                                 level_map[iter->substr(0, pos)] = boost::lexical_cast<unsigned short>(iter->substr(pos + 1, iter->size() - pos - 1));
 142                         }
 143                         catch (const boost::bad_lexical_cast& e)
 144                         {
 145                                 error << "Failed to set '" << iter->substr(0, pos) << "' group to '" << iter->substr(pos + 1, iter->size() - pos - 1) << "'!" << std::endl
 146                                           << "The error was: " << e.what() << " Ignored." << std::endl;
 147                         }
 148                 }
 149         }
 150 }
 151
 152 /* static */ bool DebugLevelManager::IsToPrint(const std::string& module, const unsigned char level)
 153 {
 154         if (level_map.count(module) == 0)
 155                 return global_level >= level;
 156         else
 157                 return level_map[module] >= level;
 158 }
 159
 160 unsigned char DebugLevelManager::global_level = 0;
 161 std::map<std::string, unsigned char> DebugLevelManager::level_map;
 162
 163 DebugOut debug_out  (" Debug ");
 164 #endif //ifdef DEBUG
 165 DebugOut warning_out("WARNING");
 166 DebugOut error_out  ("!ERROR!");