CbmKfTrackingSetupBuilder.h

Go to the documentation of this file.

/* Copyright (C) 2024 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergei Zharko [committer] */
 
 
#pragma once
 
#include "CbmDefs.h"
#include "CbmEnumArray.h"
#include "KfSetupBuilder.h"
#include "TString.h"
 
#include <mutex>
#include <tuple>
 
namespace cbm::kf
{
  class TrackingSetupBuilder {
   public:
    static TrackingSetupBuilder* Instance();
 
    std::shared_ptr<const cbm::algo::kf::Setup<double>> GetSharedGeoSetup();
 
    template<typename T>
    cbm::algo::kf::Setup<T> MakeSetup(cbm::algo::kf::EFieldMode fldMode)
    {
      if (!fbInitialized) {
        this->Init();
      }
      return fBuilder.MakeSetup<T>(fldMode);
    }
 
    bool IsInGeometry(cbm::algo::ca::EDetectorID detID) const { return fvbDetInGeometry[detID]; }
 
    bool HasHits(cbm::algo::ca::EDetectorID detID) const { return fvbDetHasHits[detID]; }
 
    void SetIgnoreHitPresence(bool ok = true)
    {
      fbIgnoreHitPresence = ok;
      fbInitialized       = false;
      CheckDetectorPresence();
    }
 
    // Disable copy and move
    TrackingSetupBuilder(const TrackingSetupBuilder&) = delete;
    TrackingSetupBuilder(TrackingSetupBuilder&&)      = delete;
    TrackingSetupBuilder& operator=(const TrackingSetupBuilder&) = delete;
    TrackingSetupBuilder& operator=(TrackingSetupBuilder&&) = delete;
 
   private:
    template<typename T>
    using DetectorIDArray_t = cbm::core::EnumArray<cbm::algo::ca::EDetectorID, T>;
 
    static constexpr DetectorIDArray_t<const char*> kDetHitBrName{
      {"MvdHit", "StsHit", "MuchPixelHit", "TrdHit", "TofHit"}};
 
    TrackingSetupBuilder() = default;
 
    ~TrackingSetupBuilder() = default;
 
    void CheckDetectorPresence();
 
    void SetMaterialCacheFile(const TString& filename, size_t geoHash)
    {
      fBuilder.SetMaterialCacheFile(filename.Data(), geoHash);
    }
 
    void Init();
 
    // Material map creator properties (TODO: Provide setters, if needed)
    static constexpr double kMatCreatorPitch{0.1};    
    static constexpr int kMatCreatorMaxNbins{100};    
    static constexpr int kMatCreatorNrays{3};         
    static constexpr bool kMatCreatorSafeMode{true};  
    static constexpr double kTargFieldInitStep{2.5};  
    static constexpr double kTargMaterialOffset{1};   
 
    inline static TrackingSetupBuilder* fpInstance{nullptr};
    inline static std::mutex fMutex{};
 
    cbm::algo::kf::SetupBuilder fBuilder{};       
    DetectorIDArray_t<bool> fvbDetInGeometry{{false}};  
    DetectorIDArray_t<bool> fvbDetHasHits{{false}};     
 
    std::shared_ptr<cbm::algo::kf::Setup<double>> fpGeoSetup{nullptr};
 
    bool fbInitialized{false};        
    bool fbIgnoreHitPresence{false};  
  };
 
 
}  // namespace cbm::kf