trigger names changed (Markus)
[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/AddTRDcheckESD.C"
79#include "PWG1/TRD/macros/AddTRDinfoGen.C"
80#include "PWG1/TRD/macros/AddTRDcheckDET.C"
81#include "PWG1/TRD/macros/AddTRDefficiency.C"
82#include "PWG1/TRD/macros/AddTRDresolution.C"
83#include "PWG1/TRD/macros/AddTRDcheckPID.C"
1ee39b3a 84#endif
85
1ee39b3a 86Bool_t MEM = kFALSE;
87
99535225 88TChain* MakeChainLST(const char* filename = NULL);
89TChain* MakeChainXML(const char* filename = NULL);
90Bool_t UseMC(Char_t *opt);
91Bool_t UseFriends(Char_t *opt);
92void run(Char_t *optList="ALL", const Char_t *files=NULL, Long64_t nev=1234567890, Long64_t first = 0)
1ee39b3a 93{
99535225 94 TMemStat *mem = NULL;
1ee39b3a 95 if(MEM){
5bb83ff3 96 if(gSystem->Load("libMemStat.so")<0) return;
97 if(gSystem->Load("libMemStatGui.so")<0) return;
1ee39b3a 98 mem = new TMemStat("new, gnubuildin");
99 mem->AddStamp("Start");
100 }
101 TStopwatch timer;
102 timer.Start();
103
1ee39b3a 104 // VERY GENERAL SETTINGS
e9d62bd2 105 //AliLog::SetGlobalLogLevel(AliLog::kError);
1ee39b3a 106 if(gSystem->Load("libANALYSIS.so")<0) return;
107 if(gSystem->Load("libANALYSISalice.so")<0) return;
c732f879 108 if(gSystem->Load("libTENDER.so")<0) return;
d4f4bb29 109 if(gSystem->Load("libPWG1.so")<0) return;
1ee39b3a 110
99535225 111 Bool_t fHasMCdata = UseMC(optList);
112 Bool_t fHasFriends = UseFriends(optList);
1ee39b3a 113
1ee39b3a 114 // DEFINE DATA CHAIN
99535225 115 TChain *chain = NULL;
1ee39b3a 116 if(!files) chain = MakeChainLST();
117 else{
118 TString fn(files);
119 if(fn.EndsWith("xml")) chain = MakeChainXML(files);
120 else chain = MakeChainLST(files);
121 }
122 if(!chain) return;
1ee39b3a 123 chain->Lookup();
124 chain->GetListOfFiles()->Print();
f2e89a4c 125 Int_t nfound=(Int_t)chain->GetEntries();
126 printf("\tENTRIES FOUND [%d] REQUESTED [%d]\n", nfound, nev>nfound?nfound:nev);
1ee39b3a 127
128
129 // BUILD ANALYSIS MANAGER
5bb83ff3 130 AliAnalysisManager *mgr = new AliAnalysisManager("TRD Reconstruction Performance & Calibration");
265a93da 131 AliESDInputHandlerRP *esdH(NULL);
132 mgr->SetInputEventHandler(esdH = new AliESDInputHandlerRP);
99535225 133 if(fHasFriends){
134 esdH->SetReadFriends(kTRUE);
135 esdH->SetActiveBranches("ESDfriend");
136 }
265a93da 137 AliMCEventHandler *mcH(NULL);
1ee39b3a 138 if(fHasMCdata) mgr->SetMCtruthEventHandler(mcH = new AliMCEventHandler());
139 //mgr->SetDebugLevel(10);
72f5dcbe 140 mgr->SetSkipTerminate(kTRUE);
1ee39b3a 141
d4f4bb29 142 gROOT->LoadMacro("$ALICE_ROOT/PWG1/macros/AddTrainPerformanceTRD.C");
c3ebe2ae 143 if(!AddTrainPerformanceTRD(optList)) {
d4f4bb29 144 Error("run.C", "Error loading TRD train.");
145 return;
1ee39b3a 146 }
147
148 if (!mgr->InitAnalysis()) return;
149 // verbosity
5bb83ff3 150 printf("\tRUNNING TRAIN FOR TASKS:\n");
151 TObjArray *taskList=mgr->GetTasks();
152 for(Int_t itask=0; itask<taskList->GetEntries(); itask++){
153 AliAnalysisTask *task=(AliAnalysisTask*)taskList->At(itask);
154 printf(" %s [%s]\n", task->GetName(), task->GetTitle());
155 }
b91fdd71 156 //mgr->PrintStatus();
1ee39b3a 157 mgr->StartAnalysis("local", chain, nev, first);
1ee39b3a 158 timer.Stop();
159 timer.Print();
160
1ee39b3a 161 // verbosity
5bb83ff3 162 printf("\tCLEANING TASK LIST:\n");
1ee39b3a 163 mgr->GetTasks()->Delete();
164
165 if(mcH) delete mcH;
166 delete esdH;
1ee39b3a 167 delete chain;
168 if(MEM) delete mem;
169 if(MEM) TMemStatViewerGUI::ShowGUI();
170}
171
172//____________________________________________
173TChain* MakeChainLST(const char* filename)
174{
175 // Create the chain
176 TChain* chain = new TChain("esdTree");
177
178 if(!filename){
179 chain->Add(Form("%s/AliESDs.root", gSystem->pwd()));
180 return chain;
181 }
182
183
184 // read ESD files from the input list.
185 ifstream in;
186 in.open(filename);
187 TString esdfile;
188 while(in.good()) {
189 in >> esdfile;
190 if (!esdfile.Contains("root")) continue; // protection
191 chain->Add(esdfile.Data());
192 }
193
194 in.close();
195
196 return chain;
197}
198
199//____________________________________________
200TChain* MakeChainXML(const char* xmlfile)
201{
202 if (!TFile::Open(xmlfile)) {
203 Error("MakeChainXML", Form("No file %s was found", xmlfile));
99535225 204 return NULL;
1ee39b3a 205 }
206
99535225 207 if(gSystem->Load("libNetx.so")<0) return NULL;
208 if(gSystem->Load("libRAliEn.so")<0) return NULL;
1ee39b3a 209 TGrid::Connect("alien://") ;
210
211 TGridCollection *collection = (TGridCollection*) TAlienCollection::Open(xmlfile);
212 if (!collection) {
213 Error("MakeChainXML", Form("No collection found in %s", xmlfile)) ;
99535225 214 return NULL;
1ee39b3a 215 }
216 //collection->CheckIfOnline();
217
218 TGridResult* result = collection->GetGridResult("",0 ,0);
219 if(!result->GetEntries()){
220 Error("MakeChainXML", Form("No entries found in %s", xmlfile)) ;
99535225 221 return NULL;
1ee39b3a 222 }
223 // Makes the ESD chain
224 TChain* chain = new TChain("esdTree");
225 for (Int_t idx = 0; idx < result->GetEntries(); idx++) {
226 chain->Add(result->GetKey(idx, "turl"));
227 }
228 return chain;
229}
99535225 230
231//______________________________________________________
232Bool_t UseMC(Char_t *opt){
233 return !(Bool_t)strstr(opt, "NOMC");
234}
235
236//____________________________________________
237Bool_t UseFriends(Char_t *opt){
238 return !(Bool_t)strstr(opt, "NOFR");
239}