NtpTools.cpp

Go to the documentation of this file.

#include "NtpMaker/NtpTools.h"

#include <fstream>

namespace NAIA {
namespace NtpTools {
std::pair<unsigned int, unsigned int> GetInnerNHits(TrTrackR *track) {
  if (!track)
    return std::make_pair(0, 0);
  unsigned int patt_xy = track->GetBitPatternXYJ();
  unsigned int patt_y = track->GetBitPatternJ();
  unsigned int nxy = 0, ny = 0;
  for (int ilay = 0 + 1; ilay < 9 - 1; ilay++) {
    if ((patt_xy & (1 << ilay)) > 0)
      nxy++;
    if ((patt_y & (1 << ilay)) > 0)
      ny++;
  }
  return std::make_pair(nxy, ny);
}

// 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 NtpTools
} // namespace NAIA