LengthIntegrator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef G4UserActions_LengthIntegrator_H
#define G4UserActions_LengthIntegrator_H
 
#include "GaudiKernel/ITHistSvc.h"
#include "GaudiKernel/ServiceHandle.h"
 
#include "G4Pow.hh"
#include "TString.h"
 
#include "G4UserEventAction.hh"
#include "G4UserSteppingAction.hh"
//#include "AthenaBaseComps/AthMessaging.h"
#include <string>
#include <map>
 
// Forward declarations
class TProfile;
class TProfile2D;
 
 
namespace G4UA
{

  class LengthIntegrator final : public G4UserEventAction,
    public G4UserSteppingAction
  {
 
    public:

      LengthIntegrator(const std::string& histSvcName, bool doHistos);
 
      LengthIntegrator(const LengthIntegrator&) = delete;
      LengthIntegrator& operator=(const LengthIntegrator&) = delete;

      virtual void BeginOfEventAction(const G4Event*) override;

      virtual void EndOfEventAction(const G4Event*) override;

      virtual void UserSteppingAction(const G4Step*) override;
 
    private:
 
      // Holder for G4 math tools
      G4Pow* m_g4pow;
 
      //Tree for material information
      TTree* m_tree;
 
      //Tree Branches
      int   m_genNPart = 0;
      float m_genEta = 0.0F;
      float m_genPhi = 0.0F;
      float m_genZ = 0.0F;
      float m_genR = 0.0F;
      
      //X0 Branches
      float m_total_X0 = 0.0F;
      float m_total_L0 = 0.0F;
 
      std::vector<double> m_collected_X0;
      std::vector<double> m_collected_L0;
 
      std::vector<float> m_collected_hitr;
      std::vector<float> m_collected_hitx;
      std::vector<float> m_collected_hity;
      std::vector<float> m_collected_hitz;
 
      std::vector<float> m_collected_outhitr;
      std::vector<float> m_collected_outhitx;
      std::vector<float> m_collected_outhity;
      std::vector<float> m_collected_outhitz;
 
      std::vector<float> m_collected_density;
      std::vector<std::string> m_collected_material;
      std::vector<std::string> m_collected_volume;
      
      std::vector<std::string> m_collected_groupedmaterial;
      std::vector<std::string> m_collected_volumetype;
 
      bool m_splitModerator;
      bool m_splitPP1;  
 
      void fillNtuple();
      std::string getMaterialClassification(const std::string& name);
      std::string getVolumeType(const std::string& s);
 
      // Add elements and values into the map
      void addToDetThickMap(const std::string&, double, double);

      void regAndFillHist(const std::string&, const std::pair<double, double>&);

      TProfile2D* getOrCreateProfile(const std::string& regName, const TString& histoname, const TString& xtitle, int nbinsx, float xmin, float xmax,const TString& ytitle, int nbinsy,float ymin, float ymax,const TString& ztitle);

      ServiceHandle<ITHistSvc> m_hSvc;
      
      //Do we create histograms
      bool m_doHistos;

      double m_etaPrimary;
      double m_phiPrimary;

      std::map<std::string, std::pair<double, double> > m_detThickMap;

      TProfile2D* m_rzProfRL;
      std::map<std::string, TProfile*> m_etaMapRL;
      std::map<std::string, TProfile*> m_phiMapRL;
 
 

      TProfile2D* m_rzProfIL;
      std::map<std::string, TProfile*> m_etaMapIL;
      std::map<std::string, TProfile*> m_phiMapIL;
 
      // 2D plots of rad-length and int-length
      std::map<std::string,TProfile2D*,std::less<std::string> > m_rzMapRL;
      std::map<std::string,TProfile2D*,std::less<std::string> > m_xyMapRL;
 
      std::map<std::string,TProfile2D*,std::less<std::string> > m_rzMapIL;
      std::map<std::string,TProfile2D*,std::less<std::string> > m_xyMapIL;
 
  }; // class LengthIntegrator
 
} // namespace G4UA
 
#endif