Added y shift to the histograms. Needed for the acquisiton of efficiencies.
[u/mrichter/AliRoot.git] / PWGLF / STRANGENESS / LambdaK0 / AliAnalysisTaskExtractPerformanceV0.h
1 /**************************************************************************
2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
17 //
18 // Modified version of AliAnalysisTaskCheckCascade.h
19 // Used bits of code from AliAnalysisTaskCheckPerformanceStrange
20 //
21 // --- David Dobrigkeit Chinellato
22 //
23 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
24
25 #ifndef ALIANALYSISTASKEXTRACTPERFORMANCEV0_H
26 #define ALIANALYSISTASKEXTRACTPERFORMANCEV0_H
27
28 class TList;
29 class TH1F;
30 class TH2F;
31 class TH3F;
32 class TVector3;
33 class THnSparse;
34
35 class AliESDpid;
36 class AliESDtrackCuts;
37 class AliESDEvent;
38 class AliPhysicsSelection;
39 class AliCFContainer;
40
41 //#include "TString.h"
42 //#include "AliESDtrackCuts.h"
43 #include "AliAnalysisTaskSE.h"
44
45 class AliAnalysisTaskExtractPerformanceV0 : public AliAnalysisTaskSE {
46  public:
47   AliAnalysisTaskExtractPerformanceV0();
48   AliAnalysisTaskExtractPerformanceV0(const char *name);
49   virtual ~AliAnalysisTaskExtractPerformanceV0();
50   
51   virtual void   UserCreateOutputObjects();
52   virtual void   UserExec(Option_t *option);
53   virtual void   Terminate(Option_t *);
54   Double_t MyRapidity(Double_t rE, Double_t rPz) const;
55   void CheckChargeV0(AliESDv0 *thisv0);
56
57   void SetIsNuclear           (Bool_t lIsNuclear   = kTRUE ) { fkIsNuclear   = lIsNuclear;   }
58   void SetINT7Trigger         (Bool_t lSwitchINT7  = kTRUE ) { fkSwitchINT7   = lSwitchINT7; }
59   void SetUseOnTheFly         (Bool_t lUseOnTheFly = kTRUE ) { fkUseOnTheFly = lUseOnTheFly; }
60   void SetTakeAllTracks       (Bool_t lTakeAllTracks = kTRUE ) { fkTakeAllTracks = lTakeAllTracks; }
61   void SetpARapidityShift     (Double_t lRapShift = 0.465 ) { fpArapidityShift = lRapShift; }
62   
63  private:
64         // Note : In ROOT, "//!" means "do not stream the data from Master node to Worker node" ...
65         // your data member object is created on the worker nodes and streaming is not needed.
66         // http://root.cern.ch/download/doc/11InputOutput.pdf, page 14
67   TList  *fListHistV0;  //! List of Cascade histograms
68   TTree  *fTree;              //! Output Tree, V0
69
70   AliPIDResponse *fPIDResponse;     // PID response object
71   AliESDtrackCuts *fESDtrackCuts;   // ESD track cuts used for primary track definition
72   Double_t fpArapidityShift; //pA rapidity shift (should be 0.465, usually)
73
74   //Objects Controlling Task Behaviour 
75   
76   Bool_t fkIsNuclear;   //if true, replace multiplicity est. by centrality (default FALSE) 
77   Bool_t fkSwitchINT7 ; //if true, skip FASTOnly (default FALSE)
78   Bool_t fkUseOnTheFly; //if true, will use On-the-fly V0s instead of Offline V0s (default FALSE)
79   Bool_t fkTakeAllTracks; // if true, no TPC crossed rows and ratio cut
80
81 //===========================================================================================
82 //   Variables for Tree, V0s
83 //===========================================================================================
84    Int_t    fTreeVariablePrimaryStatus;      //!
85    Int_t    fTreeVariablePrimaryStatusMother;      //!
86    Float_t fTreeVariableChi2V0;             //!
87    Float_t fTreeVariableDcaV0Daughters; //!
88    Float_t fTreeVariableDcaV0ToPrimVertex; //!
89    Float_t fTreeVariableDcaPosToPrimVertex; //!
90    Float_t fTreeVariableDcaNegToPrimVertex; //!
91    Float_t fTreeVariableV0CosineOfPointingAngle; //!
92    Float_t fTreeVariableV0Radius; //!
93    Float_t fTreeVariablePt; //!
94    Float_t fTreeVariablePtMC; //!
95    Float_t fTreeVariableRapK0Short; //!
96    Float_t fTreeVariableRapLambda; //!
97    Float_t fTreeVariableRapMC; //!
98    Float_t fTreeVariableInvMassK0s; //!
99    Float_t fTreeVariableInvMassLambda; //!
100    Float_t fTreeVariableInvMassAntiLambda; //!
101    Float_t fTreeVariableAlphaV0; //!
102    Float_t fTreeVariablePtArmV0;//!
103    Float_t fTreeVariableNegTotMomentum; //!               
104    Float_t fTreeVariablePosTotMomentum; //!
105    Float_t fTreeVariableNegTransvMomentum; //!   
106    Float_t fTreeVariablePosTransvMomentum; //!
107    Float_t fTreeVariableNegTransvMomentumMC; //!   
108    Float_t fTreeVariablePosTransvMomentumMC; //!
109    
110    Float_t fTreeVariableNSigmasPosProton; //!
111    Float_t fTreeVariableNSigmasPosPion; //! 
112    Float_t fTreeVariableNSigmasNegProton; //!
113    Float_t fTreeVariableNSigmasNegPion; //! 
114
115    Float_t fTreeVariablePtMother; //!
116    Float_t fTreeVariableV0CreationRadius; //!
117    Int_t fTreeVariablePID; //!
118    Int_t fTreeVariablePIDPositive; //!
119    Int_t fTreeVariablePIDNegative; //!
120    Int_t fTreeVariablePIDMother; //!
121    Int_t fTreeVariableIndexStatus; //!
122    Int_t fTreeVariableIndexStatusMother; //!
123
124    Int_t   fTreeVariableRunNumber; //! 
125    ULong64_t fTreeVariableEventNumber; //!
126
127    //Note: TDistOverTotMom needs a mass hypothesis to be converted to proper decaylength.
128    Float_t fTreeVariableDistOverTotMom;//!
129
130    Float_t fTreeVariablePosEta; //!
131    Float_t fTreeVariableNegEta; //!
132
133    Float_t fTreeVariableVertexZ; //!
134
135    Int_t fTreeVariableLeastNbrCrossedRows;//!
136    Float_t fTreeVariableLeastRatioCrossedRowsOverFindable;//!
137    Int_t fTreeVariableMultiplicity;//!
138    Int_t fTreeVariableMultiplicityMC;//!
139
140   Float_t fTreeVariableV0x; //!
141   Float_t fTreeVariableV0y; //!
142   Float_t fTreeVariableV0z; //!
143
144   Float_t fTreeVariableV0Px; //!
145   Float_t fTreeVariableV0Py; //!
146   Float_t fTreeVariableV0Pz; //!
147
148   Float_t fTreeVariableMCV0x; //!
149   Float_t fTreeVariableMCV0y; //!
150   Float_t fTreeVariableMCV0z; //!
151
152   Float_t fTreeVariableMCV0Px; //!
153   Float_t fTreeVariableMCV0Py; //!
154   Float_t fTreeVariableMCV0Pz; //!
155
156   Float_t fTreeVariablePVx; //!
157   Float_t fTreeVariablePVy; //!
158   Float_t fTreeVariablePVz; //!
159
160   Float_t fTreeVariableMCPVx; //!
161   Float_t fTreeVariableMCPVy; //!
162   Float_t fTreeVariableMCPVz; //!
163
164   Bool_t fTreeVariableIsNonInjected; //!
165
166 //===========================================================================================
167 //   Histograms
168 //===========================================================================================
169
170    TH1F      *fHistV0MultiplicityBeforeTrigSel;              //! V0 multiplicity distribution
171    TH1F      *fHistV0MultiplicityForTrigEvt;                 //! V0 multiplicity distribution
172    TH1F      *fHistV0MultiplicityForSelEvt;                  //! V0 multiplicity distribution
173    TH1F      *fHistV0MultiplicityForSelEvtNoTPCOnly;         //! V0 multiplicity distribution
174    TH1F      *fHistV0MultiplicityForSelEvtNoTPCOnlyNoPileup; //! V0 multiplicity distribution
175
176    TH1F      *fHistMultiplicityBeforeTrigSel;     //! multiplicity distribution      
177    TH1F      *fHistMultiplicityForTrigEvt;        //! multiplicity distribution
178    TH1F      *fHistMultiplicity;                  //! multiplicity distribution
179    TH1F      *fHistMultiplicityNoTPCOnly;         //! multiplicity distribution
180    TH1F      *fHistMultiplicityNoTPCOnlyNoPileup; //! multiplicity distribution
181
182   //Raw Data for J/Psi paper Technique
183         TH2F    *f2dHistMultiplicityVsTrueBeforeTrigSel;                //! multiplicity distribution    
184         TH2F    *f2dHistMultiplicityVsTrueForTrigEvt;                   //! multiplicity distribution
185         TH2F    *f2dHistMultiplicityVsTrue;                                                     //! multiplicity distribution
186         TH2F    *f2dHistMultiplicityVsTrueNoTPCOnly;                            //! multiplicity distribution
187         TH2F    *f2dHistMultiplicityVsTrueNoTPCOnlyNoPileup;                    //! multiplicity distribution
188
189   //Raw Data for Vertex Z position estimator change
190         TH2F    *f2dHistMultiplicityVsVertexZBeforeTrigSel;             //! multiplicity distribution    
191         TH2F    *f2dHistMultiplicityVsVertexZForTrigEvt;                        //! multiplicity distribution
192         TH2F    *f2dHistMultiplicityVsVertexZ;                                                  //! multiplicity distribution
193         TH2F    *f2dHistMultiplicityVsVertexZNoTPCOnly;                         //! multiplicity distribution
194         TH2F    *f2dHistMultiplicityVsVertexZNoTPCOnlyNoPileup;                 //! multiplicity distribution
195
196    TH1F      *fHistGenVertexZBeforeTrigSel;     //! multiplicity distribution      
197    TH1F      *fHistGenVertexZForTrigEvt;        //! multiplicity distribution
198    TH1F      *fHistGenVertexZ;                  //! multiplicity distribution
199    TH1F      *fHistGenVertexZNoTPCOnly;         //! multiplicity distribution
200    TH1F      *fHistGenVertexZNoTPCOnlyNoPileup; //! multiplicity distribution
201
202 //---> Filled At Analysis Scope
203
204    TH3F      *f3dHistPrimAnalysisPtVsYVsMultLambda;     //! Lambda
205    TH3F      *f3dHistPrimAnalysisPtVsYVsMultAntiLambda; //! AntiLambda
206    TH3F      *f3dHistPrimAnalysisPtVsYVsMultK0Short;    //! K0Short
207
208    TH3F      *f3dHistPrimAnalysisPtVsYCMSVsMultLambda;     //! Lambda
209    TH3F      *f3dHistPrimAnalysisPtVsYCMSVsMultAntiLambda; //! AntiLambda
210    TH3F      *f3dHistPrimAnalysisPtVsYCMSVsMultK0Short;    //! K0Short
211
212 //---> Containers for monte carlo information for calculating efficiency! 
213
214    TH3F      *f3dHistPrimRawPtVsYVsMultLambda;     //! Lambda
215    TH3F      *f3dHistPrimRawPtVsYVsMultAntiLambda; //! AntiLambda
216    TH3F      *f3dHistPrimRawPtVsYVsMultK0Short;    //! K0Short
217
218    TH3F      *f3dHistPrimRawPtVsYCMSVsMultLambda;     //! Lambda
219    TH3F      *f3dHistPrimRawPtVsYCMSVsMultAntiLambda; //! AntiLambda
220    TH3F      *f3dHistPrimRawPtVsYCMSVsMultK0Short;    //! K0Short
221
222    TH3F      *f3dHistPrimRawPtVsYVsMultNonInjLambda;     //! Non-injected Lambda
223    TH3F      *f3dHistPrimRawPtVsYVsMultNonInjAntiLambda; //! Non-injected AntiLambda
224    TH3F      *f3dHistPrimRawPtVsYVsMultNonInjK0Short;    //! Non-injected K0Short
225
226    TH3F      *f3dHistPrimRawPtVsYVsMultMCLambda;     //! Lambda
227    TH3F      *f3dHistPrimRawPtVsYVsMultMCAntiLambda; //! AntiLambda
228    TH3F      *f3dHistPrimRawPtVsYVsMultMCK0Short;    //! K0Short
229
230    TH3F      *f3dHistPrimRawPtVsYVsVertexZLambda;     //! Lambda
231    TH3F      *f3dHistPrimRawPtVsYVsVertexZAntiLambda; //! AntiLambda
232    TH3F      *f3dHistPrimRawPtVsYVsVertexZK0Short;    //! K0Short
233
234    TH3F      *f3dHistPrimCloseToPVPtVsYVsMultLambda;     //! Lambda
235    TH3F      *f3dHistPrimCloseToPVPtVsYVsMultAntiLambda; //! AntiLambda
236    TH3F      *f3dHistPrimCloseToPVPtVsYVsMultK0Short;    //! K0Short
237
238 //---> Filled vs Decay Length
239
240    TH3F      *f3dHistPrimRawPtVsYVsDecayLengthLambda;     //! Lambda
241    TH3F      *f3dHistPrimRawPtVsYVsDecayLengthAntiLambda; //! AntiLambda
242    TH3F      *f3dHistPrimRawPtVsYVsDecayLengthK0Short;    //! K0Short
243
244 //---> Needed for FeedDown Corrections
245
246    TH3F      *f3dHistGenPtVsYVsMultXiMinus;      //! Generated Xi- Distrib
247    TH3F      *f3dHistGenPtVsYVsMultXiPlus;       //! Generated Xi+ Distrib
248    TH3F      *f3dHistGenPtVsYVsMultOmegaMinus;      //! Generated Omega- Distrib
249    TH3F      *f3dHistGenPtVsYVsMultOmegaPlus;       //! Generated Omega+ Distrib
250
251    TH3F      *f3dHistGenSelectedPtVsYVsMultXiMinus;      //! Generated Xi- Distrib, at event selection level
252    TH3F      *f3dHistGenSelectedPtVsYVsMultXiPlus;       //! Generated Xi+ Distrib, at event selection level
253    TH3F      *f3dHistGenSelectedPtVsYVsMultOmegaMinus;      //! Generated Omega- Distrib, at event selection level
254    TH3F      *f3dHistGenSelectedPtVsYVsMultOmegaPlus;       //! Generated Omega+ Distrib, at event selection level
255
256    TH1F      *fHistPVx;                      //! PVx distrib
257    TH1F      *fHistPVy;                      //! PVy distrib
258    TH1F      *fHistPVz;                      //! PVz distrib
259    TH1F      *fHistPVxAnalysis;                      //! PVx distrib
260    TH1F      *fHistPVyAnalysis;                      //! PVy distrib
261    TH1F      *fHistPVzAnalysis;                      //! PVz distrib
262    TH1F      *fHistPVxAnalysisHasHighPtLambda;                      //! PVx distrib
263    TH1F      *fHistPVyAnalysisHasHighPtLambda;                      //! PVy distrib
264    TH1F      *fHistPVzAnalysisHasHighPtLambda;                      //! PVz distrib
265
266    TH1F      *fHistSwappedV0Counter;                      //! Swapped v0 counter
267
268    AliAnalysisTaskExtractPerformanceV0(const AliAnalysisTaskExtractPerformanceV0&);            // not implemented
269    AliAnalysisTaskExtractPerformanceV0& operator=(const AliAnalysisTaskExtractPerformanceV0&); // not implemented
270    
271    ClassDef(AliAnalysisTaskExtractPerformanceV0, 11);
272 };
273
274 #endif