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