]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWG3/vertexingHF/AliAnalysisVertexingHF.h
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
New class for AOD<->MC association
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAnalysisVertexingHF.h
CommitLineData
6a213b59 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 */
5
6//-------------------------------------------------------------------------
7// Class AliAnalysisVertexingHF
8// Reconstruction of heavy-flavour decay candidates
9//
10// Origin: E.Bruna, G.E.Bruno, A.Dainese, F.Prino, R.Romita
b056c5e3 11// Contact: andrea.dainese@lnl.infn.it
6a213b59 12//-------------------------------------------------------------------------
13
14#include <TNamed.h>
6a213b59 15#include "AliESDVertex.h"
16#include "AliAODRecoDecayHF.h"
17#include "AliAODRecoDecayHF2Prong.h"
18#include "AliAODRecoDecayHF3Prong.h"
19#include "AliAODRecoDecayHF4Prong.h"
20
dcb444c9 21class AliESDtrack;
22class AliVEvent;
23class AliAODVertex;
24
6a213b59 25//-----------------------------------------------------------------------------
26class AliAnalysisVertexingHF : public TNamed {
27 public:
28 //
29 AliAnalysisVertexingHF();
13977a79 30 AliAnalysisVertexingHF(const AliAnalysisVertexingHF& source);
31 AliAnalysisVertexingHF& operator=(const AliAnalysisVertexingHF& source);
6a213b59 32 virtual ~AliAnalysisVertexingHF();
33
dcb444c9 34 void FindCandidates(AliVEvent *event,
35 TClonesArray *aodVerticesHFTClArr,
36 TClonesArray *aodD0toKpiTClArr,
37 TClonesArray *aodJPSItoEleTClArr,
38 TClonesArray *aodCharm3ProngTClArr,
39 TClonesArray *aodCharm4ProngTClArr);
40
6a213b59 41 void PrintStatus() const;
b82f6d67 42 void SetSecVtxWithKF() { fSecVtxWithKF=kTRUE; }
6a213b59 43 void SetD0toKpiOn() { fD0toKpi=kTRUE; }
44 void SetD0toKpiOff() { fD0toKpi=kFALSE; }
45 void SetJPSItoEleOn() { fJPSItoEle=kTRUE; }
46 void SetJPSItoEleOff() { fJPSItoEle=kFALSE; }
47 void Set3ProngOn() { f3Prong=kTRUE; }
48 void Set3ProngOff() { f3Prong=kFALSE; }
49 void Set4ProngOn() { f4Prong=kTRUE; }
50 void Set4ProngOff() { f4Prong=kFALSE; }
b056c5e3 51 Bool_t GetD0toKpi() { return fD0toKpi; }
52 Bool_t GetJPSItoEle() { return fJPSItoEle; }
53 Bool_t Get3Prong() { return f3Prong; }
54 Bool_t Get4Prong() { return f4Prong; }
6a213b59 55 void SetRecoPrimVtxSkippingTrks()
56 { fRecoPrimVtxSkippingTrks=kTRUE; fRmTrksFromPrimVtx=kFALSE;}
57 void SetRmTrksFromPrimVtx()
58 {fRmTrksFromPrimVtx=kTRUE; fRecoPrimVtxSkippingTrks=kFALSE; }
59 void SetITSrefitRequired() { fITSrefit=kTRUE; }
60 void SetITSrefitNotRequired() { fITSrefit=kFALSE; }
61 void SetBothSPDRequired() { fBothSPD=kTRUE; }
62 void SetBothSPDNotRequired() {fBothSPD=kFALSE;}
63 void SetMinITSCls(Int_t n=6) { fMinITSCls=n; }
64 void SetMinPtCut(Double_t pt=0.) { fMinPtCut=pt; }
65 void SetMind0Cut(Double_t d0=0.) { fMind0rphiCut=d0; }
66 void SetD0toKpiCuts(Double_t cut0=1000.,Double_t cut1=100000.,
67 Double_t cut2=1.1,Double_t cut3=0.,Double_t cut4=0.,
68 Double_t cut5=100000.,Double_t cut6=100000.,
69 Double_t cut7=100000000.,Double_t cut8=-1.1);
70 void SetD0toKpiCuts(const Double_t cuts[9]);
71 void SetBtoJPSICuts(Double_t cut0=1000.,Double_t cut1=100000.,
72 Double_t cut2=1.1,Double_t cut3=0.,Double_t cut4=0.,
73 Double_t cut5=100000.,Double_t cut6=100000.,
74 Double_t cut7=100000000.,Double_t cut8=-1.1);
75 void SetBtoJPSICuts(const Double_t cuts[9]);
76 void SetDplusCuts(Double_t cut0=1000.,Double_t cut1=0.,
77 Double_t cut2=0.,Double_t cut3=0.,Double_t cut4=0.,
78 Double_t cut5=0.,Double_t cut6=10000000000.,
79 Double_t cut7=0.,Double_t cut8=0.,
80 Double_t cut9=-1.1,Double_t cut10=0.,
81679460 81 Double_t cut11=10000000000.);
6a213b59 82 void SetDplusCuts(const Double_t cuts[12]);
6ea608bf 83 void SetDsCuts(Double_t cut0=1000.,Double_t cut1=0.,
84 Double_t cut2=0.,Double_t cut3=0.,Double_t cut4=0.,
85 Double_t cut5=0.,Double_t cut6=10000000000.,
86 Double_t cut7=0.,Double_t cut8=0.,
87 Double_t cut9=-1.1,Double_t cut10=0.,
81679460 88 Double_t cut11=10000000000., Double_t cut12=1000.);
89 void SetDsCuts(const Double_t cuts[13]);
6ea608bf 90 void SetLcCuts(Double_t cut0=1000.,Double_t cut1=0.,
91 Double_t cut2=0.,Double_t cut3=0.,Double_t cut4=0.,
92 Double_t cut5=0.,Double_t cut6=10000000000.,
93 Double_t cut7=0.,Double_t cut8=0.,
94 Double_t cut9=-1.1,Double_t cut10=0.,
81679460 95 Double_t cut11=10000000000.);
6ea608bf 96 void SetLcCuts(const Double_t cuts[12]);
97
6a213b59 98 //
99 private:
100 //
dcb444c9 101 Bool_t fInputAOD; // input from AOD (kTRUE) or ESD (kFALSE)
102
103 Int_t *fAODMap; // map between index and ID for AOD tracks
104
b82f6d67 105 Double_t fBzkG; // z componenent of field in kG
106
107 Bool_t fSecVtxWithKF; // if kTRUE use KF vertexer, else AliVertexerTracks
108
6a213b59 109 Bool_t fRecoPrimVtxSkippingTrks; // flag for primary vertex reco on the fly
110 // for each candidate, w/o its daughters
111 Bool_t fRmTrksFromPrimVtx; // flag for fast removal of daughters from
112 // the primary vertex
113
114 AliESDVertex *fV1; // primary vertex
115
6a213b59 116 // flag to enable candidates production
117 Bool_t fD0toKpi;
118 Bool_t fJPSItoEle;
119 Bool_t f3Prong;
120 Bool_t f4Prong;
121
122 // single-track cuts
123 Bool_t fITSrefit; // require kITSrefit
124 Bool_t fBothSPD; // require both SPD layers
125 Int_t fMinITSCls; // minimum number of ITS clusters
126 Double_t fMinPtCut; // minimum track pt [GeV/c]
127 Double_t fMind0rphiCut; // minimum track |rphi impact parameter| [cm]
128 // candidates cuts
129 Double_t fD0toKpiCuts[9]; // cuts on D0->Kpi candidates
130 // (to be passed to AliAODRecoDecayHF2Prong::SelectD0())
131 // 0 = inv. mass half width [GeV]
132 // 1 = dca [cm]
133 // 2 = cosThetaStar
134 // 3 = pTK [GeV/c]
135 // 4 = pTPi [GeV/c]
136 // 5 = d0K [cm] upper limit!
137 // 6 = d0Pi [cm] upper limit!
138 // 7 = d0d0 [cm^2]
139 // 8 = cosThetaPoint
140 Double_t fBtoJPSICuts[9]; // cuts on JPSI candidates
141 // (to be passed to AliAODRecoDecayHF2Prong::SelectBtoJPSI())
142 // 0 = inv. mass half width [GeV]
143 // 1 = dca [cm]
144 // 2 = cosThetaStar (negative electron)
145 // 3 = pTP [GeV/c]
146 // 4 = pTN [GeV/c]
147 // 5 = d0O [cm] upper limit!
148 // 6 = d0N [cm] upper limit!
149 // 7 = d0d0 [cm^2]
150 // 8 = cosThetaPoint
151 Double_t fDplusCuts[12]; // cuts on Dplus candidates
6ea608bf 152 // (to be passed to AliAODRecoDecayHF3Prong::SelectDplus())
6a213b59 153 // 0 = inv. mass half width [GeV]
154 // 1 = pTK [GeV/c]
155 // 2 = pTPi [GeV/c]
156 // 3 = d0K [cm] lower limit!
157 // 4 = d0Pi [cm] lower limit!
158 // 5 = dist12 (cm)
159 // 6 = sigmavert (cm)
160 // 7 = dist prim-sec (cm)
161 // 8 = pM=Max{pT1,pT2,pT3} (GeV/c)
162 // 9 = cosThetaPoint
163 // 10 = Sum d0^2 (cm^2)
164 // 11 = dca cut (cm)
81679460 165 Double_t fDsCuts[13]; // cuts on Ds candidates
6ea608bf 166 // (to be passed to AliAODRecoDecayHF3Prong::SelectDs())
167 // 0 = inv. mass half width [GeV]
168 // 1 = pTK [GeV/c]
169 // 2 = pTPi [GeV/c]
170 // 3 = d0K [cm] lower limit!
171 // 4 = d0Pi [cm] lower limit!
172 // 5 = dist12 (cm)
173 // 6 = sigmavert (cm)
174 // 7 = dist prim-sec (cm)
175 // 8 = pM=Max{pT1,pT2,pT3} (GeV/c)
176 // 9 = cosThetaPoint
177 // 10 = Sum d0^2 (cm^2)
178 // 11 = dca cut (cm)
81679460 179 // 12 = inv. mass cut arounf phi and K0* [GeV]
180 Double_t fLcCuts[12]; // cuts on Lambdac candidates
6ea608bf 181 // (to be passed to AliAODRecoDecayHF3Prong::SelectLc())
182 // 0 = inv. mass half width [GeV]
183 // 1 = pTP [GeV/c]
184 // 2 = pTPi abd pTK [GeV/c]
185 // 3 = d0P [cm] lower limit!
186 // 4 = d0Pi and d0K [cm] lower limit!
187 // 5 = dist12 (cm)
188 // 6 = sigmavert (cm)
189 // 7 = dist prim-sec (cm)
190 // 8 = pM=Max{pT1,pT2,pT3} (GeV/c)
191 // 9 = cosThetaPoint
192 // 10 = Sum d0^2 (cm^2)
193 // 11 = dca cut (cm)
6a213b59 194 //
dcb444c9 195 void AddDaughterRefs(AliAODVertex *v,AliVEvent *event,
196 TObjArray *trkArray) const;
197 AliAODRecoDecayHF2Prong* Make2Prong(TObjArray *twoTrackArray1,AliVEvent *event,
198 AliAODVertex *secVert,Double_t dcap1n1,
199 Bool_t &okD0,Bool_t &okJPSI) const;
200 AliAODRecoDecayHF3Prong* Make3Prong(TObjArray *threeTrackArray,AliVEvent *event,
201 AliAODVertex *secVert,
202 Double_t dispersion,
203 AliAODVertex *vertexp1n1,
204 AliAODVertex *vertexp2n1,
205 Double_t dcap1n1,Double_t dcap2n1,Double_t dcap1p2,
206 Bool_t &ok3Prong) const;
207 AliAODRecoDecayHF4Prong* Make4Prong(TObjArray *fourTrackArray,AliVEvent *event,
208 AliAODVertex *secVert,
209 AliAODVertex *vertexp1n1,
210 AliAODVertex *vertexp2n1,
211 Double_t dcap1n1,Double_t dcap1n2,
212 Double_t dcap2n1,
213 Bool_t &ok4Prong) const;
214
215 AliAODVertex* OwnPrimaryVertex(TObjArray *trkArray,AliVEvent *event) const;
216 AliAODVertex* ReconstructSecondaryVertex(TObjArray *trkArray,Double_t &dispersion) const;
6a213b59 217 Bool_t SelectInvMass(Int_t decay,Int_t nprongs,
218 Double_t *px,Double_t *py,Double_t *pz) const;
dcb444c9 219 void SelectTracksAndCopyVertex(AliVEvent *event,
6a213b59 220 TObjArray &trksP,Int_t &nTrksP,
dcb444c9 221 TObjArray &trksN,Int_t &nTrksN);
222 void SetPrimaryVertex(AliESDVertex *v1) { fV1 = v1; }
223 Bool_t SingleTrkCuts(AliESDtrack &trk) const;
6a213b59 224 //
dcb444c9 225 ClassDef(AliAnalysisVertexingHF,4) // Reconstruction of HF decay candidates
6a213b59 226};
227
228
229#endif
230
231
232
233
234
235
236
237