]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGPP/TRD/AliTRDinfoGen.cxx
fix coverity
[u/mrichter/AliRoot.git] / PWGPP / TRD / AliTRDinfoGen.cxx
CommitLineData
94b94be0 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/* $Id: AliTRDinfoGen.cxx 27496 2008-07-22 08:35:45Z cblume $ */\r
17\r
18////////////////////////////////////////////////////////////////////////////\r
19//\r
20// Tender wagon for TRD performance/calibration train\r
21//\r
22// In this wagon the information from\r
23// - ESD\r
24// - Friends [if available]\r
25// - MC [if available]\r
26// are grouped into AliTRDtrackInfo objects and fed to worker tasks\r
27//\r
28// Authors:\r
29// Markus Fasel <M.Fasel@gsi.de>\r
30// Alexandru Bercuci <A.Bercuci@gsi.de>\r
31//\r
32////////////////////////////////////////////////////////////////////////////\r
33\r
34#include <TClonesArray.h>\r
35#include <TObjArray.h>\r
36#include <TObject.h>\r
37#include <TString.h>\r
c9b23652 38#include <TVectorT.h>\r
35983729 39#include <TH1.h>\r
40#include <TH2.h>\r
c9b23652 41#include <TCanvas.h>\r
94b94be0 42#include <TPad.h>\r
43#include <TFile.h>\r
44#include <TTree.h>\r
45#include <TROOT.h>\r
46#include <TChain.h>\r
47#include <TParticle.h>\r
48\r
49#include "AliLog.h"\r
50#include "AliAnalysisManager.h"\r
51#include "AliGeomManager.h"\r
52#include "AliCDBManager.h"\r
4e38ea4c 53#include "AliCDBStorage.h"\r
94b94be0 54#include "AliCDBEntry.h"\r
55#include "AliCDBPath.h"\r
56#include "AliESDEvent.h"\r
57#include "AliMCEvent.h"\r
58#include "AliESDInputHandler.h"\r
f073d500 59#include "AliESDInputHandlerRP.h"\r
94b94be0 60#include "AliMCEventHandler.h"\r
61\r
62#include "AliESDfriend.h"\r
63#include "AliESDfriendTrack.h"\r
64#include "AliESDHeader.h"\r
65#include "AliESDRun.h"\r
66#include "AliESDtrack.h"\r
67#include "AliESDv0.h"\r
68#include "AliESDtrackCuts.h"\r
69#include "AliMCParticle.h"\r
70#include "AliMultiplicity.h"\r
6687261d 71#include "AliCentrality.h"\r
94b94be0 72#include "AliPID.h"\r
73#include "AliStack.h"\r
74#include "AliTrackReference.h"\r
75#include "TTreeStream.h"\r
76\r
77#include <cstdio>\r
78#include <climits>\r
79#include <cstring>\r
80#include <iostream>\r
81#include <memory>\r
82\r
83#include "AliTRDReconstructor.h"\r
84#include "AliTRDrecoParam.h"\r
85#include "AliTRDcalibDB.h"\r
86#include "AliTRDtrackerV1.h"\r
c9b23652 87#include "AliTRDpadPlane.h"\r
94b94be0 88#include "AliTRDgeometry.h"\r
89#include "AliTRDtrackV1.h"\r
90#include "AliTRDseedV1.h"\r
91#include "AliTRDcluster.h"\r
92#include "AliTRDinfoGen.h"\r
2bfe5463 93#include "AliTRDpwgppHelper.h"\r
94b94be0 94#include "info/AliTRDtrackInfo.h"\r
95#include "info/AliTRDeventInfo.h"\r
96#include "info/AliTRDv0Info.h"\r
97#include "info/AliTRDeventCuts.h"\r
98\r
99ClassImp(AliTRDinfoGen)\r
100\r
101const Float_t AliTRDinfoGen::fgkITS = 100.; // to be checked\r
102const Float_t AliTRDinfoGen::fgkTPC = 290.;\r
103const Float_t AliTRDinfoGen::fgkTRD = 365.;\r
104\r
94b94be0 105const Float_t AliTRDinfoGen::fgkTrkDCAxy = 3.;\r
106const Float_t AliTRDinfoGen::fgkTrkDCAz = 10.;\r
107const Int_t AliTRDinfoGen::fgkNclTPC = 70;\r
108const Float_t AliTRDinfoGen::fgkPt = 0.2;\r
109const Float_t AliTRDinfoGen::fgkEta = 0.9;\r
110AliTRDReconstructor* AliTRDinfoGen::fgReconstructor(NULL);\r
111AliTRDgeometry* AliTRDinfoGen::fgGeo(NULL);\r
112//____________________________________________________________________\r
113AliTRDinfoGen::AliTRDinfoGen()\r
114 :AliAnalysisTaskSE()\r
94b94be0 115 ,fESDev(NULL)\r
116 ,fMCev(NULL)\r
117 ,fEventCut(NULL)\r
118 ,fTrackCut(NULL)\r
119 ,fV0Cut(NULL)\r
120 ,fOCDB("local://$ALICE_ROOT/OCDB")\r
121 ,fTrackInfo(NULL)\r
122 ,fEventInfo(NULL)\r
123 ,fV0Info(NULL)\r
124 ,fTracksBarrel(NULL)\r
125 ,fTracksSA(NULL)\r
126 ,fTracksKink(NULL)\r
127 ,fV0List(NULL)\r
f073d500 128 ,fClusters(NULL)\r
94b94be0 129 ,fContainer(NULL)\r
246fe7f7 130 ,fRecos(NULL)\r
94b94be0 131 ,fDebugStream(NULL)\r
132{\r
133 //\r
134 // Default constructor\r
135 //\r
136 SetNameTitle("TRDinfoGen", "MC-REC TRD-track list generator");\r
137}\r
138\r
139//____________________________________________________________________\r
140AliTRDinfoGen::AliTRDinfoGen(char* name)\r
141 :AliAnalysisTaskSE(name)\r
94b94be0 142 ,fESDev(NULL)\r
143 ,fMCev(NULL)\r
144 ,fEventCut(NULL)\r
145 ,fTrackCut(NULL)\r
146 ,fV0Cut(NULL)\r
147 ,fOCDB("local://$ALICE_ROOT/OCDB")\r
148 ,fTrackInfo(NULL)\r
149 ,fEventInfo(NULL)\r
150 ,fV0Info(NULL)\r
151 ,fTracksBarrel(NULL)\r
152 ,fTracksSA(NULL)\r
153 ,fTracksKink(NULL)\r
154 ,fV0List(NULL)\r
f073d500 155 ,fClusters(NULL)\r
94b94be0 156 ,fContainer(NULL)\r
246fe7f7 157 ,fRecos(NULL)\r
94b94be0 158 ,fDebugStream(NULL)\r
159{\r
160 //\r
161 // Default constructor\r
162 //\r
163 SetTitle("MC-REC TRD-track list generator");\r
2bfe5463 164 DefineOutput(AliTRDpwgppHelper::kTracksBarrel, TObjArray::Class());\r
165 DefineOutput(AliTRDpwgppHelper::kTracksSA, TObjArray::Class());\r
166 DefineOutput(AliTRDpwgppHelper::kTracksKink, TObjArray::Class());\r
167 DefineOutput(AliTRDpwgppHelper::kEventInfo, AliTRDeventInfo::Class());\r
168 DefineOutput(AliTRDpwgppHelper::kV0List, TObjArray::Class());\r
169 DefineOutput(AliTRDpwgppHelper::kClusters, TObjArray::Class());\r
170 DefineOutput(AliTRDpwgppHelper::kMonitor, TObjArray::Class()); // histogram list\r
94b94be0 171}\r
172\r
173//____________________________________________________________________\r
174AliTRDinfoGen::~AliTRDinfoGen()\r
175{\r
176// Destructor\r
177 if(fgGeo) delete fgGeo;\r
178 if(fgReconstructor) delete fgReconstructor;\r
179 if(fDebugStream) delete fDebugStream;\r
94b94be0 180 if(fV0Cut) delete fV0Cut;\r
181 if(fTrackCut) delete fTrackCut;\r
182 if(fEventCut) delete fEventCut;\r
183 if(fTrackInfo) delete fTrackInfo; fTrackInfo = NULL;\r
184 if(fEventInfo) delete fEventInfo; fEventInfo = NULL;\r
185 if(fV0Info) delete fV0Info; fV0Info = NULL;\r
f073d500 186 if(fTracksBarrel){\r
94b94be0 187 fTracksBarrel->Delete(); delete fTracksBarrel;\r
188 fTracksBarrel = NULL;\r
189 }\r
f073d500 190 if(fTracksSA){\r
94b94be0 191 fTracksSA->Delete(); delete fTracksSA;\r
192 fTracksSA = NULL;\r
193 }\r
194 if(fTracksKink){ \r
195 fTracksKink->Delete(); delete fTracksKink;\r
196 fTracksKink = NULL;\r
197 }\r
198 if(fV0List){ \r
199 fV0List->Delete(); \r
200 delete fV0List;\r
201 fV0List = NULL;\r
202 }\r
f073d500 203 if(fClusters){\r
204 fClusters->Delete(); delete fClusters;\r
205 fClusters = NULL;\r
206 }\r
c9b23652 207 if(fContainer && !(AliAnalysisManager::GetAnalysisManager() && AliAnalysisManager::GetAnalysisManager()->IsProofMode())){\r
94b94be0 208 fContainer->Delete(); \r
209 delete fContainer;\r
210 fContainer = NULL;\r
211 }\r
212}\r
213\r
214//____________________________________________________________________\r
215Bool_t AliTRDinfoGen::GetRefFigure(Int_t)\r
216{\r
2bfe5463 217// General graphs for PWGPP/TRD train\r
94b94be0 218 if(!gPad){\r
219 AliWarning("Please provide a canvas to draw results.");\r
220 return kFALSE;\r
221 }\r
3ceb45ae 222 fContainer->At(kStatTrk)->Draw("bar");\r
94b94be0 223 return kTRUE;\r
224}\r
225\r
226//____________________________________________________________________\r
227void AliTRDinfoGen::UserCreateOutputObjects()\r
228{ \r
229 //\r
230 // Create Output Containers (TObjectArray containing 1D histograms)\r
231 //\r
232 \r
233 fTrackInfo = new AliTRDtrackInfo();\r
234 fEventInfo = new AliTRDeventInfo();\r
235 fV0Info = new AliTRDv0Info();\r
236 fTracksBarrel = new TObjArray(200); fTracksBarrel->SetOwner(kTRUE);\r
237 fTracksSA = new TObjArray(20); fTracksSA->SetOwner(kTRUE);\r
238 fTracksKink = new TObjArray(20); fTracksKink->SetOwner(kTRUE);\r
239 fV0List = new TObjArray(10); fV0List->SetOwner(kTRUE);\r
f073d500 240 fClusters = new TObjArray(AliTRDgeometry::kNdet); fClusters->SetOwner(kTRUE);\r
94b94be0 241\r
242 // define general monitor\r
3ceb45ae 243 fContainer = new TObjArray(kNclasses); fContainer->SetOwner(kTRUE);\r
94b94be0 244 TH1 *h=new TH1I("hStat", "Run statistics;Observable;Entries", Int_t(kNObjects), -0.5, Float_t(kNObjects)-0.5);\r
245 TAxis *ax(h->GetXaxis());\r
246 ax->SetBinLabel(Int_t(kTracksESD) + 1, "ESD");\r
247 ax->SetBinLabel(Int_t(kTracksMC) + 1, "MC");\r
248 ax->SetBinLabel(Int_t(kV0) + 1, "V0");\r
249 ax->SetBinLabel(Int_t(kTPC) + 1, "TPC");\r
33056e04 250 ax->SetBinLabel(Int_t(kITS) + 1, "ITS");\r
94b94be0 251 ax->SetBinLabel(Int_t(kTRDin) + 1, "TRDin");\r
252 ax->SetBinLabel(Int_t(kTRDout) + 1, "TRDout");\r
253 ax->SetBinLabel(Int_t(kBarrel) + 1, "Barrel");\r
254 ax->SetBinLabel(Int_t(kBarrelMC) + 1, "BarrelMC");\r
255 ax->SetBinLabel(Int_t(kSA) + 1, "SA");\r
256 ax->SetBinLabel(Int_t(kSAMC) + 1, "SAMC");\r
257 ax->SetBinLabel(Int_t(kKink) + 1, "Kink");\r
258 ax->SetBinLabel(Int_t(kKinkMC) + 1, "KinkMC");\r
259 ax->SetBinLabel(Int_t(kBarrelFriend) + 1, "BFriend");\r
260 ax->SetBinLabel(Int_t(kSAFriend) + 1, "SFriend");\r
3ceb45ae 261 fContainer->AddAt(h, kStatTrk);\r
246fe7f7 262 h=new TH1I("hEv", "Run statistics;Event Class;Entries", 4, -0.5, 3.5);\r
263 ax = h->GetXaxis();\r
264 ax->SetBinLabel(1, "Low");\r
265 ax->SetBinLabel(2, "High");\r
266 ax->SetBinLabel(3, "Cosmic");\r
267 ax->SetBinLabel(4, "Calib");\r
3ceb45ae 268 fContainer->AddAt(h, kEvType);\r
6687261d 269 TH2I* h2=new TH2I("hBCtrack", "Track Statistics;Fill Bunch;TOF BC;Entries", 3500, -0.5, 3499.5, 31, -10.5, 20.5);\r
35983729 270 fContainer->AddAt(h2, kBC);\r
6465da91 271 h=new TH1I("hTriggers", "Triggers statistics;;Entries", 21, -0.5, 20.5);\r
272 fContainer->AddAt(h, kTrigger);\r
c9b23652 273 TObjArray *chmb = new TObjArray(AliTRDgeometry::kNdet);\r
274 chmb->SetName("Chambers"); chmb->SetOwner();\r
275 fContainer->AddAt(chmb, kChmb);\r
f073d500 276\r
2bfe5463 277 PostData(AliTRDpwgppHelper::kTracksBarrel, fTracksBarrel);\r
278 PostData(AliTRDpwgppHelper::kTracksSA, fTracksSA);\r
279 PostData(AliTRDpwgppHelper::kTracksKink, fTracksKink);\r
280 PostData(AliTRDpwgppHelper::kEventInfo, fEventInfo);\r
281 PostData(AliTRDpwgppHelper::kV0List, fV0List);\r
282 PostData(AliTRDpwgppHelper::kClusters, fClusters);\r
283 PostData(AliTRDpwgppHelper::kMonitor, fContainer);\r
94b94be0 284}\r
285\r
286//____________________________________________________________________\r
287Bool_t AliTRDinfoGen::Load(const Char_t *file, const Char_t *dir, const Char_t *name)\r
288{\r
289// Load data from performance file\r
290\r
291 if(!TFile::Open(file)){\r
292 AliWarning(Form("Couldn't open file %s.", file));\r
293 return kFALSE;\r
294 }\r
295 if(dir){\r
296 if(!gFile->cd(dir)){\r
297 AliWarning(Form("Couldn't cd to %s in %s.", dir, file));\r
298 return kFALSE;\r
299 }\r
300 }\r
301 TObjArray *o(NULL);\r
302 const Char_t *tn=(name ? name : GetName());\r
303 if(!(o = (TObjArray*)gDirectory->Get(tn))){\r
304 AliWarning(Form("Missing histogram container %s.", tn));\r
305 return kFALSE;\r
306 }\r
307 fContainer = (TObjArray*)o->Clone(GetName());\r
308 gFile->Close();\r
309 return kTRUE;\r
310}\r
311\r
312//____________________________________________________________________\r
313void AliTRDinfoGen::UserExec(Option_t *){\r
314 //\r
315 // Run the Analysis\r
316 //\r
317\r
318 fTracksBarrel->Delete();\r
319 fTracksSA->Delete();\r
320 fTracksKink->Delete();\r
321 fV0List->Delete();\r
f073d500 322 fClusters->Delete();\r
94b94be0 323 fEventInfo->Delete("");\r
324\r
325 fESDev = dynamic_cast<AliESDEvent*>(InputEvent());\r
326 if(!fESDev){\r
327 AliError("Failed retrieving ESD event");\r
328 return;\r
329 }\r
330 // WARNING\r
331 // This part may conflict with other detectors !!\r
4e38ea4c 332 if(!IsInitOCDB()){\r
94b94be0 333 AliCDBEntry* obj(NULL);\r
4e38ea4c 334 AliCDBManager* ocdb = AliCDBManager::Instance();\r
335 if(ocdb->IsDefaultStorageSet()){\r
c9b23652 336 AliInfo("OCDB :: initialized externally.");\r
94b94be0 337 } else {\r
4e38ea4c 338 AliInfo("OCDB :: initializing locally ...");\r
339 // prepare OCDB access\r
340 ocdb->SetDefaultStorage(fOCDB.Data());\r
b9058d72 341 //ocdb->SetSpecificStorage("TRD/Align/Data", "local:///home/niham/abercuci/local");\r
4e38ea4c 342 ocdb->SetRun(fESDev->GetRunNumber());\r
343 // create geo manager\r
344 if(!(obj = ocdb->Get(AliCDBPath("GRP", "Geometry", "Data")))){\r
345 AliError("GEOMETRY failed initialization.");\r
346 } else {\r
347 AliGeomManager::SetGeometry((TGeoManager*)obj->GetObject());\r
348 AliGeomManager::GetNalignable("TRD");\r
b9058d72 349 AliGeomManager::GetNalignable("TPC");\r
350 AliGeomManager::GetNalignable("ITS");\r
351 AliGeomManager::ApplyAlignObjsFromCDB("ITS TPC TRD");\r
4e38ea4c 352 }\r
353 //init magnetic field\r
354 if(!TGeoGlobalMagField::Instance()->IsLocked() &&\r
355 !fESDev->InitMagneticField()){\r
356 AliError("MAGNETIC FIELD failed initialization.");\r
357 }\r
358 // set no of time bins\r
359 AliTRDtrackerV1::SetNTimeBins(AliTRDcalibDB::Instance()->GetNumberOfTimeBinsDCS());\r
94b94be0 360 }\r
4e38ea4c 361 AliInfo(Form("OCDB : Loc[%s] Run[%d] TB[%d]", ocdb->GetDefaultStorage()->GetURI().Data(), ocdb->GetRun(), AliTRDtrackerV1::GetNTimeBins()));\r
362\r
363 // load misalignment\r
94b94be0 364 fgGeo = new AliTRDgeometry;\r
365 fgGeo->CreateClusterMatrixArray();\r
c9b23652 366 MakeChambers();\r
246fe7f7 367 // load reco param list from OCDB\r
99cd1793 368 AliInfo("Initializing TRD reco params ...");\r
94b94be0 369 fgReconstructor = new AliTRDReconstructor();\r
370 if(!(obj = ocdb->Get(AliCDBPath("TRD", "Calib", "RecoParam")))){\r
371 AliError("RECO PARAM failed initialization.");\r
94b94be0 372 } else {\r
373 obj->PrintMetaData();\r
246fe7f7 374 fRecos = (TObjArray*)obj->GetObject();\r
94b94be0 375 }\r
376 SetInitOCDB();\r
377 }\r
3ceb45ae 378\r
379 AliDebug(2, "+++++++++++++++++++++++++++++++++++++++");\r
380 AliDebug(2, Form("Analyse Ev[%d]", fESDev->GetEventNumberInFile()));\r
381\r
246fe7f7 382 // set reco param valid for this event\r
3ceb45ae 383 TH1 *h = (TH1I*)fContainer->At(kEvType);\r
246fe7f7 384 if(!fRecos){\r
385 fgReconstructor->SetRecoParam(AliTRDrecoParam::GetLowFluxParam());\r
386 h->Fill(0);\r
387 } else {\r
388 for(Int_t ireco(0); ireco<fRecos->GetEntriesFast(); ireco++){\r
389 AliTRDrecoParam *reco((AliTRDrecoParam*)fRecos->At(ireco));\r
390 Int_t es(reco->GetEventSpecie());\r
391 if(!(es&fESDev->GetEventSpecie())) continue;\r
392 fgReconstructor->SetRecoParam(reco);\r
2bfe5463 393 if(AliLog::GetDebugLevel("PWGPP/TRD", "AliTRDinfoGen")>5) reco->Dump();\r
246fe7f7 394 TString s;\r
395 if(es&AliRecoParam::kLowMult){ s="LowMult"; h->Fill(0);}\r
396 else if(es&AliRecoParam::kHighMult){ s="HighMult"; h->Fill(1);}\r
397 else if(es&AliRecoParam::kCosmic){ s="Cosmic"; h->Fill(2);}\r
398 else if(es&AliRecoParam::kCalib){ s="Calib"; h->Fill(3);}\r
399 else s="Unknown";\r
3ceb45ae 400 AliDebug(2, Form(" Using reco param \"%s\" for event %d.", s.Data(), fESDev->GetEventNumberInFile()));\r
246fe7f7 401 break;\r
402 }\r
403 }\r
94b94be0 404\r
405 // link MC if available\r
406 fMCev = MCEvent();\r
6465da91 407 \r
408 // trigger monitor\r
409 h = (TH1I*)fContainer->At(kTrigger);\r
410 TAxis *ax(h->GetXaxis());\r
411 TObjArray *evTriggers = fESDev->GetFiredTriggerClasses().Tokenize(" ");\r
412 for(Int_t iet(evTriggers->GetEntriesFast()); iet--;){\r
413 Int_t ix(1);\r
414 for(; ix<=ax->GetNbins(); ix++){\r
415 if(!Int_t(h->GetBinContent(ix))){\r
416 ax->SetBinLabel(ix, (*evTriggers)[iet]->GetName());\r
417 break;\r
418 }\r
35983729 419 if(strcmp((*evTriggers)[iet]->GetName(), ax->GetBinLabel(ix))==0) break;\r
6465da91 420 }\r
421 h->AddBinContent(ix);\r
422 }\r
94b94be0 423\r
35983729 424 // event selection based on vertex cuts and trigger\r
33056e04 425 if(UseLocalEvSelection() && !fEventCut->IsSelected(fESDev, IsCollision())){\r
426 AliDebug(2, "Event failed selection on vertex and trigger");\r
427 return;\r
428 }\r
94b94be0 429\r
430 if(!fESDfriend){\r
431 AliError("Failed retrieving ESD friend event");\r
432 return;\r
433 }\r
434 if(HasMCdata() && !fMCev){\r
435 AliError("Failed retrieving MC event");\r
436 return;\r
437 }\r
438\r
439 new(fEventInfo)AliTRDeventInfo(fESDev->GetHeader(), const_cast<AliESDRun *>(fESDev->GetESDRun()));\r
440 // Determine centrality\r
441 // Author: Ionut Arsene <I.C.Arsene@gsi.de>\r
f0857a6a 442 TString beamtype = fESDev->GetESDRun()->GetBeamType();\r
443 if(beamtype.Contains("Pb-Pb") || beamtype.Contains("A-A")){\r
94b94be0 444 const AliMultiplicity *mult = fESDev->GetMultiplicity();\r
6687261d 445 fEventInfo->SetMultiplicity(mult?mult->GetNumberOfTracklets():0);\r
446 const AliCentrality *cent = fESDev->GetCentrality();\r
447 // centrality for different options V0 = "V0M", ITS = "TKL" etc\r
448 fEventInfo->SetCentrality(cent?cent->GetCentralityPercentile("TKL"):-1.);\r
5591f3d6 449 AliDebug(2, Form(" Beam Type[%s] Mult[%d/%4d] Cent[%d/%5.2f]",\r
450 beamtype.Data(),\r
451 fEventInfo->GetMultiplicity(), mult?mult->GetNumberOfTracklets():0,\r
452 fEventInfo->GetCentrality(), cent?cent->GetCentralityPercentile("TKL"):-1.));\r
6687261d 453 } else {\r
454 fEventInfo->SetMultiplicity(0);\r
455 fEventInfo->SetCentrality(-1.);\r
5591f3d6 456 AliDebug(2, Form(" Beam Type[%s]", beamtype.Data()));\r
94b94be0 457 }\r
566c3d46 458 UShort_t evBC(fESDev->GetBunchCrossNumber());\r
94b94be0 459\r
460 Bool_t *trackMap(NULL);\r
461 AliStack * mStack(NULL);\r
462 Int_t nTracksMC = HasMCdata() ? fMCev->GetNumberOfTracks() : 0, nTracksESD = fESDev->GetNumberOfTracks();\r
463 if(HasMCdata()){\r
464 mStack = fMCev->Stack();\r
465 if(!mStack){\r
466 AliError("Failed retrieving MC Stack");\r
467 return;\r
468 }\r
469 trackMap = new Bool_t[nTracksMC];\r
470 memset(trackMap, 0, sizeof(Bool_t) * nTracksMC);\r
471 }\r
472 \r
473 Double32_t dedx[100]; Int_t nSlices(0);\r
33056e04 474 Int_t nTRDout(0), nTRDin(0), nTPC(0), nITS(0)\r
94b94be0 475 ,nclsTrklt\r
476 ,nBarrel(0), nSA(0), nKink(0)\r
477 ,nBarrelFriend(0), nSAFriend(0)\r
478 ,nBarrelMC(0), nSAMC(0), nKinkMC(0);\r
479 AliESDtrack *esdTrack = NULL;\r
480 AliESDfriendTrack *esdFriendTrack = NULL;\r
481 TObject *calObject = NULL;\r
482 AliTRDtrackV1 *track = NULL;\r
483 AliTRDseedV1 *tracklet = NULL;\r
484 AliTRDcluster *cl = NULL;\r
485\r
486\r
487 // LOOP 0 - over ESD v0s\r
488 Float_t bField(fESDev->GetMagneticField());\r
489 AliESDv0 *v0(NULL);\r
490 Int_t v0pid[AliPID::kSPECIES];\r
491 for(Int_t iv0(0); iv0<fESDev->GetNumberOfV0s(); iv0++){\r
492 if(!(v0 = fESDev->GetV0(iv0))) continue;\r
493 // register v0\r
494 if(fV0Cut) new(fV0Info) AliTRDv0Info(*fV0Cut);\r
495 else new(fV0Info) AliTRDv0Info();\r
496 fV0Info->SetMagField(bField);\r
497 fV0Info->SetV0tracks(fESDev->GetTrack(v0->GetPindex()), fESDev->GetTrack(v0->GetNindex()));\r
498 fV0Info->SetV0Info(v0);\r
499 fV0List->Add(new AliTRDv0Info(*fV0Info));// kFOUND=kFALSE;\r
500 }\r
501\r
502\r
503 // LOOP 1 - over ESD tracks\r
504 AliTRDv0Info *v0info=NULL;\r
505 for(Int_t itrk = 0; itrk < nTracksESD; itrk++){\r
506 new(fTrackInfo) AliTRDtrackInfo();\r
507 esdTrack = fESDev->GetTrack(itrk);\r
3ceb45ae 508 AliDebug(3, Form("\n%3d ITS[%d] TPC[%d] TRD[%d] TOF-BC[%d]\n", itrk, esdTrack->GetNcls(0), esdTrack->GetNcls(1), esdTrack->GetNcls(2), esdTrack->GetTOFBunchCrossing()));\r
33056e04 509 if(esdTrack->GetStatus()&AliESDtrack::kITSout) nITS++;\r
94b94be0 510 if(esdTrack->GetStatus()&AliESDtrack::kTPCout) nTPC++;\r
511 if(esdTrack->GetStatus()&AliESDtrack::kTRDout) nTRDout++;\r
512 if(esdTrack->GetStatus()&AliESDtrack::kTRDin) nTRDin++;\r
33056e04 513// printf(" %3d ITS[%c] TPC[%c] TRDin[%c] TRDout[%c] TRDStop[%c]\n", itrk,\r
514// (esdTrack->GetStatus()&AliESDtrack::kITSout)?'y':'n',\r
515// (esdTrack->GetStatus()&AliESDtrack::kTPCout)?'y':'n',\r
516// (esdTrack->GetStatus()&AliESDtrack::kTRDin)?'y':'n',\r
517// (esdTrack->GetStatus()&AliESDtrack::kTRDout)?'y':'n',\r
518// (esdTrack->GetStatus()&AliESDtrack::kTRDStop)?'y':'n');\r
94b94be0 519 // look at external track param\r
520 const AliExternalTrackParam *op = esdTrack->GetOuterParam();\r
521 Double_t xyz[3];\r
522 if(op){\r
523 op->GetXYZ(xyz);\r
524 op->Global2LocalPosition(xyz, op->GetAlpha());\r
525 AliDebug(3, Form("op @ X[%7.3f]\n", xyz[0]));\r
526 }\r
527\r
528 // read MC info\r
529 Int_t fPdg = -1;\r
530 Int_t label = -1; UInt_t alab=UINT_MAX;\r
531 if(HasMCdata()){\r
532 label = esdTrack->GetLabel(); \r
533 alab = TMath::Abs(label);\r
534 // register the track\r
535 if(alab < UInt_t(nTracksMC)){ \r
536 trackMap[alab] = kTRUE; \r
537 } else { \r
538 AliError(Form("MC label[%d] outside scope for Ev[%d] Trk[%d].", label, (Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry(), itrk));\r
539 continue; \r
540 }\r
541 AliMCParticle *mcParticle = NULL; \r
542 if(!(mcParticle = (AliMCParticle*) fMCev->GetTrack(alab))){\r
543 AliError(Form("MC particle label[%d] missing for Ev[%d] Trk[%d].", label, (Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry(), itrk));\r
544 continue;\r
545 }\r
546 fPdg = mcParticle->Particle()->GetPdgCode();\r
547 Int_t nRefs = mcParticle->GetNumberOfTrackReferences();\r
548 Int_t iref = 0; AliTrackReference *ref = NULL; \r
549 while(iref<nRefs){\r
550 ref = mcParticle->GetTrackReference(iref);\r
551 if(ref->LocalX() > fgkTPC) break;\r
552 iref++;\r
553 }\r
554\r
555 fTrackInfo->SetMC();\r
556 fTrackInfo->SetPDG(fPdg);\r
557 fTrackInfo->SetPrimary(mcParticle->Particle()->IsPrimary());\r
558 fTrackInfo->SetLabel(label);\r
eb05d549 559 fTrackInfo->SetTRDlabel(esdTrack->GetTRDLabel());\r
94b94be0 560 Int_t jref = iref;//, kref = 0;\r
561 while(jref<nRefs){\r
562 ref = mcParticle->GetTrackReference(jref);\r
563 if(ref->LocalX() > fgkTRD) break;\r
564 AliDebug(4, Form(" trackRef[%2d (%2d)] @ %7.3f OK", jref-iref, jref, ref->LocalX()));\r
565 fTrackInfo->AddTrackRef(ref);\r
566 jref++;\r
567 }\r
568 AliDebug(3, Form("NtrackRefs[%d(%d)]", fTrackInfo->GetNTrackRefs(), nRefs));\r
569 }\r
570\r
571 // copy some relevant info to TRD track info\r
572 fTrackInfo->SetStatus(esdTrack->GetStatus());\r
573 fTrackInfo->SetTrackId(esdTrack->GetID());\r
574 Double_t p[AliPID::kSPECIES]; esdTrack->GetTRDpid(p);\r
575 fTrackInfo->SetESDpid(p);\r
576 fTrackInfo->SetESDpidQuality(esdTrack->GetTRDntrackletsPID());\r
eb05d549 577 fTrackInfo->SetESDeta(esdTrack->Eta());\r
578 Double_t loc[3];\r
579 if(esdTrack->GetXYZAt(298., fESDev->GetMagneticField(), loc)) fTrackInfo->SetESDphi(TMath::ATan2(loc[1], loc[0]));\r
580 fTrackInfo->SetESDpt(esdTrack->Pt());\r
94b94be0 581 if(!nSlices) nSlices = esdTrack->GetNumberOfTRDslices();\r
582 memset(dedx, 0, 100*sizeof(Double32_t));\r
583 Int_t in(0);\r
584 for(Int_t il=0; il<AliTRDgeometry::kNlayer; il++)\r
585 for(Int_t is=0; is<nSlices; is++) \r
586 dedx[in++]=esdTrack->GetTRDslice(il, is);\r
587 for(Int_t il=0; il<AliTRDgeometry::kNlayer; il++) dedx[in++]=esdTrack->GetTRDmomentum(il);\r
588 fTrackInfo->SetSlices(in, dedx);\r
589 fTrackInfo->SetNumberOfClustersRefit(esdTrack->GetNcls(2));\r
590 // some other Informations which we may wish to store in order to find problematic cases\r
591 fTrackInfo->SetKinkIndex(esdTrack->GetKinkIndex(0));\r
592 fTrackInfo->SetTPCncls(static_cast<UShort_t>(esdTrack->GetNcls(1)));\r
2b33fd9c 593 fTrackInfo->SetTPCdedx(esdTrack->GetTPCsignal());\r
566c3d46 594 fTrackInfo->SetTOFbc(esdTrack->GetTOFBunchCrossing()==AliVTrack::kTOFBCNA?0:esdTrack->GetTOFBunchCrossing());\r
94b94be0 595 nclsTrklt = 0;\r
596 \r
597 // set V0pid info\r
598 for(Int_t iv(0); iv<fV0List->GetEntriesFast(); iv++){\r
599 if(!(v0info = (AliTRDv0Info*)fV0List->At(iv))) continue;\r
600 if(!v0info->GetV0Daughter(1) && !v0info->GetV0Daughter(-1)) continue;\r
601 if(!v0info->HasTrack(fTrackInfo)) continue;\r
602 memset(v0pid, 0, AliPID::kSPECIES*sizeof(Int_t));\r
603 fTrackInfo->SetV0();\r
604 for(Int_t is=AliPID::kSPECIES; is--;){v0pid[is] = v0info->GetPID(is, fTrackInfo);}\r
605 fTrackInfo->SetV0pid(v0pid);\r
606 fTrackInfo->SetV0();\r
607 //const AliTRDtrackInfo::AliESDinfo *ei = fTrackInfo->GetESDinfo();\r
608 break;\r
609 }\r
610\r
f073d500 611 // read track REC info\r
33056e04 612 if((esdFriendTrack = (fESDfriend->GetNumberOfTracks() > itrk) ? fESDfriend->GetTrack(itrk): NULL)) {\r
35983729 613 fTrackInfo->SetTPCoutParam(esdFriendTrack->GetTPCOut());\r
b9058d72 614 fTrackInfo->SetITSoutParam(esdFriendTrack->GetITSOut());\r
eb05d549 615 const AliTrackPointArray *tps(NULL);\r
616 if((tps=esdFriendTrack->GetTrackPointArray()) && HasTrackPoints()) fTrackInfo->SetTrackPointArray(tps);\r
94b94be0 617 Int_t icalib = 0;\r
618 while((calObject = esdFriendTrack->GetCalibObject(icalib++))){\r
619 if(strcmp(calObject->IsA()->GetName(),"AliTRDtrackV1") != 0) continue; // Look for the TRDtrack\r
620 if(!(track = dynamic_cast<AliTRDtrackV1*>(calObject))) break;\r
621 AliDebug(4, Form("TRD track OK"));\r
622 // Set the clusters to unused\r
623 for(Int_t ipl = 0; ipl < AliTRDgeometry::kNlayer; ipl++){\r
624 if(!(tracklet = track->GetTracklet(ipl))) continue;\r
625 tracklet->ResetClusterIter();\r
626 while((cl = tracklet->NextCluster())) cl->Use(0);\r
627 }\r
628 fTrackInfo->SetTrack(track);\r
35983729 629 ((TH2I*)fContainer->At(kBC))->Fill(evBC, esdTrack->GetTOFBunchCrossing());\r
94b94be0 630 break;\r
631 }\r
33056e04 632 AliDebug(3, Form("Ntracklets[%d]", fTrackInfo->GetNTracklets()));\r
633 } else AliDebug(3, Form("No Friends for trk[%3d] Ntrk[%3d]", itrk, fESDfriend->GetNumberOfTracks()));\r
94b94be0 634 if(op) fTrackInfo->SetOuterParam(op);\r
635\r
636 if(DebugLevel() >= 1){\r
637 AliTRDtrackInfo info(*fTrackInfo);\r
638 (*DebugStream()) << "trackInfo"\r
639 << "TrackInfo.=" << &info\r
640 << "\n";\r
641 info.Delete("");\r
642 }\r
643\r
644 ULong_t status(esdTrack->GetStatus());\r
33056e04 645 if((status&AliESDtrack::kTPCout)){ // TPC prolongation\r
94b94be0 646 if(!esdTrack->GetKinkIndex(0)){ // Barrel Track Selection\r
647 Bool_t selected(kTRUE);\r
648 if(UseLocalTrkSelection()){\r
649 if(esdTrack->Pt() < fgkPt){ \r
3ceb45ae 650 AliDebug(3, Form("Reject Trk[%3d] Ev[%4d] Pt[%5.2f]", itrk, fESDev->GetEventNumberInFile(), esdTrack->Pt()));\r
94b94be0 651 selected = kFALSE;\r
652 }\r
653 if(selected && TMath::Abs(esdTrack->Eta()) > fgkEta){\r
3ceb45ae 654 AliDebug(3, Form("Reject Trk[%3d] Ev[%4d] Eta[%5.2f]", itrk, fESDev->GetEventNumberInFile(), TMath::Abs(esdTrack->Eta())));\r
94b94be0 655 selected = kFALSE;\r
656 }\r
657 if(selected && esdTrack->GetTPCNcls() < fgkNclTPC){ \r
3ceb45ae 658 AliDebug(3, Form("Reject Trk[%3d] Ev[%4d] NclTPC[%d]", itrk, fESDev->GetEventNumberInFile(), esdTrack->GetTPCNcls()));\r
94b94be0 659 selected = kFALSE;\r
660 }\r
661 Float_t par[2], cov[3];\r
662 esdTrack->GetImpactParameters(par, cov);\r
663 if(IsCollision()){ // cuts on DCA\r
664 if(selected && TMath::Abs(par[0]) > fgkTrkDCAxy){ \r
3ceb45ae 665 AliDebug(3, Form("Reject Trk[%3d] Ev[%4d] DCAxy[%f]", itrk, fESDev->GetEventNumberInFile(), TMath::Abs(par[0])));\r
94b94be0 666 selected = kFALSE;\r
667 }\r
668 if(selected && TMath::Abs(par[1]) > fgkTrkDCAz){ \r
3ceb45ae 669 AliDebug(3, Form("Reject Trk[%3d] Ev[%4d] DCAz[%f]", itrk, fESDev->GetEventNumberInFile(), TMath::Abs(par[1])));\r
94b94be0 670 selected = kFALSE;\r
671 }\r
672 } else if(selected && fMCev && !fMCev->IsPhysicalPrimary(alab)){;\r
3ceb45ae 673 AliDebug(3, Form("Reject Trk[%3d] Ev[%4d] Primary", itrk, fESDev->GetEventNumberInFile()));\r
94b94be0 674 selected = kFALSE;\r
675 }\r
676 }\r
677 if(fTrackCut && !fTrackCut->IsSelected(esdTrack)) selected = kFALSE;\r
678 if(selected){ \r
679 fTracksBarrel->Add(new AliTRDtrackInfo(*fTrackInfo));\r
680 nBarrel++;\r
681 if(fTrackInfo->GetTrack()) \r
682 nBarrelFriend++;\r
683 }\r
684 } else {\r
685 fTracksKink->Add(new AliTRDtrackInfo(*fTrackInfo));\r
686 nKink++;\r
687 }\r
33056e04 688 } else if((status&AliESDtrack::kITSout)) { // ITS prolongation \r
689 Bool_t selected(kTRUE);\r
690 if(UseLocalTrkSelection()){\r
691 if(esdTrack->Pt() < fgkPt){\r
692 AliDebug(3, Form("Reject Trk[%3d] Ev[%4d] Pt[%5.2f]", itrk, fESDev->GetEventNumberInFile(), esdTrack->Pt()));\r
693 selected = kFALSE;\r
694 }\r
695 if(selected && TMath::Abs(esdTrack->Eta()) > fgkEta){\r
696 AliDebug(3, Form("Reject Trk[%3d] Ev[%4d] Eta[%5.2f]", itrk, fESDev->GetEventNumberInFile(), TMath::Abs(esdTrack->Eta())));\r
697 selected = kFALSE;\r
698 }\r
699 Float_t par[2], cov[3];\r
700 esdTrack->GetImpactParameters(par, cov);\r
701 if(IsCollision()){ // cuts on DCA\r
702 if(selected && TMath::Abs(par[0]) > fgkTrkDCAxy){\r
703 AliDebug(3, Form("Reject Trk[%3d] Ev[%4d] DCAxy[%f]", itrk, fESDev->GetEventNumberInFile(), TMath::Abs(par[0])));\r
704 selected = kFALSE;\r
705 }\r
706 if(selected && TMath::Abs(par[1]) > fgkTrkDCAz){\r
707 AliDebug(3, Form("Reject Trk[%3d] Ev[%4d] DCAz[%f]", itrk, fESDev->GetEventNumberInFile(), TMath::Abs(par[1])));\r
708 selected = kFALSE;\r
709 }\r
710 } else if(selected && fMCev && !fMCev->IsPhysicalPrimary(alab)){;\r
711 AliDebug(3, Form("Reject Trk[%3d] Ev[%4d] Primary", itrk, fESDev->GetEventNumberInFile()));\r
712 selected = kFALSE;\r
713 }\r
714 }\r
715 if(fTrackCut && !fTrackCut->IsSelected(esdTrack)) selected = kFALSE;\r
716 if(selected){\r
717 fTracksBarrel->Add(new AliTRDtrackInfo(*fTrackInfo));\r
718 nBarrel++;\r
719 if(fTrackInfo->GetTrack())\r
720 nBarrelFriend++;\r
721 }\r
722 } else if((status&AliESDtrack::kTRDout) && !(status&AliESDtrack::kTRDin)){ // TRD SA tracking\r
94b94be0 723 fTracksSA->Add(new AliTRDtrackInfo(*fTrackInfo));\r
724 nSA++;\r
725 if(fTrackInfo->GetTrack()) \r
726 nSAFriend++;\r
727 }\r
728 fTrackInfo->Delete("");\r
729 }\r
f073d500 730 // read clusters REC info\r
56838f9d 731 TTree * treeR(NULL); AliESDInputHandlerRP *esdRPhandler(NULL);\r
732 if(fInputHandler) esdRPhandler = dynamic_cast<AliESDInputHandlerRP*>(fInputHandler);\r
733 if(esdRPhandler){\r
734 if((treeR = esdRPhandler->GetTreeR("TRD"))) {\r
178d284a 735 TObjArray *recPoints(NULL);\r
736 if((treeR->GetBranch("TRDcluster"))){\r
737 treeR->SetBranchAddress("TRDcluster", &recPoints);\r
738 for(Int_t idet(0); idet<treeR->GetEntries(); idet++){\r
739 treeR->GetEntry(idet);\r
740 if(!recPoints->GetEntries()){\r
741 AliDebug(1, Form("Missing entry %d from TreeR", idet));\r
742 continue;\r
743 }\r
744 AliTRDcluster *c = (AliTRDcluster*)(*recPoints)[0];\r
745 if(!c){\r
746 AliDebug(1, Form("Missing first cluster in entry %d from TreeR", idet));\r
747 continue;\r
748 }\r
749 fClusters->AddAt(recPoints->Clone(Form("%03d", c->GetDetector())), c->GetDetector());\r
f073d500 750 }\r
178d284a 751 } else AliDebug(3, "No TRDcluster branch");\r
752 } else AliDebug(3, "No RecPoints");\r
753 } else AliDebug(3, "No AliESDInputHandlerRP");\r
f073d500 754\r
94b94be0 755\r
756 // LOOP 2 - over MC tracks which are passing TRD where the track is not reconstructed\r
757 if(HasMCdata()){\r
758 AliDebug(10, "Output of the MC track map:");\r
759 for(Int_t itk = 0; itk < nTracksMC; itk++) AliDebug(10, Form("trackMap[%d] = %s", itk, trackMap[itk] == kTRUE ? "TRUE" : "kFALSE"));\r
760 \r
761 for(Int_t itk = 0; itk < nTracksMC; itk++){\r
762 if(trackMap[itk]) continue;\r
763 AliMCParticle *mcParticle = (AliMCParticle*) fMCev->GetTrack(TMath::Abs(itk));\r
764 Int_t fPdg = mcParticle->Particle()->GetPdgCode();\r
765 Int_t nRefs = mcParticle->GetNumberOfTrackReferences();\r
766 Int_t iref = 0; AliTrackReference *ref = NULL; \r
767 Int_t nRefsTRD = 0;\r
768 new(fTrackInfo) AliTRDtrackInfo();\r
769 fTrackInfo->SetMC();\r
770 fTrackInfo->SetPDG(fPdg);\r
771 while(iref<nRefs){ // count TRD TR\r
772 Bool_t kIN(kFALSE);\r
773 ref = mcParticle->GetTrackReference(iref);\r
774 if(ref->LocalX() > fgkTPC && ref->LocalX() < fgkTRD){\r
775 fTrackInfo->AddTrackRef(ref);\r
776 nRefsTRD++;kIN=kTRUE;\r
777 }\r
778 AliDebug(4, Form(" trackRef[%2d] @ x[%7.3f] %s", iref, ref->LocalX(), kIN?"IN":"OUT"));\r
779 iref++;\r
780 }\r
781 if(!nRefsTRD){\r
782 // In this stage we at least require 1 hit inside TRD. What will be done with this tracks is a task for the \r
783 // analysis job\r
784 fTrackInfo->Delete("");\r
785 continue;\r
786 }\r
787 fTrackInfo->SetPrimary(mcParticle->Particle()->IsPrimary());\r
788 fTrackInfo->SetLabel(itk);\r
789 if(DebugLevel() >= 1){\r
790 AliTRDtrackInfo info(*fTrackInfo);\r
791 (*DebugStream()) << "trackInfo"\r
792 << "TrackInfo.=" << &info\r
793 << "\n";\r
794 info.Delete("");\r
795 }\r
796 AliDebug(3, Form("Add MC track @ label[%d] nTRDrefs[%d].", itk, nRefsTRD));\r
797 // check where the track starts\r
798 ref = mcParticle->GetTrackReference(0);\r
799 if(ref->LocalX() < fgkITS){ \r
800 fTracksBarrel->Add(new AliTRDtrackInfo(*fTrackInfo));\r
801 nBarrelMC++;\r
802 } else if(ref->LocalX() < fgkTPC) {\r
803 fTracksKink->Add(new AliTRDtrackInfo(*fTrackInfo));\r
804 nKinkMC++;\r
805 } else if(nRefsTRD>6){\r
806 fTracksSA->Add(new AliTRDtrackInfo(*fTrackInfo));\r
807 nSAMC++;\r
808 }\r
809 fTrackInfo->Delete("");\r
810 }\r
811 delete[] trackMap;\r
812 }\r
813 AliDebug(1, Form(\r
814 "\nEv[%3d] Tracks: ESD[%d] MC[%d] V0[%d]\n"\r
33056e04 815 " TPCout[%d] ITSout[%d] TRDin[%d] TRDout[%d]\n"\r
94b94be0 816 " Barrel[%3d+%3d=%3d] SA[%2d+%2d=%2d] Kink[%2d+%2d=%2d]"\r
817 ,(Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry(), nTracksESD, nTracksMC, fV0List->GetEntries()\r
33056e04 818 , nTPC, nITS, nTRDin, nTRDout\r
94b94be0 819 ,nBarrel, nBarrelMC, fTracksBarrel->GetEntries()\r
820 ,nSA, nSAMC, fTracksSA->GetEntries()\r
821 ,nKink, nKinkMC, fTracksKink->GetEntries()\r
822 ));\r
823 // save track statistics\r
3ceb45ae 824 h = (TH1I*)fContainer->At(kStatTrk);\r
94b94be0 825 h->Fill(Float_t(kTracksESD), nTracksESD);\r
826 h->Fill(Float_t(kTracksMC), nTracksMC);\r
827 h->Fill(Float_t(kV0), fV0List->GetEntries());\r
828 h->Fill(Float_t(kTPC), nTPC);\r
33056e04 829 h->Fill(Float_t(kITS), nITS);\r
94b94be0 830 h->Fill(Float_t(kTRDin), nTRDin);\r
831 h->Fill(Float_t(kTRDout), nTRDout);\r
832 h->Fill(Float_t(kBarrel), nBarrel);\r
833 h->Fill(Float_t(kBarrelMC), nBarrelMC);\r
834 h->Fill(Float_t(kSA), nSA);\r
835 h->Fill(Float_t(kSAMC), nSAMC);\r
836 h->Fill(Float_t(kKink), nKink);\r
837 h->Fill(Float_t(kKinkMC), nKinkMC);\r
838 h->Fill(Float_t(kBarrelFriend), nBarrelFriend);\r
839 h->Fill(Float_t(kSAFriend), nSAFriend);\r
94b94be0 840}\r
841\r
c9b23652 842\r
843//____________________________________________________________________\r
844void AliTRDinfoGen::MakeChambers()\r
845{\r
846// Build chamber position and status\r
847 if(!fContainer){\r
848 AliError("Missing container");\r
849 return;\r
850 }\r
851 AliTRDcalibDB *calib(AliTRDcalibDB::Instance());\r
852 if(!calib){\r
853 AliError("No access to calibration data");\r
854 return;\r
855 }\r
856\r
c9b23652 857 Double_t alpha(0.), cs(-2.), sn(0.);\r
77cb6c6a 858 TVectorF pos(5*AliTRDgeometry::kNdet);\r
c9b23652 859 for(Int_t isec(0); isec<AliTRDgeometry::kNsector; isec++){\r
860 alpha = (0.5+isec)*AliTRDgeometry::GetAlpha();\r
861 cs = TMath::Cos(alpha);\r
862 sn = TMath::Sin(alpha);\r
863\r
864 for(Int_t istk(0); istk<AliTRDgeometry::kNstack; istk++){\r
865 for(Int_t ilyr(0); ilyr<AliTRDgeometry::kNlayer; ilyr++){\r
866 Int_t idet(AliTRDgeometry::GetDetector(ilyr, istk, isec));\r
77cb6c6a 867 Int_t jdet = 5*idet;\r
c9b23652 868 TGeoHMatrix *matrix(fgGeo->GetClusterMatrix(idet));\r
869 if(!matrix){\r
870 AliDebug(2, Form("Missing matrix for %03d [%02d_%d_%d]", idet, isec, istk, ilyr));\r
871 continue;\r
872 }\r
873 AliDebug(2, Form("Read info for %03d [%02d_%d_%d]", idet, isec, istk, ilyr));\r
874 AliTRDpadPlane *pp(fgGeo->GetPadPlane(ilyr, istk));\r
875 Double_t zm(0.5 * (pp->GetRow0() + pp->GetRowEnd())),\r
876 loc0[] = {AliTRDgeometry::AnodePos(), pp->GetCol0(), zm-pp->GetRow0()},\r
877 loc1[] = {AliTRDgeometry::AnodePos(), pp->GetColEnd(), zm-pp->GetRowEnd()},\r
878 glb[3];\r
879 matrix->LocalToMaster(loc0, glb);\r
880 Float_t phi = TMath::ATan2(glb[0]*sn + glb[1]*cs, glb[0]*cs - glb[1]*sn),\r
881 tgl = glb[2]/glb[0]/TMath::Sqrt(1.+glb[1]*glb[1]/glb[0]/glb[0]),\r
882 eta = -TMath::Log(TMath::Tan(0.5 * (0.5*TMath::Pi() - TMath::ATan(tgl))));\r
77cb6c6a 883 pos[jdet+0] = eta; pos[jdet+1] = phi;\r
c9b23652 884 matrix->LocalToMaster(loc1, glb);\r
885 phi = TMath::ATan2(glb[0]*sn + glb[1]*cs, glb[0]*cs - glb[1]*sn);\r
886 tgl = glb[2]/glb[0]/TMath::Sqrt(1.+glb[1]*glb[1]/glb[0]/glb[0]);\r
887 eta = -TMath::Log(TMath::Tan(0.5 * (0.5*TMath::Pi() - TMath::ATan(tgl))));\r
77cb6c6a 888 pos[jdet+2] = eta; pos[jdet+3] = phi;\r
889 pos[jdet+4] = 0.;\r
07b3a6a6 890 if(calib->IsChamberGood(idet)){\r
77cb6c6a 891 if(calib->IsHalfChamberNoData(idet, 0)) pos[jdet+4] += 2.;\r
892 if(calib->IsHalfChamberNoData(idet, 1)) pos[jdet+4] += 3.;\r
893 } else pos[jdet+4] = 1.;\r
c9b23652 894 }\r
895 }\r
896 }\r
77cb6c6a 897 fContainer->AddAt(new TVectorF(pos), kChmb);\r
c9b23652 898}\r
899\r
900//____________________________________________________________________\r
901void AliTRDinfoGen::MakeSummary()\r
902{\r
903// Build summary plots\r
904 if(!fContainer){\r
905 AliError("Missing results");\r
906 return;\r
907 }\r
908 TH1 *h1(NULL); TVirtualPad *p(NULL); TCanvas *cOut(NULL);\r
909\r
910 const Int_t nx(2048), ny(750);\r
911 cOut = new TCanvas(GetName(), "Run Statistics", nx, ny);\r
912 cOut->Divide(3,1, 1.e-5, 1.e-5);\r
913 //=========\r
914 p=cOut->cd(1);p->SetRightMargin(0.025);p->SetTopMargin(0.01);p->SetLogy();\r
915 h1 = (TH1*)fContainer->At(kStatTrk);\r
916 h1->SetBarOffset(0.06); h1->SetBarWidth(0.88); h1->SetFillColor(3);\r
917 h1->Draw("bar1");\r
918 //=========\r
919 p=cOut->cd(2);p->SetRightMargin(0.025);p->SetTopMargin(0.01);p->SetLogy();\r
920 h1 = (TH1*)fContainer->At(kEvType);\r
921 h1->SetBarOffset(0.04); h1->SetBarWidth(0.92);h1->SetFillColor(6);\r
922 h1->Draw("bar1");\r
923 //=========\r
924 p=cOut->cd(3);p->SetRightMargin(0.025);p->SetTopMargin(0.01);p->SetLogz();\r
925 TH2 *h2((TH2*)fContainer->At(kBC));\r
926 h1 = h2->ProjectionX("_px");\r
927 Int_t n(0); Int_t bins[3500];\r
928 for(Int_t ib(1); ib<=3500; ib++){\r
929 if(h1->GetBinContent(ib) < 1) continue;\r
930 bins[n++] = ib;\r
931 }\r
932 delete h1;\r
933\r
934 TAxis *ay(h2->GetYaxis());\r
935 TH2 *hs = new TH2I("hBC_Summary", Form("%s;%s;%s", h2->GetTitle(), h2->GetXaxis()->GetTitle(), h2->GetYaxis()->GetTitle()),\r
936 n, -0.5, n-0.5, ay->GetNbins(), ay->GetXmin(), ay->GetXmax());\r
937 hs->SetLineColor(kBlack);hs->SetLineWidth(1);\r
938 hs->SetMarkerColor(kRed);\r
939 TAxis *ax(hs->GetXaxis());\r
940 for(Int_t ib(0); ib<n; ib++){\r
941 ax->SetBinLabel(ib+1, Form("%d", bins[ib]));\r
942 for(Int_t iy(1); iy<=ay->GetNbins(); iy++){\r
943 hs->SetBinContent(ib+1, iy, h2->GetBinContent(bins[ib], iy));\r
944 }\r
945 }\r
946 hs->Draw("textbox");\r
947 cOut->SaveAs(Form("%s.gif", cOut->GetName()));\r
948}\r
949\r
94b94be0 950//____________________________________________________________________\r
35983729 951void AliTRDinfoGen::SetLocalEvSelection(const AliTRDeventCuts &ec)\r
94b94be0 952{\r
35983729 953// Set event cuts from outside\r
954 if(!fEventCut) fEventCut = new AliTRDeventCuts(ec);\r
955 else new(fEventCut) AliTRDeventCuts(ec);\r
956 fEventCut->Print();\r
94b94be0 957}\r
958\r
959//____________________________________________________________________\r
35983729 960void AliTRDinfoGen::SetLocalV0Selection(const AliTRDv0Info &v0)\r
94b94be0 961{\r
35983729 962// Set V0 cuts from outside\r
963\r
964 if(!fV0Cut) fV0Cut = new AliTRDv0Info(v0);\r
965 else new(fV0Cut) AliTRDv0Info(v0);\r
966 fV0Cut->Print();\r
94b94be0 967}\r
968\r
969//____________________________________________________________________\r
970TTreeSRedirector* AliTRDinfoGen::DebugStream()\r
971{\r
972// Manage debug stream for task\r
973 if(!fDebugStream){\r
974 TDirectory *savedir = gDirectory;\r
975 fDebugStream = new TTreeSRedirector("TRD.DebugInfoGen.root");\r
976 savedir->cd();\r
977 }\r
978 return fDebugStream;\r
979}\r
980\r
35983729 981//____________________________________________________________________\r
982void AliTRDinfoGen::Terminate(Option_t* /*option*/)\r
983{\r
984// Process run information\r
985 AliInfo("");\r
2bfe5463 986 if(!(fContainer = dynamic_cast<TObjArray *>(GetOutputData(AliTRDpwgppHelper::kMonitor)))) return;\r
35983729 987 AliInfo(Form("fContainer(%p)", (void*)fContainer));\r
988 if(UseLocalEvSelection()){\r
989 TH1 *h1 = (TH1*)fContainer->At(kTrigger); TAxis *ax(h1->GetXaxis());\r
990 AliInfo(Form("h1(%p)", (void*)h1));\r
991 for(Int_t ix(1); ix<=ax->GetNbins(); ix++){\r
992 if(fEventCut->CheckTrigger(ax->GetBinLabel(ix))) ax->SetBinLabel(ix, Form("#color[2]{%s}", ax->GetBinLabel(ix)));\r
993 }\r
994 }\r
995}\r
94b94be0 996\r