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 42 of file compare.py.

Referenced by save().

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

Definition at line 194 of file compare.py.

References LegendSettings(), and save().

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

Definition at line 52 of file compare.py.

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

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

Definition at line 63 of file compare.py.

References LegendSettings(), and str.

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

References mps_check.array.

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

Definition at line 121 of file compare.py.

References LegendSettings(), edm.print(), and save().

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

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

Definition at line 46 of file compare.py.

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

Referenced by hoverlay(), and overlay().

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

Variable Documentation

compare.argc

Definition at line 16 of file compare.py.

compare.argvs

Definition at line 15 of file compare.py.

compare.den_sel

Definition at line 327 of file compare.py.

compare.hist

Definition at line 374 of file compare.py.

compare.hists
compare.hvardict

Definition at line 349 of file compare.py.

compare.loose_id

Definition at line 306 of file compare.py.

compare.num_sel

Definition at line 326 of file compare.py.

compare.reco_cut

Definition at line 305 of file compare.py.

compare.runtype

Definition at line 22 of file compare.py.

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

compare.sampledict

Definition at line 308 of file compare.py.

compare.tfile
compare.tlabel

Definition at line 26 of file compare.py.

Referenced by cscdqm::EventProcessor.getCSCFromMap().

compare.tree
compare.vardict

Definition at line 274 of file compare.py.

compare.xlabel

Definition at line 27 of file compare.py.