new task introduced by Alex Wilk (calculation of reference data for NN
[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 //     "NOFR" : Data set does not have AliESDfriends.root 
14 //     "NOMC" : Data set does not have Monte Carlo Informations (real data), so all tasks which rely
15 //              on MC information are switched off
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 //
24 // Authors:
25 //   Alex Bercuci (A.Bercuci@gsi.de) 
26 //   Markus Fasel (m.Fasel@gsi.de) 
27
28 #ifndef __CINT__
29 #include <Riostream.h>
30
31 #include "TStopwatch.h"
32 #include "TMemStat.h"
33 #include "TMemStatViewerGUI.h"
34
35 #include "TROOT.h"
36 #include "TSystem.h"
37 #include "TError.h"
38 #include "TChain.h"
39
40 #include "AliMagFMaps.h"
41 #include "AliTracker.h"
42 #include "AliLog.h"
43 #include "AliCDBManager.h"
44 #include "AliAnalysisManager.h"
45 #include "AliAnalysisDataContainer.h"
46 #include "AliMCEventHandler.h"
47 #include "AliESDInputHandler.h"
48
49 #include "TRD/AliTRDtrackerV1.h"
50 #include "TRD/AliTRDcalibDB.h"
51 #include "TRD/qaRec/AliTRDtrackInfoGen.h"
52 #include "TRD/qaRec/AliTRDtrackingEfficiency.h"
53 #include "TRD/qaRec/AliTRDtrackingEfficiencyCombined.h"
54 #include "TRD/qaRec/AliTRDtrackingResolution.h"
55 #include "TRD/qaRec/AliTRDcalibration.h"
56 #include "TRD/qaRec/AliTRDpidChecker.h"
57 #include "TRD/qaRec/AliTRDpidRefMaker.h"
58 #include "TRD/qaRec/AliTRDcheckDetector.h"
59 #endif
60
61 #include "run.h"
62
63 Bool_t MEM = kFALSE;
64
65 TChain* CreateESDChain(const char* filename = 0x0, Int_t nfiles=-1 );
66 void run(const Char_t *files=0x0, Char_t *tasks="ALL", Int_t nmax=-1)
67 {
68   TMemStat *mem = 0x0;
69   if(MEM){ 
70     gSystem->Load("libMemStat.so");
71     gSystem->Load("libMemStatGui.so");
72     mem = new TMemStat("new, gnubuildin");
73     mem->AddStamp("Start");
74   }
75
76   TStopwatch timer;
77   timer.Start();
78
79   gSystem->Load("libANALYSIS.so");
80   gSystem->Load("libTRDqaRec.so");
81   
82   Bool_t fHasMCdata = kTRUE;
83   Bool_t fHasFriends = kTRUE;
84   TObjArray *tasksArray = TString(tasks).Tokenize(" ");
85
86   Int_t fSteerTask = 0; SETBIT(fSteerTask, kInfoGen);
87   for(Int_t isel = 0; isel < tasksArray->GetEntriesFast(); isel++){
88     TString s = (dynamic_cast<TObjString *>(tasksArray->UncheckedAt(isel)))->String();
89     if(s.CompareTo("ALL") == 0){
90       for(Int_t itask = 1; itask < fknTasks; itask++) SETBIT(fSteerTask, itask);
91       continue;
92     } else if(s.CompareTo("NOFR") == 0){ 
93       fHasFriends = kFALSE;
94     } else if(s.CompareTo("NOMC") == 0){ 
95       fHasMCdata = kFALSE;
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()));
105     }
106   }
107   // define task list pointers;
108   AliTRDrecoTask *taskPtr[fknTasks], *task = 0x0;
109   memset(taskPtr, 0, fknTasks*sizeof(AliAnalysisTask*));
110
111   //____________________________________________//
112   //gROOT->LoadMacro(Form("%s/TRD/qaRec/CreateESDChain.C", gSystem->ExpandPathName("$ALICE_ROOT")));
113   TChain *chain = CreateESDChain(files, nmax);
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
127   AliAnalysisManager *mgr = new AliAnalysisManager("TRD QA Reconstruction Manager");
128   //mgr->SetSpecialOutputLocation(source); // To Be Changed
129   AliVEventHandler *esdH = 0x0, *mcH = 0x0;
130   mgr->SetInputEventHandler(esdH = new AliESDInputHandler);
131   if(fHasMCdata) mgr->SetMCtruthEventHandler(mcH = new AliMCEventHandler());
132   //mgr->SetDebugLevel(10);
133
134   //____________________________________________
135   // TRD track summary generator
136   mgr->AddTask(task = new AliTRDtrackInfoGen());
137   taskPtr[(Int_t)kInfoGen] = task;
138   task->SetDebugLevel(0);
139   task->SetMCdata(fHasMCdata);
140   // Create containers for input/output
141   AliAnalysisDataContainer *cinput1 = mgr->CreateContainer("data", TChain::Class(), AliAnalysisManager::kInputContainer);
142   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
143   mgr->ConnectInput( task, 0, cinput1);
144   mgr->ConnectOutput(task, 0, coutput1);
145
146   //____________________________________________
147   // TRD detector checker
148         if(TESTBIT(fSteerTask, kCheckDetector)){
149     mgr->AddTask(task = new AliTRDcheckDetector());
150     taskPtr[(Int_t)kCheckDetector] = task;
151     task->SetDebugLevel(0);
152     task->SetMCdata(fHasMCdata);
153     
154     // Create containers for input/output
155     mgr->ConnectInput( task, 0, coutput1);
156     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
157   }
158
159   //____________________________________________
160   // TRD barrel tracking efficiency
161   if(fHasMCdata && TESTBIT(fSteerTask, kTrackingEfficiency)){
162     mgr->AddTask(task = new AliTRDtrackingEfficiency());
163     taskPtr[(Int_t)kTrackingEfficiency] = task;
164     task->SetDebugLevel(0);
165
166     //Create containers for input/output
167     mgr->ConnectInput( task, 0, coutput1);
168     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
169   }
170
171   //____________________________________________
172   // TRD combined tracking efficiency
173   if(fHasMCdata && TESTBIT(fSteerTask, kTrackingCombinedEfficiency)){
174     mgr->AddTask(task = new AliTRDtrackingEfficiencyCombined());
175     taskPtr[(Int_t)kTrackingCombinedEfficiency] = task;
176     task->SetDebugLevel(0);
177
178     // Create containers for input/output
179     mgr->ConnectInput( task, 0, coutput1);
180     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
181   }
182
183   //____________________________________________
184   // TRD tracking resolution
185   if(TESTBIT(fSteerTask, kTrackingResolution)){
186     mgr->AddTask(task = new AliTRDtrackingResolution());
187     taskPtr[(Int_t)kTrackingResolution] = task;
188     task->SetMCdata(fHasMCdata);
189     task->SetPostProcess(kFALSE);
190     task->SetDebugLevel(2);
191     
192     // Create containers for input/output
193     mgr->ConnectInput( task, 0, coutput1);
194     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
195   }
196
197   //____________________________________________
198   // TRD calibration
199   if(TESTBIT(fSteerTask, kCalibration)){
200     mgr->AddTask(task = new AliTRDcalibration());
201     taskPtr[(Int_t)kCalibration] = task;
202     ((AliTRDcalibration*)task)->SetLow(0);
203     ((AliTRDcalibration*)task)->SetHigh(30);
204     ((AliTRDcalibration*)task)->SetFillZero(kFALSE);
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 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     mgr->ConnectInput( task, 0, coutput1);
222     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
223   }
224
225
226   //____________________________________________
227   // TRD pid checker
228   if(TESTBIT(fSteerTask, kPIDRefMaker)){
229     mgr->AddTask(task = new AliTRDpidRefMaker());
230     taskPtr[(Int_t)kPIDRefMaker] = task;
231     task->SetDebugLevel(0);
232     task->SetMCdata(fHasMCdata);
233     
234     // Create containers for input/output
235     mgr->ConnectInput( task, 0, coutput1);
236     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
237   }
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 }