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