1 // Steer TRD QA train for Reconstruction (Clusterizer, Tracking and PID).
4 // run.C(optList, files, nev, first, runNo, ocdb_uri, grp_uri)
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),
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
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");
42 // Alex Bercuci (A.Bercuci@gsi.de)
43 // Markus Fasel (m.Fasel@gsi.de)
45 #if ! defined (__CINT__) || defined (__MAKECINT__)
47 #include <Riostream.h>
49 #include "TStopwatch.h"
51 #include "TMemStatViewerGUI.h"
59 #include "TAlienCollection.h"
60 #include "TGridCollection.h"
61 #include "TGridResult.h"
62 #include "TGeoGlobalMagField.h"
65 #include "AliTracker.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"
75 #include "TRD/AliTRDtrackerV1.h"
76 #include "TRD/AliTRDcalibDB.h"
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"
88 #include "macros/AliTRDperformanceTrain.h"
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, Int_t runNo=0, const Char_t *ocdb_uri="local://$ALICE_ROOT/OCDB", const Char_t *grp_uri=Form("local://%s", gSystem->pwd()))
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");
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;
115 Bool_t fHasMCdata = HasReadMCData(optList);
116 Bool_t fHasFriends = HasReadFriendData(optList);
118 // INITIALIZATION OF RUNNING ENVIRONMENT
119 // initialize OCDB manager
120 // AliCDBManager *cdbManager = AliCDBManager::Instance();
121 // cdbManager->SetDefaultStorage(ocdb_uri);
122 // if(!cdbManager->IsDefaultStorageSet()){
123 // Error("run.C", "Error setting OCDB.");
126 // cdbManager->SetRun(runNo);
127 // cdbManager->SetSpecificStorage("GRP/GRP/Data", grp_uri);
128 // cdbManager->SetCacheFlag(kFALSE);
129 // cdbManager->Print();
130 // // initialize magnetic field from the GRP manager.
131 // AliGRPManager grpMan;
132 // if(!grpMan.ReadGRPEntry()) return;
133 // if(!grpMan.SetMagField()) return;
134 // //AliRunInfo *runInfo = grpMan.GetRunInfo();
135 // AliGeomManager::LoadGeometry();
139 if(!files) chain = MakeChainLST();
142 if(fn.EndsWith("xml")) chain = MakeChainXML(files);
143 else chain = MakeChainLST(files);
147 chain->GetListOfFiles()->Print();
148 Int_t nfound=(Int_t)chain->GetEntries();
149 printf("\tENTRIES FOUND [%d] REQUESTED [%d]\n", nfound, nev>nfound?nfound:nev);
152 // BUILD ANALYSIS MANAGER
153 AliAnalysisManager *mgr = new AliAnalysisManager("TRD Reconstruction Performance & Calibration");
154 AliESDInputHandlerRP *esdH(NULL);
155 mgr->SetInputEventHandler(esdH = new AliESDInputHandlerRP);
156 esdH->SetReadFriends(kTRUE);
157 esdH->SetActiveBranches("ESDfriend");
158 AliMCEventHandler *mcH(NULL);
159 if(fHasMCdata) mgr->SetMCtruthEventHandler(mcH = new AliMCEventHandler());
160 //mgr->SetDebugLevel(10);
161 mgr->SetSkipTerminate(kTRUE);
163 gROOT->LoadMacro("$ALICE_ROOT/PWG1/macros/AddTrainPerformanceTRD.C");
164 if(!AddTrainPerformanceTRD(optList)) {
165 Error("run.C", "Error loading TRD train.");
169 if (!mgr->InitAnalysis()) return;
171 printf("\tRUNNING TRAIN FOR TASKS:\n");
172 TObjArray *taskList=mgr->GetTasks();
173 for(Int_t itask=0; itask<taskList->GetEntries(); itask++){
174 AliAnalysisTask *task=(AliAnalysisTask*)taskList->At(itask);
175 printf(" %s [%s]\n", task->GetName(), task->GetTitle());
177 //mgr->PrintStatus();
178 mgr->StartAnalysis("local", chain, nev, first);
183 printf("\tCLEANING TASK LIST:\n");
184 mgr->GetTasks()->Delete();
190 if(MEM) TMemStatViewerGUI::ShowGUI();
193 //____________________________________________
194 TChain* MakeChainLST(const char* filename)
197 TChain* chain = new TChain("esdTree");
200 chain->Add(Form("%s/AliESDs.root", gSystem->pwd()));
205 // read ESD files from the input list.
211 if (!esdfile.Contains("root")) continue; // protection
212 chain->Add(esdfile.Data());
220 //____________________________________________
221 TChain* MakeChainXML(const char* xmlfile)
223 if (!TFile::Open(xmlfile)) {
224 Error("MakeChainXML", Form("No file %s was found", xmlfile));
228 if(gSystem->Load("libNetx.so")<0) return 0x0;
229 if(gSystem->Load("libRAliEn.so")<0) return 0x0;
230 TGrid::Connect("alien://") ;
232 TGridCollection *collection = (TGridCollection*) TAlienCollection::Open(xmlfile);
234 Error("MakeChainXML", Form("No collection found in %s", xmlfile)) ;
237 //collection->CheckIfOnline();
239 TGridResult* result = collection->GetGridResult("",0 ,0);
240 if(!result->GetEntries()){
241 Error("MakeChainXML", Form("No entries found in %s", xmlfile)) ;
244 // Makes the ESD chain
245 TChain* chain = new TChain("esdTree");
246 for (Int_t idx = 0; idx < result->GetEntries(); idx++) {
247 chain->Add(result->GetKey(idx, "turl"));