CbmKfTrackingGeoSetupFactory.h

Go to the documentation of this file.

/* Copyright (C) 2024-2025 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 "CbmKfUtil.h"
#include "KfISetupFactory.h"
#include "KfSetupBuilder.h"
#include "TString.h"
 
#include <tuple>
 
namespace cbm::kf
{
  class TrackingGeoSetupFactory : public cbm::algo::kf::ISetupFactory {
   public:
    TrackingGeoSetupFactory(bool requireHitPresence = false);

    Setup<double> Create(FloatTag<double>, EFieldMode fldMode) const override
    {
      return fBuilder.template MakeSetup<double>(fldMode);
    }

    Setup<float> Create(FloatTag<float>, EFieldMode fldMode) const override
    {
      return fBuilder.template MakeSetup<float>(fldMode);
    }

    Setup<fvec> Create(FloatTag<algo::kf::fvec>, EFieldMode fldMode) const override
    {
      return fBuilder.template MakeSetup<fvec>(fldMode);
    }

    bool IsInGeometry(algo::ca::EDetectorID detID) const { return fvbDetInGeometry[detID]; }

    bool HasHits(algo::ca::EDetectorID detID) const { return fvbDetHasHits[detID]; }

    const algo::kf::FieldFn_t& GetFieldFunction() const override { return fBuilder.GetFieldFunction(); }
 
 
   private:
    template<typename T>
    using DetectorIDArray_t = cbm::core::EnumArray<algo::ca::EDetectorID, T>;
 
    static constexpr DetectorIDArray_t<const char*> kDetHitBrName{
      {"MvdHit", "StsHit", "MuchPixelHit", "TrdHit", "TofHit"}};  
 
    // 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};   
 
    algo::kf::SetupBuilder fBuilder{};                  
    DetectorIDArray_t<bool> fvbDetInGeometry{{false}};  
    DetectorIDArray_t<bool> fvbDetHasHits{{false}};     

    void CheckDetectorPresence(bool requireHitPresence);

    void SetMaterialCacheFile(const TString& filename, size_t geoHash)
    {
      fBuilder.SetMaterialCacheFile(filename.Data(), geoHash);
    }

    std::string MaterialCacheFile() const;
  };
 
 
}  // namespace cbm::kf