]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWG0/dNdPt/AlidNdPtCutAnalysis.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
task derives from AliAnalysisTaskSE
[u/mrichter/AliRoot.git] / PWG0 / dNdPt / AlidNdPtCutAnalysis.cxx
CommitLineData
0aaa8b91 1/**************************************************************************\r
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
3 * *\r
4 * Author: The ALICE Off-line Project. *\r
5 * Contributors are mentioned in the code where appropriate. *\r
6 * *\r
7 * Permission to use, copy, modify and distribute this software and its *\r
8 * documentation strictly for non-commercial purposes is hereby granted *\r
9 * without fee, provided that the above copyright notice appears in all *\r
10 * copies and that both the copyright notice and this permission notice *\r
11 * appear in the supporting documentation. The authors make no claims *\r
12 * about the suitability of this software for any purpose. It is *\r
13 * provided "as is" without express or implied warranty. *\r
14 **************************************************************************/\r
00b81913 15//------------------------------------------------------------------------------\r
16// AlidNdPtCutAnalysis class. \r
17//\r
18// a. functionality:\r
19// - fills generic cut histograms\r
20// - generates cuts (selection criteria)\r
21//\r
22// b. data members:\r
23// - generic cut histograms\r
24// - control histograms\r
25//\r
26// Author: J.Otwinowski 04/11/2008 \r
27//------------------------------------------------------------------------------\r
0aaa8b91 28#include "TH1.h"\r
29#include "TH2.h"\r
30\r
31#include "AliHeader.h" \r
32#include "AliGenEventHeader.h" \r
33#include "AliStack.h" \r
34#include "AliESDEvent.h" \r
35#include "AliMCEvent.h" \r
36#include "AliESDtrackCuts.h" \r
37#include "AliLog.h" \r
7f7dd416 38#include "AliTracker.h" \r
0aaa8b91 39\r
40#include "AlidNdPtEventCuts.h"\r
41#include "AlidNdPtAcceptanceCuts.h"\r
7f7dd416 42#include "AlidNdPtBackgroundCuts.h"\r
17e8c701 43#include "AlidNdPtAnalysis.h"\r
d269b0e6 44#include "AliPhysicsSelection.h"\r
0aaa8b91 45\r
bad4ba69 46#include "AliPWG0Helper.h"\r
47#include "AlidNdPtHelper.h"\r
0aaa8b91 48#include "AlidNdPtCutAnalysis.h"\r
49\r
50using namespace std;\r
51\r
52ClassImp(AlidNdPtCutAnalysis)\r
53\r
54//_____________________________________________________________________________\r
55 AlidNdPtCutAnalysis::AlidNdPtCutAnalysis(): AlidNdPt(),\r
56 fAnalysisFolder(0),\r
985bdfd5 57 fEventCount(0),\r
0aaa8b91 58 fRecEventHist(0),\r
59 fMCEventHist(0),\r
60 fRecMCEventHist(0),\r
61 fRecMCTrackHist(0)\r
62{\r
63 // default constructor\r
64 Init();\r
65}\r
66\r
67//_____________________________________________________________________________\r
68AlidNdPtCutAnalysis::AlidNdPtCutAnalysis(Char_t* name, Char_t* title): AlidNdPt(name,title),\r
69 fAnalysisFolder(0),\r
985bdfd5 70 fEventCount(0),\r
0aaa8b91 71 fRecEventHist(0),\r
72 fMCEventHist(0),\r
73 fRecMCEventHist(0),\r
74 fRecMCTrackHist(0)\r
75{\r
76 // constructor\r
77 Init();\r
78}\r
79\r
80//_____________________________________________________________________________\r
81AlidNdPtCutAnalysis::~AlidNdPtCutAnalysis() {\r
82 // \r
985bdfd5 83 if(fEventCount) delete fEventCount; fEventCount=0;\r
0aaa8b91 84 if(fRecEventHist) delete fRecEventHist; fRecEventHist=0;\r
85 if(fMCEventHist) delete fMCEventHist; fMCEventHist=0;\r
86 if(fRecMCEventHist) delete fRecMCEventHist; fRecMCEventHist=0;\r
87 if(fRecMCTrackHist) delete fRecMCTrackHist; fRecMCTrackHist=0;\r
88\r
89 if(fAnalysisFolder) delete fAnalysisFolder; fAnalysisFolder=0;\r
90}\r
91\r
92//_____________________________________________________________________________\r
93void AlidNdPtCutAnalysis::Init(){\r
94 //\r
95 // Init histograms\r
96 //\r
97 const Int_t ptNbins = 56; \r
98 const Double_t ptMin = 0.; \r
99 const Double_t ptMax = 16.; \r
100\r
101 Double_t binsPt[ptNbins+1] = {0.,0.05,0.1,0.15,0.2,0.25,0.3,0.35,0.4,0.45,0.5,0.55,0.6,0.65,0.7,0.75,0.8,0.85,0.9,0.95,1.0,1.1,1.2,1.3,1.4,1.5,1.6,1.7,1.8,1.9,2.0,2.2,2.4,2.6,2.8,3.0,3.2,3.4,3.6,3.8,4.0,4.5,5.0,5.5,6.0,6.5,7.0,7.5,8.0,9.0,10.0,11.0,12.0,13.0,14.0,15.0,16.0};\r
102\r
985bdfd5 103 // \r
104 Int_t binsEventCount[2]={2,2};\r
105 Double_t minEventCount[2]={0,0}; \r
106 Double_t maxEventCount[2]={2,2}; \r
107 fEventCount = new THnSparseF("fEventCount","trig vs trig+vertex",2,binsEventCount,minEventCount,maxEventCount);\r
108 fEventCount->GetAxis(0)->SetTitle("trig");\r
109 fEventCount->GetAxis(1)->SetTitle("trig+vert");\r
110 fEventCount->Sumw2();\r
111\r
0aaa8b91 112 //Xv:Yv:Zv:ResZv:Mult\r
a8ac5525 113 Double_t kFact = 1.0;\r
00b81913 114\r
f537848a 115 Int_t binsRecEventHist[5]={80,80,100,80,150};\r
00b81913 116 Double_t minRecEventHist[5]={-3.*kFact,-3.*kFact,-35.,0.,0.}; \r
117 Double_t maxRecEventHist[5]={3.*kFact,3.*kFact,35.,10.,150.}; \r
0aaa8b91 118 fRecEventHist = new THnSparseF("fRecEventHist","Xv:Yv:Zv:ResZv:Mult",5,binsRecEventHist,minRecEventHist,maxRecEventHist);\r
119 fRecEventHist->GetAxis(0)->SetTitle("Xv (cm)");\r
120 fRecEventHist->GetAxis(1)->SetTitle("Yv (cm)");\r
121 fRecEventHist->GetAxis(2)->SetTitle("Zv (cm)");\r
122 fRecEventHist->GetAxis(3)->SetTitle("ResZv (cm)");\r
123 fRecEventHist->GetAxis(4)->SetTitle("Mult");\r
124 fRecEventHist->Sumw2();\r
125\r
126 //Xv:Yv:Zv\r
f537848a 127 Int_t binsMCEventHist[3]={80,80,100};\r
0aaa8b91 128 Double_t minMCEventHist[3]={-0.1,-0.1,-35.}; \r
129 Double_t maxMCEventHist[3]={0.1,0.1,35.}; \r
130 fMCEventHist = new THnSparseF("fMCEventHist","mcXv:mcYv:mcZv",3,binsMCEventHist,minMCEventHist,maxMCEventHist);\r
131 fMCEventHist->GetAxis(0)->SetTitle("mcXv (cm)");\r
132 fMCEventHist->GetAxis(1)->SetTitle("mcYv (cm)");\r
133 fMCEventHist->GetAxis(2)->SetTitle("mcZv (cm)");\r
134 fMCEventHist->Sumw2();\r
135\r
136 //Xv-mcXv:Yv-mcYv:Zv-mcZv:Mult\r
137 Int_t binsRecMCEventHist[4]={100,100,100,150};\r
a8ac5525 138 Double_t minRecMCEventHist[4]={-1.0*kFact,-1.0*kFact,-1.0*kFact,0.}; \r
139 Double_t maxRecMCEventHist[4]={1.0*kFact,1.0*kFact,1.0*kFact,150.}; \r
7f7dd416 140 fRecMCEventHist = new THnSparseF("fRecMCEventHist","Xv-mcXv:Yv-mcYv:Zv-mcZv:Mult",4,binsRecMCEventHist,minRecMCEventHist,maxRecMCEventHist);\r
141 fRecMCEventHist->GetAxis(0)->SetTitle("Xv-mcXv (cm)");\r
142 fRecMCEventHist->GetAxis(1)->SetTitle("Yv-mcYv (cm)");\r
143 fRecMCEventHist->GetAxis(2)->SetTitle("Zv-mcZv (cm)");\r
0aaa8b91 144 fRecMCEventHist->GetAxis(3)->SetTitle("Mult");\r
145 fRecMCEventHist->Sumw2();\r
146\r
147 //\r
148 // THnSparse track histograms\r
149 //\r
150\r
7a761ceb 151 //nClust:chi2PerClust:nClust/nFindableClust:DCAy:DCAz:eta:phi:pt:kinkIdx:isPrim:polarity\r
de93884d 152 Int_t binsRecMCTrackHist[11]={160,80,80,100,100,90,90,ptNbins, 3, 2, 2};\r
153 Double_t minRecMCTrackHist[11]={0., 0., 0., -1.,-1.,-1.5, 0., ptMin, -1., 0., 0.};\r
154 Double_t maxRecMCTrackHist[11]={160.,10.,1.2, 1.,1.,1.5, 2.*TMath::Pi(), ptMax, 2., 2., 2.};\r
0aaa8b91 155\r
7a761ceb 156 fRecMCTrackHist = new THnSparseF("fRecMCTrackHist","nClust:chi2PerClust:nClust/nFindableClust:DCAy:DCAz:eta:phi:pt:kinkIdx:isPrim:polarity",11,binsRecMCTrackHist,minRecMCTrackHist,maxRecMCTrackHist);\r
0aaa8b91 157 fRecMCTrackHist->SetBinEdges(7,binsPt);\r
158\r
159 fRecMCTrackHist->GetAxis(0)->SetTitle("nClust");\r
160 fRecMCTrackHist->GetAxis(1)->SetTitle("chi2PerClust");\r
161 fRecMCTrackHist->GetAxis(2)->SetTitle("nClust/nFindableClust");\r
162 fRecMCTrackHist->GetAxis(3)->SetTitle("DCAy (cm)");\r
163 fRecMCTrackHist->GetAxis(4)->SetTitle("DCAz (cm)");\r
164 fRecMCTrackHist->GetAxis(5)->SetTitle("#eta");\r
165 fRecMCTrackHist->GetAxis(6)->SetTitle("#phi (rad)");\r
166 fRecMCTrackHist->GetAxis(7)->SetTitle("p_{T} (GeV/c)");\r
7a761ceb 167 fRecMCTrackHist->GetAxis(8)->SetTitle("kinkIdx"); // 0 - no kink, -1 - kink mother, 1 - kink daugther \r
0aaa8b91 168 fRecMCTrackHist->GetAxis(9)->SetTitle("isPrim");\r
d269b0e6 169 fRecMCTrackHist->GetAxis(10)->SetTitle("polarity");\r
0aaa8b91 170 fRecMCTrackHist->Sumw2();\r
171\r
172 // init output folder\r
173 fAnalysisFolder = CreateFolder("folderdNdPt","Analysis dNdPt Folder");\r
174\r
175}\r
176\r
177//_____________________________________________________________________________\r
178void AlidNdPtCutAnalysis::Process(AliESDEvent *const esdEvent, AliMCEvent * const mcEvent)\r
179{\r
180 //\r
181 // Process real and/or simulated events\r
182 //\r
183 if(!esdEvent) {\r
184 AliDebug(AliLog::kError, "esdEvent not available");\r
185 return;\r
186 }\r
187\r
bad4ba69 188 // get selection cuts\r
0aaa8b91 189 AlidNdPtEventCuts *evtCuts = GetEventCuts(); \r
190 AlidNdPtAcceptanceCuts *accCuts = GetAcceptanceCuts(); \r
191 AliESDtrackCuts *esdTrackCuts = GetTrackCuts(); \r
192\r
bad4ba69 193 if(!evtCuts || !accCuts || !esdTrackCuts) {\r
0aaa8b91 194 AliDebug(AliLog::kError, "cuts not available");\r
195 return;\r
196 }\r
197\r
bad4ba69 198 // trigger selection\r
199 Bool_t isEventTriggered = kTRUE;\r
17e8c701 200 AliPhysicsSelection *trigSel = NULL;\r
201 AliTriggerAnalysis *trigAna = NULL; // needed for andV0\r
202\r
7f7dd416 203 if(evtCuts->IsTriggerRequired()) \r
204 {\r
17e8c701 205 //\r
206 trigSel = GetPhysicsTriggerSelection();\r
7f7dd416 207 if(!trigSel) {\r
208 AliDebug(AliLog::kError, "cannot get trigSel");\r
209 return;\r
210 }\r
17e8c701 211 \r
f537848a 212 if(IsUseMCInfo()) { \r
d269b0e6 213 trigSel->SetAnalyzeMC();\r
214 isEventTriggered = trigSel->IsCollisionCandidate(esdEvent);\r
17e8c701 215\r
216 trigAna = trigSel->GetTriggerAnalysis();\r
217 if(!trigAna) \r
218 return;\r
219\r
220 if(GetTrigger() == AliTriggerAnalysis::kV0AND)\r
221 isEventTriggered = trigAna->IsOfflineTriggerFired(esdEvent, GetTrigger());\r
f537848a 222 }\r
223 else {\r
17e8c701 224 //\r
225 // 0-multiplicity bin for LHC background correction\r
226 //\r
227 if(GetAnalysisMode() == AlidNdPtHelper::kTPCTrackSPDvtx || GetAnalysisMode() == AlidNdPtHelper::kTPCTrackSPDvtxUpdate || \r
228 GetAnalysisMode() == AlidNdPtHelper::kTPCITSHybridTrackSPDvtx || GetAnalysisMode() == AlidNdPtHelper::kTPCITSHybridTrackSPDvtxDCArPt) \r
229 {\r
230 trigSel->SetBin0CallbackViaPointer(&AlidNdPtAnalysis::IsBinZeroTrackSPDvtx);\r
231 } else {\r
232 trigSel->SetBin0CallbackViaPointer(&AlidNdPtAnalysis::IsBinZeroSPDvtx);\r
233 }\r
234\r
d269b0e6 235 isEventTriggered = trigSel->IsCollisionCandidate(esdEvent);\r
17e8c701 236\r
237 trigAna = trigSel->GetTriggerAnalysis();\r
238 if(!trigAna) \r
239 return;\r
240\r
241 if(GetTrigger() == AliTriggerAnalysis::kV0AND)\r
242 isEventTriggered = trigAna->IsOfflineTriggerFired(esdEvent, GetTrigger());\r
f537848a 243 }\r
bad4ba69 244 }\r
245\r
0aaa8b91 246 // use MC information\r
247 AliHeader* header = 0;\r
248 AliGenEventHeader* genHeader = 0;\r
249 AliStack* stack = 0;\r
250 TArrayF vtxMC(3);\r
bad4ba69 251 AliPWG0Helper::MCProcessType evtType = AliPWG0Helper::kInvalidProcess;\r
0aaa8b91 252\r
253 if(IsUseMCInfo())\r
254 {\r
255 if(!mcEvent) {\r
256 AliDebug(AliLog::kError, "mcEvent not available");\r
257 return;\r
258 }\r
259\r
260 // get MC event header\r
261 header = mcEvent->Header();\r
262 if (!header) {\r
263 AliDebug(AliLog::kError, "Header not available");\r
264 return;\r
265 }\r
266\r
267 // MC particle stack\r
268 stack = mcEvent->Stack();\r
269 if (!stack) {\r
270 AliDebug(AliLog::kError, "Stack not available");\r
271 return;\r
272 }\r
273\r
274 // get event type (ND=0x1, DD=0x2, SD=0x4)\r
bad4ba69 275 evtType = AliPWG0Helper::GetEventProcessType(header);\r
0aaa8b91 276 AliDebug(AliLog::kDebug+1, Form("Found process type %d", evtType));\r
277\r
278 // get MC vertex\r
279 genHeader = header->GenEventHeader();\r
280 if (!genHeader) {\r
281 AliDebug(AliLog::kError, "Could not retrieve genHeader from Header");\r
282 return;\r
283 }\r
284 genHeader->PrimaryVertex(vtxMC);\r
285\r
286 // Fill MC event histogram\r
287 Double_t vMCEventHist[3]={vtxMC[0],vtxMC[1],vtxMC[2]};\r
288 fMCEventHist->Fill(vMCEventHist);\r
289\r
290 } // end bUseMC\r
291\r
292 // get reconstructed vertex \r
bad4ba69 293 Bool_t bRedoTPCVertex = evtCuts->IsRedoTPCVertex();\r
294 Bool_t bUseConstraints = evtCuts->IsUseBeamSpotConstraint();\r
295 const AliESDVertex* vtxESD = AlidNdPtHelper::GetVertex(esdEvent,evtCuts,accCuts,esdTrackCuts,GetAnalysisMode(),kFALSE,bRedoTPCVertex,bUseConstraints); \r
791aaf54 296 Bool_t isRecVertex = AlidNdPtHelper::TestRecVertex(vtxESD, esdEvent->GetPrimaryVertexSPD(), GetAnalysisMode(), kFALSE);\r
0aaa8b91 297 Bool_t isEventOK = evtCuts->AcceptEvent(esdEvent,mcEvent,vtxESD) && isRecVertex;\r
298\r
299 TObjArray *allChargedTracks=0;\r
300 Int_t multAll=0;\r
985bdfd5 301 \r
302 //\r
303 // event counter\r
304 // \r
305 //printf("isEventOK %d, isEventTriggered %d \n",isEventOK,isEventTriggered);\r
306\r
307 Bool_t isTrigAndVertex = isEventTriggered && isEventOK;\r
308 Double_t vEventCount[2] = { isEventTriggered, isTrigAndVertex};\r
309 fEventCount->Fill(vEventCount);\r
0aaa8b91 310\r
7f7dd416 311 //\r
312 // cosmic background and splitted tracks\r
313 //\r
314 if(GetParticleMode() == AlidNdPtHelper::kBackgroundTrack) \r
315 {\r
316 AlidNdPtBackgroundCuts *backCuts = GetBackgroundCuts(); \r
317 if(!backCuts) return;\r
318\r
319 for (Int_t iTrack = 0; iTrack < esdEvent->GetNumberOfTracks(); iTrack++) \r
320 {\r
321 AliESDtrack *track1 = esdEvent->GetTrack(iTrack);\r
322 if(!track1) continue; \r
323 if(track1->Charge()==0) continue; \r
324\r
325 for (Int_t jTrack = iTrack+1; jTrack < esdEvent->GetNumberOfTracks(); jTrack++) \r
326 {\r
327 AliESDtrack *track2 = esdEvent->GetTrack(jTrack);\r
328 if(!track2) continue; \r
329 if(track2->Charge()==0) continue; \r
330\r
331 //printf("track2->Charge() %d\n",track2->Charge());\r
332\r
333 backCuts->IsBackgroundTrack(track1, track2);\r
334 }\r
335 }\r
336 }\r
337\r
0aaa8b91 338 // check event cuts\r
339 if(isEventOK && isEventTriggered)\r
340 {\r
341 // get all charged tracks\r
bad4ba69 342 allChargedTracks = AlidNdPtHelper::GetAllChargedTracks(esdEvent,GetAnalysisMode());\r
0aaa8b91 343 if(!allChargedTracks) return;\r
344\r
345 Int_t entries = allChargedTracks->GetEntries();\r
346 for(Int_t i=0; i<entries;++i) \r
347 {\r
348 AliESDtrack *track = (AliESDtrack*)allChargedTracks->At(i);\r
349 if(!track) continue;\r
350\r
7a761ceb 351 if(!esdTrackCuts->AcceptTrack(track)) continue;\r
352\r
7f7dd416 353 //\r
354 Bool_t isOK = kFALSE;\r
355 Double_t x[3]; track->GetXYZ(x);\r
356 Double_t b[3]; AliTracker::GetBxByBz(x,b);\r
357\r
358 //\r
359 // if TPC-ITS hybrid tracking (kTPCITSHybrid)\r
360 // replace track parameters with TPC-ony track parameters\r
361 //\r
791aaf54 362 if( GetAnalysisMode() == AlidNdPtHelper::kTPCITSHybrid || GetAnalysisMode() == AlidNdPtHelper::kTPCITSHybridTrackSPDvtx || GetAnalysisMode() == AlidNdPtHelper::kTPCITSHybridTrackSPDvtxDCArPt) \r
7f7dd416 363 {\r
364 // Relate TPC-only tracks to SPD vertex\r
791aaf54 365 isOK = track->RelateToVertexTPCBxByBz(vtxESD, b, kVeryBig);\r
7f7dd416 366 if(!isOK) continue;\r
367\r
368 // replace esd track parameters with TPCinner\r
369 AliExternalTrackParam *tpcTrack = new AliExternalTrackParam(*(track->GetTPCInnerParam()));\r
370 if (!tpcTrack) return;\r
371 track->Set(tpcTrack->GetX(),tpcTrack->GetAlpha(),tpcTrack->GetParameter(),tpcTrack->GetCovariance());\r
372\r
373 if(tpcTrack) delete tpcTrack; \r
374 } \r
375\r
847e74b2 376 //\r
791aaf54 377 if (GetAnalysisMode()==AlidNdPtHelper::kTPCSPDvtxUpdate || GetAnalysisMode() == AlidNdPtHelper::kTPCTrackSPDvtxUpdate) \r
847e74b2 378 {\r
379 //\r
380 // update track parameters\r
381 //\r
7f7dd416 382 AliExternalTrackParam cParam;\r
791aaf54 383 isOK = track->RelateToVertexTPCBxByBz(vtxESD, b, kVeryBig, &cParam);\r
7f7dd416 384 if(!isOK) continue;\r
385\r
386 track->Set(cParam.GetX(),cParam.GetAlpha(),cParam.GetParameter(),cParam.GetCovariance());\r
847e74b2 387 }\r
388\r
0aaa8b91 389 FillHistograms(track, stack);\r
390 multAll++;\r
391 }\r
0aaa8b91 392\r
7f7dd416 393 Double_t vRecEventHist[5] = {vtxESD->GetXv(),vtxESD->GetYv(),vtxESD->GetZv(),vtxESD->GetZRes(),multAll};\r
394 fRecEventHist->Fill(vRecEventHist);\r
0aaa8b91 395\r
7f7dd416 396 if(IsUseMCInfo()) {\r
397 Double_t vRecMCEventHist[5] = {vtxESD->GetXv()-vtxMC[0],vtxESD->GetYv()-vtxMC[1],vtxESD->GetZv()-vtxMC[2],multAll};\r
398 fRecMCEventHist->Fill(vRecMCEventHist);\r
399 }\r
bad4ba69 400 }\r
0aaa8b91 401\r
402 if(allChargedTracks) delete allChargedTracks; allChargedTracks = 0;\r
f537848a 403\r
0aaa8b91 404}\r
405\r
406//_____________________________________________________________________________\r
00b81913 407void AlidNdPtCutAnalysis::FillHistograms(AliESDtrack *const esdTrack, AliStack *const stack) const\r
0aaa8b91 408{\r
409 //\r
410 // Fill ESD track and MC histograms \r
411 //\r
412 if(!esdTrack) return;\r
bad4ba69 413 if(esdTrack->Charge() == 0.) return;\r
0aaa8b91 414\r
0aaa8b91 415 Float_t pt = esdTrack->Pt();\r
416 Float_t eta = esdTrack->Eta();\r
417 Float_t phi = esdTrack->Phi();\r
7f7dd416 418 //Int_t nClust = esdTrack->GetTPCclusters(0);\r
419 Int_t nClust = esdTrack->GetTPCNclsIter1();\r
0aaa8b91 420 Int_t nFindableClust = esdTrack->GetTPCNclsF();\r
421\r
422 Float_t chi2PerCluster = 0.;\r
7f7dd416 423 //if(nClust>0.) chi2PerCluster = esdTrack->GetTPCchi2()/Float_t(nClust);\r
424 if(nClust>0.) chi2PerCluster = esdTrack->GetTPCchi2Iter1()/Float_t(nClust);\r
0aaa8b91 425\r
426 Float_t clustPerFindClust = 0.;\r
427 if(nFindableClust>0.) clustPerFindClust = Float_t(nClust)/nFindableClust;\r
428\r
429 Float_t b[2], bCov[3];\r
430 esdTrack->GetImpactParameters(b,bCov);\r
431\r
7a761ceb 432 // kink idx\r
433 Int_t kinkIdx = 0;\r
434 //if(esdTrack->GetKinkIndex(0) > 0.) isKink = kTRUE;\r
435 if(esdTrack->GetKinkIndex(0) > 0) kinkIdx = 1; // kink daughter\r
436 else if(esdTrack->GetKinkIndex(0) < 0) kinkIdx = -1; // kink mother\r
437 else kinkIdx = 0; // not kink\r
438\r
439 //printf("esdTrack->GetKinkIndex(0) %d \n", esdTrack->GetKinkIndex(0));\r
440 //printf("esdTrack->GetKinkIndex(1) %d \n", esdTrack->GetKinkIndex(1));\r
441 //printf("esdTrack->GetKinkIndex(2) %d \n", esdTrack->GetKinkIndex(2));\r
442 //printf("kinkIdx %d \n", kinkIdx);\r
0aaa8b91 443\r
444 //\r
445 // Fill rec vs MC information\r
446 //\r
bad4ba69 447 Bool_t isPrim = kTRUE;\r
448\r
449 if(IsUseMCInfo()) {\r
450 if(!stack) return;\r
451 Int_t label = TMath::Abs(esdTrack->GetLabel()); \r
452 TParticle* particle = stack->Particle(label);\r
453 if(!particle) return;\r
454 if(particle->GetPDG() && particle->GetPDG()->Charge()==0.) return;\r
455 isPrim = stack->IsPhysicalPrimary(label);\r
0aaa8b91 456\r
7a761ceb 457 //if(isPrim && pt > 1.5 && kinkIdx == -1) printf("nClust %d \n", nClust);\r
458 }\r
459 \r
0aaa8b91 460 // fill histo\r
d269b0e6 461 Int_t polarity = -2;\r
462 if (esdTrack->Charge() < 0.) polarity = 0; \r
463 else polarity = 1; \r
7a761ceb 464 Double_t vRecMCTrackHist[11] = { nClust,chi2PerCluster,clustPerFindClust,b[0],b[1],eta,phi,pt,kinkIdx,isPrim, polarity }; \r
0aaa8b91 465 fRecMCTrackHist->Fill(vRecMCTrackHist);\r
466}\r
467\r
468//_____________________________________________________________________________\r
00b81913 469Long64_t AlidNdPtCutAnalysis::Merge(TCollection* const list) \r
0aaa8b91 470{\r
471 // Merge list of objects (needed by PROOF)\r
472\r
473 if (!list)\r
474 return 0;\r
475\r
476 if (list->IsEmpty())\r
477 return 1;\r
478\r
479 TIterator* iter = list->MakeIterator();\r
480 TObject* obj = 0;\r
481\r
7f7dd416 482 //TList *collPhysSelection = new TList;\r
483\r
0aaa8b91 484 // collection of generated histograms\r
485 Int_t count=0;\r
486 while((obj = iter->Next()) != 0) {\r
487 AlidNdPtCutAnalysis* entry = dynamic_cast<AlidNdPtCutAnalysis*>(obj);\r
488 if (entry == 0) continue; \r
489 \r
490 // event histo\r
985bdfd5 491 fEventCount->Add(entry->fEventCount);\r
0aaa8b91 492 fRecEventHist->Add(entry->fRecEventHist);\r
493 fRecMCEventHist->Add(entry->fRecMCEventHist);\r
494 fMCEventHist->Add(entry->fMCEventHist);\r
495\r
496 // track histo\r
497 fRecMCTrackHist->Add(entry->fRecMCTrackHist);\r
7f7dd416 498\r
499 // physics selection\r
500 //collPhysSelection->Add(entry->GetPhysicsTriggerSelection());\r
0aaa8b91 501 \r
502 count++;\r
503 }\r
504\r
7f7dd416 505 //AliPhysicsSelection *trigSelection = GetPhysicsTriggerSelection();\r
506 //trigSelection->Merge(collPhysSelection);\r
507\r
508 //if(collPhysSelection) delete collPhysSelection;\r
509\r
0aaa8b91 510return count;\r
511}\r
512\r
513//_____________________________________________________________________________\r
514void AlidNdPtCutAnalysis::Analyse() \r
515{\r
516 //\r
517 // Analyse histograms\r
518 //\r
519 TH1::AddDirectory(kFALSE);\r
520 TObjArray *aFolderObj = new TObjArray;\r
521 TH1D *h1D = 0; \r
522 TH2D *h2D = 0; \r
523\r
524\r
525 //\r
526 // get cuts\r
527 //\r
528 AlidNdPtEventCuts *evtCuts = GetEventCuts(); \r
529 AlidNdPtAcceptanceCuts *accCuts = GetAcceptanceCuts(); \r
530 AliESDtrackCuts *esdTrackCuts = GetTrackCuts(); \r
531\r
532 if(!evtCuts || !accCuts || !esdTrackCuts) {\r
533 Error("AlidNdPtCutAnalysis::Analyse()", "cuts not available");\r
534 return;\r
535 }\r
536\r
537 //\r
538 // set min and max values\r
539 //\r
540 Double_t minPt = accCuts->GetMinPt();\r
541 Double_t maxPt = accCuts->GetMaxPt();\r
542 Double_t minEta = accCuts->GetMinEta();\r
543 Double_t maxEta = accCuts->GetMaxEta()-0.00001;\r
544\r
bad4ba69 545 Double_t maxDCAr = accCuts->GetMaxDCAr();\r
546\r
985bdfd5 547 //\r
548 // Event counters\r
549 //\r
550 h2D = (TH2D*)fEventCount->Projection(0,1);\r
551 h2D->SetName("trig_vs_trigANDvertex");\r
552 aFolderObj->Add(h2D);\r
553\r
314e9442 554 fEventCount->GetAxis(0)->SetRange(2,2); // triggered\r
70a34bda 555 h1D = (TH1D*)fEventCount->Projection(1);\r
985bdfd5 556 h1D->SetTitle("rec. vertex for triggered events");\r
557 h1D->SetName("trigANDvertex");\r
558 aFolderObj->Add(h1D);\r
559\r
0aaa8b91 560 //\r
561 // Create rec. event histograms\r
562 //\r
985bdfd5 563 h1D = (TH1D *)fRecEventHist->Projection(0);\r
564 h1D->SetName("rec_xv");\r
565 aFolderObj->Add(h1D);\r
566\r
567 h1D = (TH1D *)fRecEventHist->Projection(1);\r
568 h1D->SetName("rec_yv");\r
569 aFolderObj->Add(h1D);\r
570\r
571 h1D = (TH1D *)fRecEventHist->Projection(2);\r
572 h1D->SetName("rec_zv");\r
573 aFolderObj->Add(h1D);\r
574\r
575 h2D = (TH2D *)fRecEventHist->Projection(3,4);\r
576 h2D->SetName("rec_resZv_vs_Mult");\r
577 aFolderObj->Add(h2D);\r
578\r
0aaa8b91 579 h2D = (TH2D *)fRecEventHist->Projection(0,1);\r
580 h2D->SetName("rec_xv_vs_yv");\r
581 aFolderObj->Add(h2D);\r
582\r
583 h2D = (TH2D *)fRecEventHist->Projection(0,2);\r
584 h2D->SetName("rec_xv_vs_zv");\r
585 aFolderObj->Add(h2D);\r
586\r
bad4ba69 587 h2D = (TH2D *)fRecEventHist->Projection(3,4);\r
0aaa8b91 588 h2D->SetName("rec_resZv_vs_Mult");\r
589 aFolderObj->Add(h2D);\r
590\r
bad4ba69 591 //\r
592 // MC available\r
593 //\r
594 if(IsUseMCInfo()) {\r
595\r
0aaa8b91 596 //\r
597 // Create mc event histograms\r
598 //\r
599 h2D = (TH2D *)fMCEventHist->Projection(0,1);\r
600 h2D->SetName("mc_xv_vs_yv");\r
601 aFolderObj->Add(h2D);\r
602\r
603 h2D = (TH2D *)fMCEventHist->Projection(0,2);\r
604 h2D->SetName("mc_xv_vs_zv");\r
605 aFolderObj->Add(h2D);\r
606\r
607 //\r
608 // Create rec-mc event histograms\r
609 //\r
610 h2D = (TH2D *)fRecMCEventHist->Projection(0,3);\r
611 h2D->SetName("rec_mc_deltaXv_vs_mult");\r
612 aFolderObj->Add(h2D);\r
613\r
614 h2D = (TH2D *)fRecMCEventHist->Projection(1,3);\r
615 h2D->SetName("rec_mc_deltaYv_vs_mult");\r
616 aFolderObj->Add(h2D);\r
617\r
618 h2D = (TH2D *)fRecMCEventHist->Projection(2,3);\r
619 h2D->SetName("rec_mc_deltaZv_vs_mult");\r
620 aFolderObj->Add(h2D);\r
621\r
bad4ba69 622 } // end use MC info \r
623\r
624\r
625\r
0aaa8b91 626 //\r
627 // Create rec-mc track track histograms \r
628 //\r
629\r
630 // DCA cuts\r
bad4ba69 631 fRecMCTrackHist->GetAxis(3)->SetRangeUser(-maxDCAr,maxDCAr);\r
632 fRecMCTrackHist->GetAxis(4)->SetRangeUser(-maxDCAr,maxDCAr);\r
633\r
634 h2D = (TH2D *)fRecMCTrackHist->Projection(7,5);\r
635 h2D->SetName("pt_vs_eta");\r
636 aFolderObj->Add(h2D);\r
637\r
0aaa8b91 638 fRecMCTrackHist->GetAxis(7)->SetRangeUser(minPt,maxPt); \r
639\r
640 h2D = (TH2D *)fRecMCTrackHist->Projection(0,5);\r
641 h2D->SetName("nClust_vs_eta");\r
642 aFolderObj->Add(h2D);\r
643\r
644 h2D = (TH2D *)fRecMCTrackHist->Projection(1,5);\r
645 h2D->SetName("chi2PerClust_vs_eta");\r
646 aFolderObj->Add(h2D);\r
647\r
648 h2D = (TH2D *)fRecMCTrackHist->Projection(2,5);\r
649 h2D->SetName("ratio_nClust_nFindableClust_vs_eta");\r
650 aFolderObj->Add(h2D);\r
651\r
985bdfd5 652 h2D = (TH2D *)fRecMCTrackHist->Projection(5,6);\r
653 h2D->SetName("eta_vs_phi");\r
654 aFolderObj->Add(h2D);\r
655\r
0aaa8b91 656 //\r
985bdfd5 657 fRecMCTrackHist->GetAxis(5)->SetRangeUser(minEta,maxEta); \r
bad4ba69 658\r
0aaa8b91 659 h2D = (TH2D *)fRecMCTrackHist->Projection(0,6);\r
660 h2D->SetName("nClust_vs_phi");\r
661 aFolderObj->Add(h2D);\r
662\r
663 h2D = (TH2D *)fRecMCTrackHist->Projection(1,6);\r
664 h2D->SetName("chi2PerClust_vs_phi");\r
665 aFolderObj->Add(h2D);\r
666\r
667 h2D = (TH2D *)fRecMCTrackHist->Projection(2,6);\r
668 h2D->SetName("ratio_nClust_nFindableClust_vs_phi");\r
669 aFolderObj->Add(h2D);\r
670\r
671 //\r
672 fRecMCTrackHist->GetAxis(7)->SetRangeUser(0.0,maxPt); \r
673\r
674 h2D = (TH2D *)fRecMCTrackHist->Projection(0,7);\r
675 h2D->SetName("nClust_vs_pt");\r
676 aFolderObj->Add(h2D);\r
677\r
678 h2D = (TH2D *)fRecMCTrackHist->Projection(1,7);\r
679 h2D->SetName("chi2PerClust_vs_pt");\r
680 aFolderObj->Add(h2D);\r
681\r
682 h2D = (TH2D *)fRecMCTrackHist->Projection(2,7);\r
683 h2D->SetName("ratio_nClust_nFindableClust_vs_pt");\r
684 aFolderObj->Add(h2D);\r
685\r
bad4ba69 686 h2D = (TH2D *)fRecMCTrackHist->Projection(6,7);\r
687 h2D->SetName("phi_vs_pt");\r
688 aFolderObj->Add(h2D);\r
689\r
690\r
0aaa8b91 691 // fiducial volume\r
692 fRecMCTrackHist->GetAxis(5)->SetRangeUser(minEta,maxEta); \r
693 fRecMCTrackHist->GetAxis(7)->SetRangeUser(minPt,maxPt); \r
694\r
695 // DCA cuts\r
bad4ba69 696 fRecMCTrackHist->GetAxis(3)->SetRangeUser(-maxDCAr,maxDCAr);\r
697 fRecMCTrackHist->GetAxis(4)->SetRangeUser(-maxDCAr,maxDCAr);\r
0aaa8b91 698\r
699 h2D = (TH2D *)fRecMCTrackHist->Projection(0,1);\r
700 h2D->SetName("nClust_vs_chi2PerClust");\r
701 aFolderObj->Add(h2D);\r
702\r
703 h2D = (TH2D *)fRecMCTrackHist->Projection(0,2);\r
704 h2D->SetName("nClust_vs_ratio_nClust_nFindableClust");\r
705 aFolderObj->Add(h2D);\r
706\r
985bdfd5 707 //\r
0aaa8b91 708 // DCAy cuts\r
985bdfd5 709 //\r
710 fRecMCTrackHist->GetAxis(0)->SetRange(50,160); // nClust/track > 50\r
711 fRecMCTrackHist->GetAxis(1)->SetRangeUser(0.,3.9999); // chi2/cluster < 4.0\r
bad4ba69 712 fRecMCTrackHist->GetAxis(3)->SetRange(1,fRecMCTrackHist->GetAxis(3)->GetNbins());\r
985bdfd5 713 //fRecMCTrackHist->GetAxis(4)->SetRangeUser(-1.0,1.0);\r
714 fRecMCTrackHist->GetAxis(4)->SetRange(1,fRecMCTrackHist->GetAxis(4)->GetNbins());\r
0aaa8b91 715\r
716 // sec\r
717 fRecMCTrackHist->GetAxis(9)->SetRange(1,1);\r
718 h1D = (TH1D *)fRecMCTrackHist->Projection(3);\r
719 h1D->SetName("dcay_sec");\r
720 aFolderObj->Add(h1D);\r
721\r
722 // prim\r
723 fRecMCTrackHist->GetAxis(9)->SetRange(2,2);\r
724 h1D = (TH1D *)fRecMCTrackHist->Projection(3);\r
725 h1D->SetName("dcay_prim");\r
726 aFolderObj->Add(h1D);\r
727\r
728 // DCAz cuts\r
985bdfd5 729 //fRecMCTrackHist->GetAxis(3)->SetRangeUser(-1.0,1.0);\r
bad4ba69 730 fRecMCTrackHist->GetAxis(4)->SetRange(1,fRecMCTrackHist->GetAxis(4)->GetNbins());\r
0aaa8b91 731\r
732 // sec\r
733 fRecMCTrackHist->GetAxis(9)->SetRange(1,1);\r
734 h1D = (TH1D *)fRecMCTrackHist->Projection(4);\r
735 h1D->SetName("dcaz_sec");\r
736 aFolderObj->Add(h1D);\r
737\r
738 // prim\r
739 fRecMCTrackHist->GetAxis(9)->SetRange(2,2);\r
740 h1D = (TH1D *)fRecMCTrackHist->Projection(4);\r
741 h1D->SetName("dcaz_prim");\r
742 aFolderObj->Add(h1D);\r
743\r
744\r
745 // export objects to analysis folder\r
746 fAnalysisFolder = ExportToFolder(aFolderObj);\r
747\r
748 // delete only TObjArray\r
749 if(aFolderObj) delete aFolderObj;\r
750}\r
751\r
752//_____________________________________________________________________________\r
c931948b 753TFolder* AlidNdPtCutAnalysis::ExportToFolder(TObjArray * const array) \r
0aaa8b91 754{\r
755 // recreate folder avery time and export objects to new one\r
756 //\r
757 AlidNdPtCutAnalysis * comp=this;\r
758 TFolder *folder = comp->GetAnalysisFolder();\r
759\r
760 TString name, title;\r
761 TFolder *newFolder = 0;\r
762 Int_t i = 0;\r
763 Int_t size = array->GetSize();\r
764\r
765 if(folder) { \r
766 // get name and title from old folder\r
767 name = folder->GetName(); \r
768 title = folder->GetTitle(); \r
769\r
770 // delete old one\r
771 delete folder;\r
772\r
773 // create new one\r
774 newFolder = CreateFolder(name.Data(),title.Data());\r
775 newFolder->SetOwner();\r
776\r
777 // add objects to folder\r
778 while(i < size) {\r
779 newFolder->Add(array->At(i));\r
780 i++;\r
781 }\r
782 }\r
783\r
784return newFolder;\r
785}\r
786\r
787//_____________________________________________________________________________\r
788TFolder* AlidNdPtCutAnalysis::CreateFolder(TString name,TString title) { \r
789// create folder for analysed histograms\r
790//\r
791TFolder *folder = 0;\r
792 folder = new TFolder(name.Data(),title.Data());\r
793\r
794 return folder;\r
795}\r