CMS 3D CMS Logo

Functions | Variables
compare Namespace Reference

Functions

def ensureDir (directory)
 
def hoverlay (hists, xtitle, ytitle, name)
 
def LegendSettings (leg, ncolumn)
 
def makeCompareVars (tree, var, sel, leglist, nbin, xmin, xmax, xtitle, ytitle, scale, name)
 
def makeEffPlotsVars (tree, varx, vary, sel, nbinx, xmin, xmax, nbiny, ymin, ymax, xtitle, ytitle, leglabel=None, header='', addon='', option='pt', marker=20)
 
def overlay (hists, ytitle, header, addon)
 
def save (canvas, name)
 

Variables

 argc
 
 argvs
 
 den_sel
 
 hist
 
 hists
 
 hvardict
 
 loose_id
 
 num_sel
 
 reco_cut
 
 runtype
 
 sampledict
 
 tfile
 
 tlabel
 
 tree
 
 vardict
 
 xlabel
 

Function Documentation

def compare.ensureDir (   directory)

Definition at line 41 of file compare.py.

Referenced by save().

41 def ensureDir(directory):
42  if not os.path.exists(directory):
43  os.makedirs(directory)
44 
def ensureDir(directory)
Definition: compare.py:41
def compare.hoverlay (   hists,
  xtitle,
  ytitle,
  name 
)

Definition at line 193 of file compare.py.

References LegendSettings(), and save().

193 def hoverlay(hists, xtitle, ytitle, name):
194 
195  c = TCanvas()
196 
197  ymax = 0
198  for hist in hists:
199  if ymax < hist.GetMaximum():
200  ymax = hist.GetMaximum()
201 
202 
203  leg = TLegend(0.6,0.65,0.91,0.9)
204  LegendSettings(leg,1)
205 
206  for ii, ihist in enumerate(hists):
207  ihist.SetMaximum(ymax*1.2)
208  ihist.SetMinimum(0.)
209  ihist.SetMarkerSize(0.)
210  ihist.GetXaxis().SetTitle(xtitle)
211  ihist.GetYaxis().SetTitle(ytitle)
212 
213  if ii ==0:
214  ihist.Draw('h')
215  else:
216  ihist.Draw('hsame')
217 
218  leg.AddEntry(ihist, ihist.GetName(), "l")
219 
220 
221  leg.Draw()
222 
223 
224  save(c, 'compare_' + runtype + '/hist_' + name)
225 
226 
227 
228 
def save(canvas, name)
Definition: compare.py:45
def LegendSettings(leg, ncolumn)
Definition: compare.py:51
def hoverlay(hists, xtitle, ytitle, name)
Definition: compare.py:193
def compare.LegendSettings (   leg,
  ncolumn 
)

Definition at line 51 of file compare.py.

Referenced by hoverlay(), makeCompareVars(), and overlay().

51 def LegendSettings(leg, ncolumn):
52  leg.SetNColumns(ncolumn)
53  leg.SetBorderSize(0)
54  leg.SetFillColor(10)
55  leg.SetLineColor(0)
56  leg.SetFillStyle(0)
57  leg.SetTextSize(0.03)
58  leg.SetTextFont(42)
59 
60 
61 
def LegendSettings(leg, ncolumn)
Definition: compare.py:51
def compare.makeCompareVars (   tree,
  var,
  sel,
  leglist,
  nbin,
  xmin,
  xmax,
  xtitle,
  ytitle,
  scale,
  name 
)

Definition at line 62 of file compare.py.

References LegendSettings(), and str.

62 def makeCompareVars(tree, var, sel, leglist, nbin, xmin, xmax, xtitle, ytitle, scale, name):
63 
64 # print leglist
65 
66  c = TCanvas()
67 
68  hists = []
69  col = [1,2,4,8,6]
70  ymax = 0
71 
72  for ii, isel in enumerate(sel):
73  hist = TH1F('h_' + str(ii), 'h_' + str(ii), nbin, xmin, xmax)
74  hist.GetXaxis().SetTitle(xtitle)
75  hist.GetYaxis().SetTitle(ytitle)
76  hist.GetYaxis().SetNdivisions(507)
77  hist.SetLineColor(col[ii])
78  hist.SetLineWidth(len(sel)+1-ii)
79  hist.SetLineStyle(1)
80  hist.SetMarkerSize(0)
81  hist.SetMinimum(0)
82  hist.Sumw2()
83 
84 # print hist.GetName(), var, isel
85  tree.Project(hist.GetName(), var, isel)
86  hist.Scale(1./hist.GetEntries())
87 
88  if ymax < hist.GetMaximum():
89  ymax = hist.GetMaximum()
90 
91  hists.append(hist)
92 
93 
94  leg = TLegend(0.6,0.65,0.91,0.9)
95  LegendSettings(leg,1)
96 
97  for ii, ihist in enumerate(hists):
98  ihist.SetMaximum(ymax*1.2)
99  ihist.SetMinimum(0.)
100 
101  if ii ==0:
102  ihist.Draw('h')
103  else:
104  ihist.Draw('hsame')
105 
106  if leglist[ii] != 'None':
107  leg.AddEntry(ihist, leglist[ii], "l")
108 
109 
110  if leglist[0]!='None':
111  leg.Draw()
112 
113 
114 # save(c, 'compare_' + runtype + '/compare_' + name)
115 
116 
117 
118 
119 
def LegendSettings(leg, ncolumn)
Definition: compare.py:51
#define str(s)
def makeCompareVars(tree, var, sel, leglist, nbin, xmin, xmax, xtitle, ytitle, scale, name)
Definition: compare.py:62
def compare.makeEffPlotsVars (   tree,
  varx,
  vary,
  sel,
  nbinx,
  xmin,
  xmax,
  nbiny,
  ymin,
  ymax,
  xtitle,
  ytitle,
  leglabel = None,
  header = '',
  addon = '',
  option = 'pt',
  marker = 20 
)

Definition at line 229 of file compare.py.

References mps_check.array.

229 def makeEffPlotsVars(tree, varx, vary, sel, nbinx, xmin, xmax, nbiny, ymin, ymax, xtitle, ytitle, leglabel = None, header='', addon='', option='pt', marker=20):
230 
231  binning = [20,30,40,50,60,70,80,100,150,200]
232 
233  c = TCanvas()
234 
235  if option=='pt':
236  _hist_ = TH1F('h_effp_' + addon, 'h_effp' + addon, len(binning)-1, array('d',binning))
237  _ahist_ = TH1F('ah_effp_' + addon, 'ah_effp' + addon, len(binning)-1, array('d',binning))
238  elif option=='eta':
239  _hist_ = TH1F('h_effp_' + addon, 'h_effp' + addon, nbinx, xmin, xmax)
240  _ahist_ = TH1F('ah_effp_' + addon, 'ah_effp' + addon, nbinx, xmin, xmax)
241  elif option=='nvtx':
242  _hist_ = TH1F('h_effp_' + addon, 'h_effp' + addon, len(vbinning)-1, array('d',vbinning))
243  _ahist_ = TH1F('ah_effp_' + addon, 'ah_effp' + addon, len(vbinning)-1, array('d',vbinning))
244 
245 
246  tree.Draw(varx + ' >> ' + _hist_.GetName(), sel)
247  tree.Draw(varx + ' >> ' + _ahist_.GetName(), sel + ' && ' + vary)
248 
249  g_efficiency = TGraphAsymmErrors()
250  g_efficiency.BayesDivide(_ahist_, _hist_)
251  g_efficiency.GetXaxis().SetTitle(xtitle)
252  g_efficiency.GetYaxis().SetTitle('efficiency')
253  g_efficiency.GetYaxis().SetNdivisions(507)
254  g_efficiency.SetLineWidth(3)
255  g_efficiency.SetName(header)
256  g_efficiency.SetMinimum(0.)
257  g_efficiency.GetYaxis().SetTitleOffset(1.3)
258  g_efficiency.SetMarkerStyle(marker)
259  g_efficiency.SetMarkerSize(1)
260  g_efficiency.Draw('ap')
261 
262 # save(c, 'plots/' + addon)
263 
264  return copy.deepcopy(g_efficiency)
265 
266 
267 
268 
269 
def makeEffPlotsVars(tree, varx, vary, sel, nbinx, xmin, xmax, nbiny, ymin, ymax, xtitle, ytitle, leglabel=None, header='', addon='', option='pt', marker=20)
Definition: compare.py:229
def compare.overlay (   hists,
  ytitle,
  header,
  addon 
)

Definition at line 120 of file compare.py.

References LegendSettings(), and save().

Referenced by FWCaloTowerDetailView.build(), FWMuonDetailView.build(), FWPhotonDetailView.build(), FWElectronDetailView.build(), and SiStripHitEffFromCalibTree.makeSummary().

120 def overlay(hists, ytitle, header, addon):
121 
122  print 'number of histograms = ', len(hists)
123 
124  canvas = TCanvas()
125  leg = TLegend(0.2,0.7,0.5,0.9)
126  LegendSettings(leg, 1)
127 
128  col = [1,2,4,6,8,9,12]
129 
130  ymax = -1
131  ymin = 100
132 
133  for ii, hist in enumerate(hists):
134  hist.GetYaxis().SetTitle('efficiency')
135  hist.SetLineColor(col[ii])
136  hist.SetMarkerColor(col[ii])
137  hist.SetLineWidth(2)
138  hist.SetMarkerSize(1)
139 
140 
141  for ip in range(hist.GetN()):
142  x = Double(-1)
143  y = Double(-1)
144  hist.GetPoint(ip, x, y)
145 
146  if ymin > y:
147  ymin = y
148  if ymax < y:
149  ymax = y
150 
151 #
152 # if ymax < hist.GetMaximum():
153 # ymax = hist.GetMaximum()
154 # if ymin > hist.GetMinimum():
155 # ymin = hist.GetMinimum()
156 
157  if ii==0:
158  hist.Draw("Azp")
159  else:
160 
161  hist.Draw("pzsame")
162 
163 # print hist.GetName(), hist.GetTitle()
164  legname = hist.GetName()
165 
166  leg.AddEntry(hist, legname, 'lep')
167 
168 
169  for hist in hists:
170  hist.SetMaximum(ymax*2)
171 # hist.SetMinimum(ymin*0.5)
172 
173  leg.Draw()
174 
175  tex = TLatex( hists[-1].GetXaxis().GetXmin() + 0.01*(hists[-1].GetXaxis().GetXmax() - hists[-1].GetXaxis().GetXmin()), ymax*2.1, addon.replace('tau_',''))
176 
177  tex.SetTextFont(42)
178  tex.SetTextSize(0.03)
179  tex.Draw()
180 
181  tex2 = TLatex( hists[-1].GetXaxis().GetXmin() + 0.87*(hists[-1].GetXaxis().GetXmax() - hists[-1].GetXaxis().GetXmin()), ymax*2.1, tlabel)
182 
183  tex2.SetTextFont(42)
184  tex2.SetTextSize(0.03)
185  tex2.Draw()
186 
187 
188 
189  save(canvas, 'compare_' + runtype + '/' + header)
190 
191 
192 
def save(canvas, name)
Definition: compare.py:45
def LegendSettings(leg, ncolumn)
Definition: compare.py:51
def overlay(hists, ytitle, header, addon)
Definition: compare.py:120
def compare.save (   canvas,
  name 
)

Definition at line 45 of file compare.py.

References ensureDir(), and python.rootplot.root2matplotlib.replace().

Referenced by hoverlay(), and overlay().

45 def save(canvas, name):
46  ensureDir('compare_' + runtype)
47  canvas.SaveAs(name.replace(' ','').replace('&&','')+'.pdf')
48  canvas.SaveAs(name.replace(' ','').replace('&&','')+'.gif')
49 
50 
def replace(string, replacements)
def save(canvas, name)
Definition: compare.py:45
def ensureDir(directory)
Definition: compare.py:41

Variable Documentation

compare.argc

Definition at line 15 of file compare.py.

compare.argvs

Definition at line 14 of file compare.py.

compare.den_sel

Definition at line 326 of file compare.py.

compare.hist

Definition at line 373 of file compare.py.

compare.hists
compare.hvardict

Definition at line 348 of file compare.py.

compare.loose_id

Definition at line 305 of file compare.py.

compare.num_sel

Definition at line 325 of file compare.py.

compare.reco_cut

Definition at line 304 of file compare.py.

compare.runtype

Definition at line 21 of file compare.py.

Referenced by RPCTechTriggerConfig.RPCTechTriggerConfig(), and RunTypeDef.setRunType().

compare.sampledict

Definition at line 307 of file compare.py.

compare.tfile
compare.tlabel

Definition at line 25 of file compare.py.

Referenced by cscdqm::EventProcessor.getCSCFromMap().

compare.tree
compare.vardict

Definition at line 273 of file compare.py.

compare.xlabel

Definition at line 26 of file compare.py.