extend user flexibility / fault tolerance
[u/mrichter/AliRoot.git] / TRD / qaRec / macros / makeResults.C
1 // Usage:
2 //   makeResults.C("tasks", "file_list", 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 //     "CAL"  : TRD calibration
8 //     "PID"  : TRD PID - pion efficiency 
9 //     "PIDR" : TRD PID - reference data
10 //     "DET"  : Basic TRD Detector checks
11 //     "NOFR" : Data set does not have AliESDfriends.root 
12 //     "NOMC" : Data set does not have Monte Carlo Informations (real data), so all tasks which rely
13 //              on MC information are switched off
14 //   file_list : is the list of the files to be processed. 
15 //     They should contain the full path. Here is an example:
16 // /lustre/alice/local/TRDdata/SIM/P-Flat/TRUNK/RUN0/TRD.Performance.root
17 // or for GRID alien:///alice/cern.ch/user/m/mfasel/MinBiasProd/results/ppMinBias80000/1/TRD.Performance.root
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.Task%.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 <TFileMerger.h>
52 #include <TCanvas.h>
53 #include <TH1.h>
54 #include <TGraph.h>
55 #include <TObjArray.h>
56 #include <TObjString.h>
57 #include <TString.h>
58 #include <TROOT.h>
59 #include <TSystem.h>
60 #include <TStyle.h>
61
62 #include "qaRec/AliTRDrecoTask.h"
63
64 #endif
65
66 #include "AliTRDperformanceTrain.h"
67 //#include "../../PWG1/macros/AddPerformanceTask.h"
68
69 Char_t *libs[] = {"libProofPlayer.so", "libANALYSIS.so", "libTRDqaRec.so"};
70 // define setup
71 gStyle->SetOptStat(0);
72 Bool_t mc      = kTRUE;
73 Bool_t friends = kTRUE;
74 TCanvas *c = 0x0;
75 void mergeProd(const Char_t *mark="TRD.Performance.root", const Char_t *files=0);
76 void processTRD(TNamed* task);
77 void processAliTask(TNamed* task);
78 void makeResults(Char_t *opt = "ALL", const Char_t *files=0x0, Bool_t kGRID=kFALSE)
79 {
80   if(kGRID){
81     if(!gSystem->Getenv("GSHELL_ROOT")){
82       Error("makeResults.C", "AliEn not initialized.");
83       return;
84     }
85     TGrid::Connect("alien://");
86   }
87
88         // Load Libraries in interactive mode
89   Int_t nlibs = static_cast<Int_t>(sizeof(libs)/sizeof(Char_t *));
90   for(Int_t ilib=0; ilib<nlibs; ilib++){
91     if(gSystem->Load(libs[ilib]) >= 0) continue;
92     Error("makeResults.C", Form("Failed to load %s.", libs[ilib]));
93     return;
94   }
95
96   if(files) mergeProd("TRD.Performance.root", files);
97   Int_t fSteerTask = ParseOptions(opt);
98
99   if(!c) c=new TCanvas("c", "Performance", 10, 10, 800, 500);
100
101   TClass *ctask = new TClass;
102   AliAnalysisTask *task = 0x0;
103   AliTRDrecoTask  *trd  = 0x0;
104   for(Int_t itask = NTRDQATASKS; itask--;){
105     if(!TSTBIT(fSteerTask, itask)) continue;
106     new(ctask) TClass(fgkTRDtaskClassName[itask]);
107     task = (AliAnalysisTask*)ctask->New();
108
109     if(task->IsA()->InheritsFrom("AliTRDrecoTask")) processTRD(task);
110     else processAliTask(task);
111   }
112   delete ctask;
113   delete c;
114 }
115
116 //______________________________________________________
117 void mergeProd(const Char_t *mark, const Char_t *files)
118 {
119   const Int_t kBatch = 20;
120
121   TFileMerger *fFM = new TFileMerger(1);
122   fFM->OutputFile(Form("%s/0_%s",  gSystem->ExpandPathName("$PWD"), mark));
123
124   Int_t jbatch = 0, nbatch = 0;
125   string filename;
126   ifstream file(files);
127   while(getline(file, filename)){
128     if(Int_t(filename.find(mark)) < 0) continue;
129     fFM->AddFile(filename.c_str()); nbatch++;
130     if(nbatch==kBatch){
131       //printf("MERGE BATCH %d [%d]\n", jbatch, nbatch);
132       fFM->Merge(); jbatch++;
133       new(fFM) TFileMerger(kTRUE);
134       fFM->OutputFile(Form("%s/%d_%s",  gSystem->ExpandPathName("$PWD"), jbatch, mark));
135       nbatch=0;
136     }
137   }
138   if(nbatch){
139     //printf("MERGE BATCH %d[%d]\n", jbatch, nbatch);
140     fFM->Merge();
141     jbatch++;
142   }
143   if(!jbatch){
144     delete fFM;
145     return;
146   }
147
148   new(fFM) TFileMerger(kTRUE);
149   fFM->OutputFile(Form("%s/%s",  gSystem->ExpandPathName("$PWD"), mark));
150   for(Int_t ib=jbatch; ib--;){
151     fFM->AddFile(Form("%s/%d_%s",  gSystem->ExpandPathName("$PWD"), ib, mark));
152     gSystem->Exec(Form("rm -f %s/%d_%s", gSystem->ExpandPathName("$PWD"), ib, mark));
153   }
154   fFM->Merge();
155   delete fFM;
156 }
157
158 //______________________________________________________
159 void processTRD(TNamed *otask)
160 {
161   AliTRDrecoTask *task = dynamic_cast<AliTRDrecoTask*>(otask);
162   task->SetDebugLevel(0);
163   task->SetMCdata(mc);
164   task->SetFriends(friends);
165
166   if(!task->Load(Form("%s/TRD.Performance.root", gSystem->ExpandPathName("$PWD")))){
167     Error("makeResults.C", Form("Load data container for task %s failed.", task->GetName()));
168     delete task;
169     return;
170   }
171
172   if(!task->PostProcess()){
173     Error("makeResults.C", Form("Processing data container for task %s failed.", task->GetName()));
174     delete task;
175     return;
176   }
177   for(Int_t ipic=0; ipic<task->GetNRefFigures(); ipic++){
178     c->Clear();
179     if(!task->GetRefFigure(ipic)) continue;
180     c->SaveAs(Form("%s_Fig%02d.gif", task->GetName(), ipic));
181   }
182   delete task;
183 }
184
185 //______________________________________________________
186 void processAliTask(TNamed *otask)
187 {
188   AliAnalysisTask *task = dynamic_cast<AliAnalysisTask*>(otask);
189   Info("makeResults.C", Form("Processing of task %s not implemented yet.", task->GetName()));
190   delete task;
191 }