The present commit corresponds to an important change in the way the
[u/mrichter/AliRoot.git] / TRD / qaRec / run.C
1 // Steer TRD QA train for Reconstruction (Clusterizer, Tracking and PID).
2 // 
3 // Usage:
4 //   run.C(tasks, files)
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
9 //     "CLRES": clusters Resolution
10 //     "CAL"  : TRD calibration
11 //     "ALGN" : TRD alignment
12 //     "PID"  : TRD PID - pion efficiency 
13 //     "PIDR" : TRD PID - reference data
14 //     "DET"  : Basic TRD Detector checks
15 //     "NOFR" : Data set does not have AliESDfriends.root 
16 //     "NOMC" : Data set does not have Monte Carlo Informations (real data), so all tasks which rely
17 //              on MC information are switched off
18 //
19 // In compiled mode : 
20 // Don't forget to load first the libraries
21 // gSystem->Load("libMemStat.so")
22 // gSystem->Load("libMemStatGui.so")
23 // gSystem->Load("libANALYSIS.so")
24 // gSystem->Load("libTRDqaRec.so")
25 // gSystem->Load("libNetx.so") ;
26 // gSystem->Load("libRAliEn.so");
27 //
28 // Authors:
29 //   Alex Bercuci (A.Bercuci@gsi.de) 
30 //   Markus Fasel (m.Fasel@gsi.de) 
31
32 #ifndef __CINT__
33 #include <Riostream.h>
34
35 #include "TStopwatch.h"
36 #include "TMemStat.h"
37 #include "TMemStatViewerGUI.h"
38
39 #include "TROOT.h"
40 #include "TSystem.h"
41 #include "TError.h"
42 #include "TChain.h"
43 #include "TGrid.h"
44 #include "TAlienCollection.h"
45 #include "TGridCollection.h"
46 #include "TGridResult.h"
47
48 #include "AliMagF.h"
49 #include "AliTracker.h"
50 #include "AliLog.h"
51 #include "AliCDBManager.h"
52 #include "AliGeomManager.h"
53 #include "AliAnalysisManager.h"
54 #include "AliAnalysisDataContainer.h"
55 #include "AliMCEventHandler.h"
56 #include "AliESDInputHandler.h"
57
58 #include "TRD/AliTRDtrackerV1.h"
59 #include "TRD/AliTRDcalibDB.h"
60 #include "TRD/qaRec/AliTRDtrackInfo/AliTRDeventInfo.h"
61 #include "TRD/qaRec/AliTRDtrackInfoGen.h"
62 #include "TRD/qaRec/AliTRDtrackingEfficiency.h"
63 #include "TRD/qaRec/AliTRDtrackingEfficiencyCombined.h"
64 #include "TRD/qaRec/AliTRDtrackingResolution.h"
65 #include "TRD/qaRec/AliTRDcalibration.h"
66 #include "TRD/qaRec/AliTRDalignmentTask.h"
67 #include "TRD/qaRec/AliTRDpidChecker.h"
68 #include "TRD/qaRec/AliTRDpidRefMaker.h"
69 #include "TRD/qaRec/AliTRDcheckDetector.h"
70 #include "TRD/qaRec/AliTRDclusterResolution.h"
71 #endif
72
73 #include "run.h"
74
75 Bool_t MEM = kFALSE;
76
77 TChain* MakeChainLST(const char* filename = 0x0);
78 TChain* MakeChainXML(const char* filename = 0x0);
79 void run(Char_t *tasks="ALL", const Char_t *files=0x0)
80 {
81   TMemStat *mem = 0x0;
82   if(MEM){ 
83     gSystem->Load("libMemStat.so");
84     gSystem->Load("libMemStatGui.so");
85     mem = new TMemStat("new, gnubuildin");
86     mem->AddStamp("Start");
87   }
88
89   TStopwatch timer;
90   timer.Start();
91
92   if(gSystem->Load("libANALYSIS.so")<0) return;
93   if(gSystem->Load("libTRDqaRec.so")<0) return;
94   
95   // DB INITIALIZATION
96   // initialize OCDB manager
97   AliCDBManager *cdbManager = AliCDBManager::Instance();
98   cdbManager->SetDefaultStorage("local://$ALICE_ROOT");
99   cdbManager->SetRun(0);
100   cdbManager->SetCacheFlag(kFALSE);
101
102   // initialize TRD settings
103   AliTRDcalibDB *cal = AliTRDcalibDB::Instance();
104   AliTRDtrackerV1::SetNTimeBins(cal->GetNumberOfTimeBins());
105   AliGeomManager::LoadGeometry();
106
107   Bool_t fHasMCdata = kTRUE;
108   Bool_t fHasFriends = kTRUE;
109   TObjArray *tasksArray = TString(tasks).Tokenize(" ");
110
111   Int_t fSteerTask = 0; SETBIT(fSteerTask, kInfoGen);
112   for(Int_t isel = 0; isel < tasksArray->GetEntriesFast(); isel++){
113     TString s = (dynamic_cast<TObjString *>(tasksArray->UncheckedAt(isel)))->String();
114     if(s.CompareTo("ALL") == 0){
115       for(Int_t itask = 1; itask < NQATASKS; itask++) SETBIT(fSteerTask, itask);
116       continue;
117     } else if(s.CompareTo("NOFR") == 0){ 
118       fHasFriends = kFALSE;
119     } else if(s.CompareTo("NOMC") == 0){ 
120       fHasMCdata = kFALSE;
121     } else { 
122       Bool_t foundOpt = kFALSE;  
123       for(Int_t itask = 1; itask < NTRDTASKS; itask++){
124         if(s.CompareTo(fgkTRDtaskOpt[itask]) != 0) continue;
125         SETBIT(fSteerTask, itask);
126         foundOpt = kTRUE;
127         break;
128       }
129       if(!foundOpt) Info("run.C", Form("Task %s not implemented (yet).", s.Data()));
130     }
131   }
132   // extra rules for calibration tasks
133   if(TSTBIT(fSteerTask, kClErrParam)) SETBIT(fSteerTask, kResolution);
134   if(TSTBIT(fSteerTask, kPIDRefMaker)) SETBIT(fSteerTask, kPIDChecker);
135   if(TSTBIT(fSteerTask, kAlignment)) SETBIT(fSteerTask, kResolution);
136
137   // define task list pointers;
138   AliTRDrecoTask *taskPtr[NTRDTASKS], *task = 0x0;
139   memset(taskPtr, 0, NTRDTASKS*sizeof(AliAnalysisTask*));
140
141   //____________________________________________//
142   // DEFINE DATA CHAIN
143   TChain *chain = 0x0;
144   if(!files) chain = MakeChainLST();
145   else{
146     TString fn(files);
147     if(fn.EndsWith("xml")) chain = MakeChainXML(files);
148     else chain = MakeChainLST(files);
149   }
150   if(!chain) return;
151
152   //chain->SetBranchStatus("*", 0);
153   chain->SetBranchStatus("*FMD*",0);
154   chain->SetBranchStatus("*Calo*",0);
155   chain->SetBranchStatus("Tracks", 1);
156   chain->SetBranchStatus("ESDfriend*",1);
157   chain->Lookup();
158   chain->GetListOfFiles()->Print();
159   printf("\n ----> CHAIN HAS %d ENTRIES <----\n\n", (Int_t)chain->GetEntries());
160   
161   AliLog::SetGlobalLogLevel(AliLog::kError);
162
163   //____________________________________________
164   // Make the analysis manager
165   AliAnalysisManager *mgr = new AliAnalysisManager("TRD QA Reconstruction Manager");
166   //mgr->SetSpecialOutputLocation(source); // To Be Changed
167   AliVEventHandler *esdH = 0x0, *mcH = 0x0;
168   mgr->SetInputEventHandler(esdH = new AliESDInputHandler);
169   if(fHasMCdata) mgr->SetMCtruthEventHandler(mcH = new AliMCEventHandler());
170   //mgr->SetDebugLevel(10);
171
172   //____________________________________________
173   // TRD track summary generator
174   mgr->AddTask(task = new AliTRDtrackInfoGen());
175   taskPtr[(Int_t)kInfoGen] = task;
176   task->SetDebugLevel(1);
177   task->SetMCdata(fHasMCdata);
178   // Create containers for input/output
179   AliAnalysisDataContainer *cinput1 = mgr->CreateContainer("data", TChain::Class(), AliAnalysisManager::kInputContainer);
180   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("trackInfo", TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
181   AliAnalysisDataContainer *coutput1a = mgr->CreateContainer("eventInfo", AliTRDeventInfo::Class(), AliAnalysisManager::kExchangeContainer);
182   mgr->ConnectInput( task, 0, cinput1);
183   mgr->ConnectOutput(task, 0, coutput1);
184   mgr->ConnectOutput(task, 1, coutput1a);
185
186   //____________________________________________
187   // TRD detector checker
188         if(TSTBIT(fSteerTask, kCheckDetector)){
189     mgr->AddTask(task = new AliTRDcheckDetector());
190     taskPtr[(Int_t)kCheckDetector] = task;
191     task->SetDebugLevel(4);
192     task->SetMCdata(fHasMCdata);
193     
194     // Create containers for input/output
195     mgr->ConnectInput( task, 0, coutput1);
196     mgr->ConnectInput( task, 1, coutput1a);
197     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
198   }
199
200   //____________________________________________
201   // TRD barrel tracking efficiency
202   if(fHasMCdata && TSTBIT(fSteerTask, kTrackingEff)){
203     mgr->AddTask(task = new AliTRDtrackingEfficiency());
204     taskPtr[(Int_t)kTrackingEff] = task;
205     task->SetDebugLevel(0);
206
207     //Create containers for input/output
208     mgr->ConnectInput( task, 0, coutput1);
209     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
210   }
211
212   //____________________________________________
213   // TRD combined tracking efficiency
214   if(fHasMCdata && TSTBIT(fSteerTask, kTrackingEffMC)){
215     mgr->AddTask(task = new AliTRDtrackingEfficiencyCombined());
216     taskPtr[(Int_t)kTrackingEffMC] = task;
217     task->SetDebugLevel(0);
218
219     // Create containers for input/output
220     mgr->ConnectInput( task, 0, coutput1);
221     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
222   }
223
224   //____________________________________________
225   // TRD tracking resolution
226   if(TSTBIT(fSteerTask, kResolution)){
227     mgr->AddTask(task = new AliTRDtrackingResolution());
228     taskPtr[(Int_t)kResolution] = task;
229     task->SetMCdata(fHasMCdata);
230     task->SetPostProcess(kFALSE);
231     task->SetDebugLevel(1);
232     
233     // Create containers for input/output
234     mgr->ConnectInput( task, 0, coutput1);
235     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
236
237     // Create output containers for calibration tasks
238     const Int_t nc = 4;
239     const Char_t *cn[nc] = {"Cl", "Trklt", "MC_Cl", "MC_Trklt"}; 
240     AliAnalysisDataContainer *co[nc]; 
241     for(Int_t ic = 0; ic<nc; ic++){
242       co[ic] = mgr->CreateContainer(Form("%s%s", task->GetName(), cn[ic]), TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
243       mgr->ConnectOutput(task, 1+ic, co[ic]);
244     }
245     
246     // test reconstruction calibration plugin
247     if(TSTBIT(fSteerTask, kClErrParam)){
248       mgr->AddTask(task = new AliTRDclusterResolution());
249       taskPtr[(Int_t)kClErrParam] = task;
250       ((AliTRDclusterResolution*)task)->SetExB();
251       mgr->ConnectInput(task, 0, co[0]);
252       mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
253   
254       mgr->AddTask(task = new AliTRDclusterResolution("ClErrParamMC"));
255       taskPtr[(Int_t)kClErrParam+1] = task;
256       ((AliTRDclusterResolution*)task)->SetExB();
257       mgr->ConnectInput(task, 0, co[2]);
258       mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
259     }
260   }
261
262   //____________________________________________
263   // TRD calibration
264   if(TSTBIT(fSteerTask, kCalibration)){
265     mgr->AddTask(task = new AliTRDcalibration());
266     taskPtr[(Int_t)kCalibration] = task;
267     ((AliTRDcalibration*)task)->SetLow(0);
268     ((AliTRDcalibration*)task)->SetHigh(30);
269     ((AliTRDcalibration*)task)->SetFillZero(kFALSE);
270     task->SetDebugLevel(0);
271
272     // Create containers for input/output
273     mgr->ConnectInput(task, 0, coutput1);
274     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
275   }
276   
277   //____________________________________________
278   // TRD alignment
279   if(TSTBIT(fSteerTask, kAlignment)){
280     mgr->AddTask(task = new AliTRDalignmentTask());
281     taskPtr[(Int_t)kAlignment] = task;
282     task->SetDebugLevel(0);
283
284     // Create containers for input/output
285     mgr->ConnectInput(task, 0, coutput1);
286     mgr->ConnectOutput(task, 0, mgr->CreateContainer(Form("h%s", task->GetName()), TObjArray::Class(), AliAnalysisManager::kExchangeContainer));
287
288     mgr->ConnectOutput(task, 1, mgr->CreateContainer(task->GetName(), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
289   }
290   
291   //____________________________________________
292   // TRD PID
293   if(TSTBIT(fSteerTask, kPIDChecker)){
294     mgr->AddTask(task = new AliTRDpidChecker());
295     taskPtr[(Int_t)kPIDChecker] = task;
296     task->SetDebugLevel(0);
297     task->SetMCdata(fHasMCdata);
298     
299     // Create containers for input/output
300     mgr->ConnectInput( task, 0, coutput1);
301     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
302
303     //____________________________________________
304     // TRD pid reference 
305     if(TSTBIT(fSteerTask, kPIDRefMaker)){
306       mgr->AddTask(task = new AliTRDpidRefMaker());
307       taskPtr[(Int_t)kPIDRefMaker] = task;
308       task->SetDebugLevel(0);
309       task->SetMCdata(fHasMCdata);
310       
311       // Create containers for input/output
312       mgr->ConnectInput( task, 0, coutput1);
313       mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
314       mgr->ConnectOutput(task, 1, mgr->CreateContainer(Form("%sNN", task->GetName()), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%sNN.root", task->GetName())));
315       mgr->ConnectOutput(task, 2, mgr->CreateContainer(Form("%sLQ", task->GetName()), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%sLQ.root", task->GetName())));
316     }
317   }
318
319
320   if (!mgr->InitAnalysis()) return;
321   printf("\n\tRUNNING TRAIN FOR TASKS:\n");
322   for(Int_t itask = 1; itask < NTRDTASKS; itask++){
323     if(TSTBIT(fSteerTask, itask)) printf("\t   %s [%s]\n",  taskPtr[itask]->GetName(), taskPtr[itask]->GetTitle());
324   }
325   printf("\n\n");
326   //mgr->PrintStatus();
327
328   mgr->StartAnalysis("local",chain);
329
330   timer.Stop();
331   timer.Print();  
332
333   cal->Terminate();
334   delete field;
335   delete cdbManager;
336   for(Int_t it=NTRDTASKS; it--; ){ 
337     if(taskPtr[it]){ 
338       printf("Cleaning %s [%s] ...\n", fgkTRDtaskClassName[it], taskPtr[it]->GetTitle());
339       delete taskPtr[it];
340     }
341   }
342   if(mcH) delete mcH;
343   delete esdH;
344   delete mgr;
345   delete chain;
346   if(MEM) delete mem;
347   if(MEM) TMemStatViewerGUI::ShowGUI();
348 }
349
350 //____________________________________________
351 TChain* MakeChainLST(const char* filename)
352 {
353   // Create the chain
354   TChain* chain = new TChain("esdTree");
355
356   if(!filename){
357     chain->Add(Form("%s/AliESDs.root", gSystem->pwd()));
358     return chain;
359   }
360
361
362   // read ESD files from the input list.
363   ifstream in;
364   in.open(filename);
365   TString esdfile;
366   while(in.good()) {
367     in >> esdfile;
368     if (!esdfile.Contains("root")) continue; // protection
369     chain->Add(esdfile.Data());
370   }
371
372   in.close();
373
374   return chain;
375 }
376
377 //____________________________________________
378 TChain* MakeChainXML(const char* xmlfile)
379 {
380   if (!TFile::Open(xmlfile)) {
381     Error("MakeChainXML", Form("No file %s was found", xmlfile));
382     return 0x0;
383   }
384
385   if(gSystem->Load("libNetx.so")<0) return 0x0;
386   if(gSystem->Load("libRAliEn.so")<0) return 0x0;
387   TGrid::Connect("alien://") ;
388
389   TGridCollection *collection = (TGridCollection*) TAlienCollection::Open(xmlfile);
390   if (!collection) {
391     Error("MakeChainXML", Form("No collection found in %s", xmlfile)) ; 
392     return 0x0; 
393   }
394   //collection->CheckIfOnline();
395
396   TGridResult* result = collection->GetGridResult("",0 ,0);
397   if(!result->GetEntries()){
398     Error("MakeChainXML", Form("No entries found in %s", xmlfile)) ; 
399     return 0x0; 
400   }
401   // Makes the ESD chain 
402   TChain* chain = new TChain("esdTree");
403   for (Int_t idx = 0; idx < result->GetEntries(); idx++) {
404     chain->Add(result->GetKey(idx, "turl")); 
405   }
406   return chain;
407 }