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