GGEMSPrint.hh

Go to the documentation of this file.

#ifndef GUARD_GGEMS_TOOLS_GGEMSPRINT_HH
#define GUARD_GGEMS_TOOLS_GGEMSPRINT_HH
 
// ************************************************************************
// * This file is part of GGEMS.                                          *
// *                                                                      *
// * GGEMS is free software: you can redistribute it and/or modify        *
// * it under the terms of the GNU General Public License as published by *
// * the Free Software Foundation, either version 3 of the License, or    *
// * (at your option) any later version.                                  *
// *                                                                      *
// * GGEMS is distributed in the hope that it will be useful,             *
// * but WITHOUT ANY WARRANTY; without even the implied warranty of       *
// * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        *
// * GNU General Public License for more details.                         *
// *                                                                      *
// * You should have received a copy of the GNU General Public License    *
// * along with GGEMS.  If not, see <https://www.gnu.org/licenses/>.      *
// *                                                                      *
// ************************************************************************
 
#include <iostream>
#include <ostream>
#include <iomanip>
#include <mutex>
 
namespace {
  std::mutex mutex; 
}
 
#include "GGEMS/global/GGEMSExport.hh"
#include "GGEMS/tools/GGEMSTypes.hh"
 
#ifdef _WIN32
#ifdef _MSC_VER
#define NOMINMAX
#endif
#include <windows.h>
#endif
 
#define GGendl (static_cast<std::ostream& (*)(std::ostream&)>(std::endl))
 
#define GGcin (std::cin)
 
#ifdef _MSC_VER
#pragma warning(disable: 4251) // Deleting warning exporting STL members!!!
#endif
 
enum GGEMSConsoleColor : GGuchar
{
  black = 0,
  blue,
  green,
  aqua,
  red,
  purple,
  yellow,
  white,
  gray
};
 
namespace
{
  #ifdef _WIN32
  WORD constexpr kColor [] = {
    0x00,
    0x01,
    0x02,
    0x03,
    0x04,
    0x05,
    0x06,
    0x07,
    0x08
  }; 
  #else
  std::string const kColor[] = {
    "\033[30m",
    "\033[34m",
    "\033[32m",
    "\033[36m",
    "\033[31m",
    "\033[35m",
    "\033[33m",
    "\033[97m",
    "\033[37m"
  }; 
  std::string const kDefaultColor("\033[0m"); 
  #endif
}
 
class GGEMS_EXPORT GGEMSStream
{
  public:
    GGEMSStream(std::ostream& stream, GGEMSConsoleColor const& color);
 
    ~GGEMSStream(){};
 
  public:
    GGEMSStream& operator()(std::string const& class_name, std::string const& method_name, GGint const& verbosity_level);
 
    template<typename T>
    GGEMSStream& operator<<(T const& message);
 
    void SetVerbosity(GGint const& verbosity_limit);
 
  private:
    std::string class_name_; 
    std::string method_name_; 
    GGint verbosity_limit_; 
    GGint verbosity_level_; 
    GGint stream_counter_; 
  protected: // Avoid warnings using clang on Windows system
    std::ostream& stream_; 
    GGEMSConsoleColor color_index_; 
};
 
 
template <typename T>
GGEMSStream& GGEMSStream::operator<<(T const& message)
{
  #ifdef _WIN32
  // Get current color of terminal
  CONSOLE_SCREEN_BUFFER_INFO info;
  GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &info);
  HANDLE hConsole = GetStdHandle(STD_OUTPUT_HANDLE);
  FlushConsoleInputBuffer(hConsole);
  #endif
 
  if (stream_counter_ == 0) {
    if (verbosity_level_ <= verbosity_limit_) {
      stream_ << std::scientific << "[";
      if (!class_name_.empty() && !method_name_.empty()) {
        #ifdef _WIN32
        SetConsoleTextAttribute(hConsole, ::kColor[color_index_]);
        stream_ << "GGEMS " << class_name_ << "::" << method_name_;
        SetConsoleTextAttribute(hConsole, info.wAttributes);
        #else
        stream_ << kColor[color_index_] << "GGEMS " << class_name_ << "::" << method_name_ << kDefaultColor;
        #endif
      }
      else {
        #ifdef _WIN32
        SetConsoleTextAttribute(hConsole, ::kColor[color_index_]);
        stream_ << "GGEMS";
        SetConsoleTextAttribute(hConsole, info.wAttributes);
        #else
        stream_ << kColor[color_index_] << "GGEMS" << kDefaultColor;
        #endif
      }
      stream_ << "](" << verbosity_level_ << ") " << message;
      stream_counter_++;
    }
  }
  else {
    if (verbosity_level_ <= verbosity_limit_) stream_ << message;
  }
  return *this;
}
 
 
extern GGEMS_EXPORT GGEMSStream GGcout; 
extern GGEMS_EXPORT GGEMSStream GGcerr; 
extern GGEMS_EXPORT GGEMSStream GGwarn; 
extern "C" GGEMS_EXPORT void set_ggems_verbose(GGint verbosity);
 
#endif // GUARD_GGEMS_TOOLS_GGPRINT_HH