logtag.hh

// -*- tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=8 sw=2 sts=2:
 
#ifndef DUNE_PDELAB_COMMON_LOGTAG_HH
#define DUNE_PDELAB_COMMON_LOGTAG_HH
 
#include <memory>
#include <ostream>
#include <sstream>
#include <string>
 
namespace Dune {
  namespace PDELab {
 
 
    extern std::ostream &logtag(std::ostream &s);
 
    extern std::ostream &hostPidWallUserLogtagFormatFunc(std::ostream &s);
 
 
    extern std::ostream &hostRankWallUserLogtagFormatFunc(std::ostream &s);
 
    extern std::ostream &nullFormatFunc(std::ostream &s);
 
 
    extern void logtagSetupMPI(bool syncWidthes = true);
 
 
    struct LogtagFormatterBase {
 
      virtual void writeTag(std::ostream &s) const = 0;
 
      virtual ~LogtagFormatterBase() = default;
 
    };
 
    template<class FormatFunc>
    class GeneralLogtagFormatter :
      public LogtagFormatterBase
    {
      FormatFunc formatFunc;
 
    public:
 
      GeneralLogtagFormatter(const FormatFunc &formatFunc_) :
        formatFunc(formatFunc_)
      { }
 
      virtual void writeTag(std::ostream &s) const override { formatFunc(s); }
    };
    template<class FormatFunc>
    std::shared_ptr<LogtagFormatterBase>
    makeGeneralLogtagFormatter(const FormatFunc &formatFunc)
    { return std::make_shared<GeneralLogtagFormatter<FormatFunc> >(formatFunc); }
    extern std::shared_ptr<LogtagFormatterBase>
    makeGeneralLogtagFormatter(std::ostream &(&formatFunc)(std::ostream&));
 
    extern const std::shared_ptr<LogtagFormatterBase> &getLogtagFormatter();
 
    extern void
    setLogtagFormatter(const std::shared_ptr<LogtagFormatterBase> &formatter
                       = std::shared_ptr<LogtagFormatterBase>());
 
    template<class FormatFunc>
    void setLogtagFormatFunc(const FormatFunc &formatFunc)
    { setLogtagFormatter(makeGeneralLogtagFormatter(formatFunc)); }
 
 
    class WithLogtag {
      std::shared_ptr<LogtagFormatterBase> savedFormatter;
 
    public:
      template<class FormatFunc>
      WithLogtag(const FormatFunc &formatFunc) :
        savedFormatter(getLogtagFormatter())
      { setLogtagFormatFunc(formatFunc); }
 
      ~WithLogtag();
    };
 
 
    class LocalTag {
      std::string str_;
 
    public:
      inline const std::string &str() const { return str_; }
 
      template<class V>
      LocalTag &operator<<(const V &v) {
        std::stringstream s;
        s << v;
        str_ += s.str();
        return *this;
      }
    };
 
    inline std::ostream &operator<<(std::ostream &s, const LocalTag &tag)
    { return s << logtag << tag.str(); }
 
  } // namespace PDELab
} // namespace Dune
 
#endif // DUNE_PDELAB_COMMON_LOGTAG_HH