]>
Commit | Line | Data |
---|---|---|
77203477 | 1 | // Steer TRD QA train for Reconstruction (Clusterizer, Tracking and PID). |
2 | // | |
3 | // Usage: | |
4 | // run.C(tasks, files, entries) | |
5 | // tasks : "ALL" or one/more of the following: | |
6 | // "EFF" : TRD Tracking Efficiency | |
7 | // "EFFC" : TRD Tracking Efficiency Combined (barrel + stand alone) - only in case of simulations | |
8 | // "RES" : TRD tracking Resolution | |
ba2773d0 | 9 | // "CAL" : TRD calibration |
77203477 | 10 | // "PID" : TRD PID - pion efficiency |
11 | // "PIDR" : TRD PID - reference data | |
2b468513 | 12 | // "DET" : Basic TRD Detector checks |
28efdace | 13 | // "NOFR" : Data set does not have AliESDfriends.root |
4b8f8a35 | 14 | // "NOMC" : Data set does not have Monte Carlo Informations (real data), so all tasks which rely |
15 | // on MC information are switched off | |
2c0cf367 | 16 | // |
17 | // In compiled mode : | |
18 | // Don't forget to load first the libraries | |
19 | // gSystem->Load("libMemStat.so") | |
20 | // gSystem->Load("libMemStatGui.so") | |
21 | // gSystem->Load("libANALYSIS.so") | |
22 | // gSystem->Load("libTRDqaRec.so") | |
23 | // | |
77203477 | 24 | // Authors: |
25 | // Alex Bercuci (A.Bercuci@gsi.de) | |
26 | // Markus Fasel (m.Fasel@gsi.de) | |
27 | ||
819f5b0f | 28 | #ifndef __CINT__ |
8cc0ed91 | 29 | #include <Riostream.h> |
30 | ||
819f5b0f | 31 | #include "TStopwatch.h" |
32 | #include "TMemStat.h" | |
33 | #include "TMemStatViewerGUI.h" | |
34 | ||
35 | #include "TROOT.h" | |
36 | #include "TSystem.h" | |
8cc0ed91 | 37 | #include "TError.h" |
819f5b0f | 38 | #include "TChain.h" |
39 | ||
8cc0ed91 | 40 | #include "AliMagFMaps.h" |
41 | #include "AliTracker.h" | |
819f5b0f | 42 | #include "AliLog.h" |
43 | #include "AliCDBManager.h" | |
44 | #include "AliAnalysisManager.h" | |
45 | #include "AliAnalysisDataContainer.h" | |
8cc0ed91 | 46 | #include "AliMCEventHandler.h" |
47 | #include "AliESDInputHandler.h" | |
819f5b0f | 48 | |
8cc0ed91 | 49 | #include "TRD/AliTRDtrackerV1.h" |
50 | #include "TRD/AliTRDcalibDB.h" | |
819f5b0f | 51 | #include "TRD/qaRec/AliTRDtrackInfoGen.h" |
8cc0ed91 | 52 | #include "TRD/qaRec/AliTRDtrackingEfficiency.h" |
53 | #include "TRD/qaRec/AliTRDtrackingEfficiencyCombined.h" | |
54 | #include "TRD/qaRec/AliTRDtrackingResolution.h" | |
55 | #include "TRD/qaRec/AliTRDcalibration.h" | |
3d86166d | 56 | #include "TRD/qaRec/AliTRDpidChecker.h" |
28efdace | 57 | #include "TRD/qaRec/AliTRDpidRefMaker.h" |
2b468513 | 58 | #include "TRD/qaRec/AliTRDcheckDetector.h" |
819f5b0f | 59 | #endif |
60 | ||
3d86166d | 61 | #include "run.h" |
62 | ||
819f5b0f | 63 | Bool_t MEM = kFALSE; |
8cc0ed91 | 64 | |
65 | TChain* CreateESDChain(const char* filename = 0x0, Int_t nfiles=-1 ); | |
0ddce9c3 | 66 | void run(Char_t *tasks="ALL", const Char_t *files=0x0, Int_t nmax=-1) |
77203477 | 67 | { |
819f5b0f | 68 | TMemStat *mem = 0x0; |
8cc0ed91 | 69 | if(MEM){ |
70 | gSystem->Load("libMemStat.so"); | |
71 | gSystem->Load("libMemStatGui.so"); | |
72 | mem = new TMemStat("new, gnubuildin"); | |
73 | mem->AddStamp("Start"); | |
74 | } | |
77203477 | 75 | |
814ecea4 | 76 | TStopwatch timer; |
77 | timer.Start(); | |
814ecea4 | 78 | |
814ecea4 | 79 | gSystem->Load("libANALYSIS.so"); |
7c4d6d8e | 80 | gSystem->Load("libTRDqaRec.so"); |
77203477 | 81 | |
d2381af5 | 82 | Bool_t fHasMCdata = kTRUE; |
28efdace | 83 | Bool_t fHasFriends = kTRUE; |
773d3f8c | 84 | TObjArray *tasksArray = TString(tasks).Tokenize(" "); |
28efdace | 85 | |
86 | Int_t fSteerTask = 0; SETBIT(fSteerTask, kInfoGen); | |
773d3f8c | 87 | for(Int_t isel = 0; isel < tasksArray->GetEntriesFast(); isel++){ |
88 | TString s = (dynamic_cast<TObjString *>(tasksArray->UncheckedAt(isel)))->String(); | |
77203477 | 89 | if(s.CompareTo("ALL") == 0){ |
28efdace | 90 | for(Int_t itask = 1; itask < fknTasks; itask++) SETBIT(fSteerTask, itask); |
77203477 | 91 | continue; |
28efdace | 92 | } else if(s.CompareTo("NOFR") == 0){ |
93 | fHasFriends = kFALSE; | |
94 | } else if(s.CompareTo("NOMC") == 0){ | |
d2381af5 | 95 | fHasMCdata = kFALSE; |
28efdace | 96 | } else { |
97 | Bool_t foundOpt = kFALSE; | |
98 | for(Int_t itask = 1; itask < fknTasks; itask++){ | |
99 | if(s.CompareTo(fTaskOpt[itask]) != 0) continue; | |
100 | SETBIT(fSteerTask, itask); | |
101 | foundOpt = kTRUE; | |
102 | break; | |
103 | } | |
104 | if(!foundOpt) Info("run.C", Form("Task %s not implemented (yet).", s.Data())); | |
77203477 | 105 | } |
106 | } | |
8cc0ed91 | 107 | // define task list pointers; |
3d86166d | 108 | AliTRDrecoTask *taskPtr[fknTasks], *task = 0x0; |
8cc0ed91 | 109 | memset(taskPtr, 0, fknTasks*sizeof(AliAnalysisTask*)); |
8cc0ed91 | 110 | |
814ecea4 | 111 | //____________________________________________// |
8cc0ed91 | 112 | //gROOT->LoadMacro(Form("%s/TRD/qaRec/CreateESDChain.C", gSystem->ExpandPathName("$ALICE_ROOT"))); |
77203477 | 113 | TChain *chain = CreateESDChain(files, nmax); |
814ecea4 | 114 | //chain->SetBranchStatus("*", 0); |
115 | chain->SetBranchStatus("*FMD*",0); | |
116 | chain->SetBranchStatus("*Calo*",0); | |
117 | chain->SetBranchStatus("Tracks", 1); | |
118 | chain->SetBranchStatus("ESDfriend*",1); | |
119 | chain->Lookup(); | |
120 | chain->GetListOfFiles()->Print(); | |
121 | printf("\n ----> CHAIN HAS %d ENTRIES <----\n\n", (Int_t)chain->GetEntries()); | |
122 | ||
123 | AliLog::SetGlobalLogLevel(AliLog::kError); | |
124 | ||
125 | //____________________________________________ | |
126 | // Make the analysis manager | |
3d86166d | 127 | AliAnalysisManager *mgr = new AliAnalysisManager("TRD QA Reconstruction Manager"); |
814ecea4 | 128 | //mgr->SetSpecialOutputLocation(source); // To Be Changed |
8cc0ed91 | 129 | AliVEventHandler *esdH = 0x0, *mcH = 0x0; |
130 | mgr->SetInputEventHandler(esdH = new AliESDInputHandler); | |
131 | if(fHasMCdata) mgr->SetMCtruthEventHandler(mcH = new AliMCEventHandler()); | |
814ecea4 | 132 | //mgr->SetDebugLevel(10); |
133 | ||
134 | //____________________________________________ | |
135 | // TRD track summary generator | |
773d3f8c | 136 | mgr->AddTask(task = new AliTRDtrackInfoGen()); |
3d86166d | 137 | taskPtr[(Int_t)kInfoGen] = task; |
b718144c | 138 | task->SetDebugLevel(1); |
3d86166d | 139 | task->SetMCdata(fHasMCdata); |
814ecea4 | 140 | // Create containers for input/output |
141 | AliAnalysisDataContainer *cinput1 = mgr->CreateContainer("data", TChain::Class(), AliAnalysisManager::kInputContainer); | |
421fb961 | 142 | AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("trackInfo", TObjArray::Class(), AliAnalysisManager::kExchangeContainer); |
143 | AliAnalysisDataContainer *coutput1a = mgr->CreateContainer("eventInfo", AliTRDeventInfo::Class(), AliAnalysisManager::kExchangeContainer); | |
773d3f8c | 144 | mgr->ConnectInput( task, 0, cinput1); |
145 | mgr->ConnectOutput(task, 0, coutput1); | |
421fb961 | 146 | mgr->ConnectOutput(task, 1, coutput1a); |
814ecea4 | 147 | |
28efdace | 148 | //____________________________________________ |
149 | // TRD detector checker | |
150 | if(TESTBIT(fSteerTask, kCheckDetector)){ | |
151 | mgr->AddTask(task = new AliTRDcheckDetector()); | |
152 | taskPtr[(Int_t)kCheckDetector] = task; | |
9e548ae7 | 153 | task->SetDebugLevel(4); |
28efdace | 154 | task->SetMCdata(fHasMCdata); |
155 | ||
156 | // Create containers for input/output | |
157 | mgr->ConnectInput( task, 0, coutput1); | |
421fb961 | 158 | mgr->ConnectInput( task, 1, coutput1a); |
28efdace | 159 | mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName()))); |
160 | } | |
161 | ||
814ecea4 | 162 | //____________________________________________ |
163 | // TRD barrel tracking efficiency | |
28efdace | 164 | if(fHasMCdata && TESTBIT(fSteerTask, kTrackingEfficiency)){ |
773d3f8c | 165 | mgr->AddTask(task = new AliTRDtrackingEfficiency()); |
3d86166d | 166 | taskPtr[(Int_t)kTrackingEfficiency] = task; |
167 | task->SetDebugLevel(0); | |
168 | ||
77203477 | 169 | //Create containers for input/output |
773d3f8c | 170 | mgr->ConnectInput( task, 0, coutput1); |
28efdace | 171 | mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName()))); |
77203477 | 172 | } |
814ecea4 | 173 | |
174 | //____________________________________________ | |
175 | // TRD combined tracking efficiency | |
28efdace | 176 | if(fHasMCdata && TESTBIT(fSteerTask, kTrackingCombinedEfficiency)){ |
773d3f8c | 177 | mgr->AddTask(task = new AliTRDtrackingEfficiencyCombined()); |
3d86166d | 178 | taskPtr[(Int_t)kTrackingCombinedEfficiency] = task; |
179 | task->SetDebugLevel(0); | |
773d3f8c | 180 | |
77203477 | 181 | // Create containers for input/output |
773d3f8c | 182 | mgr->ConnectInput( task, 0, coutput1); |
28efdace | 183 | mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName()))); |
77203477 | 184 | } |
814ecea4 | 185 | |
77203477 | 186 | //____________________________________________ |
e26505aa | 187 | // TRD tracking resolution |
77203477 | 188 | if(TESTBIT(fSteerTask, kTrackingResolution)){ |
773d3f8c | 189 | mgr->AddTask(task = new AliTRDtrackingResolution()); |
3d86166d | 190 | taskPtr[(Int_t)kTrackingResolution] = task; |
191 | task->SetMCdata(fHasMCdata); | |
2b468513 | 192 | task->SetPostProcess(kFALSE); |
b718144c | 193 | task->SetDebugLevel(1); |
3d86166d | 194 | |
77203477 | 195 | // Create containers for input/output |
773d3f8c | 196 | mgr->ConnectInput( task, 0, coutput1); |
28efdace | 197 | mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName()))); |
77203477 | 198 | } |
814ecea4 | 199 | |
ba2773d0 | 200 | //____________________________________________ |
201 | // TRD calibration | |
202 | if(TESTBIT(fSteerTask, kCalibration)){ | |
773d3f8c | 203 | mgr->AddTask(task = new AliTRDcalibration()); |
3d86166d | 204 | taskPtr[(Int_t)kCalibration] = task; |
773d3f8c | 205 | ((AliTRDcalibration*)task)->SetLow(0); |
206 | ((AliTRDcalibration*)task)->SetHigh(30); | |
773d3f8c | 207 | ((AliTRDcalibration*)task)->SetFillZero(kFALSE); |
3d86166d | 208 | task->SetDebugLevel(0); |
209 | ||
ba2773d0 | 210 | // Create containers for input/output |
773d3f8c | 211 | mgr->ConnectInput(task, 0, coutput1); |
28efdace | 212 | mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName()))); |
ba2773d0 | 213 | } |
214 | ||
773d3f8c | 215 | //____________________________________________ |
216 | // TRD pid checker | |
217 | if(TESTBIT(fSteerTask, kPIDChecker)){ | |
218 | mgr->AddTask(task = new AliTRDpidChecker()); | |
3d86166d | 219 | taskPtr[(Int_t)kPIDChecker] = task; |
220 | task->SetDebugLevel(0); | |
c7cf2032 | 221 | task->SetMCdata(fHasMCdata); |
3d86166d | 222 | |
773d3f8c | 223 | // Create containers for input/output |
773d3f8c | 224 | mgr->ConnectInput( task, 0, coutput1); |
28efdace | 225 | mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName()))); |
773d3f8c | 226 | } |
227 | ||
28efdace | 228 | |
2b468513 | 229 | //____________________________________________ |
93e41bce | 230 | // TRD pid reference |
28efdace | 231 | if(TESTBIT(fSteerTask, kPIDRefMaker)){ |
232 | mgr->AddTask(task = new AliTRDpidRefMaker()); | |
233 | taskPtr[(Int_t)kPIDRefMaker] = task; | |
2b468513 | 234 | task->SetDebugLevel(0); |
d85cd79c | 235 | task->SetMCdata(fHasMCdata); |
2b468513 | 236 | |
237 | // Create containers for input/output | |
2b468513 | 238 | mgr->ConnectInput( task, 0, coutput1); |
28efdace | 239 | mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName()))); |
93e41bce | 240 | mgr->ConnectOutput(task, 1, mgr->CreateContainer(Form("%sNN", task->GetName()), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%sNN.root", task->GetName()))); |
241 | mgr->ConnectOutput(task, 2, mgr->CreateContainer(Form("%sLQ", task->GetName()), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%sLQ.root", task->GetName()))); | |
2b468513 | 242 | } |
243 | ||
28efdace | 244 | |
814ecea4 | 245 | if (!mgr->InitAnalysis()) return; |
3d86166d | 246 | printf("\n\tRUNNING TRAIN FOR TASKS:\n"); |
247 | for(Int_t itask = 1; itask < fknTasks; itask++){ | |
248 | if(TESTBIT(fSteerTask, itask)) printf("\t %s [%s]\n", taskPtr[itask]->GetTitle(), taskPtr[itask]->GetName()); | |
249 | } | |
250 | printf("\n\n"); | |
d85cd79c | 251 | //mgr->PrintStatus(); |
819f5b0f | 252 | |
253 | ||
254 | AliCDBManager *cdbManager = AliCDBManager::Instance(); | |
255 | cdbManager->SetDefaultStorage("local://$ALICE_ROOT"); | |
256 | //cdbManager->SetSpecificStorage("TRD/Calib/FEE","local:///u/bailhach/aliroot/database30head/database"); | |
257 | cdbManager->SetRun(0); | |
2c0cf367 | 258 | cdbManager->SetCacheFlag(kFALSE); |
819f5b0f | 259 | |
d2381af5 | 260 | // initialize TRD settings |
8cc0ed91 | 261 | AliMagFMaps *field = new AliMagFMaps("Maps","Maps", 2, 1., 10., AliMagFMaps::k5kG); |
262 | AliTracker::SetFieldMap(field, kTRUE); | |
2c0cf367 | 263 | AliTRDcalibDB *cal = AliTRDcalibDB::Instance(); |
264 | AliTRDtrackerV1::SetNTimeBins(cal->GetNumberOfTimeBins()); | |
8cc0ed91 | 265 | |
266 | ||
814ecea4 | 267 | mgr->StartAnalysis("local",chain); |
268 | ||
269 | timer.Stop(); | |
819f5b0f | 270 | timer.Print(); |
271 | ||
2c0cf367 | 272 | cal->Terminate(); |
8cc0ed91 | 273 | delete field; |
274 | delete cdbManager; | |
3d86166d | 275 | for(Int_t it=fknTasks-1; it>=0; it--) if(taskPtr[it]) delete taskPtr[it]; |
8cc0ed91 | 276 | if(mcH) delete mcH; |
277 | delete esdH; | |
278 | delete mgr; | |
279 | delete chain; | |
819f5b0f | 280 | if(MEM) delete mem; |
281 | if(MEM) TMemStatViewerGUI::ShowGUI(); | |
814ecea4 | 282 | } |
8cc0ed91 | 283 | |
284 | ||
285 | TChain* CreateESDChain(const char* filename, Int_t nfiles) | |
286 | { | |
287 | // Create the chain | |
288 | TChain* chain = new TChain("esdTree"); | |
289 | ||
290 | if(!filename){ | |
291 | chain->Add(Form("%s/AliESDs.root", gSystem->pwd())); | |
292 | return chain; | |
293 | } | |
294 | ||
295 | ||
296 | // read ESD files from the input list. | |
297 | ifstream in; | |
298 | in.open(filename); | |
299 | TString esdfile; | |
300 | while(in.good() && (nfiles--) ) { | |
301 | in >> esdfile; | |
302 | if (!esdfile.Contains("root")) continue; // protection | |
303 | chain->Add(esdfile.Data()); | |
304 | } | |
305 | ||
306 | in.close(); | |
307 | ||
308 | return chain; | |
309 | } |