CbmKfUtil.h

Go to the documentation of this file.

/* Copyright (C) 2023 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergey Gorbunov [committer] */
 
#ifndef CbmKfUtil_H
#define CbmKfUtil_H 1
 
#include "CbmDefs.h"
#include "CbmRecoSetupManager.h"
#include "KfTrackParam.h"
 
#include <optional>
 
class KFParticle;
 
class FairTrackParam;
 
 
namespace cbm::algo::kf
{
  template<typename Float>
  class Setup;
 
  template<typename Float>
  struct FloatTag;
}  // namespace cbm::algo::kf

namespace cbm::kf
{
  using cbm::algo::kf::EFieldMode;
  using cbm::algo::kf::FloatTag;
  using cbm::algo::kf::fvec;
  using cbm::algo::kf::Setup;

  cbm::algo::kf::TrackParamD ConvertTrackParam(const FairTrackParam& par);

  FairTrackParam ConvertTrackParam(const cbm::algo::kf::TrackParamD& t);
 
  template<cbm::algo::ca::EDetectorID DetID>
  const auto* GetTrackingInterface()
  {
    const auto& recoSetup = cbm::RecoSetupManager::Instance()->GetSetup();
    if constexpr (DetID == cbm::algo::ca::EDetectorID::kMvd) {
      return recoSetup.template Get<ECbmModuleId::kMvd>();
    }
    else if constexpr (DetID == cbm::algo::ca::EDetectorID::kSts) {
      return recoSetup.template Get<ECbmModuleId::kSts>();
    }
    else if constexpr (DetID == cbm::algo::ca::EDetectorID::kMuch) {
      return recoSetup.template Get<ECbmModuleId::kMuch>();
    }
    else if constexpr (DetID == cbm::algo::ca::EDetectorID::kTrd) {
      return recoSetup.template Get<ECbmModuleId::kTrd>();
    }
    else if constexpr (DetID == cbm::algo::ca::EDetectorID::kTof) {
      return recoSetup.template Get<ECbmModuleId::kTof>();
    }
    __builtin_unreachable();  // No return is reachable
  }
 
  template<ECbmModuleId ModuleId>
  const auto* GetTrackingInterface()
  {
    const auto& recoSetup = cbm::RecoSetupManager::Instance()->GetSetup();
    return recoSetup.template Get<ModuleId>();
  }
 
  std::optional<KFParticle> CreateKfParticle(const cbm::algo::kf::TrackParamD& trackParam, double mass,
                                             int absCharge = 1);
 
}  // namespace cbm::kf
 
#endif