inserted the first calibration task for reconstruction algorithms
[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/AliTRDtrackInfo/AliTRDeventInfo.h"
52 #include "TRD/qaRec/AliTRDtrackInfoGen.h"
53 #include "TRD/qaRec/AliTRDtrackingEfficiency.h"
54 #include "TRD/qaRec/AliTRDtrackingEfficiencyCombined.h"
55 #include "TRD/qaRec/AliTRDtrackingResolution.h"
56 #include "TRD/qaRec/AliTRDcalibration.h"
57 #include "TRD/qaRec/AliTRDpidChecker.h"
58 #include "TRD/qaRec/AliTRDpidRefMaker.h"
59 #include "TRD/qaRec/AliTRDcheckDetector.h"
60 #endif
61
62 #include "run.h"
63
64 Bool_t MEM = kFALSE;
65
66 TChain* CreateESDChain(const char* filename = 0x0, Int_t nfiles=-1 );
67 void run(Char_t *tasks="ALL", const Char_t *files=0x0, Int_t nmax=-1)
68 {
69   TMemStat *mem = 0x0;
70   if(MEM){ 
71     gSystem->Load("libMemStat.so");
72     gSystem->Load("libMemStatGui.so");
73     mem = new TMemStat("new, gnubuildin");
74     mem->AddStamp("Start");
75   }
76
77   TStopwatch timer;
78   timer.Start();
79
80   if(gSystem->Load("libANALYSIS.so")<0) return;
81   if(gSystem->Load("libTRDqaRec.so")<0) return;
82   
83   Bool_t fHasMCdata = kTRUE;
84   Bool_t fHasFriends = kTRUE;
85   TObjArray *tasksArray = TString(tasks).Tokenize(" ");
86
87   Int_t fSteerTask = 0; SETBIT(fSteerTask, kInfoGen);
88   for(Int_t isel = 0; isel < tasksArray->GetEntriesFast(); isel++){
89     TString s = (dynamic_cast<TObjString *>(tasksArray->UncheckedAt(isel)))->String();
90     if(s.CompareTo("ALL") == 0){
91       for(Int_t itask = 1; itask < NTRDTASKS; itask++) SETBIT(fSteerTask, itask);
92       continue;
93     } else if(s.CompareTo("NOFR") == 0){ 
94       fHasFriends = kFALSE;
95     } else if(s.CompareTo("NOMC") == 0){ 
96       fHasMCdata = kFALSE;
97     } else { 
98       Bool_t foundOpt = kFALSE;  
99       for(Int_t itask = 1; itask < NTRDTASKS; itask++){
100         if(s.CompareTo(fgkTRDtaskOpt[itask]) != 0) continue;
101         SETBIT(fSteerTask, itask);
102         foundOpt = kTRUE;
103         break;
104       }
105       if(!foundOpt) Info("run.C", Form("Task %s not implemented (yet).", s.Data()));
106     }
107   }
108   // define task list pointers;
109   AliTRDrecoTask *taskPtr[NTRDTASKS], *task = 0x0;
110   memset(taskPtr, 0, NTRDTASKS*sizeof(AliAnalysisTask*));
111
112   //____________________________________________//
113   //gROOT->LoadMacro(Form("%s/TRD/qaRec/CreateESDChain.C", gSystem->ExpandPathName("$ALICE_ROOT")));
114   TChain *chain = CreateESDChain(files, nmax);
115   //chain->SetBranchStatus("*", 0);
116   chain->SetBranchStatus("*FMD*",0);
117   chain->SetBranchStatus("*Calo*",0);
118   chain->SetBranchStatus("Tracks", 1);
119   chain->SetBranchStatus("ESDfriend*",1);
120   chain->Lookup();
121   chain->GetListOfFiles()->Print();
122   printf("\n ----> CHAIN HAS %d ENTRIES <----\n\n", (Int_t)chain->GetEntries());
123   
124   AliLog::SetGlobalLogLevel(AliLog::kError);
125
126   //____________________________________________
127   // Make the analysis manager
128   AliAnalysisManager *mgr = new AliAnalysisManager("TRD QA Reconstruction Manager");
129   //mgr->SetSpecialOutputLocation(source); // To Be Changed
130   AliVEventHandler *esdH = 0x0, *mcH = 0x0;
131   mgr->SetInputEventHandler(esdH = new AliESDInputHandler);
132   if(fHasMCdata) mgr->SetMCtruthEventHandler(mcH = new AliMCEventHandler());
133   //mgr->SetDebugLevel(10);
134
135   //____________________________________________
136   // TRD track summary generator
137   mgr->AddTask(task = new AliTRDtrackInfoGen());
138   taskPtr[(Int_t)kInfoGen] = task;
139   task->SetDebugLevel(1);
140   task->SetMCdata(fHasMCdata);
141   // Create containers for input/output
142   AliAnalysisDataContainer *cinput1 = mgr->CreateContainer("data", TChain::Class(), AliAnalysisManager::kInputContainer);
143   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("trackInfo", TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
144   AliAnalysisDataContainer *coutput1a = mgr->CreateContainer("eventInfo", AliTRDeventInfo::Class(), AliAnalysisManager::kExchangeContainer);
145   mgr->ConnectInput( task, 0, cinput1);
146   mgr->ConnectOutput(task, 0, coutput1);
147   mgr->ConnectOutput(task, 1, coutput1a);
148
149   //____________________________________________
150   // TRD detector checker
151         if(TSTBIT(fSteerTask, kCheckDetector)){
152     mgr->AddTask(task = new AliTRDcheckDetector());
153     taskPtr[(Int_t)kCheckDetector] = task;
154     task->SetDebugLevel(4);
155     task->SetMCdata(fHasMCdata);
156     
157     // Create containers for input/output
158     mgr->ConnectInput( task, 0, coutput1);
159     mgr->ConnectInput( task, 1, coutput1a);
160     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
161   }
162
163   //____________________________________________
164   // TRD barrel tracking efficiency
165   if(fHasMCdata && TSTBIT(fSteerTask, kTrackingEfficiency)){
166     mgr->AddTask(task = new AliTRDtrackingEfficiency());
167     taskPtr[(Int_t)kTrackingEfficiency] = task;
168     task->SetDebugLevel(0);
169
170     //Create containers for input/output
171     mgr->ConnectInput( task, 0, coutput1);
172     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
173   }
174
175   //____________________________________________
176   // TRD combined tracking efficiency
177   if(fHasMCdata && TSTBIT(fSteerTask, kTrackingCombinedEfficiency)){
178     mgr->AddTask(task = new AliTRDtrackingEfficiencyCombined());
179     taskPtr[(Int_t)kTrackingCombinedEfficiency] = task;
180     task->SetDebugLevel(0);
181
182     // Create containers for input/output
183     mgr->ConnectInput( task, 0, coutput1);
184     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
185   }
186
187   //____________________________________________
188   // TRD tracking resolution
189   if(TSTBIT(fSteerTask, kTrackingResolution)){
190     mgr->AddTask(task = new AliTRDtrackingResolution());
191     taskPtr[(Int_t)kTrackingResolution] = task;
192     task->SetMCdata(fHasMCdata);
193     task->SetPostProcess(kFALSE);
194     task->SetDebugLevel(1);
195     
196     // Create containers for input/output
197     mgr->ConnectInput( task, 0, coutput1);
198     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
199
200     AliAnalysisDataContainer *co1 = mgr->CreateContainer(Form("%sClRez", task->GetName()), TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
201     mgr->ConnectOutput(task, 1, co1);
202
203     AliAnalysisDataContainer *co2 = mgr->CreateContainer(Form("%sClRes", task->GetName()), TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
204     mgr->ConnectOutput(task, 2, co2);
205
206     AliAnalysisDataContainer *co3 = mgr->CreateContainer(Form("%sTrkltRes", task->GetName()), TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
207     mgr->ConnectOutput(task, 3, co3);
208     
209     // test reconstruction calibration plugin
210     mgr->AddTask(task = new AliTRDclusterResolution());
211     mgr->ConnectInput(task, 0, co2);
212     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
213   }
214
215   //____________________________________________
216   // TRD calibration
217   if(TSTBIT(fSteerTask, kCalibration)){
218     mgr->AddTask(task = new AliTRDcalibration());
219     taskPtr[(Int_t)kCalibration] = task;
220     ((AliTRDcalibration*)task)->SetLow(0);
221     ((AliTRDcalibration*)task)->SetHigh(30);
222     ((AliTRDcalibration*)task)->SetFillZero(kFALSE);
223     task->SetDebugLevel(0);
224
225     // Create containers for input/output
226     mgr->ConnectInput(task, 0, coutput1);
227     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
228   }
229   
230   //____________________________________________
231   // TRD pid checker
232   if(TSTBIT(fSteerTask, kPIDChecker)){
233     mgr->AddTask(task = new AliTRDpidChecker());
234     taskPtr[(Int_t)kPIDChecker] = task;
235     task->SetDebugLevel(0);
236     task->SetMCdata(fHasMCdata);
237     
238     // Create containers for input/output
239     mgr->ConnectInput( task, 0, coutput1);
240     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
241   }
242
243
244   //____________________________________________
245   // TRD pid reference 
246   if(TSTBIT(fSteerTask, kPIDRefMaker)){
247     mgr->AddTask(task = new AliTRDpidRefMaker());
248     taskPtr[(Int_t)kPIDRefMaker] = task;
249     task->SetDebugLevel(0);
250     task->SetMCdata(fHasMCdata);
251     
252     // Create containers for input/output
253     mgr->ConnectInput( task, 0, coutput1);
254     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
255     mgr->ConnectOutput(task, 1, mgr->CreateContainer(Form("%sNN", task->GetName()), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%sNN.root", task->GetName())));
256     mgr->ConnectOutput(task, 2, mgr->CreateContainer(Form("%sLQ", task->GetName()), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%sLQ.root", task->GetName())));
257   }
258
259
260   if (!mgr->InitAnalysis()) return;
261   printf("\n\tRUNNING TRAIN FOR TASKS:\n");
262   for(Int_t itask = 1; itask < NTRDTASKS; itask++){
263     if(TSTBIT(fSteerTask, itask)) printf("\t   %s [%s]\n", taskPtr[itask]->GetTitle(), taskPtr[itask]->GetName());
264   }
265   printf("\n\n");
266   //mgr->PrintStatus();
267
268
269   AliCDBManager *cdbManager = AliCDBManager::Instance();
270   cdbManager->SetDefaultStorage("local://$ALICE_ROOT");
271   //cdbManager->SetSpecificStorage("TRD/Calib/FEE","local:///u/bailhach/aliroot/database30head/database");
272   cdbManager->SetRun(0);
273   cdbManager->SetCacheFlag(kFALSE);
274  
275   // initialize TRD settings
276   AliMagFMaps *field = new AliMagFMaps("Maps","Maps", 2, 1., 10., AliMagFMaps::k5kG);
277   AliTracker::SetFieldMap(field, kTRUE);
278   AliTRDcalibDB *cal = AliTRDcalibDB::Instance();
279   AliTRDtrackerV1::SetNTimeBins(cal->GetNumberOfTimeBins());
280   AliGeomManager::LoadGeometry();
281
282
283   mgr->StartAnalysis("local",chain);
284
285   timer.Stop();
286   timer.Print();  
287
288   cal->Terminate();
289   delete field;
290   delete cdbManager;
291   for(Int_t it=NTRDTASKS-1; it>=0; it--) if(taskPtr[it]) delete taskPtr[it];
292   if(mcH) delete mcH;
293   delete esdH;
294   delete mgr;
295   delete chain;
296   if(MEM) delete mem;
297   if(MEM) TMemStatViewerGUI::ShowGUI();
298 }
299
300
301 TChain* CreateESDChain(const char* filename, Int_t nfiles)
302 {
303   // Create the chain
304   TChain* chain = new TChain("esdTree");
305
306   if(!filename){
307     chain->Add(Form("%s/AliESDs.root", gSystem->pwd()));
308     return chain;
309   }
310
311
312   // read ESD files from the input list.
313   ifstream in;
314   in.open(filename);
315   TString esdfile;
316   while(in.good() && (nfiles--) ) {
317     in >> esdfile;
318     if (!esdfile.Contains("root")) continue; // protection
319     chain->Add(esdfile.Data());
320   }
321
322   in.close();
323
324   return chain;
325 }