]>
Commit | Line | Data |
---|---|---|
1 | /** | |
2 | * @file DrawOccupancy.C | |
3 | * @author Christian Holm Christensen <cholm@dalsgaard.hehi.nbi.dk> | |
4 | * @date Thu Jul 7 10:24:58 2011 | |
5 | * | |
6 | * @brief A script to draw the Poisson vs Energy Loss correlation | |
7 | * | |
8 | * | |
9 | * @ingroup pwg2_forward_scripts_qa | |
10 | * | |
11 | */ | |
12 | #ifndef __CINT__ | |
13 | # include <TH1.h> | |
14 | # include <TH2.h> | |
15 | # include <TList.h> | |
16 | # include <TFile.h> | |
17 | # include <TString.h> | |
18 | # include <TError.h> | |
19 | # include <TPad.h> | |
20 | # include <TCanvas.h> | |
21 | # include <TLine.h> | |
22 | # include <TLatex.h> | |
23 | # include <TStyle.h> | |
24 | #else | |
25 | class TList; | |
26 | #endif | |
27 | ||
28 | /** | |
29 | * Draw the poisson @f$N_{ch}@f$ estimate against the @f$\Delta@f$ | |
30 | * @f$N_{ch}@f$ estimate and do a regression line between the two | |
31 | * | |
32 | * @param p List | |
33 | * @param d Detector | |
34 | * @param r Ring | |
35 | * @param xmin Minimum | |
36 | * @param xmax Maximum | |
37 | * | |
38 | * @return The regression coefficient | |
39 | * | |
40 | * @ingroup pwg2_forward_scripts_qa | |
41 | */ | |
42 | Double_t | |
43 | DrawRingOccupancy(TList* p, UShort_t d, Char_t r) | |
44 | { | |
45 | if (!p) return 0; | |
46 | ||
47 | TList* ring = static_cast<TList*>(p->FindObject(Form("FMD%d%c",d,r))); | |
48 | if (!ring) { | |
49 | Error("DrawOccupancy", "List FMD%d%c not found in %s",d,r,p->GetName()); | |
50 | return 0; | |
51 | } | |
52 | ||
53 | TH1* corr = static_cast<TH1*>(ring->FindObject("occupancy")); | |
54 | if (!corr) { | |
55 | Error("DrawRingOccupancy", "Histogram occupancy not found in FMD%d%c", | |
56 | d, r); | |
57 | return 0; | |
58 | } | |
59 | corr->Rebin(4); | |
60 | ||
61 | TPad* pad = static_cast<TPad*>(gPad); | |
62 | pad->SetGridy(); | |
63 | pad->SetGridx(); | |
64 | pad->SetLogy(); | |
65 | pad->SetFillColor(0); | |
66 | pad->SetRightMargin(0.01); | |
67 | #if 0 | |
68 | if (d == 3) { | |
69 | pad->SetPad(pad->GetXlowNDC(), pad->GetYlowNDC(), .99, | |
70 | pad->GetYlowNDC()+pad->GetHNDC()); | |
71 | pad->SetRightMargin(0.15); | |
72 | } | |
73 | #endif | |
74 | ||
75 | corr->Draw("hist"); | |
76 | ||
77 | TLatex* ltx = new TLatex(.95, .95, Form("FMD%d%c", d, r)); | |
78 | ltx->SetNDC(); | |
79 | ltx->SetTextAlign(33); | |
80 | ltx->SetTextSize(.08); | |
81 | ltx->Draw(); | |
82 | ||
83 | return corr->GetMean(); | |
84 | } | |
85 | ||
86 | /** | |
87 | * Draw the correlation between the Poisson @f$N_{ch}@f$ estimate | |
88 | * and the @f$\Delta@f$ @f$N_{ch}@f$ estimate and do a regression | |
89 | * line between the two for each ring | |
90 | * | |
91 | * @param filename File to read | |
92 | * @param xmax Minimum X | |
93 | * @param xmin Maximum X | |
94 | * | |
95 | * @ingroup pwg2_forward_scripts_qa | |
96 | */ | |
97 | void | |
98 | DrawOccupancy(const char* filename="forward.root", | |
99 | const char* folder="ForwardResults") | |
100 | { | |
101 | gStyle->SetPalette(1); | |
102 | gStyle->SetOptFit(0); | |
103 | gStyle->SetOptStat(0); | |
104 | gStyle->SetOptTitle(0); | |
105 | gStyle->SetTitleW(.4); | |
106 | gStyle->SetTitleH(.1); | |
107 | gStyle->SetTitleX(.4); | |
108 | // gStyle->SetTitleY(.1); | |
109 | gStyle->SetTitleColor(0); | |
110 | gStyle->SetTitleStyle(0); | |
111 | gStyle->SetTitleBorderSize(0); | |
112 | ||
113 | TFile* file = TFile::Open(filename, "READ"); | |
114 | if (!file) { | |
115 | Error("DrawOccupancy", "failed to open %s", filename); | |
116 | return; | |
117 | } | |
118 | ||
119 | TList* forward = static_cast<TList*>(file->Get(folder)); | |
120 | if (!forward) { | |
121 | Error("DrawOccupancy", "List %s not found in %s", folder, filename); | |
122 | return; | |
123 | } | |
124 | ||
125 | TList* dc = static_cast<TList*>(forward->FindObject("fmdDensityCalculator")); | |
126 | if (!dc) { | |
127 | Error("DrawOccupancy", "List fmdDensityCalculator not found in Forward"); | |
128 | return; | |
129 | } | |
130 | ||
131 | TCanvas* c = new TCanvas("occupancy", | |
132 | "Mean Occupancy", 900, 700); | |
133 | c->SetFillColor(0); | |
134 | c->SetBorderSize(0); | |
135 | c->SetBorderMode(0); | |
136 | c->SetHighLightColor(0); | |
137 | c->SetBottomMargin(.15); | |
138 | c->SetTopMargin(.02); | |
139 | c->SetRightMargin(.02); | |
140 | c->SetLeftMargin(.15); | |
141 | c->Divide(3, 2, 0, 0); | |
142 | ||
143 | ||
144 | Double_t corrs[5]; | |
145 | c->cd(1); corrs[0] = DrawRingOccupancy(dc, 1, 'I'); | |
146 | c->cd(2); corrs[1] = DrawRingOccupancy(dc, 2, 'I'); | |
147 | c->cd(5); corrs[2] = DrawRingOccupancy(dc, 2, 'O'); | |
148 | c->cd(3); corrs[3] = DrawRingOccupancy(dc, 3, 'I'); | |
149 | c->cd(6); corrs[4] = DrawRingOccupancy(dc, 3, 'O'); | |
150 | ||
151 | TVirtualPad* p = c->cd(4); | |
152 | p->SetTopMargin(0.05); | |
153 | p->SetRightMargin(0.10); | |
154 | p->SetLeftMargin(0.15); | |
155 | p->SetBottomMargin(0.15); | |
156 | p->SetFillColor(0); | |
157 | ||
158 | TH1D* hc = new TH1D("occ", "Mean occupancy", 5, .5, 5.5); | |
159 | hc->SetFillColor(kRed+1); | |
160 | hc->SetFillStyle(3001); | |
161 | hc->SetMinimum(0.0); | |
162 | hc->GetXaxis()->SetBinLabel(1,"FMD1i"); hc->SetBinContent(1,corrs[0]); | |
163 | hc->GetXaxis()->SetBinLabel(2,"FMD2i"); hc->SetBinContent(2,corrs[1]); | |
164 | hc->GetXaxis()->SetBinLabel(3,"FMD2o"); hc->SetBinContent(3,corrs[2]); | |
165 | hc->GetXaxis()->SetBinLabel(4,"FMD3i"); hc->SetBinContent(4,corrs[3]); | |
166 | hc->GetXaxis()->SetBinLabel(5,"FMD3o"); hc->SetBinContent(5,corrs[4]); | |
167 | hc->GetXaxis()->SetLabelSize(0.08); | |
168 | hc->GetYaxis()->SetTitle("#bar{occupancy}"); | |
169 | hc->SetMarkerSize(1.5); | |
170 | hc->Draw("text hist"); | |
171 | hc->SetMaximum(hc->GetMaximum()*1.5); | |
172 | ||
173 | // TH2D* highCuts = static_cast<TH2D*>(dc->FindObject("highCuts")); | |
174 | // if (highCuts) highCuts->Draw("colz"); | |
175 | c->cd(); | |
176 | c->SaveAs("occupancy.png"); | |
177 | } | |
178 | // | |
179 | // EOF | |
180 | // |