Update from Alberica. Addition of VZERO equalized signals and ZNC.
[u/mrichter/AliRoot.git] / ANALYSIS / AliCentralitySelectionTask.h
1 #ifndef ALICENTRALITYSELECTIONTASK_H
2 #define ALICENTRALITYSELECTIONTASK_H
3
4 /* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 //*****************************************************
8 //   Class AliCentralitySelectionTask
9 //   author: Alberica Toia
10 //*****************************************************
11
12 #include "AliAnalysisTaskSE.h"
13
14 class TFile;
15 class TH1F;
16 class TH2F;
17 class TList;
18 class TString;
19
20 class AliESDEvent;
21 class AliESDtrackCuts;
22
23 class AliCentralitySelectionTask : public AliAnalysisTaskSE {
24
25  public:
26
27   AliCentralitySelectionTask();
28   AliCentralitySelectionTask(const char *name);
29   AliCentralitySelectionTask& operator= (const AliCentralitySelectionTask& ana);
30   AliCentralitySelectionTask(const AliCentralitySelectionTask& c);
31   virtual ~AliCentralitySelectionTask();
32
33   // Implementation of interface methods
34   virtual void UserCreateOutputObjects();
35   virtual void UserExec(Option_t *option);
36   virtual void Terminate(Option_t *option);
37
38   void SetInput(const char* input)         {fAnalysisInput = input;}
39   void SetMCInput()                        {fIsMCInput = kTRUE;}
40   void DontUseScaling()                    {fUseScaling=kFALSE;}  
41   void DontUseCleaning()                   {fUseCleaning=kFALSE;}
42   void SetFillHistos()                     {fFillHistos=kTRUE; DefineOutput(1, TList::Class());
43 }
44
45  private:
46
47   Int_t SetupRun(const AliVEvent* const esd);
48   Bool_t IsOutlierV0MSPD(Float_t spd, Float_t v0, Int_t cent) const;
49   Bool_t IsOutlierV0MTPC(Int_t tracks, Float_t v0, Int_t cent) const;
50   Bool_t IsOutlierV0MZDC(Float_t zdc, Float_t v0) const;
51   Bool_t IsOutlierV0MZDCECal(Float_t zdc, Float_t v0) const;
52
53   TString  fAnalysisInput;      // "ESD", "AOD"
54   Bool_t   fIsMCInput;          // true when input is MC
55   Int_t    fCurrentRun;         // current run number
56   Bool_t   fUseScaling;         // flag to use scaling 
57   Bool_t   fUseCleaning;        // flag to use cleaning  
58   Bool_t   fFillHistos;         // flag to fill the QA histos
59   Float_t  fV0MScaleFactor;     // scale factor V0M
60   Float_t  fSPDScaleFactor;     // scale factor SPD
61   Float_t  fTPCScaleFactor;     // scale factor TPC
62   Float_t  fV0MScaleFactorMC;   // scale factor V0M for MC
63   Float_t  fV0MSPDOutlierPar0;  // outliers parameter
64   Float_t  fV0MSPDOutlierPar1;  // outliers parameter
65   Float_t  fV0MTPCOutlierPar0;  // outliers parameter
66   Float_t  fV0MTPCOutlierPar1;  // outliers parameter
67   Float_t  fV0MSPDSigmaOutlierPar0;  // outliers parameter
68   Float_t  fV0MSPDSigmaOutlierPar1;  // outliers parameter
69   Float_t  fV0MSPDSigmaOutlierPar2;  // outliers parameter
70   Float_t  fV0MTPCSigmaOutlierPar0;  // outliers parameter
71   Float_t  fV0MTPCSigmaOutlierPar1;  // outliers parameter
72   Float_t  fV0MTPCSigmaOutlierPar2;  // outliers parameter                                                 
73   Float_t  fV0MZDCOutlierPar0;       // outliers parameter
74   Float_t  fV0MZDCOutlierPar1;       // outliers parameter
75   Float_t  fV0MZDCEcalOutlierPar0;   // outliers parameter
76   Float_t  fV0MZDCEcalOutlierPar1;   // outliers parameter
77
78   AliESDtrackCuts* fTrackCuts;  //! optional track cuts
79   AliESDtrackCuts* fEsdTrackCuts;  //! optional track cuts
80   AliESDtrackCuts* fEsdTrackCutsExtra1;  //! optional track cuts
81   AliESDtrackCuts* fEsdTrackCutsExtra2;  //! optional track cuts
82
83   Float_t  fZVCut;              //! z-vertex cut (in cm)
84   Float_t  fOutliersCut;        //! outliers cut (in n-sigma)
85   Int_t    fQuality;            //! quality for centrality determination
86
87   Bool_t   fIsSelected;         //! V0BG rejection
88
89   Bool_t   fMSL;                //! 
90   Bool_t   fMSH;                //! 
91   Bool_t   fMUL;                //! 
92   Bool_t   fMLL;                //! 
93
94   Bool_t   fEJE;                //! 
95   Bool_t   fEGA;                //! 
96   Bool_t   fPHS;                //! 
97
98   Bool_t   fMB;                 //! if the event is MB
99   Bool_t   fCVHN;               //! if the event is central trigger
100   Bool_t   fCVLN;               //! if the event is semicentral trigger
101   Bool_t   fCVHNbit;            //! if the event is central trigger
102   Bool_t   fCVLNbit;            //! if the event is semicentral trigger
103   Bool_t   fCCENT;              //! if the event is central trigger
104   Bool_t   fCSEMI;              //! if the event is semicentral trigger
105   Bool_t   fCCENTbit;           //! if the event is central trigger
106   Bool_t   fCSEMIbit;           //! if the event is semicentral trigger
107
108   Float_t  fCentV0M;            // percentile centrality from V0
109   Float_t  fCentV0A;            // percentile centrality from V0A
110   Float_t  fCentV0C;            // percentile centrality from V0C
111   Float_t  fCentV0MEq;          // percentile centrality from V0 equalized channel
112   Float_t  fCentV0AEq;          // percentile centrality from V0A equalized channel
113   Float_t  fCentV0CEq;          // percentile centrality from V0C equalized channel
114   Float_t  fCentFMD;            // percentile centrality from FMD
115   Float_t  fCentTRK;            // percentile centrality from tracks
116   Float_t  fCentTKL;            // percentile centrality from tracklets
117   Float_t  fCentCL0;            // percentile centrality from clusters in layer 0
118   Float_t  fCentCL1;            // percentile centrality from clusters in layer 1
119   Float_t  fCentCND;            // percentile centrality from candle 
120   Float_t  fCentZNA;            // percentile centrality from ZNA 
121   Float_t  fCentZNC;            // percentile centrality from ZNC 
122   Float_t  fCentNPA;            // percentile centrality from Npart (MC) 
123   Float_t  fCentV0MvsFMD;       // percentile centrality from V0 vs FMD
124   Float_t  fCentTKLvsV0M;       // percentile centrality from tracklets vs V0
125   Float_t  fCentZEMvsZDC;       // percentile centrality from ZEM vs ZDC
126
127   Float_t  fCentV0Mtrue;        // percentile centrality from true (sim) V0A+V0C
128   Float_t  fCentV0Atrue;        // percentile centrality from true (sim) V0A
129   Float_t  fCentV0Ctrue;        // percentile centrality from true (sim) V0C
130   Float_t  fCentV0MEqtrue;      // percentile centrality from true (sim) V0A+V0C equalized channel
131   Float_t  fCentV0AEqtrue;      // percentile centrality from true (sim) V0A equalized channel
132   Float_t  fCentV0CEqtrue;      // percentile centrality from true (sim) V0C equalized channel
133   Float_t  fCentFMDtrue;        // percentile centrality from true (sim) FMD
134   Float_t  fCentTRKtrue;        // percentile centrality from true (sim) tracks
135   Float_t  fCentTKLtrue;        // percentile centrality from true (sim) tracklets
136   Float_t  fCentCL0true;        // percentile centrality from true (sim) Clusters in layer 0
137   Float_t  fCentCL1true;        // percentile centrality from true (sim) Clusters in layer 1
138   Float_t  fCentCNDtrue;        // percentile centrality from true (sim) tracks (candle condition)
139   Float_t  fCentZNAtrue;        // percentile centrality from true (sim) ZNA
140   Float_t  fCentZNCtrue;        // percentile centrality from true (sim) ZNC
141
142
143   TH1F    *fHtempV0M;           // histogram with centrality vs multiplicity using V0
144   TH1F    *fHtempV0A;           // histogram with centrality vs multiplicity using V0A
145   TH1F    *fHtempV0C;           // histogram with centrality vs multiplicity using V0C
146   TH1F    *fHtempV0MEq;         // histogram with centrality vs multiplicity using V0  equalized channel
147   TH1F    *fHtempV0AEq;         // histogram with centrality vs multiplicity using V0A equalized channel
148   TH1F    *fHtempV0CEq;         // histogram with centrality vs multiplicity using V0C equalized channel
149   TH1F    *fHtempFMD;           // histogram with centrality vs multiplicity using FMD
150   TH1F    *fHtempTRK;           // histogram with centrality vs multiplicity using tracks
151   TH1F    *fHtempTKL;           // histogram with centrality vs multiplicity using tracklets
152   TH1F    *fHtempCL0;           // histogram with centrality vs multiplicity using clusters in layer 0
153   TH1F    *fHtempCL1;           // histogram with centrality vs multiplicity using clusters in layer 1
154   TH1F    *fHtempCND;           // histogram with centrality vs multiplicity using candle
155   TH1F    *fHtempZNA;           // histogram with centrality vs multiplicity using ZNA
156   TH1F    *fHtempZNC;           // histogram with centrality vs multiplicity using ZNC
157   TH1F    *fHtempV0MvsFMD;      // histogram with centrality vs multiplicity using V0 vs FMD   
158   TH1F    *fHtempTKLvsV0M;      // histogram with centrality vs multiplicity using tracklets vs V0
159   TH2F    *fHtempZEMvsZDC;      // histogram with centrality vs multiplicity using ZEM vs ZDC 
160   TH1F    *fHtempNPA;           // histogram with centrality vs multiplicity using Npart
161
162   TH1F    *fHtempV0Mtrue;       // histogram with centrality true (sim) vs multiplicity using V0
163   TH1F    *fHtempV0Atrue;       // histogram with centrality true (sim) vs multiplicity using V0A
164   TH1F    *fHtempV0Ctrue;       // histogram with centrality true (sim) vs multiplicity using V0C
165   TH1F    *fHtempV0MEqtrue;     // histogram with centrality true (sim) vs multiplicity using V0 equalized channel
166   TH1F    *fHtempV0AEqtrue;     // histogram with centrality true (sim) vs multiplicity using V0A equalized channel
167   TH1F    *fHtempV0CEqtrue;     // histogram with centrality true (sim) vs multiplicity using V0C equalized channel
168   TH1F    *fHtempFMDtrue;       // histogram with centrality true (sim) vs multiplicity using FMD
169   TH1F    *fHtempTRKtrue;       // histogram with centrality true (sim) vs multiplicity using tracks
170   TH1F    *fHtempTKLtrue;       // histogram with centrality true (sim) vs multiplicity using tracklets
171   TH1F    *fHtempCL0true;       // histogram with centrality true (sim) vs multiplicity using clusters in layer 0
172   TH1F    *fHtempCL1true;       // histogram with centrality true (sim) vs multiplicity using clusters in layer 1
173   TH1F    *fHtempCNDtrue;       // histogram with centrality true (sim) vs multiplicity using candle
174   TH1F    *fHtempZNAtrue;       // histogram with centrality true (sim) vs multiplicity using ZNA
175   TH1F    *fHtempZNCtrue;       // histogram with centrality true (sim) vs multiplicity using ZNC
176
177   TList   *fOutputList; // output list
178   
179
180   TH1F *fHOutCentV0M    ;       //control histogram for centrality
181   TH1F *fHOutCentV0A    ;       //control histogram for centrality
182   TH1F *fHOutCentV0C    ;       //control histogram for centrality
183   TH1F *fHOutCentV0MEq  ;       //control histogram for centrality
184   TH1F *fHOutCentV0AEq  ;       //control histogram for centrality
185   TH1F *fHOutCentV0CEq  ;       //control histogram for centrality
186   TH1F *fHOutCentV0MCVHN;       //control histogram for centrality
187   TH1F *fHOutCentV0MCVLN;       //control histogram for centrality
188   TH1F *fHOutCentV0MCVHNinMB;   //control histogram for centrality
189   TH1F *fHOutCentV0MCVLNinMB;   //control histogram for centrality
190   TH1F *fHOutCentV0MCCENT;      //control histogram for centrality
191   TH1F *fHOutCentV0MCSEMI;      //control histogram for centrality
192   TH1F *fHOutCentV0MCCENTinMB;  //control histogram for centrality
193   TH1F *fHOutCentV0MCSEMIinMB;  //control histogram for centrality
194   TH1F *fHOutCentV0MMSL;        //control histogram for centrality
195   TH1F *fHOutCentV0MMSH;        //control histogram for centrality
196   TH1F *fHOutCentV0MMUL;        //control histogram for centrality
197   TH1F *fHOutCentV0MMLL;        //control histogram for centrality
198   TH1F *fHOutCentV0MEJE;        //control histogram for centrality
199   TH1F *fHOutCentV0MEGA;        //control histogram for centrality
200   TH1F *fHOutCentV0MPHS;        //control histogram for centrality
201   TH1F *fHOutCentV0MMSLinMB;    //control histogram for centrality
202   TH1F *fHOutCentV0MMSHinMB;    //control histogram for centrality
203   TH1F *fHOutCentV0MMULinMB;    //control histogram for centrality
204   TH1F *fHOutCentV0MMLLinMB;    //control histogram for centrality
205   TH1F *fHOutCentV0MEJEinMB;    //control histogram for centrality
206   TH1F *fHOutCentV0MEGAinMB;    //control histogram for centrality
207   TH1F *fHOutCentV0MPHSinMB;    //control histogram for centrality
208   TH1F *fHOutCentFMD     ;      //control histogram for centrality
209   TH1F *fHOutCentTRK     ;      //control histogram for centrality
210   TH1F *fHOutCentTKL     ;      //control histogram for centrality
211   TH1F *fHOutCentCL0     ;      //control histogram for centrality
212   TH1F *fHOutCentCL1     ;      //control histogram for centrality
213   TH1F *fHOutCentCND     ;      //control histogram for centrality
214   TH1F *fHOutCentNPA     ;      //control histogram for centrality
215   TH1F *fHOutCentZNA     ;      //control histogram for centrality
216   TH1F *fHOutCentZNC     ;      //control histogram for centrality
217   TH1F *fHOutCentV0MvsFMD;      //control histogram for centrality
218   TH1F *fHOutCentTKLvsV0M;      //control histogram for centrality
219   TH1F *fHOutCentZEMvsZDC;      //control histogram for centrality
220   TH2F *fHOutCentV0MvsCentCL1;  //control histogram for centrality
221   TH2F *fHOutCentV0MvsCentTRK;  //control histogram for centrality
222   TH2F *fHOutCentTRKvsCentCL1;  //control histogram for centrality
223   TH2F *fHOutCentV0MvsCentZDC;  //control histogram for centrality
224   TH2F *fHOutCentV0AvsCentV0C;  //control histogram for centrality
225   TH2F *fHOutCentV0AvsCentTRK;  //control histogram for centrality
226   TH2F *fHOutCentV0AvsCentCND;  //control histogram for centrality
227   TH2F *fHOutCentV0AvsCentCL1;  //control histogram for centrality
228   TH2F *fHOutCentV0CvsCentTRK;  //control histogram for centrality
229   TH2F *fHOutCentV0CvsCentCND;  //control histogram for centrality
230   TH2F *fHOutCentV0CvsCentCL1;  //control histogram for centrality
231   TH2F *fHOutCentNPAvsCentV0A;  //control histogram for centrality
232   TH2F *fHOutCentNPAvsCentV0C;  //control histogram for centrality
233   TH2F *fHOutCentNPAvsCentTRK;  //control histogram for centrality
234   TH2F *fHOutCentNPAvsCentCND;  //control histogram for centrality
235   TH2F *fHOutCentNPAvsCentCL1;  //control histogram for centrality
236   TH2F *fHOutCentZNAvsCentV0A;  //control histogram for centrality
237   TH2F *fHOutCentZNAvsCentV0C;  //control histogram for centrality
238   TH2F *fHOutCentZNAvsCentTRK;  //control histogram for centrality
239   TH2F *fHOutCentZNAvsCentCND;  //control histogram for centrality
240   TH2F *fHOutCentZNAvsCentCL1;  //control histogram for centrality
241
242   TH2F *fHOutMultV0AC;          //control histogram for multiplicity
243   TH1F *fHOutMultV0M ;          //control histogram for multiplicity
244   TH1F *fHOutMultV0A ;          //control histogram for multiplicity
245   TH1F *fHOutMultV0C ;          //control histogram for multiplicity
246   TH1F *fHOutMultV0MEq ;        //control histogram for multiplicity
247   TH1F *fHOutMultV0AEq ;        //control histogram for multiplicity
248   TH1F *fHOutMultV0CEq ;        //control histogram for multiplicity
249   TH1F *fHOutMultV0Mnc ;        //control histogram for multiplicity
250   TH1F *fHOutMultV0Anc ;        //control histogram for multiplicity
251   TH1F *fHOutMultV0Cnc ;        //control histogram for multiplicity
252   TH1F *fHOutMultV0O ;          //control histogram for multiplicity
253   TH2F *fHOutMultV0Cells ;      //control histogram for multiplicity
254   TH1F *fHOutMultFMD ;          //control histogram for multiplicity
255   TH1F *fHOutMultTRK ;          //control histogram for multiplicity
256   TH1F *fHOutMultTKL ;          //control histogram for multiplicity
257   TH1F *fHOutMultCL0 ;          //control histogram for multiplicity
258   TH1F *fHOutMultCL1 ;          //control histogram for multiplicity
259   TH1F *fHOutMultCND ;          //control histogram for multiplicity
260   TH1F *fHOutMultNPA ;          //control histogram for multiplicity
261   TH1F *fHOutMultZNA ;          //control histogram for multiplicity
262   TH1F *fHOutMultZNC ;          //control histogram for multiplicity
263
264   TH2F *fHOutMultV0MvsZDN;      //control histogram for multiplicity
265   TH2F *fHOutMultZEMvsZDN;      //control histogram for multiplicity
266   TH2F *fHOutMultV0MvsZDC;      //control histogram for multiplicity
267   TH2F *fHOutMultZEMvsZDC;      //control histogram for multiplicity
268   TH2F *fHOutMultZEMvsZDCw;     //control histogram for multiplicity
269   TH2F *fHOutMultV0MvsCL1;      //control histogram for multiplicity
270   TH2F *fHOutMultV0MvsTRK;      //control histogram for multiplicity
271   TH2F *fHOutMultTRKvsCL1;      //control histogram for multiplicity
272   TH2F *fHOutMultV0MvsV0O;      //control histogram for multiplicity
273   TH2F *fHOutMultV0OvsCL1;      //control histogram for multiplicity
274   TH2F *fHOutMultV0OvsTRK;      //control histogram for multiplicity
275   TH2F *fHOutMultCL1vsTKL;      //control histogram for multiplicity
276
277   TH1F *fHOutCentV0Mqual1     ; //control histogram for centrality quality 1
278   TH1F *fHOutCentTRKqual1     ; //control histogram for centrality quality 1
279   TH1F *fHOutCentCL1qual1     ; //control histogram for centrality quality 1
280   TH2F *fHOutMultV0MvsCL1qual1; //control histogram for multiplicity quality 1
281   TH2F *fHOutMultV0MvsTRKqual1; //control histogram for multiplicity quality 1
282   TH2F *fHOutMultTRKvsCL1qual1; //control histogram for multiplicity quality 1
283
284   TH1F *fHOutCentV0Mqual2     ; //control histogram for centrality quality 2
285   TH1F *fHOutCentTRKqual2     ; //control histogram for centrality quality 2
286   TH1F *fHOutCentCL1qual2     ; //control histogram for centrality quality 2
287   TH2F *fHOutMultV0MvsCL1qual2; //control histogram for multiplicity quality 2
288   TH2F *fHOutMultV0MvsTRKqual2; //control histogram for multiplicity quality 2
289   TH2F *fHOutMultTRKvsCL1qual2; //control histogram for multiplicity quality 2
290
291   TH1F *fHOutQuality ;          //control histogram for quality
292   TH1F *fHOutVertex ;           //control histogram for vertex SPD
293   TH1F *fHOutVertexT0 ;         //control histogram for vertex T0
294
295   ClassDef(AliCentralitySelectionTask, 26); 
296 };
297
298 #endif
299