Console.hh

Go to the documentation of this file.

/*
 * Copyright 2011 Nate Koenig
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
*/

#ifndef _SDF_CONSOLE_HH_
#define _SDF_CONSOLE_HH_

#include <iostream>
#include <fstream>
#include <string>

#include <boost/shared_ptr.hpp>

#include "sdf/system_util.hh"

namespace sdf
{

  #define sdfdbg (sdf::Console::Instance()->Log("Dbg", \
    __FILE__, __LINE__))

  #define sdfmsg (sdf::Console::Instance()->ColorMsg("Msg", \
    __FILE__, __LINE__, 32))

  #define sdfwarn (sdf::Console::Instance()->ColorMsg("Warning", \
        __FILE__, __LINE__, 33))

  #define sdferr (sdf::Console::Instance()->ColorMsg("Error", \
        __FILE__, __LINE__, 31))

  class SDFORMAT_VISIBLE Console
  {
    public: class SDFORMAT_VISIBLE ConsoleStream
    {
      public: ConsoleStream(std::ostream *_stream) :
              stream(_stream) {}

      public: template <class T>
        ConsoleStream &operator<<(const T &_rhs)
        {
          if (this->stream)
            *this->stream << _rhs;

          if (Console::Instance()->logFileStream.is_open())
          {
            Console::Instance()->logFileStream << _rhs;
            Console::Instance()->logFileStream.flush();
          }

          return *this;
        }

      public: void Prefix(const std::string &_lbl,
                          const std::string &_file,
                          unsigned int _line, int _color)
        {
          int index = _file.find_last_of("/") + 1;

          if (this->stream)
          {
            *this->stream << "\033[1;" << _color << "m" << _lbl << " [" <<
              _file.substr(index , _file.size() - index)<< ":" << _line <<
              "]\033[0m ";
          }

          if (Console::Instance()->logFileStream.is_open())
          {
            Console::Instance()->logFileStream << _lbl << " [" <<
              _file.substr(index , _file.size() - index)<< ":" << _line << "] ";
          }
        }

      private: std::ostream *stream;
    };

    private: Console();

    public: virtual ~Console();

    public: static boost::shared_ptr<Console> Instance();

    public: void SetQuiet(bool _q);

    public: ConsoleStream &ColorMsg(const std::string &lbl,
                                    const std::string &file,
                                    unsigned int line, int color);

    public: ConsoleStream &Log(const std::string &lbl,
                               const std::string &file,
                               unsigned int line);

    private: ConsoleStream msgStream;

    private: ConsoleStream logStream;

    private: std::ofstream logFileStream;

    private: static boost::shared_ptr<Console> myself;
  };
}
#endif