]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWGGA/EMCALTasks/AliAnalysisTaskPi0V2.h
changes from fzhou
[u/mrichter/AliRoot.git] / PWGGA / EMCALTasks / AliAnalysisTaskPi0V2.h
1 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
2  * See cxx source for full Copyright notice                               */
3
4 /* $Id: AliAnalysisTaskPi0V2.h 45956 2010-12-10 12:55:37Z agheata $ */
5 /* AliAnalysisTaskPi0V2.h
6  *
7  * Template task producing a P_t spectrum and pseudorapidity distribution.
8  * Includes explanations of physics and primary track selections
9  *
10  * Based on tutorial example from offline pages
11  * Edited by Arvinder Palaha
12  */
13 #ifndef ALIANALYSISTASKPI0V2_H
14 #define ALIANALYSISTASKPI0V2_H
15
16 class TH1F;
17 class TH2F;
18 class TH3F;
19 class TList;
20 class AliESDCaloCluster;
21 class AliVCluster;
22 class AliESDtrackCuts;
23 class AliESDEvent;
24 class THnSparse;
25 class TClonesArray;
26 class TString;
27
28 #ifndef ALIANALYSISTASKSE_H
29 #include "AliAnalysisTaskSE.h"
30 #endif
31
32 class AliAnalysisTaskPi0V2 : public AliAnalysisTaskSE {
33  public:
34     AliAnalysisTaskPi0V2();
35     AliAnalysisTaskPi0V2(const char *name);
36     virtual ~AliAnalysisTaskPi0V2();
37     
38     virtual void     UserCreateOutputObjects();
39     virtual void     UserExec(Option_t *option);
40     virtual void     Terminate(Option_t *);
41     Double_t            GetCrossEnergy(const AliVCluster *cluster, Short_t &idmax) const;
42     Double_t            GetMaxCellEnergy(const AliVCluster *cluster, Short_t &id) const;
43     Bool_t              IsWithinFiducialVolume(Short_t id) const;
44     Bool_t              IsGoodCluster(const AliVCluster *c) const;
45     Bool_t              IsGoodClusterV1(const AliVCluster *c) const;
46     Bool_t              IsGoodPion(const TLorentzVector& p1, const TLorentzVector& p2) const;
47     void                FillPion(const TLorentzVector& p1, const TLorentzVector& p2, Double_t EPV0r, Double_t EPV0A, Double_t EPV0C, Double_t EPTPC);
48     void                FillCluster(const TLorentzVector& p1, Double_t EPV0r, Double_t EPV0A, Double_t EPV0C, Double_t EPTPC);
49     void                GetMom(TLorentzVector& p, const AliVCluster *c, Double_t *vertex);              
50     void                SetEventMethod(Double_t e )     { fEvtSelect  =e ;}
51     void                SetVtxCut(Double_t v )          { fVtxCut     =v ;}
52     void                SetClusNcell(Double_t c )       { fNcellCut   =c ;}
53     void                SetClusE(Double_t e )           { fECut       =e ;}
54     void                SetClusEta(Double_t e )         { fEtaCut     =e ;}
55     void                SetClusM02(Double_t m )         { fM02Cut     =m ;}
56     void                SetPi0Asy(Double_t a )          { fPi0AsyCut  =a ;}
57     void                SetTracksName(const char *n)    { fTracksName =n ;}
58     void                SetTrigClass(const char *n)     { fTrigClass  =n ;} 
59     void                SetV1ClusName(const char *n)    { fV1ClusName =n ;} 
60     void                SetV2ClusName(const char *n)    { fV2ClusName =n ;} 
61     Int_t               ConvertToInternalRunNumber(Int_t n);
62     void                FillEPQA();
63     void                SetIsV1Clus(Bool_t e)           { isV1Clus   =e  ;}
64     
65
66     
67  private:
68     TList                       *fOutput;               //! Output list
69     AliESDEvent                 *fESD;                  //!ESD object
70     TString                     fTracksName;            // name of track collection
71     TString                     fV1ClusName;            // name of V1 Clus collection
72     TString                     fV2ClusName;            // name of V1 Clus collection
73     TString                     fTrigClass;             // trigger class name for event selection
74     TClonesArray                *fTracks;               //! pico tracks specific for Skim ESD
75     TClonesArray                *fV1Clus;               //! Cluster Array for V1
76     TClonesArray                *fV2Clus;               //! Cluster Array for V2
77     Int_t                       fRunNumber;             //! Run numbers
78     Double_t                    fEvtSelect;             // 1 = MB+Semi+Central, 2 = MB+Semi, 3 = MB;
79     Double_t                    fVtxCut;                // vertex cut
80     Double_t                    fNcellCut;              // N cells Cut
81     Double_t                    fECut;                  // Cluster E cut
82     Double_t                    fEtaCut;                // Cluster Eta Cut
83     Double_t                    fM02Cut;                // Cluster long axis cut
84     Bool_t                      fPi0AsyCut;             // pion Asymetry cut 0=off 1=on
85     Bool_t                      isV1Clus;               // pion Asymetry cut 0=off 1=on
86     Double_t                    fCentrality;            //! Centrality
87     Double_t                    fEPTPC;                 //! Evt plane TPC
88     Double_t                    fEPTPCreso;             //! resolution of TPC method
89     Double_t                    fEPV0;                  //! EP V0
90     Double_t                    fEPV0A;                 //! EP V0A
91     Double_t                    fEPV0C;                 //! EP V0C
92     Double_t                    fEPV0Ar;                //! EP V0A reduced
93     Double_t                    fEPV0Cr;                //! EP V0C reduced
94     Double_t                    fEPV0r;                 //! EP V0 reduced
95     Double_t                    fEPV0AR4;               //! EP V0A ring4 only
96     Double_t                    fEPV0AR5;               //! EP V0A ring5 only
97     Double_t                    fEPV0AR6;               //! EP V0A ring6 only
98     Double_t                    fEPV0AR7;               //! EP V0A ring7 only
99     Double_t                    fEPV0CR0;               //! EP V0C ring0 only   
100     Double_t                    fEPV0CR1;               //! EP V0C ring1 only   
101     Double_t                    fEPV0CR2;               //! EP V0C ring2 only   
102     Double_t                    fEPV0CR3;               //! EP V0C ring3 only   
103
104     TH1F                        *hEvtCount;             //!
105     TH1F                        *hAllcentV0;            //!
106     TH1F                        *hAllcentV0r;           //!
107     TH1F                        *hAllcentV0A;           //!
108     TH1F                        *hAllcentV0C;           //!
109     TH1F                        *hAllcentTPC;           //!
110   
111     TH2F                        *h2DcosV0r;             //! QA for cos(Phi) V0r vs Run NUmber
112     TH2F                        *h2DsinV0r;             //! QA for cos(Phi) V0r vs Run NUmber
113     TH2F                        *h2DcosV0A;             //!
114     TH2F                        *h2DsinV0A;             //!
115     TH2F                        *h2DcosV0C;             //!
116     TH2F                        *h2DsinV0C;             //!
117     TH2F                        *h2DcosTPC;             //!
118     TH2F                        *h2DsinTPC;             //!
119
120     TH2F                        *hEPTPC;                //! 2-D histo EPTPC  vs cent
121     TH2F                        *hresoTPC;              //! 2-D histo TPC resolution vs cent
122     TH2F                        *hEPV0;                 //! 2-D histo EPV0   vs cent
123     TH2F                        *hEPV0A;                //! 2-D histo EPV0A  vs cent
124     TH2F                        *hEPV0C;                //! 2-D histo EPV0C  vs cent
125     TH2F                        *hEPV0Ar;               //! 2-D histo EPV0Ar vs cent
126     TH2F                        *hEPV0Cr;               //! 2-D histo EPV0Cr vs cent
127     TH2F                        *hEPV0r;                //! 2-D histo EPV0r  vs cent
128     TH2F                        *hEPV0AR4;              //! 2-D histo EPV0AR4 vs cent
129     TH2F                        *hEPV0AR7;              //! 2-D histo EPV0AR7 vs cent
130     TH2F                        *hEPV0CR0;              //! 2-D histo EPV0AR0 vs cent
131     TH2F                        *hEPV0CR3;              //! 2-D histo EPV0AR3 vs cent
132
133     TH2F                        *hdifV0Ar_V0Cr;         //! 2-D histo diff V0Ar, V0Cr vs cent
134     TH2F                        *hdifV0A_V0CR0;         //! 2-D histo diff V0A, V0CR0 vs cent
135     TH2F                        *hdifV0A_V0CR3;         //! 2-D histo diff V0A, V0CR3 vs cent
136     TH2F                        *hdifV0ACR0_V0CR3;      //! 2-D histo diff V0CR0, V0CR3 vs cent
137     TH2F                        *hdifV0C_V0AR4;         //! 2-D histo diff V0C, V0AR4 vs cent
138     TH2F                        *hdifV0C_V0AR7;         //! 2-D histo diff V0C, V0AR7 vs cent
139     TH2F                        *hdifV0AR4_V0AR7;       //! 2-D histo diff V0AR7, V0AR4 vs cent
140
141     TH2F                        *hdifV0A_V0C;           //! 2-D histo diff V0A - V0C
142     TH2F                        *hdifV0A_TPC;           //! 2-D histo diff V0A - TPC
143     TH2F                        *hdifV0C_TPC;           //! 2-D histo diff V0C - TPC
144     TH2F                        *hdifV0C_V0A;           //! 2-D histo diff V0C - V0A
145
146     TH2F                        *hM02vsPtA;             //! 2-D histo clus M02 vs Pt before cut
147     TH2F                        *hM02vsPtB;             //! 2-D histo clus M02 vs Pt after cut
148     TH2F                        *hClusDxDZA;            //! 2-D histo clus Dx vs Dz before
149     TH2F                        *hClusDxDZB;            //! 2-D histo clus Dx vs Dz after
150
151     TH3F                        *hdifEMC_EPV0;          //! 3-D histo dif phi in EMC with EPV0
152     TH3F                        *hdifEMC_EPV0A;         //! 3-D histo dif phi in EMC with EPV0A
153     TH3F                        *hdifEMC_EPV0C;         //! 3-D histo dif phi in EMC with EPV0C
154
155     TH3F                        *hdifful_EPV0;          //! 3-D histo dif phi in full with EPV0
156     TH3F                        *hdifful_EPV0A;         //! 3-D histo dif phi in full with EPV0A
157     TH3F                        *hdifful_EPV0C;         //! 3-D histo dif phi in full with EPV0C
158
159     TH3F                        *hdifout_EPV0;          //! 3-D histo dif phi out EMC with EPV0
160     TH3F                        *hdifout_EPV0A;         //! 3-D histo dif phi out EMC with EPV0A
161     TH3F                        *hdifout_EPV0C;         //! 3-D histo dif phi out EMC with EPV0C
162
163     TH3F                        *hdifEMC_EPTPC;         //! 3-D histo dif phi in EMC with EPTPC
164     TH3F                        *hdifful_EPTPC;         //! 3-D histo dif phi in full with EPTPC
165     TH3F                        *hdifout_EPTPC;         //! 3-D histo dif phi out EMC with EPTPC
166
167     TH3F                        *hdifClus_EPV0;         //! 3-D histo dif phi of Clus with V0r
168     TH3F                        *hdifClus_EPV0A;        //! 3-D histo dif phi of Clus with V0A
169     TH3F                        *hdifClus_EPV0C;        //! 3-D histo dif phi of Clus with V0C
170     TH3F                        *hdifClus_EPTPC;        //! 3-D histo dif phi of Clus with TPC
171
172     THnSparse                   *fHEPV0r;               //! Flow 4-D Histo
173     THnSparse                   *fHEPV0A;               //! Flow 4-D Histo
174     THnSparse                   *fHEPV0C;               //! Flow 4-D Histo
175     THnSparse                   *fHEPTPC;               //! Flow 4-D Histo
176
177     
178     
179     AliAnalysisTaskPi0V2(const AliAnalysisTaskPi0V2&); // not implemented
180     AliAnalysisTaskPi0V2& operator=(const AliAnalysisTaskPi0V2&); // not implemented
181     
182     ClassDef(AliAnalysisTaskPi0V2, 4); // example of analysis
183 };
184
185 #endif
186