PunchThroughParticle.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
// PunchThroughParticle.h, (c) ATLAS Detector software
 
#ifndef ISF_PUNCHTHROUGHTOOLS_PUNCHTHROUGHPARTICLE_H
#define ISF_PUNCHTHROUGHTOOLS_PUNCHTHROUGHPARTICLE_H
 
// ISF
#include "PDFcreator.h"
 
// forward declarations
class TH2F;
 
/*-------------------------------------------------------------------------
 *  class PunchThroughParticle
 *-------------------------------------------------------------------------*/
namespace ISF
{
  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<PDFcreator> 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<PDFcreator> pdf);                       
    void setPCA1PDF(std::unique_ptr<PDFcreator> pdf);                   
    void setPCA2PDF(std::unique_ptr<PDFcreator> pdf);                     
    void setPCA3PDF(std::unique_ptr<PDFcreator> pdf);                    
    void setPCA4PDF(std::unique_ptr<PDFcreator> 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 PDFcreator * getNumParticlesPDF()   { return m_pdfNumParticles.get(); };    
    inline PDFcreator * getPCA0PDF()     { return m_pdf_pca0.get(); };      
    inline PDFcreator * getPCA1PDF() { return m_pdf_pca1.get(); };  
    inline PDFcreator * getPCA2PDF()   { return m_pdf_pca2.get(); };    
    inline PDFcreator * getPCA3PDF()  { return m_pdf_pca3.get(); };  
    inline PDFcreator * 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<PDFcreator>                  m_pdfNumParticles;           
    std::unique_ptr<PDFcreator>                  m_pdf_pca0;             
    std::unique_ptr<PDFcreator>                  m_pdf_pca1;         
    std::unique_ptr<PDFcreator>                  m_pdf_pca2;           
    std::unique_ptr<PDFcreator>                  m_pdf_pca3;          
    std::unique_ptr<PDFcreator>                  m_pdf_pca4;            
 
  };
}
 
#endif