1 // Steer TRD QA train for Reconstruction (Clusterizer, Tracking and PID).
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
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")
25 // Alex Bercuci (A.Bercuci@gsi.de)
26 // Markus Fasel (m.Fasel@gsi.de)
29 #include <Riostream.h>
31 #include "TStopwatch.h"
33 #include "TMemStatViewerGUI.h"
40 #include "AliMagFMaps.h"
41 #include "AliTracker.h"
43 #include "AliCDBManager.h"
44 #include "AliAnalysisManager.h"
45 #include "AliAnalysisDataContainer.h"
46 #include "AliMCEventHandler.h"
47 #include "AliESDInputHandler.h"
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"
65 TChain* CreateESDChain(const char* filename = 0x0, Int_t nfiles=-1 );
66 void run(Char_t *tasks="ALL", const Char_t *files=0x0, Int_t nmax=-1)
70 gSystem->Load("libMemStat.so");
71 gSystem->Load("libMemStatGui.so");
72 mem = new TMemStat("new, gnubuildin");
73 mem->AddStamp("Start");
79 gSystem->Load("libANALYSIS.so");
80 gSystem->Load("libTRDqaRec.so");
82 Bool_t fHasMCdata = kTRUE;
83 Bool_t fHasFriends = kTRUE;
84 TObjArray *tasksArray = TString(tasks).Tokenize(" ");
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);
92 } else if(s.CompareTo("NOFR") == 0){
94 } else if(s.CompareTo("NOMC") == 0){
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);
104 if(!foundOpt) Info("run.C", Form("Task %s not implemented (yet).", s.Data()));
107 // define task list pointers;
108 AliTRDrecoTask *taskPtr[fknTasks], *task = 0x0;
109 memset(taskPtr, 0, fknTasks*sizeof(AliAnalysisTask*));
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);
120 chain->GetListOfFiles()->Print();
121 printf("\n ----> CHAIN HAS %d ENTRIES <----\n\n", (Int_t)chain->GetEntries());
123 AliLog::SetGlobalLogLevel(AliLog::kError);
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);
134 //____________________________________________
135 // TRD track summary generator
136 mgr->AddTask(task = new AliTRDtrackInfoGen());
137 taskPtr[(Int_t)kInfoGen] = task;
138 task->SetDebugLevel(1);
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("trackInfo", TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
143 AliAnalysisDataContainer *coutput1a = mgr->CreateContainer("eventInfo", AliTRDeventInfo::Class(), AliAnalysisManager::kExchangeContainer);
144 mgr->ConnectInput( task, 0, cinput1);
145 mgr->ConnectOutput(task, 0, coutput1);
146 mgr->ConnectOutput(task, 1, coutput1a);
148 //____________________________________________
149 // TRD detector checker
150 if(TESTBIT(fSteerTask, kCheckDetector)){
151 mgr->AddTask(task = new AliTRDcheckDetector());
152 taskPtr[(Int_t)kCheckDetector] = task;
153 task->SetDebugLevel(4);
154 task->SetMCdata(fHasMCdata);
156 // Create containers for input/output
157 mgr->ConnectInput( task, 0, coutput1);
158 mgr->ConnectInput( task, 1, coutput1a);
159 mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
162 //____________________________________________
163 // TRD barrel tracking efficiency
164 if(fHasMCdata && TESTBIT(fSteerTask, kTrackingEfficiency)){
165 mgr->AddTask(task = new AliTRDtrackingEfficiency());
166 taskPtr[(Int_t)kTrackingEfficiency] = task;
167 task->SetDebugLevel(0);
169 //Create containers for input/output
170 mgr->ConnectInput( task, 0, coutput1);
171 mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
174 //____________________________________________
175 // TRD combined tracking efficiency
176 if(fHasMCdata && TESTBIT(fSteerTask, kTrackingCombinedEfficiency)){
177 mgr->AddTask(task = new AliTRDtrackingEfficiencyCombined());
178 taskPtr[(Int_t)kTrackingCombinedEfficiency] = task;
179 task->SetDebugLevel(0);
181 // Create containers for input/output
182 mgr->ConnectInput( task, 0, coutput1);
183 mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
186 //____________________________________________
187 // TRD tracking resolution
188 if(TESTBIT(fSteerTask, kTrackingResolution)){
189 mgr->AddTask(task = new AliTRDtrackingResolution());
190 taskPtr[(Int_t)kTrackingResolution] = task;
191 task->SetMCdata(fHasMCdata);
192 task->SetPostProcess(kFALSE);
193 task->SetDebugLevel(1);
195 // Create containers for input/output
196 mgr->ConnectInput( task, 0, coutput1);
197 mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
200 //____________________________________________
202 if(TESTBIT(fSteerTask, kCalibration)){
203 mgr->AddTask(task = new AliTRDcalibration());
204 taskPtr[(Int_t)kCalibration] = task;
205 ((AliTRDcalibration*)task)->SetLow(0);
206 ((AliTRDcalibration*)task)->SetHigh(30);
207 ((AliTRDcalibration*)task)->SetFillZero(kFALSE);
208 task->SetDebugLevel(0);
210 // Create containers for input/output
211 mgr->ConnectInput(task, 0, coutput1);
212 mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
215 //____________________________________________
217 if(TESTBIT(fSteerTask, kPIDChecker)){
218 mgr->AddTask(task = new AliTRDpidChecker());
219 taskPtr[(Int_t)kPIDChecker] = task;
220 task->SetDebugLevel(0);
221 task->SetMCdata(fHasMCdata);
223 // Create containers for input/output
224 mgr->ConnectInput( task, 0, coutput1);
225 mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
229 //____________________________________________
231 if(TESTBIT(fSteerTask, kPIDRefMaker)){
232 mgr->AddTask(task = new AliTRDpidRefMaker());
233 taskPtr[(Int_t)kPIDRefMaker] = task;
234 task->SetDebugLevel(0);
235 task->SetMCdata(fHasMCdata);
237 // Create containers for input/output
238 mgr->ConnectInput( task, 0, coutput1);
239 mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
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())));
245 if (!mgr->InitAnalysis()) return;
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());
251 //mgr->PrintStatus();
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);
258 cdbManager->SetCacheFlag(kFALSE);
260 // initialize TRD settings
261 AliMagFMaps *field = new AliMagFMaps("Maps","Maps", 2, 1., 10., AliMagFMaps::k5kG);
262 AliTracker::SetFieldMap(field, kTRUE);
263 AliTRDcalibDB *cal = AliTRDcalibDB::Instance();
264 AliTRDtrackerV1::SetNTimeBins(cal->GetNumberOfTimeBins());
267 mgr->StartAnalysis("local",chain);
275 for(Int_t it=fknTasks-1; it>=0; it--) if(taskPtr[it]) delete taskPtr[it];
281 if(MEM) TMemStatViewerGUI::ShowGUI();
285 TChain* CreateESDChain(const char* filename, Int_t nfiles)
288 TChain* chain = new TChain("esdTree");
291 chain->Add(Form("%s/AliESDs.root", gSystem->pwd()));
296 // read ESD files from the input list.
300 while(in.good() && (nfiles--) ) {
302 if (!esdfile.Contains("root")) continue; // protection
303 chain->Add(esdfile.Data());