Debug.h

Go to the documentation of this file.

 
 
 
 
#pragma once
 
#include <BaseClass.h>
 
#include <algorithm>
#include <array>
#include <cerrno>
#include <fstream>
#include <iostream>
#include <sstream>
#include <string>
#include <vector>
 
namespace ttk {
 
  COMMON_EXPORTS extern bool welcomeMsg_;
  COMMON_EXPORTS extern bool goodbyeMsg_;
  COMMON_EXPORTS extern int globalDebugLevel_;
 
  namespace debug {
    enum class Priority : int {
      ERROR, // 0
      WARNING, // 1
      PERFORMANCE, // 2
      INFO, // 3
      DETAIL, // 4
      VERBOSE // 5
    };
 
    enum class Separator : char {
      L0 = '%',
      L1 = '=',
      L2 = '-',
      SLASH = '/',
      BACKSLASH = '\\'
    };
 
    enum class LineMode : int {
      NEW,
      APPEND, // append
      REPLACE // replace line and append
    };
 
    namespace output {
      const std::string BOLD = "\33[0;1m";
      const std::string GREY = "\33[2;1m";
      const std::string ITALIC = "\33[3;1m";
      const std::string UNDERLINED = "\33[4;1m";
      const std::string FLASHING = "\33[5;1m";
      const std::string INVERTED = "\33[7;1m";
      const std::string STRIKETHROUGH = "\33[9;1m";
      const std::string DARKGREY = "\33[30;1m";
      const std::string RED = "\33[31;1m";
      const std::string GREEN = "\33[32;1m";
      const std::string YELLOW = "\33[33;1m";
      const std::string BLUE = "\33[34;1m";
      const std::string PINK = "\33[35;1m";
      const std::string LIGHTBLUE = "\33[36;1m";
      const std::string BRIGHTWHITE = "\33[37;1m";
      const std::string ENDCOLOR = "\33[0m";
    } // namespace output
 
    const int LINEWIDTH = 80;
  } // namespace debug
 
  class Debug : public BaseClass {
 
  public:
    // 1) constructors, destructors, operators, etc.
    Debug();
 
    ~Debug() override;
 
    // 2) functions
    virtual int setDebugLevel(const int &debugLevel);
 
    int setWrapper(const Wrapper *wrapper) override;
 
    // =========================================================================
    // New Debug Methods
    // =========================================================================
 
    inline int printMsg(const std::string &msg,
                        const debug::Priority &priority = debug::Priority::INFO,
                        const debug::LineMode &lineMode = debug::LineMode::NEW,
                        std::ostream &stream = std::cout) const {
      if((this->debugLevel_ < (int)priority)
         && (globalDebugLevel_ < (int)priority))
        return 0;
 
      return this->printMsgInternal(msg, priority, lineMode, stream);
    }
 
    inline int printMsg(const std::vector<std::string> &msgs,
                        const debug::Priority &priority = debug::Priority::INFO,
                        const debug::LineMode &lineMode = debug::LineMode::NEW,
                        std::ostream &stream = std::cout) const {
      if((this->debugLevel_ < (int)priority)
         && (globalDebugLevel_ < (int)priority))
        return 0;
 
      size_t prints = 0;
      for(auto &msg : msgs)
        prints += this->printMsgInternal(msg, priority, lineMode, stream);
      return prints == msgs.size() ? 1 : 0;
    }
 
    inline int printErr(const std::string &msg,
                        const debug::LineMode &lineMode = debug::LineMode::NEW,
                        std::ostream &stream = std::cerr) const {
      return this->printMsgInternal(
        msg, debug::Priority::ERROR, lineMode, stream);
    }
 
    inline int printWrn(const std::string &msg,
                        const debug::LineMode &lineMode = debug::LineMode::NEW,
                        std::ostream &stream = std::cerr) const {
      return this->printMsgInternal(
        msg, debug::Priority::WARNING, lineMode, stream);
    }
 
    inline int printMsg(const std::string &msg,
                        const double &progress,
                        const double &time,
                        const int &threads,
                        const double &memory,
                        const debug::LineMode &lineMode = debug::LineMode::NEW,
                        const debug::Priority &priority
                        = debug::Priority::PERFORMANCE,
                        std::ostream &stream = std::cout) const {
      if((this->debugLevel_ < (int)priority)
         && (globalDebugLevel_ < (int)priority))
        return 0;
 
      std::array<std::string, 4> chunks{};
      size_t q = 0;
 
      if(memory >= 0)
        chunks[q++] = std::to_string(static_cast<int>(memory)) + "MB";
      if(time >= 0) {
        std::stringstream sStream;
        sStream.precision(3);
        sStream << std::fixed;
        sStream << time;
        chunks[q++] = sStream.str() + "s";
      }
      if(threads >= 0)
        chunks[q++] = std::to_string(threads) + "T";
      if(progress >= 0)
        chunks[q++] = std::to_string((int)(progress * 100)) + "%";
 
      std::string stats = "";
      if(q > 0) {
        stats += " [";
        stats += chunks[0];
        for(size_t i = 1; i < q; i++)
          stats += "|" + chunks[i];
 
        stats += "]";
      }
 
      return this->printMsgInternal(
        msg, stats, msg.length() < 1 ? ">" : ".", priority, lineMode, stream);
    }
 
    inline int printMsg(const std::string &msg,
                        const double &progress,
                        const double &time,
                        const debug::LineMode &lineMode = debug::LineMode::NEW,
                        const debug::Priority &priority
                        = debug::Priority::PERFORMANCE,
                        std::ostream &stream = std::cout) const {
      return this->printMsg(
        msg, progress, time, -1, -1, lineMode, priority, stream);
    }
 
    inline int printMsg(const std::string &msg,
                        const double &progress,
                        const double &time,
                        const int &threads,
                        const debug::LineMode &lineMode = debug::LineMode::NEW,
                        const debug::Priority &priority
                        = debug::Priority::PERFORMANCE,
                        std::ostream &stream = std::cout) const {
      return this->printMsg(
        msg, progress, time, threads, -1, lineMode, priority, stream);
    }
 
    inline int printMsg(const std::string &msg,
                        const double &progress,
                        const debug::LineMode &lineMode = debug::LineMode::NEW,
                        const debug::Priority &priority
                        = debug::Priority::PERFORMANCE,
                        std::ostream &stream = std::cout) const {
      return this->printMsg(
        msg, progress, -1, -1, -1, lineMode, priority, stream);
    }
 
    inline int printMsg(const std::string &msg,
                        const double &progress,
                        const debug::Priority &priority,
                        const debug::LineMode &lineMode = debug::LineMode::NEW,
                        std::ostream &stream = std::cout) const {
      return this->printMsg(msg, progress, -1, -1, lineMode, priority, stream);
    }
 
    inline int printMsg(const std::vector<std::vector<std::string>> &rows,
                        const debug::Priority &priority = debug::Priority::INFO,
                        const bool hasHeader = true,
                        const debug::LineMode &lineMode = debug::LineMode::NEW,
                        std::ostream &stream = std::cout) const {
      if((this->debugLevel_ < (int)priority)
         && (globalDebugLevel_ < (int)priority))
        return 0;
 
      int nRows = rows.size();
      int nCols = nRows > 0 ? rows[0].size() : 0;
      if(nCols < 1)
        return 0;
 
      std::vector<std::string> formattedRows(nRows);
      std::vector<size_t> colSizes(nCols, 0);
      for(int i = 0; i < nRows; i++)
        for(int j = 0; j < nCols; j++)
          colSizes[j] = std::max(colSizes[j], rows[i][j].size());
 
      auto formatCell = [](const std::string &value, const size_t &width,
                           const std::string &fillSymbol) {
        std::string cell = value;
        int diff = width - cell.size();
        for(int i = 0; i < diff; i++)
          cell += fillSymbol;
        return cell;
      };
 
      // Values
      int resultIndex = 0;
      for(int i = 0; i < nRows; i++) {
        auto &row = formattedRows[resultIndex++];
        row
          = formatCell(rows[i][0], colSizes[0], " ") + (hasHeader ? ": " : "");
        if(nCols > 1)
          row += formatCell(rows[i][1], colSizes[1], " ");
        for(int j = 2; j < nCols; j++)
          row += "," + formatCell(rows[i][j], colSizes[j], " ");
      }
 
      return this->printMsg(formattedRows, priority, lineMode, stream);
    }
 
    inline int printMsg(const debug::Separator &separator,
                        const debug::LineMode &lineMode = debug::LineMode::NEW,
                        const debug::Priority &priority = debug::Priority::INFO,
                        std::ostream &stream = std::cout) const {
      if((this->debugLevel_ < (int)priority)
         && (globalDebugLevel_ < (int)priority))
        return 0;
 
      return this->printMsgInternal("", "",
                                    std::string(1, (const char &)separator),
                                    priority, lineMode, stream);
    }
 
    inline int printMsg(const debug::Separator &separator,
                        const debug::Priority &priority,
                        const debug::LineMode &lineMode = debug::LineMode::NEW,
                        std::ostream &stream = std::cout) const {
      return this->printMsg(separator, lineMode, priority, stream);
    }
 
    inline int printMsg(const std::string &msg,
                        const debug::Separator &separator,
                        const debug::LineMode &lineMode = debug::LineMode::NEW,
                        const debug::Priority &priority = debug::Priority::INFO,
                        std::ostream &stream = std::cout) const {
      if((this->debugLevel_ < (int)priority)
         && (globalDebugLevel_ < (int)priority))
        return 0;
 
      return this->printMsgInternal(msg, "",
                                    std::string(1, (const char &)separator),
                                    priority, lineMode, stream);
    }
 
    inline void setDebugMsgPrefix(const std::string &prefix) {
      this->debugMsgNamePrefix_ = prefix;
#ifdef TTK_ENABLE_MPI
      this->debugMsgPrefix_
        = debugMsgNamePrefix_.length() > 0
            ? "[" + debugMsgNamePrefix_ + "-" + std::to_string(MPIrank_) + "] "
            : "";
#else
      this->debugMsgPrefix_ = debugMsgNamePrefix_.length() > 0
                                ? "[" + debugMsgNamePrefix_ + "] "
                                : "";
#endif // TTK_ENABLE_MPI
    }
 
  protected:
    mutable int debugLevel_;
 
    COMMON_EXPORTS static debug::LineMode lastLineMode;
 
    std::string debugMsgPrefix_;
    std::string debugMsgNamePrefix_;
 
    inline int
      printMsgInternal(const std::string &msg,
                       const std::string &right,
                       const std::string &filler,
                       const debug::Priority &priority = debug::Priority::INFO,
                       const debug::LineMode &lineMode = debug::LineMode::NEW,
                       std::ostream &stream = std::cout) const {
 
      std::string combinedMsg = msg;
 
      if(filler.length() > 0) {
        if(msg.length() > 0)
          combinedMsg += " ";
 
        int gapWidth = debug::LINEWIDTH - this->debugMsgPrefix_.length()
                       - combinedMsg.length() - right.length();
        gapWidth = std::max(gapWidth / filler.length(), (size_t)1);
 
        for(int i = 0; i < gapWidth; i++)
          combinedMsg += filler;
 
        combinedMsg += debug::output::BLUE + right + debug::output::ENDCOLOR;
      }
 
      return this->printMsgInternal(combinedMsg, priority, lineMode, stream);
    }
 
    inline int printMsgInternal(const std::string &msg,
                                const debug::Priority &priority,
                                const debug::LineMode &lineMode,
                                std::ostream &stream = std::cout) const {
 
      if((this->debugLevel_ < (int)priority)
         && (globalDebugLevel_ < (int)priority))
        return 0;
 
      // on error or warning print end of line
      if((int)priority < 2 && this->lastLineMode == debug::LineMode::REPLACE)
        stream << "\n";
 
      // print prefix
      if(lineMode != debug::LineMode::APPEND)
        stream << debug::output::GREEN << this->debugMsgPrefix_
               << debug::output::ENDCOLOR;
 
      // print error or warning prefix
      if((int)priority == 0)
        stream << debug::output::RED << "[ERROR]" << debug::output::ENDCOLOR
               << " ";
      else if((int)priority == 1)
        stream << debug::output::YELLOW << "[WARNING]"
               << debug::output::ENDCOLOR << " ";
 
      // print msg
      stream << msg.data();
 
      // go either into new line or replace current line
      if(lineMode == debug::LineMode::NEW)
        stream << "\n";
      else if(lineMode == debug::LineMode::REPLACE)
        stream << "\r";
 
      // flush stream
      stream.flush();
 
      this->lastLineMode = lineMode;
 
      return 1;
    }
 
    int welcomeMsg(std::ostream &stream);
  };
 
} // namespace ttk
 
#include <Os.h>
#include <Timer.h>
 
namespace ttk {
  class DebugTimer : public Timer {};
  class DebugMemory : public Memory {};
} // namespace ttk
 
#include <OrderDisambiguation.h>