]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWG2/SPECTRA/AliAnalysisTaskCheckCascade.h
da2efe8a6d3b88caa4634cc448cbc3da54a7bdc6
[u/mrichter/AliRoot.git] / PWG2 / SPECTRA / AliAnalysisTaskCheckCascade.h
1 #ifndef ALIANALYSISTASKCHECKCASCADE_H
2 #define ALIANALYSISTASKCHECKCASCADE_H
3
4 /*  See cxx source for full Copyright notice */
5
6 //-----------------------------------------------------------------
7 //                 AliAnalysisTaskCheckCascade class
8 //            (AliAnalysisTaskCheckCascade)
9 //            This task has four roles :
10 //              1. QAing the Cascades from ESD and AOD
11 //                 Origin:  AliAnalysisTaskESDCheckV0 by B.H. Nov2007, hippolyt@in2p3.fr
12 //              2. Prepare the plots which stand as raw material for yield extraction (wi/wo PID)
13 //              3. Supply an AliCFContainer meant to define the optimised topological selections
14 //              4. Rough azimuthal correlation study (Eta, Phi)
15 //            Adapted to Cascade : A.Maire Mar2008, antonin.maire@ires.in2p3.fr
16 //            Modified :           A.Maire Dec2009, antonin.maire@ires.in2p3.fr
17 //-----------------------------------------------------------------
18
19 class TList;
20 class TH1F;
21 class TH2F;
22 class TH3F;
23 class TVector3;
24 class THnSparse;
25  
26 class AliESDpid;
27 class AliCFContainer;
28
29
30 #include "TString.h"
31
32 #include "AliAnalysisTaskSE.h"
33
34 class AliAnalysisTaskCheckCascade : public AliAnalysisTaskSE {
35  public:
36   AliAnalysisTaskCheckCascade();
37   AliAnalysisTaskCheckCascade(const char *name);
38   virtual ~AliAnalysisTaskCheckCascade() {}
39   
40   virtual void   UserCreateOutputObjects();
41   virtual void   UserExec(Option_t *option);
42           void   DoAngularCorrelation(const Char_t   *lCascType, 
43                                             Double_t  lInvMassCascade, 
44                                       const Int_t    *lArrTrackID,
45                                             TVector3 &lTVect3MomXi, 
46                                             Double_t  lEtaXi);
47   virtual void   Terminate(Option_t *);
48   
49   void SetRealDataFlag    (Short_t realData = 0)             {fRealData = realData;}
50   void SetCollidingSystems(Short_t collidingSystems = 0)     {fCollidingSystems = collidingSystems;}
51   void SetAnalysisType    (const char* analysisType = "ESD") {fAnalysisType = analysisType;}
52     
53  private:
54         TString         fAnalysisType;                  // "ESD" or "AOD" analysis type 
55         Short_t         fCollidingSystems;              // 0 = pp collisions or 1 = AA collisions
56         AliESDpid*      fESDpid;                        //! Tool data member to manage the TPC Bethe-Bloch info
57         Short_t         fRealData;                      // 0 = MC data or 1 = real data (needed for trigger issues)
58         
59         TList   *fListHistCascade;              //! List of Cascade histograms
60         
61         // - General histos (filled for any event)
62         TH1F    *fHistTrackMultiplicity;                //! Track multiplicity distribution
63         TH1F    *fHistCascadeMultiplicity;              //! Cascade multiplicity distribution
64
65
66         // - Vertex Positions
67         TH1F    *fHistVtxStatus;                        //! Is there a tracking vertex in the event ?
68
69                 // Vtx coming from the full tracking
70         TH1F    *fHistPosTrkgPrimaryVtxX;               //! primary vertex position distribution in x 
71         TH1F    *fHistPosTrkgPrimaryVtxY;               //! primary vertex position distribution in y
72         TH1F    *fHistPosTrkgPrimaryVtxZ;               //! primary vertex position distribution in z
73         TH1F    *fHistTrkgPrimaryVtxRadius;             //! primary vertex (3D) radius distribution 
74
75                 // Best primary Vtx available for the event
76         TH1F    *fHistPosBestPrimaryVtxX;               //! (best) primary vertex position distribution in x 
77         TH1F    *fHistPosBestPrimaryVtxY;               //! (best) primary vertex position distribution in y
78         TH1F    *fHistPosBestPrimaryVtxZ;               //! (best) primary vertex position distribution in z
79         TH1F    *fHistBestPrimaryVtxRadius;             //! (best) primary vertex radius distribution 
80         
81                 // Correlation Best Vtx / Full Tracking Vtx
82         TH2F    *f2dHistTrkgPrimVtxVsBestPrimVtx;       //!  Radius of prim. Vtx from tracks Vs Radius of best Prim. Vtx
83         
84         
85         // PART 1 : Adavanced QA
86         // - Typical histos on the variables used for the selection of cascades
87         TH1F    *fHistEffMassXi;                        //! reconstructed cascade effective mass
88         TH1F    *fHistChi2Xi;                           //! chi2 value
89         TH1F    *fHistDcaXiDaughters;                   //! dca between Xi's daughters
90         TH1F    *fHistDcaBachToPrimVertex;              //! dca of the bachelor track to primary vertex
91         TH1F    *fHistXiCosineOfPointingAngle;          //! cosine of Xi pointing angle in a cascade
92         TH1F    *fHistXiRadius;                         //! (transverse) radius of the cascade vertex 
93                 
94         // - Histos about ~ the "V0 selection part" of the cascade,  coming by inheritance from AliESDv0
95         TH1F    *fHistMassLambdaAsCascDghter;           //! Test Invariant Mass of Lambda coming from Cascade
96         TH1F    *fHistV0Chi2Xi;                         //! V0 chi2 distribution, for the V0 associated to a cascade
97         TH1F    *fHistDcaV0DaughtersXi;                 //! Dca between V0 daughters, for the V0 associated to a cascade
98         TH1F    *fHistDcaV0ToPrimVertexXi;              //! Dca of V0 to primary vertex, for the V0 associated to a cascade     
99         TH1F    *fHistV0CosineOfPointingAngleXi;        //! Cosine of V0 pointing angle, for the V0 associated to a cascade
100         TH1F    *fHistV0RadiusXi;                       //! V0 (transverse) distance distribution, for the V0 associated to a cascade
101
102         TH1F    *fHistDcaPosToPrimVertexXi;             //! Dca of V0 positive daughter to primary vertex, for the V0 associated to a cascade
103         TH1F    *fHistDcaNegToPrimVertexXi;             //! Dca of V0 negative daughter to primary vertex, for the V0 associated to a cascade
104         
105
106         // - Effective mass histos for cascades.
107         TH1F    *fHistMassXiMinus;                      //! reconstructed cascade effective mass, under Xi- hyp.
108         TH1F    *fHistMassXiPlus;                       //! reconstructed cascade effective mass, under Xi+ hyp.
109         TH1F    *fHistMassOmegaMinus;                   //! reconstructed cascade effective mass, under Omega- hyp.
110         TH1F    *fHistMassOmegaPlus;                    //! reconstructed cascade effective mass, under Omega+ hyp.
111         
112         TH1F    *fHistMassWithCombPIDXiMinus;           //! reconstructed Xi- effective mass, with bach. comb PID
113         TH1F    *fHistMassWithCombPIDXiPlus;            //! reconstructed Xi+ effective mass, with bach. comb PID
114         TH1F    *fHistMassWithCombPIDOmegaMinus;        //! reconstructed Omega- effective mass, with bach. comb PID
115         TH1F    *fHistMassWithCombPIDOmegaPlus;         //! reconstructed Omega+ effective mass, with bach. comb PID
116
117         // - Complements for QA
118         TH1F    *fHistXiTransvMom;                      //! Xi transverse momentum 
119         TH1F    *fHistXiTotMom;                         //! Xi momentum norm
120         
121         TH1F    *fHistBachTransvMom;                    //! bachelor transverse momentum 
122         TH1F    *fHistBachTotMom;                       //! bachelor momentum norm
123                                 
124         TH1F    *fHistChargeXi;                         //! Charge sign of the cascade candidate
125         TH1F    *fHistV0toXiCosineOfPointingAngle;      //! Cos. of Pointing angle between the V0 mom and the Xi-V0 vtx line
126   
127         TH1F    *fHistRapXi;                            //! rapidity of Xi candidates
128         TH1F    *fHistRapOmega;                         //! rapidity of Omega candidates
129         TH1F    *fHistEta;                              //! eta distrib. of all the cascade candidates
130         TH1F    *fHistTheta;                            //! theta distrib. of all the cascade candidates
131         TH1F    *fHistPhi;                              //! phi distrib. of all the cascade candidates
132         
133         TH2F    *f2dHistArmenteros;                     //! alpha(casc. cand.) Vs PtArm(casc. cand.)
134         
135         TH2F    *f2dHistEffMassLambdaVsEffMassXiMinus;  //! Xi- Eff mass Vs V0 Eff mass, under Xi- hyp.
136         TH2F    *f2dHistEffMassXiVsEffMassOmegaMinus;   //! Xi- Eff mass Vs Omega- Eff mass, for negative cascades
137         TH2F    *f2dHistEffMassLambdaVsEffMassXiPlus;   //! Xi+ Eff mass Vs V0 Eff mass, under Xi+ hyp.
138         TH2F    *f2dHistEffMassXiVsEffMassOmegaPlus;    //! Xi+ Eff mass Vs Omega+ Eff mass, for positive cascades
139         
140         TH2F    *f2dHistXiRadiusVsEffMassXiMinus;       //! transv. casc. decay radius Vs Xi- Eff mass, under Xi- hyp.
141         TH2F    *f2dHistXiRadiusVsEffMassXiPlus;        //! transv. casc. decay radius Vs Xi+ Eff mass, under Xi+ hyp.
142         TH2F    *f2dHistXiRadiusVsEffMassOmegaMinus;    //! transv. casc. decay radius Vs Omega- Eff mass, under Omega- hyp.
143         TH2F    *f2dHistXiRadiusVsEffMassOmegaPlus;     //! transv. casc. decay radius Vs Omega+ Eff mass, under Omega+ hyp.
144         
145         
146         // PART 2 : TH3F needed for pt spectrum and yield extraction
147         // Without any PID
148         TH3F    *f3dHistXiPtVsEffMassVsYXiMinus;        //! casc. transv. momemtum Vs Xi- Eff mass Vs Y
149         TH3F    *f3dHistXiPtVsEffMassVsYXiPlus;         //! casc. transv. momemtum Vs Xi+ Eff mass Vs Y
150         TH3F    *f3dHistXiPtVsEffMassVsYOmegaMinus;     //! casc. transv. momemtum Vs Omega- Eff mass Vs Y
151         TH3F    *f3dHistXiPtVsEffMassVsYOmegaPlus;      //! casc. transv. momemtum Vs Omega+ Eff mass Vs Y
152         
153         // With single PID : proton PID for Xi (pion = useless) / bachelor PID for Omega
154         // = a priori, the most important PID info for each species
155         TH3F    *f3dHistXiPtVsEffMassVsYWithCombPIDXiMinus;     //! casc. transv. momemtum Vs Xi- Eff mass Vs Y
156         TH3F    *f3dHistXiPtVsEffMassVsYWithCombPIDXiPlus;      //! casc. transv. momemtum Vs Xi+ Eff mass Vs Y
157         TH3F    *f3dHistXiPtVsEffMassVsYWithCombPIDOmegaMinus;  //! casc. transv. momemtum Vs Omega- Eff mass Vs Y
158         TH3F    *f3dHistXiPtVsEffMassVsYWithCombPIDOmegaPlus;   //! casc. transv. momemtum Vs Omega+ Eff mass Vs Y
159         
160         // With double PID : proton PID + bachelor PID for Omega
161         // = "second order" refinement for omegas
162         TH3F    *f3dHistXiPtVsEffMassVsYWith2CombPIDOmegaMinus; //! casc. transv. momemtum Vs Omega- Eff mass Vs Y
163         TH3F    *f3dHistXiPtVsEffMassVsYWith2CombPIDOmegaPlus;  //! casc. transv. momemtum Vs Omega+ Eff mass Vs Y
164         
165         // With TPC PID : 3-sigma band on the Bethe-Bloch curve
166         // = directly detector-based
167         TH3F    *f3dHistXiPtVsEffMassVsYWithTpcPIDOmegaMinus;   //! casc. transv. momemtum Vs Omega- Eff mass Vs Y
168         
169         // Compilation of all PID plots (3D = casc. transv. momemtum Vs Casc Eff mass Vs Y), stored into an AliCFContainer
170         AliCFContainer  *fCFContCascadePIDXiMinus;      //! for Xi-   : Container to store any 3D histos with the different PID flavours
171         AliCFContainer  *fCFContCascadePIDXiPlus;       //! for Xi+   : Container to store any 3D histos with the different PID flavours
172         AliCFContainer  *fCFContCascadePIDOmegaMinus;   //! for Omega-: Container to store any 3D histos with the different PID flavours
173         AliCFContainer  *fCFContCascadePIDOmegaPlus;    //! for Omega+: Container to store any 3D histos with the different PID flavours
174         
175         
176         
177         // PART 3 : Towards the optimisation of topological selections
178         AliCFContainer  *fCFContCascadeCuts;            //! Container meant to store all the relevant distributions corresponding to the cut variables
179         
180         
181         // PART 4 :  Azimuthal correlation study
182         THnSparseF      *fHnSpAngularCorrXiMinus;       //! Delta Phi(Casc,any trck) Vs Delta Eta(Casc,any trck) Vs Casc Pt Vs Pt of the tracks Vs Eff Mass
183         THnSparseF      *fHnSpAngularCorrXiPlus;        //! Delta Phi(Casc,any trck) Vs Delta Eta(Casc,any trck) Vs Casc Pt Vs Pt of the tracks Vs Eff Mass
184         THnSparseF      *fHnSpAngularCorrOmegaMinus;    //! Delta Phi(Casc,any trck) Vs Delta Eta(Casc,any trck) Vs Casc Pt Vs Pt of the tracks Vs Eff Mass
185         THnSparseF      *fHnSpAngularCorrOmegaPlus;     //! Delta Phi(Casc,any trck) Vs Delta Eta(Casc,any trck) Vs Casc Pt Vs Pt of the tracks Vs Eff Mass
186
187
188   AliAnalysisTaskCheckCascade(const AliAnalysisTaskCheckCascade&);            // not implemented
189   AliAnalysisTaskCheckCascade& operator=(const AliAnalysisTaskCheckCascade&); // not implemented
190   
191   ClassDef(AliAnalysisTaskCheckCascade, 8);
192 };
193
194 #endif