def plotRatio(var,name,hist1,hist2):
"""Plot minimal ratio of two histograms."""
canvas = TCanvas('canvas','canvas',100,100,800,800)
canvas.Divide(2)
canvas.cd(1)
gPad.SetPad('pad1','pad1',0,0.42,1,1,0,-1,0)
gPad.SetTopMargin( 0.10 ); gPad.SetBottomMargin( 0.01 )
hist1.Draw()
hist2.Draw('SAME')
hist1.SetLineColor(kBlue)
hist2.SetLineColor(kRed)
hist1.SetLineWidth(2)
hist2.SetLineWidth(2)
hist1.SetLineStyle(1)
hist2.SetLineStyle(2)
hist1.Draw('HIST')
hist2.Draw('HIST SAMES')
gPad.Update()
stats1 = hist1.GetListOfFunctions().FindObject('stats')
stats2 = hist2.GetListOfFunctions().FindObject('stats')
stats1.SetY1NDC(.74); stats1.SetY2NDC(.94)
stats2.SetY1NDC(.50); stats2.SetY2NDC(.70)
stats1.Draw()
stats2.Draw()
canvas.cd(2)
gPad.SetPad('pad2','pad2',0,0,1,0.41,0,-1,0)
gPad.SetTopMargin( 0.05 ); gPad.SetBottomMargin( 0.24 )
ratio = hist1.Clone('ratio')
ratio.Divide(hist2)
for i, (y1, y2, r) in enumerate(zip(hist1,hist2,ratio),0):
if hist1.GetBinContent(i)==0 and hist2.GetBinContent(i)==0:
ratio.SetBinContent(i,1)
ratio.GetXaxis().SetTitle(var)
ratio.GetXaxis().SetLabelSize(0.045)
ratio.GetYaxis().SetLabelSize(0.045)
ratio.GetXaxis().SetTitleSize(0.060)
ratio.SetMinimum(0.2)
ratio.SetMaximum(1.8)
ratio.Draw()
statsr = ratio.GetListOfFunctions().FindObject('stats')
statsr.SetY1NDC(.65); statsr.SetY2NDC(.98)
canvas.SaveAs(name)
canvas.Close()
gDirectory.Delete(hist1.GetName())
gDirectory.Delete(hist2.GetName())
gDirectory.Delete(ratio.GetName())
def setcanvas(self, **kwargs):
"""Make canvas and pads for ratio plots."""
square = kwargs.get('square', False)
double = kwargs.get('ratio', False) # include lower panel
width = kwargs.get('width', 900 if square else 800 if double else 800)
height = kwargs.get('height',
900 if square else 750 if double else 600)
lmargin = kwargs.get('lmargin', 1.)
rmargin = kwargs.get('rmargin', 1.)
tmargin = kwargs.get('tmargin', 1.)
bmargin = kwargs.get('bmargin', 1.)
pads = kwargs.get('pads', []) # pass list as reference
#if not CMSStyle.lumi_13TeV:
# tmargin *= 0.7
if square:
lmargin *= 1.15
tmargin *= 0.90
#rmargin *= 3.6
#CMSStyle.relPosX = 0.15
canvas = TCanvas('canvas', 'canvas', 100, 100, width, height)
canvas.SetFillColor(0)
#canvas.SetFillStyle(0)
canvas.SetBorderMode(0)
canvas.SetFrameBorderMode(0)
if double:
canvas.SetMargin(0.0, 0.0, 0.0, 0.0) # LRBT
canvas.Divide(2)
canvas.cd(1)
gPad.SetPad('pad1', 'pad1', 0.0, 0.33, 1.0, 1.0)
gPad.SetMargin(0.145 * lmargin, 0.04 * rmargin, 0.029,
0.075 * tmargin)
gPad.SetFillColor(0)
gPad.SetFillStyle(4000) # transparant (for pdf)
#gPad.SetFillStyle(0)
gPad.SetBorderMode(0)
gPad.Draw()
canvas.cd(2)
gPad.SetPad('pad2', 'pad2', 0.0, 0.0, 1.0, 0.33)
gPad.SetMargin(0.145 * lmargin, 0.04 * rmargin, 0.355 * bmargin,
0.04)
gPad.SetFillColor(0) #gPad.SetFillColorAlpha(0,0.0)
gPad.SetFillStyle(4000) # transparant (for pdf)
gPad.SetBorderMode(0)
gPad.Draw()
canvas.cd(1)
else:
canvas.SetMargin(0.145 * lmargin, 0.05 * rmargin, 0.145 * bmargin,
0.06 * tmargin)
return canvas
#print(reso[2])
#
deltaE_f.append( (reso[0] - energyNominal)/energyNominal )
# resolution (relative error)
relEnergyErrorVec.append( reso[1]/reso[0] )
#
relEnergyErrorErrorVec.append( reso[3]/reso[0] )
small=1.e-5
c1.Divide(1,2,small,small)
c1.cd(1)
gPad.SetPad(small, 0.3, 1.-small, 1.-small)
gPad.SetBottomMargin(small)
labelSize=10
makerSize = 0.3
gStyle.SetEndErrorSize(0)
gr = TGraphErrors( n, energyVec, recoEnergyVec, energyErrorVec, recoEnergyErrorVec )
gr.SetName('gr')
gr.SetLineColor( 1 )
gr.SetLineWidth( 1 )
gr.SetLineStyle( 2 )
gr.SetMarkerColor( 2 )
gr.SetMarkerStyle( 20 )
gr.SetMarkerSize( makerSize )
def MakeOneDHist(pathToDir, distribution):
numFittingSamples = 0
HeaderLabel = TPaveLabel(header_x_left, header_y_bottom, header_x_right,
header_y_top, HeaderText, "NDC")
HeaderLabel.SetTextAlign(32)
HeaderLabel.SetBorderSize(0)
HeaderLabel.SetFillColor(0)
HeaderLabel.SetFillStyle(0)
LumiLabel = TPaveLabel(topLeft_x_left, topLeft_y_bottom, topLeft_x_right,
topLeft_y_top, LumiText, "NDC")
LumiLabel.SetBorderSize(0)
LumiLabel.SetFillColor(0)
LumiLabel.SetFillStyle(0)
NormLabel = TPaveLabel()
NormLabel.SetDrawOption("NDC")
NormLabel.SetX1NDC(topLeft_x_left)
NormLabel.SetX2NDC(topLeft_x_right)
NormLabel.SetBorderSize(0)
NormLabel.SetFillColor(0)
NormLabel.SetFillStyle(0)
NormText = ""
if arguments.normalizeToUnitArea:
NormText = "Scaled to unit area"
elif arguments.normalizeToData:
NormText = "MC scaled to data"
NormLabel.SetLabel(NormText)
YieldsLabel = TPaveText(0.39, 0.7, 0.59, 0.9, "NDC")
YieldsLabel.SetBorderSize(0)
YieldsLabel.SetFillColor(0)
YieldsLabel.SetFillStyle(0)
YieldsLabel.SetTextAlign(12)
RatiosLabel = TPaveText()
RatiosLabel.SetDrawOption("NDC")
RatiosLabel.SetBorderSize(0)
RatiosLabel.SetFillColor(0)
RatiosLabel.SetFillStyle(0)
RatiosLabel.SetTextAlign(32)
Legend = TLegend()
Legend.SetBorderSize(0)
Legend.SetFillColor(0)
Legend.SetFillStyle(0)
fittingIntegral = 0
scaleFactor = 1
HistogramsToFit = []
TargetDataset = distribution['target_dataset']
FittingLegendEntries = []
DataLegendEntries = []
FittingHistogramDatasets = []
Stack_list = []
Stack_list.append(THStack("stack_before", distribution['name']))
Stack_list.append(THStack("stack_after", distribution['name']))
fileName = condor_dir + "/" + distribution['target_dataset'] + ".root"
if not os.path.exists(fileName):
return
inputFile = TFile(fileName)
if inputFile.IsZombie() or not inputFile.GetNkeys():
return
Target = inputFile.Get("OSUAnalysis/" + distribution['channel'] + "/" +
distribution['name']).Clone()
Target.SetDirectory(0)
inputFile.Close()
Target.SetMarkerStyle(20)
Target.SetMarkerSize(0.8)
Target.SetFillStyle(0)
Target.SetLineColor(colors[TargetDataset])
Target.SetLineStyle(1)
Target.SetLineWidth(2)
targetIntegral = Target.Integral()
if (arguments.normalizeToUnitArea and Target.Integral() > 0):
Target.Scale(1. / Target.Integral())
if arguments.rebinFactor:
RebinFactor = int(arguments.rebinFactor)
#don't rebin histograms which will have less than 5 bins or any gen-matching histograms
if Target.GetNbinsX() >= RebinFactor * 5 and Target.GetName().find(
"GenMatch") is -1:
Target.Rebin(RebinFactor)
### formatting target histogram and adding to legend
legendIndex = 0
Legend.AddEntry(Target, labels[TargetDataset], "LEP")
legendIndex = legendIndex + 1
if not outputFile.Get("OSUAnalysis"):
outputFile.mkdir("OSUAnalysis")
if not outputFile.Get("OSUAnalysis/" + distribution['channel']):
outputFile.Get("OSUAnalysis").mkdir(distribution['channel'])
for sample in distribution[
'datasets']: # loop over different samples requested to be fit
dataset_file = "%s/%s.root" % (condor_dir, sample)
inputFile = TFile(dataset_file)
HistogramObj = inputFile.Get(pathToDir + "/" +
distribution['channel'] + "/" +
distribution['name'])
if not HistogramObj:
print "WARNING: Could not find histogram " + pathToDir + "/" + distribution[
'channel'] + "/" + distribution[
'name'] + " in file " + dataset_file + ". Will skip it and continue."
continue
Histogram = HistogramObj.Clone()
Histogram.SetDirectory(0)
inputFile.Close()
if arguments.rebinFactor:
RebinFactor = int(arguments.rebinFactor)
#don't rebin histograms which will have less than 5 bins or any gen-matching histograms
if Histogram.GetNbinsX() >= RebinFactor * 5 and Histogram.GetName(
).find("GenMatch") is -1:
Histogram.Rebin(RebinFactor)
xAxisLabel = Histogram.GetXaxis().GetTitle()
unitBeginIndex = xAxisLabel.find("[")
unitEndIndex = xAxisLabel.find("]")
if unitBeginIndex is not -1 and unitEndIndex is not -1: #x axis has a unit
yAxisLabel = "Entries / " + str(Histogram.GetXaxis().GetBinWidth(
1)) + " " + xAxisLabel[unitBeginIndex + 1:unitEndIndex]
else:
yAxisLabel = "Entries per bin (" + str(
Histogram.GetXaxis().GetBinWidth(1)) + " width)"
if not arguments.makeFancy:
histoTitle = Histogram.GetTitle()
else:
histoTitle = ""
legLabel = labels[sample]
if (arguments.printYields):
yieldHist = Histogram.Integral()
legLabel = legLabel + " (%.1f)" % yieldHist
FittingLegendEntries.append(legLabel)
if (types[sample] == "bgMC"):
numFittingSamples += 1
fittingIntegral += Histogram.Integral()
Histogram.SetLineStyle(1)
if (arguments.noStack):
Histogram.SetFillStyle(0)
Histogram.SetLineColor(colors[sample])
Histogram.SetLineWidth(2)
else:
Histogram.SetFillStyle(1001)
Histogram.SetFillColor(colors[sample])
Histogram.SetLineColor(1)
Histogram.SetLineWidth(1)
elif (types[sample] == "signalMC"):
numFittingSamples += 1
Histogram.SetFillStyle(0)
Histogram.SetLineColor(colors[sample])
Histogram.SetLineStyle(1)
Histogram.SetLineWidth(2)
if (arguments.normalizeToUnitArea and Histogram.Integral() > 0):
Histogram.Scale(1. / Histogram.Integral())
HistogramsToFit.append(Histogram)
FittingHistogramDatasets.append(sample)
#scaling histograms as per user's specifications
if targetIntegral > 0 and fittingIntegral > 0:
scaleFactor = targetIntegral / fittingIntegral
for fittingHist in HistogramsToFit:
if arguments.normalizeToData:
fittingHist.Scale(scaleFactor)
if arguments.normalizeToUnitArea and not arguments.noStack and fittingIntegral > 0:
fittingHist.Scale(1. / fittingIntegral)
elif arguments.normalizeToUnitArea and arguments.noStack and fittingHist.Integral(
) > 0:
fittingHist.Scale(1. / fittingHist.Integral())
def fitf(x, par):
xBin = HistogramsToFit[0].FindBin(x[0])
value = 0.0
for i in range(0, len(HistogramsToFit)):
value += par[i] * HistogramsToFit[i].GetBinContent(xBin) + par[
i +
len(HistogramsToFit)] * HistogramsToFit[i].GetBinError(xBin)
return value
lowerLimit = Target.GetBinLowEdge(1)
upperLimit = Target.GetBinLowEdge(Target.GetNbinsX()) + Target.GetBinWidth(
Target.GetNbinsX())
if 'lowerLimit' in distribution:
lowerLimit = distribution['lowerLimit']
if 'upperLimit' in distribution:
upperLimit = distribution['upperLimit']
func = TF1("fit", fitf, lowerLimit, upperLimit, 2 * len(HistogramsToFit))
for i in range(0, len(HistogramsToFit)):
if 'fixed_datasets' in distribution and distribution['datasets'][
i] in distribution['fixed_datasets']:
func.FixParameter(i, 1.0)
else:
func.SetParameter(i, 1.0)
# func.SetParLimits (i, 0.0, 1.0e2) # comment this out so we don't have to pre-normalize the QCD input sample
func.SetParName(i, labels[FittingHistogramDatasets[i]])
shiftedScaleFactors = []
if arguments.parametricErrors:
# loop over all input histograms and shift them +- 1 sigma
for i in range(0, len(HistogramsToFit)):
sfs = []
# -1 => -1 sigma, +1 => +1 sigma
for j in [-1, 1]:
# loop over the parameters holding the errors for each dataset, fixing all to 0
for k in range(len(HistogramsToFit), 2 * len(HistogramsToFit)):
func.FixParameter(k, 0)
# fix the error of the dataset of interest to +-1
func.FixParameter(i + len(HistogramsToFit), j)
# perform new fit
for k in range(0, distribution['iterations'] - 1):
if j == -1:
print "Scale down " + labels[FittingHistogramDatasets[
i]] + " iteration " + str(k + 1) + "..."
if j == 1:
print "Scale up " + labels[FittingHistogramDatasets[
i]] + " iteration " + str(k + 1) + "..."
Target.Fit("fit", "QEMR0")
Target.Fit("fit", "VEMR0")
# save the new scale factors for each dataset
for k in range(0, len(HistogramsToFit)):
sfs.append(func.GetParameter(k))
shiftedScaleFactors.append(sfs)
# reset the parameters with the errors of each dataset to 0
for i in range(len(HistogramsToFit), 2 * len(HistogramsToFit)):
func.FixParameter(i, 0)
# do the fit to get the central values
for i in range(0, distribution['iterations'] - 1):
print "Iteration " + str(i + 1) + "..."
Target.Fit("fit", "QEMR0")
Target.Fit("fit", "VEMR0")
if arguments.parametricErrors:
# make a list of the largest errors on each contribution by shifting any other contribution
parErrors = []
# loop over all the datasets
for i in range(0, len(HistogramsToFit)):
centralValue = func.GetParameter(i)
maxError = 0
# find the maximum deviation from the central value and save that
for shiftedScaleFactor in shiftedScaleFactors[i]:
currentError = abs(shiftedScaleFactor - centralValue)
if currentError > maxError:
maxError = currentError
parErrors.append(maxError)
finalMax = 0
if not arguments.noStack:
for fittingHist in HistogramsToFit:
finalMax += fittingHist.GetMaximum()
else:
for fittingHist in HistogramsToFit:
if (fittingHist.GetMaximum() > finalMax):
finalMax = fittingHist.GetMaximum()
if (Target.GetMaximum() > finalMax):
finalMax = Target.GetMaximum()
Target.SetMaximum(1.1 * finalMax)
Target.SetMinimum(0.0001)
Canvas = TCanvas(distribution['name'] + "_FitFunction")
Canvas.cd(1)
Target.Draw()
func.Draw("same")
outputFile.cd("OSUAnalysis/" + distribution['channel'])
Canvas.Write()
if arguments.savePDFs:
if histogram == input_histograms[0]:
Canvas.Print(pdfFileName + "(", "pdf")
else:
Canvas.Print(pdfFileName, "pdf")
Target.SetStats(0)
### formatting bgMC histograms and adding to legend
legendIndex = numFittingSamples - 1
for Histogram in reversed(HistogramsToFit):
if (arguments.noStack):
Legend.AddEntry(Histogram, FittingLegendEntries[legendIndex], "L")
else:
Legend.AddEntry(Histogram, FittingLegendEntries[legendIndex], "F")
legendIndex = legendIndex - 1
### Drawing histograms to canvas
makeRatioPlots = arguments.makeRatioPlots
makeDiffPlots = arguments.makeDiffPlots
yAxisMin = 0.0001
if arguments.setYMin:
yAxisMin = float(arguments.setYMin)
### Draw everything to the canvases !!!!
for i in range(0, 2): # 0 => before, 1 => after
integrals = []
ratios = []
errors = []
if i == 1:
# loop over each dataset, saving it's yield and the errors on it
for j in range(0, len(HistogramsToFit)):
integrals.append(HistogramsToFit[j].Integral())
HistogramsToFit[j].Scale(func.GetParameter(j))
ratios.append(func.GetParameter(j))
errors.append(func.GetParError(j))
for fittingHist in HistogramsToFit:
if not arguments.noStack:
Stack_list[i].Add(fittingHist)
#creating the histogram to represent the statistical errors on the stack
if not arguments.noStack:
ErrorHisto = HistogramsToFit[0].Clone("errors")
ErrorHisto.SetFillStyle(3001)
ErrorHisto.SetFillColor(13)
ErrorHisto.SetLineWidth(0)
if i == 1:
Legend.AddEntry(ErrorHisto, "Stat. Errors", "F")
for Histogram in HistogramsToFit:
if Histogram is not HistogramsToFit[0]:
ErrorHisto.Add(Histogram)
if i == 0:
Canvas = TCanvas(distribution['name'] + "_Before")
if i == 1:
Canvas = TCanvas(distribution['name'] + "_After")
if makeRatioPlots or makeDiffPlots:
Canvas.SetFillStyle(0)
Canvas.Divide(1, 2)
Canvas.cd(1)
gPad.SetPad(0, 0.25, 1, 1)
gPad.SetMargin(0.15, 0.05, 0.01, 0.07)
gPad.SetFillStyle(0)
gPad.Update()
gPad.Draw()
if arguments.setLogY:
gPad.SetLogy()
Canvas.cd(2)
gPad.SetPad(0, 0, 1, 0.25)
# format: gPad.SetMargin(l,r,b,t)
gPad.SetMargin(0.15, 0.05, 0.4, 0.01)
gPad.SetFillStyle(0)
gPad.SetGridy(1)
gPad.Update()
gPad.Draw()
Canvas.cd(1)
### finding the maximum value of anything going on the canvas, so we know how to set the y-axis
finalMax = 0
if numFittingSamples is not 0 and not arguments.noStack:
finalMax = ErrorHisto.GetMaximum() + ErrorHisto.GetBinError(
ErrorHisto.GetMaximumBin())
else:
for bgMCHist in HistogramsToFit:
if (bgMCHist.GetMaximum() > finalMax):
finalMax = bgMCHist.GetMaximum()
if (Target.GetMaximum() > finalMax):
finalMax = Target.GetMaximum() + Target.GetBinError(
Target.GetMaximumBin())
finalMax = 1.15 * finalMax
if arguments.setYMax:
finalMax = float(arguments.setYMax)
if not arguments.noStack: # draw stacked background samples
Stack_list[i].SetTitle(histoTitle)
Stack_list[i].Draw("HIST")
Stack_list[i].GetXaxis().SetTitle(xAxisLabel)
Stack_list[i].GetYaxis().SetTitle(yAxisLabel)
Stack_list[i].SetMaximum(finalMax)
Stack_list[i].SetMinimum(yAxisMin)
if makeRatioPlots or makeDiffPlots:
Stack_list[i].GetHistogram().GetXaxis().SetLabelSize(0)
#draw shaded error bands
ErrorHisto.Draw("A E2 SAME")
else: #draw the unstacked backgrounds
HistogramsToFit[0].SetTitle(histoTitle)
HistogramsToFit[0].Draw("HIST")
HistogramsToFit[0].GetXaxis().SetTitle(xAxisLabel)
HistogramsToFit[0].GetYaxis().SetTitle(yAxisLabel)
HistogramsToFit[0].SetMaximum(finalMax)
HistogramsToFit[0].SetMinimum(yAxisMin)
for bgMCHist in HistogramsToFit:
bgMCHist.Draw("A HIST SAME")
Target.Draw("A E X0 SAME")
#legend coordinates, empirically determined :-)
x_left = 0.6761745
x_right = 0.9328859
x_width = x_right - x_left
y_max = 0.9
entry_height = 0.05
if (numFittingSamples is not 0): #then draw the data & bgMC legend
numExtraEntries = 2 # count the target and (lack of) title
Legend.SetX1NDC(x_left)
numExtraEntries = numExtraEntries + 1 # count the stat. errors entry
Legend.SetY1NDC(y_max - entry_height *
(numExtraEntries + numFittingSamples))
Legend.SetX2NDC(x_right)
Legend.SetY2NDC(y_max)
Legend.Draw()
RatiosLabel.SetX1NDC(x_left - 0.1)
RatiosLabel.SetX2NDC(x_right)
RatiosLabel.SetY2NDC(Legend.GetY1NDC() - 0.1)
RatiosLabel.SetY1NDC(RatiosLabel.GetY2NDC() - entry_height *
(numFittingSamples))
# Deciding which text labels to draw and drawing them
drawLumiLabel = False
drawNormLabel = False
offsetNormLabel = False
drawHeaderLabel = False
if not arguments.normalizeToUnitArea: #don't draw the lumi label if there's no data and it's scaled to unit area
drawLumiLabel = True
# move the normalization label down before drawing if we drew the lumi. label
offsetNormLabel = True
if arguments.normalizeToUnitArea or arguments.normalizeToData:
drawNormLabel = True
if arguments.makeFancy:
drawHeaderLabel = True
drawLumiLabel = False
# now that flags are set, draw the appropriate labels
if drawLumiLabel:
LumiLabel.Draw()
if drawNormLabel:
if offsetNormLabel:
NormLabel.SetY1NDC(topLeft_y_bottom - topLeft_y_offset)
NormLabel.SetY2NDC(topLeft_y_top - topLeft_y_offset)
else:
NormLabel.SetY1NDC(topLeft_y_bottom)
NormLabel.SetY2NDC(topLeft_y_top)
NormLabel.Draw()
if drawHeaderLabel:
HeaderLabel.Draw()
YieldsLabel.Clear()
mcYield = Stack_list[i].GetStack().Last().Integral()
dataYield = Target.Integral()
if i == 0:
YieldsLabel.AddText("Before Fit to Data")
if i == 1:
YieldsLabel.AddText("After Fit to Data")
YieldsLabel.AddText("data yield: " + '%.1f' % dataYield)
YieldsLabel.AddText("bkgd yield: " + '%.1f' % mcYield)
YieldsLabel.AddText("data/bkgd: " + '%.2f' % (dataYield / mcYield))
if i == 1:
for j in range(0, len(FittingLegendEntries)):
if abs(ratios[j] - 1) < 0.001 and abs(
errors[j]
) < 0.001: #then it probably was held fixed
continue
if arguments.showFittedYields:
yield_ = ratios[j] * integrals[j]
yielderror_ = errors[j] * yield_
text = FittingLegendEntries[
j] + " yield: " + '%.0f' % yield_ + ' #pm %.0f' % yielderror_
else:
text = FittingLegendEntries[
j] + " ratio: " + '%.2f' % ratios[
j] + ' #pm %.2f' % errors[j]
text = text + " (fit)"
if arguments.parametricErrors:
yield_ = ratios[j] * integrals[j]
yieldParError_ = parErrors[j] * yield_
if arguments.showFittedYields:
text += ' #pm %.2f' % yieldParError_
else:
text += ' #pm %.2f' % parErrors[j]
text = text + " (sys)"
RatiosLabel.AddText(text)
YieldsLabel.Draw()
RatiosLabel.Draw()
# drawing the ratio or difference plot if requested
if (makeRatioPlots or makeDiffPlots):
Canvas.cd(2)
BgSum = Stack_list[i].GetStack().Last()
if makeRatioPlots:
if arguments.ratioRelErrMax:
Comparison = ratioHistogram(Target, BgSum,
arguments.ratioRelErrMax)
else:
Comparison = ratioHistogram(Target, BgSum)
elif makeDiffPlots:
Comparison = Target.Clone("diff")
Comparison.Add(BgSum, -1)
Comparison.SetTitle("")
Comparison.GetYaxis().SetTitle("Data-Bkgd")
Comparison.GetXaxis().SetTitle(xAxisLabel)
Comparison.GetYaxis().CenterTitle()
Comparison.GetYaxis().SetTitleSize(0.1)
Comparison.GetYaxis().SetTitleOffset(0.5)
Comparison.GetXaxis().SetTitleSize(0.15)
Comparison.GetYaxis().SetLabelSize(0.1)
Comparison.GetXaxis().SetLabelSize(0.15)
if makeRatioPlots:
RatioYRange = 1.15
if arguments.ratioYRange:
RatioYRange = float(arguments.ratioYRange)
Comparison.GetYaxis().SetRangeUser(-1 * RatioYRange,
RatioYRange)
elif makeDiffPlots:
YMax = Comparison.GetMaximum()
YMin = Comparison.GetMinimum()
if YMax <= 0 and YMin <= 0:
Comparison.GetYaxis().SetRangeUser(-1.2 * YMin, 0)
elif YMax >= 0 and YMin >= 0:
Comparison.GetYaxis().SetRangeUser(0, 1.2 * YMax)
else: #axis crosses y=0
if abs(YMax) > abs(YMin):
Comparison.GetYaxis().SetRangeUser(
-1.2 * YMax, 1.2 * YMax)
else:
Comparison.GetYaxis().SetRangeUser(
-1.2 * YMin, 1.2 * YMin)
Comparison.GetYaxis().SetNdivisions(205)
Comparison.Draw("E0")
if i == 0:
Canvas.Write(distribution['name'] + "_Before")
if arguments.savePDFs:
pathToDirString = plainTextString(pathToDir)
Canvas.SaveAs(condor_dir + "/fitting_histogram_pdfs/" +
pathToDirString + "/" + distribution['name'] +
"_Before.pdf")
if i == 1:
Canvas.Write(distribution['name'] + "_After")
if arguments.savePDFs:
pathToDirString = plainTextString(pathToDir)
Canvas.SaveAs(condor_dir + "/fitting_histogram_pdfs/" +
pathToDirString + "/" + distribution['name'] +
"_After.pdf")
def rooFit109():
print ">>> setup model..."
x = RooRealVar("x", "x", -10, 10)
sigma = RooRealVar("sigma", "sigma", 3, 0.1, 10)
mean = RooRealVar("mean", "mean", 0, -10, 10)
gauss = RooGaussian("gauss", "gauss", x, RooConst(0),
sigma) # RooConst(0) gives segfaults
data = gauss.generate(RooArgSet(x), 100000) # RooDataSet
#sigma = 3.15 # overwrites RooRealVar with a float
sigma.setVal(3.15)
print ">>> plot data and slightly distorted model..."
frame1 = x.frame(Title("Data with distorted Gaussian pdf"),
Bins(40)) # RooPlot
data.plotOn(frame1, DataError(RooAbsData.SumW2))
gauss.plotOn(frame1)
print ">>> calculate chi^2..."
# Show the chi^2 of the curve w.r.t. the histogram
# If multiple curves or datasets live in the frame you can specify
# the name of the relevant curve and/or dataset in chiSquare()
print ">>> chi^2 = %.2f" % frame1.chiSquare()
print ">>> construct histograms with the residuals and pull of the data wrt the curve"
hresid = frame1.residHist() # RooHist
hpull = frame1.pullHist() # RooHist
frame2 = x.frame(Title("Residual Distribution")) # RooPlot
frame2.addPlotable(hresid, "P")
frame3 = x.frame(Title("Pull Distribution")) # RooPlot
frame3.addPlotable(hpull, "P")
print ">>> draw functions and toy data on canvas..."
canvas = TCanvas("canvas", "canvas", 100, 100, 2000, 400)
canvas.Divide(3)
for i, frame in enumerate([frame1, frame2, frame3], 1):
canvas.cd(i)
gPad.SetLeftMargin(0.14)
gPad.SetRightMargin(0.04)
frame.GetYaxis().SetTitleOffset(1.5)
frame.GetYaxis().SetLabelOffset(0.010)
frame.GetYaxis().SetTitleSize(0.045)
frame.GetYaxis().SetLabelSize(0.042)
frame.GetXaxis().SetTitleSize(0.045)
frame.GetXaxis().SetLabelSize(0.042)
frame.Draw()
canvas.SaveAs("rooFit109.png")
canvas.Close()
# ratio/pull/residual plot
print ">>> draw with pull plot..."
canvas = TCanvas("canvas", "canvas", 100, 100, 1000, 1000)
canvas.Divide(2)
canvas.cd(1)
gPad.SetPad("pad1", "pad1", 0, 0.33, 1, 1, 0, -1, 0)
gPad.SetTopMargin(0.10)
gPad.SetBottomMargin(0.03)
gPad.SetLeftMargin(0.14)
gPad.SetRightMargin(0.04)
gPad.SetBorderMode(0)
gStyle.SetTitleFontSize(0.062)
frame1.GetYaxis().SetTitle("Events / %.3g GeV" % frame1.getFitRangeBinW())
frame1.GetYaxis().SetTitleSize(0.059)
frame1.GetYaxis().SetTitleOffset(1.21)
#frame1.GetYaxis().SetLabelOffset(0.010)
frame1.GetXaxis().SetLabelSize(0)
frame1.GetYaxis().SetLabelSize(0.045)
frame1.Draw()
canvas.cd(2)
gPad.SetPad("pad2", "pad2", 0, 0, 1, 0.33, 0, -1, 0)
gPad.SetTopMargin(0.01)
gPad.SetBottomMargin(0.30)
gPad.SetLeftMargin(0.14)
gPad.SetRightMargin(0.04)
gPad.SetBorderMode(0)
gPad.SetGridy(kTRUE)
line1 = TLine(frame3.GetXaxis().GetXmin(), 0,
frame3.GetXaxis().GetXmax(), 0)
line2 = TLine(frame3.GetXaxis().GetXmin(), 0,
frame3.GetXaxis().GetXmax(), 0)
line1.SetLineColor(0) # white to clear dotted grid lines
line2.SetLineColor(12) # dark grey
line2.SetLineStyle(2)
frame3.SetTitle("")
frame3.GetYaxis().SetTitle("pull")
frame3.GetXaxis().SetTitle("#Deltam^{2}_{ll} [GeV]")
frame3.GetXaxis().SetTitleSize(0.13)
frame3.GetYaxis().SetTitleSize(0.12)
frame3.GetXaxis().SetTitleOffset(1.0)
frame3.GetYaxis().SetTitleOffset(0.58)
frame3.GetXaxis().SetLabelSize(0.10)
frame3.GetYaxis().SetLabelSize(0.10)
frame3.GetXaxis().SetLabelOffset(0.02)
frame3.GetYaxis().SetLabelOffset(0.01)
frame3.GetYaxis().SetRangeUser(-5, 5)
frame3.GetYaxis().CenterTitle(True)
frame3.GetYaxis().SetNdivisions(505)
frame3.Draw("")
line1.Draw("SAME")
line2.Draw("SAME")
frame3.Draw("SAME")
canvas.SaveAs("rooFit109_ratiolike.png")
canvas.Close()
def drawHistsWithRatio(hists, name, **kwargs):
"""Draw histograms with ratios."""
title = kwargs.get('title', "")
xtitle = kwargs.get('xtitle', "")
ytitle = kwargs.get('ytitle', "")
rtitle = kwargs.get('rtitle', "Ratio")
xmin = kwargs.get('xmin', hists[0].GetXaxis().GetXmin())
xmax = kwargs.get('xmax', hists[0].GetXaxis().GetXmax())
ymin = kwargs.get('ymin', None)
ymax = kwargs.get('ymax', None)
rmin = kwargs.get('rmin', 0.45)
rmax = kwargs.get('rmax', 1.55)
logx = kwargs.get('logx', False)
logy = kwargs.get('logy', False)
denom = kwargs.get('denom', 1) - 1 # denominator for ratio
textsize = kwargs.get('textsize', 0.045)
texts = kwargs.get('text', [])
#textheight = kwargs.get('textheight', 1.09 )
#ctext = kwargs.get('ctext', [ ] ) # corner text
#cposition = kwargs.get('cposition', 'topleft' ).lower() # cornertext
#ctextsize = kwargs.get('ctextsize', 1.4*legendtextsize )
colors = kwargs.get('colors', linecolors)
if not isinstance(texts, list) or isinstance(texts, tuple):
texts = [texts]
if ymax == None:
ymax = 1.12 * max(h.GetMaximum() for h in hists)
# MAIN plot
canvas = TCanvas('canvas', 'canvas', 100, 100, 800, 800)
canvas.SetFillColor(0)
canvas.SetBorderMode(0)
canvas.SetFrameBorderMode(0)
canvas.Divide(2)
canvas.SetMargin(0.0, 0.0, 0.0, 0.0)
canvas.cd(1)
gPad.SetPad('pad1', 'pad1', 0, 0.33, 1, 1)
gPad.SetMargin(0.12, 0.04, 0.02, 0.08)
gPad.SetFillColor(0)
gPad.SetBorderMode(0)
gPad.SetTickx(0)
gPad.SetTicky(0)
gPad.SetGrid()
gPad.Draw()
canvas.cd(2)
gPad.SetPad('pad2', 'pad2', 0, 0, 1, 0.33)
gPad.SetMargin(0.12, 0.04, 0.30, 0.03)
gPad.SetFillColor(0)
gPad.SetFillStyle(4000)
gPad.SetFrameFillStyle(0)
gPad.SetBorderMode(0)
gPad.Draw()
# MAIN plot
canvas.cd(1)
for i, hist in enumerate(hists):
color = colors[i % len(colors)]
hist.SetLineColor(color)
hist.SetLineWidth(2)
hist.Draw('HIST SAME')
frame = hists[0]
frame.GetYaxis().SetTitleSize(0.060)
frame.GetXaxis().SetTitleSize(0)
frame.GetXaxis().SetLabelSize(0)
frame.GetYaxis().SetLabelSize(0.052)
frame.GetXaxis().SetLabelOffset(0.010)
frame.GetXaxis().SetTitleOffset(0.98)
frame.GetYaxis().SetTitleOffset(1.05)
frame.GetXaxis().SetNdivisions(508)
frame.GetYaxis().SetTitle(ytitle)
frame.GetXaxis().SetTitle(xtitle)
if logx:
gPad.Update()
gPad.SetLogx()
if logy:
gPad.Update()
gPad.SetLogy()
if ymin: frame.SetMinimum(ymin)
if ymax: frame.SetMaximum(ymax)
width = 0.25
height = 1.1 * textsize * len([l for l in texts + hists if l])
x1, y1 = 0.65, 0.88
x2, y2 = x1 + width, y1 - height
legend = TLegend(x1, y1, x2, y2)
legend.SetTextSize(textsize)
legend.SetBorderSize(0)
legend.SetFillStyle(0)
legend.SetFillColor(0)
legend.SetMargin(0.05 / width)
if title:
legend.SetTextFont(62)
legend.SetHeader(title)
legend.SetTextFont(42)
for hist in hists:
legend.AddEntry(hist, hist.GetTitle(), 'l')
for text in texts:
legend.AddEntry(0, text, '')
legend.Draw()
gPad.SetTicks(1, 1)
gPad.Modified()
frame.Draw('AXIS SAME')
CMS_style.CMS_lumi(gPad, 13, 0)
# RATIO plot
canvas.cd(2)
ratios = []
for i, hist in enumerate(hists):
if i == denom: continue
ratio = hist.Clone(hist.GetName() + "_ratio")
ratio.Divide(hists[denom])
ratio.Draw('HIST SAME')
ratios.append(ratio)
frame_ratio = ratios[0]
frame_ratio.GetYaxis().SetRangeUser(rmin, rmax)
frame_ratio.GetYaxis().CenterTitle()
frame_ratio.GetYaxis().SetTitleSize(0.13)
frame_ratio.GetXaxis().SetTitleSize(0.13)
frame_ratio.GetXaxis().SetLabelSize(0.12)
frame_ratio.GetYaxis().SetLabelSize(0.11)
frame_ratio.GetXaxis().SetLabelOffset(0.012)
frame_ratio.GetXaxis().SetTitleOffset(1.02)
frame_ratio.GetYaxis().SetTitleOffset(0.48)
frame_ratio.GetXaxis().SetNdivisions(508)
frame_ratio.GetYaxis().CenterTitle(True)
frame_ratio.GetYaxis().SetTitle(rtitle)
frame_ratio.GetXaxis().SetTitle(xtitle)
frame_ratio.GetYaxis().SetNdivisions(505)
if logx:
gPad.Update()
gPad.SetLogx()
line = TLine(xmin, 1., xmax, 1.)
line.SetLineColor(hists[denom].GetLineColor())
line.SetLineWidth(hists[denom].GetLineWidth())
line.SetLineStyle(1)
line.Draw('SAME')
gPad.SetTicks(1, 1)
gPad.Update()
gPad.SetGrid()
gPad.Modified()
frame_ratio.Draw('SAME AXIS')
canvas.SaveAs(name + ".png")
canvas.SaveAs(name + ".pdf")
canvas.Close()
def MakeOneHist(histogramName):
HeaderLabel = TPaveLabel(header_x_left, header_y_bottom, header_x_right,
header_y_top, HeaderText, "NDC")
HeaderLabel.SetTextAlign(32)
HeaderLabel.SetBorderSize(0)
HeaderLabel.SetFillColor(0)
HeaderLabel.SetFillStyle(0)
LumiLabel = TPaveLabel(topLeft_x_left, topLeft_y_bottom, topLeft_x_right,
topLeft_y_top, LumiText, "NDC")
LumiLabel.SetBorderSize(0)
LumiLabel.SetFillColor(0)
LumiLabel.SetFillStyle(0)
NormLabel = TPaveLabel()
NormLabel.SetDrawOption("NDC")
NormLabel.SetX1NDC(topLeft_x_left)
NormLabel.SetX2NDC(topLeft_x_right)
NormLabel.SetBorderSize(0)
NormLabel.SetFillColor(0)
NormLabel.SetFillStyle(0)
NormText = ""
if arguments.normalizeToUnitArea:
NormText = "Scaled to unit area"
Legend = TLegend()
Legend.SetBorderSize(0)
Legend.SetFillColor(0)
Legend.SetFillStyle(0)
Canvas = TCanvas(histogramName)
Histograms = []
LegendEntries = []
colorIndex = 0
for source in input_sources: # loop over different input sources in config file
dataset_file = "condor/%s/%s.root" % (source['condor_dir'],
source['dataset'])
inputFile = TFile(dataset_file)
NumHistogramObj = inputFile.Get("OSUAnalysis/" +
source['num_channel'] + "/" +
histogramName)
if 'condor_dir_den' in source: # If specified, take the denominator histogram from a different condor directory.
dataset_fileDen = "condor/%s/%s.root" % (source['condor_dir_den'],
source['dataset'])
inputFileDen = TFile(dataset_fileDen)
DenHistogramObj = inputFileDen.Get("OSUAnalysis/" +
source['den_channel'] + "/" +
histogramName)
else: # Default is to use the same condor directory
DenHistogramObj = inputFile.Get("OSUAnalysis/" +
source['den_channel'] + "/" +
histogramName)
if not NumHistogramObj:
print "WARNING: Could not find histogram " + source[
'num_channel'] + "/" + histogramName + " in file " + dataset_file + ". Will skip it and continue."
return
if not DenHistogramObj:
print "WARNING: Could not find histogram " + source[
'den_channel'] + "/" + histogramName + " in file " + dataset_file + ". Will skip it and continue."
return
Histogram = NumHistogramObj.Clone()
if Histogram.Class().InheritsFrom("TH2"):
Histogram.SetName(Histogram.GetName() + "__" + source['dataset'])
Histogram.SetDirectory(0)
DenHistogram = DenHistogramObj.Clone()
DenHistogram.SetDirectory(0)
inputFile.Close()
if arguments.rebinFactor:
RebinFactor = int(arguments.rebinFactor)
#don't rebin histograms which will have less than 5 bins or any gen-matching histograms
if Histogram.GetNbinsX() >= RebinFactor * 5 and Histogram.GetTitle(
).find("GenMatch") is -1 and not Histogram.Class().InheritsFrom(
"TH2"):
Histogram.Rebin(RebinFactor)
DenHistogram.Rebin(RebinFactor)
xAxisLabel = Histogram.GetXaxis().GetTitle()
unitBeginIndex = xAxisLabel.find("[")
unitEndIndex = xAxisLabel.find("]")
if unitBeginIndex is not -1 and unitEndIndex is not -1: #x axis has a unit
yAxisLabel = "#epsilon_{ " + cutName + "} (" + str(
Histogram.GetXaxis().GetBinWidth(1)
) + " " + xAxisLabel[unitBeginIndex + 1:unitEndIndex] + " width)"
else:
yAxisLabel = "#epsilon_{ " + cutName + "} (" + str(
Histogram.GetXaxis().GetBinWidth(1)) + " width)"
if arguments.normalizeToUnitArea:
yAxisLabel = yAxisLabel + " (Unit Area Norm.)"
#Histogram = ROOT.TGraphAsymmErrors(NumHistogramObj,DenHistogramObj)
if arguments.noTGraph or Histogram.Class().InheritsFrom("TH2"):
Histogram.Divide(DenHistogram)
else:
Histogram = TGraphAsymmErrors(Histogram, DenHistogram)
if not arguments.makeFancy:
fullTitle = Histogram.GetTitle()
splitTitle = fullTitle.split(":")
# print splitTitle
histoTitle = splitTitle[1].lstrip(" ")
else:
histoTitle = ""
if 'color' in source:
Histogram.SetMarkerColor(colors[source['color']])
Histogram.SetLineColor(colors[source['color']])
else:
Histogram.SetMarkerColor(colors[colorList[colorIndex]])
Histogram.SetLineColor(colors[colorList[colorIndex]])
colorIndex = colorIndex + 1
if colorIndex is len(colorList):
colorIndex = 0
markerStyle = 20
if 'marker' in source:
markerStyle = markers[source['marker']]
if 'fill' in source:
markerStyle = markerStyle + fills[source['fill']]
Histogram.SetMarkerStyle(markerStyle)
Histogram.SetLineWidth(line_width)
Histogram.SetFillStyle(0)
LegendEntries.append(source['legend_entry'])
Histograms.append(Histogram)
### scaling histograms as per user's specifications
for histogram in Histograms:
if arguments.normalizeToUnitArea and histogram.Integral() > 0:
histogram.Scale(1. / histogram.Integral())
### formatting histograms and adding to legend
legendIndex = 0
for histogram in Histograms:
Legend.AddEntry(histogram, LegendEntries[legendIndex], "LEP")
legendIndex = legendIndex + 1
### finding the maximum value of anything going on the canvas, so we know how to set the y-axis
finalMax = 0
if arguments.noTGraph:
for histogram in Histograms:
currentMax = histogram.GetMaximum() + histogram.GetBinError(
histogram.GetMaximumBin())
if (currentMax > finalMax):
finalMax = currentMax
finalMax = 1.5 * finalMax
if finalMax is 0:
finalMax = 1
if arguments.setYMax:
finalMax = float(arguments.setYMax)
### Drawing histograms to canvas
makeRatioPlots = arguments.makeRatioPlots
makeDiffPlots = arguments.makeDiffPlots
yAxisMin = 0.0001
if arguments.setYMin:
yAxisMin = float(arguments.setYMin)
if makeRatioPlots or makeDiffPlots:
Canvas.SetFillStyle(0)
Canvas.Divide(1, 2)
Canvas.cd(1)
gPad.SetPad(0, 0.25, 1, 1)
gPad.SetMargin(0.15, 0.05, 0.01, 0.07)
gPad.SetFillStyle(0)
gPad.Update()
gPad.Draw()
if arguments.setLogY:
gPad.SetLogy()
Canvas.cd(2)
gPad.SetPad(0, 0, 1, 0.25)
#format: gPad.SetMargin(l,r,b,t)
gPad.SetMargin(0.15, 0.05, 0.4, 0.01)
gPad.SetFillStyle(0)
gPad.SetGridy(1)
gPad.Update()
gPad.Draw()
Canvas.cd(1)
histCounter = 0
plotting_options = ""
if arguments.plot_hist:
plotting_options = "HIST"
for histogram in Histograms:
if histogram.Class().InheritsFrom("TH2"):
histogram.SetTitle(histoTitle)
histogram.Draw("colz")
DatasetName = histogram.GetName()
DatasetName = DatasetName[
DatasetName.rfind('__') +
2:] # substring starting with the last underscore
DatasetLabel = TPaveLabel(topLeft_x_left, topLeft_y_bottom,
topLeft_x_right, topLeft_y_top,
DatasetName, "NDC")
DatasetLabel.SetBorderSize(0)
DatasetLabel.SetFillColor(0)
DatasetLabel.SetFillStyle(0)
DatasetLabel.Draw()
outputFile.cd()
Canvas.SetName(histogram.GetName())
Canvas.Write()
if arguments.makeFancy:
HeaderLabel.Draw()
else:
if histogram.InheritsFrom("TGraph") and histCounter == 0:
plotting_options = "AP"
histogram.SetTitle(histoTitle)
histogram.Draw(plotting_options)
histogram.GetXaxis().SetTitle(xAxisLabel)
histogram.GetYaxis().SetTitle(yAxisLabel)
if histogram.InheritsFrom("TH1"):
histogram.SetMaximum(finalMax)
histogram.SetMinimum(yAxisMin)
if makeRatioPlots or makeDiffPlots:
histogram.GetXaxis().SetLabelSize(0)
if histCounter is 0:
if histogram.InheritsFrom("TH1"):
plotting_options = plotting_options + " SAME"
elif histogram.InheritsFrom("TGraph"):
plotting_options = "P"
histCounter = histCounter + 1
if histogram.Class().InheritsFrom("TH2"):
return
#legend coordinates, empirically determined :-)
x_left = 0.1677852
x_right = 0.9647651
y_min = 0.6765734
y_max = 0.9
Legend.SetX1NDC(x_left)
Legend.SetY1NDC(y_min)
Legend.SetX2NDC(x_right)
Legend.SetY2NDC(y_max)
Legend.Draw()
if arguments.makeFancy:
HeaderLabel.Draw()
#drawing the ratio or difference plot if requested
if makeRatioPlots or makeDiffPlots:
Canvas.cd(2)
if makeRatioPlots:
Comparison = ratioHistogram(Histograms[0], Histograms[1])
elif makeDiffPlots:
Comparison = Histograms[0].Clone("diff")
Comparison.Add(Histograms[1], -1)
Comparison.SetTitle("")
Comparison.GetYaxis().SetTitle("hist1-hist2")
Comparison.GetXaxis().SetTitle(xAxisLabel)
Comparison.GetYaxis().CenterTitle()
Comparison.GetYaxis().SetTitleSize(0.1)
Comparison.GetYaxis().SetTitleOffset(0.5)
Comparison.GetXaxis().SetTitleSize(0.15)
Comparison.GetYaxis().SetLabelSize(0.1)
Comparison.GetXaxis().SetLabelSize(0.15)
if makeRatioPlots:
RatioYRange = 1.15
if arguments.ratioYRange:
RatioYRange = float(arguments.ratioYRange)
Comparison.GetYaxis().SetRangeUser(-1 * RatioYRange, RatioYRange)
elif makeDiffPlots:
YMax = Comparison.GetMaximum()
YMin = Comparison.GetMinimum()
if YMax <= 0 and YMin <= 0:
Comparison.GetYaxis().SetRangeUser(-1.2 * YMin, 0)
elif YMax >= 0 and YMin >= 0:
Comparison.GetYaxis().SetRangeUser(0, 1.2 * YMax)
else: #axis crosses y=0
if abs(YMax) > abs(YMin):
Comparison.GetYaxis().SetRangeUser(-1.2 * YMax, 1.2 * YMax)
else:
Comparison.GetYaxis().SetRangeUser(-1.2 * YMin, 1.2 * YMin)
Comparison.GetYaxis().SetNdivisions(205)
Comparison.Draw("E0")
outputFile.cd()
Canvas.Write()
if arguments.savePDFs:
Canvas.SaveAs("efficiency_histograms_pdfs/" + histogramName + ".pdf")
########################################################################################
outputFile = TFile(arguments.outputDir + "/output.root", "RECREATE")
for pt in pts_to_test:
for variable in variables_to_test:
for plot in plots:
plot_name = plot['name'] + "_" + variable + "_" + pt
canvas = TCanvas(plot_name, "", 0, 0, 700, 800)
canvas.Divide(1, 2)
canvas.cd(1)
gPad.SetLogy()
gPad.SetPad(0, 0.25, 1, 1)
gPad.SetMargin(0.15, 0.05, 0.0, 0.07)
gPad.SetFillStyle(0)
gPad.SetTickx(1)
gPad.SetTicky(1)
gPad.Update()
gPad.Draw()
canvas.cd(2)
gPad.SetPad(0, 0, 1, 0.25)
gPad.SetMargin(0.15, 0.05, 0.4, 0.0)
gPad.SetFillStyle(0)
gPad.SetTickx(1)
gPad.SetTicky(1)
gPad.Update()
gPad.Draw()
def MakeOneDHist(histogramDirectory, histogramName,integrateDir):
if arguments.verbose:
print "Creating histogram", histogramName, "in directory", histogramDirectory
HeaderLabel = TPaveLabel(header_x_left,header_y_bottom,header_x_right,header_y_top,HeaderText,"NDC")
HeaderLabel.SetTextAlign(32)
HeaderLabel.SetTextFont(42)
HeaderLabel.SetTextSize(0.697674)
HeaderLabel.SetBorderSize(0)
HeaderLabel.SetFillColor(0)
HeaderLabel.SetFillStyle(0)
CMSLabel = TPaveLabel(header_x_left,header_y_bottom,header_x_right,header_y_top,HeaderText,"NDC")
CMSLabel.SetTextAlign(32)
CMSLabel.SetTextFont(42)
CMSLabel.SetTextSize(0.697674)
CMSLabel.SetBorderSize(0)
CMSLabel.SetFillColor(0)
CMSLabel.SetFillStyle(0)
if makeFancy:
LumiLabel = TPaveLabel(topLeft_x_left,topLeft_y_bottom,topLeft_x_right,topLeft_y_top,"CMS Preliminary","NDC")
LumiLabel.SetTextFont(62)
LumiLabel.SetTextSize(0.7)
LumiLabel.SetTextAlign(12)
else:
LumiLabel = TPaveLabel(topLeft_x_left,topLeft_y_bottom,topLeft_x_right,topLeft_y_top,LumiText,"NDC")
LumiLabel.SetTextAlign(32)
LumiLabel.SetTextFont(42)
LumiLabel.SetBorderSize(0)
LumiLabel.SetFillColor(0)
LumiLabel.SetFillStyle(0)
Legend = TLegend()
Legend.SetBorderSize(0)
Legend.SetFillColor(0)
Legend.SetFillStyle(0)
canvasName = histogramName
if integrateDir is "left":
canvasName += "_CumulativeLeft"
elif integrateDir is "right":
canvasName += "_CumulativeRight"
Canvas = TCanvas(canvasName)
Histograms = []
RefIndex = -99
LegendEntries = []
colorIndex = 0
markerStyleIndex = 0
fillIndex = 0
for source in input_sources: # loop over different input sources in config file
dataset_file = "condor/%s/%s.root" % (source['condor_dir'],source['dataset'])
inputFile = TFile(dataset_file)
if arguments.generic:
if histogramDirectory == "":
histPath = histogramName
else:
histPath = histogramDirectory + "/" + histogramName
HistogramObj = inputFile.Get(histPath)
else:
HistogramObj = inputFile.Get(source['channel'] + "Plotter/" + histogramDirectory + "/" + histogramName)
if not HistogramObj:
print "WARNING: Could not find histogram " + source['channel'] + "/" + histogramName + " in file " + dataset_file + ". Will skip it and continue."
return
Histogram = HistogramObj.Clone()
Histogram.SetDirectory(0)
inputFile.Close()
Histogram.Sumw2()
if arguments.verbose:
print " Got histogram", Histogram.GetName(), "from file", dataset_file
if arguments.rebinFactor:
RebinFactor = int(arguments.rebinFactor)
#don't rebin histograms which will have less than 5 bins or any gen-matching histograms
if Histogram.GetNbinsX() >= RebinFactor*5 and Histogram.GetTitle().find("GenMatch") is -1:
Histogram.Rebin(RebinFactor)
# correct bin contents of object multiplcity plots
if Histogram.GetName().startswith("num") and "PV" not in Histogram.GetName():
# include overflow bin
for bin in range(2,Histogram.GetNbinsX()+2):
content = Histogram.GetBinContent(bin)
Histogram.SetBinContent(bin, content/float(bin-1))
xAxisLabel = Histogram.GetXaxis().GetTitle()
unitBeginIndex = xAxisLabel.find("[")
unitEndIndex = xAxisLabel.find("]")
xAxisLabelVar = xAxisLabel
if "_pfx" in Histogram.GetName() or "_pfy" in Histogram.GetName() or "_sigma" in Histogram.GetName():
yAxisLabel = Histogram.GetYaxis().GetTitle()
else:
if unitBeginIndex is not -1 and unitEndIndex is not -1: #x axis has a unit
yAxisLabel = "Entries / " + str(Histogram.GetXaxis().GetBinWidth(1)) + " " + xAxisLabel[unitBeginIndex+1:unitEndIndex]
xAxisLabelVar = xAxisLabel[0:unitBeginIndex]
else:
yAxisLabel = "Entries per bin (" + str(Histogram.GetXaxis().GetBinWidth(1)) + " width)"
if arguments.normalizeToUnitArea:
yAxisLabel = yAxisLabel + " (Unit Area Norm.)"
if arguments.normalizeToUnitArea and arguments.makeSignificancePlots:
unit = "Efficiency"
else:
unit = "Yield"
if integrateDir is "left":
yAxisLabel = unit + ", " + xAxisLabelVar + "< x (" + str(Histogram.GetXaxis().GetBinWidth(1)) + " bin width)"
if integrateDir is "right":
yAxisLabel = unit + ", " + xAxisLabelVar + "> x (" + str(Histogram.GetXaxis().GetBinWidth(1)) + " bin width)"
nbins = Histogram.GetNbinsX()
if not noOverFlow:
Histogram.SetBinContent(nbins, Histogram.GetBinContent(nbins) + Histogram.GetBinContent(nbins+1)) # Add overflow
Histogram.SetBinError(nbins, math.sqrt(math.pow(Histogram.GetBinError(nbins),2) + math.pow(Histogram.GetBinError(nbins+1),2))) # Set the errors to be the sum in quadrature
if not noUnderFlow:
Histogram.SetBinContent(1, Histogram.GetBinContent(1) + Histogram.GetBinContent(0)) # Add underflow
Histogram.SetBinError(1, math.sqrt(math.pow(Histogram.GetBinError(1), 2) + math.pow(Histogram.GetBinError(0), 2))) # Set the errors to be the sum in quadrature
if not arguments.makeFancy and not arguments.generic:
fullTitle = Histogram.GetTitle()
splitTitle = fullTitle.split(":")
if len(splitTitle) > 1:
histoTitle = splitTitle[1].lstrip(" ")
else:
histoTitle = splitTitle[0]
else:
histoTitle = ""
if 'color' in source:
Histogram.SetMarkerColor(colors[source['color']])
Histogram.SetLineColor(colors[source['color']])
else:
Histogram.SetMarkerColor(colors[colorList[colorIndex]])
Histogram.SetLineColor(colors[colorList[colorIndex]])
colorIndex = colorIndex + 1
if colorIndex is len(colorList):
colorIndex = 0
markerStyleIndex = markerStyleIndex + 1
if markerStyleIndex is len(markerStyleList):
markerStyleIndex = 0
fillIndex = fillIndex + 1
if 'scale' in source:
Histogram.Scale(source['scale'])
markerStyle = 20
if 'marker' in source:
markerStyle = markers[source['marker']]
else:
markerStyle = markers[markerStyleList[markerStyleIndex]]
fillStyle = 0
if 'fill' in source:
markerStyle = markerStyle + fills[source['fill']]
else:
markerStyle = markerStyle + fills[fillList[fillIndex]]
Histogram.SetMarkerStyle(markerStyle)
Histogram.SetMarkerSize(0.5)
Histogram.SetLineWidth(line_width)
Histogram.SetFillStyle(0)
if arguments.normalizeToUnitArea and Histogram.Integral() > 0:
Histogram.Scale(1./Histogram.Integral())
Histogram = MakeIntegralHist(Histogram, integrateDir)
LegendEntries.append(source['legend_entry'])
Histograms.append(Histogram)
if 'reference' in source:
if source['reference']:
RefIndex = len(Histograms)-1
### formatting histograms and adding to legend
legendIndex = 0
for histogram in Histograms:
Legend.AddEntry(histogram,LegendEntries[legendIndex],"LEP")
# Legend.AddEntry(histogram,LegendEntries[legendIndex],"P")
legendIndex = legendIndex+1
### finding the maximum value of anything going on the canvas, so we know how to set the y-axis
finalMax = 0
for histogram in Histograms:
currentMax = histogram.GetMaximum() + histogram.GetBinError(histogram.GetMaximumBin())
if(currentMax > finalMax):
finalMax = currentMax
finalMax = 1.5*finalMax
if arguments.setYMax:
finalMax = float(arguments.setYMax)
### Drawing histograms to canvas
makeRatioPlots = arguments.makeRatioPlots
makeDiffPlots = arguments.makeDiffPlots
addOneToRatio = -1
if arguments.addOneToRatio:
addOneToRatio = arguments.addOneToRatio
yAxisMin = 0.0001
if arguments.setYMin:
yAxisMin = float(arguments.setYMin)
if makeRatioPlots or makeDiffPlots:
Canvas.SetFillStyle(0)
Canvas.Divide(1,2)
Canvas.cd(1)
gPad.SetPad(0,0.25,1,1)
gPad.SetMargin(0.15,0.05,0.01,0.07)
gPad.SetFillStyle(0)
gPad.Update()
gPad.Draw()
if arguments.setLogY:
gPad.SetLogy()
Canvas.cd(2)
gPad.SetPad(0,0,1,0.25)
#format: gPad.SetMargin(l,r,b,t)
gPad.SetMargin(0.15,0.05,0.4,0.01)
gPad.SetFillStyle(0)
gPad.SetGridy(1)
gPad.Update()
gPad.Draw()
Canvas.cd(1)
histCounter = 0
plotting_options = ""
if arguments.generic:
plotting_options = "p,e"
if arguments.plot_hist:
plotting_options = "HIST"
for histogram in Histograms:
histogram.SetTitle(histoTitle)
if arguments.verbose:
print " Drawing hist " + histogram.GetName() + ", with plotting_options = " + plotting_options + ", with mean = " + str(histogram.GetMean()) + ", with color = " + str(histogram.GetLineColor())
histogram.Draw(plotting_options)
histogram.GetXaxis().SetTitle(xAxisLabel)
histogram.GetYaxis().SetTitle(yAxisLabel)
histogram.SetMaximum(finalMax)
if "_pfx" not in Histogram.GetName() and "_pfy" not in Histogram.GetName() and "_sigma" not in Histogram.GetName():
histogram.SetMinimum(yAxisMin)
if makeRatioPlots or makeDiffPlots:
histogram.GetXaxis().SetLabelSize(0)
if histCounter is 0:
plotting_options = plotting_options + " SAME"
histCounter = histCounter + 1
#legend coordinates, empirically determined :-)
x_left = 0.1677852
x_right = 0.9647651
y_min = 0.6765734
y_max = 0.9
Legend.SetX1NDC(x_left)
Legend.SetY1NDC(y_min)
Legend.SetX2NDC(x_right)
Legend.SetY2NDC(y_max)
Legend.Draw()
# Deciding which text labels to draw and drawing them
if arguments.makeFancy:
HeaderLabel.Draw()
LumiLabel.Draw()
#drawing the ratio or difference plot if requested
if makeRatioPlots or makeDiffPlots:
Comparisons = []
Canvas.cd(2)
if RefIndex == -99:
Reference = Histograms[0]
else:
Reference = Histograms[RefIndex]
for Histogram in Histograms:
if Histogram is Reference:
continue
if makeRatioPlots:
makeRatio = functools.partial (ratioHistogram,Histogram, Reference)
if arguments.ratioRelErrMax is not -1: # it gets initialized to this dummy value of -1
makeRatio = functools.partial (makeRatio, relErrMax = float(arguments.ratioRelErrMax))
if addOneToRatio != -1: # it gets initialized to this dummy value of -1
makeRatio = functools.partial (makeRatio, addOne = bool (addOneToRatio))
Comparison = makeRatio()
elif makeDiffPlots:
Comparison = Reference.Clone("diff")
Comparison.Add(Histograms[1],-1)
Comparison.SetTitle("")
Comparison.GetYaxis().SetTitle("X-ref")
Comparison.SetLineColor(Histogram.GetLineColor())
Comparison.SetFillColor(Histogram.GetFillColor())
Comparison.SetFillStyle(Histogram.GetFillStyle())
Comparison.SetMarkerColor(Histogram.GetMarkerColor())
Comparison.SetMarkerStyle(Histogram.GetMarkerStyle())
Comparison.GetXaxis().SetTitle(xAxisLabel)
Comparison.GetYaxis().CenterTitle()
Comparison.GetYaxis().SetTitleSize(0.1)
Comparison.GetYaxis().SetTitleOffset(0.5)
Comparison.GetXaxis().SetTitleSize(0.15)
Comparison.GetYaxis().SetLabelSize(0.1)
Comparison.GetXaxis().SetLabelSize(0.15)
if makeRatioPlots:
RatioYRange = 1.15
if arguments.ratioYRange:
RatioYRange = float(arguments.ratioYRange)
if addOneToRatio == -1: # it gets initialized to this dummy value of -1
Comparison.GetYaxis().SetRangeUser(-1*RatioYRange, RatioYRange)
else:
Comparison.GetYaxis().SetRangeUser(-1*RatioYRange + 1.0, RatioYRange + 1.0)
elif makeDiffPlots:
YMax = Comparison.GetMaximum()
YMin = Comparison.GetMinimum()
if YMax <= 0 and YMin <= 0:
Comparison.GetYaxis().SetRangeUser(-1.2*YMin,0)
elif YMax >= 0 and YMin >= 0:
Comparison.GetYaxis().SetRangeUser(0,1.2*YMax)
else: #axis crosses y=0
if abs(YMax) > abs(YMin):
Comparison.GetYaxis().SetRangeUser(-1.2*YMax,1.2*YMax)
else:
Comparison.GetYaxis().SetRangeUser(-1.2*YMin,1.2*YMin)
Comparison.GetYaxis().SetNdivisions(205)
Comparisons.append(Comparison)
option = "E0"
for index,Comparison in enumerate(Comparisons):
if index == 0:
option += " SAME"
Comparison.Draw(option)
outputFile.cd(histogramDirectory)
Canvas.Write()
if arguments.verbose:
print " Finished writing canvas: ", Canvas.GetName()
if arguments.savePDFs:
Canvas.SaveAs("comparison_histograms_pdfs/"+histogramName+".pdf")
def MakeOneHist(dirName, histogramName):
HeaderLabel = TPaveLabel(header_x_left,header_y_bottom,header_x_right,header_y_top,HeaderText,"NDC")
HeaderLabel.SetTextAlign(32)
HeaderLabel.SetTextFont(42)
HeaderLabel.SetTextSize(0.697674)
HeaderLabel.SetBorderSize(0)
HeaderLabel.SetFillColor(0)
HeaderLabel.SetFillStyle(0)
CMSLabel = TPaveLabel(header_x_left,header_y_bottom,header_x_right,header_y_top,HeaderText,"NDC")
CMSLabel.SetTextAlign(32)
CMSLabel.SetTextFont(42)
CMSLabel.SetTextSize(0.697674)
CMSLabel.SetBorderSize(0)
CMSLabel.SetFillColor(0)
CMSLabel.SetFillStyle(0)
if makeFancy:
LumiLabel = TPaveLabel(topLeft_x_left,topLeft_y_bottom,topLeft_x_right,topLeft_y_top,"CMS Preliminary","NDC")
LumiLabel.SetTextFont(62)
LumiLabel.SetTextSize(0.7)
LumiLabel.SetTextAlign(12)
else:
LumiLabel = TPaveLabel(topLeft_x_left,topLeft_y_bottom,topLeft_x_right,topLeft_y_top,LumiText,"NDC")
LumiLabel.SetTextAlign(32)
LumiLabel.SetTextFont(42)
LumiLabel.SetBorderSize(0)
LumiLabel.SetFillColor(0)
LumiLabel.SetFillStyle(0)
#legend coordinates, empirically determined :-)
x_left = 0.4
x_right = 0.7
y_min = 0.15
y_max = 0.3
Legend = TLegend(x_left,y_min,x_right,y_max)
Legend.SetBorderSize(0)
Legend.SetFillColor(0)
Legend.SetFillStyle(0)
Canvas = TCanvas(histogramName)
Histograms = []
HistogramClones = []
NBins = []
MaxXValues = []
MinXValues = []
LegendEntries = []
colorIndex = 0
for source in input_sources: # loop over different input sources in config file
dataset_file = "condor/%s/%s.root" % (source['condor_dir'],source['dataset'])
inputFile = TFile(dataset_file)
NumHistogramObj = inputFile.Get(source['num_channel'] + "Plotter/" + dirName + "/" + histogramName)
if 'condor_dir_den' in source: # If specified, take the denominator histogram from a different condor directory.
dataset_fileDen = "condor/%s/%s.root" % (source['condor_dir_den'],source['dataset'])
inputFileDen = TFile(dataset_fileDen)
DenHistogramObj = inputFileDen.Get(source['den_channel'] + "Plotter/" + dirName + "/" + histogramName)
else: # Default is to use the same condor directory
DenHistogramObj = inputFile.Get(source['den_channel'] + "Plotter/" + dirName + "/" + histogramName)
if not NumHistogramObj:
print "WARNING: Could not find histogram " + source['num_channel'] + "Plotter/" + dirName + "/" + histogramName + " in file " + dataset_file + ". Will skip it and continue."
return
if not DenHistogramObj:
print "WARNING: Could not find histogram " + source['den_channel'] + "Plotter/" + dirName + "/" + histogramName + " in file " + dataset_file + ". Will skip it and continue."
return
Histogram = NumHistogramObj.Clone()
if Histogram.Class().InheritsFrom("TH2"):
Histogram.SetName(Histogram.GetName() + "__" + source['dataset'])
Histogram.SetDirectory(0)
DenHistogram = DenHistogramObj.Clone()
DenHistogram.SetDirectory(0)
inputFile.Close()
nbinsN = Histogram.GetNbinsX()
nbinsD = DenHistogram.GetNbinsX()
if not noOverFlow:
# Add overflow
Histogram.SetBinContent( nbinsN, Histogram.GetBinContent(nbinsN) + Histogram.GetBinContent(nbinsN+1))
DenHistogram.SetBinContent(nbinsD, DenHistogram.GetBinContent(nbinsD) + DenHistogram.GetBinContent(nbinsD+1))
# Set the errors to be the sum in quadrature
Histogram.SetBinError( nbinsN, math.sqrt(math.pow(Histogram.GetBinError(nbinsN),2) + math.pow(Histogram.GetBinError(nbinsN+1),2)))
DenHistogram.SetBinError(nbinsD, math.sqrt(math.pow(DenHistogram.GetBinError(nbinsD),2) + math.pow(DenHistogram.GetBinError(nbinsD+1),2)))
if not noUnderFlow:
# Add underflow
Histogram.SetBinContent( 1, Histogram.GetBinContent(1) + Histogram.GetBinContent(0))
DenHistogram.SetBinContent(1, DenHistogram.GetBinContent(1) + DenHistogram.GetBinContent(0))
# Set the errors to be the sum in quadrature
Histogram.SetBinError( 1, math.sqrt(math.pow(Histogram.GetBinError(1), 2) + math.pow(Histogram.GetBinError(0), 2)))
DenHistogram.SetBinError(1, math.sqrt(math.pow(DenHistogram.GetBinError(1), 2) + math.pow(DenHistogram.GetBinError(0), 2)))
if arguments.rebinFactor:
RebinFactor = int(arguments.rebinFactor)
#don't rebin histograms which will have less than 5 bins or any gen-matching histograms
if Histogram.GetNbinsX() >= RebinFactor*5 and Histogram.GetTitle().find("GenMatch") is -1 and not Histogram.Class().InheritsFrom("TH2"):
Histogram.Rebin(RebinFactor)
DenHistogram.Rebin(RebinFactor)
xAxisLabel = Histogram.GetXaxis().GetTitle()
unitBeginIndex = xAxisLabel.find("[")
unitEndIndex = xAxisLabel.find("]")
if unitBeginIndex is not -1 and unitEndIndex is not -1: #x axis has a unit
yAxisLabel = "#epsilon_{ " + cutName + "} (" + str(Histogram.GetXaxis().GetBinWidth(1)) + " " + xAxisLabel[unitBeginIndex+1:unitEndIndex] + " width)"
else:
yAxisLabel = "#epsilon_{ " + cutName + "} (" + str(Histogram.GetXaxis().GetBinWidth(1)) + " width)"
if arguments.normalizeToUnitArea:
yAxisLabel = yAxisLabel + " (Unit Area Norm.)"
#check if bin content is consistent
TEfficiency.CheckConsistency(Histogram,DenHistogram)
for i in range(1,Histogram.GetNbinsX()+1):
if Histogram.GetBinContent(i) > DenHistogram.GetBinContent(i):
DenHistogram.SetBinContent(i,Histogram.GetBinContent(i))
#HistogramClone and HistogramClones only used for ratio plot
HistogramClone = Histogram.Clone()
HistogramClone.SetDirectory(0)
HistogramClone.Divide(DenHistogram)
Nbins = HistogramClone.GetNbinsX()
MaxXValue = HistogramClone.GetXaxis().GetBinLowEdge(Nbins+1)
MinXValue = HistogramClone.GetXaxis().GetBinLowEdge(1)
#this Histogram becomes the main TEfficiency
if Histogram.Class().InheritsFrom("TH2"):
Histogram.Divide(DenHistogram)
else:
#using default methods (which give correct uncertainties)
#see https://root.cern.ch/doc/master/classTEfficiency.html (c.f. section IV)
Histogram = TEfficiency(Histogram,DenHistogram)
if not arguments.makeFancy:
fullTitle = Histogram.GetTitle()
splitTitle = fullTitle.split(":")
# print splitTitle
if len(splitTitle) > 1:
histoTitle = splitTitle[1].lstrip(" ")
else:
histoTitle = splitTitle[0]
else:
histoTitle = ""
if 'color' in source:
Histogram.SetMarkerColor(colors[source['color']])
Histogram.SetLineColor(colors[source['color']])
else:
Histogram.SetMarkerColor(colors[colorList[colorIndex]])
Histogram.SetLineColor(colors[colorList[colorIndex]])
colorIndex = colorIndex + 1
if colorIndex is len(colorList):
colorIndex = 0
markerStyle = 20
if 'marker' in source:
markerStyle = markers[source['marker']]
if 'fill' in source:
markerStyle = markerStyle + fills[source['fill']]
Histogram.SetMarkerStyle(markerStyle)
Histogram.SetLineWidth(line_width)
Histogram.SetFillStyle(0)
LegendEntries.append(source['legend_entry'])
Histograms.append(Histogram)
HistogramClones.append(HistogramClone)
NBins.append(Nbins)
MaxXValues.append(MaxXValue)
MinXValues.append(MinXValue)
### scaling histograms as per user's specifications
for histogram in Histograms:
if arguments.normalizeToUnitArea and histogram.Integral() > 0:
histogram.Scale(1./histogram.Integral())
### formatting histograms and adding to legend
legendIndex = 0
for histogram in Histograms:
Legend.AddEntry(histogram,LegendEntries[legendIndex],"LEP")
legendIndex = legendIndex+1
### finding the maximum value of anything going on the canvas, so we know how to set the y-axis
finalMax = 1.1
if arguments.setYMax:
finalMax = float(arguments.setYMax)
### Drawing histograms to canvas
makeRatioPlots = arguments.makeRatioPlots
makeDiffPlots = arguments.makeDiffPlots
addOneToRatio = -1
if arguments.addOneToRatio:
addOneToRatio = arguments.addOneToRatio
dontRebinRatio = -1
if arguments.dontRebinRatio:
dontRebinRatio = arguments.dontRebinRatio
ratioRelErrMax = -1
if arguments.ratioRelErrMax:
ratioRelErrMax = arguments.ratioRelErrMax
yAxisMin = 0.0001
if arguments.setYMin:
yAxisMin = float(arguments.setYMin)
if makeRatioPlots or makeDiffPlots:
Canvas.SetFillStyle(0)
Canvas.Divide(1,2)
Canvas.cd(1)
gPad.SetPad(0,0.25,1,1)
gPad.SetMargin(0.15,0.05,0.01,0.07)
gPad.SetFillStyle(0)
gPad.Update()
gPad.Draw()
if arguments.setLogY:
gPad.SetLogy()
Canvas.cd(2)
gPad.SetPad(0,0,1,0.25)
#format: gPad.SetMargin(l,r,b,t)
gPad.SetMargin(0.15,0.05,0.4,0.01)
gPad.SetFillStyle(0)
gPad.SetGridy(1)
gPad.Update()
gPad.Draw()
Canvas.cd(1)
histCounter = 0
plotting_options = ""
if arguments.plot_hist:
plotting_options = "HIST"
h = TH2F("h1","",NBins[0],MinXValues[0],MaxXValues[0],110,0.,finalMax)
h.SetTitle(";"+xAxisLabel+";"+yAxisLabel)
h.Draw()
for histogram in Histograms:
if histogram.Class().InheritsFrom("TH2"):
histogram.SetTitle(histoTitle)
histogram.Draw("colz")
DatasetName = histogram.GetName()
DatasetName = DatasetName[DatasetName.rfind('__')+2:] # substring starting with the last underscore
DatasetLabel = TPaveLabel(topLeft_x_left,topLeft_y_bottom,topLeft_x_right,topLeft_y_top,DatasetName,"NDC")
DatasetLabel.SetBorderSize(0)
DatasetLabel.SetFillColor(0)
DatasetLabel.SetFillStyle(0)
DatasetLabel.Draw()
outputFile.cd()
Canvas.SetName(histogram.GetName())
Canvas.Write()
else:
if histogram.InheritsFrom("TEfficiency") and histCounter==0:
plotting_options = "P SAME"
histogram.SetTitle(histoTitle)
histogram.Draw(plotting_options)
if histogram.InheritsFrom("TH1"):
histogram.SetMaximum(finalMax)
histogram.SetMinimum(yAxisMin)
if histCounter is 0:
if histogram.InheritsFrom("TH1"):
plotting_options = plotting_options + " SAME"
elif histogram.InheritsFrom("TEfficiency"):
plotting_options = "P" + " SAME"
histCounter = histCounter + 1
if histogram.Class().InheritsFrom("TH2"):
return
Legend.Draw()
if arguments.makeFancy:
HeaderLabel.Draw()
LumiLabel.Draw()
#drawing the ratio or difference plot if requested
if makeRatioPlots or makeDiffPlots:
Canvas.cd(2)
if makeRatioPlots:
makeRatio = functools.partial (ratioHistogram,HistogramClones[0],HistogramClones[1])
if addOneToRatio != -1: # it gets initialized to this dummy value of -1
makeRatio = functools.partial (makeRatio, addOne = bool (addOneToRatio))
if ratioRelErrMax is not -1: # it gets initialized to this dummy value of -1
makeRatio = functools.partial (makeRatio, relErrMax = float (ratioRelErrMax))
if dontRebinRatio is True:
makeRatio = functools.partial (makeRatio, dontRebinRatio)
Comparison = makeRatio ()
elif makeDiffPlots:
Comparison = Histograms[0].Clone("diff")
Comparison.Add(Histograms[1],-1)
Comparison.SetTitle("")
Comparison.GetYaxis().SetTitle("hist1-hist2")
Comparison.GetXaxis().SetTitle(xAxisLabel)
Comparison.GetYaxis().CenterTitle()
Comparison.GetYaxis().SetTitleSize(0.1)
Comparison.GetYaxis().SetTitleOffset(0.5)
Comparison.GetXaxis().SetTitleSize(0.15)
Comparison.GetYaxis().SetLabelSize(0.1)
Comparison.GetXaxis().SetLabelSize(0.15)
if makeRatioPlots:
RatioYRange = 1.15
if arguments.ratioYRange:
RatioYRange = float(arguments.ratioYRange)
if addOneToRatio == -1: # it gets initialized to this dummy value of -1
Comparison.GetYaxis().SetRangeUser(-1*RatioYRange, RatioYRange)
else:
Comparison.GetYaxis().SetRangeUser(-1*RatioYRange + 1.0, RatioYRange + 1.0)
elif makeDiffPlots:
YMax = Comparison.GetMaximum()
YMin = Comparison.GetMinimum()
if YMax <= 0 and YMin <= 0:
Comparison.GetYaxis().SetRangeUser(-1.2*YMin,0)
elif YMax >= 0 and YMin >= 0:
Comparison.GetYaxis().SetRangeUser(0,1.2*YMax)
else: #axis crosses y=0
if abs(YMax) > abs(YMin):
Comparison.GetYaxis().SetRangeUser(-1.2*YMax,1.2*YMax)
else:
Comparison.GetYaxis().SetRangeUser(-1.2*YMin,1.2*YMin)
Comparison.GetYaxis().SetNdivisions(205)
Comparison.Draw("E0")
outputFile.cd(dirName)
Canvas.Write()
if arguments.savePDFs:
Canvas.SaveAs("efficiency_histograms_pdfs/"+histogramName+".pdf")
def makePlot(hName, CR, isQCDMC, isData, isLog, isRatio):
'''
We first draw stacked histograms then data then unc band.
The ratio of data and background is drawn next in a separate
pad.
'''
#Divide canvas for the ratio plot
canvas = TCanvas()
if isRatio and isData:
canvas.Divide(1,2)
canvas.cd(1)
gPad.SetRightMargin(0.03);
gPad.SetPad(xPadRange[0],yPadRange[2],xPadRange[1],yPadRange[3]);
gPad.SetTopMargin(0.09);
gPad.SetBottomMargin(padGap);
#gPad.SetTickx(0);
gPad.RedrawAxis();
if isLog:
gPad.SetLogy(True);
#Get nominal histograms
#dataHist, bkgHists, qcdMCHist, qcdDDHist = getBaseHists(fileDict, hName, CR)
dataHist, bkgHists, qcdMCHist = getBaseHists(fileDict, hName, CR)
hSumOtherBkg = bkgHists[0].Clone("hSumOtherBkg")
hSumOtherBkg.Reset()
hAllBkgs = bkgHists
for bkgHist in bkgHists:
hColor = Samples[bkgHist.GetName().split("_")[0]][1]
hSumOtherBkg.Add(bkgHist)
hSumAllBkg = hSumOtherBkg.Clone("hSumAllBkg")
if isQCDMC:
hSumAllBkg.Add(qcdMCHist[0])
hAllBkgs.append(qcdMCHist[0])
#else:
# hSumAllBkg.Add(qcdDDHist[0])
# hAllBkgs.append(qcdDDHist[0])
#Stack nominal hists
xTitle = histograms[hName][0]
yTitle = histograms[hName][1]
hStack = THStack(hName,hName)
hForStack = sortHists(hAllBkgs, False)
for h in hForStack:
sampleName = h.GetName().split("_")[0]
decoHist(h, xTitle, yTitle, Samples[sampleName][1])
hStack.Add(h)
hStack.SetMinimum(1.0)
if isLog:
hStack.SetMaximum(100*hStack.GetMaximum())
else:
hStack.SetMaximum(1.3*hStack.GetMaximum())
hStack.Draw("HIST")
decoHistStack(hStack, xTitle, yTitle)
#Get histograms for the difference between nominal and syst up/down
hSumOtherBkgUps, hQCDUps = getSystHists(fileDict, hName, CR, "Up")
hSumOtherBkgDowns, hQCDDowns = getSystHists(fileDict, hName, CR, "Down")
hDiffUp = hSumOtherBkg.Clone("hDiffUp")
hDiffUp.Reset()
hDiffDown = hSumOtherBkg.Clone("hDiffDown")
hDiffDown.Reset()
for hUp in hSumOtherBkgUps:
hDiff = hUp.Clone("hDiff")
hDiff.Add(hSumOtherBkg, -1)
hDiffUp.Add(hDiff)
print "hDiffUp = ", hDiffUp.Integral()
for hDown in hSumOtherBkgDowns:
hDiff = hSumOtherBkg.Clone("hDiff")
hDiff.Add(hDown, -1)
hDiffDown.Add(hDiff)
if isQCDMC:
for hUp in hQCDUps:
hDiff = hUp.Clone("hDiff")
hDiff.Add(qcdMCHist[0], -1)
hDiffUp.Add(hDiff)
for hDown in hQCDDowns:
hDiff = qcdMCHist[0].Clone("hDiff")
hDiff.Add(hDown, -1)
hDiffDown.Add(hDiff)
#Get unc band for the top plot
uncGraphTop = getUncBand(hSumAllBkg, hDiffUp, hDiffDown,False)
uncGraphTop.SetFillColor(2);
uncGraphTop.SetFillStyle(3001);
uncGraphTop.Draw(" E2 same ");
#Draw data
decoHist(dataHist[0], xTitle, yTitle, Samples["Data"][1])
dataHist[0].SetMarkerStyle(20)
if isData:
dataHist[0].Draw("EPsame")
#Draw legend
hForLegend = sortHists(hAllBkgs, True)
plotLegend = getLegend(dataHist, hForLegend, uncGraphTop)
plotLegend.Draw()
#Draw CMS, Lumi, channel
if channel in ["mu", "Mu", "m"]:
chName = "#mu + jets"
else:
chName = "e + jets"
crName = formatCRString(CR)
if CR=="":
chName = "%s, SR"%chName
else:
chName = "%s, CR"%chName
chCRName = "#splitline{#font[42]{%s}}{#font[42]{%s}}"%(chName, crName)
extraText = "#splitline{Preliminary}{%s}"%chCRName
#CMS_lumi(canvas, iPeriod, iPosX, extraText)
lumi_13TeV = "35.9 fb^{-1}"
if "16" in year:
lumi_13TeV = "35.9 fb^{-1} (2016)"
if "17" in year:
lumi_13TeV = "41.5 fb^{-1} (2017)"
if "18" in year:
lumi_13TeV = "59.7 fb^{-1} (2018)"
CMS_lumi(lumi_13TeV, canvas, iPeriod, iPosX, extraText)
#Draw the ratio of data and all background
if isData and isRatio:
canvas.cd(2)
gPad.SetTopMargin(padGap);
gPad.SetBottomMargin(0.30);
gPad.SetRightMargin(0.03);
#gPad.SetTickx(0);
gPad.SetPad(xPadRange[0],yPadRange[0],xPadRange[1],yPadRange[2]);
gPad.RedrawAxis();
hRatio = dataHist[0].Clone("hRatio")
hRatio.Divide(hSumAllBkg)
decoHistRatio(hRatio, xTitle, "Obs./Exp.", 1)
hRatio.Draw()
uncGraphRatio = getUncBand(hSumAllBkg, hDiffUp, hDiffDown,True)
uncGraphRatio.SetFillColor(2);
uncGraphRatio.SetFillStyle(3001);
uncGraphRatio.Draw("E2same");
baseLine = TF1("baseLine","1", -100, 2000);
#baseLine.SetLineColor(kRed+1);
baseLine.SetLineColor(3);
baseLine.Draw("SAME");
hRatio.Draw("same")
#canvas.SaveAs("%s/%s.pdf"%(outPlotFullDir, hName))
canvas.SaveAs("%s/%s_%s_%s.png"%(outPlotFullDir, hName, year, channel))
def compareUpperLimits(labels, masses, filelabels, **kwargs):
print color("compareUpperLimits()", color="magenta", prepend=">>>\n>>> ")
# SIGNAL strength & mass
extralabel = kwargs.get('extralabel', "")
plotlabel = kwargs.get('plotlabel', "")
# LOOP over LABELS
for label in labels:
print color("plotUpperLimits - %s" % (label),
color="grey",
prepend="\n>>> ")
ymax = -99999
ymin = 99999
medians = []
for filelabel, title in filelabels:
N = len(masses)
median = TGraph(N) # median line
median.SetTitle(title)
medians.append(median)
for i, mass in enumerate(masses):
filename = getOutputFilename(label, mass, extralabel=filelabel)
limits = getLimits(filename)
median.SetPoint(i, mass, limits[2]) # median
if limits[2] > ymax: ymax = limits[2]
if limits[2] < ymin: ymin = limits[2]
median0 = medians[0]
doLog = ymin and ymax / ymin > 6
ymax = ymax * 1.40
xtitle = "m_{X} [GeV]"
ytitle = "95% upper limit on #sigma#timesBR(X #rightarrow #tau#tau) [pb]"
if "bbA" in filelabels[0][0]:
xtitle = "m_{A} [GeV]"
ytitle = "95% upper limit on #sigma#timesBR(A #rightarrow #tau#tau) [pb]"
W, H = 800, 800
T, B = 0.10 * H, 0.16 * H
L, R = 0.12 * W, 0.04 * W
# MAIN plot
canvas = TCanvas("canvas", "canvas", 100, 100, W, H)
canvas.Divide(2)
canvas.cd(1)
gPad.SetPad("pad1", "pad1", 0, 0.37, 1, 1, 0, -1, 0)
gPad.SetTopMargin(T / H)
gPad.SetBottomMargin(0.01)
gPad.SetLeftMargin(L / W)
gPad.SetRightMargin(R / W)
gPad.SetTickx(0)
gPad.SetTicky(0)
gPad.SetGrid()
if doLog:
ymin = 10**(floor(log(ymin, 10)))
ymax = 10**(ceil(log(ymax, 10)))
gPad.SetLogy()
else:
ymin *= 0 if ymin > 0 else 1.20
width = 0.25
height = 0.05 + len(medians) * 0.05
#x1 = 0.16; x2 = x1 + width # Left
x2 = 0.70
x1 = x2 - width # Right
x2 = x1 + width
y1 = 0.86
y2 = y1 - height
legend = TLegend(x1, y1, x2, y2)
frame = gPad.DrawFrame(min(masses), ymin, max(masses), ymax)
frame.GetYaxis().CenterTitle()
frame.GetYaxis().SetTitleSize(0.055)
frame.GetXaxis().SetTitleSize(0.058)
frame.GetXaxis().SetLabelSize(0.050 * 0)
frame.GetYaxis().SetLabelSize(0.053)
frame.GetXaxis().SetLabelOffset(0.012)
frame.GetXaxis().SetTitleOffset(1.02)
frame.GetYaxis().SetTitleOffset(1.08)
frame.GetXaxis().SetNdivisions(508)
frame.GetYaxis().CenterTitle(True)
frame.GetYaxis().SetTitle(ytitle)
frame.GetXaxis().SetTitle(xtitle)
option = 'L'
for i, median in enumerate(medians):
median.SetLineColor(colors[i % len(colors)])
median.SetLineStyle(styles[i % len(styles)])
median.SetLineWidth(2 if styles[i % len(styles)] != kDotted else 3)
median.Draw('L')
legend.AddEntry(median, median.GetTitle(), 'L')
if i == 0: option += ' SAME'
CMS_lumi.CMS_lumi(gPad, 13, 0)
gPad.SetTicks(1, 1)
gPad.Modified()
frame.Draw('sameaxis')
legend.SetFillStyle(0)
legend.SetBorderSize(0)
legend.SetTextSize(0.041)
legend.SetTextFont(62)
legend.SetHeader("%s" % (label_dict[label + extralabel]))
legend.SetTextFont(42)
legend.Draw()
# RATIO plot
canvas.cd(2)
gPad.SetPad("pad2", "pad2", 0, 0, 1, 0.36, 0, -1, 0)
gPad.SetTopMargin(0.05)
gPad.SetBottomMargin(0.24)
gPad.SetLeftMargin(L / W)
gPad.SetRightMargin(R / W)
frame_ratio = gPad.DrawFrame(min(masses), 0.36, max(masses), 1.20)
frame_ratio.GetYaxis().CenterTitle()
frame_ratio.GetYaxis().SetTitleSize(0.055 * 1.79)
frame_ratio.GetXaxis().SetTitleSize(0.058 * 1.79)
frame_ratio.GetXaxis().SetLabelSize(0.052 * 1.79)
frame_ratio.GetYaxis().SetLabelSize(0.050 * 1.79)
frame_ratio.GetXaxis().SetLabelOffset(0.012)
frame_ratio.GetXaxis().SetTitleOffset(1.00)
frame_ratio.GetYaxis().SetTitleOffset(0.63)
frame_ratio.GetXaxis().SetNdivisions(508)
frame_ratio.GetYaxis().CenterTitle(True)
frame_ratio.GetYaxis().SetTitle("ratio")
frame_ratio.GetXaxis().SetTitle(xtitle)
frame_ratio.GetYaxis().SetNdivisions(5)
option = 'L'
ratios = []
for i, median in enumerate(medians):
ratio = makeRatioTGraphs(median, median0)
ratio.SetLineColor(median.GetLineColor())
ratio.SetLineStyle(median.GetLineStyle())
ratio.SetLineWidth(median.GetLineWidth())
ratio.Draw('L')
ratios.append(ratio)
if i == 1: option += ' SAME'
gPad.SetTicks(1, 1)
gPad.Modified()
frame_ratio.Draw('sameaxis')
print " "
canvas.SaveAs("%s/upperLimit-%s%s_compare.png" %
(PLOTS_DIR, label, plotlabel))
canvas.SaveAs("%s/upperLimit-%s%s_compare.pdf" %
(PLOTS_DIR, label, plotlabel))
canvas.Close()
def MakeOneHist(varXaxis, varConst, xvalues, constval):
if len(xvalues)<=1:
return # Do not make plot if there is only 1 xvalue
Legend = TLegend()
Legend.SetBorderSize(0)
Legend.SetFillColor(0)
Legend.SetFillStyle(0)
Canvas = TCanvas("sigEff_"+varConst+str(constval)+"_vs_"+varXaxis)
Histograms = []
LegendEntries = []
colorIndex = 0
for source in input_sources: # loop over different input sources in config file
# print "Debug: running over input_source from: ", source['condor_dir'], " for varXaxis=", varXaxis, ", varConst=", varConst
# Create histogram of efficiencies with appropriate binning
xAxisBins = array('d')
xBinWidth = xvalues[1] - xvalues[0]
for xval in xvalues:
xAxisBins.append(xval - xBinWidth/2.0)
xAxisBins.append(xvalues[-1] + xBinWidth/2.0) # add an extra bin boundary for the last bin.
Histogram = TH1F("Histogram", ";x title;signal efficiency", len(xvalues), xAxisBins)
Histogram.SetDirectory(0)
# Now fill the histogram of efficiencies:
for xval in xvalues:
if varXaxis=="mass" and varConst=="lifetime":
mass = xval
lifetime = constval
elif varXaxis=="lifetime" and varConst=="mass":
lifetime = xval
mass = constval
else:
print "Unrecognized value of varXaxis=", varXaxis
return
dataset_file = "condor/%s/%s.root" % (source['condor_dir'],dataset)
dataset_file = dataset_file.replace("MASS", str(mass))
dataset_file = dataset_file.replace("LIFETIME", str(lifetime))
inputFile = TFile(dataset_file)
HistCutflow = inputFile.Get("OSUAnalysis/" + source['channel'] + "CutFlow")
if not HistCutflow:
print "WARNING: Could not find histogram OSUAnalysis/" + source['channel'] + "CutFlow in file " + dataset_file + ". Will skip it and continue."
return
# calculate efficiency
nbinsCutflow = HistCutflow.GetNbinsX()
if arguments.xsecFile:
xsec = float(signal_cross_sections[str(mass)]['value'])
denom = xsec * intLumi
else:
denom = HistCutflow.GetBinContent(1) # denominator is the first entry in the cutflow histogram
eff = HistCutflow.GetBinContent(nbinsCutflow) / denom
effErr = HistCutflow.GetBinError (nbinsCutflow) / denom
ibin = Histogram.FindBin(xval)
Histogram.SetBinContent(ibin, eff)
Histogram.SetBinError (ibin, effErr)
print "Setting bin content for varConst = ", varConst, " = ", constval, " in ibin=", ibin, ", xval=", xval, ", eff=", eff, " +- ", effErr, ", fractional error =", effErr/eff, " numerator = ", HistCutflow.GetBinContent(nbinsCutflow), " denom = ", denom, " from file: ", dataset_file
inputFile.Close()
if varXaxis=="mass":
xAxisLabel = "chargino mass [GeV]"
elif varXaxis=="lifetime":
xAxisLabel = "chargino c#tau [cm]"
else:
print "Unrecognized value of varXaxis=", varXaxis
return
yAxisLabel = "efficiency"
if 'color' in source:
Histogram.SetMarkerColor(colors[source['color']])
Histogram.SetLineColor(colors[source['color']])
else:
Histogram.SetMarkerColor(colors[colorList[colorIndex]])
Histogram.SetLineColor(colors[colorList[colorIndex]])
colorIndex = colorIndex + 1
if colorIndex is len(colorList):
colorIndex = 0
markerStyle = 20
if 'marker' in source:
markerStyle = markers[source['marker']]
if 'fill' in source:
markerStyle = markerStyle + fills[source['fill']]
Histogram.SetMarkerStyle(markerStyle)
Histogram.SetMarkerSize(1)
# Histogram.SetLineWidth(line_width)
Histogram.SetFillStyle(0)
LegendEntries.append(source['legend_entry'])
Histograms.append(Histogram)
# Finish loop over input_sources
# formatting histograms and adding to legend
legendIndex = 0
for histogram in Histograms:
# print "Adding entry to legend: ", LegendEntries[legendIndex]
# Legend.AddEntry(histogram,LegendEntries[legendIndex],"LEP")
Legend.AddEntry(histogram,LegendEntries[legendIndex],"P")
legendIndex = legendIndex+1
# finding the maximum value of anything going on the canvas, so we know how to set the y-axis
finalMax = 0
for histogram in Histograms:
# currentMax = histogram.GetMaximum() + histogram.GetBinError(histogram.GetMaximumBin())
currentMax = histogram.GetMaximum()
if(currentMax > finalMax):
finalMax = currentMax
finalMax = 1.5*finalMax
if finalMax is 0:
finalMax = 1
## if arguments.setYMax:
## finalMax = float(arguments.setYMax)
# Drawing histograms to canvas
makeRatioPlots = arguments.makeRatioPlots
makeDiffPlots = arguments.makeDiffPlots
yAxisMin = 0.0
## if arguments.setYMin:
## yAxisMin = float(arguments.setYMin)
if makeRatioPlots or makeDiffPlots:
Canvas.SetFillStyle(0)
Canvas.Divide(1,2)
Canvas.cd(1)
gPad.SetPad(0,0.25,1,1)
gPad.SetMargin(0.15,0.05,0.01,0.07)
gPad.SetFillStyle(0)
gPad.Update()
gPad.Draw()
## if arguments.setLogY:
## gPad.SetLogy()
Canvas.cd(2)
gPad.SetPad(0,0,1,0.25)
#format: gPad.SetMargin(l,r,b,t)
gPad.SetMargin(0.15,0.05,0.4,0.01)
gPad.SetFillStyle(0)
gPad.SetGridy(1)
gPad.Update()
gPad.Draw()
Canvas.cd(1)
histCounter = 0
plotting_options = "pe, x0" # x0 suppresses the error bar along x
## if arguments.plot_hist:
## plotting_options = "HIST"
for histogram in Histograms:
# histogram.SetTitle(histoTitle)
histogram.Draw(plotting_options)
histogram.GetXaxis().SetTitle(xAxisLabel)
histogram.GetYaxis().SetTitle(yAxisLabel)
if histogram.InheritsFrom("TH1"):
histogram.SetMaximum(finalMax)
histogram.SetMinimum(yAxisMin)
if makeRatioPlots or makeDiffPlots:
histogram.GetXaxis().SetLabelSize(0)
if histCounter is 0:
if histogram.InheritsFrom("TH1"):
plotting_options = plotting_options + " SAME"
elif histogram.InheritsFrom("TGraph"):
plotting_options = "P"
histCounter = histCounter + 1
x_left = 0.15
x_right = 0.55
y_min = 0.6
y_max = 0.8
Legend.SetX1NDC(x_left)
Legend.SetY1NDC(y_min)
Legend.SetX2NDC(x_right)
Legend.SetY2NDC(y_max)
Legend.Draw()
customText = "chargino " + varConst + " " + str(constval)
if varConst == "mass":
customText += " GeV"
elif varConst == "lifetime":
customText += " cm"
customText = customText.replace("lifetime", "c#tau")
CustomLabel = TPaveLabel(x_left, 0.8, x_right, 0.9, customText, "NDC")
CustomLabel.SetBorderSize(0)
CustomLabel.SetFillColor(0)
CustomLabel.SetFillStyle(0)
CustomLabel.Draw()
#drawing the ratio or difference plot if requested
if makeRatioPlots or makeDiffPlots:
Canvas.cd(2)
if makeRatioPlots:
# Comparison = ratioHistogram(Histograms[0],Histograms[1], 1000)
Comparison = Histograms[0].Clone()
Comparison.Divide(Histograms[1])
Comparison.GetYaxis().SetTitle("ratio")
elif makeDiffPlots:
Comparison = Histograms[0].Clone("diff")
Comparison.Add(Histograms[1],-1)
Comparison.SetTitle("")
Comparison.GetYaxis().SetTitle("hist1-hist2")
Comparison.GetXaxis().SetTitle(xAxisLabel)
Comparison.GetYaxis().CenterTitle()
Comparison.GetYaxis().SetTitleSize(0.1)
Comparison.GetYaxis().SetTitleOffset(0.5)
Comparison.GetXaxis().SetTitleSize(0.15)
Comparison.GetYaxis().SetLabelSize(0.1)
Comparison.GetXaxis().SetLabelSize(0.15)
if makeRatioPlots:
# RatioYRange = 1.15
if arguments.ratioYRange:
RatioYRange = float(arguments.ratioYRange)
Comparison.GetYaxis().SetRangeUser(0.7, 1.3)
elif makeDiffPlots:
YMax = Comparison.GetMaximum()
YMin = Comparison.GetMinimum()
if YMax <= 0 and YMin <= 0:
Comparison.GetYaxis().SetRangeUser(-1.2*YMin,0)
elif YMax >= 0 and YMin >= 0:
Comparison.GetYaxis().SetRangeUser(0,1.2*YMax)
else: #axis crosses y=0
if abs(YMax) > abs(YMin):
Comparison.GetYaxis().SetRangeUser(-1.2*YMax,1.2*YMax)
else:
Comparison.GetYaxis().SetRangeUser(-1.2*YMin,1.2*YMin)
Comparison.GetYaxis().SetNdivisions(205)
# Comparison.Draw("E0")
Comparison.Draw("PE, X0") # X0 suppresses error bar in x direction
outputFile.cd()
Canvas.Write()
if arguments.savePDFs:
Canvas.SaveAs("efficiency_histograms_pdfs/"+histogramName+".pdf")
def checkDataPoissonErrors():
'''Check on behaviour of Poisson errors in histograms.'''
N = 100
hist0 = TH1D("data0", "default", N, 0, N)
hist1 = TH1D("data1", "TH1.kPoisson", N, 0, N)
hist2 = TH1D("data2", "#chi^{2} quantile", N, 0, N)
for i in xrange(0, N + 2):
y = max(0, i - 1)
hist0.SetBinContent(i, y)
hist1.SetBinContent(i, y)
hist2.SetBinContent(i, y)
# GET data Possoin errors
graph1 = getDataPoissonErrors(hist1,
drawZeroBins=True,
centerBin=False,
kPoisson=True)
graph2 = getDataPoissonErrors(hist2,
drawZeroBins=True,
centerBin=False,
kPoisson=False)
# FILL table
table = []
table.append(
"%22s %14s %14s %14s" %
(" ", "default".center(14), "kPoisson".center(14), "chi^2".center(14)))
table.append("%4s %7s %7s %6s %6s %6s %6s %6s %6s" %
("bin", "content", "sqrt(N)", "errLow", "errUp", "errLow",
"errUp", "errLow", "errUp"))
for i in xrange(1, N + 1):
y = hist0.GetBinContent(i)
table.append(
"%4d %5d %7.2f %6.2f %6.2f %6.2f %6.2f %6.2f %6.2f" %
(i, y, sqrt(y), hist0.GetBinErrorLow(i), hist0.GetBinErrorUp(i),
graph1.GetErrorYlow(i - 1), graph1.GetErrorYhigh(i - 1),
graph2.GetErrorYlow(i - 1), graph2.GetErrorYhigh(i - 1)))
# PRINT table
for line in table:
print ">>> " + line
print ">>> "
# GRAPHS
function = TF1("function1", "sqrt(x)", 0, N)
function.SetTitle("#sqrt{N}")
graphElow1, graphEup1 = makeErrorGraph(graph1)
graphElow2, graphEup2 = makeErrorGraph(graph2)
graphElow1.SetTitle("TH1.kPoisson low")
graphEup1.SetTitle("TH1.kPoisson high")
graphElow2.SetTitle("N-F^{-1}_{#chi^{2}}(1-#alpha,2N)/2")
graphEup2.SetTitle("F^{-1}_{#chi^{2}}(#alpha,2(N-1))/2-N")
graphs = [function, graphElow1, graphEup1, graphElow2, graphEup2]
# DRAW settings
W, H = 800, 800
T, B = 0.04 * H, 0.14 * H
L, R = 0.12 * W, 0.04 * W
xmin, xmax = -N * 0.01, N * 1.01
ymax = function.GetMaximum() * 1.16
width = 0.25
legendTextSize = 0.045
height = 0.062 * (1 + len(graphs))
x2 = 0.86
x1 = x2 - width
y1 = 0.20
y2 = y1 + height
# CANVAS
canvas = TCanvas("canvas", "canvas", 100, 100, W, H)
canvas.Divide(2)
canvas.cd(1)
gPad.SetPad("pad1", "pad1", 0, 0.37, 1, 1, 0, -1, 0)
gPad.SetTopMargin(T / H)
gPad.SetBottomMargin(0.01)
gPad.SetLeftMargin(L / W)
gPad.SetRightMargin(R / W)
gPad.SetTickx(0)
gPad.SetTicky(0)
gPad.SetGrid()
# FRAME
frame = gPad.DrawFrame(xmin, 0, xmax, ymax)
frame.GetYaxis().SetTitleSize(0.055)
frame.GetXaxis().SetTitleSize(0.058)
frame.GetXaxis().SetLabelSize(0.050 * 0)
frame.GetYaxis().SetLabelSize(0.053)
frame.GetXaxis().SetLabelOffset(0.012)
frame.GetXaxis().SetTitleOffset(1.02)
frame.GetYaxis().SetTitleOffset(1.08)
frame.GetXaxis().SetNdivisions(505)
frame.GetXaxis().SetTitle("number of events")
frame.GetYaxis().SetTitle("error on number of events")
legend = TLegend(x1, y1, x2, y2)
# DRAW
colors = [kRed, kBlue, kViolet, kYellow, kGreen]
for i, graph in enumerate(graphs):
style = kSolid
width = 2
if i < 3: width = 3
if i > 2: style = kDashed
graph.SetLineColor(colors[i % len(colors)])
graph.SetLineStyle(style)
graph.SetLineWidth(width)
graph.Draw('L SAME')
legend.AddEntry(graph, graph.GetTitle(), 'L')
#CMS_lumi.CMS_lumi(gPad,13,0)
gPad.SetTicks(1, 1)
gPad.Modified()
frame.Draw('SAME AXIS')
# LEGEND
legend.SetFillStyle(0)
legend.SetBorderSize(0)
legend.SetTextSize(legendTextSize)
legend.SetTextFont(62)
legend.SetHeader("method")
legend.SetTextFont(42)
legend.Draw()
# CHI2 TEXT
text1 = "#splitline{#alpha = (1-0.6827)/2}{upper #chi^{2} CDF F_{#chi^{2}}}"
text = TLatex()
text.SetNDC()
text.SetTextFont(42)
text.SetTextSize(legendTextSize)
text.SetTextAlign(13) # centered: 22
xoffset = (legend.GetX2() - legend.GetX1()) * legend.GetMargin() * 1.04
text.DrawLatexNDC(legend.GetX1() + xoffset, legend.GetY1() - 0.012, text1)
# RATIO plot
canvas.cd(2)
gPad.SetPad("pad2", "pad2", 0, 0, 1, 0.36, 0, -1, 0)
gPad.SetTopMargin(0.05)
gPad.SetBottomMargin(0.24)
gPad.SetLeftMargin(L / W)
gPad.SetRightMargin(R / W)
# FRAME ratio
frame_ratio = gPad.DrawFrame(xmin, 0.50, xmax, 1.50)
frame_ratio.GetYaxis().CenterTitle()
frame_ratio.GetYaxis().SetTitleSize(0.055 * 1.79)
frame_ratio.GetXaxis().SetTitleSize(0.058 * 1.79)
frame_ratio.GetXaxis().SetLabelSize(0.052 * 1.79)
frame_ratio.GetYaxis().SetLabelSize(0.050 * 1.79)
frame_ratio.GetXaxis().SetLabelOffset(0.012)
frame_ratio.GetXaxis().SetTitleOffset(1.00)
frame_ratio.GetYaxis().SetTitleOffset(0.63)
frame_ratio.GetXaxis().SetNdivisions(508)
frame_ratio.GetYaxis().CenterTitle(True)
frame_ratio.GetYaxis().SetTitle("ratio")
frame_ratio.GetXaxis().SetTitle("number of events N")
frame_ratio.GetYaxis().SetNdivisions(5)
# DRAW ratio
ratios = []
graphf = makeGraphFromTF1(function, N)
for i, graph in enumerate(graphs):
if isinstance(graph, TF1): graph = makeGraphFromTF1(graph, N)
ratio = makeRatioTGraphs(graph, graphf)
ratio.SetLineColor(graph.GetLineColor())
ratio.SetLineStyle(graph.GetLineStyle())
ratio.SetLineWidth(graph.GetLineWidth())
ratio.Draw('L SAME')
ratios.append(ratio)
gPad.SetTicks(1, 1)
gPad.Modified()
frame_ratio.Draw('sameaxis')
# SAVE
canvas.SaveAs("PoissonError.png")
canvas.Close()
def smear(mu, sigma0, smearfactor, N=100000, lcut=None, ucut=None, nbins=80):
print ">>> smearing for N(%s,%s) with a factor of %s" % (mu, sigma0,
smearfactor)
# HISTS
xmin, xmax = mu - sigma0 * 5, mu + sigma0 * 4
sigma1 = sigma0 * (1 + smearfactor)
histname0 = "unsmeared"
histname1 = "smeared"
histtitle0 = "unsmeared, #sigma_{0} = %s" % (sigma0)
histtitle1 = "smeared, #sigma_{new} = %s" % (sigma1)
hist0 = TH1F(histname0, histtitle0, nbins, xmin, xmax)
hist1 = TH1F(histname1, histtitle1, nbins, xmin, xmax)
# FIT FUNCTIONS
xminf = xmin if lcut == None else lcut
xmaxf = xmax if ucut == None else ucut
gaus0 = TF1("gaus0", "gaus", xminf, xmaxf)
gaus1 = TF1("gaus1", "gaus", xminf, xmaxf)
gaus0.SetTitle("%s fit" % histname0)
gaus1.SetTitle("%s fit" % histname1)
gaus0.SetParameters(N, mu, sigma0)
gaus1.SetParameters(N, mu, sigma1)
#gaus0.SetParLimits(2,sigma0*0.9,sigma0*1.1)
#gaus1.SetParLimits(2,sigma1*0.9,sigma1*1.1)
hists = [(hist0, gaus0), (hist1, gaus1)]
# SMEAR & FILL
#sigma1 = smearfactor
#sigma1 = (smearfactor**2)/2.
#sigma1 = sqrt(2.*(smearfactor))
#sigma1 = sqrt(-2.*log(smearfactor))
#sigma1 = 1./(2*smearfactor**2)
sigma2 = sqrt(sigma1**2 - sigma0**2)
print ">>> sigma0 = %.3f, sigma1 = %.3f, sigma2 = %.3f" % (
sigma0, sigma1, sigma2)
print ">>> generating %s events..." % (N)
for i in xrange(N):
xval0 = gRandom.Gaus(mu, sigma0)
if lcut != None and xval0 < lcut: continue
if ucut != None and xval0 > ucut: continue
#rand = gRandom.Gaus(1,smearfactor)
#xval1 = xval0 * rand
#rand = gRandom.Gaus(0,1+smearfactor)
#xval1 = xval0 + rand
rand = gRandom.Gaus(0, 1)
xval1 = xval0 + sigma2 * rand
hist0.Fill(xval0)
if lcut != None and xval1 < lcut: continue
if ucut != None and xval1 > ucut: continue
hist1.Fill(xval1)
# PLOT SETTINGS
xtitle = "x variable"
ytitle = "events"
title = "Gauss(%s,%s)" % (mu, "#sigma")
canvasname = "smear_%.1f-%.1f_by_%.2f" % (mu, sigma0, smearfactor)
if lcut != None: canvasname += "_gt%.1f" % (lcut)
if ucut != None: canvasname += "_lt%.1f" % (ucut)
canvasname = canvasname.replace('.', 'p')
ymin, ymax = 0, 1.14 * max(hist0.GetMaximum(), hist1.GetMaximum())
rmin, rmax = 0.60, 1.40
lmargin, rmargin = 0.14, 0.04
# CANVAS
print ">>> plotting..."
canvas = TCanvas("canvas", "canvas", 100, 100, 800, 800)
canvas.Divide(2)
# MAIN plot
canvas.cd(1)
gPad.SetPad("pad1", "pad1", 0, 0.33, 1, 1, 0, -1, 0)
gPad.SetFillColor(0)
gPad.SetBorderMode(0)
gPad.SetFrameFillStyle(0)
gPad.SetFrameBorderMode(0)
gPad.SetTopMargin(0.06)
gPad.SetBottomMargin(0.02)
gPad.SetLeftMargin(lmargin)
gPad.SetRightMargin(rmargin)
gPad.SetGrid()
gPad.cd()
textsize = 0.050
x1, width = 0.18, 0.25
y1, height = 0.89, textsize * 1.08 * 5
legend = TLegend(x1, y1, x1 + width, y1 - height)
legend.SetTextSize(textsize)
legend.SetBorderSize(0)
legend.SetFillStyle(0)
legend.SetFillColor(0)
legend.SetTextFont(62)
legend.SetHeader(title)
legend.SetTextFont(42)
# FRAME
frame = gPad.DrawFrame(xmin, ymin, xmax, ymax)
frame.GetYaxis().SetTitleSize(0.070)
frame.GetXaxis().SetTitleSize(0.070)
frame.GetXaxis().SetLabelSize(0.000)
frame.GetYaxis().SetLabelSize(0.060)
frame.GetXaxis().SetTitleOffset(1.00)
frame.GetYaxis().SetTitleOffset(1.06)
frame.GetXaxis().SetNdivisions(508)
frame.GetXaxis().SetTitle(xtitle)
frame.GetYaxis().SetTitle(ytitle)
# DRAW & FIT
print ">>> fitting..."
fits = []
for i, (hist, gaus) in enumerate(hists):
color = colors[i % len(colors)]
hist.SetLineColor(color)
hist.SetLineWidth(2)
hist.SetLineStyle(1)
gaus.SetLineColor(color + 1)
gaus.SetLineWidth(2)
gaus.SetLineStyle(2)
hist.Fit(gaus.GetName(), '0', '', xminf, xmaxf)
hist.Draw('SAME')
gaus.Draw('SAME')
gtitle = "#sigma_{fit} = %.3f" % (gaus.GetParameter(2)
) #gaus.GetTitle()
legend.AddEntry(hist, hist.GetTitle(), 'l')
legend.AddEntry(gaus, gtitle, 'l')
print ">>> real unsmeared sigma: %5.3f" % (sigma0)
print ">>> fitted unsmeared sigma: %5.3f" % (gaus0.GetParameter(2))
print ">>> real smear factor: %5.3f" % (smearfactor)
print ">>> fitted smeared sigma: %5.3f" % (gaus1.GetParameter(2))
legend.Draw()
frame.Draw('SAMEAXIS')
gPad.Modified()
# RATIO plot
canvas.cd(2)
gPad.SetPad("pad2", "pad2", 0, 0, 1, 0.32, 0, -1, 0)
gPad.SetTopMargin(0.04)
gPad.SetBottomMargin(0.29)
gPad.SetLeftMargin(lmargin)
gPad.SetRightMargin(rmargin)
# RATIO FRAME
rframe = gPad.DrawFrame(xmin, rmin, xmax, rmax)
rframe.GetYaxis().CenterTitle()
rframe.GetXaxis().SetTitleSize(0.144)
rframe.GetYaxis().SetTitleSize(0.140)
rframe.GetXaxis().SetLabelSize(0.130)
rframe.GetYaxis().SetLabelSize(0.120)
rframe.GetXaxis().SetLabelOffset(0.012)
rframe.GetXaxis().SetTitleOffset(0.85)
rframe.GetYaxis().SetTitleOffset(0.53)
rframe.GetXaxis().SetNdivisions(508)
rframe.GetYaxis().CenterTitle(True)
rframe.GetYaxis().SetTitle("ratio")
rframe.GetXaxis().SetTitle(xtitle)
rframe.GetYaxis().SetNdivisions(5)
# RATIO
ratios = []
for i, (hist, gaus) in enumerate(hists):
#if i==0: continue
#ratio = hist.Clone(hist.GetName()+"_ratio")
#ratio.Divide(hist0)
ratio = divideHists(hist, hist0, name=hist.GetName() + "_ratio")
ratio.Draw('HISTSAME')
ratios.append(ratio)
#ratiof = createTH1FromTF1(gaus,nbins,xmin,xmax)
#ratiof.Divide(hist0)
ratiof = divideTF1ByTH1(gaus, hist0, name=gaus.GetName() + "_ratio")
ratiof.Draw('HISTSAME')
ratios.append(ratiof)
print ratiof
#line = TLine(xmin,1.,xmax,1.)
#line.SetLineColor(hist0.GetLineColor())
#line.SetLineWidth(hist0.GetLineWidth())
#line.SetLineStyle(1)
#line.Draw('SAME')
gPad.SetGrid()
gPad.Modified()
rframe.Draw('sameaxis')
canvas.SaveAs(canvasname + ".png")
canvas.SaveAs(canvasname + ".pdf")
canvas.Close()
print ">>> "
def MakeOneDHist(histogramName,integrateDir):
HeaderLabel = TPaveLabel(header_x_left,header_y_bottom,header_x_right,header_y_top,HeaderText,"NDC")
HeaderLabel.SetTextAlign(32)
HeaderLabel.SetBorderSize(0)
HeaderLabel.SetFillColor(0)
HeaderLabel.SetFillStyle(0)
LumiLabel = TPaveLabel(topLeft_x_left,topLeft_y_bottom,topLeft_x_right,topLeft_y_top,LumiText,"NDC")
LumiLabel.SetBorderSize(0)
LumiLabel.SetFillColor(0)
LumiLabel.SetFillStyle(0)
NormLabel = TPaveLabel()
NormLabel.SetDrawOption("NDC")
NormLabel.SetX1NDC(topLeft_x_left)
NormLabel.SetX2NDC(topLeft_x_right)
NormLabel.SetBorderSize(0)
NormLabel.SetFillColor(0)
NormLabel.SetFillStyle(0)
NormText = ""
if arguments.normalizeToUnitArea:
NormText = "Scaled to unit area"
Legend = TLegend()
Legend.SetBorderSize(0)
Legend.SetFillColor(0)
Legend.SetFillStyle(0)
canvasName = histogramName
if integrateDir is "left":
canvasName += "_CumulativeLeft"
elif integrateDir is "right":
canvasName += "_CumulativeRight"
Canvas = TCanvas(canvasName)
Histograms = []
LegendEntries = []
colorIndex = 0
for source in input_sources: # loop over different input sources in config file
dataset_file = "condor/%s/%s.root" % (source['condor_dir'],source['dataset'])
inputFile = TFile(dataset_file)
if arguments.generic:
HistogramObj = inputFile.Get(source['channel'] + "/" +histogramName)
else:
HistogramObj = inputFile.Get("OSUAnalysis/" + source['channel'] + "/" +histogramName)
if not HistogramObj:
print "WARNING: Could not find histogram " + source['channel'] + "/" + histogramName + " in file " + dataset_file + ". Will skip it and continue."
return
Histogram = HistogramObj.Clone()
Histogram.SetDirectory(0)
inputFile.Close()
if arguments.rebinFactor:
RebinFactor = int(arguments.rebinFactor)
#don't rebin histograms which will have less than 5 bins or any gen-matching histograms
if Histogram.GetNbinsX() >= RebinFactor*5 and Histogram.GetTitle().find("GenMatch") is -1:
Histogram.Rebin(RebinFactor)
xAxisLabel = Histogram.GetXaxis().GetTitle()
unitBeginIndex = xAxisLabel.find("[")
unitEndIndex = xAxisLabel.find("]")
xAxisLabelVar = xAxisLabel
if unitBeginIndex is not -1 and unitEndIndex is not -1: #x axis has a unit
yAxisLabel = "Entries / " + str(Histogram.GetXaxis().GetBinWidth(1)) + " " + xAxisLabel[unitBeginIndex+1:unitEndIndex]
xAxisLabelVar = xAxisLabel[0:unitBeginIndex]
else:
yAxisLabel = "Entries per bin (" + str(Histogram.GetXaxis().GetBinWidth(1)) + " width)"
if arguments.normalizeToUnitArea:
yAxisLabel = yAxisLabel + " (Unit Area Norm.)"
if arguments.normalizeToUnitArea and arguments.makeSignificancePlots:
unit = "Efficiency"
else:
unit = "Yield"
if integrateDir is "left":
yAxisLabel = unit + ", " + xAxisLabelVar + "< x (" + str(Histogram.GetXaxis().GetBinWidth(1)) + " bin width)"
if integrateDir is "right":
yAxisLabel = unit + ", " + xAxisLabelVar + "> x (" + str(Histogram.GetXaxis().GetBinWidth(1)) + " bin width)"
if not arguments.makeFancy and not arguments.generic:
fullTitle = Histogram.GetTitle()
splitTitle = fullTitle.split(":")
# print splitTitle
histoTitle = splitTitle[1].lstrip(" ")
else:
histoTitle = ""
if 'color' in source:
Histogram.SetMarkerColor(colors[source['color']])
Histogram.SetLineColor(colors[source['color']])
else:
Histogram.SetMarkerColor(colors[colorList[colorIndex]])
Histogram.SetLineColor(colors[colorList[colorIndex]])
colorIndex = colorIndex + 1
if colorIndex is len(colorList):
colorIndex = 0
markerStyle = 20
if 'marker' in source:
markerStyle = markers[source['marker']]
if 'fill' in source:
markerStyle = markerStyle + fills[source['fill']]
Histogram.SetMarkerStyle(markerStyle)
Histogram.SetLineWidth(line_width)
Histogram.SetFillStyle(0)
if arguments.normalizeToUnitArea and Histogram.Integral() > 0:
Histogram.Scale(1./Histogram.Integral())
Histogram = MakeIntegralHist(Histogram, integrateDir)
LegendEntries.append(source['legend_entry'])
Histograms.append(Histogram)
### formatting histograms and adding to legend
legendIndex = 0
for histogram in Histograms:
Legend.AddEntry(histogram,LegendEntries[legendIndex],"LEP")
legendIndex = legendIndex+1
### finding the maximum value of anything going on the canvas, so we know how to set the y-axis
finalMax = 0
for histogram in Histograms:
currentMax = histogram.GetMaximum() + histogram.GetBinError(histogram.GetMaximumBin())
if(currentMax > finalMax):
finalMax = currentMax
finalMax = 1.5*finalMax
if arguments.setYMax:
finalMax = float(arguments.setYMax)
### Drawing histograms to canvas
makeRatioPlots = arguments.makeRatioPlots
makeDiffPlots = arguments.makeDiffPlots
#makeSignifPlots = arguments.makeSignificancePlots
yAxisMin = 0.0001
if arguments.setYMin:
yAxisMin = float(arguments.setYMin)
if makeRatioPlots or makeDiffPlots:
Canvas.SetFillStyle(0)
Canvas.Divide(1,2)
Canvas.cd(1)
gPad.SetPad(0,0.25,1,1)
gPad.SetMargin(0.15,0.05,0.01,0.07)
gPad.SetFillStyle(0)
gPad.Update()
gPad.Draw()
if arguments.setLogY:
gPad.SetLogy()
Canvas.cd(2)
gPad.SetPad(0,0,1,0.25)
#format: gPad.SetMargin(l,r,b,t)
gPad.SetMargin(0.15,0.05,0.4,0.01)
gPad.SetFillStyle(0)
gPad.SetGridy(1)
gPad.Update()
gPad.Draw()
Canvas.cd(1)
histCounter = 0
plotting_options = ""
if arguments.generic:
plotting_options = "p,e"
if arguments.plot_hist:
plotting_options = "HIST"
for histogram in Histograms:
histogram.SetTitle(histoTitle)
if arguments.verbose:
print "Debug: drawing hist " + histogram.GetName() + ", with plotting_options = " + plotting_options + ", with mean = " + str(histogram.GetMean()) + ", with color = " + str(histogram.GetLineColor())
histogram.Draw(plotting_options)
histogram.GetXaxis().SetTitle(xAxisLabel)
histogram.GetYaxis().SetTitle(yAxisLabel)
histogram.SetMaximum(finalMax)
histogram.SetMinimum(yAxisMin)
if makeRatioPlots or makeDiffPlots:
histogram.GetXaxis().SetLabelSize(0)
if histCounter is 0:
plotting_options = plotting_options + " SAME"
histCounter = histCounter + 1
#legend coordinates, empirically determined :-)
x_left = 0.1677852
x_right = 0.9647651
y_min = 0.6765734
y_max = 0.9
Legend.SetX1NDC(x_left)
Legend.SetY1NDC(y_min)
Legend.SetX2NDC(x_right)
Legend.SetY2NDC(y_max)
Legend.Draw()
# Deciding which text labels to draw and drawing them
drawHeaderLabel = False
if arguments.makeFancy:
drawHeaderLabel = True
#now that flags are set, draw the appropriate labels
if drawHeaderLabel:
HeaderLabel.Draw()
#drawing the ratio or difference plot if requested
if makeRatioPlots or makeDiffPlots:
Canvas.cd(2)
if makeRatioPlots:
if arguments.ratioRelErrMax:
Comparison = ratioHistogram(Histograms[0],Histograms[1],float(arguments.ratioRelErrMax))
else:
Comparison = ratioHistogram(Histograms[0],Histograms[1])
elif makeDiffPlots:
Comparison = Histograms[0].Clone("diff")
Comparison.Add(Histograms[1],-1)
Comparison.SetTitle("")
Comparison.GetYaxis().SetTitle("hist1-hist2")
Comparison.GetXaxis().SetTitle(xAxisLabel)
Comparison.GetYaxis().CenterTitle()
Comparison.GetYaxis().SetTitleSize(0.1)
Comparison.GetYaxis().SetTitleOffset(0.5)
Comparison.GetXaxis().SetTitleSize(0.15)
Comparison.GetYaxis().SetLabelSize(0.1)
Comparison.GetXaxis().SetLabelSize(0.15)
if makeRatioPlots:
RatioYRange = 1.15
if arguments.ratioYRange:
RatioYRange = float(arguments.ratioYRange)
Comparison.GetYaxis().SetRangeUser(-1*RatioYRange, RatioYRange)
elif makeDiffPlots:
YMax = Comparison.GetMaximum()
YMin = Comparison.GetMinimum()
if YMax <= 0 and YMin <= 0:
Comparison.GetYaxis().SetRangeUser(-1.2*YMin,0)
elif YMax >= 0 and YMin >= 0:
Comparison.GetYaxis().SetRangeUser(0,1.2*YMax)
else: #axis crosses y=0
if abs(YMax) > abs(YMin):
Comparison.GetYaxis().SetRangeUser(-1.2*YMax,1.2*YMax)
else:
Comparison.GetYaxis().SetRangeUser(-1.2*YMin,1.2*YMin)
Comparison.GetYaxis().SetNdivisions(205)
Comparison.Draw("E0")
outputFile.cd()
Canvas.Write()
if arguments.savePDFs:
Canvas.SaveAs("comparison_histograms_pdfs/"+histogramName+".pdf")