Paths.cpp

Go to the documentation of this file.

#include "Paths.hpp"
#include <fstream>
 
#ifndef MUSIC_DATASET_NAME
#define MUSIC_DATASET_NAME "unknown"
#endif
#ifndef MUSIC_GIT_HASH
#define MUSIC_GIT_HASH "unknown"
#endif
#ifndef MUSIC_DATASET_DIR
#define MUSIC_DATASET_DIR ""
#endif
// Where the wordmark lives. Baked in at build like the paths above so the
// binary stays self-locating; nix points it at the installed copy.
#ifndef MUSIC_ASSETS_DIR
#define MUSIC_ASSETS_DIR ""
#endif
 
TString Paths::DatasetName() { return TString(MUSIC_DATASET_NAME); }
 
TString Paths::ResultsDir() {
  const Char_t *env = gSystem->Getenv("MUSIC_RESULTS_DIR");
  if (env && env[0] != '\0')
    return TString(env);
  return DatasetDir();
}
 
void Paths::PrintLogo() {
  std::ifstream in(TString(MUSIC_ASSETS_DIR) + "/music.txt");
  if (!in)
    return; // art is decoration; never let a missing asset stop a run
  std::cout << std::endl << in.rdbuf() << std::endl;
}
 
void Paths::PrintBanner(const TString &dataset_dir) {
  std::cout << "============================================================"
            << std::endl;
  std::cout << " MUSIC tooling | dataset=" << MUSIC_DATASET_NAME
            << " | git=" << MUSIC_GIT_HASH << std::endl;
  std::cout << " dataset dir : " << dataset_dir << std::endl;
  std::cout << " CoMPASS base: " << Constants::cfg.COMPASS_BASE_DIR
            << std::endl;
  std::cout << " runs        :";
  for (Int_t i = 0; i < Int_t(Constants::cfg.RUN_NUMBERS.size()); i++)
    std::cout << " " << Constants::cfg.RUN_NUMBERS[i];
  std::cout << std::endl;
  std::cout << " event mode  : " << Constants::ActiveReferenceChannel() << " ("
            << Constants::ActiveEventTimeWindowUs() << " us window)"
            << " | plots ";
  if (Constants::cfg.SAVE_FULL_PLOTS)
    std::cout << "every subfile";
  else
    std::cout << "first " << Constants::cfg.PLOT_SAMPLE_FILES
              << " subfile(s) per epoch";
  std::cout << std::endl;
  std::cout << "============================================================"
            << std::endl;
}
 
TString Paths::DatasetDir() {
  const Char_t *env = gSystem->Getenv("MUSIC_DATASET_DIR");
  TString d;
  if (env && env[0] != '\0') {
    d = TString(env);
    // Guard against running a binary for one dataset in another's dev shell:
    // the env override would silently mix configs, so the env path must contain
    TString name = TString(MUSIC_DATASET_NAME);
    if (name.Length() > 0 && name != "unknown" && !d.Contains("/" + name)) {
      std::cerr << "FATAL: this binary was built for dataset '" << name
                << "' but MUSIC_DATASET_DIR points to '" << d << "'."
                << std::endl;
      std::cerr << "You are probably running the wrong ./result symlink or "
                   "are in the wrong dataset dev shell."
                << std::endl;
      std::cerr << "Rebuild with `nix build .#" << name
                << "` or enter the matching shell (`nix develop .#" << name
                << "`)." << std::endl;
      gSystem->Exit(1);
    }
  } else {
    d = MUSIC_DATASET_DIR;
    if (d.Length() == 0) {
      std::cerr
          << "FATAL: this binary was built without a dataset dir baked in "
             "(MUSIC_DATASET_DIR). Rebuild via the Makefile."
          << std::endl;
      gSystem->Exit(1);
    }
  }
  static Bool_t printed = kFALSE;
  if (!printed) {
    PrintBanner(d);
    printed = kTRUE;
  }
  return d;
}