another codechecker warning fixed
[u/mrichter/AliRoot.git] / PWG1 / TRD / run.C
CommitLineData
1ee39b3a 1// Steer TRD QA train for Reconstruction (Clusterizer, Tracking and PID).
2//
3// Usage:
e9d62bd2 4// run.C(optList, files, nev, first, runNo, ocdb_uri, grp_uri)
5//
6// optList : "ALL" [default] or one/more of the following:
1ee39b3a 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
45d01ff8 16// "V0" : monitor V0 performance for use in TRD PID calibration
1ee39b3a 17// "DET" : Basic TRD Detector checks
e9d62bd2 18// ****** SPECIAL OPTIONS **********
1ee39b3a 19// "NOFR" : Data set does not have AliESDfriends.root
e9d62bd2 20// "NOMC" : Data set does not have Monte Carlo Informations (default have MC),
1ee39b3a 21//
e9d62bd2 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
1ee39b3a 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")
e9d62bd2 36// gSystem->Load("libTENDER.so");
1ee39b3a 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
d4f4bb29 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"
1ee39b3a 85
1ee39b3a 86#endif
87
d4f4bb29 88#include "macros/AliTRDperformanceTrain.h"
1ee39b3a 89
90
91Bool_t MEM = kFALSE;
92
93TChain* MakeChainLST(const char* filename = 0x0);
94TChain* MakeChainXML(const char* filename = 0x0);
06e1e9e2 95void run(Char_t *optList="ALL", const Char_t *files=0x0, Long64_t nev=1234567890, Long64_t first = 0)
1ee39b3a 96{
97 TMemStat *mem = 0x0;
98 if(MEM){
5bb83ff3 99 if(gSystem->Load("libMemStat.so")<0) return;
100 if(gSystem->Load("libMemStatGui.so")<0) return;
1ee39b3a 101 mem = new TMemStat("new, gnubuildin");
102 mem->AddStamp("Start");
103 }
104 TStopwatch timer;
105 timer.Start();
106
1ee39b3a 107 // VERY GENERAL SETTINGS
e9d62bd2 108 //AliLog::SetGlobalLogLevel(AliLog::kError);
058c3053 109 gStyle->SetOptStat(0);
1ee39b3a 110 if(gSystem->Load("libANALYSIS.so")<0) return;
111 if(gSystem->Load("libANALYSISalice.so")<0) return;
c732f879 112 if(gSystem->Load("libTENDER.so")<0) return;
d4f4bb29 113 if(gSystem->Load("libPWG1.so")<0) return;
1ee39b3a 114
523a3d00 115 Bool_t fHasMCdata = HasReadMCData(optList);
116 Bool_t fHasFriends = HasReadFriendData(optList);
1ee39b3a 117
1ee39b3a 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;
1ee39b3a 127 chain->Lookup();
128 chain->GetListOfFiles()->Print();
f2e89a4c 129 Int_t nfound=(Int_t)chain->GetEntries();
130 printf("\tENTRIES FOUND [%d] REQUESTED [%d]\n", nfound, nev>nfound?nfound:nev);
1ee39b3a 131
132
133 // BUILD ANALYSIS MANAGER
5bb83ff3 134 AliAnalysisManager *mgr = new AliAnalysisManager("TRD Reconstruction Performance & Calibration");
265a93da 135 AliESDInputHandlerRP *esdH(NULL);
136 mgr->SetInputEventHandler(esdH = new AliESDInputHandlerRP);
137 esdH->SetReadFriends(kTRUE);
138 esdH->SetActiveBranches("ESDfriend");
139 AliMCEventHandler *mcH(NULL);
1ee39b3a 140 if(fHasMCdata) mgr->SetMCtruthEventHandler(mcH = new AliMCEventHandler());
141 //mgr->SetDebugLevel(10);
72f5dcbe 142 mgr->SetSkipTerminate(kTRUE);
1ee39b3a 143
d4f4bb29 144 gROOT->LoadMacro("$ALICE_ROOT/PWG1/macros/AddTrainPerformanceTRD.C");
c3ebe2ae 145 if(!AddTrainPerformanceTRD(optList)) {
d4f4bb29 146 Error("run.C", "Error loading TRD train.");
147 return;
1ee39b3a 148 }
149
150 if (!mgr->InitAnalysis()) return;
151 // verbosity
5bb83ff3 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 }
b91fdd71 158 //mgr->PrintStatus();
1ee39b3a 159 mgr->StartAnalysis("local", chain, nev, first);
1ee39b3a 160 timer.Stop();
161 timer.Print();
162
1ee39b3a 163 // verbosity
5bb83ff3 164 printf("\tCLEANING TASK LIST:\n");
1ee39b3a 165 mgr->GetTasks()->Delete();
166
167 if(mcH) delete mcH;
168 delete esdH;
1ee39b3a 169 delete chain;
170 if(MEM) delete mem;
171 if(MEM) TMemStatViewerGUI::ShowGUI();
172}
173
174//____________________________________________
175TChain* 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//____________________________________________
202TChain* 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}