]>
Commit | Line | Data |
---|---|---|
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 | |
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 | 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 | |
55 | AlidNdPtCutAnalysis::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 | |
67 | AlidNdPtCutAnalysis::~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 | |
78 | void 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 | |
151 | void 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 | |
70fdd197 | 174 | static AliTriggerAnalysis* triggerAnalysis = new AliTriggerAnalysis;\r |
175 | isEventTriggered = triggerAnalysis->IsTriggerFired(esdEvent, GetTrigger());\r | |
bad4ba69 | 176 | }\r |
177 | \r | |
0aaa8b91 | 178 | // use MC information\r |
179 | AliHeader* header = 0;\r | |
180 | AliGenEventHeader* genHeader = 0;\r | |
181 | AliStack* stack = 0;\r | |
182 | TArrayF vtxMC(3);\r | |
bad4ba69 | 183 | //AlidNdPtHelper::MCProcessType evtType = AlidNdPtHelper::kInvalidProcess;\r |
184 | AliPWG0Helper::MCProcessType evtType = AliPWG0Helper::kInvalidProcess;\r | |
0aaa8b91 | 185 | \r |
186 | if(IsUseMCInfo())\r | |
187 | {\r | |
188 | if(!mcEvent) {\r | |
189 | AliDebug(AliLog::kError, "mcEvent not available");\r | |
190 | return;\r | |
191 | }\r | |
192 | \r | |
193 | // get MC event header\r | |
194 | header = mcEvent->Header();\r | |
195 | if (!header) {\r | |
196 | AliDebug(AliLog::kError, "Header not available");\r | |
197 | return;\r | |
198 | }\r | |
199 | \r | |
200 | // MC particle stack\r | |
201 | stack = mcEvent->Stack();\r | |
202 | if (!stack) {\r | |
203 | AliDebug(AliLog::kError, "Stack not available");\r | |
204 | return;\r | |
205 | }\r | |
206 | \r | |
207 | // get event type (ND=0x1, DD=0x2, SD=0x4)\r | |
bad4ba69 | 208 | evtType = AliPWG0Helper::GetEventProcessType(header);\r |
0aaa8b91 | 209 | AliDebug(AliLog::kDebug+1, Form("Found process type %d", evtType));\r |
210 | \r | |
211 | // get MC vertex\r | |
212 | genHeader = header->GenEventHeader();\r | |
213 | if (!genHeader) {\r | |
214 | AliDebug(AliLog::kError, "Could not retrieve genHeader from Header");\r | |
215 | return;\r | |
216 | }\r | |
217 | genHeader->PrimaryVertex(vtxMC);\r | |
218 | \r | |
219 | // Fill MC event histogram\r | |
220 | Double_t vMCEventHist[3]={vtxMC[0],vtxMC[1],vtxMC[2]};\r | |
221 | fMCEventHist->Fill(vMCEventHist);\r | |
222 | \r | |
223 | } // end bUseMC\r | |
224 | \r | |
225 | // get reconstructed vertex \r | |
bad4ba69 | 226 | Bool_t bRedoTPCVertex = evtCuts->IsRedoTPCVertex();\r |
227 | Bool_t bUseConstraints = evtCuts->IsUseBeamSpotConstraint();\r | |
228 | const AliESDVertex* vtxESD = AlidNdPtHelper::GetVertex(esdEvent,evtCuts,accCuts,esdTrackCuts,GetAnalysisMode(),kFALSE,bRedoTPCVertex,bUseConstraints); \r | |
0aaa8b91 | 229 | if(!vtxESD) return; \r |
230 | \r | |
bad4ba69 | 231 | Bool_t isRecVertex = AlidNdPtHelper::TestRecVertex(vtxESD, GetAnalysisMode(), kFALSE);\r |
0aaa8b91 | 232 | Bool_t isEventOK = evtCuts->AcceptEvent(esdEvent,mcEvent,vtxESD) && isRecVertex;\r |
233 | \r | |
234 | TObjArray *allChargedTracks=0;\r | |
235 | Int_t multAll=0;\r | |
236 | \r | |
237 | // check event cuts\r | |
238 | if(isEventOK && isEventTriggered)\r | |
239 | {\r | |
240 | // get all charged tracks\r | |
bad4ba69 | 241 | allChargedTracks = AlidNdPtHelper::GetAllChargedTracks(esdEvent,GetAnalysisMode());\r |
0aaa8b91 | 242 | if(!allChargedTracks) return;\r |
243 | \r | |
244 | Int_t entries = allChargedTracks->GetEntries();\r | |
245 | for(Int_t i=0; i<entries;++i) \r | |
246 | {\r | |
247 | AliESDtrack *track = (AliESDtrack*)allChargedTracks->At(i);\r | |
248 | if(!track) continue;\r | |
249 | \r | |
250 | FillHistograms(track, stack);\r | |
251 | multAll++;\r | |
252 | }\r | |
0aaa8b91 | 253 | \r |
254 | Double_t vRecEventHist[5] = {vtxESD->GetXv(),vtxESD->GetYv(),vtxESD->GetZv(),vtxESD->GetZRes(),multAll};\r | |
255 | fRecEventHist->Fill(vRecEventHist);\r | |
256 | \r | |
257 | if(IsUseMCInfo()) {\r | |
258 | Double_t vRecMCEventHist[5] = {vtxESD->GetXv()-vtxMC[0],vtxESD->GetYv()-vtxMC[1],vtxESD->GetZv()-vtxMC[2],multAll};\r | |
259 | fRecMCEventHist->Fill(vRecMCEventHist);\r | |
260 | }\r | |
bad4ba69 | 261 | }\r |
0aaa8b91 | 262 | \r |
263 | if(allChargedTracks) delete allChargedTracks; allChargedTracks = 0;\r | |
264 | }\r | |
265 | \r | |
266 | //_____________________________________________________________________________\r | |
267 | void AlidNdPtCutAnalysis::FillHistograms(AliESDtrack *const esdTrack, AliStack *const stack)\r | |
268 | {\r | |
269 | //\r | |
270 | // Fill ESD track and MC histograms \r | |
271 | //\r | |
272 | if(!esdTrack) return;\r | |
bad4ba69 | 273 | if(esdTrack->Charge() == 0.) return;\r |
0aaa8b91 | 274 | \r |
0aaa8b91 | 275 | Float_t pt = esdTrack->Pt();\r |
276 | Float_t eta = esdTrack->Eta();\r | |
277 | Float_t phi = esdTrack->Phi();\r | |
278 | Int_t nClust = esdTrack->GetTPCclusters(0);\r | |
279 | Int_t nFindableClust = esdTrack->GetTPCNclsF();\r | |
280 | \r | |
281 | Float_t chi2PerCluster = 0.;\r | |
282 | if(nClust>0.) chi2PerCluster = esdTrack->GetTPCchi2()/Float_t(nClust);\r | |
283 | \r | |
284 | Float_t clustPerFindClust = 0.;\r | |
285 | if(nFindableClust>0.) clustPerFindClust = Float_t(nClust)/nFindableClust;\r | |
286 | \r | |
287 | Float_t b[2], bCov[3];\r | |
288 | esdTrack->GetImpactParameters(b,bCov);\r | |
289 | \r | |
290 | // kink daughter\r | |
291 | Bool_t isKink = kFALSE;\r | |
292 | if(esdTrack->GetKinkIndex(0)>0) isKink=kTRUE;\r | |
293 | \r | |
294 | //\r | |
295 | // Fill rec vs MC information\r | |
296 | //\r | |
bad4ba69 | 297 | \r |
298 | Bool_t isPrim = kTRUE;\r | |
299 | \r | |
300 | if(IsUseMCInfo()) {\r | |
301 | if(!stack) return;\r | |
302 | Int_t label = TMath::Abs(esdTrack->GetLabel()); \r | |
303 | TParticle* particle = stack->Particle(label);\r | |
304 | if(!particle) return;\r | |
305 | if(particle->GetPDG() && particle->GetPDG()->Charge()==0.) return;\r | |
306 | isPrim = stack->IsPhysicalPrimary(label);\r | |
307 | }\r | |
0aaa8b91 | 308 | \r |
309 | // fill histo\r | |
310 | Double_t vRecMCTrackHist[10] = {nClust,chi2PerCluster,clustPerFindClust,b[0],b[1],eta,phi,pt,isKink,isPrim}; \r | |
311 | fRecMCTrackHist->Fill(vRecMCTrackHist);\r | |
312 | }\r | |
313 | \r | |
314 | //_____________________________________________________________________________\r | |
315 | Long64_t AlidNdPtCutAnalysis::Merge(TCollection* list) \r | |
316 | {\r | |
317 | // Merge list of objects (needed by PROOF)\r | |
318 | \r | |
319 | if (!list)\r | |
320 | return 0;\r | |
321 | \r | |
322 | if (list->IsEmpty())\r | |
323 | return 1;\r | |
324 | \r | |
325 | TIterator* iter = list->MakeIterator();\r | |
326 | TObject* obj = 0;\r | |
327 | \r | |
328 | // collection of generated histograms\r | |
329 | Int_t count=0;\r | |
330 | while((obj = iter->Next()) != 0) {\r | |
331 | AlidNdPtCutAnalysis* entry = dynamic_cast<AlidNdPtCutAnalysis*>(obj);\r | |
332 | if (entry == 0) continue; \r | |
333 | \r | |
334 | // event histo\r | |
335 | fRecEventHist->Add(entry->fRecEventHist);\r | |
336 | fRecMCEventHist->Add(entry->fRecMCEventHist);\r | |
337 | fMCEventHist->Add(entry->fMCEventHist);\r | |
338 | \r | |
339 | // track histo\r | |
340 | fRecMCTrackHist->Add(entry->fRecMCTrackHist);\r | |
341 | \r | |
342 | count++;\r | |
343 | }\r | |
344 | \r | |
345 | return count;\r | |
346 | }\r | |
347 | \r | |
348 | //_____________________________________________________________________________\r | |
349 | void AlidNdPtCutAnalysis::Analyse() \r | |
350 | {\r | |
351 | //\r | |
352 | // Analyse histograms\r | |
353 | //\r | |
354 | TH1::AddDirectory(kFALSE);\r | |
355 | TObjArray *aFolderObj = new TObjArray;\r | |
356 | TH1D *h1D = 0; \r | |
357 | TH2D *h2D = 0; \r | |
358 | \r | |
359 | \r | |
360 | //\r | |
361 | // get cuts\r | |
362 | //\r | |
363 | AlidNdPtEventCuts *evtCuts = GetEventCuts(); \r | |
364 | AlidNdPtAcceptanceCuts *accCuts = GetAcceptanceCuts(); \r | |
365 | AliESDtrackCuts *esdTrackCuts = GetTrackCuts(); \r | |
366 | \r | |
367 | if(!evtCuts || !accCuts || !esdTrackCuts) {\r | |
368 | Error("AlidNdPtCutAnalysis::Analyse()", "cuts not available");\r | |
369 | return;\r | |
370 | }\r | |
371 | \r | |
372 | //\r | |
373 | // set min and max values\r | |
374 | //\r | |
375 | Double_t minPt = accCuts->GetMinPt();\r | |
376 | Double_t maxPt = accCuts->GetMaxPt();\r | |
377 | Double_t minEta = accCuts->GetMinEta();\r | |
378 | Double_t maxEta = accCuts->GetMaxEta()-0.00001;\r | |
379 | \r | |
bad4ba69 | 380 | Double_t maxDCAr = accCuts->GetMaxDCAr();\r |
381 | \r | |
0aaa8b91 | 382 | //\r |
383 | // Create rec. event histograms\r | |
384 | //\r | |
385 | h2D = (TH2D *)fRecEventHist->Projection(0,1);\r | |
386 | h2D->SetName("rec_xv_vs_yv");\r | |
387 | aFolderObj->Add(h2D);\r | |
388 | \r | |
389 | h2D = (TH2D *)fRecEventHist->Projection(0,2);\r | |
390 | h2D->SetName("rec_xv_vs_zv");\r | |
391 | aFolderObj->Add(h2D);\r | |
392 | \r | |
bad4ba69 | 393 | h2D = (TH2D *)fRecEventHist->Projection(3,4);\r |
0aaa8b91 | 394 | h2D->SetName("rec_resZv_vs_Mult");\r |
395 | aFolderObj->Add(h2D);\r | |
396 | \r | |
bad4ba69 | 397 | \r |
398 | //\r | |
399 | // MC available\r | |
400 | //\r | |
401 | if(IsUseMCInfo()) {\r | |
402 | \r | |
0aaa8b91 | 403 | //\r |
404 | // Create mc event histograms\r | |
405 | //\r | |
406 | h2D = (TH2D *)fMCEventHist->Projection(0,1);\r | |
407 | h2D->SetName("mc_xv_vs_yv");\r | |
408 | aFolderObj->Add(h2D);\r | |
409 | \r | |
410 | h2D = (TH2D *)fMCEventHist->Projection(0,2);\r | |
411 | h2D->SetName("mc_xv_vs_zv");\r | |
412 | aFolderObj->Add(h2D);\r | |
413 | \r | |
414 | //\r | |
415 | // Create rec-mc event histograms\r | |
416 | //\r | |
417 | h2D = (TH2D *)fRecMCEventHist->Projection(0,3);\r | |
418 | h2D->SetName("rec_mc_deltaXv_vs_mult");\r | |
419 | aFolderObj->Add(h2D);\r | |
420 | \r | |
421 | h2D = (TH2D *)fRecMCEventHist->Projection(1,3);\r | |
422 | h2D->SetName("rec_mc_deltaYv_vs_mult");\r | |
423 | aFolderObj->Add(h2D);\r | |
424 | \r | |
425 | h2D = (TH2D *)fRecMCEventHist->Projection(2,3);\r | |
426 | h2D->SetName("rec_mc_deltaZv_vs_mult");\r | |
427 | aFolderObj->Add(h2D);\r | |
428 | \r | |
bad4ba69 | 429 | } // end use MC info \r |
430 | \r | |
431 | \r | |
432 | \r | |
0aaa8b91 | 433 | //\r |
434 | // Create rec-mc track track histograms \r | |
435 | //\r | |
436 | \r | |
437 | // DCA cuts\r | |
bad4ba69 | 438 | fRecMCTrackHist->GetAxis(3)->SetRangeUser(-maxDCAr,maxDCAr);\r |
439 | fRecMCTrackHist->GetAxis(4)->SetRangeUser(-maxDCAr,maxDCAr);\r | |
440 | \r | |
441 | h2D = (TH2D *)fRecMCTrackHist->Projection(7,5);\r | |
442 | h2D->SetName("pt_vs_eta");\r | |
443 | aFolderObj->Add(h2D);\r | |
444 | \r | |
0aaa8b91 | 445 | fRecMCTrackHist->GetAxis(7)->SetRangeUser(minPt,maxPt); \r |
446 | \r | |
447 | h2D = (TH2D *)fRecMCTrackHist->Projection(0,5);\r | |
448 | h2D->SetName("nClust_vs_eta");\r | |
449 | aFolderObj->Add(h2D);\r | |
450 | \r | |
451 | h2D = (TH2D *)fRecMCTrackHist->Projection(1,5);\r | |
452 | h2D->SetName("chi2PerClust_vs_eta");\r | |
453 | aFolderObj->Add(h2D);\r | |
454 | \r | |
455 | h2D = (TH2D *)fRecMCTrackHist->Projection(2,5);\r | |
456 | h2D->SetName("ratio_nClust_nFindableClust_vs_eta");\r | |
457 | aFolderObj->Add(h2D);\r | |
458 | \r | |
459 | //\r | |
bad4ba69 | 460 | fRecMCTrackHist->GetAxis(7)->SetRangeUser(minEta,maxEta); \r |
461 | \r | |
0aaa8b91 | 462 | h2D = (TH2D *)fRecMCTrackHist->Projection(0,6);\r |
463 | h2D->SetName("nClust_vs_phi");\r | |
464 | aFolderObj->Add(h2D);\r | |
465 | \r | |
466 | h2D = (TH2D *)fRecMCTrackHist->Projection(1,6);\r | |
467 | h2D->SetName("chi2PerClust_vs_phi");\r | |
468 | aFolderObj->Add(h2D);\r | |
469 | \r | |
470 | h2D = (TH2D *)fRecMCTrackHist->Projection(2,6);\r | |
471 | h2D->SetName("ratio_nClust_nFindableClust_vs_phi");\r | |
472 | aFolderObj->Add(h2D);\r | |
473 | \r | |
bad4ba69 | 474 | h2D = (TH2D *)fRecMCTrackHist->Projection(5,6);\r |
475 | h2D->SetName("eta_vs_phi");\r | |
476 | aFolderObj->Add(h2D);\r | |
477 | \r | |
0aaa8b91 | 478 | //\r |
479 | fRecMCTrackHist->GetAxis(7)->SetRangeUser(0.0,maxPt); \r | |
480 | \r | |
481 | h2D = (TH2D *)fRecMCTrackHist->Projection(0,7);\r | |
482 | h2D->SetName("nClust_vs_pt");\r | |
483 | aFolderObj->Add(h2D);\r | |
484 | \r | |
485 | h2D = (TH2D *)fRecMCTrackHist->Projection(1,7);\r | |
486 | h2D->SetName("chi2PerClust_vs_pt");\r | |
487 | aFolderObj->Add(h2D);\r | |
488 | \r | |
489 | h2D = (TH2D *)fRecMCTrackHist->Projection(2,7);\r | |
490 | h2D->SetName("ratio_nClust_nFindableClust_vs_pt");\r | |
491 | aFolderObj->Add(h2D);\r | |
492 | \r | |
bad4ba69 | 493 | h2D = (TH2D *)fRecMCTrackHist->Projection(6,7);\r |
494 | h2D->SetName("phi_vs_pt");\r | |
495 | aFolderObj->Add(h2D);\r | |
496 | \r | |
497 | \r | |
0aaa8b91 | 498 | // fiducial volume\r |
499 | fRecMCTrackHist->GetAxis(5)->SetRangeUser(minEta,maxEta); \r | |
500 | fRecMCTrackHist->GetAxis(7)->SetRangeUser(minPt,maxPt); \r | |
501 | \r | |
502 | // DCA cuts\r | |
bad4ba69 | 503 | fRecMCTrackHist->GetAxis(3)->SetRangeUser(-maxDCAr,maxDCAr);\r |
504 | fRecMCTrackHist->GetAxis(4)->SetRangeUser(-maxDCAr,maxDCAr);\r | |
0aaa8b91 | 505 | \r |
506 | h2D = (TH2D *)fRecMCTrackHist->Projection(0,1);\r | |
507 | h2D->SetName("nClust_vs_chi2PerClust");\r | |
508 | aFolderObj->Add(h2D);\r | |
509 | \r | |
510 | h2D = (TH2D *)fRecMCTrackHist->Projection(0,2);\r | |
511 | h2D->SetName("nClust_vs_ratio_nClust_nFindableClust");\r | |
512 | aFolderObj->Add(h2D);\r | |
513 | \r | |
514 | // DCAy cuts\r | |
bad4ba69 | 515 | fRecMCTrackHist->GetAxis(3)->SetRange(1,fRecMCTrackHist->GetAxis(3)->GetNbins());\r |
0aaa8b91 | 516 | fRecMCTrackHist->GetAxis(4)->SetRangeUser(-1.0,1.0);\r |
517 | \r | |
518 | // sec\r | |
519 | fRecMCTrackHist->GetAxis(9)->SetRange(1,1);\r | |
520 | h1D = (TH1D *)fRecMCTrackHist->Projection(3);\r | |
521 | h1D->SetName("dcay_sec");\r | |
522 | aFolderObj->Add(h1D);\r | |
523 | \r | |
524 | // prim\r | |
525 | fRecMCTrackHist->GetAxis(9)->SetRange(2,2);\r | |
526 | h1D = (TH1D *)fRecMCTrackHist->Projection(3);\r | |
527 | h1D->SetName("dcay_prim");\r | |
528 | aFolderObj->Add(h1D);\r | |
529 | \r | |
530 | // DCAz cuts\r | |
531 | fRecMCTrackHist->GetAxis(3)->SetRangeUser(-1.0,1.0);\r | |
bad4ba69 | 532 | fRecMCTrackHist->GetAxis(4)->SetRange(1,fRecMCTrackHist->GetAxis(4)->GetNbins());\r |
0aaa8b91 | 533 | \r |
534 | // sec\r | |
535 | fRecMCTrackHist->GetAxis(9)->SetRange(1,1);\r | |
536 | h1D = (TH1D *)fRecMCTrackHist->Projection(4);\r | |
537 | h1D->SetName("dcaz_sec");\r | |
538 | aFolderObj->Add(h1D);\r | |
539 | \r | |
540 | // prim\r | |
541 | fRecMCTrackHist->GetAxis(9)->SetRange(2,2);\r | |
542 | h1D = (TH1D *)fRecMCTrackHist->Projection(4);\r | |
543 | h1D->SetName("dcaz_prim");\r | |
544 | aFolderObj->Add(h1D);\r | |
545 | \r | |
546 | \r | |
547 | // export objects to analysis folder\r | |
548 | fAnalysisFolder = ExportToFolder(aFolderObj);\r | |
549 | \r | |
550 | // delete only TObjArray\r | |
551 | if(aFolderObj) delete aFolderObj;\r | |
552 | }\r | |
553 | \r | |
554 | //_____________________________________________________________________________\r | |
555 | TFolder* AlidNdPtCutAnalysis::ExportToFolder(TObjArray * array) \r | |
556 | {\r | |
557 | // recreate folder avery time and export objects to new one\r | |
558 | //\r | |
559 | AlidNdPtCutAnalysis * comp=this;\r | |
560 | TFolder *folder = comp->GetAnalysisFolder();\r | |
561 | \r | |
562 | TString name, title;\r | |
563 | TFolder *newFolder = 0;\r | |
564 | Int_t i = 0;\r | |
565 | Int_t size = array->GetSize();\r | |
566 | \r | |
567 | if(folder) { \r | |
568 | // get name and title from old folder\r | |
569 | name = folder->GetName(); \r | |
570 | title = folder->GetTitle(); \r | |
571 | \r | |
572 | // delete old one\r | |
573 | delete folder;\r | |
574 | \r | |
575 | // create new one\r | |
576 | newFolder = CreateFolder(name.Data(),title.Data());\r | |
577 | newFolder->SetOwner();\r | |
578 | \r | |
579 | // add objects to folder\r | |
580 | while(i < size) {\r | |
581 | newFolder->Add(array->At(i));\r | |
582 | i++;\r | |
583 | }\r | |
584 | }\r | |
585 | \r | |
586 | return newFolder;\r | |
587 | }\r | |
588 | \r | |
589 | //_____________________________________________________________________________\r | |
590 | TFolder* AlidNdPtCutAnalysis::CreateFolder(TString name,TString title) { \r | |
591 | // create folder for analysed histograms\r | |
592 | //\r | |
593 | TFolder *folder = 0;\r | |
594 | folder = new TFolder(name.Data(),title.Data());\r | |
595 | \r | |
596 | return folder;\r | |
597 | }\r |