Logging.cpp

Go to the documentation of this file.

#include "Utility/Logging/Logging.h"

#include <fstream>
#include <iostream>

namespace NAIA {
namespace Logging {
// Mute-unmute routines.
// From: https://bbs.archlinux.org/viewtopic.php?id=79378
// Modified to mute also C-style output.

FILE *muteOut(fopen("/dev/null", "w"));
std::ofstream fout("/dev/null");

FILE *stdoutSave = nullptr;
FILE *stderrSave = nullptr;
std::streambuf *cout_sbuf = nullptr;
std::streambuf *cerr_sbuf = nullptr;

bool isMuted = false;

void MuteOutput() {
  if (!std::exchange(isMuted, true)) {
    stdoutSave = stdout;
    stdout = muteOut;
    stderrSave = stderr;
    stderr = muteOut;

    cout_sbuf = std::cout.rdbuf();
    cerr_sbuf = std::cerr.rdbuf();
    std::cout.rdbuf(fout.rdbuf());
    std::cerr.rdbuf(fout.rdbuf());
  }
}

void UnmuteOutput() {
  if (std::exchange(isMuted, false)) {
    if (stdoutSave) {
      stdout = stdoutSave;
      stdoutSave = nullptr;
    }
    if (stderrSave) {
      stderr = stderrSave;
      stderrSave = nullptr;
    }

    if (cout_sbuf) {
      std::cout.rdbuf(cout_sbuf);
      cout_sbuf = nullptr;
    }
    if (cerr_sbuf) {
      std::cerr.rdbuf(cerr_sbuf);
      cerr_sbuf = nullptr;
    }
  }
}
} // namespace Logging
} // namespace NAIA