RemixSim.cpp

Go to the documentation of this file.

#include "RemixSim.hpp"
 
Bool_t RemixSim::SimFileSpecTagLess(const SimFileSpec &a,
                                    const SimFileSpec &b) {
  return a.tag.CompareTo(b.tag) < 0;
}
 
TString RemixSim::ControlDir() { return Paths::DatasetDir() + "/sim_control"; }
 
TString RemixSim::TagFromControlFile(const TString &filepath) {
  std::ifstream in(filepath.Data());
  if (!in.is_open())
    return "";
  TString prefix = TString("traces_") + Paths::DatasetName() + "_";
  std::string raw;
  while (std::getline(in, raw)) {
    TString line(raw.c_str());
    line = line.Strip(TString::kLeading);
    if (!line.BeginsWith("output"))
      continue;
    Ssiz_t start = line.Index(prefix);
    if (start == kNPOS)
      return "";
    TString rest = line(start, line.Length() - start);
    Ssiz_t dot = rest.Index(".root");
    if (dot == kNPOS)
      return "";
    TString base = rest(0, dot);
    base.Remove(0, prefix.Length());
    return base;
  }
  return "";
}
 
std::vector<RemixSim::SimFileSpec> RemixSim::BuildFileSpecs() {
  std::vector<SimFileSpec> specs;
  TString dir = ControlDir();
  void *d = gSystem->OpenDirectory(dir.Data());
  if (!d) {
    std::cerr << "BuildFileSpecs: cannot open control dir " << dir << std::endl;
    return specs;
  }
  const char *entry;
  while ((entry = gSystem->GetDirEntry(d))) {
    TString name(entry);
    if (!name.EndsWith(".toml"))
      continue;
    TString tag = TagFromControlFile(dir + "/" + name);
    if (tag.Length() > 0) {
      SimFileSpec s;
      s.tag = tag;
      specs.push_back(s);
    }
  }
  gSystem->FreeDirectory(d);
  std::sort(specs.begin(), specs.end(), SimFileSpecTagLess);
  for (Int_t i = 0; i < (Int_t)specs.size(); i++) {
    std::cout << specs.at(i).tag << std::endl;
  }
  return specs;
}
 
TString RemixSim::TracesName(const SimFileSpec &s) {
  return Form("traces_%s_%s", Paths::DatasetName().Data(), s.tag.Data());
}
 
TString RemixSim::SimRootPath(const SimFileSpec &s) {
  return Paths::DatasetDir() + "/sim_root_files/" + TracesName(s) + ".root";
}
 
Int_t RemixSim::ReactionStripOf(const TString &tag) {
  Ssiz_t pos = tag.Last('_');
  if (pos == kNPOS)
    return -1;
  TString last = tag(pos + 1, tag.Length() - pos - 1);
  if (last.Length() < 2 || last[0] != 's')
    return -1;
  TString digits = last(1, last.Length() - 1);
  if (!digits.IsDigit())
    return -1;
  return digits.Atoi();
}
 
TString RemixSim::TagWithoutStrip(const TString &tag) {
  if (ReactionStripOf(tag) < 0)
    return tag;
  Ssiz_t pos = tag.Last('_');
  return tag(0, pos);
}
 
Bool_t RemixSim::IsEresTag(const TString &tag) {
  return TagWithoutStrip(tag).EndsWith("_eres");
}