Output.hpp

#pragma once
 
#include <iostream>
#include <string>
 
#ifdef AMDIS_NO_THROW
  #include <cassert>
#else
  #include <stdexcept>
#endif
 
#include <fmt/core.h>
#include <fmt/ostream.h>
 
#include <amdis/Environment.hpp>
#include <amdis/common/TypeTraits.hpp>
 
#ifndef AMDIS_ENABLE_MSG_DBG
  #ifndef NDEBUG
    #define AMDIS_ENABLE_MSG_DBG 1
  #else
    #define AMDIS_ENABLE_MSG_DBG 0
  #endif
#endif
 
#define AMDIS_FUNCNAME(nn) [[maybe_unused]] const char *funcName = nn;
 
#ifdef NDEBUG
  #define AMDIS_FUNCNAME_DBG(nn)
  #define AMDIS_DBG_VAR(var)
#else
  #define AMDIS_FUNCNAME_DBG(nn) [[maybe_unused]] const char *funcName = nn;
  #define AMDIS_DBG_VAR(var) var
#endif
 
 
namespace AMDiS
{
  namespace Impl
  {
    template <class OStream, class... Args>
    OStream& msg_all(OStream& out, std::string const& str, Args&&... args)
    {
      if (Environment::mpiSize() > 1)
        fmt::print(out, fmt::runtime("[" + std::to_string(Environment::mpiRank()) + "]  " + str), FWD(args)...);
      else
        fmt::print(out, fmt::runtime(str), FWD(args)...);
      return out;
    }
 
    template <class OStream, class... Args>
    OStream& msg(OStream& out, std::string const& str, Args&&... args)
    {
      if (Environment::msgAllRanks())
        return msg_all(out, str, FWD(args)...);
 
      if (Environment::mpiSize() > 1 && Environment::mpiRank() == 0)
        fmt::print(out, fmt::runtime("[0]  " + str), FWD(args)...);
      else if (Environment::mpiSize() == 1)
        fmt::print(out, fmt::runtime(str), FWD(args)...);
      return out;
    }
 
    template <class T>
    void printType (T const&)
    {
      std::cout << __PRETTY_FUNCTION__ << std::endl;
    }
 
    template <class T>
    void printType ()
    {
      std::cout << __PRETTY_FUNCTION__ << std::endl;
    }
 
  } // end namespace Impl
 
 
 
  template <class... Args>
  void msg(std::string const& str, Args&&... args)
  {
    Impl::msg(std::cout, str + "\n", FWD(args)...);
  }
 
  template <class... Args>
  void info(int noInfoLevel, std::string const& str, Args&&... args)
  {
    if (Environment::infoLevel() >= noInfoLevel)
      Impl::msg(std::cout, str + "\n", FWD(args)...);
  }
 
 
 
  template <class... Args>
  void msg_(std::string const& str, Args&&... args)
  {
    Impl::msg(std::cout, str, FWD(args)...);
  }
 
  template <class... Args>
  void info_(int noInfoLevel, std::string const& str, Args&&... args)
  {
    if (Environment::infoLevel() >= noInfoLevel)
      Impl::msg(std::cout, str, FWD(args)...);
  }
 
 
 
 
  template <class... Args>
  void error_exit(std::string const& str, Args&&... args)
  {
#ifdef AMDIS_NO_THROW
    Impl::msg_all(std::cerr, "ERROR: " + str + "\n", FWD(args)...);
    std::abort();
#else
    throw std::runtime_error( std::string("ERROR: ") + fmt::format(fmt::runtime(str), FWD(args)...));
#endif
  }
 
 
 
  template <class... Args>
  void test_exit(bool condition, std::string const& str, Args&&... args)
  {
    if (!condition) { error_exit(str, FWD(args)...); }
  }
 
 
  template <class... Args>
  void warning(std::string const& str, Args&&... args)
  {
    Impl::msg(std::cout, "WARNING: " + str + "\n", FWD(args)...);
  }
 
 
 
  template <class... Args>
  void test_warning(bool condition, std::string const& str, Args&&... args)
  {
    if (!condition) { warning(str, FWD(args)...); }
  }
 
 
#if AMDIS_ENABLE_MSG_DBG
 
  template <class... Args>
  void msg_dbg(Args&&... args) { msg(FWD(args)...); }
 
 
 
  template <class... Args>
  void test_exit_dbg(bool condition, Args&&... args)
  {
    test_exit(condition, FWD(args)...);
  }
#else
  template <class... Args>
  void msg_dbg(Args&&...) {}
 
  template <class... Args>
  void test_exit_dbg(bool, Args&&...) {}
#endif
 
} // end namespace AMDiS