]> 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
removed duplications in AlidNdPtHelper and AliPWG0Helper and small changes
[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
15\r
16#include <iostream>\r
17\r
18#include "TFile.h"\r
19#include "TCint.h"\r
20#include "TH1.h"\r
21#include "TH2.h"\r
22\r
23#include "AliHeader.h" \r
24#include "AliGenEventHeader.h" \r
25#include "AliStack.h" \r
26#include "AliESDEvent.h" \r
27#include "AliMCEvent.h" \r
28#include "AliESDtrackCuts.h" \r
29#include "AliLog.h" \r
30\r
31#include "AlidNdPtEventCuts.h"\r
32#include "AlidNdPtAcceptanceCuts.h"\r
33\r
bad4ba69 34#include "AliPWG0Helper.h"\r
35#include "AlidNdPtHelper.h"\r
0aaa8b91 36#include "AlidNdPtCutAnalysis.h"\r
37\r
38using namespace std;\r
39\r
40ClassImp(AlidNdPtCutAnalysis)\r
41\r
42//_____________________________________________________________________________\r
43 AlidNdPtCutAnalysis::AlidNdPtCutAnalysis(): AlidNdPt(),\r
44 fAnalysisFolder(0),\r
45 fRecEventHist(0),\r
46 fMCEventHist(0),\r
47 fRecMCEventHist(0),\r
48 fRecMCTrackHist(0)\r
49{\r
50 // default constructor\r
51 Init();\r
52}\r
53\r
54//_____________________________________________________________________________\r
55AlidNdPtCutAnalysis::AlidNdPtCutAnalysis(Char_t* name, Char_t* title): AlidNdPt(name,title),\r
56 fAnalysisFolder(0),\r
57 fRecEventHist(0),\r
58 fMCEventHist(0),\r
59 fRecMCEventHist(0),\r
60 fRecMCTrackHist(0)\r
61{\r
62 // constructor\r
63 Init();\r
64}\r
65\r
66//_____________________________________________________________________________\r
67AlidNdPtCutAnalysis::~AlidNdPtCutAnalysis() {\r
68 // \r
69 if(fRecEventHist) delete fRecEventHist; fRecEventHist=0;\r
70 if(fMCEventHist) delete fMCEventHist; fMCEventHist=0;\r
71 if(fRecMCEventHist) delete fRecMCEventHist; fRecMCEventHist=0;\r
72 if(fRecMCTrackHist) delete fRecMCTrackHist; fRecMCTrackHist=0;\r
73\r
74 if(fAnalysisFolder) delete fAnalysisFolder; fAnalysisFolder=0;\r
75}\r
76\r
77//_____________________________________________________________________________\r
78void AlidNdPtCutAnalysis::Init(){\r
79 //\r
80 // Init histograms\r
81 //\r
82 const Int_t ptNbins = 56; \r
83 const Double_t ptMin = 0.; \r
84 const Double_t ptMax = 16.; \r
85\r
86 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
87\r
88 //Xv:Yv:Zv:ResZv:Mult\r
89 Int_t binsRecEventHist[5]={100,100,140,100,150};\r
90 Double_t minRecEventHist[5]={-3.,-3.,-35.,0.,0.}; \r
91 Double_t maxRecEventHist[5]={3.,3.,35.,10.,150.}; \r
92 fRecEventHist = new THnSparseF("fRecEventHist","Xv:Yv:Zv:ResZv:Mult",5,binsRecEventHist,minRecEventHist,maxRecEventHist);\r
93 fRecEventHist->GetAxis(0)->SetTitle("Xv (cm)");\r
94 fRecEventHist->GetAxis(1)->SetTitle("Yv (cm)");\r
95 fRecEventHist->GetAxis(2)->SetTitle("Zv (cm)");\r
96 fRecEventHist->GetAxis(3)->SetTitle("ResZv (cm)");\r
97 fRecEventHist->GetAxis(4)->SetTitle("Mult");\r
98 fRecEventHist->Sumw2();\r
99\r
100 //Xv:Yv:Zv\r
101 Int_t binsMCEventHist[3]={100,100,140};\r
102 Double_t minMCEventHist[3]={-0.1,-0.1,-35.}; \r
103 Double_t maxMCEventHist[3]={0.1,0.1,35.}; \r
104 fMCEventHist = new THnSparseF("fMCEventHist","mcXv:mcYv:mcZv",3,binsMCEventHist,minMCEventHist,maxMCEventHist);\r
105 fMCEventHist->GetAxis(0)->SetTitle("mcXv (cm)");\r
106 fMCEventHist->GetAxis(1)->SetTitle("mcYv (cm)");\r
107 fMCEventHist->GetAxis(2)->SetTitle("mcZv (cm)");\r
108 fMCEventHist->Sumw2();\r
109\r
110 //Xv-mcXv:Yv-mcYv:Zv-mcZv:Mult\r
111 Int_t binsRecMCEventHist[4]={100,100,100,150};\r
112 Double_t minRecMCEventHist[4]={-10.0,-10.0,-10.0,0.}; \r
113 Double_t maxRecMCEventHist[4]={10.0,10.0,10.0,150.}; \r
114 fRecMCEventHist = new THnSparseF("fRecMCEventHist","mcXv-Xv:mcYv-Yv:mcZv-Zv:Mult",4,binsRecMCEventHist,minRecMCEventHist,maxRecMCEventHist);\r
115 fRecMCEventHist->GetAxis(0)->SetTitle("mcXv-Xv (cm)");\r
116 fRecMCEventHist->GetAxis(1)->SetTitle("mcYv-Yv (cm)");\r
117 fRecMCEventHist->GetAxis(2)->SetTitle("mcZv-Zv (cm)");\r
118 fRecMCEventHist->GetAxis(3)->SetTitle("Mult");\r
119 fRecMCEventHist->Sumw2();\r
120\r
121 //\r
122 // THnSparse track histograms\r
123 //\r
124\r
125 //nClust:chi2PerClust:nClust/nFindableClust:DCAy:DCAz:eta:phi:pt:isKink:isPrim\r
126 Int_t binsRecMCTrackHist[10]={160,100,100,100,100,30,90,ptNbins, 2,2};\r
127 Double_t minRecMCTrackHist[10]={0., 0., 0., -10.,-10.,-1.5, 0., ptMin, 0., 0.};\r
128 Double_t maxRecMCTrackHist[10]={160.,10.,1.2, 10.,10.,1.5, 2.*TMath::Pi(), ptMax, 2.,2.};\r
129\r
130 fRecMCTrackHist = new THnSparseF("fRecMCTrackHist","nClust:chi2PerClust:nClust/nFindableClust:DCAy:DCAz:eta:phi:pt:isKink:isPrim",10,binsRecMCTrackHist,minRecMCTrackHist,maxRecMCTrackHist);\r
131 fRecMCTrackHist->SetBinEdges(7,binsPt);\r
132\r
133 fRecMCTrackHist->GetAxis(0)->SetTitle("nClust");\r
134 fRecMCTrackHist->GetAxis(1)->SetTitle("chi2PerClust");\r
135 fRecMCTrackHist->GetAxis(2)->SetTitle("nClust/nFindableClust");\r
136 fRecMCTrackHist->GetAxis(3)->SetTitle("DCAy (cm)");\r
137 fRecMCTrackHist->GetAxis(4)->SetTitle("DCAz (cm)");\r
138 fRecMCTrackHist->GetAxis(5)->SetTitle("#eta");\r
139 fRecMCTrackHist->GetAxis(6)->SetTitle("#phi (rad)");\r
140 fRecMCTrackHist->GetAxis(7)->SetTitle("p_{T} (GeV/c)");\r
141 fRecMCTrackHist->GetAxis(8)->SetTitle("isKink");\r
142 fRecMCTrackHist->GetAxis(9)->SetTitle("isPrim");\r
143 fRecMCTrackHist->Sumw2();\r
144\r
145 // init output folder\r
146 fAnalysisFolder = CreateFolder("folderdNdPt","Analysis dNdPt Folder");\r
147\r
148}\r
149\r
150//_____________________________________________________________________________\r
151void AlidNdPtCutAnalysis::Process(AliESDEvent *const esdEvent, AliMCEvent * const mcEvent)\r
152{\r
153 //\r
154 // Process real and/or simulated events\r
155 //\r
156 if(!esdEvent) {\r
157 AliDebug(AliLog::kError, "esdEvent not available");\r
158 return;\r
159 }\r
160\r
bad4ba69 161 // get selection cuts\r
0aaa8b91 162 AlidNdPtEventCuts *evtCuts = GetEventCuts(); \r
163 AlidNdPtAcceptanceCuts *accCuts = GetAcceptanceCuts(); \r
164 AliESDtrackCuts *esdTrackCuts = GetTrackCuts(); \r
165\r
bad4ba69 166 if(!evtCuts || !accCuts || !esdTrackCuts) {\r
0aaa8b91 167 AliDebug(AliLog::kError, "cuts not available");\r
168 return;\r
169 }\r
170\r
bad4ba69 171 // trigger selection\r
172 Bool_t isEventTriggered = kTRUE;\r
173 if(evtCuts->IsTriggerRequired()) {\r
174 isEventTriggered = AliPWG0Helper::IsEventTriggered(esdEvent->GetTriggerMask(), GetTrigger());\r
175 }\r
176\r
0aaa8b91 177 // use MC information\r
178 AliHeader* header = 0;\r
179 AliGenEventHeader* genHeader = 0;\r
180 AliStack* stack = 0;\r
181 TArrayF vtxMC(3);\r
bad4ba69 182 //AlidNdPtHelper::MCProcessType evtType = AlidNdPtHelper::kInvalidProcess;\r
183 AliPWG0Helper::MCProcessType evtType = AliPWG0Helper::kInvalidProcess;\r
0aaa8b91 184\r
185 if(IsUseMCInfo())\r
186 {\r
187 if(!mcEvent) {\r
188 AliDebug(AliLog::kError, "mcEvent not available");\r
189 return;\r
190 }\r
191\r
192 // get MC event header\r
193 header = mcEvent->Header();\r
194 if (!header) {\r
195 AliDebug(AliLog::kError, "Header not available");\r
196 return;\r
197 }\r
198\r
199 // MC particle stack\r
200 stack = mcEvent->Stack();\r
201 if (!stack) {\r
202 AliDebug(AliLog::kError, "Stack not available");\r
203 return;\r
204 }\r
205\r
206 // get event type (ND=0x1, DD=0x2, SD=0x4)\r
bad4ba69 207 evtType = AliPWG0Helper::GetEventProcessType(header);\r
0aaa8b91 208 AliDebug(AliLog::kDebug+1, Form("Found process type %d", evtType));\r
209\r
210 // get MC vertex\r
211 genHeader = header->GenEventHeader();\r
212 if (!genHeader) {\r
213 AliDebug(AliLog::kError, "Could not retrieve genHeader from Header");\r
214 return;\r
215 }\r
216 genHeader->PrimaryVertex(vtxMC);\r
217\r
218 // Fill MC event histogram\r
219 Double_t vMCEventHist[3]={vtxMC[0],vtxMC[1],vtxMC[2]};\r
220 fMCEventHist->Fill(vMCEventHist);\r
221\r
222 } // end bUseMC\r
223\r
224 // get reconstructed vertex \r
bad4ba69 225 Bool_t bRedoTPCVertex = evtCuts->IsRedoTPCVertex();\r
226 Bool_t bUseConstraints = evtCuts->IsUseBeamSpotConstraint();\r
227 const AliESDVertex* vtxESD = AlidNdPtHelper::GetVertex(esdEvent,evtCuts,accCuts,esdTrackCuts,GetAnalysisMode(),kFALSE,bRedoTPCVertex,bUseConstraints); \r
0aaa8b91 228 if(!vtxESD) return; \r
229\r
bad4ba69 230 Bool_t isRecVertex = AlidNdPtHelper::TestRecVertex(vtxESD, GetAnalysisMode(), kFALSE);\r
0aaa8b91 231 Bool_t isEventOK = evtCuts->AcceptEvent(esdEvent,mcEvent,vtxESD) && isRecVertex;\r
232\r
233 TObjArray *allChargedTracks=0;\r
234 Int_t multAll=0;\r
235\r
236 // check event cuts\r
237 if(isEventOK && isEventTriggered)\r
238 {\r
239 // get all charged tracks\r
bad4ba69 240 allChargedTracks = AlidNdPtHelper::GetAllChargedTracks(esdEvent,GetAnalysisMode());\r
0aaa8b91 241 if(!allChargedTracks) return;\r
242\r
243 Int_t entries = allChargedTracks->GetEntries();\r
244 for(Int_t i=0; i<entries;++i) \r
245 {\r
246 AliESDtrack *track = (AliESDtrack*)allChargedTracks->At(i);\r
247 if(!track) continue;\r
248\r
249 FillHistograms(track, stack);\r
250 multAll++;\r
251 }\r
0aaa8b91 252\r
253 Double_t vRecEventHist[5] = {vtxESD->GetXv(),vtxESD->GetYv(),vtxESD->GetZv(),vtxESD->GetZRes(),multAll};\r
254 fRecEventHist->Fill(vRecEventHist);\r
255\r
256 if(IsUseMCInfo()) {\r
257 Double_t vRecMCEventHist[5] = {vtxESD->GetXv()-vtxMC[0],vtxESD->GetYv()-vtxMC[1],vtxESD->GetZv()-vtxMC[2],multAll};\r
258 fRecMCEventHist->Fill(vRecMCEventHist);\r
259 }\r
bad4ba69 260 }\r
0aaa8b91 261\r
262 if(allChargedTracks) delete allChargedTracks; allChargedTracks = 0;\r
263}\r
264\r
265//_____________________________________________________________________________\r
266void AlidNdPtCutAnalysis::FillHistograms(AliESDtrack *const esdTrack, AliStack *const stack)\r
267{\r
268 //\r
269 // Fill ESD track and MC histograms \r
270 //\r
271 if(!esdTrack) return;\r
bad4ba69 272 if(esdTrack->Charge() == 0.) return;\r
0aaa8b91 273\r
0aaa8b91 274 Float_t pt = esdTrack->Pt();\r
275 Float_t eta = esdTrack->Eta();\r
276 Float_t phi = esdTrack->Phi();\r
277 Int_t nClust = esdTrack->GetTPCclusters(0);\r
278 Int_t nFindableClust = esdTrack->GetTPCNclsF();\r
279\r
280 Float_t chi2PerCluster = 0.;\r
281 if(nClust>0.) chi2PerCluster = esdTrack->GetTPCchi2()/Float_t(nClust);\r
282\r
283 Float_t clustPerFindClust = 0.;\r
284 if(nFindableClust>0.) clustPerFindClust = Float_t(nClust)/nFindableClust;\r
285\r
286 Float_t b[2], bCov[3];\r
287 esdTrack->GetImpactParameters(b,bCov);\r
288\r
289 // kink daughter\r
290 Bool_t isKink = kFALSE;\r
291 if(esdTrack->GetKinkIndex(0)>0) isKink=kTRUE;\r
292\r
293 //\r
294 // Fill rec vs MC information\r
295 //\r
bad4ba69 296\r
297 Bool_t isPrim = kTRUE;\r
298\r
299 if(IsUseMCInfo()) {\r
300 if(!stack) return;\r
301 Int_t label = TMath::Abs(esdTrack->GetLabel()); \r
302 TParticle* particle = stack->Particle(label);\r
303 if(!particle) return;\r
304 if(particle->GetPDG() && particle->GetPDG()->Charge()==0.) return;\r
305 isPrim = stack->IsPhysicalPrimary(label);\r
306 }\r
0aaa8b91 307\r
308 // fill histo\r
309 Double_t vRecMCTrackHist[10] = {nClust,chi2PerCluster,clustPerFindClust,b[0],b[1],eta,phi,pt,isKink,isPrim}; \r
310 fRecMCTrackHist->Fill(vRecMCTrackHist);\r
311}\r
312\r
313//_____________________________________________________________________________\r
314Long64_t AlidNdPtCutAnalysis::Merge(TCollection* list) \r
315{\r
316 // Merge list of objects (needed by PROOF)\r
317\r
318 if (!list)\r
319 return 0;\r
320\r
321 if (list->IsEmpty())\r
322 return 1;\r
323\r
324 TIterator* iter = list->MakeIterator();\r
325 TObject* obj = 0;\r
326\r
327 // collection of generated histograms\r
328 Int_t count=0;\r
329 while((obj = iter->Next()) != 0) {\r
330 AlidNdPtCutAnalysis* entry = dynamic_cast<AlidNdPtCutAnalysis*>(obj);\r
331 if (entry == 0) continue; \r
332 \r
333 // event histo\r
334 fRecEventHist->Add(entry->fRecEventHist);\r
335 fRecMCEventHist->Add(entry->fRecMCEventHist);\r
336 fMCEventHist->Add(entry->fMCEventHist);\r
337\r
338 // track histo\r
339 fRecMCTrackHist->Add(entry->fRecMCTrackHist);\r
340 \r
341 count++;\r
342 }\r
343\r
344return count;\r
345}\r
346\r
347//_____________________________________________________________________________\r
348void AlidNdPtCutAnalysis::Analyse() \r
349{\r
350 //\r
351 // Analyse histograms\r
352 //\r
353 TH1::AddDirectory(kFALSE);\r
354 TObjArray *aFolderObj = new TObjArray;\r
355 TH1D *h1D = 0; \r
356 TH2D *h2D = 0; \r
357\r
358\r
359 //\r
360 // get cuts\r
361 //\r
362 AlidNdPtEventCuts *evtCuts = GetEventCuts(); \r
363 AlidNdPtAcceptanceCuts *accCuts = GetAcceptanceCuts(); \r
364 AliESDtrackCuts *esdTrackCuts = GetTrackCuts(); \r
365\r
366 if(!evtCuts || !accCuts || !esdTrackCuts) {\r
367 Error("AlidNdPtCutAnalysis::Analyse()", "cuts not available");\r
368 return;\r
369 }\r
370\r
371 //\r
372 // set min and max values\r
373 //\r
374 Double_t minPt = accCuts->GetMinPt();\r
375 Double_t maxPt = accCuts->GetMaxPt();\r
376 Double_t minEta = accCuts->GetMinEta();\r
377 Double_t maxEta = accCuts->GetMaxEta()-0.00001;\r
378\r
bad4ba69 379 Double_t maxDCAr = accCuts->GetMaxDCAr();\r
380\r
0aaa8b91 381 //\r
382 // Create rec. event histograms\r
383 //\r
384 h2D = (TH2D *)fRecEventHist->Projection(0,1);\r
385 h2D->SetName("rec_xv_vs_yv");\r
386 aFolderObj->Add(h2D);\r
387\r
388 h2D = (TH2D *)fRecEventHist->Projection(0,2);\r
389 h2D->SetName("rec_xv_vs_zv");\r
390 aFolderObj->Add(h2D);\r
391\r
bad4ba69 392 h2D = (TH2D *)fRecEventHist->Projection(3,4);\r
0aaa8b91 393 h2D->SetName("rec_resZv_vs_Mult");\r
394 aFolderObj->Add(h2D);\r
395\r
bad4ba69 396\r
397 //\r
398 // MC available\r
399 //\r
400 if(IsUseMCInfo()) {\r
401\r
0aaa8b91 402 //\r
403 // Create mc event histograms\r
404 //\r
405 h2D = (TH2D *)fMCEventHist->Projection(0,1);\r
406 h2D->SetName("mc_xv_vs_yv");\r
407 aFolderObj->Add(h2D);\r
408\r
409 h2D = (TH2D *)fMCEventHist->Projection(0,2);\r
410 h2D->SetName("mc_xv_vs_zv");\r
411 aFolderObj->Add(h2D);\r
412\r
413 //\r
414 // Create rec-mc event histograms\r
415 //\r
416 h2D = (TH2D *)fRecMCEventHist->Projection(0,3);\r
417 h2D->SetName("rec_mc_deltaXv_vs_mult");\r
418 aFolderObj->Add(h2D);\r
419\r
420 h2D = (TH2D *)fRecMCEventHist->Projection(1,3);\r
421 h2D->SetName("rec_mc_deltaYv_vs_mult");\r
422 aFolderObj->Add(h2D);\r
423\r
424 h2D = (TH2D *)fRecMCEventHist->Projection(2,3);\r
425 h2D->SetName("rec_mc_deltaZv_vs_mult");\r
426 aFolderObj->Add(h2D);\r
427\r
bad4ba69 428 } // end use MC info \r
429\r
430\r
431\r
0aaa8b91 432 //\r
433 // Create rec-mc track track histograms \r
434 //\r
435\r
436 // DCA cuts\r
bad4ba69 437 fRecMCTrackHist->GetAxis(3)->SetRangeUser(-maxDCAr,maxDCAr);\r
438 fRecMCTrackHist->GetAxis(4)->SetRangeUser(-maxDCAr,maxDCAr);\r
439\r
440 h2D = (TH2D *)fRecMCTrackHist->Projection(7,5);\r
441 h2D->SetName("pt_vs_eta");\r
442 aFolderObj->Add(h2D);\r
443\r
0aaa8b91 444 fRecMCTrackHist->GetAxis(7)->SetRangeUser(minPt,maxPt); \r
445\r
446 h2D = (TH2D *)fRecMCTrackHist->Projection(0,5);\r
447 h2D->SetName("nClust_vs_eta");\r
448 aFolderObj->Add(h2D);\r
449\r
450 h2D = (TH2D *)fRecMCTrackHist->Projection(1,5);\r
451 h2D->SetName("chi2PerClust_vs_eta");\r
452 aFolderObj->Add(h2D);\r
453\r
454 h2D = (TH2D *)fRecMCTrackHist->Projection(2,5);\r
455 h2D->SetName("ratio_nClust_nFindableClust_vs_eta");\r
456 aFolderObj->Add(h2D);\r
457\r
458 //\r
bad4ba69 459 fRecMCTrackHist->GetAxis(7)->SetRangeUser(minEta,maxEta); \r
460\r
0aaa8b91 461 h2D = (TH2D *)fRecMCTrackHist->Projection(0,6);\r
462 h2D->SetName("nClust_vs_phi");\r
463 aFolderObj->Add(h2D);\r
464\r
465 h2D = (TH2D *)fRecMCTrackHist->Projection(1,6);\r
466 h2D->SetName("chi2PerClust_vs_phi");\r
467 aFolderObj->Add(h2D);\r
468\r
469 h2D = (TH2D *)fRecMCTrackHist->Projection(2,6);\r
470 h2D->SetName("ratio_nClust_nFindableClust_vs_phi");\r
471 aFolderObj->Add(h2D);\r
472\r
bad4ba69 473 h2D = (TH2D *)fRecMCTrackHist->Projection(5,6);\r
474 h2D->SetName("eta_vs_phi");\r
475 aFolderObj->Add(h2D);\r
476\r
0aaa8b91 477 //\r
478 fRecMCTrackHist->GetAxis(7)->SetRangeUser(0.0,maxPt); \r
479\r
480 h2D = (TH2D *)fRecMCTrackHist->Projection(0,7);\r
481 h2D->SetName("nClust_vs_pt");\r
482 aFolderObj->Add(h2D);\r
483\r
484 h2D = (TH2D *)fRecMCTrackHist->Projection(1,7);\r
485 h2D->SetName("chi2PerClust_vs_pt");\r
486 aFolderObj->Add(h2D);\r
487\r
488 h2D = (TH2D *)fRecMCTrackHist->Projection(2,7);\r
489 h2D->SetName("ratio_nClust_nFindableClust_vs_pt");\r
490 aFolderObj->Add(h2D);\r
491\r
bad4ba69 492 h2D = (TH2D *)fRecMCTrackHist->Projection(6,7);\r
493 h2D->SetName("phi_vs_pt");\r
494 aFolderObj->Add(h2D);\r
495\r
496\r
0aaa8b91 497 // fiducial volume\r
498 fRecMCTrackHist->GetAxis(5)->SetRangeUser(minEta,maxEta); \r
499 fRecMCTrackHist->GetAxis(7)->SetRangeUser(minPt,maxPt); \r
500\r
501 // DCA cuts\r
bad4ba69 502 fRecMCTrackHist->GetAxis(3)->SetRangeUser(-maxDCAr,maxDCAr);\r
503 fRecMCTrackHist->GetAxis(4)->SetRangeUser(-maxDCAr,maxDCAr);\r
0aaa8b91 504\r
505 h2D = (TH2D *)fRecMCTrackHist->Projection(0,1);\r
506 h2D->SetName("nClust_vs_chi2PerClust");\r
507 aFolderObj->Add(h2D);\r
508\r
509 h2D = (TH2D *)fRecMCTrackHist->Projection(0,2);\r
510 h2D->SetName("nClust_vs_ratio_nClust_nFindableClust");\r
511 aFolderObj->Add(h2D);\r
512\r
513 // DCAy cuts\r
bad4ba69 514 fRecMCTrackHist->GetAxis(3)->SetRange(1,fRecMCTrackHist->GetAxis(3)->GetNbins());\r
0aaa8b91 515 fRecMCTrackHist->GetAxis(4)->SetRangeUser(-1.0,1.0);\r
516\r
517 // sec\r
518 fRecMCTrackHist->GetAxis(9)->SetRange(1,1);\r
519 h1D = (TH1D *)fRecMCTrackHist->Projection(3);\r
520 h1D->SetName("dcay_sec");\r
521 aFolderObj->Add(h1D);\r
522\r
523 // prim\r
524 fRecMCTrackHist->GetAxis(9)->SetRange(2,2);\r
525 h1D = (TH1D *)fRecMCTrackHist->Projection(3);\r
526 h1D->SetName("dcay_prim");\r
527 aFolderObj->Add(h1D);\r
528\r
529 // DCAz cuts\r
530 fRecMCTrackHist->GetAxis(3)->SetRangeUser(-1.0,1.0);\r
bad4ba69 531 fRecMCTrackHist->GetAxis(4)->SetRange(1,fRecMCTrackHist->GetAxis(4)->GetNbins());\r
0aaa8b91 532\r
533 // sec\r
534 fRecMCTrackHist->GetAxis(9)->SetRange(1,1);\r
535 h1D = (TH1D *)fRecMCTrackHist->Projection(4);\r
536 h1D->SetName("dcaz_sec");\r
537 aFolderObj->Add(h1D);\r
538\r
539 // prim\r
540 fRecMCTrackHist->GetAxis(9)->SetRange(2,2);\r
541 h1D = (TH1D *)fRecMCTrackHist->Projection(4);\r
542 h1D->SetName("dcaz_prim");\r
543 aFolderObj->Add(h1D);\r
544\r
545\r
546 // export objects to analysis folder\r
547 fAnalysisFolder = ExportToFolder(aFolderObj);\r
548\r
549 // delete only TObjArray\r
550 if(aFolderObj) delete aFolderObj;\r
551}\r
552\r
553//_____________________________________________________________________________\r
554TFolder* AlidNdPtCutAnalysis::ExportToFolder(TObjArray * array) \r
555{\r
556 // recreate folder avery time and export objects to new one\r
557 //\r
558 AlidNdPtCutAnalysis * comp=this;\r
559 TFolder *folder = comp->GetAnalysisFolder();\r
560\r
561 TString name, title;\r
562 TFolder *newFolder = 0;\r
563 Int_t i = 0;\r
564 Int_t size = array->GetSize();\r
565\r
566 if(folder) { \r
567 // get name and title from old folder\r
568 name = folder->GetName(); \r
569 title = folder->GetTitle(); \r
570\r
571 // delete old one\r
572 delete folder;\r
573\r
574 // create new one\r
575 newFolder = CreateFolder(name.Data(),title.Data());\r
576 newFolder->SetOwner();\r
577\r
578 // add objects to folder\r
579 while(i < size) {\r
580 newFolder->Add(array->At(i));\r
581 i++;\r
582 }\r
583 }\r
584\r
585return newFolder;\r
586}\r
587\r
588//_____________________________________________________________________________\r
589TFolder* AlidNdPtCutAnalysis::CreateFolder(TString name,TString title) { \r
590// create folder for analysed histograms\r
591//\r
592TFolder *folder = 0;\r
593 folder = new TFolder(name.Data(),title.Data());\r
594\r
595 return folder;\r
596}\r