KfMaterialMapFactory.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 "KfIMaterialMapFactory.h"
#include "Rtypes.h"
#include "TObject.h"
 
class TGeoNavigator;
namespace cbm::algo::kf
{
  class MaterialMap;
}
 
namespace kf::tools
{
  class MaterialMapFactory : public cbm::algo::kf::IMaterialMapFactory {
   public:
    MaterialMapFactory(int verbose = 0);

    ~MaterialMapFactory();

    void SetDoRadialProjection(double targetZ)
    {
      fDoRadialProjection = true;
      fTargetZ            = targetZ;
    }

    void SetDoHorisontalProjection() { fDoRadialProjection = false; }

    void SetNraysPerDim(int nRaysDim) { fNraysBinPerDim = (nRaysDim > 0) ? nRaysDim : 1; }
 

    cbm::algo::kf::MaterialMap GenerateMaterialMap(double zRef, double zMin, double zMax, double xyMax,
                                                   int nBinsDim) override;

    void SetSafeMaterialInitialization(bool val = true) { fDoSafeInitialization = val; }
 
   private:
    void InitThreads();

    void CleanUpThreads();

    TGeoNavigator* GetCurrentNavigator(int iThread);
 
   private:
    static constexpr double kMinRadLength = 0.3;  
 
    std::vector<TGeoNavigator*> fNavigators{};  
    double fTargetZ{0.};                        
    int fNthreadsOld{0};                        
    int fNthreads{0};                           
    int fNraysBinPerDim{3};                     
    int fVerbose{0};                            
    bool fDoRadialProjection{false};            
    bool fDoSafeInitialization{false};          
  };
 
}  // namespace kf::tools