Renames and new scripts
[u/mrichter/AliRoot.git] / PWG2 / FORWARD / analysis2 / scripts / CompareCorrs.C
1 /**
2  * @file   CompareCorrs.C
3  * @author Christian Holm Christensen <cholm@dalsgaard.hehi.nbi.dk>
4  * @date   Fri Jan 28 23:01:59 2011
5  * 
6  * @brief  Utilities for comparing correction objects 
7  * 
8  * 
9  */
10 #ifndef __CINT__
11 #include <TCanvas.h>
12 #include <TFile.h>
13 #include <TLatex.h>
14 #include <TStyle.h>
15 #include <TMath.h>
16 #include "AliForwardCorrectionManager.h"
17 #endif
18
19
20 //======================================================================
21 struct Canvas 
22 {
23   //____________________________________________________________________
24   Canvas(const char* name, const char* title,
25          const char* n1,   const char* n2)
26     : fName(name),
27       fTitle(title),
28       fN1(n1), 
29       fN2(n2),
30       fCanvas(0),
31       fBody(0)
32   {
33     gStyle->SetPalette(1);
34     gStyle->SetTitleX(.10);
35     gStyle->SetTitleY(.99);
36     gStyle->SetTitleW(.85);
37     gStyle->SetTitleH(.085);
38     gStyle->SetTitleFillColor(0);
39     gStyle->SetTitleBorderSize(0);
40   }
41   //____________________________________________________________________
42   void Open() 
43   {
44     fCanvas = new TCanvas(fName, fTitle, 800, TMath::Sqrt(2)*800);
45     fCanvas->SetFillColor(0);
46   
47     fCanvas->Print("comparison.pdf[", "pdf");
48   }
49   //____________________________________________________________________
50   TPad* 
51   Clear(UShort_t nPad, UShort_t d, Char_t r)
52   {
53     fCanvas->Clear();
54     TPad* top = new TPad("top", "Top", 0, .95, 1, 1, 0, 0);
55     top->Draw();
56     top->cd();
57
58     TLatex* l = new TLatex(.5, .5, Form("%s for FMD%d%c (%s / %s)", 
59                                         fTitle, d, r, fN1, fN2));
60     l->SetNDC();
61     l->SetTextAlign(22);
62     l->SetTextSize(0.3);
63     l->Draw();
64   
65     fCanvas->cd();
66     fBody = new TPad("body", "Body", 0, 0, 1, .95, 0, 0);
67     fBody->SetTopMargin(0.05);
68     fBody->SetRightMargin(0.05);
69     fBody->Divide(2, (nPad+1)/2, 0.001, 0.001);
70     fBody->Draw();
71     
72     return fBody;
73   }  
74   //____________________________________________________________________
75   TVirtualPad* cd(Int_t i) 
76   {
77     if (!fBody) return 0;
78     
79     return fBody->cd(i);
80   }
81   //____________________________________________________________________
82   void Print(UShort_t d, Char_t r, const char* extra="")
83   {
84     fCanvas->Print("comparison.pdf", 
85                    Form("Title:FMD%d%c %s", d, r, extra));
86   }
87   //____________________________________________________________________
88   void Close()
89   {
90     fCanvas->Print("comparison.pdf]", "pdf");
91   }    
92   //____________________________________________________________________
93   const char* fName;
94   const char* fTitle;
95   const char* fN1;
96   const char* fN2;
97   TCanvas*    fCanvas;
98   TPad*       fBody;
99 };
100
101 //======================================================================
102 void
103 GetObjects(UShort_t    what, 
104            const char* fn1, const char* fn2, 
105            TObject*&   o1,  TObject*&   o2)
106 {
107   // --- Open files --------------------------------------------------
108   TFile* file1 = TFile::Open(fn1, "READ");
109   TFile* file2 = TFile::Open(fn2, "READ");
110
111   if (!file1) { 
112     Error("CompareSecMaps", "File %s cannot be opened", fn1);
113     return;
114   }
115
116   if (!file2) { 
117     Error("CompareSecMaps", "File %s cannot be opened", fn2);
118     return;
119   }
120   
121   // --- Find Objects ------------------------------------------------
122   AliForwardCorrectionManager::ECorrection ewhat = what;
123   // (AliForwardCorrectionManager::ECorrection)what;
124   const char* objName = 
125     AliForwardCorrectionManager::Instance().GetObjectName(ewhat);
126   
127   o1 = file1->Get(objName);
128   o2 = file2->Get(objName);
129   
130   if (!o1) {
131     Error("CompareSecMaps", "File %s does not contain an object named %s", 
132           fn1, objName);
133     return;
134   }
135   if (!o2) {
136     Error("CompareSecMaps", "File %s does not contain an object named %s", 
137           fn2, objName);
138     return;
139   }
140 };
141
142
143