1 #ifndef ALIANALYSISVERTEXINGHF_H
2 #define ALIANALYSISVERTEXINGHF_H
3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
6 //-------------------------------------------------------------------------
7 // Class AliAnalysisVertexingHF
8 // Reconstruction of heavy-flavour decay candidates
10 // Origin: E.Bruna, G.E.Bruno, A.Dainese, F.Prino, R.Romita, X.M.Zhang
11 // Contact: andrea.dainese@pd.infn.it
12 //-------------------------------------------------------------------------
18 class AliAODRecoDecayHF;
19 class AliAODRecoDecayHF2Prong;
20 class AliAODRecoDecayHF3Prong;
21 class AliAODRecoDecayHF4Prong;
22 class AliAODRecoCascadeHF;
23 class AliAnalysisFilter;
24 class AliRDHFCutsD0toKpi;
25 class AliRDHFCutsJpsitoee;
26 class AliRDHFCutsDplustoKpipi;
27 class AliRDHFCutsDstoKKpi;
28 class AliRDHFCutsLctopKpi;
29 class AliRDHFCutsLctoV0;
30 class AliRDHFCutsD0toKpipipi;
37 //-----------------------------------------------------------------------------
38 class AliAnalysisVertexingHF : public TNamed {
41 AliAnalysisVertexingHF();
42 AliAnalysisVertexingHF(const AliAnalysisVertexingHF& source);
43 AliAnalysisVertexingHF& operator=(const AliAnalysisVertexingHF& source);
44 virtual ~AliAnalysisVertexingHF();
46 void FindCandidates(AliVEvent *event,
47 TClonesArray *aodVerticesHFTClArr,
48 TClonesArray *aodD0toKpiTClArr,
49 TClonesArray *aodJPSItoEleTClArr,
50 TClonesArray *aodCharm3ProngTClArr,
51 TClonesArray *aodCharm4ProngTClArr,
52 TClonesArray *aodDstarTClArr,
53 TClonesArray *aodCascadesTClArr,
54 TClonesArray *aodLikeSign2ProngTClArr,
55 TClonesArray *aodLikeSign3ProngTClArr);
57 TList* FillListOfCuts();
58 void PrintStatus() const;
59 void SetSecVtxWithKF() { fSecVtxWithKF=kTRUE; }
60 void SetD0toKpiOn() { fD0toKpi=kTRUE; }
61 void SetD0toKpiOff() { fD0toKpi=kFALSE; }
62 void SetJPSItoEleOn() { fJPSItoEle=kTRUE; }
63 void SetJPSItoEleOff() { fJPSItoEle=kFALSE; }
64 void Set3ProngOn() { f3Prong=kTRUE; }
65 void Set3ProngOff() { f3Prong=kFALSE; }
66 void Set4ProngOn() { f4Prong=kTRUE; }
67 void Set4ProngOff() { f4Prong=kFALSE; }
68 void SetDstarOn() { fDstar=kTRUE; }
69 void SetDstarOff() { fDstar=kFALSE; }
70 void SetCascadesOn() { fCascades=kTRUE; }
71 void SetCascadesOff() { fCascades=kFALSE; }
72 void SetLikeSignOn() { fLikeSign=kTRUE; }
73 void SetLikeSignOff() { fLikeSign=kFALSE; }
74 void SetMixEventOn() { fMixEvent=kTRUE; }
75 void SetMixEventOff() { fMixEvent=kFALSE; }
76 void SetInputAOD() { fInputAOD=kTRUE; }
77 Bool_t GetD0toKpi() const { return fD0toKpi; }
78 Bool_t GetJPSItoEle() const { return fJPSItoEle; }
79 Bool_t Get3Prong() const { return f3Prong; }
80 Bool_t Get4Prong() const { return f4Prong; }
81 Bool_t GetDstar() const { return fDstar; }
82 Bool_t GetCascades() const { return fCascades; }
83 Bool_t GetLikeSign() const { return fLikeSign; }
84 Bool_t GetMixEvent() const { return fMixEvent; }
85 Bool_t GetInputAOD() const { return fInputAOD; }
86 Bool_t GetRecoPrimVtxSkippingTrks() const {return fRecoPrimVtxSkippingTrks;}
87 Bool_t GetRmTrksFromPrimVtx() const {return fRmTrksFromPrimVtx;}
88 void SetFindVertexForDstar(Bool_t vtx=kTRUE) { fFindVertexForDstar=vtx; }
89 void SetFindVertexForCascades(Bool_t vtx=kTRUE) { fFindVertexForCascades=vtx; }
90 void SetRecoPrimVtxSkippingTrks()
91 { fRecoPrimVtxSkippingTrks=kTRUE; fRmTrksFromPrimVtx=kFALSE;}
92 void UnsetRecoPrimVtxSkippingTrks()
93 { fRecoPrimVtxSkippingTrks=kFALSE; fRmTrksFromPrimVtx=kFALSE;}
94 void SetRmTrksFromPrimVtx()
95 {fRmTrksFromPrimVtx=kTRUE; fRecoPrimVtxSkippingTrks=kFALSE; }
96 void SetTrackFilter(AliAnalysisFilter* trackF) { fTrackFilter = trackF; }
97 void SetTrackFilterSoftPi(AliAnalysisFilter* trackF) { fTrackFilterSoftPi = trackF; }
98 AliAnalysisFilter* GetTrackFilter() const { return fTrackFilter; }
99 AliAnalysisFilter* GetTrackFilterSoftPi() const { return fTrackFilterSoftPi; }
100 void SetCutsD0toKpi(AliRDHFCutsD0toKpi* cuts) { fCutsD0toKpi = cuts; }
101 AliRDHFCutsD0toKpi* GetCutsD0toKpi() const { return fCutsD0toKpi; }
102 void SetCutsJpsitoee(AliRDHFCutsJpsitoee* cuts) { fCutsJpsitoee = cuts; }
103 AliRDHFCutsJpsitoee* GetCutsJpsitoee() const { return fCutsJpsitoee; }
104 void SetCutsDplustoKpipi(AliRDHFCutsDplustoKpipi* cuts) { fCutsDplustoKpipi = cuts; }
105 AliRDHFCutsDplustoKpipi* GetCutsDplustoKpipi() const { return fCutsDplustoKpipi; }
106 void SetCutsDstoKKpi(AliRDHFCutsDstoKKpi* cuts) { fCutsDstoKKpi = cuts; }
107 AliRDHFCutsDstoKKpi* GetCutsDstoKKpi() const { return fCutsDstoKKpi; }
108 void SetCutsLctopKpi(AliRDHFCutsLctopKpi* cuts) { fCutsLctopKpi = cuts; }
109 AliRDHFCutsLctopKpi* GetCutsLctopKpi() const { return fCutsLctopKpi; }
110 void SetCutsLctoV0(AliRDHFCutsLctoV0* cuts) { fCutsLctoV0 = cuts; }
111 AliRDHFCutsLctoV0* GetCutsLctoV0() const { return fCutsLctoV0; }
112 void SetCutsD0toKpipipi(AliRDHFCutsD0toKpipipi* cuts) { fCutsD0toKpipipi = cuts; }
113 AliRDHFCutsD0toKpipipi* GetCutsD0toKpipipi() const { return fCutsD0toKpipipi; }
114 void SetCutsD0fromDstar(AliRDHFCutsD0toKpi* cuts) { fCutsD0fromDstar = cuts; }
115 AliRDHFCutsD0toKpi* GetCutsD0fromDstar() const { return fCutsD0fromDstar; }
116 void SetD0toKpiCuts(Double_t cut0=1000.,Double_t cut1=100000.,
117 Double_t cut2=1.1,Double_t cut3=0.,Double_t cut4=0.,
118 Double_t cut5=100000.,Double_t cut6=100000.,
119 Double_t cut7=100000000.,Double_t cut8=-1.1);
120 void SetD0toKpiCuts(const Double_t cuts[9]);
121 void SetD0fromDstarCuts(Double_t cut0=1000.,Double_t cut1=100000.,
122 Double_t cut2=1.1,Double_t cut3=0.,Double_t cut4=0.,
123 Double_t cut5=100000.,Double_t cut6=100000.,
124 Double_t cut7=100000000.,Double_t cut8=-1.1);
125 void SetD0fromDstarCuts(const Double_t cuts[9]);
126 void SetBtoJPSICuts(Double_t cut0=1000.,Double_t cut1=100000.,
127 Double_t cut2=1.1,Double_t cut3=0.,Double_t cut4=0.,
128 Double_t cut5=100000.,Double_t cut6=100000.,
129 Double_t cut7=100000000.,Double_t cut8=-1.1);
130 void SetBtoJPSICuts(const Double_t cuts[9]);
131 void SetDplusCuts(Double_t cut0=1000.,Double_t cut1=0.,
132 Double_t cut2=0.,Double_t cut3=0.,Double_t cut4=0.,
133 Double_t cut5=0.,Double_t cut6=10000000000.,
134 Double_t cut7=0.,Double_t cut8=0.,
135 Double_t cut9=-1.1,Double_t cut10=0.,
136 Double_t cut11=10000000000.);
137 void SetDplusCuts(const Double_t cuts[12]);
138 void SetDsCuts(Double_t cut0=1000.,Double_t cut1=0.,
139 Double_t cut2=0.,Double_t cut3=0.,Double_t cut4=0.,
140 Double_t cut5=0.,Double_t cut6=10000000000.,
141 Double_t cut7=0.,Double_t cut8=0.,
142 Double_t cut9=-1.1,Double_t cut10=0.,
143 Double_t cut11=10000000000., Double_t cut12=1000.,
144 Double_t cut13=1000.);
145 void SetDsCuts(const Double_t cuts[14]);
146 void SetLcCuts(Double_t cut0=1000.,Double_t cut1=0.,
147 Double_t cut2=0.,Double_t cut3=0.,Double_t cut4=0.,
148 Double_t cut5=0.,Double_t cut6=10000000000.,
149 Double_t cut7=0.,Double_t cut8=0.,
150 Double_t cut9=-1.1,Double_t cut10=0.,
151 Double_t cut11=10000000000.);
152 void SetLcCuts(const Double_t cuts[12]);
153 void SetLctoV0Cuts(Double_t cut0=1000.,Double_t cut1=1000.,
154 Double_t cut2=1000., Double_t cut3=1000.,
155 Double_t cut4=0., Double_t cut5=0.,Double_t cut6=0.,
156 Double_t cut7=0.,Double_t cut8=0.);
157 void SetLctoV0Cuts(const Double_t cuts[9]);
158 void SetDstarCuts(Double_t cut0=1000., Double_t cut1=1000.,
159 Double_t cut2=-1., Double_t cut3=1000.,
161 void SetDstarCuts(const Double_t cuts[5]);
162 void SetD0to4ProngsCuts(Double_t cut0=1000.,Double_t cut1=100000.,
163 Double_t cut2=0.,Double_t cut3=0.,Double_t cut4=0.,
164 Double_t cut5=-1.1,Double_t cut6=0.,
165 Double_t cut7=0.,Double_t cut8=0.);
166 void SetD0to4ProngsCuts(const Double_t cuts[9]);
167 const Double_t *GetD0toKpiCuts() const {return fD0toKpiCuts;}
168 const Double_t *GetD0fromDstarCuts() const {return fD0fromDstarCuts;}
169 const Double_t *GetBtoJPSICuts() const {return fBtoJPSICuts;}
170 const Double_t *GetDplusCuts() const {return fDplusCuts;}
171 const Double_t *GetDsCuts() const {return fDsCuts;}
172 const Double_t *GetLcCuts() const {return fLcCuts;}
173 const Double_t *GetLctoV0Cuts() const {return fLctoV0Cuts;}
174 const Double_t *GetDstarCuts() const {return fDstarCuts;}
175 const Double_t *GetD0to4ProngsCuts() const {return fD0to4ProngsCuts;}
180 enum { kBitDispl = 0, kBitSoftPi = 1 };
182 Bool_t fInputAOD; // input from AOD (kTRUE) or ESD (kFALSE)
183 Int_t fAODMapSize; // size of fAODMap
184 Int_t *fAODMap; //[fAODMapSize] map between index and ID for AOD tracks
186 Double_t fBzkG; // z componenent of field in kG
188 Bool_t fSecVtxWithKF; // if kTRUE use KF vertexer, else AliVertexerTracks
190 Bool_t fRecoPrimVtxSkippingTrks; // flag for primary vertex reco on the fly
191 // for each candidate, w/o its daughters
192 Bool_t fRmTrksFromPrimVtx; // flag for fast removal of daughters from
193 // the primary vertex
195 AliESDVertex *fV1; // primary vertex
197 // flag to enable candidates production
198 Bool_t fD0toKpi; // D0->Kpi
199 Bool_t fJPSItoEle; // Jpsi->ee
200 Bool_t f3Prong; // D+,Ds,Lc
201 Bool_t f4Prong; // D0->Kpipipi
202 Bool_t fDstar; // D*->D0pi
203 Bool_t fCascades; // cascades, Lc --> v0+track
204 Bool_t fLikeSign; // Like-sign pairs
205 Bool_t fMixEvent; // event mixing
208 AliAnalysisFilter *fTrackFilter; // Track Filter for displaced vertices
209 AliAnalysisFilter *fTrackFilterSoftPi; // Track Filter for D* soft pion
211 AliRDHFCutsD0toKpi *fCutsD0toKpi; // D0->Kpi cuts
212 AliRDHFCutsJpsitoee *fCutsJpsitoee; // J/psi->ee cuts
213 AliRDHFCutsDplustoKpipi *fCutsDplustoKpipi; // D+->Kpipi cuts
214 AliRDHFCutsDstoKKpi *fCutsDstoKKpi; // Ds->KKpi cuts
215 AliRDHFCutsLctopKpi *fCutsLctopKpi; // Lc->pKpi cuts
216 AliRDHFCutsLctoV0 *fCutsLctoV0; // Lc --> v0 + bachelor cuts
217 AliRDHFCutsD0toKpipipi *fCutsD0toKpipipi; // D0->Kpipipi cuts
218 AliRDHFCutsD0toKpi *fCutsD0fromDstar; // D0 from Dstar cuts
220 Double_t fD0toKpiCuts[9]; // cuts on D0->Kpi candidates
221 // (to be passed to AliAODRecoDecayHF2Prong::SelectD0())
222 // 0 = inv. mass half width [GeV]
227 // 5 = d0K [cm] upper limit!
228 // 6 = d0Pi [cm] upper limit!
231 Double_t fD0fromDstarCuts[9]; // cuts on D0->Kpi candidates from D*
232 // (to be passed to AliAODRecoCascadeHF::SelectDstar())
233 // 0 = inv. mass half width [GeV]
238 // 5 = d0K [cm] upper limit!
239 // 6 = d0Pi [cm] upper limit!
242 Double_t fBtoJPSICuts[9]; // cuts on JPSI candidates
243 // (to be passed to AliAODRecoDecayHF2Prong::SelectBtoJPSI())
244 // 0 = inv. mass half width [GeV]
246 // 2 = cosThetaStar (negative electron)
249 // 5 = d0O [cm] upper limit!
250 // 6 = d0N [cm] upper limit!
253 Double_t fDplusCuts[12]; // cuts on Dplus candidates
254 // (to be passed to AliAODRecoDecayHF3Prong::SelectDplus())
255 // 0 = inv. mass half width [GeV]
258 // 3 = d0K [cm] lower limit!
259 // 4 = d0Pi [cm] lower limit!
261 // 6 = sigmavert (cm)
262 // 7 = dist prim-sec (cm)
263 // 8 = pM=Max{pT1,pT2,pT3} (GeV/c)
265 // 10 = Sum d0^2 (cm^2)
267 Double_t fDsCuts[14]; // cuts on Ds candidates
268 // (to be passed to AliAODRecoDecayHF3Prong::SelectDs())
269 // 0 = inv. mass half width [GeV]
272 // 3 = d0K [cm] lower limit!
273 // 4 = d0Pi [cm] lower limit!
275 // 6 = sigmavert (cm)
276 // 7 = dist prim-sec (cm)
277 // 8 = pM=Max{pT1,pT2,pT3} (GeV/c)
279 // 10 = Sum d0^2 (cm^2)
281 // 12 = inv. mass cut around phi
282 // 13 = inv. mass cut around phi K0* [GeV]
283 Double_t fLcCuts[12]; // cuts on Lambdac candidates
284 // (to be passed to AliAODRecoDecayHF3Prong::SelectLc())
285 // 0 = inv. mass half width [GeV]
287 // 2 = pTPi abd pTK [GeV/c]
288 // 3 = d0P [cm] lower limit!
289 // 4 = d0Pi and d0K [cm] lower limit!
291 // 6 = sigmavert (cm)
292 // 7 = dist prim-sec (cm)
293 // 8 = pM=Max{pT1,pT2,pT3} (GeV/c)
295 // 10 = Sum d0^2 (cm^2)
298 Double_t fLctoV0Cuts[9]; // cuts on Lambdac candidates to V0+bachelor
299 // (to be passed to AliAODRecoDecayHF3Prong::SelectLctoV0())
300 // 0 = inv. mass half width in K0s hypothesis [GeV]
301 // 1 = inv. mass half width in Lambda hypothesis [GeV]
302 // 2 = inv. mass V0 in K0s hypothesis half width [GeV]
303 // 3 = inv. mass V0 in Lambda hypothesis half width [GeV]
304 // 4 = pT min Bachelor track [GeV/c]
305 // 5 = pT min V0-Positive track [GeV/c]
306 // 6 = pT min V0-Negative track [GeV/c]
307 // 7 = dca cut on the V0 (cm)
308 // 8 = dca cut on the cascade (cm)
310 Double_t fDstarCuts[5]; // cuts on D* candidates
311 // (to be passed to AliAODRecoCascadeHF::SelectDstar())
312 // 0 = inv. mass half width of D* [GeV]
313 // 1 = half width of (M_Kpipi-M_Kpi) [GeV]
314 // 2 = PtMin of pi_s [GeV/c]
315 // 3 = PtMax of pi_s [GeV/c]
316 // 4 = theta, angle between the track of pi_s and D0 decay plane [rad]
318 Double_t fD0to4ProngsCuts[9]; // cuts on D0->K3pi candidates
319 // (to be passed to AliAODRecoDecayHF4Prong::SelectD0())
320 // 0 = inv. mass half width of D0 [GeV]
321 // 1 = DCA between opposite sign tracks
322 // 2 = Distance between primary and two tracks vertex fDist12toPrim
323 // 3 = Distance between primary and three tracks vertex fDist3toPrim
324 // 4 = Distance between primary and two tracks vertex fDist4toPrim
325 // 5 = Cosinus of the pointing angle
326 // 6 = Transverse momentum of the D0 candidate
327 // 7 = Mass Pi+Pi- = mass of the rho0
328 // 8 = PID cut (one K in the quadruplet)
330 TList *fListOfCuts; // pointer to list of cuts for output file
331 Bool_t fFindVertexForDstar; // reconstruct a secondary vertex or assume it's from the primary vertex
332 Bool_t fFindVertexForCascades; // reconstruct a secondary vertex or assume it's from the primary vertex
334 void AddRefs(AliAODVertex *v,AliAODRecoDecayHF *rd,const AliVEvent *event,
335 const TObjArray *trkArray) const;
336 void AddDaughterRefs(AliAODVertex *v,const AliVEvent *event,
337 const TObjArray *trkArray) const;
338 AliAODRecoDecayHF2Prong* Make2Prong(TObjArray *twoTrackArray1,AliVEvent *event,
339 AliAODVertex *secVert,Double_t dcap1n1,
340 Bool_t &okD0,Bool_t &okJPSI,Bool_t &okD0fromDstar) const;
341 AliAODRecoDecayHF3Prong* Make3Prong(TObjArray *threeTrackArray,AliVEvent *event,
342 AliAODVertex *secVert,
344 const AliAODVertex *vertexp1n1,
345 const AliAODVertex *vertexp2n1,
346 Double_t dcap1n1,Double_t dcap2n1,Double_t dcap1p2,
347 Bool_t &ok3Prong) const;
348 AliAODRecoDecayHF4Prong* Make4Prong(TObjArray *fourTrackArray,AliVEvent *event,
349 AliAODVertex *secVert,
350 const AliAODVertex *vertexp1n1,
351 const AliAODVertex *vertexp1n1p2,
352 Double_t dcap1n1,Double_t dcap1n2,
353 Double_t dcap2n1,Double_t dcap2n2,
354 Bool_t &ok4Prong) const;
355 AliAODRecoCascadeHF* MakeCascade(TObjArray *twoTrackArray,AliVEvent *event,
356 AliAODVertex *secVert,
357 AliAODRecoDecayHF2Prong *rd2Prong,
359 Bool_t &okDstar) const;
360 AliAODRecoCascadeHF* MakeCascade(TObjArray *twoTrackArray,AliVEvent *event,
361 AliAODVertex *secVert,
364 Bool_t &okCascades) const;
366 AliAODVertex* PrimaryVertex(const TObjArray *trkArray=0x0,AliVEvent *event=0x0) const;
367 AliAODVertex* ReconstructSecondaryVertex(TObjArray *trkArray,Double_t &dispersion,Bool_t useTRefArray=kTRUE) const;
368 Bool_t SelectInvMass(Int_t decay,Int_t nprongs,
369 Double_t *px,Double_t *py,Double_t *pz) const;
370 void SelectTracksAndCopyVertex(const AliVEvent *event,
371 TObjArray &seleTrksArray,Int_t &nSeleTrks,
372 UChar_t *seleFlags,Int_t *evtNumber);
373 Bool_t SingleTrkCuts(AliESDtrack *trk,Bool_t &okDisplaced,Bool_t &okSoftPi) const;
375 AliAODv0* TransformESDv0toAODv0(AliESDv0 *esdv0,
376 TObjArray *twoTrackArrayV0);
379 ClassDef(AliAnalysisVertexingHF,17); // Reconstruction of HF decay candidates