Analysis Software
Documentation for sPHENIX simulation software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
DrawUECentralityPlots.C
Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file DrawUECentralityPlots.C
1 void SetHist(TH2* h, string xt ="", string yt ="",int color = 4, int marker = 20,int width = 3, float size = 1.0);
2 void SetHist(TH2* h, int color = 4);
3 void SetLeg(TLegend* l,float txtsize=0.03);
4 
5 
6 void DrawUECentralityPlots(string infile = "UEvsEtaCentrality.root"){
7 
8  gStyle->SetOptStat(0);
9 
10  TFile *_file0 = TFile::Open(infile.c_str());
11 
12  TH2* hv2_cent = (TH2*)gROOT->FindObject("hv2_cent");
13  TH2* hPsi2_cent = (TH2*)gROOT->FindObject("hPsi2_cent");
14  TCanvas* can1 = new TCanvas("can1","",800,400);
15  can1->Divide(2,1,0,0);
16 
17  can1->cd(1);
18  gPad->SetLeftMargin(0.15);
19  gPad->SetRightMargin(.15);
20  gPad->SetTopMargin(.07);
21  gPad->SetBottomMargin(.10);
22  SetHist(hv2_cent,"Centrality %","v2 value",1);
23  hv2_cent->SetTitle("v2 vs Centrality");
24  hv2_cent->Draw("ColZ");
25 
26  TProfile* v2profile = hv2_cent->ProfileX("Mean Values of v2");
27 
28  gStyle->SetTitleTextColor(1);
29  v2profile->Draw("same");
30 
31  can1->cd(2);
32  gPad->SetLeftMargin(0.15);
33  gPad->SetRightMargin(.15);
34  gPad->SetTopMargin(.07);
35  gPad->SetBottomMargin(.10);
36  SetHist(hPsi2_cent,"Centrality %","Psi2 value",1);
37  hPsi2_cent->SetTitle("Psi2 vs Centrality");
38  hPsi2_cent->Draw("ColZ");
39 
40  TProfile* Psi2profile = hPsi2_cent->ProfileX("Mean Values of Psi2");
41 
42  gStyle->SetTitleTextColor(1);
43  Psi2profile->Draw("same");
44 
45  TH2* hUEiHcalEta_Cent0_20 = (TH2*)gROOT->FindObject("hUEiHcalEta_Cent0_20");
46  TH2* hUEoHcalEta_Cent0_20 = (TH2*)gROOT->FindObject("hUEoHcalEta_Cent0_20");
47  TH2* hUEemcalEta_Cent0_20 = (TH2*)gROOT->FindObject("hUEemcalEta_Cent0_20");
48 
49  TH2* hUEiHcalEta_Cent20_50 = (TH2*)gROOT->FindObject("hUEiHcalEta_Cent20_50");
50  TH2* hUEoHcalEta_Cent20_50 = (TH2*)gROOT->FindObject("hUEoHcalEta_Cent20_50");
51  TH2* hUEemcalEta_Cent20_50 = (TH2*)gROOT->FindObject("hUEemcalEta_Cent20_50");
52 
53  TH2* hUEiHcalEta_Cent50_100 = (TH2*)gROOT->FindObject("hUEiHcalEta_Cent50_100");
54  TH2* hUEoHcalEta_Cent50_100 = (TH2*)gROOT->FindObject("hUEoHcalEta_Cent50_100");
55  TH2* hUEemcalEta_Cent50_100 = (TH2*)gROOT->FindObject("hUEemcalEta_Cent50_100");
56 
57  TCanvas* can2 = new TCanvas("can2","",1200,400);
58  can2->Divide(3,1,0,0);
59 
60  can2->cd(1);
61  gPad->SetLeftMargin(0.15);
62  gPad->SetRightMargin(.10);
63  gPad->SetTopMargin(.07);
64  gPad->SetBottomMargin(.10);
65  SetHist(hUEiHcalEta_Cent0_20,"Eta","UE Size/Tower (GeV)",1);
66  hUEiHcalEta_Cent0_20->SetTitle("UE Size vs Eta 0 - 20 % Centrality Range iHCal");
67  hUEiHcalEta_Cent0_20->Draw("ColZ");
68  hUEiHcalEta_Cent0_20->ProfileX()->Draw("same");
69 
70  can2->cd(2);
71  gPad->SetLeftMargin(0.15);
72  gPad->SetRightMargin(.15);
73  gPad->SetTopMargin(.07);
74  gPad->SetBottomMargin(.10);
75  SetHist(hUEiHcalEta_Cent20_50,"Eta","UE Size/Tower (GeV)",1);
76  hUEiHcalEta_Cent20_50->SetTitle("UE Size vs Eta 20 - 50 % Centrality Range iHCal");
77  hUEiHcalEta_Cent20_50->GetYaxis()->SetRangeUser(0.0,0.05);
78  hUEiHcalEta_Cent20_50->Draw("ColZ");
79  hUEiHcalEta_Cent20_50->ProfileX()->Draw("same");
80 
81  can2->cd(3);
82  gPad->SetLeftMargin(0.15);
83  gPad->SetRightMargin(.10);
84  gPad->SetTopMargin(.07);
85  gPad->SetBottomMargin(.10);
86  SetHist(hUEiHcalEta_Cent50_100,"Eta","UE Size/Tower (GeV)",1);
87  hUEiHcalEta_Cent50_100->SetTitle("UE Size vs Eta 50 - 100 % Centrality Range iHCal");
88  hUEiHcalEta_Cent50_100->GetYaxis()->SetRangeUser(0.0,0.03);
89  hUEiHcalEta_Cent50_100->Draw("ColZ");
90  hUEiHcalEta_Cent50_100->ProfileX()->Draw("same");
91 
92  TCanvas* can3 = new TCanvas("can3","",1200,400);
93  can3->Divide(3,1,0,0);
94 
95  can3->cd(1);
96  gPad->SetLeftMargin(0.15);
97  gPad->SetRightMargin(.10);
98  gPad->SetTopMargin(.07);
99  gPad->SetBottomMargin(.10);
100  SetHist(hUEoHcalEta_Cent0_20,"Eta","UE Size/Tower (GeV)",1);
101  hUEoHcalEta_Cent0_20->SetTitle("UE Size vs Eta 0 - 20 % Centrality Range oHCal");
102  hUEoHcalEta_Cent0_20->Draw("ColZ");
103  hUEoHcalEta_Cent0_20->ProfileX()->Draw("same");
104 
105  can3->cd(2);
106  gPad->SetLeftMargin(0.15);
107  gPad->SetRightMargin(.15);
108  gPad->SetTopMargin(.07);
109  gPad->SetBottomMargin(.10);
110  SetHist(hUEoHcalEta_Cent20_50,"Eta","UE Size/Tower (GeV)",1);
111  hUEoHcalEta_Cent20_50->SetTitle("UE Size vs Eta 20 - 50 % Centrality Range oHCal");
112  hUEoHcalEta_Cent20_50->GetYaxis()->SetRangeUser(0.0,0.25);
113  hUEoHcalEta_Cent20_50->Draw("ColZ");
114  hUEoHcalEta_Cent20_50->ProfileX()->Draw("same");
115 
116  can3->cd(3);
117  gPad->SetLeftMargin(0.15);
118  gPad->SetRightMargin(.10);
119  gPad->SetTopMargin(.07);
120  gPad->SetBottomMargin(.10);
121  SetHist(hUEoHcalEta_Cent50_100,"Eta","UE Size/Tower (GeV)",1);
122  hUEoHcalEta_Cent50_100->SetTitle("UE Size vs Eta 50 - 100 % Centrality Range oHCal");
123  hUEoHcalEta_Cent50_100->GetYaxis()->SetRangeUser(0.0,0.15);
124  hUEoHcalEta_Cent50_100->Draw("ColZ");
125  hUEoHcalEta_Cent50_100->ProfileX()->Draw("same");
126 
127  TCanvas* can4 = new TCanvas("can4","",1200,400);
128  can4->Divide(3,1,0,0);
129 
130 
131  can4->cd(1);
132  gPad->SetLeftMargin(0.15);
133  gPad->SetRightMargin(.10);
134  gPad->SetTopMargin(.07);
135  gPad->SetBottomMargin(.10);
136  SetHist(hUEemcalEta_Cent0_20,"Eta","UE Size/Tower (GeV)",1);
137  hUEemcalEta_Cent0_20->SetTitle("UE Size vs Eta 0 - 20 % Centrality Range EMCal");
138  hUEemcalEta_Cent0_20->Draw("ColZ");
139  hUEemcalEta_Cent0_20->ProfileX()->Draw("same");
140 
141  can4->cd(2);
142  gPad->SetLeftMargin(0.15);
143  gPad->SetRightMargin(.15);
144  gPad->SetTopMargin(.07);
145  gPad->SetBottomMargin(.10);
146  SetHist(hUEemcalEta_Cent20_50,"Eta","UE Size/Tower (GeV)",1);
147  hUEemcalEta_Cent20_50->SetTitle("UE Size vs Eta 20 - 50 % Centrality Range EMCal");
148  hUEemcalEta_Cent20_50->GetYaxis()->SetRangeUser(0.0,0.60);
149  hUEemcalEta_Cent20_50->Draw("ColZ");
150  hUEemcalEta_Cent20_50->ProfileX()->Draw("same");
151 
152  can4->cd(3);
153  gPad->SetLeftMargin(0.15);
154  gPad->SetRightMargin(.10);
155  gPad->SetTopMargin(.07);
156  gPad->SetBottomMargin(.10);
157  SetHist(hUEemcalEta_Cent50_100,"Eta","UE Size/Tower (GeV)",1);
158  hUEemcalEta_Cent50_100->SetTitle("UE Size vs Eta 50 - 100 % Centrality Range EMCal");
159  hUEemcalEta_Cent50_100->GetYaxis()->SetRangeUser(0.0,0.30);
160  hUEemcalEta_Cent50_100->Draw("ColZ");
161  hUEemcalEta_Cent50_100->ProfileX()->Draw("same");
162 
163 
164  TProfile *IH0_20 = hUEiHcalEta_Cent0_20->ProfileX();
165  TProfile *IH20_50 = hUEiHcalEta_Cent20_50->ProfileX();
166  TProfile *IH50_100 = hUEiHcalEta_Cent50_100->ProfileX();
167  TProfile *OH0_20 = hUEoHcalEta_Cent0_20->ProfileX();
168  TProfile *OH20_50 = hUEoHcalEta_Cent20_50->ProfileX();
169  TProfile *OH50_100 = hUEoHcalEta_Cent50_100->ProfileX();
170  TProfile *EM0_20 = hUEemcalEta_Cent0_20->ProfileX();
171  TProfile *EM20_50 = hUEemcalEta_Cent20_50->ProfileX();
172  TProfile *EM50_100 = hUEemcalEta_Cent50_100->ProfileX();
173 
174  TCanvas* can5 = new TCanvas("can5","",1200,400);
175  can5->Divide(3,1,0,0);
176  can5->cd(1);
177  gPad->SetLeftMargin(0.15);
178  gPad->SetRightMargin(.10);
179  gPad->SetTopMargin(.07);
180  gPad->SetBottomMargin(.10);
181  IH0_20->SetTitle("Mean Values of UE Size vs Eta iHCal");
182  IH0_20->GetYaxis()->SetTitle("UE Size/Tower (GeV)");
183  IH0_20->GetYaxis()->SetRangeUser(0.0,0.08);
184  IH0_20->GetXaxis()->SetTitle("Eta");
185  IH0_20->Draw();
186  IH20_50->SetMarkerColor(1);
187  IH50_100->SetMarkerColor(2);
188  IH20_50->Draw("same");
189  IH50_100->Draw("same");
190 
191  TLegend* leg1 = new TLegend(0.4,0.7,0.7,0.85);
192  SetLeg(leg1);
193  leg1->AddEntry(IH0_20,"0-20 % Centrality","P");
194  leg1->AddEntry(IH20_50,"20-50 % Centrality","P");
195  leg1->AddEntry(IH50_100,"50-100 % Centrality","P");
196  leg1->Draw();
197 
198 
199  can5->cd(2);
200  gPad->SetLeftMargin(0.15);
201  gPad->SetRightMargin(.10);
202  gPad->SetTopMargin(.07);
203  gPad->SetBottomMargin(.10);
204  OH0_20->SetTitle("Mean Values of UE Size vs Eta oHCal");
205  OH0_20->GetYaxis()->SetTitle("UE Size/Tower (GeV)");
206  OH0_20->GetXaxis()->SetTitle("Eta");
207  OH0_20->GetYaxis()->SetRangeUser(0.0,0.30);
208  OH0_20->Draw();
209  OH20_50->SetMarkerColor(1);
210  OH50_100->SetMarkerColor(2);
211  OH20_50->Draw("same");
212  OH50_100->Draw("same");
213 
214  TLegend* leg2 = new TLegend(0.4,0.7,0.7,0.85);
215  SetLeg(leg2);
216  leg2->AddEntry(OH0_20,"0-20 % Centrality","P"); leg2->AddEntry(OH20_50,"20-50 % Centrality","P"); leg2->AddEntry(OH50_100,"50-100 % Centrality","P"); leg2->Draw();
217 
218  can5->cd(3);
219  gPad->SetLeftMargin(0.15);
220  gPad->SetRightMargin(.10);
221  gPad->SetTopMargin(.07);
222  gPad->SetBottomMargin(.10);
223  EM0_20->SetTitle("Mean Values of UE Size vs Eta EMCal");
224  EM0_20->GetYaxis()->SetTitle("UE Size/Tower (GeV)");
225  EM0_20->GetXaxis()->SetTitle("Eta");
226  EM0_20->GetYaxis()->SetRangeUser(0.0,0.70);
227  EM0_20->Draw();
228  EM20_50->SetMarkerColor(1);
229  EM50_100->SetMarkerColor(2);
230  EM20_50->Draw("same");
231  EM50_100->Draw("same");
232 
233 
234  TLegend* leg3 = new TLegend(0.4,0.7,0.7,0.85);
235  SetLeg(leg3);
236  leg3->AddEntry(EM0_20,"0-20 % Centrality","P"); leg3->AddEntry(EM20_50,"20-50 % Centrality","P"); leg3->AddEntry(EM50_100,"50-100 % Centrality","P"); leg3->Draw();
237  }
238 
239 void SetHist(TH2* h, string xt ="", string yt ="",int color = 4, int marker = 24,int width = 3, float size = 1.0)
240 {
241  h->SetLineWidth(width);
242  h->SetLineColor(color);
243  h->SetMarkerColor(3);
244  h->SetMarkerSize(0.5);
245  h->SetMarkerStyle(marker);
246  h->GetYaxis()->SetTitle(yt.c_str());
247  h->GetYaxis()->SetTitleOffset(1.6);
248  h->GetXaxis()->SetTitle(xt.c_str());
249 }
250 
251 void SetHist(TH2* h, int color = 4)
252 {
253  h->SetLineWidth(3);
254  h->SetLineColor(color);
255  h->SetMarkerColor(color);
256  h->SetMarkerSize(0.25);
257  h->GetYaxis()->SetTitleOffset(1.6);
258 
259 }
260 
261 
262 void SetLeg(TLegend* l,float txtsize=0.04){
263  l->SetBorderSize(1);
264  l->SetFillColor(0);
265  l->SetTextSize(txtsize);
266 }