services/run_info/app/Application.cxx

Go to the documentation of this file.

/* Copyright (C) 2025 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergei Zharko [committer] */

 
#include "Application.h"
 
#include "CbmEnumArray.h"
#include "CbmRunDatabaseContainer.h"
 
#include <Logger.h>
 
#include <boost/program_options.hpp>
 
#include <cstdlib>
#include <exception>
#include <iostream>
#include <sstream>
 
using cbm::services::run_info::Application;
using cbm::services::run_info::EInfo;
 
namespace po = boost::program_options;
 
 
// ---------------------------------------------------------------------------------------------------------------------
//
std::optional<EInfo> Application::ParseOptions(int argc, char* argv[])
{
  cbm::core::EnumArray<EInfo, bool> vbSelected{{0}};
  bool verbose = false;
 
  // ----- Define options
  po::options_description opts("  Options");
  auto optsAdd = opts.add_options();
  // Default options
  optsAdd("help,h", "display this help and exit");
 
  // Required options
  optsAdd("run,r", po::value<uint32_t>(&fRunId)->value_name("<runID>")->required(),
          "standard CBM run identifier (required)");
 
  // Info selection options
  optsAdd("geotag,g", po::bool_switch(&vbSelected[EInfo::GeoTag])->default_value(false), "prints the geometry tag");
  optsAdd("print,p", po::bool_switch(&vbSelected[EInfo::Print])->default_value(false),
          "print available information on the run ID");
  optsAdd("run-start", po::bool_switch(&vbSelected[EInfo::RunStart])->default_value(false),
          "print a run starttime in nanoseconds since the epoch");
  optsAdd("algo-config", po::bool_switch(&vbSelected[EInfo::AlgoConfig])->default_value(false),
          "print path to the algo config");
  optsAdd("tof-calibration-tag", po::bool_switch(&vbSelected[EInfo::TofCalibTag])->default_value(false),
          "print TOF calibration tag");
  optsAdd("bmon-calibration-tag", po::bool_switch(&vbSelected[EInfo::BmonCalibTag])->default_value(false),
          "print BMON calibration tag");
  optsAdd("alignment-tag", po::bool_switch(&vbSelected[EInfo::AlignmentTag])->default_value(false),
          "print alignment tag");
  optsAdd("reco-par-tag", po::bool_switch(&vbSelected[EInfo::RecoParTag])->default_value(false),
          "print reconstruction parameters tag");
  optsAdd("reco-par-dir", po::bool_switch(&vbSelected[EInfo::RecoParDir])->default_value(false),
          "print reconstruction parameters directory");
 
  // Optional
  optsAdd("verbose,v", po::bool_switch(&verbose)->default_value(false),
          "Switch logging printout ON (default: only FATAL)");
 
 
  po::variables_map vars;
  po::store(po::parse_command_line(argc, argv, opts), vars);
 
  if (vars.count("help")) {
    std::cout << opts << std::endl;
    std::exit(EXIT_SUCCESS);
  }
 
  po::notify(vars);
 
  int nSelectedInfos{0};
  EInfo selectedInfo{EInfo::Print};
  for (size_t iInfo = 0; iInfo < vbSelected.size(); ++iInfo) {
    if (vbSelected[iInfo]) {
      selectedInfo = static_cast<EInfo>(iInfo);
      ++nSelectedInfos;
    }
  }
  if (nSelectedInfos > 1) {
    throw std::logic_error("More than one information items were requested. Please request only one item");
  }
 
  if (!verbose) {
    fair::Logger::SetConsoleSeverity(fair::Severity::fatal);
  }
 
  return std::make_optional(selectedInfo);
}
 
// ---------------------------------------------------------------------------------------------------------------------
//
void Application::Print(EInfo info) const
{
  auto& db = cbm::RunDatabaseContainer::Instance();
  if (!db.LoadRun(fRunId)) {
    std::cerr << "Failed to load info for run " << fRunId << " from the database (see error above)" << std::endl;
    throw std::runtime_error("DB Info not available for this run");
  }
  std::string dir = std::getenv("VMCWORKDIR") != nullptr ? std::getenv("VMCWORKDIR") : "";
  if (dir.empty()) {
    throw std::runtime_error("VMCWORKDIR environmental variable not found. Please configure your CbmRoot");
  }
 
  switch (info) {
    case EInfo::GeoTag: std::cout << db.GetGeoSetupTag() << std::endl; break;
    case EInfo::TofCalibTag: std::cout << db.GetTofCalibrationTag() << std::endl; break;
    case EInfo::BmonCalibTag: std::cout << db.GetBmonCalibrationTag() << std::endl; break;
    case EInfo::AlignmentTag: std::cout << db.GetAlignmentTag() << std::endl; break;
    case EInfo::RecoParTag: std::cout << db.GetRecoParTag() << std::endl; break;
    case EInfo::RecoParDir: std::cout << dir << "/" << db.GetRecoParDir() << std::endl; break;
    case EInfo::AlgoConfig: std::cout << dir << "/" << db.GetAlgoMainConfigPath() << std::endl; break;
    case EInfo::RunStart: std::cout << db.GetRunStartTimeNs() << std::endl; break;
    case EInfo::Print: std::cout << db.ToString() << std::endl; break;
    case EInfo::END: break;
  };
}