8944b55fd29ab4d69e9bd3095eab2fa445f55ffd
[u/mrichter/AliRoot.git] / PWG1 / macros / AddTrainPerformanceTRD.C
1 // Steer TRD QA train for Reconstruction (Clusterizer, Tracking and PID).\r
2 // \r
3 // Usage:\r
4 //   AddTrainPerformanceTRD.C(MC, friends, tasks)\r
5 //   tasks : "ALL" or one/more of the following:\r
6 //     "EFF"  : TRD Tracking Efficiency \r
7 //     "EFFC" : TRD Tracking Efficiency Combined (barrel + stand alone) - only in case of simulations\r
8 //     "MULT"  : TRD single track selection\r
9 //     "RES"  : TRD tracking Resolution\r
10 //     "CLRES": clusters Resolution\r
11 //     "CAL"  : TRD calibration\r
12 //     "ALGN" : TRD alignment\r
13 //     "PID"  : TRD PID - pion efficiency \r
14 //     "PIDR" : TRD PID - reference data\r
15 //     "DET"  : Basic TRD Detector checks\r
16 //     "NOFR" : Data set does not have AliESDfriends.root \r
17 //     "NOMC" : Data set does not have Monte Carlo Informations (real data), so all tasks which rely\r
18 //              on MC information are switched off\r
19 //\r
20 // In compiled mode : \r
21 // Don't forget to load first the libraries\r
22 // gSystem->Load("libMemStat.so")\r
23 // gSystem->Load("libMemStatGui.so")\r
24 // gSystem->Load("libANALYSIS.so")\r
25 // gSystem->Load("libANALYSISalice.so")\r
26 // gSystem->Load("libPWG1.so");\r
27 //\r
28 // Authors:\r
29 //   Alex Bercuci (A.Bercuci@gsi.de) \r
30 //   Markus Fasel (m.Fasel@gsi.de) \r
31 \r
32 #if ! defined (__CINT__) || defined (__MAKECINT__)\r
33 //#ifndef __CINT__\r
34 #include <Riostream.h>\r
35 \r
36 #include "TStopwatch.h"\r
37 #include "TMemStat.h"\r
38 #include "TMemStatViewerGUI.h"\r
39 \r
40 #include "TROOT.h"\r
41 #include "TClass.h"\r
42 #include "TSystem.h"\r
43 #include "TString.h"\r
44 #include "TError.h"\r
45 #include "TChain.h"\r
46 #include "TGrid.h"\r
47 #include "TAlienCollection.h"\r
48 #include "TGridCollection.h"\r
49 #include "TGridResult.h"\r
50 #include "TGeoGlobalMagField.h"\r
51 \r
52 #include "AliMagF.h"\r
53 #include "AliTracker.h"\r
54 #include "AliLog.h"\r
55 #include "AliCDBManager.h"\r
56 #include "AliGRPManager.h"\r
57 #include "AliGeomManager.h"\r
58 #include "AliAnalysisManager.h"\r
59 #include "AliAnalysisDataContainer.h"\r
60 #include "AliMCEventHandler.h"\r
61 #include "AliESDInputHandler.h"\r
62 \r
63 #include "TRD/AliTRDtrackerV1.h"\r
64 #include "TRD/AliTRDcalibDB.h"\r
65 \r
66 #include "PWG1/TRD/macros/AddTRDcheckESD.C"\r
67 #include "PWG1/TRD/macros/AddTRDinfoGen.C"\r
68 #include "PWG1/TRD/macros/AddTRDcheckDET.C"\r
69 #include "PWG1/TRD/macros/AddTRDefficiency.C"\r
70 #include "PWG1/TRD/macros/AddTRDresolution.C"\r
71 #include "PWG1/TRD/macros/AddTRDcheckPID.C"\r
72 #include "PWG1/TRD/macros/AddTRDcheckTRK.C"\r
73 #include "PWG1/TRD/macros/AddTRDv0Monitor.C"\r
74 #endif\r
75 \r
76 TString opt("");\r
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);\r
78 Bool_t AddTrainPerformanceTRD(Char_t *trd="ALL", const Char_t *addMacroPath = "$ALICE_ROOT/PWG1/TRD/macros")\r
79 {\r
80   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
81   if(!mgr) { \r
82     Error("AddTrainPerformanceTRD", "AliAnalysisManager not set!");\r
83     return kFALSE;\r
84   }\r
85 \r
86   // TRD data containers\r
87   AliAnalysisDataContainer *ci[AliTRDpwg1Helper::kNOutSlots];\r
88   AliAnalysisDataContainer *ce[5];\r
89 \r
90   Info("AddTrainPerformanceTRD", Form("Add Macros taken from %s", addMacroPath));\r
91   Info("AddTrainPerformanceTRD", Form("TRD wagons \"%s\"", trd));\r
92   Int_t bitmap = AliTRDpwg1Helper::ParseOptions(trd);\r
93   for(Int_t it=0; it<AliTRDpwg1Helper::kNTRDQATASKS; it++){\r
94     if(gROOT->LoadMacro(Form("%s/Add%s.C+", addMacroPath, TString(AliTRDpwg1Helper::TaskClassName(it))(3,20).Data()))) {\r
95       Error("AddTrainPerformanceTRD()", Form("Error loading %s task.", AliTRDpwg1Helper::TaskClassName(it)));\r
96       return kFALSE;\r
97     } \r
98     if(!AliTRDpwg1Helper::DoTask(it, bitmap)) continue;\r
99 \r
100     switch(it){\r
101     case AliTRDpwg1Helper::kCheckESD:\r
102       AddTRDcheckESD(mgr); break;\r
103     case AliTRDpwg1Helper::kInfoGen:\r
104       AddTRDinfoGen(mgr, 0, NULL, ci); break;\r
105     case AliTRDpwg1Helper::kCheckDET:\r
106       // map slots\r
107       ce[0]=ci[AliTRDpwg1Helper::kTracksBarrel];\r
108       ce[1]=ci[AliTRDpwg1Helper::kTracksSA];\r
109       ce[2]=ci[AliTRDpwg1Helper::kTracksKink];\r
110       ce[3]=ci[AliTRDpwg1Helper::kEventInfo];\r
111       AddTRDcheckDET(mgr, bitmap, ce);\r
112       break;\r
113     case AliTRDpwg1Helper::kEfficiency:\r
114       // map slots\r
115       ce[0]=ci[AliTRDpwg1Helper::kTracksBarrel];\r
116       ce[1]=ci[AliTRDpwg1Helper::kTracksSA];\r
117       ce[2]=ci[AliTRDpwg1Helper::kTracksKink];\r
118       ce[3]=ci[AliTRDpwg1Helper::kEventInfo];\r
119       AddTRDefficiency(mgr, bitmap, ce);\r
120       break;\r
121     case AliTRDpwg1Helper::kResolution:\r
122       // map slots\r
123       ce[0]=ci[AliTRDpwg1Helper::kTracksBarrel];\r
124       ce[1]=ci[AliTRDpwg1Helper::kTracksSA];\r
125       ce[2]=ci[AliTRDpwg1Helper::kTracksKink];\r
126       ce[3]=ci[AliTRDpwg1Helper::kEventInfo];\r
127       AddTRDresolution(mgr, bitmap, ce); \r
128       break;\r
129     case AliTRDpwg1Helper::kCheckPID:\r
130       // map slots\r
131       ce[0]=ci[AliTRDpwg1Helper::kTracksBarrel];\r
132       ce[1]=ci[AliTRDpwg1Helper::kEventInfo];\r
133       ce[2]=ci[AliTRDpwg1Helper::kV0List];\r
134       AddTRDcheckPID(mgr, bitmap, ce, &ce[3]);\r
135       break;\r
136     case AliTRDpwg1Helper::kCheckTRK:\r
137       // map slots\r
138       ce[0]=ci[AliTRDpwg1Helper::kTracksBarrel];\r
139       ce[1]=ci[AliTRDpwg1Helper::kEventInfo];\r
140       AddTRDcheckTRK(mgr, 0, ce);\r
141       break;\r
142     case AliTRDpwg1Helper::kV0Monitor:\r
143       // slots already mapped by checkPID\r
144       AddTRDv0Monitor(mgr, 0, ce);\r
145       break;\r
146     default:\r
147       Warning("AddTrainPerformanceTRD()", Form("No performance task registered at slot %d.", it)); \r
148     }\r
149   }\r
150   return kTRUE;\r
151 }\r
152 \r
153 const Char_t* Translate(Bool_t doCheckESD, Bool_t doCheckDET, Bool_t doEffic, Bool_t doResolution, Bool_t doCheckPID, Bool_t doCheckV0)\r
154 {\r
155   opt.Clear();\r
156   if( doCheckESD==kTRUE &&\r
157       doCheckDET==kTRUE &&\r
158       doEffic==kTRUE &&\r
159       doResolution==kTRUE &&\r
160       doCheckPID==kTRUE &&\r
161       doCheckV0==kTRUE\r
162   ){\r
163     opt="ALL";\r
164   } else {\r
165     Bool_t kINDENT(kFALSE);\r
166     if(doCheckESD){ \r
167       opt.Append("ESD");\r
168       kINDENT=kTRUE;\r
169     }\r
170     if(doCheckDET){ \r
171       if(kINDENT) opt.Append(" ");\r
172       opt.Append("DET"); \r
173       kINDENT = kTRUE;\r
174     }\r
175     if(doEffic){ \r
176       if(kINDENT) opt.Append(" ");\r
177       opt.Append("EFF");\r
178       kINDENT=kTRUE;\r
179     }\r
180     if(doResolution){ \r
181       if(kINDENT) opt.Append(" ");\r
182       opt.Append("RES");\r
183       kINDENT=kTRUE;\r
184     }\r
185     if(doCheckPID){ \r
186       if(kINDENT) opt.Append(" ");\r
187       opt.Append("PID");\r
188       kINDENT=kTRUE;\r
189     }\r
190     if(doCheckV0){ \r
191       if(kINDENT) opt.Append(" ");\r
192       opt.Append("V0");\r
193       kINDENT=kTRUE;\r
194     }\r
195   }\r
196 \r
197   return (const Char_t*)opt.Data();\r
198 }\r
199 \r
200 \r