G4Atlas/G4AtlasTools/src/PunchThroughParticle.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/

// PunchThroughParticle.h, (c) ATLAS Detector software
 
#ifndef G4ATLASTOOLS_PUNCHTHROUGHPARTICLE_H
#define G4ATLASTOOLS_PUNCHTHROUGHPARTICLE_H
 
//
#include "PunchThroughPDFCreator.h"
 
// C++
#include <memory>
 
// forward declarations
class TH2F;

 
class PunchThroughParticle
{
public:
  PunchThroughParticle(int pdg, bool doAnti = false);           
  ~PunchThroughParticle();                                      

  void setMinEnergy(double minEnergy);                          
  void setMaxNumParticles(int maxNum);                          
  void setNumParticlesFactor(double numFactor);                 
  void setEnergyFactor(double energyFactor);                    
  void setPosAngleFactor(double momAngleFactor);                
  void setMomAngleFactor(double momAngleFactor);                
  void setNumParticlesPDF(std::unique_ptr<PunchThroughPDFCreator> pdf);                     
  void setCorrelation(int corrPdg, TH2F *histLowE, TH2F *histHighE,
                      double minCorrE = 0., double fullCorrE = 0., double lowE = 0.,
                      double midE = 0., double upperE = 0.);     
  void setPCA0PDF(std::unique_ptr<PunchThroughPDFCreator> pdf);                       
  void setPCA1PDF(std::unique_ptr<PunchThroughPDFCreator> pdf);                   
  void setPCA2PDF(std::unique_ptr<PunchThroughPDFCreator> pdf);                     
  void setPCA3PDF(std::unique_ptr<PunchThroughPDFCreator> pdf);                    
  void setPCA4PDF(std::unique_ptr<PunchThroughPDFCreator> pdf);                      
 

  inline int getId()                        { return m_pdgId; };              
  inline bool getdoAnti()                   { return m_doAnti; };             
  inline double getMinEnergy()              { return m_minEnergy; };          
  inline double getNumParticlesFactor()     { return m_numParticlesFactor; }; 
  inline double getEnergyFactor()           { return m_energyFactor; };       
  inline double getPosAngleFactor()         { return m_posAngleFactor; };     
  inline double getMomAngleFactor()         { return m_momAngleFactor; };     
 
  inline int getMaxNumParticles()           { return m_maxNum; };             
  inline int getCorrelatedPdg()             { return m_corrPdg; };            
  inline int getMinCorrelationEnergy()      { return m_corrMinEnergy; };      
  inline int getFullCorrelationEnergy()     { return m_corrFullEnergy; };     
  inline TH2F *getCorrelationLowEHist()     { return m_histCorrLowE; };       
  inline TH2F *getCorrelationHighEHist()    { return m_histCorrHighE; };      
  inline double *getCorrelationHistDomains(){ return m_corrHistDomains; };    
  inline PunchThroughPDFCreator * getNumParticlesPDF()   { return m_pdfNumParticles.get(); };    
  inline PunchThroughPDFCreator * getPCA0PDF()     { return m_pdf_pca0.get(); };      
  inline PunchThroughPDFCreator * getPCA1PDF() { return m_pdf_pca1.get(); };  
  inline PunchThroughPDFCreator * getPCA2PDF()   { return m_pdf_pca2.get(); };    
  inline PunchThroughPDFCreator * getPCA3PDF()  { return m_pdf_pca3.get(); };  
  inline PunchThroughPDFCreator * getPCA4PDF()    { return m_pdf_pca4.get(); };     
 
private:
  int                          m_pdgId;                     
  bool                         m_doAnti;                    

  double                       m_minEnergy;                 
  int                          m_maxNum;                    

  double                       m_numParticlesFactor;        
  double                       m_energyFactor;              
  double                       m_posAngleFactor;            
  double                       m_momAngleFactor;            

  int                          m_corrPdg;                   
  double                       m_corrMinEnergy;             
  double                       m_corrFullEnergy;            
 
  TH2F*                        m_histCorrLowE;              
  TH2F*                        m_histCorrHighE;             
  double                      *m_corrHistDomains;           
  std::unique_ptr<PunchThroughPDFCreator>                  m_pdfNumParticles;           
  std::unique_ptr<PunchThroughPDFCreator>                  m_pdf_pca0;             
  std::unique_ptr<PunchThroughPDFCreator>                  m_pdf_pca1;         
  std::unique_ptr<PunchThroughPDFCreator>                  m_pdf_pca2;           
  std::unique_ptr<PunchThroughPDFCreator>                  m_pdf_pca3;          
  std::unique_ptr<PunchThroughPDFCreator>                  m_pdf_pca4;            
 
}; // class PunchThroughParticle
 
#endif