def pid_background_fraction():
"""simplistic calculation of background contamination fraction"""
nsig_p = hadd_interactive(histDir, runlist, 'jetpatch', 'anyspin', 'plus',
'away2_nSigmaPion')
nsig_m = hadd_interactive(histDir, runlist, 'jetpatch', 'anyspin', 'minus',
'away2_nSigmaPion')
nsig_p.SetTitle('n#sigma(#pi) for #pi^{+}')
nsig_m.SetTitle('n#sigma(#pi) for #pi^{-}')
print 'Fit summary for π+'
pfits = pid_calibration(nsig_p)
print 'Fit summary for π-'
mfits = pid_calibration(nsig_m)
c = graphics.canvas2()
c.cd(1)
nsig_m.Draw()
[fit.Draw('same') for fit in mfits[1:]]
c.cd(2)
nsig_p.Draw()
[fit.Draw('same') for fit in pfits[1:]]
raw_input('wait here:')
c.Print('pid_background_fraction.png')
def dphi():
data = hadd_interactive(histDir, runlist, 'jetpatch', 'anyspin', 'sum', \
'dphi')
mgr = HistogramManager(ROOT.TFile(datamcFile), keys=['dphi'])
simu = mgr.anyspin['jetpatch'].tracks_sum['dphi']
c = graphics.canvas1()
simu.SetLineColor(ROOT.kRed)
simu.GetXaxis().SetTitle('#Delta#phi')
simu.GetYaxis().SetTitle('normalized yield')
simu.DrawNormalized('hist')
data.SetMarkerStyle(20)
data.DrawNormalized('same')
raw_input('wait here:')
def meanpt():
"""plots mean pT in each z bin for for pions and tracks"""
meanpt_p = hadd_interactive(histDir, runlist, 'jetpatch', 'anyspin',
'plus', 'meanpt')
meanpt_m = hadd_interactive(histDir, runlist, 'jetpatch', 'anyspin',
'minus', 'meanpt')
meanjetpt_p = hadd_interactive(histDir, runlist, 'jetpatch', 'anyspin',
'plus', 'meanjetpt')
meanjetpt_m = hadd_interactive(histDir, runlist, 'jetpatch', 'anyspin',
'minus', 'meanjetpt')
## now get some results from simulations
f = ROOT.TFile(datamcFile)
mgr = HistogramManager(f, keys=['meanpt', 'meanjetpt'])
meansim_p = mgr.anyspin['jetpatch'].tracks_plus['meanpt']
meansim_m = mgr.anyspin['jetpatch'].tracks_minus['meanpt']
meanjetsim_p = mgr.anyspin['jetpatch'].tracks_plus['meanjetpt']
meanjetsim_m = mgr.anyspin['jetpatch'].tracks_minus['meanjetpt']
mbsim_p = mgr.anyspin['117001'].tracks_plus['meanpt']
mbsim_m = mgr.anyspin['117001'].tracks_minus['meanpt']
mbjetsim_p = mgr.anyspin['117001'].tracks_plus['meanjetpt']
mbjetsim_m = mgr.anyspin['117001'].tracks_minus['meanjetpt']
meanjetpt_m.SetTitle('JP Data / Monte Carlo comparison, #pi -')
meanjetpt_p.SetTitle('JP Data / Monte Carlo comparison, #pi +')
meanjetsim_m.SetTitle('Monte Carlo #pi -')
meanjetsim_p.SetTitle('Monte Carlo #pi +')
for h in (meanjetpt_m, meanjetpt_p, meanjetsim_m, meanjetsim_p):
h.SetMarkerStyle(21)
h.SetXTitle('z')
h.SetYTitle('< pT >')
[h.SetMarkerStyle(20) for h in (mbjetsim_m, mbjetsim_p)]
[h.SetMarkerStyle(24) for h in (mbsim_m, mbsim_p)]
for h in (meanpt_m, meanpt_p, meansim_m, meansim_p):
h.SetMarkerStyle(25)
for h in (meansim_m, meansim_p, meanjetsim_m, meanjetsim_p):
h.SetMarkerColor(ROOT.kRed)
h.SetLineColor(ROOT.kRed)
leg = ROOT.TLegend(.6, .7, .8, .88)
leg.AddEntry(meanjetpt_p, 'jet')
leg.AddEntry(meanpt_p, '#pi')
## first compare JP data/MC
c = graphics.canvas2('Data / Monte Carlo comparison for JP')
c.cd(1)
meanjetpt_m.Draw()
meanjetpt_m.GetXaxis().SetRangeUser(zbins[0], zbins[-1])
meanjetpt_m.GetYaxis().SetRangeUser(0, 25)
meanjetsim_m.Draw('][ hist same')
meanpt_m.Draw('same')
meansim_m.Draw('][ hist same')
c.cd(2)
meanjetpt_p.Draw()
meanjetpt_p.GetXaxis().SetRangeUser(zbins[0], zbins[-1])
meanjetpt_p.GetYaxis().SetRangeUser(0, 25)
meanjetsim_p.Draw('][ hist same')
meanpt_p.Draw('same')
meansim_p.Draw('][ hist same')
leg.Draw()
raw_input('wait here:')
c.Print('jp-means.png')
## now compare JP and MB simulations
c2 = graphics.canvas2('Comparison of MB and JP simulations')
for h in (meanjetsim_m, meanjetsim_p, meansim_m, meansim_p):
h.SetMarkerColor(ROOT.kBlue)
h.SetLineColor(ROOT.kBlue)
for h in (mbjetsim_m, mbjetsim_p, mbsim_m, mbsim_p):
h.SetMarkerColor(ROOT.kRed)
h.SetLineColor(ROOT.kRed)
c2.cd(1)
meanjetsim_m.Draw('hist p')
meanjetsim_m.GetXaxis().SetRangeUser(zbins[0], zbins[-1])
meanjetsim_m.GetYaxis().SetRangeUser(0, 25)
mbjetsim_m.Draw('hist p same')
mbsim_m.Draw('hist p same')
meansim_m.Draw('hist p same')
c2.cd(2)
meanjetsim_p.Draw('hist p')
meanjetsim_p.GetXaxis().SetRangeUser(zbins[0], zbins[-1])
meanjetsim_p.GetYaxis().SetRangeUser(0, 25)
mbjetsim_p.Draw('hist p same')
mbsim_p.Draw('hist p same')
meansim_p.Draw('hist p same')
leg2 = ROOT.TLegend(.68, .7, .88, .88)
leg2.AddEntry(meanjetsim_p, 'JP')
leg2.AddEntry(mbjetsim_p, 'MB')
leg2.Draw()
raw_input('wait here:')
c.Print('jp-means.png')
c2.Print('simu-means.png')
