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);
}

std::vector<std::string> ExpandTextFilelist(const std::string &txtFile) {
  using namespace std::string_literals;

  std::ifstream infilelist(txtFile);
  std::vector<std::string> flist;
  for (std::string bufname; std::getline(infilelist, bufname);) {
    if (bufname.substr(bufname.length() - 4, 4) == ".txt"s) {
      auto tmp = ExpandTextFilelist(bufname);
      flist.insert(end(flist), begin(tmp), end(tmp));
    } else {
      flist.push_back(bufname);
    }
  }

  return flist;
}

std::vector<std::string> LoadInputFiles(std::vector<std::string> &inputFiles) {

  using namespace std::string_literals;
  if (inputFiles.size() > 1 && std::any_of(begin(inputFiles), end(inputFiles), [](const std::string &filename) {
        return filename.substr(filename.length() - 4, 4) == ".txt"s;
      })) {
    throw std::runtime_error("Input file list cannot contain mixed txt and root files");
  }

  // we need to read the filelist from disk if the only file ends with ".txt"
  std::vector<std::string> fileList;
  if (inputFiles.size() == 1 && inputFiles[0].substr(inputFiles[0].length() - 4, 4) == ".txt") {
    fileList = ExpandTextFilelist(inputFiles[0]);
  } else {
    fileList = inputFiles;
  }

  return fileList;
}

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