]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWGGA/CaloTrackCorrelations/AliAnaPi0EbE.h
a7bf28f690b29e160908979aa80b16e49f013fcc
[u/mrichter/AliRoot.git] / PWGGA / CaloTrackCorrelations / AliAnaPi0EbE.h
1 #ifndef ALIANAPI0EBE_H
2 #define ALIANAPI0EBE_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice     */
5
6 //_________________________________________________________________________
7 //
8 // Class for the analysis of high pT pi0 event by event
9 // Pi0/Eta identified by one of the following:
10 //  -Invariant mass of 2 cluster in calorimeter
11 //  -Shower shape analysis in calorimeter
12 //  -Invariant mass of one cluster in calorimeter and one photon reconstructed in TPC (in near future)
13 //
14 //-- Author: Gustavo Conesa (INFN-LNF)  &  Raphaelle Ichou (SUBATECH)
15 //_________________________________________________________________________
16
17
18 // --- ROOT system ---
19 class TList ;
20 class TObjString;
21
22 // --- ANALYSIS system ---
23 #include "AliAnaCaloTrackCorrBaseClass.h"
24
25 class AliAnaPi0EbE : public AliAnaCaloTrackCorrBaseClass {
26
27  public: 
28   AliAnaPi0EbE() ; // default ctor
29   virtual ~AliAnaPi0EbE() { ; } //virtual dtor
30           
31   TObjString *   GetAnalysisCuts();
32   
33   TList      *   GetCreateOutputObjects();
34   
35   Int_t          GetMCIndex(const Int_t aodTag);
36   
37   void           Init();
38   
39   void           InitParameters();
40
41   void           MakeAnalysisFillAOD()  ;
42    
43   void           MakeAnalysisFillHistograms() ; 
44   
45   void           Print(const Option_t * opt) const;
46   
47   // Main
48   
49   void           FillSelectedClusterHistograms(AliVCluster* cluster, 
50                                                const Int_t nLocMax,
51                                                const Int_t tag,
52                                                const Float_t asy = 0);
53     
54   void           FillWeightHistograms(AliVCluster *clus);
55     
56   void           HasPairSameMCMother(AliAODPWG4Particle * photon1, 
57                                      AliAODPWG4Particle * photon2, 
58                                      Int_t & label, Int_t & tag);
59   
60   void           MakeInvMassInCalorimeter() ;
61   
62   void           MakeInvMassInCalorimeterAndCTS() ;
63   
64   void           MakeShowerShapeIdentification() ;
65           
66   //Setters Getters
67   
68   //Analysis types
69   enum anaTypes  {kIMCalo, kSSCalo, kIMCaloTracks};  
70   anaTypes       GetAnalysisType()                     const { return fAnaType                ; }
71   void           SetAnalysisType(anaTypes ana)               { fAnaType = ana                 ; }
72   
73   TString        GetInputAODGammaConvName()            const { return fInputAODGammaConvName  ; }
74   void           SetInputAODGammaConvName(TString name)      { fInputAODGammaConvName = name  ; }       
75   
76   //Only for pi0 SS identification case
77   void           SetCalorimeter(TString & det)               { fCalorimeter = det             ; }
78   
79   void           SetMinDistanceToBadChannel(Float_t m1, Float_t m2, Float_t m3) {
80                   fMinDist = m1; fMinDist2 = m2; fMinDist3 = m3                               ; }
81   
82   void           SetTimeCut(Double_t min, Double_t max)      { fTimeCutMin = min; 
83                                                                fTimeCutMax = max              ; }
84   Double_t       GetTimeCutMin()                       const { return fTimeCutMin             ; }
85   Double_t       GetTimeCutMax()                       const { return fTimeCutMax             ; }       
86
87   void           SwitchOnFillWeightHistograms()              { fFillWeightHistograms = kTRUE  ; }
88   void           SwitchOffFillWeightHistograms()             { fFillWeightHistograms = kFALSE ; }  
89   
90   void           SwitchOnTMHistoFill()                       { fFillTMHisto          = kTRUE  ; }
91   void           SwitchOffTMHistoFill()                      { fFillTMHisto          = kFALSE ; }
92
93   void           SwitchOnSelectedClusterHistoFill()          { fFillSelectClHisto    = kTRUE  ; }
94   void           SwitchOffSelectedClusterHistoFill()         { fFillSelectClHisto    = kFALSE ; }
95   
96   //For histograms
97   enum mcTypes   { kmcPhoton = 0, kmcConversion = 1, kmcPi0    = 2,  
98                    kmcEta    = 3, kmcElectron   = 4, kmcHadron = 5 };
99
100  private:
101   
102   anaTypes       fAnaType;                 // Select analysis type
103   
104   //Only for pi0 SS identification case, kSSCalo
105   TString        fCalorimeter ;            // Calorimeter where the gamma is searched;
106   Float_t        fMinDist ;                // Minimal distance to bad channel to accept cluster
107   Float_t        fMinDist2;                // Cuts on Minimal distance to study acceptance evaluation
108   Float_t        fMinDist3;                // One more cut on distance used for acceptance-efficiency study
109   Double_t       fTimeCutMin  ;            // Remove clusters/cells with time smaller than this value, in ns
110   Double_t       fTimeCutMax  ;            // Remove clusters/cells with time larger than this value, in ns
111   
112   Bool_t         fFillWeightHistograms ;   // Fill weigth histograms
113   Bool_t         fFillTMHisto;             // Fill track matching plots
114   Bool_t         fFillSelectClHisto;       // Fill selected cluster histograms
115
116   //Only for combination of calorimeter and conversion photons, kIMCaloTracks
117   TString        fInputAODGammaConvName;   //  Name of AOD branch with conversion photons
118   
119   //Histograms
120   
121   TH1F         * fhPt  ;                   //! Number of identified  pi0/eta vs pT
122   TH1F         * fhE   ;                   //! Number of identified  pi0/eta vs E
123   TH2F         * fhEEta  ;                 //! E vs eta of identified  pi0/eta 
124   TH2F         * fhEPhi  ;                 //! E vs phi of identified  pi0/eta 
125   TH2F         * fhEtaPhi  ;               //! eta vs phi of identified  pi0/eta 
126
127   TH1F         * fhPtDecay  ;              //! Number of identified  pi0/eta decay photons vs pT
128   TH1F         * fhEDecay   ;              //! Number of identified  pi0/eta decay photons vs E
129   
130   TH2F         * fhEDispersion ;           //! E vs disp of selected cluster
131   TH2F         * fhELambda0 ;              //! E vs lambda0 of selected cluster 
132   TH2F         * fhELambda1 ;              //! E vs lambda1 of selected cluster 
133   TH2F         * fhELambda0NoTRD ;         //! E vs lambda0 of selected cluster, not behind TRD 
134   TH2F         * fhELambda0FracMaxCellCut ;//! E vs lambda0 of selected cluster, fraction of cluster energy in max cell cut 
135   TH2F         * fhEFracMaxCell ;          //! E vs frac max cell of selected cluster 
136   TH2F         * fhEFracMaxCellNoTRD ;     //! E vs frac max cell of selected cluster, not behind TRD  
137   TH2F         * fhENCells;                //! E vs N cells in selected cluster
138   TH2F         * fhETime;                  //! E vs Time of selected cluster 
139   TH2F         * fhEPairDiffTime;          //! E vs Pair of clusters time difference vs E
140   
141   TH2F         * fhDispEtaE ;              //! shower dispersion in eta direction
142   TH2F         * fhDispPhiE ;              //! shower dispersion in phi direction
143   TH2F         * fhLambda0DispEta[7] ;     //! shower shape correlation l0 vs disp eta
144   TH2F         * fhLambda0DispPhi[7] ;     //! shower shape correlation l0 vs disp phi
145   TH2F         * fhSumEtaE ;               //! shower dispersion in eta direction
146   TH2F         * fhSumPhiE ;               //! shower dispersion in phi direction
147   TH2F         * fhSumEtaPhiE ;            //! shower dispersion in eta and phi direction
148   TH2F         * fhDispEtaPhiDiffE ;       //! shower dispersion eta - phi
149   TH2F         * fhSphericityE ;           //! shower sphericity in eta vs phi
150   TH2F         * fhDispEtaDispPhi[7] ;     //! shower dispersion in eta direction vs phi direction for 5 E bins [0-2],[2-4],[4-6],[6-10],[> 10]
151   TH2F         * fhAsymmetryE ;            //! E asymmetry of 2 splitted clusters vs cluster E
152   TH2F         * fhAsymmetryLambda0[7] ;   //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
153   TH2F         * fhAsymmetryDispEta[7] ;   //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
154   TH2F         * fhAsymmetryDispPhi[7] ;   //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
155
156   //MC histograms
157   
158   TH2F         * fhEMCLambda0[6] ;            //! E vs lambda0 of pi0 pairs but really from MC particle
159   TH2F         * fhEMCLambda1[6] ;            //! E vs lambda1 of pi0 pairs but really from MC particle
160   TH2F         * fhEMCDispersion[6] ;         //! E vs dispersion of pi0 pairs but really from MC particle
161   TH2F         * fhEMCLambda0NoTRD[6] ;         //! E vs lambda0 of pi0 pairs but really from MC particle, not behind TRD
162   TH2F         * fhEMCLambda0FracMaxCellCut[6] ;//! E vs lambda0 of pi0 pairs but really from MC particle, fraction of cluster energy in max cell cut
163   TH2F         * fhEMCFracMaxCell[6] ;        //! E vs fraction of max cell 
164   
165   TH2F         * fhMCEDispEta[6] ;            //! shower dispersion in eta direction
166   TH2F         * fhMCEDispPhi[6] ;            //! shower dispersion in phi direction
167   TH2F         * fhMCLambda0DispEta[7][6] ;   //! shower shape correlation l0 vs disp eta
168   TH2F         * fhMCLambda0DispPhi[7][6] ;   //! shower shape correlation l0 vs disp phi
169   TH2F         * fhMCESumEtaPhi[6] ;          //! shower dispersion in eta vs phi direction
170   TH2F         * fhMCEDispEtaPhiDiff[6] ;     //! shower dispersion in eta -phi direction
171   TH2F         * fhMCESphericity[6] ;         //! shower sphericity, eta vs phi
172   TH2F         * fhMCDispEtaDispPhi[7][6] ;   //! shower dispersion in eta direction vs phi direction for 5 E bins [0-2],[2-4],[4-6],[6-10],[> 10]
173   TH2F         * fhMCEAsymmetry[6] ;          //! E asymmetry of 2 splitted clusters vs cluster E
174   TH2F         * fhMCAsymmetryLambda0[7][6] ; //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
175   TH2F         * fhMCAsymmetryDispEta[7][6] ; //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
176   TH2F         * fhMCAsymmetryDispPhi[7][6] ; //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
177   
178   TH1F         * fhMCPt[6];                   //! Number of identified as pi0, coming from X
179   TH2F         * fhMCPhi[6];                  //! Phi of identified as pi0, coming from X
180   TH2F         * fhMCEta[6];                  //! eta of identified as pi0, coming from X
181
182   TH1F         * fhMCPi0DecayPt;              //! SS id, clusters id as pi0, coming from single photon, pi0 decay, pt
183   TH2F         * fhMCPi0DecayPtFraction;      //! SS id, clusters id as pi0, coming from single photon, pi0 decay, pt vs pt decay / pt mother
184   TH1F         * fhMCEtaDecayPt;              //! SS id, clusters id as pi0, coming from single photon, eta decay, pt
185   TH2F         * fhMCEtaDecayPtFraction;      //! SS id, clusters id as pi0, coming from single photon, eta decay, pt vs pt decay / pt mother  
186   TH1F         * fhMCOtherDecayPt;            //! SS id, clusters id as pi0, coming from single photon, other decay, pt
187
188   TH2F         * fhMassPairMCPi0;             //! pair mass, origin is same pi0
189   TH2F         * fhMassPairMCEta;             //! pair mass, origin is same eta
190   TH2F         * fhAnglePairMCPi0;            //! pair opening angle, origin is same pi0
191   TH2F         * fhAnglePairMCEta;            //! pair opening angle, origin is same eta
192   
193   // Weight studies
194   
195   TH2F         * fhECellClusterRatio;      //! e cell / e cluster vs e cluster for selected photons
196   TH2F         * fhECellClusterLogRatio;   //! log (e cell / e cluster)  vs e cluster for selected photons
197   TH2F         * fhEMaxCellClusterRatio;   //! e max cell / e cluster vs e cluster for selected photons
198   TH2F         * fhEMaxCellClusterLogRatio;//! log (e max cell / e cluster) vs e cluster for selected photons
199   TH2F         * fhLambda0ForW0[14];       //! L0 for 7 defined w0= 3, 3.5 ... 6 for selected photons
200   //TH2F         * fhLambda1ForW0[7];        //! L1 for 7 defined w0= 3, 3.5 ... 6 for selected photons  
201   
202   // Track Matching
203   TH2F         * fhTrackMatchedDEta     ;  //! Eta distance between track and cluster vs cluster E
204   TH2F         * fhTrackMatchedDPhi     ;  //! Phi distance between track and cluster vs cluster E
205   TH2F         * fhTrackMatchedDEtaDPhi ;  //! Eta vs Phi distance between track and cluster, E cluster > 0.5 GeV
206   TH2F         * fhTrackMatchedMCParticle; //! Trace origin of matched particle
207   TH2F         * fhdEdx  ;                 //! matched track dEdx vs cluster E 
208   TH2F         * fhEOverP;                 //! matched track E cluster over P track vs cluster E
209   TH2F         * fhEOverPNoTRD;                 //! matched track E cluster over P track vs cluster E, not behind TRD 
210
211   // Local maxima
212   TH2F         * fhNLocMax;                //! number of maxima in selected clusters
213   TH2F         * fhELambda0LocMax[3] ;     //! E vs lambda0 of selected cluster, 1,2,>2 local maxima in cluster 
214   TH2F         * fhELambda1LocMax[3] ;     //! E vs lambda1 of selected cluster, 1,2,>2 local maxima in cluster 
215   TH2F         * fhEDispersionLocMax[3] ;  //! E vs lambda1 of selected cluster, 1,2,>2 local maxima in cluster 
216   TH2F         * fhEDispEtaLocMax[3] ;     //! E vs eta dispersion of selected cluster, 1,2,>2 local maxima in cluster 
217   TH2F         * fhEDispPhiLocMax[3] ;     //! E vs phi dispersion of selected cluster, 1,2,>2 local maxima in cluster 
218   TH2F         * fhESumEtaPhiLocMax[3] ;   //! E vs dispersion in eta and phi direction
219   TH2F         * fhEDispEtaPhiDiffLocMax[3] ; //! E vs dispersion eta - phi
220   TH2F         * fhESphericityLocMax[3] ;  //! E vs sphericity in eta vs phi  
221   TH2F         * fhEAsymmetryLocMax[3] ;   //! E asymmetry of 2 splitted clusters vs cluster E for different NLM
222
223   TH2F         * fhMassPairLocMax[8];      //! pair mass, origin is same pi0, combine clusters depending on number of maxima
224
225   AliAnaPi0EbE(              const AliAnaPi0EbE & pi0ebe) ; // cpy ctor
226   AliAnaPi0EbE & operator = (const AliAnaPi0EbE & pi0ebe) ; // cpy assignment
227   
228   ClassDef(AliAnaPi0EbE,18)
229 } ;
230
231
232 #endif //ALIANAPI0EBE_H
233
234
235