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, "[" + std::to_string(Environment::mpiRank()) + "]  " + str, FWD(args)...);
      else
        fmt::print(out, 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, "[0]  " + str, FWD(args)...);
      else if (Environment::mpiSize() == 1)
        fmt::print(out, str, FWD(args)...);
      return out;
    }

  } // 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(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