29e8549f7d98ddb77d042be183b93a94bc95c3b1
[u/mrichter/AliRoot.git] / PWG1 / TRD / run.C
1 // Steer TRD QA train for Reconstruction (Clusterizer, Tracking and PID).
2 // 
3 // Usage:
4 //   run.C(optList, files, nev, first, runNo, ocdb_uri, grp_uri)
5 //
6 //   optList : "ALL" [default] or one/more of the following:
7 //     "EFF"  : TRD Tracking Efficiency 
8 //     "EFFC" : TRD Tracking Efficiency Combined (barrel + stand alone) - only in case of simulations
9 //     "MULT"  : TRD single track selection
10 //     "RES"  : TRD tracking Resolution
11 //     "CLRES": clusters Resolution
12 //     "CAL"  : TRD calibration
13 //     "ALGN" : TRD alignment
14 //     "PID"  : TRD PID - pion efficiency 
15 //     "PIDR" : TRD PID - reference data
16 //     "V0"   : monitor V0 performance for use in TRD PID calibration
17 //     "DET"  : Basic TRD Detector checks
18 //      ****** SPECIAL OPTIONS **********
19 //     "NOFR" : Data set does not have AliESDfriends.root 
20 //     "NOMC" : Data set does not have Monte Carlo Informations (default have MC), 
21 //
22 //     files : the list of ESD files to be processed [default AliESds.root from cwd]
23 //     nev   : number of events to be processed [default all]
24 //     first : first event to process [default 0]
25 //     runNo : run number [default 0]
26 //     ocdb_uri : OCDB location [default local, $ALICE_ROOT]. In case of AliEn the syntax should be of the form
27 //                 alien://folder=/alice/data/2010/OCDB?user=username?cacheF=yourDirectory?cacheS=yourSize
28 //     grp_uri  : GRP/GRP/Data location [default cwd]. In case of AliEn the syntax should be of the form
29 //                 alien://folder=/alice/data/2010/OCDB?user=username?cacheF=yourDirectory?cacheS=yourSize
30 // In compiled mode : 
31 // Don't forget to load first the libraries
32 // gSystem->Load("libMemStat.so")
33 // gSystem->Load("libMemStatGui.so")
34 // gSystem->Load("libANALYSIS.so")
35 // gSystem->Load("libANALYSISalice.so")
36 // gSystem->Load("libTENDER.so");
37 // gSystem->Load("libPWG1.so");
38 // gSystem->Load("libNetx.so") ;
39 // gSystem->Load("libRAliEn.so");
40 //
41 // Authors:
42 //   Alex Bercuci (A.Bercuci@gsi.de) 
43 //   Markus Fasel (m.Fasel@gsi.de) 
44
45 #if ! defined (__CINT__) || defined (__MAKECINT__)
46 //#ifndef __CINT__
47 #include <Riostream.h>
48
49 #include "TStopwatch.h"
50 #include "TMemStat.h"
51 #include "TMemStatViewerGUI.h"
52
53 #include "TROOT.h"
54 #include "TClass.h"
55 #include "TSystem.h"
56 #include "TError.h"
57 #include "TChain.h"
58 #include "TGrid.h"
59 #include "TAlienCollection.h"
60 #include "TGridCollection.h"
61 #include "TGridResult.h"
62 #include "TGeoGlobalMagField.h"
63
64 #include "AliMagF.h"
65 #include "AliTracker.h"
66 #include "AliLog.h"
67 #include "AliCDBManager.h"
68 #include "AliGRPManager.h"
69 #include "AliGeomManager.h"
70 #include "AliAnalysisManager.h"
71 #include "AliAnalysisDataContainer.h"
72 #include "AliMCEventHandler.h"
73 #include "AliESDInputHandler.h"
74
75 #include "TRD/AliTRDtrackerV1.h"
76 #include "TRD/AliTRDcalibDB.h"
77
78 #include "PWG1/TRD/macros/AliTRDperformanceTrain.h"
79 #include "PWG1/TRD/macros/AddTRDcheckESD.C"
80 #include "PWG1/TRD/macros/AddTRDinfoGen.C"
81 #include "PWG1/TRD/macros/AddTRDcheckDET.C"
82 #include "PWG1/TRD/macros/AddTRDefficiency.C"
83 #include "PWG1/TRD/macros/AddTRDresolution.C"
84 #include "PWG1/TRD/macros/AddTRDcheckPID.C"
85
86 #endif
87
88 #include "macros/AliTRDperformanceTrain.h"
89
90
91 Bool_t MEM = kFALSE;
92
93 TChain* MakeChainLST(const char* filename = 0x0);
94 TChain* MakeChainXML(const char* filename = 0x0);
95 void run(Char_t *optList="ALL", const Char_t *files=0x0, Long64_t nev=1234567890, Long64_t first = 0)
96 {
97   TMemStat *mem = 0x0;
98   if(MEM){ 
99     if(gSystem->Load("libMemStat.so")<0) return;
100     if(gSystem->Load("libMemStatGui.so")<0) return;
101     mem = new TMemStat("new, gnubuildin");
102     mem->AddStamp("Start");
103   }
104   TStopwatch timer;
105   timer.Start();
106
107   // VERY GENERAL SETTINGS
108   //AliLog::SetGlobalLogLevel(AliLog::kError);
109   gStyle->SetOptStat(0);
110   if(gSystem->Load("libANALYSIS.so")<0) return;
111   if(gSystem->Load("libANALYSISalice.so")<0) return;
112   if(gSystem->Load("libTENDER.so")<0) return;
113   if(gSystem->Load("libPWG1.so")<0) return;
114
115   Bool_t fHasMCdata =  HasReadMCData(optList);
116   Bool_t fHasFriends = HasReadFriendData(optList);
117   
118   // DEFINE DATA CHAIN
119   TChain *chain = 0x0;
120   if(!files) chain = MakeChainLST();
121   else{
122     TString fn(files);
123     if(fn.EndsWith("xml")) chain = MakeChainXML(files);
124     else chain = MakeChainLST(files);
125   }
126   if(!chain) return;
127   chain->Lookup();
128   chain->GetListOfFiles()->Print();
129   Int_t nfound=(Int_t)chain->GetEntries();
130   printf("\tENTRIES FOUND [%d] REQUESTED [%d]\n", nfound, nev>nfound?nfound:nev);
131
132
133   // BUILD ANALYSIS MANAGER
134   AliAnalysisManager *mgr = new AliAnalysisManager("TRD Reconstruction Performance & Calibration");
135   AliESDInputHandlerRP *esdH(NULL);
136   mgr->SetInputEventHandler(esdH = new AliESDInputHandlerRP);
137   esdH->SetReadFriends(kTRUE);
138   esdH->SetActiveBranches("ESDfriend");
139   AliMCEventHandler *mcH(NULL);
140   if(fHasMCdata) mgr->SetMCtruthEventHandler(mcH = new AliMCEventHandler());
141   //mgr->SetDebugLevel(10);
142   mgr->SetSkipTerminate(kTRUE);
143
144   gROOT->LoadMacro("$ALICE_ROOT/PWG1/macros/AddTrainPerformanceTRD.C");
145   if(!AddTrainPerformanceTRD(optList)) {
146     Error("run.C", "Error loading TRD train.");
147     return;
148   }
149
150   if (!mgr->InitAnalysis()) return;
151   // verbosity
152   printf("\tRUNNING TRAIN FOR TASKS:\n");
153   TObjArray *taskList=mgr->GetTasks();
154   for(Int_t itask=0; itask<taskList->GetEntries(); itask++){ 
155     AliAnalysisTask *task=(AliAnalysisTask*)taskList->At(itask);
156     printf(" %s [%s]\n", task->GetName(), task->GetTitle());
157   }
158   //mgr->PrintStatus();
159   mgr->StartAnalysis("local", chain, nev, first);
160   timer.Stop();
161   timer.Print();  
162
163   // verbosity
164   printf("\tCLEANING TASK LIST:\n");
165   mgr->GetTasks()->Delete();
166
167   if(mcH) delete mcH;
168   delete esdH;
169   delete chain;
170   if(MEM) delete mem;
171   if(MEM) TMemStatViewerGUI::ShowGUI();
172 }
173
174 //____________________________________________
175 TChain* MakeChainLST(const char* filename)
176 {
177   // Create the chain
178   TChain* chain = new TChain("esdTree");
179
180   if(!filename){
181     chain->Add(Form("%s/AliESDs.root", gSystem->pwd()));
182     return chain;
183   }
184
185
186   // read ESD files from the input list.
187   ifstream in;
188   in.open(filename);
189   TString esdfile;
190   while(in.good()) {
191     in >> esdfile;
192     if (!esdfile.Contains("root")) continue; // protection
193     chain->Add(esdfile.Data());
194   }
195
196   in.close();
197
198   return chain;
199 }
200
201 //____________________________________________
202 TChain* MakeChainXML(const char* xmlfile)
203 {
204   if (!TFile::Open(xmlfile)) {
205     Error("MakeChainXML", Form("No file %s was found", xmlfile));
206     return 0x0;
207   }
208
209   if(gSystem->Load("libNetx.so")<0) return 0x0;
210   if(gSystem->Load("libRAliEn.so")<0) return 0x0;
211   TGrid::Connect("alien://") ;
212
213   TGridCollection *collection = (TGridCollection*) TAlienCollection::Open(xmlfile);
214   if (!collection) {
215     Error("MakeChainXML", Form("No collection found in %s", xmlfile)) ; 
216     return 0x0; 
217   }
218   //collection->CheckIfOnline();
219
220   TGridResult* result = collection->GetGridResult("",0 ,0);
221   if(!result->GetEntries()){
222     Error("MakeChainXML", Form("No entries found in %s", xmlfile)) ; 
223     return 0x0; 
224   }
225   // Makes the ESD chain 
226   TChain* chain = new TChain("esdTree");
227   for (Int_t idx = 0; idx < result->GetEntries(); idx++) {
228     chain->Add(result->GetKey(idx, "turl")); 
229   }
230   return chain;
231 }