10eaad41 |
1 | /************************************************************************** |
2 | * Author: Paraskevi Ganoti, University of Athens (pganoti@phys.uoa.gr) * |
3 | * Contributors are mentioned in the code where appropriate. * |
4 | * * |
5 | * Permission to use, copy, modify and distribute this software and its * |
6 | * documentation strictly for non-commercial purposes is hereby granted * |
7 | * without fee, provided that the above copyright notice appears in all * |
8 | * copies and that both the copyright notice and this permission notice * |
9 | * appear in the supporting documentation. The authors make no claims * |
10 | * about the suitability of this software for any purpose. It is * |
11 | * provided "as is" without express or implied warranty. * |
12 | **************************************************************************/ |
13 | |
e3c296cd |
14 | //---------------------------------------------------------------------------------------------------------------- |
10eaad41 |
15 | // class AliResonanceKinkLikeSign |
16 | // Example of an analysis task for producing a like-sign background for resonances having at least one |
17 | // kaon-kink in their decay products. |
e3c296cd |
18 | // Background is calculated from a positive kaon kink and a negative track. |
19 | //----------------------------------------------------------------------------------------------------------------- |
10eaad41 |
20 | |
be1a7181 |
21 | #include "AliESDEvent.h" |
10eaad41 |
22 | #include "TH1D.h" |
7ccf0419 |
23 | #include "TH2D.h" |
be1a7181 |
24 | #include "TF1.h" |
25 | #include "TDatabasePDG.h" |
10eaad41 |
26 | #include "TLorentzVector.h" |
be1a7181 |
27 | #include "AliESDkink.h" |
10eaad41 |
28 | |
29 | #include "AliResonanceKinkLikeSign.h" |
10eaad41 |
30 | |
31 | ClassImp(AliResonanceKinkLikeSign) |
32 | |
10eaad41 |
33 | //________________________________________________________________________ |
34 | AliResonanceKinkLikeSign::AliResonanceKinkLikeSign(const char *name) |
be1a7181 |
35 | : AliAnalysisTaskSE(name), fDebug(0), fListOfHistos(0), f1(0), f2(0), fPosKaonLikeSign(0), fLikeSignInvmassPt(0), fMaxNSigmaToVertex(0), fMinPtTrackCut(0), fMaxDCAxy(0), fMaxDCAzaxis(0), |
e3c296cd |
36 | fMinTPCclusters(0),fMaxChi2PerTPCcluster(0), fMaxCov0(0), fMaxCov2(0), fMaxCov5(0) , fMaxCov9(0), fMaxCov14(0), fdaughter1pdg(0), fdaughter2pdg(0), fnbins(0), fnlowx(0), fnhighx(0), floweta(0), fuppereta(0), fminKinkRadius(0), fmaxKinkRadius(0), fminQt(0), fmaxQt(0), fptbins(0), flowpt(0), fupperpt(0), fmaxAbsEtaCut(0) |
10eaad41 |
37 | |
38 | { |
39 | // Constructor |
40 | |
41 | // Define input and output slots here |
42 | // Input slot #0 works with a TChain |
10eaad41 |
43 | DefineOutput(1, TList::Class()); |
44 | } |
45 | |
46 | //________________________________________________________________________ |
be1a7181 |
47 | void AliResonanceKinkLikeSign::UserCreateOutputObjects() |
10eaad41 |
48 | { |
49 | // Create histograms |
50 | // Called once |
51 | |
52 | f1=new TF1("f1","((atan([0]*[1]*(1.0/(sqrt((x^2)*(1.0-([1]^2))-([0]^2)*([1]^2))))))*180.)/[2]",1.1,10.0); |
53 | f1->SetParameter(0,0.493677); |
54 | f1->SetParameter(1,0.9127037); |
55 | f1->SetParameter(2,TMath::Pi()); |
56 | |
57 | f2=new TF1("f2","((atan([0]*[1]*(1.0/(sqrt((x^2)*(1.0-([1]^2))-([0]^2)*([1]^2))))))*180.)/[2]",0.1,10.0); |
58 | f2->SetParameter(0,0.13957018); |
59 | f2->SetParameter(1,0.2731374); |
60 | f2->SetParameter(2,TMath::Pi()); |
61 | |
be1a7181 |
62 | fPosKaonLikeSign=new TH1D("fPosKaonLikeSign"," ", fnbins, fnlowx, fnhighx); |
894840ad |
63 | fLikeSignInvmassPt=new TH2D("fLikeSignInvmassPt"," ", fnbins, fnlowx, fnhighx, fptbins, flowpt, fupperpt); |
be1a7181 |
64 | |
10eaad41 |
65 | fListOfHistos=new TList(); |
7ccf0419 |
66 | fListOfHistos->Add(fPosKaonLikeSign); |
67 | fListOfHistos->Add(fLikeSignInvmassPt); |
10eaad41 |
68 | |
69 | } |
70 | |
71 | //________________________________________________________________________ |
be1a7181 |
72 | void AliResonanceKinkLikeSign::UserExec(Option_t *) |
10eaad41 |
73 | { |
74 | // Main loop |
75 | // Called for each event |
76 | |
be1a7181 |
77 | AliVEvent *event = InputEvent(); |
78 | if (!event) { |
79 | Printf("ERROR: Could not retrieve event"); |
80 | return; |
81 | } |
10eaad41 |
82 | |
be1a7181 |
83 | AliESDEvent* esd = dynamic_cast<AliESDEvent*>(event); |
84 | if (!esd) { |
85 | Printf("ERROR: Could not retrieve esd"); |
86 | return; |
87 | } |
f2a16059 |
88 | |
be1a7181 |
89 | Double_t daughter1Mass, daughter2Mass; |
90 | |
91 | if (fdaughter1pdg==kKPlus) { |
92 | daughter1Mass=TDatabasePDG::Instance()->GetParticle(fdaughter1pdg)->Mass(); |
93 | daughter2Mass=TDatabasePDG::Instance()->GetParticle(fdaughter2pdg)->Mass(); |
94 | } |
95 | |
96 | if (fdaughter1pdg!=kKPlus) { |
97 | daughter1Mass=TDatabasePDG::Instance()->GetParticle(fdaughter2pdg)->Mass(); |
98 | daughter2Mass=TDatabasePDG::Instance()->GetParticle(fdaughter1pdg)->Mass(); |
99 | } // to ensure that daughter1Mass has always the kaon mass |
100 | |
101 | if (!esd) { |
102 | Printf("ERROR: esd not available"); |
10eaad41 |
103 | return; |
104 | } |
105 | |
be1a7181 |
106 | const AliESDVertex* vertex = GetEventVertex(esd); |
10eaad41 |
107 | if(!vertex) return; |
108 | |
109 | Double_t ptrackpos[3], ptrackneg[3]; |
110 | |
111 | TLorentzVector p4pos; |
112 | TLorentzVector p4neg; |
113 | TLorentzVector p4comb; |
114 | |
be1a7181 |
115 | for (Int_t iTracks = 0; iTracks < esd->GetNumberOfTracks(); iTracks++) { |
116 | |
117 | AliESDtrack* trackpos = esd->GetTrack(iTracks); |
10eaad41 |
118 | if (!trackpos) { |
1ac5c71d |
119 | if (fDebug > 0) Printf("ERROR: Could not receive track %d", iTracks); |
10eaad41 |
120 | continue; |
121 | } |
7ccf0419 |
122 | if (trackpos->GetSign() < 0) continue; |
10eaad41 |
123 | |
be1a7181 |
124 | AliExternalTrackParam *tpcTrackpos = (AliExternalTrackParam *)trackpos->GetTPCInnerParam(); |
125 | if(!tpcTrackpos) continue; |
126 | ptrackpos[0]=tpcTrackpos->Px(); |
127 | ptrackpos[1]=tpcTrackpos->Py(); |
128 | ptrackpos[2]=tpcTrackpos->Pz(); |
10eaad41 |
129 | |
be1a7181 |
130 | Bool_t firstLevelAcceptPosTrack=IsAcceptedForKink(esd, vertex, trackpos); |
131 | if(firstLevelAcceptPosTrack==kFALSE) continue; |
10eaad41 |
132 | |
133 | TVector3 posTrackMom(ptrackpos[0],ptrackpos[1],ptrackpos[2]); |
10eaad41 |
134 | |
be1a7181 |
135 | Int_t indexKinkPos=trackpos->GetKinkIndex(0); |
136 | Bool_t posKaonKinkFlag=0; |
137 | if(indexKinkPos<0) posKaonKinkFlag=IsKink(esd, indexKinkPos, posTrackMom); |
138 | if(posKaonKinkFlag==0) continue; |
139 | if(posKaonKinkFlag==1) p4pos.SetVectM(posTrackMom, daughter1Mass); |
7ccf0419 |
140 | |
be1a7181 |
141 | for (Int_t j=0; j<esd->GetNumberOfTracks(); j++) { |
10eaad41 |
142 | if(iTracks==j) continue; |
be1a7181 |
143 | AliESDtrack* trackneg=esd->GetTrack(j); |
144 | if (!trackneg) { |
145 | if (fDebug > 0) Printf("ERROR: Could not receive track %d", j); |
146 | continue; |
147 | } |
7ccf0419 |
148 | if (trackneg->GetSign() < 0) continue; |
10eaad41 |
149 | |
be1a7181 |
150 | AliExternalTrackParam *tpcTrackneg = (AliExternalTrackParam *)trackneg->GetTPCInnerParam(); |
151 | if(!tpcTrackneg) continue; |
152 | ptrackneg[0]=tpcTrackneg->Px(); |
153 | ptrackneg[1]=tpcTrackneg->Py(); |
154 | ptrackneg[2]=tpcTrackneg->Pz(); |
155 | |
156 | Bool_t firstLevelAcceptNegTrack=IsAcceptedForKink(esd, vertex, trackneg); |
157 | if(firstLevelAcceptNegTrack==kFALSE) continue; |
10eaad41 |
158 | |
be1a7181 |
159 | TVector3 negTrackMom(ptrackneg[0],ptrackneg[1],ptrackneg[2]); |
10eaad41 |
160 | |
be1a7181 |
161 | Bool_t secondLevelAcceptNegTrack=IsAcceptedForTrack(esd, vertex, trackneg); |
162 | if(secondLevelAcceptNegTrack==kFALSE) continue; |
163 | |
164 | p4neg.SetVectM(negTrackMom, daughter2Mass); |
10eaad41 |
165 | |
166 | p4comb=p4pos; |
167 | p4comb+=p4neg; |
be1a7181 |
168 | |
894840ad |
169 | if(p4comb.Vect().Pt()<=fMinPtTrackCut) continue; |
be1a7181 |
170 | |
e3c296cd |
171 | if((TMath::Abs(p4pos.Vect().Eta())<fmaxAbsEtaCut)&&(TMath::Abs(p4neg.Vect().Eta())<fmaxAbsEtaCut)&&(p4comb.Vect().Eta()<fmaxAbsEtaCut)) { |
be1a7181 |
172 | |
173 | fPosKaonLikeSign->Fill(p4comb.M()); |
174 | fLikeSignInvmassPt->Fill(p4comb.M(), p4comb.Vect().Pt()); |
175 | |
176 | } |
10eaad41 |
177 | |
178 | } //inner track loop |
179 | |
180 | } //outer track loop |
181 | |
182 | PostData(1, fListOfHistos); |
183 | } |
184 | |
be1a7181 |
185 | //________________________________________________________________________ |
186 | void AliResonanceKinkLikeSign::Terminate(Option_t *) |
187 | { |
188 | |
189 | } |
190 | |
10eaad41 |
191 | //____________________________________________________________________// |
192 | |
193 | Float_t AliResonanceKinkLikeSign::GetSigmaToVertex(AliESDtrack* esdTrack) const { |
194 | // Calculates the number of sigma to the vertex. |
195 | |
196 | Float_t b[2]; |
197 | Float_t bRes[2]; |
198 | Float_t bCov[3]; |
199 | |
200 | esdTrack->GetImpactParameters(b,bCov); |
201 | |
202 | if (bCov[0]<=0 || bCov[2]<=0) { |
203 | //AliDebug(1, "Estimated b resolution lower or equal zero!"); |
204 | bCov[0]=0; bCov[2]=0; |
205 | } |
206 | bRes[0] = TMath::Sqrt(bCov[0]); |
207 | bRes[1] = TMath::Sqrt(bCov[2]); |
208 | |
209 | if (bRes[0] == 0 || bRes[1] ==0) return -1; |
210 | |
211 | Float_t d = TMath::Sqrt(TMath::Power(b[0]/bRes[0],2) + TMath::Power(b[1]/bRes[1],2)); |
212 | |
213 | if (TMath::Exp(-d * d / 2) < 1e-10) return 1000; |
214 | |
215 | d = TMath::ErfInverse(1 - TMath::Exp(-d * d / 2)) * TMath::Sqrt(2); |
216 | |
217 | return d; |
218 | } |
219 | |
220 | //____________________________________________________________________// |
221 | |
222 | const AliESDVertex* AliResonanceKinkLikeSign::GetEventVertex(const AliESDEvent* esd) const |
223 | |
224 | { |
225 | // Get the vertex |
226 | |
227 | const AliESDVertex* vertex = esd->GetPrimaryVertex(); |
228 | |
229 | if((vertex->GetStatus()==kTRUE)&&(vertex->GetNContributors()>2)) return vertex; |
230 | else |
231 | { |
232 | vertex = esd->GetPrimaryVertexSPD(); |
233 | if((vertex->GetStatus()==kTRUE)&&(vertex->GetNContributors()>2)) return vertex; |
234 | else |
235 | return 0; |
236 | } |
237 | } |
238 | |
be1a7181 |
239 | //________________________________________________________________________ |
240 | |
241 | Bool_t AliResonanceKinkLikeSign::IsAcceptedForKink(AliESDEvent *localesd, |
242 | const AliESDVertex *localvertex, AliESDtrack* localtrack) { |
243 | // Apply the selections for each kink |
244 | |
245 | Double_t gPt = 0.0, gPx = 0.0, gPy = 0.0, gPz = 0.0; |
246 | Double_t dca[2] = {0.0,0.0}, cov[3] = {0.0,0.0,0.0}; //The impact parameters and their covariance. |
247 | Double_t dca3D = 0.0; |
248 | |
249 | AliExternalTrackParam *tpcTrack = (AliExternalTrackParam *)localtrack->GetTPCInnerParam(); |
250 | if(!tpcTrack) { |
251 | gPt = 0.0; gPx = 0.0; gPy = 0.0; gPz = 0.0; |
252 | dca[0] = -100.; dca[1] = -100.; dca3D = -100.; |
253 | cov[0] = -100.; cov[1] = -100.; cov[2] = -100.; |
254 | } |
255 | else { |
256 | gPt = tpcTrack->Pt(); |
257 | gPx = tpcTrack->Px(); |
258 | gPy = tpcTrack->Py(); |
259 | gPz = tpcTrack->Pz(); |
260 | tpcTrack->PropagateToDCA(localvertex, |
261 | localesd->GetMagneticField(),100.,dca,cov); |
262 | } |
263 | |
264 | if(GetSigmaToVertex(localtrack) > fMaxNSigmaToVertex) { |
265 | if (fDebug > 1) Printf("IsAcceptedKink: Track rejected because it has a %lf sigmas to vertex TPC (max. requested: %lf)", GetSigmaToVertex(localtrack),fMaxNSigmaToVertex); |
266 | return kFALSE; |
267 | } |
268 | |
269 | if(TMath::Abs(dca[0]) > fMaxDCAxy) { |
270 | if (fDebug > 1) Printf("IsAcceptedKink: Track rejected because it has a value of dca(xy) (TPC) of %lf (max. requested: %lf)", TMath::Abs(dca[0]), fMaxDCAxy); |
271 | return kFALSE; |
272 | } |
273 | |
274 | if(TMath::Abs(dca[1]) > fMaxDCAzaxis) { |
275 | if (fDebug > 1) Printf("IsAcceptedKink: Track rejected because it has a value of dca(z) of %lf (max. requested: %lf)", TMath::Abs(dca[1]), fMaxDCAzaxis); |
276 | return kFALSE; |
277 | } |
278 | |
279 | if(gPt <= fMinPtTrackCut) { |
280 | if (fDebug > 1) Printf("IsAcceptedKink: Track rejected because it has a min value of pt of %lf (min. requested: %lf)", gPt, fMinPtTrackCut); |
281 | return kFALSE; |
282 | } |
283 | |
284 | return kTRUE; |
285 | } |
286 | |
287 | //________________________________________________________________________ |
288 | Bool_t AliResonanceKinkLikeSign::IsAcceptedForTrack(AliESDEvent *localesd, const AliESDVertex *localvertex, AliESDtrack *localtrack) { |
289 | // Apply the selections for each track |
290 | |
291 | Double_t gPt = 0.0, gPx = 0.0, gPy = 0.0, gPz = 0.0; |
292 | Double_t dca[2] = {0.0,0.0}, cov[3] = {0.0,0.0,0.0}; //The impact parameters and their covariance. |
293 | Double_t dca3D = 0.0; |
294 | |
295 | AliExternalTrackParam *tpcTrack = (AliExternalTrackParam *)localtrack->GetTPCInnerParam(); |
296 | if(!tpcTrack) { |
297 | gPt = 0.0; gPx = 0.0; gPy = 0.0; gPz = 0.0; |
298 | dca[0] = -100.; dca[1] = -100.; dca3D = -100.; |
299 | cov[0] = -100.; cov[1] = -100.; cov[2] = -100.; |
300 | } |
301 | else { |
302 | gPt = tpcTrack->Pt(); |
303 | gPx = tpcTrack->Px(); |
304 | gPy = tpcTrack->Py(); |
305 | gPz = tpcTrack->Pz(); |
306 | tpcTrack->PropagateToDCA(localvertex, |
307 | localesd->GetMagneticField(),100.,dca,cov); |
308 | } |
309 | |
310 | Int_t fcls[200]; |
311 | Int_t nClustersTPC=localtrack->GetTPCclusters(fcls); |
312 | Float_t chi2perTPCcluster=-1.0; |
313 | if(nClustersTPC!=0) chi2perTPCcluster=(localtrack->GetTPCchi2())/Float_t(nClustersTPC); |
314 | |
315 | Double_t extCov[15]; |
316 | localtrack->GetExternalCovariance(extCov); |
317 | |
318 | if((localtrack->GetStatus() & AliESDtrack::kTPCrefit) == 0) { |
319 | if (fDebug > 1) Printf("IsAccepted: Track rejected because of no refited in TPC"); |
320 | return kFALSE; |
321 | } |
322 | |
323 | if(nClustersTPC < fMinTPCclusters) { |
44b5c57e |
324 | if (fDebug > 1) Printf("IsAccepted: Track rejected because it has a value of nclusters (TPC) of %d (min. requested: %d)", nClustersTPC, fMinTPCclusters); |
be1a7181 |
325 | return kFALSE; |
326 | } |
327 | |
328 | if(chi2perTPCcluster > fMaxChi2PerTPCcluster) { |
329 | if (fDebug > 1) Printf("IsAccepted: Track rejected because it has a value of chi2perTPCcluster of %lf (max. requested: %lf)", chi2perTPCcluster, fMaxChi2PerTPCcluster); |
330 | return kFALSE; |
331 | } |
332 | |
333 | if(extCov[0] > fMaxCov0) { |
9ee1c45f |
334 | if (fDebug > 1) Printf("IsAccepted: Track rejected because it has a value of cov[0] of %lf (max. requested: %lf)", extCov[0], fMaxCov0); |
be1a7181 |
335 | return kFALSE; |
336 | } |
337 | |
338 | if(extCov[2] > fMaxCov2) { |
9ee1c45f |
339 | if (fDebug > 1) Printf("IsAccepted: Track rejected because it has a value of cov[2] of %lf (max. requested: %lf)", extCov[2], fMaxCov2); |
be1a7181 |
340 | return kFALSE; |
341 | } |
342 | |
343 | if(extCov[5] > fMaxCov5) { |
9ee1c45f |
344 | if (fDebug > 1) Printf("IsAccepted: Track rejected because it has a value of cov[5] of %lf (max. requested: %lf)", extCov[5], fMaxCov5); |
be1a7181 |
345 | return kFALSE; |
346 | } |
347 | |
348 | if(extCov[9] > fMaxCov9) { |
9ee1c45f |
349 | if (fDebug > 1) Printf("IsAccepted: Track rejected because it has a value of cov[9] of %lf (max. requested: %lf)", extCov[9], fMaxCov9); |
be1a7181 |
350 | return kFALSE; |
351 | } |
352 | |
353 | if(extCov[14] > fMaxCov14) { |
9ee1c45f |
354 | if (fDebug > 1) Printf("IsAccepted: Track rejected because it has a value of cov[14] of %lf (max. requested: %lf)", extCov[14], fMaxCov14); |
be1a7181 |
355 | return kFALSE; |
356 | } |
357 | |
358 | return kTRUE; |
359 | |
360 | } |
361 | |
362 | //________________________________________________________________________ |
363 | Bool_t AliResonanceKinkLikeSign::IsKink(AliESDEvent *localesd, Int_t kinkIndex, TVector3 trackMom) |
364 | { |
365 | // Test some kinematical criteria for each kink |
366 | |
367 | AliESDkink *kink=localesd->GetKink(TMath::Abs(kinkIndex)-1); |
368 | const TVector3 motherMfromKink(kink->GetMotherP()); |
369 | const TVector3 daughterMKink(kink->GetDaughterP()); |
370 | Float_t qt=kink->GetQt(); |
371 | |
372 | Double_t maxDecAngKmu=f1->Eval(motherMfromKink.Mag(),0.,0.,0.); |
373 | Double_t maxDecAngpimu=f2->Eval(motherMfromKink.Mag(),0.,0.,0.); |
374 | |
375 | Float_t kinkAngle=TMath::RadToDeg()*kink->GetAngle(2); |
376 | |
377 | Float_t energyDaughterMu=TMath::Sqrt(daughterMKink.Mag()*daughterMKink.Mag()+0.105658*0.105658); |
378 | Float_t p1XM= motherMfromKink.Px(); |
379 | Float_t p1YM= motherMfromKink.Py(); |
380 | Float_t p1ZM= motherMfromKink.Pz(); |
381 | Float_t p2XM= daughterMKink.Px(); |
382 | Float_t p2YM= daughterMKink.Py(); |
383 | Float_t p2ZM= daughterMKink.Pz(); |
384 | Float_t p3Daughter=TMath::Sqrt(((p1XM-p2XM)*(p1XM-p2XM))+((p1YM-p2YM)*(p1YM-p2YM))+((p1ZM-p2ZM)*(p1ZM-p2ZM))); |
385 | Double_t invariantMassKmu= TMath::Sqrt((energyDaughterMu+p3Daughter)*(energyDaughterMu+p3Daughter)-motherMfromKink.Mag()*motherMfromKink.Mag()); |
386 | |
e3c296cd |
387 | if((kinkAngle>maxDecAngpimu)&&(qt>fminQt)&&(qt<fmaxQt)&&((kink->GetR()>fminKinkRadius)&&(kink->GetR()<fmaxKinkRadius))&&(TMath::Abs(trackMom.Eta())<fmaxAbsEtaCut)&&(invariantMassKmu<0.6)) { |
be1a7181 |
388 | |
389 | if(trackMom.Mag()<=1.1) { |
390 | return kTRUE; |
391 | } |
392 | else |
393 | if (kinkAngle<maxDecAngKmu) { |
394 | return kTRUE; |
395 | } |
396 | } |
397 | return kFALSE; |
398 | } |