AtomicGameEngine/Source/ThirdParty/crunch/crnlib/crn_console.cpp

// File: crn_console.cpp
// See Copyright Notice and license at the end of inc/crnlib.h
#include "crn_core.h"
#include "crn_console.h"
#include "crn_data_stream.h"
#include "crn_threading.h"

namespace crnlib
{
   eConsoleMessageType                       console::m_default_category = cInfoConsoleMessage;
   crnlib::vector<console::console_func>      console::m_output_funcs;
   bool                                      console::m_crlf = true;
   bool                                      console::m_prefixes = true;
   bool                                      console::m_output_disabled;
   data_stream*                              console::m_pLog_stream;
   mutex*                                    console::m_pMutex;
   uint                                      console::m_num_messages[cCMTTotal];
   bool                                      console::m_at_beginning_of_line = true;

   const uint cConsoleBufSize = 4096;

   void console::init()
   {
      if (!m_pMutex)
      {
         m_pMutex = crnlib_new<mutex>();
      }
   }

   void console::deinit()
   {
      if (m_pMutex)
      {
         crnlib_delete(m_pMutex);
         m_pMutex = NULL;
      }
   }

   void console::disable_crlf()
   {
      init();

      m_crlf = false;
   }

   void console::enable_crlf()
   {
      init();

      m_crlf = true;
   }

   void console::vprintf(eConsoleMessageType type, const char* p, va_list args)
   {
      init();

      scoped_mutex lock(*m_pMutex);

      m_num_messages[type]++;

      char buf[cConsoleBufSize];
      vsprintf_s(buf, cConsoleBufSize, p, args);

      bool handled = false;

      if (m_output_funcs.size())
      {
         for (uint i = 0; i < m_output_funcs.size(); i++)
            if (m_output_funcs[i].m_func(type, buf, m_output_funcs[i].m_pData))
               handled = true;
      }

      const char* pPrefix = NULL;
      if ((m_prefixes) && (m_at_beginning_of_line))
      {
         switch (type)
         {
            case cDebugConsoleMessage:    pPrefix = "Debug: ";   break;
            case cWarningConsoleMessage:  pPrefix = "Warning: "; break;
            case cErrorConsoleMessage:    pPrefix = "Error: ";   break;
            default: break;
         }
      }

      if ((!m_output_disabled) && (!handled))
      {
         if (pPrefix)
            ::printf("%s", pPrefix);
         ::printf(m_crlf ? "%s\n" : "%s", buf);
      }

      uint n = strlen(buf);
      m_at_beginning_of_line = (m_crlf) || ((n) && (buf[n - 1] == '\n'));

      if ((type != cProgressConsoleMessage) && (m_pLog_stream))
      {
         // Yes this is bad.
         dynamic_string tmp_buf(buf);

         tmp_buf.translate_lf_to_crlf();

         m_pLog_stream->printf(m_crlf ? "%s\r\n" : "%s", tmp_buf.get_ptr());
         m_pLog_stream->flush();
      }
   }

   void console::printf(eConsoleMessageType type, const char* p, ...)
   {
      va_list args;
      va_start(args, p);
      vprintf(type, p, args);
      va_end(args);
   }

   void console::printf(const char* p, ...)
   {
      va_list args;
      va_start(args, p);
      vprintf(m_default_category, p, args);
      va_end(args);
   }

   void console::set_default_category(eConsoleMessageType category)
   {
      init();

      m_default_category = category;
   }

   eConsoleMessageType console::get_default_category()
   {
      init();

      return m_default_category;
   }

   void console::add_console_output_func(console_output_func pFunc, void* pData)
   {
      init();

      scoped_mutex lock(*m_pMutex);

      m_output_funcs.push_back(console_func(pFunc, pData));
   }

   void console::remove_console_output_func(console_output_func pFunc)
   {
      init();

      scoped_mutex lock(*m_pMutex);

      for (int i = m_output_funcs.size() - 1; i >= 0; i--)
      {
         if (m_output_funcs[i].m_func == pFunc)
         {
            m_output_funcs.erase(m_output_funcs.begin() + i);
         }
      }

      if (!m_output_funcs.size())
      {
         m_output_funcs.clear();
      }
   }

   void console::progress(const char* p, ...)
   {
      va_list args;
      va_start(args, p);
      vprintf(cProgressConsoleMessage, p, args);
      va_end(args);
   }

   void console::info(const char* p, ...)
   {
      va_list args;
      va_start(args, p);
      vprintf(cInfoConsoleMessage, p, args);
      va_end(args);
   }

   void console::message(const char* p, ...)
   {
      va_list args;
      va_start(args, p);
      vprintf(cMessageConsoleMessage, p, args);
      va_end(args);
   }

   void console::cons(const char* p, ...)
   {
      va_list args;
      va_start(args, p);
      vprintf(cConsoleConsoleMessage, p, args);
      va_end(args);
   }

   void console::debug(const char* p, ...)
   {
      va_list args;
      va_start(args, p);
      vprintf(cDebugConsoleMessage, p, args);
      va_end(args);
   }

   void console::warning(const char* p, ...)
   {
      va_list args;
      va_start(args, p);
      vprintf(cWarningConsoleMessage, p, args);
      va_end(args);
   }

   void console::error(const char* p, ...)
   {
      va_list args;
      va_start(args, p);
      vprintf(cErrorConsoleMessage, p, args);
      va_end(args);
   }

} // namespace crnlib