1 // Steer TRD QA train for Reconstruction (Clusterizer, Tracking and PID).
4 // AddTrainPerformanceTRD.C(MC, friends, tasks)
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 // "MULT" : TRD single track selection
9 // "RES" : TRD tracking Resolution
10 // "CLRES": clusters Resolution
11 // "CAL" : TRD calibration
12 // "ALGN" : TRD alignment
13 // "PID" : TRD PID - pion efficiency
14 // "PIDR" : TRD PID - reference data
15 // "DET" : Basic TRD Detector checks
16 // "NOFR" : Data set does not have AliESDfriends.root
17 // "NOMC" : Data set does not have Monte Carlo Informations (real data), so all tasks which rely
18 // on MC information are switched off
21 // Don't forget to load first the libraries
22 // gSystem->Load("libMemStat.so")
23 // gSystem->Load("libMemStatGui.so")
24 // gSystem->Load("libANALYSIS.so")
25 // gSystem->Load("libANALYSISalice.so")
26 // gSystem->Load("libPWGPP.so");
29 // Alex Bercuci (A.Bercuci@gsi.de)
30 // Markus Fasel (m.Fasel@gsi.de)
32 #if ! defined (__CINT__) || defined (__MAKECINT__)
34 #include <Riostream.h>
36 #include "TStopwatch.h"
38 #include "TMemStatViewerGUI.h"
47 #include "TAlienCollection.h"
48 #include "TGridCollection.h"
49 #include "TGridResult.h"
50 #include "TGeoGlobalMagField.h"
53 #include "AliTracker.h"
55 #include "AliCDBManager.h"
56 #include "AliGRPManager.h"
57 #include "AliGeomManager.h"
58 #include "AliAnalysisManager.h"
59 #include "AliAnalysisDataContainer.h"
60 #include "AliMCEventHandler.h"
61 #include "AliESDInputHandler.h"
63 #include "TRD/AliTRDtrackerV1.h"
64 #include "TRD/AliTRDcalibDB.h"
66 #include "PWGPP/TRD/macros/AddTRDcheckESD.C"
67 #include "PWGPP/TRD/macros/AddTRDinfoGen.C"
68 #include "PWGPP/TRD/macros/AddTRDcheckDET.C"
69 #include "PWGPP/TRD/macros/AddTRDefficiency.C"
70 #include "PWGPP/TRD/macros/AddTRDresolution.C"
71 #include "PWGPP/TRD/macros/AddTRDcheckPID.C"
72 #include "PWGPP/TRD/macros/AddTRDcheckTRK.C"
73 #include "PWGPP/TRD/macros/AddTRDv0Monitor.C"
77 const Char_t* Translate(Bool_t doCheckESD=kTRUE, Bool_t doCheckDET=kTRUE, Bool_t doEffic=kTRUE, Bool_t doResolution=kTRUE, Bool_t doCheckPID=kTRUE, Bool_t doV0Monitor=kTRUE);
78 Bool_t AddTrainPerformanceTRD(Char_t *trd="ALL", const Char_t *addMacroPath = "$ALICE_ROOT/PWGPP/TRD/macros")
80 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
82 Error("AddTrainPerformanceTRD", "AliAnalysisManager not set!");
86 // TRD data containers
87 AliAnalysisDataContainer *ci[AliTRDpwgppHelper::kNOutSlots];
88 AliAnalysisDataContainer *ce[5];
90 Info("AddTrainPerformanceTRD", Form("Add Macros taken from %s", addMacroPath));
91 Info("AddTrainPerformanceTRD", Form("TRD wagons \"%s\"", trd));
92 Int_t bitmap = AliTRDpwgppHelper::ParseOptions(trd);
93 for(Int_t it=0; it<AliTRDpwgppHelper::kNTRDQATASKS; it++){
94 if(gROOT->LoadMacro(Form("%s/Add%s.C+", addMacroPath, TString(AliTRDpwgppHelper::TaskClassName(it))(3,20).Data()))) {
95 Error("AddTrainPerformanceTRD()", Form("Error loading %s task.", AliTRDpwgppHelper::TaskClassName(it)));
98 if(!AliTRDpwgppHelper::DoTask(it, bitmap)) continue;
101 case AliTRDpwgppHelper::kCheckESD:
102 AddTRDcheckESD(mgr); break;
103 case AliTRDpwgppHelper::kInfoGen:
104 AddTRDinfoGen(mgr, 0, NULL, ci); break;
105 case AliTRDpwgppHelper::kCheckDET:
107 ce[0]=ci[AliTRDpwgppHelper::kTracksBarrel];
108 ce[1]=ci[AliTRDpwgppHelper::kTracksSA];
109 ce[2]=ci[AliTRDpwgppHelper::kTracksKink];
110 ce[3]=ci[AliTRDpwgppHelper::kEventInfo];
111 ce[4]=ci[AliTRDpwgppHelper::kClusters];
112 AddTRDcheckDET(mgr, bitmap, ce);
114 case AliTRDpwgppHelper::kEfficiency:
116 ce[0]=ci[AliTRDpwgppHelper::kTracksBarrel];
117 ce[1]=ci[AliTRDpwgppHelper::kTracksITS];
118 ce[2]=ci[AliTRDpwgppHelper::kTracksKink];
119 ce[3]=ci[AliTRDpwgppHelper::kEventInfo];
120 ce[4]=ci[AliTRDpwgppHelper::kClusters];
121 AddTRDefficiency(mgr, bitmap, ce);
123 case AliTRDpwgppHelper::kResolution:
125 ce[0]=ci[AliTRDpwgppHelper::kTracksBarrel];
126 ce[1]=ci[AliTRDpwgppHelper::kTracksITS];
127 ce[2]=ci[AliTRDpwgppHelper::kTracksKink];
128 ce[3]=ci[AliTRDpwgppHelper::kEventInfo];
129 ce[4]=ci[AliTRDpwgppHelper::kClusters];
130 AddTRDresolution(mgr, bitmap, ce);
132 case AliTRDpwgppHelper::kCheckPID:
134 ce[0]=ci[AliTRDpwgppHelper::kTracksBarrel];
135 ce[1]=ci[AliTRDpwgppHelper::kEventInfo];
136 ce[2]=ci[AliTRDpwgppHelper::kV0List];
137 AddTRDcheckPID(mgr, bitmap, ce, &ce[3]);
139 case AliTRDpwgppHelper::kCheckTRK:
141 ce[0]=ci[AliTRDpwgppHelper::kTracksBarrel];
142 ce[1]=ci[AliTRDpwgppHelper::kEventInfo];
143 ce[2]=ci[AliTRDpwgppHelper::kClusters];
144 AddTRDcheckTRK(mgr, 0, ce);
146 case AliTRDpwgppHelper::kV0Monitor:
147 // slots already mapped by checkPID
148 AddTRDv0Monitor(mgr, 0, ce);
151 Warning("AddTrainPerformanceTRD()", Form("No performance task registered at slot %d.", it));
157 const Char_t* Translate(Bool_t doCheckESD, Bool_t doCheckDET, Bool_t doEffic, Bool_t doResolution, Bool_t doCheckPID, Bool_t doCheckV0)
160 if( doCheckESD==kTRUE &&
163 doResolution==kTRUE &&
169 Bool_t kINDENT(kFALSE);
175 if(kINDENT) opt.Append(" ");
180 if(kINDENT) opt.Append(" ");
185 if(kINDENT) opt.Append(" ");
190 if(kINDENT) opt.Append(" ");
195 if(kINDENT) opt.Append(" ");
201 return (const Char_t*)opt.Data();