AliAnaParticleJetFinderCorrelation.h

Go to the documentation of this file.

#ifndef ALIANAPARTICLEJETFINDERCORRELATION_H
#define ALIANAPARTICLEJETFINDERCORRELATION_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice     */

//_________________________________________________________________________

// --- ROOT system ---
class TH2F;
class TTree;
class TRandom2;

//---- Analysis system ----
#include "AliAnaCaloTrackCorrBaseClass.h"

class AliAnaParticleJetFinderCorrelation : public AliAnaCaloTrackCorrBaseClass {
       
 public:
    
           AliAnaParticleJetFinderCorrelation() ;
    
  virtual ~AliAnaParticleJetFinderCorrelation() ;

  // General methods
  
  void     InitParameters();
  
  TList *  GetCreateOutputObjects();

  void     MakeAnalysisFillAOD() ;
  
  void     MakeAnalysisFillHistograms() ;
  
  // To access non standard branch
  Int_t    SelectJet(AliAODPWG4Particle * particle, TClonesArray * aodRecJets) ;
    
  void     Print(const Option_t * opt)           const;
  
  // Settings
  
  Bool_t   OnlyIsolated()                        const { return fSelectIsolated              ; }
  void     SelectIsolated(Bool_t select)               { fSelectIsolated = select            ; }
  
  Float_t  GetConeSize()                         const { return fConeSize                    ; }
  Float_t  GetPtThresholdInCone()                const { return fPtThresholdInCone           ; }     
  Double_t GetDeltaPhiMaxCut()                   const { return fDeltaPhiMaxCut              ; }
  Double_t GetDeltaPhiMinCut()                   const { return fDeltaPhiMinCut              ; }
  Double_t GetRatioMaxCut()                      const { return fRatioMaxCut                 ; }
  Double_t GetRatioMinCut()                      const { return fRatioMinCut                 ; }       
  Bool_t   AreJetRefTracks()                     const { return fUseJetRefTracks             ; }
  Bool_t   IsCorrelationMadeInHistoMaker()       const { return fMakeCorrelationInHistoMaker ; } 
  Double_t GetJetConeSize()                      const { return fJetConeSize                 ; } 
  Double_t GetJetMinPt()                         const { return fJetMinPt                    ; }
  Double_t GetJetMinPtBkgSub()                   const { return fJetMinPtBkgSub              ; }
  Double_t GetJetAreaFraction()                  const { return fJetAreaFraction             ; }
  Double_t GetGammaConeSize()                    const { return fGammaConeSize               ; }

  void     SetConeSize(Float_t cone)                   { fConeSize = cone                    ; }
  void     SetPtThresholdInCone(Float_t pt)            { fPtThresholdInCone = pt             ; }
    
  void     SetDeltaPhiCutRange(Double_t phimin, Double_t phimax)
            { fDeltaPhiMaxCut = phimax  ;  fDeltaPhiMinCut = phimin                          ; }
    
  void     SetRatioCutRange(Double_t ratiomin, Double_t ratiomax)
            { fRatioMaxCut    = ratiomax;  fRatioMinCut    = ratiomin                        ; }
    
  void     UseJetRefTracks(Bool_t use)                 { fUseJetRefTracks = use              ; }  
  void     SetMakeCorrelationInHistoMaker(Bool_t make) { fMakeCorrelationInHistoMaker = make ; }
  void     SetJetConeSize(Double_t cone)               { fJetConeSize     = cone             ; }
  void     SetJetMinPt(Double_t minpt)                 { fJetMinPt        = minpt            ; }
  void     SetJetMinPtBkgSub(Double_t minpt)           { fJetMinPtBkgSub  = minpt            ; }
  void     SetJetAreaFraction(Double_t areafr)         { fJetAreaFraction = areafr           ; }
  void     SetGammaConeSize(Float_t cone)              { fGammaConeSize   = cone             ; }

  // Settings for non standard jet branch
  TString  GetJetBranchName()                    const { return fJetBranchName               ; }
  void     SetJetBranchName(const char *name)          { fJetBranchName = name               ; }
//void     SwitchOnNonStandardJetFromReader()          { fNonStandardJetFromReader = kTRUE   ; }
//void     SwitchOffNonStandardJetFromReader()         { fNonStandardJetFromReader = kFALSE  ; }
//Bool_t   IsNonStandardJetFromReader()                { return fNonStandardJetFromReader    ; }

  TString  GetBkgJetBranchName()                 const { return fBkgJetBranchName            ; }
  void     SetBkgJetBranchName(const char *name)       { fBkgJetBranchName = name            ; }
  void     SwitchOnBackgroundJetFromReader()           { fBackgroundJetFromReader = kTRUE    ; }
  void     SwitchOffBackgroundJetFromReader()          { fBackgroundJetFromReader = kFALSE   ; }
  Bool_t   IsBackgroundJetFromReader()                 { return fBackgroundJetFromReader     ; }

  //switches for photons
  void     SwitchOnBackgroundSubtractionGamma()        { fUseBackgroundSubtractionGamma = kTRUE ; }
  void     SwitchOffBackgroundSubtractionGamma()       { fUseBackgroundSubtractionGamma = kFALSE; }
  Bool_t   IsBackgroundSubtractionGamma()        const { return fUseBackgroundSubtractionGamma  ; }

  void     CalculateBkg(TVector3 gamma, TVector3 jet,Double_t *vector,Int_t type);

  void     FindMCgenInfo();//gives information on generated level

  void     SwitchOnSaveGJTree()                         { fSaveGJTree = kTRUE  ; }
  void     SwitchOffSaveGJTree()                        { fSaveGJTree = kFALSE ; }
  Bool_t   IsSaveGJTree()                         const { return fSaveGJTree   ; }

  void     SwitchOnMostEnergetic()                      { fMostEnergetic = kTRUE  ; fMostOpposite = kFALSE ; }
  void     SwitchOffMostEnergetic()                     { fMostEnergetic = kFALSE ; fMostOpposite = kTRUE  ; }
  void     SwitchOffMostOpposite()                      { fMostEnergetic = kTRUE  ; fMostOpposite = kFALSE ; }
  void     SwitchOnMostOpposite()                       { fMostEnergetic = kFALSE ; fMostOpposite = kTRUE  ; }
  Bool_t   IsMostEnergetic()                      const { return fMostEnergetic ; }
  Bool_t   IsMostOpposite()                       const { return fMostOpposite  ; }

  //switches for histograms
  void     SwitchOnHistogramJetBkg()                    { fUseHistogramJetBkg = kTRUE  ; }
  void     SwitchOffHistogramJetBkg()                   { fUseHistogramJetBkg = kFALSE ; }
  Bool_t   IsHistogramJetBkg()                    const { return fUseHistogramJetBkg   ; }

  void     SwitchOnHistogramTracks()                    { fUseHistogramTracks = kTRUE  ; }
  void     SwitchOffHistogramTracks()                   { fUseHistogramTracks = kFALSE ; }
  Bool_t   IsHistogramTracks()                    const { return fUseHistogramTracks   ; }

  void     SwitchOnHistogramJetTracks()                 { fUseHistogramJetTracks = kTRUE  ; }
  void     SwitchOffHistogramJetTracks()                { fUseHistogramJetTracks = kFALSE ; }
  Bool_t   IsHistogramJetTracks()                 const { return fUseHistogramJetTracks   ; }

  void     SwitchOnMCStudies()                          { fMCStudies = kTRUE  ; }
  void     SwitchOffMCStudies()                         { fMCStudies = kFALSE ; }
  Bool_t   IsMCStudies()                          const { return fMCStudies   ; }

private:

  // selection parameters
  Double_t   fDeltaPhiMaxCut ;                    
  Double_t   fDeltaPhiMinCut ;                    
  Double_t   fRatioMaxCut ;                       
  Double_t   fRatioMinCut ;                       
  
  Double_t   fConeSize  ;                         
  Double_t   fPtThresholdInCone ;                 
  Bool_t     fUseJetRefTracks ;                   
  Bool_t     fMakeCorrelationInHistoMaker ;       
  Bool_t     fSelectIsolated ;                    

  Double_t   fJetConeSize ;                       
  Double_t   fJetMinPt ;                          
  Double_t   fJetMinPtBkgSub ;                    
  Double_t   fJetAreaFraction ;                   
//Bool_t     fNonStandardJetFromReader;           ///<  use non standard jet from reader //new
  TString    fJetBranchName ;                     
  Bool_t     fBackgroundJetFromReader;            
  TString    fBkgJetBranchName ;                  

  Double_t   fGammaConeSize ;                     
  Bool_t     fUseBackgroundSubtractionGamma;      
  Bool_t     fSaveGJTree;                         
  Bool_t     fMostEnergetic;                      
  Bool_t     fMostOpposite;                       

  Bool_t     fUseHistogramJetBkg;                 
  Bool_t     fUseHistogramTracks;                 
  Bool_t     fUseHistogramJetTracks;              

  Bool_t     fMCStudies;                          

  TRandom2 * fGenerator;                          

  TLorentzVector fMomentum;                       
  
  // Histograms
  TH2F *     fhDeltaEta;                          
//TH2F *     fhDeltaPhi;                          //!<!  Difference of jet phi and trigger particle phi as function of trigger particle pT
  TH2F *     fhDeltaPhiCorrect;                   
  TH2F *     fhDeltaPhi0PiCorrect;                

  TH2F *     fhDeltaPt;                           
  TH2F *     fhPtRatio;                           
  TH2F *     fhPt;                                
  
  TH2F *     fhFFz ;                              
  TH2F *     fhFFxi;                              
  TH2F *     fhFFpt;                              
  TH2F *     fhNTracksInCone;                     
  
  TH2F *     fhJetFFz ;                           
  TH2F *     fhJetFFxi;                           
  TH2F *     fhJetFFpt;                           
  TH2F *     fhJetFFzCor ;                        
  TH2F *     fhJetFFxiCor;                        

  TH1F * fhGamPtPerTrig ;                         
  TH2F * fhPtGamPtJet ;                           

  // background from RC
  TH2F *     fhBkgFFz[5] ;                        
  TH2F *     fhBkgFFxi[5];                        
  TH2F *     fhBkgFFpt[5];                        
  TH2F *     fhBkgNTracksInCone[5];               
  TH2F *     fhBkgSumPtInCone[5];                 
  TH2F *     fhBkgSumPtnTracksInCone[5];          

  // temporary histograms
  TH2F * fhNjetsNgammas;                          
  TH1F * fhCuts;                                  

  TH2F * fhDeltaEtaBefore;                        
  TH2F * fhDeltaPhiBefore;                        
  TH2F * fhDeltaPtBefore;                         
  TH2F * fhPtRatioBefore;                         
  TH2F * fhPtBefore;                              
  TH2F * fhDeltaPhi0PiCorrectBefore;              

  // temporary jet histograms
  TH1F * fhJetPtBefore;                           
  TH1F * fhJetPtBeforeCut;                        
  TH1F * fhJetPt;                                 
  TH1F * fhJetPtMostEne;                          
  TH1F * fhJetPhi;                                
  TH1F * fhJetEta;                                
  TH2F * fhJetEtaVsPt;                            
  TH2F * fhJetPhiVsEta;                           
  TH2F * fhJetEtaVsNpartInJet;                    
  TH2F * fhJetEtaVsNpartInJetBkg;                 
  TH2F * fhJetChBkgEnergyVsPt;                    
  TH2F * fhJetChAreaVsPt;                         
  TH2F * fhTrackPhiVsEta;                         
  TH1F * fhTrackAveTrackPt;                       
  TH1F * fhJetNjetOverPtCut[10];                  
  TH2F * fhJetChBkgEnergyVsArea;                  
  TH2F * fhJetRhoVsPt;                            
  TH2F * fhJetRhoVsCentrality;                    
  TH1F * fhJetNparticlesInJet;                    
  TH2F * fhJetDeltaEtaDeltaPhi;                   
  TH2F * fhJetDeltaEtaDeltaPhiAllTracks;          

  TH1F * fhJetAveTrackPt;                         
  TH2F * fhJetNtracksInJetAboveThr[6];            
  TH2F * fhJetRatioNTrkAboveToNTrk[5];            
  TH2F * fhJetNtrackRatioMostEne  [5];            
  TH2F * fhJetNtrackRatioJet5GeV  [5];            
  TH2F * fhJetNtrackRatioLead5GeV [5];            

  // temporary background jet histograms
  TH1F * fhBkgJetBackground[4];                   
  TH1F * fhBkgJetSigma[4];                        
  TH1F * fhBkgJetArea[4];                         

  // temporary photon histograms
  TH1F * fhPhotonPtMostEne;                       
  TH1F * fhPhotonAverageEnergy;                   
  TH1F * fhPhotonRatioAveEneToMostEne;            
  TH1F * fhPhotonAverageEnergyMinus1;             
  TH1F * fhPhotonRatioAveEneMinus1ToMostEne;      
  TH1F * fhPhotonNgammaMoreAverageToNgamma;       
  TH1F * fhPhotonNgammaMoreAverageMinus1ToNgamma; 
  TH1F * fhPhotonNgammaOverPtCut[10];             
  TH2F * fhPhotonBkgRhoVsNtracks;                 
  TH2F * fhPhotonBkgRhoVsNclusters;               
  TH2F * fhPhotonBkgRhoVsCentrality;              
  TH2F * fhPhotonBkgRhoVsNcells;                  
  TH1F * fhPhotonPt;                              
  TH1F * fhPhotonPtCorrected;                     
  TH1F * fhPhotonPtCorrectedZoom;                 
  TH1F * fhPhotonPtDiff;                          
  TH2F * fhPhotonPtDiffVsCentrality;              
  TH2F * fhPhotonPtDiffVsNcells;                  
  TH2F * fhPhotonPtDiffVsNtracks;                 
  TH2F * fhPhotonPtDiffVsNclusters;               

  TH1F * fhPhotonSumPtInCone;                     
  TH1F * fhPhotonSumPtCorrectInCone;              
  TH1F * fhPhotonSumPtChargedInCone;              


  // temporary jet histograms after selection
  TH2F * fhSelectedJetPhiVsEta;                   
  TH2F * fhSelectedJetChBkgEnergyVsPtJet;         
  TH2F * fhSelectedJetChAreaVsPtJet;              
  TH1F * fhSelectedJetNjet;                       
  TH1F * fhSelectedNtracks;                       
  TH2F * fhSelectedTrackPhiVsEta;                 

  TH1F * fhCuts2;                                 

  // temporary photon histogram after selection
  TH2F * fhSelectedPhotonNLMVsPt;                 
  TH2F * fhSelectedPhotonLambda0VsPt;             
  TH2F * fhRandomPhiEta[5];                       

  // MC generated histograms
  TH1F * fhMCPhotonCuts;                          
  TH1F * fhMCPhotonPt;                            
  TH2F * fhMCPhotonEtaPhi;                        
  TH1F * fhMCJetOrigin;                           
  TH2F * fhMCJetNPartVsPt;                        
  TH2F * fhMCJetChNPartVsPt;                      
  TH2F * fhMCJetNPart150VsPt;                     
  TH2F * fhMCJetChNPart150VsPt;                   
  TH2F * fhMCJetChNPart150ConeVsPt;               
  TH1F * fhMCJetRatioChFull;                      
  TH1F * fhMCJetRatioCh150Ch;                     
  TH2F * fhMCJetEtaPhi;                           
  TH2F * fhMCJetChEtaPhi;                         
  TH2F * fhMCJet150EtaPhi;                        
  TH2F * fhMCJetCh150EtaPhi;                      
  TH2F * fhMCJetCh150ConeEtaPhi;                  
 
  // tree with data gamma and jet
  TTree *  fTreeGJ;                               
  Double_t fGamPt;                                
  Double_t fGamLambda0;                           
  Int_t    fGamNLM;                               
  Double_t fGamSumPtCh;                           
  Double_t fGamTime;                              
  Int_t    fGamNcells;                            
  Double_t fGamEta;                               
  Double_t fGamPhi;                               
  Double_t fGamSumPtNeu;                          
  Int_t    fGamNtracks;                           
  Int_t    fGamNclusters;                         
  Double_t fGamAvEne;                             
  Double_t fJetPhi;                               
  Double_t fJetEta;                               
  Double_t fJetPt;                                
  Double_t fJetBkgChEne;                          
  Double_t fJetArea;                              
  Int_t    fJetNtracks;                           
  Int_t    fJetNtracks1;                          
  Int_t    fJetNtracks2;                          
  Double_t fJetRho;                               
  Int_t    fEventNumber;                          
  Int_t    fNtracks;                              
  Double_t fZvertex;                              
  Double_t fCentrality;                           
  Bool_t   fIso;                                  
  Double_t fGamRho;                               

  Double_t fMCGamPt;                              
  Double_t fMCGamEta;                             
  Double_t fMCGamPhi;                             
  Int_t    fMCPartonType;                         
  Double_t fMCJetPt;                              
  Double_t fMCJetChPt;                            
  Double_t fMCJet150Pt;                           
  Double_t fMCJetCh150Pt;                         
  Int_t    fMCJetNPart;                           
  Int_t    fMCJetChNPart;                         
  Int_t    fMCJet150NPart;                        
  Int_t    fMCJetCh150NPart;                      
  Double_t fMCJetEta;                             
  Double_t fMCJetPhi;                             
  Double_t fMCJetChEta;                           
  Double_t fMCJetChPhi;                           
  Double_t fMCJet150Eta;                          
  Double_t fMCJet150Phi;                          
  Double_t fMCJetCh150Eta;                        
  Double_t fMCJetCh150Phi;                        

  Double_t fMCJetCh150ConePt;                     
  Int_t    fMCJetCh150ConeNPart;                  
  Double_t fMCJetCh150ConeEta;                    
  Double_t fMCJetCh150ConePhi;                    

  AliAnaParticleJetFinderCorrelation(              const AliAnaParticleJetFinderCorrelation & g) ;
    
  AliAnaParticleJetFinderCorrelation & operator = (const AliAnaParticleJetFinderCorrelation & g) ;
  
  ClassDef(AliAnaParticleJetFinderCorrelation,3) ;

 } ;

#endif //ALIANAPARTICLEJETFINDERCORRELATION_H