def compareROCs(f1, g1, f2, g2, plotdir):
tf1 = ROOT.TFile.Open(f1)
graph1 = tf1.Get(g1)
graph1.SetName("roc1")
tf2 = ROOT.TFile.Open(f2)
graph2 = tf2.Get(g2)
graph2.SetName("roc2")
c = getCanvas()
graph1.SetMarkerStyle(ROOT.kOpenSquare)
graph1.SetMarkerSize(0.8)
graph1.SetMarkerColor(ROOT.kGray + 2)
graph1.SetLineColor(ROOT.kGray + 2)
graph2.SetMarkerColor(ROOT.kRed + 1)
graph2.SetLineColor(ROOT.kRed + 1)
graph1.Draw('APC')
#graph2.Draw('PC')
graph1.GetXaxis().SetTitle(
"Signal efficiency ( #varepsilon_{S}^{#delta} )")
graph1.GetXaxis().SetTitleFont(42)
graph1.GetXaxis().SetTitleSize(0.05)
graph1.GetXaxis().SetDecimals()
graph1.GetYaxis().SetTitle(
"Background rejection ( 1 - #varepsilon_{B}^{#delta} )")
graph1.GetYaxis().SetTitleFont(42)
graph1.GetYaxis().SetTitleSize(0.05)
graph1.GetYaxis().SetDecimals()
lat50 = ROOT.TLatex()
lat50.SetNDC()
lat50.SetTextFont(42)
lat50.SetTextSize(0.03)
lat50.SetTextColor(ROOT.kRed + 1)
lat50.DrawLatex(0.6, 0.79, "#delta > 10 photons/pixel")
ar50 = ROOT.TArrow(0.52, 0.95, 0.6, 0.95, 0.02, "|>")
ar50.SetLineWidth(2)
ar50.SetLineColor(ROOT.kRed + 1)
ar50.SetFillColor(ROOT.kRed + 1)
ar50.Draw()
lat40 = ROOT.TLatex()
lat40.SetNDC()
lat40.SetTextFont(42)
lat40.SetTextSize(0.03)
lat40.SetTextColor(ROOT.kAzure + 7)
lat40.DrawLatex(0.3, 0.5, "#delta > 11 photons/pixel")
ar40 = ROOT.TArrow(0.39, 0.95, 0.39, 0.6, 0.02, "|>")
ar40.SetLineWidth(2)
ar40.SetLineColor(ROOT.kAzure + 7)
ar40.SetFillColor(ROOT.kAzure + 7)
ar40.Draw()
for ext in ['png', 'pdf']:
c.SaveAs("{plotdir}/comp_roc.{ext}".format(plotdir=plotdir, ext=ext))
def fitDensity(rfile, plotdir):
tf = ROOT.TFile.Open(rfile)
histo1 = tf.Get("density")
histo1.SetMarkerColor(ROOT.kBlack)
histo1.SetLineColor(ROOT.kBlack)
histo1.GetXaxis().SetTitle('#delta (photons/pixel)')
histo1.GetXaxis().SetTitleSize(0.05)
histo1.GetYaxis().SetTitle('superclusters')
histo2 = tf.Get("cosm_density")
histo3 = tf.Get("fe_density")
c = getCanvas()
xmin1, xmax1 = (15, 25)
xmin2, xmax2 = (15, 25)
xmin3, xmax3 = (12, 18)
if histo3.Integral():
histo3.SetMarkerColor(ROOT.kGray + 2)
histo3.SetMarkerStyle(ROOT.kFullCircle)
histo3.SetLineColor(ROOT.kGray + 2)
histo3.Scale(1. / histo3.Integral())
histo3.Sumw2()
histo3.Draw("pe 1")
g3 = ROOT.TF1("g3", "gaus", xmin3, xmax3)
g3.SetLineColor(ROOT.kGray + 2)
histo3.Fit('g3', 'R+S')
histo1.Sumw2()
histo1.SetMaximum(0.5)
histo1.GetYaxis().SetRangeUser(0., 0.5)
histo1.Scale(1. / histo1.Integral())
histo1.Draw("pe 1 same")
g1 = ROOT.TF1("g1", "gaus", xmin1, xmax1)
g1.SetLineColor(ROOT.kBlack)
histo1.Fit('g1', 'R+S')
histo2.SetMarkerColor(ROOT.kTeal + 4)
histo2.SetMarkerStyle(ROOT.kFullCircle)
histo2.SetLineColor(ROOT.kTeal + 4)
histo2.Sumw2()
histo2.Scale(1. / histo2.Integral())
histo2.Draw("pe 1 same")
g2 = ROOT.TF1("g2", "gaus", xmin2, xmax2)
g2.SetLineColor(ROOT.kTeal + 4)
histo2.Fit('g2', 'R+S')
mean = g1.GetParameter(1)
mErr = g1.GetParError(1)
sigma = g1.GetParameter(2)
mSigma = g1.GetParError(2)
lat = ROOT.TLatex()
unit = ''
ndigits = 2
lat.SetNDC()
lat.SetTextFont(42)
lat.SetTextSize(0.03)
lat.DrawLatex(
0.55, 0.80,
"m_{{AmBe}} = {m:.{nd}f} #pm {em:.{nd}f} {unit}".format(m=mean,
em=mErr,
unit=unit,
nd=ndigits))
lat.DrawLatex(
0.55, 0.75,
"#sigma_{{AmBe}} = {s:.{nd}f} #pm {es:.{nd}f} {unit}".format(
s=sigma, es=mSigma, unit=unit, nd=ndigits))
mean = g2.GetParameter(1)
mErr = g2.GetParError(1)
sigma = g2.GetParameter(2)
mSigma = g2.GetParError(2)
lat.DrawLatex(
0.55, 0.60,
"m_{{no source}} = {m:.{nd}f} #pm {em:.{nd}f} {unit}".format(
m=mean, em=mErr, unit=unit, nd=ndigits))
lat.DrawLatex(
0.55, 0.55,
"#sigma_{{no source}} = {s:.{nd}f} #pm {es:.{nd}f} {unit}".format(
s=sigma, es=mSigma, unit=unit, nd=ndigits))
if histo3.Integral():
mean = g3.GetParameter(1)
mErr = g3.GetParError(1)
sigma = g3.GetParameter(2)
mSigma = g3.GetParError(2)
lat.DrawLatex(
0.55, 0.40,
"m_{{Fe}} = {m:.{nd}f} #pm {em:.{nd}f} {unit}".format(m=mean,
em=mErr,
unit=unit,
nd=ndigits))
lat.DrawLatex(
0.55, 0.35,
"#sigma_{{Fe}} = {s:.{nd}f} #pm {es:.{nd}f} {unit}".format(
s=sigma, es=mSigma, unit=unit, nd=ndigits))
for ext in ['png', 'pdf']:
c.SaveAs('{pdir}/cosm_scalefit.{ext}'.format(pdir=plotdir, ext=ext))
def fitFe(rfile, calib=True):
tf = ROOT.TFile.Open(rfile)
histo_sig = tf.Get("energyfe_diff" if calib else "integralfe_diff")
histo_sig.SetMarkerColor(ROOT.kBlack)
histo_sig.SetLineColor(ROOT.kBlack)
histo_sig.GetXaxis().SetTitle(
'energy (keV)' if calib else 'I_{SC} (counts)')
histo_sig.GetXaxis().SetTitleSize(0.05)
histo_sig.GetYaxis().SetTitle('superclusters (bkg subtracted)')
c = getCanvas()
histo_sig.Draw("pe 1")
par = array('d', 6 * [0.])
xmin1, xmax1 = (3, 9) if calib else (1600, 3400)
xmin2, xmax2 = (11, 16) if calib else (4400, 6000)
g1 = ROOT.TF1("g1", "gaus", xmin1, xmax1)
g2 = ROOT.TF1("g2", "gaus", xmin2, xmax2)
total = ROOT.TF1('total', 'gaus(0)+gaus(3)', xmin1, xmax2)
total.SetLineColor(ROOT.kBlue + 1)
histo_sig.Fit('g1', 'RS')
histo_sig.Fit('g2', 'R+S')
mean = g1.GetParameter(1)
mErr = g1.GetParError(1)
sigma = g1.GetParameter(2)
mSigma = g1.GetParError(2)
lat = ROOT.TLatex()
unit = 'keV' if calib else 'counts'
ndigits = 2 if calib else 0
lat.SetNDC()
lat.SetTextFont(42)
lat.SetTextSize(0.03)
lat.DrawLatex(
0.55, 0.70,
"m_{{1}} = {m:.{nd}f} #pm {em:.{nd}f} {unit}".format(m=mean,
em=mErr,
unit=unit,
nd=ndigits))
lat.DrawLatex(
0.55, 0.65,
"#sigma_{{1}} = {s:.{nd}f} #pm {es:.{nd}f} {unit}".format(s=sigma,
es=mSigma,
unit=unit,
nd=ndigits))
mean = g2.GetParameter(1)
mErr = g2.GetParError(1)
sigma = g2.GetParameter(2)
mSigma = g2.GetParError(2)
lat.DrawLatex(
0.55, 0.50,
"m_{{2}} = {m:.{nd}f} #pm {em:.{nd}f} {unit}".format(m=mean,
em=mErr,
unit=unit,
nd=ndigits))
lat.DrawLatex(
0.55, 0.45,
"#sigma_{{2}} = {s:.{nd}f} #pm {es:.{nd}f} {unit}".format(s=sigma,
es=mSigma,
unit=unit,
nd=ndigits))
par1 = g1.GetParameters()
par2 = g2.GetParameters()
par[0], par[1], par[2] = par1[0], par1[1], par1[2]
par[3], par[4], par[5] = par2[0], par2[1], par2[2]
total.SetParameters(par)
histo_sig.Fit(total, 'R+')
c.SaveAs('fe_diff_simplefit.pdf')
def makeEff(f1,
histo1,
f2,
histo2,
f3=None,
histo3=None,
plotdir="./",
xmax=31):
# numerator: selected events
tf1 = ROOT.TFile.Open(f1)
hpass = tf1.Get(histo1)
hpass.GetYaxis().SetTitle('efficiency')
hpass.GetXaxis().SetRangeUser(0, xmax)
# used to calculate the average efficiency up to a given energy
averageXmax = 20 # keV
upperBin = hpass.GetXaxis().FindFixBin(averageXmax)
passInt = hpass.Integral(0, upperBin + 1)
# denominator: all events
tf2 = ROOT.TFile.Open(f2)
htotal = tf2.Get(histo2)
htotal.GetYaxis().SetTitle('signal efficiency ( #varepsilon_{S}^{total} )')
htotal.GetXaxis().SetRangeUser(0, xmax)
totalInt = htotal.Integral(1, upperBin + 1)
eff = passInt / totalInt
errEff = math.sqrt(eff * (1 - eff) / totalInt)
print("Average efficiency from 0 to ", averageXmax, " keV = ", eff,
" +/- ", errEff)
## default is 68% CL
col = ROOT.kRed + 1
teffi = ROOT.TEfficiency(hpass, htotal)
teffi.SetStatisticOption(ROOT.TEfficiency.kFCP)
teffi.SetFillColorAlpha(col, 0.1)
teffi.SetMarkerStyle(ROOT.kFullSquare)
teffi.SetMarkerSize(2)
teffi.SetLineWidth(3)
teffi.SetMarkerColor(col)
teffi.SetLineColor(col)
if histo3 and f3:
# numerator: selected events for the other WP
tf3 = ROOT.TFile.Open(f3)
hpass2 = tf3.Get(histo3)
hpass2.GetYaxis().SetTitle('efficiency')
hpass2.GetXaxis().SetRangeUser(0, xmax)
## default is 68% CL
col2 = ROOT.kAzure - 6
teffi2 = ROOT.TEfficiency(hpass2, htotal)
teffi2.SetStatisticOption(ROOT.TEfficiency.kFCP)
teffi2.SetFillColorAlpha(col2, 0.1)
teffi2.SetMarkerStyle(ROOT.kFullCircle)
teffi2.SetMarkerSize(2)
teffi2.SetLineWidth(1)
teffi2.SetMarkerColor(col2)
teffi2.SetLineColor(col2)
c = getCanvas()
if histo1.startswith('fe'):
c.SetLogy()
teffi.Draw("A3")
teffi.Draw("same pe1")
teffi2.Draw("same pe1")
teffi2.Draw("same3")
# to change the x-axis range
ROOT.gPad.Update()
teffi.GetPaintedGraph().GetXaxis().SetTitle("E (keV)")
teffi.GetPaintedGraph().GetXaxis().SetRangeUser(0, xmax)
teffi.GetPaintedGraph().GetXaxis().SetTitleOffset(1.3)
teffi.GetPaintedGraph().GetXaxis().SetTitleFont(42)
teffi.GetPaintedGraph().GetXaxis().SetTitleSize(0.05)
teffi.GetPaintedGraph().GetYaxis().SetTitle(
'signal efficiency ( #varepsilon_{S}^{total} )')
teffi.GetPaintedGraph().GetYaxis().SetRangeUser(0, 1)
teffi.GetPaintedGraph().GetYaxis().SetTitleFont(42)
teffi.GetPaintedGraph().GetYaxis().SetTitleSize(0.05)
ROOT.gPad.Update()
## add legend
histos = [teffi, teffi2]
labels = ['#varepsilon_{B}^{total}=4%', '#varepsilon_{B}^{total}=1%']
styles = ['F', 'F']
legend = doLegend(histos, labels, styles, corner='BR')
#legend.Draw('same')
for ext in ['png', 'pdf']:
c.SaveAs('{odir}/{var}_effi.{ext}'.format(odir=plotdir,
var=histo1,
ext=ext))