def __init__(self, width=None, height=None,
xtitle=None, ytitle=None,
tick_length=15,
logy=False):
super(SimplePlot, self).__init__(width=width, height=height)
left, right, bottom, top = self.margin
self.SetMargin(0, 0, 0, 0)
# top pad for histograms
with self:
main = Pad(0., 0., 1., 1.)
if logy:
main.SetLogy()
main.margin = (left, right, bottom, top)
main.Draw()
# draw axes
with main:
main_hist = Hist(1, 0, 1)
main_hist.Draw('AXIS')
if xtitle is not None:
main_hist.xaxis.title = xtitle
if ytitle is not None:
main_hist.yaxis.title = ytitle
# set the tick lengths
tick_length_pixels(main, main_hist.xaxis, main_hist.yaxis,
tick_length)
self.main = main
self.main_hist = main_hist
self.logy = logy
def __init__(self, width=None, height=None,
xtitle=None, ytitle=None,
tick_length=15,
logy=False):
super(SimplePlot, self).__init__(width=width, height=height)
left, right, bottom, top = self.margin
self.SetMargin(0, 0, 0, 0)
# top pad for histograms
with self:
main = Pad(0., 0., 1., 1.)
if logy:
main.SetLogy()
main.margin = (left, right, bottom, top)
main.Draw()
# draw axes
with main:
main_hist = Hist(1, 0, 1)
main_hist.Draw('AXIS')
if xtitle is not None:
main_hist.xaxis.title = xtitle
if ytitle is not None:
main_hist.yaxis.title = ytitle
# set the tick lengths
tick_length_pixels(main, main_hist.xaxis, main_hist.yaxis,
tick_length)
self.main = main
self.main_hist = main_hist
self.logy = logy
def __init__(self, width=None, height=None,
offset=0,
ratio_height=None, ratio_margin=26,
ratio_limits=(0, 2), ratio_divisions=4,
prune_ratio_ticks=False,
ratio_line_values=(1,),
ratio_line_width=2,
ratio_line_style='dashed',
xtitle=None, ytitle=None, ratio_title=None,
tick_length=15,
logy=False):
# first init as normal canvas
super(RatioPlot, self).__init__(width=width, height=height)
# get margins in pixels
left, right, bottom, top = self.margin_pixels
default_height = self.height
default_frame_height = default_height - bottom - top
if ratio_height is None:
ratio_height = default_height / 4.
self.height += int(ratio_height) + ratio_margin + offset
self.margin = (0, 0, 0, 0)
main_height = default_frame_height + top + ratio_margin / 2. + offset
ratio_height += ratio_margin / 2. + bottom
# top pad for histograms
with self:
main = Pad(0., ratio_height / self.height, 1., 1.)
if logy:
main.SetLogy()
main.margin_pixels = (left, right, ratio_margin / 2., top)
main.Draw()
# bottom pad for ratio plot
with self:
ratio = Pad(0, 0, 1, ratio_height / self.height)
ratio.margin_pixels = (left, right, bottom, ratio_margin / 2.)
ratio.Draw()
# draw main axes
with main:
main_hist = Hist(1, 0, 1)
main_hist.Draw('AXIS')
# hide x-axis labels and title on main pad
xaxis, yaxis = main_hist.xaxis, main_hist.yaxis
xaxis.SetLabelOffset(1000)
xaxis.SetTitleOffset(1000)
# adjust y-axis title spacing
yaxis.SetTitleOffset(
yaxis.GetTitleOffset() * self.height / default_height)
# draw ratio axes
with ratio:
ratio_hist = Hist(1, 0, 1)
ratio_hist.Draw('AXIS')
# adjust x-axis label and title spacing
xaxis, yaxis = ratio_hist.xaxis, ratio_hist.yaxis
xaxis.SetLabelOffset(
xaxis.GetLabelOffset() * self.height / ratio_height)
xaxis.SetTitleOffset(
xaxis.GetTitleOffset() * self.height / ratio_height)
# adjust y-axis title spacing
yaxis.SetTitleOffset(
yaxis.GetTitleOffset() * self.height / default_height)
if ratio_limits is not None:
low, high = ratio_limits
if prune_ratio_ticks:
delta = 0.01 * (high - low) / float(ratio_divisions % 100)
low += delta
high -= delta
yaxis.SetLimits(low, high)
yaxis.SetRangeUser(low, high)
yaxis.SetNdivisions(ratio_divisions)
if xtitle is not None:
ratio_hist.xaxis.title = xtitle
if ytitle is not None:
main_hist.yaxis.title = ytitle
if ratio_title is not None:
ratio_hist.yaxis.title = ratio_title
# set the tick lengths
tick_length_pixels(main, main_hist.xaxis, main_hist.yaxis,
tick_length)
tick_length_pixels(ratio, ratio_hist.xaxis, ratio_hist.yaxis,
tick_length)
# draw ratio lines
lines = []
if ratio_line_values:
with ratio:
for value in ratio_line_values:
line = Line(0, value, 1, value)
line.linestyle = ratio_line_style
line.linewidth = ratio_line_width
line.Draw()
lines.append(line)
self.lines = lines
self.main = main
self.main_hist = main_hist
self.ratio = ratio
self.ratio_hist = ratio_hist
self.ratio_limits = ratio_limits
self.logy = logy
def __init__(self,
width=None,
height=None,
offset=0,
ratio_height=None,
ratio_margin=26,
ratio_limits=(0, 2),
ratio_divisions=4,
prune_ratio_ticks=False,
ratio_line_values=(1, ),
ratio_line_width=2,
ratio_line_style='dashed',
xtitle=None,
ytitle=None,
ratio_title=None,
tick_length=15,
logy=False):
# first init as normal canvas
super(RatioPlot, self).__init__(width=width, height=height)
# get margins in pixels
left, right, bottom, top = self.margin_pixels
default_height = self.height
default_frame_height = default_height - bottom - top
if ratio_height is None:
ratio_height = default_height / 4.
self.height += int(ratio_height) + ratio_margin + offset
self.margin = (0, 0, 0, 0)
main_height = default_frame_height + top + ratio_margin / 2. + offset
ratio_height += ratio_margin / 2. + bottom
# top pad for histograms
with self:
main = Pad(0., ratio_height / self.height, 1., 1.)
if logy:
main.SetLogy()
main.margin_pixels = (left, right, ratio_margin / 2., top)
main.Draw()
# bottom pad for ratio plot
with self:
ratio = Pad(0, 0, 1, ratio_height / self.height)
ratio.margin_pixels = (left, right, bottom, ratio_margin / 2.)
ratio.Draw()
# draw main axes
with main:
main_hist = Hist(1, 0, 1)
main_hist.Draw('AXIS')
# hide x-axis labels and title on main pad
xaxis, yaxis = main_hist.xaxis, main_hist.yaxis
xaxis.SetLabelOffset(1000)
xaxis.SetTitleOffset(1000)
# adjust y-axis title spacing
yaxis.SetTitleOffset(yaxis.GetTitleOffset() * self.height /
default_height)
# draw ratio axes
with ratio:
ratio_hist = Hist(1, 0, 1)
ratio_hist.Draw('AXIS')
# adjust x-axis label and title spacing
xaxis, yaxis = ratio_hist.xaxis, ratio_hist.yaxis
xaxis.SetLabelOffset(xaxis.GetLabelOffset() * self.height /
ratio_height)
xaxis.SetTitleOffset(xaxis.GetTitleOffset() * self.height /
ratio_height)
# adjust y-axis title spacing
yaxis.SetTitleOffset(yaxis.GetTitleOffset() * self.height /
default_height)
if ratio_limits is not None:
low, high = ratio_limits
if prune_ratio_ticks:
delta = 0.01 * (high - low) / float(ratio_divisions % 100)
low += delta
high -= delta
yaxis.SetLimits(low, high)
yaxis.SetRangeUser(low, high)
yaxis.SetNdivisions(ratio_divisions)
if xtitle is not None:
ratio_hist.xaxis.title = xtitle
if ytitle is not None:
main_hist.yaxis.title = ytitle
if ratio_title is not None:
ratio_hist.yaxis.title = ratio_title
# set the tick lengths
tick_length_pixels(main, main_hist.xaxis, main_hist.yaxis, tick_length)
tick_length_pixels(ratio, ratio_hist.xaxis, ratio_hist.yaxis,
tick_length)
# draw ratio lines
lines = []
if ratio_line_values:
with ratio:
for value in ratio_line_values:
line = Line(0, value, 1, value)
line.linestyle = ratio_line_style
line.linewidth = ratio_line_width
line.Draw()
lines.append(line)
self.lines = lines
self.main = main
self.main_hist = main_hist
self.ratio = ratio
self.ratio_hist = ratio_hist
self.ratio_limits = ratio_limits
self.logy = logy
def __init__(self, width=None, height=None,
ratio_height=0.2, ratio_margin=0.05,
ratio_range=(0, 2), ratio_divisions=4,
ratio_line_values=(1,),
ratio_line_width=2,
ratio_line_style='dashed',
xtitle=None, ytitle=None, ratio_title=None,
tick_length=20):
style = ROOT.gStyle
# plot dimensions in pixels
if height is not None:
figheight = baseheight = height
else:
figheight = baseheight = style.GetCanvasDefH()
if width is not None:
figwidth = basewidth = width
else:
figwidth = basewidth = style.GetCanvasDefW()
# margins
left_margin = style.GetPadLeftMargin()
bottom_margin = style.GetPadBottomMargin()
top_margin = style.GetPadTopMargin()
right_margin = style.GetPadRightMargin()
figheight += (ratio_height + ratio_margin) * figheight
ratio_height += bottom_margin + ratio_margin / 2.
super(RatioPlot, self).__init__(
width=int(figwidth), height=int(figheight))
self.SetMargin(0, 0, 0, 0)
# top pad for histograms
with self:
main = Pad(0., ratio_height, 1., 1.)
main.SetBottomMargin(ratio_margin / 2.)
main.SetTopMargin(top_margin)
main.SetLeftMargin(left_margin)
main.SetRightMargin(right_margin)
main.Draw()
# bottom pad for ratio plot
with self:
ratio = Pad(0, 0, 1, ratio_height)
ratio.SetBottomMargin(bottom_margin / ratio_height)
ratio.SetTopMargin(ratio_margin / (2. * ratio_height))
ratio.SetLeftMargin(left_margin)
ratio.SetRightMargin(right_margin)
ratio.Draw()
# draw main axes
with main:
main_hist = Hist(1, 0, 1)
main_hist.Draw('AXIS')
# hide x-axis labels and title on main pad
xaxis, yaxis = main_hist.xaxis, main_hist.yaxis
xaxis.SetLabelOffset(1000)
xaxis.SetTitleOffset(1000)
# adjust y-axis title spacing
yaxis.SetTitleOffset(
yaxis.GetTitleOffset() * figheight / baseheight)
# draw ratio axes
with ratio:
ratio_hist = Hist(1, 0, 1)
ratio_hist.Draw('AXIS')
# adjust x-axis label and title spacing
xaxis, yaxis = ratio_hist.xaxis, ratio_hist.yaxis
xaxis.SetLabelOffset(
xaxis.GetLabelOffset() / ratio_height)
xaxis.SetTitleOffset(
xaxis.GetTitleOffset() / ratio_height)
# adjust y-axis title spacing
yaxis.SetTitleOffset(
yaxis.GetTitleOffset() * figheight / baseheight)
if ratio_range is not None:
yaxis.SetLimits(*ratio_range)
yaxis.SetRangeUser(*ratio_range)
yaxis.SetNdivisions(ratio_divisions)
if xtitle is not None:
ratio_hist.xaxis.title = xtitle
if ytitle is not None:
main_hist.yaxis.title = ytitle
if ratio_title is not None:
ratio_hist.yaxis.title = ratio_title
# set the tick lengths
tick_length_pixels(main, main_hist.xaxis, main_hist.yaxis,
tick_length)
tick_length_pixels(ratio, ratio_hist.xaxis, ratio_hist.yaxis,
tick_length)
# draw ratio lines
lines = []
if ratio_line_values:
with ratio:
for value in ratio_line_values:
line = Line(0, value, 1, value)
line.linestyle = ratio_line_style
line.linewidth = ratio_line_width
line.Draw()
lines.append(line)
self.lines = lines
self.main = main
self.main_hist = main_hist
self.ratio = ratio
self.ratio_hist = ratio_hist
self.ratio_range = ratio_range
