RecoSetup.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 "algo/setup/RecoSetup.h"
 
#include <boost/archive/binary_iarchive.hpp>
#include <boost/archive/binary_oarchive.hpp>
#include <boost/filesystem.hpp>
#include <boost/serialization/vector.hpp>
 
#include <fstream>
#include <sstream>
 
using cbm::algo::RecoSetup;
 
// ---------------------------------------------------------------------------------------------------------------------
//
RecoSetup::RecoSetup(const std::string& filename) : RecoSetup(std::move(RecoSetup::Load(filename))) {}
 
// ---------------------------------------------------------------------------------------------------------------------
//
bool RecoSetup::Has(ECbmModuleId moduleId) const
{
  switch (moduleId) {
    case ECbmModuleId::kMvd: return Has<ECbmModuleId::kMvd>();
    case ECbmModuleId::kSts: return Has<ECbmModuleId::kSts>();
    case ECbmModuleId::kMuch: return Has<ECbmModuleId::kMuch>();
    case ECbmModuleId::kTrd: return Has<ECbmModuleId::kTrd>();
    case ECbmModuleId::kTof: return Has<ECbmModuleId::kTof>();
    default: return false;
  }
}
 
// ---------------------------------------------------------------------------------------------------------------------
//
RecoSetup RecoSetup::Load(const std::string& filename)
{
  RecoSetup setup{};
 
  std::ifstream ifs(filename, std::ios::binary);
  if (!ifs) {
    std::stringstream msg;
    msg << "RecoSetup::Load: input file \"" << filename << "\" was not found";
    throw std::runtime_error(msg.str());
  }
 
  try {
    boost::archive::binary_iarchive ia(ifs);
    ia >> setup;
  }
  catch (const std::exception& err) {
    std::stringstream msg;
    msg << "RecoSetup::Load: input file \"" << filename
        << "\" has inconsistent format or was corrupted. Details: " << err.what();
    throw std::runtime_error(msg.str());
  }
 
  return setup;
}
 
// ---------------------------------------------------------------------------------------------------------------------
//
void RecoSetup::Store(const RecoSetup& setup, const std::string& filename)
{
  namespace fs = boost::filesystem;
  auto path    = fs::absolute(fs::weakly_canonical(filename));
  fs::create_directories(path.parent_path());
 
  std::ofstream ofs(path.string(), std::ios::binary);
  if (!ofs) {
    std::stringstream msg;
    msg << "RecoSetup::Store: failed opening file  \"" << filename << "\" to store a setup";
    throw std::runtime_error(msg.str());
  }
  boost::archive::binary_oarchive oa(ofs);
  oa << setup;
}
 
// ---------------------------------------------------------------------------------------------------------------------
//
std::string RecoSetup::ToString() const
{
  std::stringstream msg;
  msg << "======== CBM Reconstruction Setup ============================================";
  ForEachUnit([&](const auto& unit) { msg << '\n' << unit.TableOfTrackingStations(); });
  msg << "==============================================================================";
  return msg.str();
}