962352a72edb8c5eeb0a2bf419417e85650e969e
[u/mrichter/AliRoot.git] / TRD / qaRec / makeResults.C
1 #if ! defined (__CINT__) || defined (__MAKECINT__)
2 #include <TClass.h>
3 #include <TFileMerger.h>
4 #include <TPad.h>
5 #include <TH1.h>
6 #include <TGraph.h>
7 #include <TObjArray.h>
8 #include <TObjString.h>
9 #include <TPython.h>
10 #include <TString.h>
11 #include <TROOT.h>
12 #include <TSystem.h>
13
14 #include "qaRec/AliTRDrecoTask.h"
15
16 // #include <string>
17 // #include <iostream>
18 // #include <cstdio>
19
20 #endif
21
22 #include "run.h"
23
24 Char_t *libs[] = {"libProofPlayer.so", "libANALYSIS.so", "libTRDqaRec.so", "libPyROOT"};
25
26 void makeResults(Char_t* dir=0x0, Char_t *tasks = "ALL")
27 {
28         // Load Libraries in interactive mode
29 //#ifndef __CINT__
30   Int_t nlibs = static_cast<Int_t>(sizeof(libs)/sizeof(Char_t *));
31   for(Int_t ilib=0; ilib<nlibs; ilib++){
32     if(!gSystem->Load(libs[ilib])) continue;
33     printf("Failed to load %s.\n", libs[ilib]);
34     return;
35   }
36   //gSystem->Setenv("PYTHONPATH", "$ROOTSYS/lib:$PYTHONPATH");
37 //#endif
38
39
40   //printf("dir[%s] tasks[%s]\n", dir, tasks);
41   gStyle->SetOptStat(0);
42   Bool_t mc      = kTRUE;
43   Bool_t friends = kTRUE;
44
45   // select tasks to process; we should move it to an 
46   // individual function and move the task identifiers 
47   // outside the const space
48   TObjArray *tasksArray = TString(tasks).Tokenize(" ");
49   Int_t fSteerTask = 0;
50   for(Int_t isel = 0; isel < tasksArray->GetEntriesFast(); isel++){
51     TString s = (dynamic_cast<TObjString *>(tasksArray->UncheckedAt(isel)))->String();
52     if(s.CompareTo("ALL") == 0){
53       for(Int_t itask = 1; itask < fknTasks; itask++) SETBIT(fSteerTask, itask);
54       continue;
55     } else if(s.CompareTo("EFF") == 0){
56       SETBIT(fSteerTask, kTrackingEfficiency);
57       continue;
58     } else if(s.CompareTo("EFFC") == 0){
59       SETBIT(fSteerTask, kTrackingCombinedEfficiency);
60       continue;
61     } else if(s.CompareTo("RES") == 0){
62       SETBIT(fSteerTask, kTrackingResolution);
63       continue;
64     } else if(s.CompareTo("CAL" ) == 0){
65       SETBIT(fSteerTask, kCalibration);
66       continue;
67     } else if(s.CompareTo("PID" ) == 0){
68       SETBIT(fSteerTask, kPIDChecker);
69       continue;
70     } else if(s.CompareTo("DET" ) == 0){
71         SETBIT(fSteerTask, kCheckDetector);
72         continue;
73     } else{
74       std::cout << "E-makeResults.C: Task "<< s.Data() << " %s not implemented (yet).\n";
75       continue;
76     }
77   }
78
79   // catch the list of files using the ROOT Python Interface
80   TPython *pyshell = new TPython();
81   pyshell->Exec("import commands");
82
83   // file merger object
84   TFileMerger *fFM = new TFileMerger();
85   TClass *ctask = 0x0;
86   TObject *o = 0x0;
87   TObjArray *fContainer = 0x0;
88   AliTRDrecoTask *task = 0x0;
89
90   printf("\n\tPROCESSING DATA FOR TASKS [%d]:\n", fSteerTask);
91   for(Int_t itask = 1; itask <fknTasks; itask++){
92     if(!TESTBIT(fSteerTask, itask)) continue;
93
94     ctask = new TClass(fTaskClass[itask]);
95     task = (AliTRDrecoTask*)ctask->New();
96     task->SetDebugLevel(2);
97     task->SetMCdata(mc);
98     task->SetFriends(friends);
99     printf("\t%s [%s]\n", task->GetTitle(), task->GetName());
100
101      // setup filelist
102     TString pathname(dir ? "." : gSystem->ExpandPathName("$PWD"));
103     TString filestring((const Char_t*) pyshell->Eval(Form("commands.getstatusoutput(\"find %s | grep TRD.Task%s.root\")[1]", pathname.Data(), task->GetName())));
104     TObjArray *filenames = filestring.Tokenize("\n");
105     Int_t nFiles = filenames->GetEntriesFast();
106     if(!nFiles){
107       printf("No Files found for Task %s\n", task->GetName());
108       delete task;
109       delete ctask;
110       continue;
111     }
112
113     if(nFiles>1){
114       fFM = new(fFM) TFileMerger(kTRUE);
115       fFM->OutputFile(Form("merge/TRD.Task%s.root",  task->GetName()));
116       for(Int_t ifile = 0; ifile < nFiles; ifile++){
117         TString filename = (dynamic_cast<TObjString *>(filenames->UncheckedAt(ifile)))->String();
118         if(filename.Contains("merge")) continue;
119         printf("\tProcessing %s ...\n", filename.Data());
120         fFM->AddFile(filename.Data());
121       }
122       fFM->Merge();
123       fFM->~TFileMerger();
124       task->Load(Form("%s/merge/TRD.Task%s.root",gSystem->ExpandPathName("$PWD"), task->GetName()));
125     } else task->Load((dynamic_cast<TObjString *>(filenames->UncheckedAt(0)))->String().Data());
126
127
128     if(!(fContainer = task->Container())) {
129       delete task;
130       delete ctask;
131       continue;
132     } 
133     
134     task->PostProcess();
135     for(Int_t ipic=0; ipic<task->GetNRefFigures(); ipic++){
136       TCanvas *c = new TCanvas("c", "", 500, 500);
137       Int_t ifirst, ilast;
138       TH1 *h = 0x0; TGraph *g = 0x0;
139       task->GetRefFigure(ipic, ifirst, ilast);
140       if(!(o = fContainer->At(ifirst))) continue;
141       
142       if(o->InheritsFrom("TH1")){ 
143         h = dynamic_cast<TH1*>(o);
144         h->Draw("pl");
145       } else if(o->InheritsFrom("TGraph")){ 
146         g = dynamic_cast<TGraph*>(o);
147         g->Draw("apl");
148       } else{
149         printf("No idea how to plot object of type %s.\n", o->IsA()->GetName());
150         printf("Please teach me.\n");
151         continue;
152       }
153
154       for(Int_t ig=ifirst+1; ig<ilast; ig++){
155         if(!(o = fContainer->At(ig))) continue;
156         if(o->InheritsFrom("TH1")){
157           h = dynamic_cast<TH1*>(o);
158           h->Draw("plsame");
159         } else if(o->InheritsFrom("TGraph")){
160           g = dynamic_cast<TGraph*>(o);
161           g->Draw("pl");
162         }
163       }
164       c->SaveAs(Form("%s_fig%d.gif", task->GetName(), ipic));
165       delete c;
166     }
167     delete task;
168     delete ctask;
169   }
170   delete pyshell;
171 }
172