5 |
|
|
6 |
|
void Layout::DrawInit() |
7 |
|
{ |
8 |
< |
/* gStyle->SetCanvasColor(0); |
8 |
> |
gStyle->SetCanvasColor(0); |
9 |
|
gStyle->SetCanvasBorderMode(0); |
10 |
|
gStyle->SetCanvasBorderSize(3); |
11 |
|
|
70 |
|
Double_t blue[NRGBs] = { 0.51, 1.00, 0.12, 0.00, 0.00 }; |
71 |
|
TColor::CreateGradientColorTable(NRGBs, stops, red, green, blue, NCont); |
72 |
|
gStyle->SetNumberContours(NCont); |
73 |
< |
*/ |
73 |
> |
|
74 |
|
|
75 |
|
} |
76 |
|
|
77 |
|
|
78 |
< |
void Layout::Draw(const std::vector<HistoBlock>& histos, |
79 |
< |
const std::string& xlabel, float K, float MU, float SIG) |
78 |
> |
void Layout::Draw(const std::string title, const std::vector<HistoBlock>& histos, |
79 |
> |
const std::string& xlabel, bool fit, float K, float MU, float SIG) |
80 |
|
{ |
81 |
|
std::cout << "----------------- Draw : " << histos[0].histo->GetName() << std::endl; |
82 |
|
|
91 |
|
// SetLineColor |
92 |
|
for (unsigned int i=0;i<histos.size();i++) |
93 |
|
{ |
94 |
– |
std::cout << "TEST : " << histos[i].linecolor << std::endl; |
94 |
|
histos[i].histo->SetLineColor( histos[i].linecolor ); |
95 |
|
} |
96 |
|
|
109 |
|
float xmax = mystack->GetXaxis()->GetXmin(); |
110 |
|
|
111 |
|
// Draw legend |
112 |
< |
TLegend* mylegend = new TLegend(0.17,0.85-(0.04*histos.size()),0.33,0.85); |
112 |
> |
TLegend* mylegend = new TLegend(0.55,0.90-(0.04*histos.size()),0.90,0.90); |
113 |
|
mylegend->SetFillColor(0); |
114 |
|
for (unsigned int i=0;i<histos.size();i++) |
115 |
|
{ |
117 |
|
histos[i].algoName.c_str(),"l"); |
118 |
|
} |
119 |
|
mylegend->Draw(); |
121 |
– |
std::cout << "size = " << mylegend->GetTextSize() << std::endl; |
122 |
– |
std::cout << "font = " << mylegend->GetTextFont() << std::endl; |
123 |
– |
|
124 |
– |
std::cout << mylegend->GetX1() << " " << mylegend->GetX2() << std::endl; |
125 |
– |
std::cout << mylegend->GetY1() << " " << mylegend->GetY2() << std::endl; |
126 |
– |
|
127 |
– |
TPaveText* toto = new TPaveText(0.17,0.72,0.43,0.83); |
128 |
– |
toto->SetFillColor(2); |
129 |
– |
toto->SetTextFont(2); |
130 |
– |
toto->SetTextSize(0); |
131 |
– |
toto->AddText("toto"); |
132 |
– |
toto->Draw(); |
120 |
|
|
134 |
– |
// To Fit |
135 |
– |
std::vector<double> sigmas; |
136 |
– |
for (unsigned int i=0;i<histos.size();i++) |
137 |
– |
{ |
138 |
– |
TF1* fitter = new TF1(dummy().c_str(),"gaus",xmin,xmax); |
139 |
– |
fitter->SetParameter(0,K); |
140 |
– |
fitter->SetParameter(1,MU); |
141 |
– |
fitter->SetParameter(2,SIG); |
142 |
– |
histos[i].histo->Fit(fitter,"0"); |
143 |
– |
sigmas.push_back(fitter->GetParameter(2)); |
144 |
– |
} |
121 |
|
|
122 |
< |
// Display sigmas |
147 |
< |
for (unsigned int i=0;i<sigmas.size();i++) |
122 |
> |
if (fit) |
123 |
|
{ |
124 |
< |
std::stringstream str; |
125 |
< |
str << "#sigma_{RECO} = "; |
126 |
< |
str << sigmas[i]; |
127 |
< |
TLatex mytext(56,200,str.str().c_str()); |
128 |
< |
std::cout << mycanvas->GetXlowNDC() << " " << mycanvas->GetWNDC() << std::endl; |
124 |
> |
// To Fit |
125 |
> |
std::vector<double> sigmas; |
126 |
> |
std::vector<double> errors; |
127 |
> |
for (unsigned int i=0;i<histos.size();i++) |
128 |
> |
{ |
129 |
> |
TF1* fitter = new TF1(dummy().c_str(),"gaus",xmin,xmax); |
130 |
> |
fitter->SetParameter(0,K); |
131 |
> |
fitter->SetParameter(1,MU); |
132 |
> |
fitter->SetParameter(2,SIG); |
133 |
> |
histos[i].histo->Fit(fitter,"0"); |
134 |
> |
sigmas.push_back(fitter->GetParameter(2)); |
135 |
> |
errors.push_back(fitter->GetParError(2)); |
136 |
> |
} |
137 |
> |
|
138 |
> |
// Display sigmas |
139 |
> |
TPaveText* myblock = new TPaveText(0.65,0.65-(0.04*histos.size()),0.90,0.65,"NDC"); |
140 |
> |
myblock->SetBorderSize(0); |
141 |
> |
myblock->SetFillColor(0); |
142 |
> |
for (unsigned int i=0;i<sigmas.size();i++) |
143 |
> |
{ |
144 |
> |
std::stringstream str; |
145 |
> |
str << "#sigma_{" << histos[i].shortName <<"} = "; |
146 |
> |
str.precision(3); str << sigmas[i]; |
147 |
> |
str << " #pm "; |
148 |
> |
str.precision(3); str << errors[i]; |
149 |
> |
TText* mytext = myblock->AddText(str.str().c_str()); |
150 |
> |
mytext->SetTextColor(histos[i].linecolor); |
151 |
> |
mytext->SetTextAlign(11); |
152 |
> |
} |
153 |
> |
myblock->Draw(); |
154 |
|
} |
155 |
|
|
156 |
|
// Save |
157 |
< |
std::string filename = GetFileName(histos[0].histo->GetTitle()); |
157 |
> |
std::string filename = GetFileName(title); |
158 |
|
mycanvas->SaveAs((filename+".gif").c_str()); |
159 |
|
mycanvas->SaveAs((filename+".root").c_str()); |
160 |
|
|