monitoring and calibration for PbPb 2011
[u/mrichter/AliRoot.git] / PWG1 / TRD / macros / makeResults.C
1 // Usage:
2 //   makeResults.C("tasks", "file_list", ""task_id, kGRID)
3 //   tasks : "ALL" or one/more of the following separated by space:
4 //     "EFF"  : TRD Tracking Efficiency 
5 //     "EFFC" : TRD Tracking Efficiency Combined (barrel + stand alone) - only in case of simulations
6 //     "RES"  : TRD tracking Resolution
7 //     "PID"  : TRD PID - pion efficiency 
8 //     "DET"  : Basic TRD Detector checks
9 //     "NOFR" : Data set does not have AliESDfriends.root 
10 //     "NOMC" : Data set does not have Monte Carlo Informations (real data), so all tasks which rely
11 //              on MC information are switched off
12 //   file_list : is the list of the files to be processed. 
13 //     They should contain the full path. Here is an example:
14 // /lustre/alice/local/TRDdata/SIM/P-Flat/TRUNK/RUN0/TRD.Performance.root
15 // or for GRID alien:///alice/cern.ch/user/m/mfasel/MinBiasProd/results/ppMinBias80000/1/TRD.Performance.root
16 //   task_id : identifier of task speciality as defined by the AddMacro.C. 
17 //             (e.g. AddTRDresolution.C defines "" for barrel tracks, "K" for kink tracks and "SA" for stand alone tracks)
18 //   kGRID : specify if files are comming from a GRID collection [default kFALSE]
19 //
20 // HOW TO BUILD THE FILE LIST
21 //   1. locally
22 // ls -1 BaseDir/RUN*/TRD.Performance.root > files.lst
23 // 
24 //   2. on Grid
25 // char *BaseDir="/alice/cern.ch/user/m/mfasel/MinBiasProd/results/ppMinBias80000/";
26 // TGrid::Connect("alien://");
27 // TGridResult *res = gGrid->Query(BaseDir, "%/TRD.Performance.root");
28 // TGridCollection *col = gGrid->OpenCollectionQuery(res);
29 // col->Reset();
30 // TMap *map = 0x0;
31 // while(map = (TMap*)col->Next()){
32 //   TIter it((TCollection*)map);
33 //   TObjString *info = 0x0;
34 //   while(info=(TObjString*)it()){
35 //     printf("alien://%s\n", col->GetLFN(info->GetString().Data()));
36 //   }
37 // }
38 //
39 // The files which will be processed are the intersection between the
40 // condition on the tasks and the files in the file list.
41 //
42 // Authors:
43 //   Alex Bercuci (A.Bercuci@gsi.de) 
44 //   Markus Fasel (m.Fasel@gsi.de) 
45 //
46
47 #if ! defined (__CINT__) || defined (__MAKECINT__)
48 #include <fstream>
49 #include "TError.h"
50 #include <TClass.h>
51 #include <TCanvas.h>
52 #include <TH1.h>
53 #include <TGraph.h>
54 #include <TObjArray.h>
55 #include <TObjString.h>
56 #include <TString.h>
57 #include <TROOT.h>
58 #include <TStyle.h>
59 #include <TGrid.h>
60 #include <TGridResult.h>
61 #include <TGridCollection.h>
62
63 #include "AliLog.h"
64
65 #include "PWG1/TRD/AliTRDrecoTask.h"
66 #include "PWG1/TRD/AliTRDcheckESD.h"
67
68 #endif
69
70 Char_t *libs[] = {"libProofPlayer.so", "libANALYSIS.so", "libANALYSISalice.so", "libCORRFW", "libTENDER.so", "libPWG1.so"};
71 // define setup
72 TCanvas *c = 0x0;
73 Bool_t mc(kFALSE), friends(kFALSE);
74 Bool_t summary(kTRUE);
75
76 void processTRD(TNamed* task, const Char_t *filename);
77 void processESD(TNamed* task, const Char_t *filename);
78 void makeSummaryESD(const Char_t* filename="QAResults.root", Double_t* trendValues=0x0, Bool_t useCF=kFALSE, Bool_t useIsolatedBC=kFALSE, Bool_t cutTOFbc=kFALSE, const Char_t* dir="TRD_Performance", Bool_t isGrid=kFALSE);
79 void makeResults(Char_t *opt = "ALL", const Char_t *files="QAResults.root", Char_t *cid = "", Bool_t kGRID=kFALSE, Bool_t dosummary = kTRUE)
80 {
81   if(kGRID){
82     if(!gSystem->Getenv("GSHELL_ROOT")){
83       Error("makeResults.C", "AliEn not initialized.");
84       return;
85     }
86     TGrid::Connect("alien://");
87   }
88
89   // Load Libraries in interactive mode
90   Int_t nlibs = static_cast<Int_t>(sizeof(libs)/sizeof(Char_t *));
91   for(Int_t ilib=0; ilib<nlibs; ilib++){
92     if(gSystem->Load(libs[ilib]) >= 0) continue;
93     Error("makeResults.C", Form("Failed to load %s.", libs[ilib]));
94     return;
95   }
96
97   mc      = AliTRDpwg1Helper::HasReadMCData(opt);
98   friends = AliTRDpwg1Helper::HasReadFriendData(opt);
99
100   gStyle->SetOptStat(0);
101   gStyle->SetOptFit(0);
102   TString outputFile;
103   if(!TString(files).EndsWith(".root")){ 
104     outputFile = Form("%s/QAResults.root", gSystem->ExpandPathName("$PWD"));
105     AliTRDpwg1Helper::MergeProd("QAResults.root", files);
106   } else {
107     outputFile = files;
108   }
109   Int_t fSteerTask = AliTRDpwg1Helper::ParseOptions(opt);
110
111   if(!dosummary){
112     summary = kFALSE;
113     if(!c) c=new TCanvas("c", "Performance", 10, 10, 800, 500);
114   }
115
116   TClass *ctask = new TClass;
117   AliAnalysisTask *task = 0x0;
118   for(Int_t itask = AliTRDpwg1Helper::kNTRDQATASKS; itask--;){
119     if(!AliTRDpwg1Helper::DoTask(itask, fSteerTask)) continue;
120     new(ctask) TClass(AliTRDpwg1Helper::TaskClassName(itask));
121     task = (AliAnalysisTask*)ctask->New();
122     task->SetName(Form("%s%s", task->GetName(), cid));
123     printf(" *** task %s, output file %s\n", task->GetName(), outputFile.Data());
124     if(task->IsA()->InheritsFrom("AliTRDrecoTask")) processTRD(task, outputFile.Data());
125     else if(strcmp(task->IsA()->GetName(), "AliTRDcheckESD")==0) processESD(task, outputFile.Data());
126     else if(strcmp(task->IsA()->GetName(), "AliTRDinfoGen")==0) processGEN(task, outputFile.Data());
127     else{
128       Error("makeResults.C", Form("Handling of class task \"%s\" not implemented.", task->IsA()->GetName()));
129       delete task;
130     }
131   }
132   delete ctask;
133   delete c;
134 }
135
136
137 //______________________________________________________
138 void processTRD(TNamed *otask, const Char_t *filename)
139 {
140   printf("process[%s] : %s\n", otask->GetName(), otask->GetTitle());
141   Int_t debug(0);
142   AliTRDrecoTask *task = dynamic_cast<AliTRDrecoTask*>(otask);
143   task->SetDebugLevel(debug);
144   AliLog::SetClassDebugLevel(otask->IsA()->GetName(), debug);
145   task->SetMCdata(mc);
146   task->SetFriends(friends);
147
148   //if(!task->Load(Form("%s/AnalysisResults.root", gSystem->ExpandPathName("$PWD")))){
149   if(!task->Load(filename)){
150     Error("makeResults.C", Form("Load data container for task %s failed.", task->GetName()));
151     delete task;
152     return;
153   }
154
155   if(!task->PostProcess()){
156     Error("makeResults.C", Form("Processing data container for task %s failed.", task->GetName()));
157     delete task;
158     return;
159   }
160   if(summary) task->MakeSummary();
161   else{
162     for(Int_t ipic=0; ipic<task->GetNRefFigures(); ipic++){
163       c->Clear();
164       if(!task->GetRefFigure(ipic)) continue;
165       c->SaveAs(Form("%s_Fig%02d.gif", task->GetName(), ipic), "gif");
166     }
167   }
168   delete task;
169 }
170
171 //______________________________________________________
172 void processESD(TNamed *otask, const Char_t *filename)
173 {
174   printf("process[%s] : %s\n", otask->GetName(), otask->GetTitle());
175
176   AliTRDcheckESD *esd = dynamic_cast<AliTRDcheckESD*>(otask);
177   if(!esd){
178     Info("makeResults.C", Form("Processing of task %s failed.", otask->GetName()));
179     delete otask;
180     return;
181   }
182   //if(!esd->Load(Form("%s/AnalysisResults.root", gSystem->ExpandPathName("$PWD")), "TRD_Performance")){
183   if(!esd->Load(filename, "TRD_Performance")){
184     Error("makeResults.C", Form("Load data container for task %s failed.", esd->GetName()));
185     delete esd;
186     return;
187   }
188   esd->Terminate(NULL);
189   
190   if(summary) esd->MakeSummary();
191   else{
192     for(Int_t ipic(0); ipic<esd->GetNRefFigures(); ipic++){
193       c->Clear(); 
194       if(!esd->GetRefFigure(ipic)) continue;
195       c->SaveAs(Form("%s_Fig%02d.gif", esd->GetName(), ipic));
196     }
197   }
198   delete esd;
199 }
200
201 //______________________________________________________
202 void processGEN(TNamed *otask, const Char_t *filename)
203 {
204   printf("process[%s] : %s\n", otask->GetName(), otask->GetTitle());
205
206   AliTRDinfoGen *info = dynamic_cast<AliTRDinfoGen*>(otask);
207
208   if(!info->Load(filename, "TRD_Performance")){
209     Error("makeResults.C", Form("Load data container for task %s failed.", info->GetName()));
210     delete info;
211     return;
212   }
213   if(!summary){
214     for(Int_t ipic(0); ipic<info->GetNRefFigures(); ipic++){
215       c->Clear(); 
216       if(!info->GetRefFigure(ipic)) continue;
217       c->SaveAs(Form("%s_Fig%02d.gif", info->GetName(), ipic));
218     }
219   }
220
221   delete info;
222 }
223
224 //______________________________________________________
225 void makeSummaryESD(const Char_t* filename, Double_t* trendValues, Bool_t useCF, Bool_t useIsolatedBC, Bool_t cutTOFbc, const Char_t* dir, Bool_t isGrid) {
226   //
227   //  Make the summary picture and get trending variables from the ESD task
228   //
229   if(isGrid){
230     if(!gSystem->Getenv("GSHELL_ROOT")){
231       Error("makeResults.C", "AliEn not initialized.");
232       return;
233     }
234     TGrid::Connect("alien://");
235   }
236   
237   // Load Libraries in interactive mode
238   Int_t nlibs = static_cast<Int_t>(sizeof(libs)/sizeof(Char_t *));
239   for(Int_t ilib=0; ilib<nlibs; ilib++){
240     if(gSystem->Load(libs[ilib]) >= 0) continue;
241     Error("makeResults.C", Form("Failed to load %s.", libs[ilib]));
242     return;
243   }
244   
245   AliTRDcheckESD *esd=new AliTRDcheckESD();
246   if(!esd->Load(filename,dir)) return;
247   //esd->Terminate();
248   if(useCF) esd->MakeSummaryFromCF(trendValues, useIsolatedBC, cutTOFbc);
249   else esd->MakeSummary(trendValues);
250 }