def minbias_pt_reweight():
"""plots ratio of JP and MB jet spectrum and fits it"""
f = ROOT.TFile(datamcFile)
mgr = HistogramManager(f, keys=['true_jet_pt'])
mb = mgr.anyspin['117001']['true_jet_pt']
jp = mgr.anyspin['jetpatch']['true_jet_pt']
jp.Divide(mb)
jp.SetStats(True)
jp.GetXaxis().SetRangeUser(5,30)
jp.GetYaxis().SetRangeUser(0.,1.)
jp.SetXTitle('corrected jet pT')
jp.SetYTitle('nJets (JP) / nJets (MB)')
fit = ROOT.TF1('fit', 'pol1', 9.8, 25.3)
# sigmoid = ROOT.TF1('sigmoid', '[0]*log([1]*x+[2])', 9.8 , 25.3)
# sigmoid.SetParameter(0, 1.)
# sigmoid.SetParameter(1, 0.08)
# sigmoid.SetParameter(2, 0.)
c1 = graphics.canvas1()
jp.Fit(fit, 'r')
ROOT.gStyle.SetOptFit(111)
raw_input('wait here:')
c1.Print('minbias_pt_reweight.png')
jp.GetYaxis().SetRangeUser(1.E-5,1.)
ROOT.gPad.SetLogy()
raw_input('wait here:')
c1.Print('minbias_pt_reweight_log.png')
def qg_fragmentation_comparison():
simuFile = '/Users/kocolosk/work/2009-03-19-pi-fragmentation/merged.root'
import operator
from array import array
from analysis.plots import graphics
from analysis.histos2 import HistogramManager
bin_centers = array('d', [2.59, 3.87, 5.44, 7.56, 10.82])
dss_g = [2.5988E+06, 1.3668E+05, 1.0331E+04, 7.4421E+02, 3.4858E+01]
dss_q = [7.9033E+05, 6.0820E+04, 6.3704E+03, 6.3982E+02, 4.3774E+01]
kretzer_g = [1.6305E+06, 7.3650E+04, 4.8672E+03, 3.0306E+02, 1.1753E+01]
kretzer_q = [1.4892E+06, 1.1274E+05, 1.1583E+04, 1.1391E+03, 7.5693E+01]
dss_graph = ROOT.TGraph(
5, bin_centers,
array('d', map(operator.div, dss_g, map(operator.add, dss_q, dss_g))))
kretzer_graph = ROOT.TGraph(
5, bin_centers,
array(
'd',
map(operator.div, kretzer_g, map(operator.add, kretzer_q,
kretzer_g))))
dss_graph.SetLineStyle(2)
[graph.SetLineWidth(2) for graph in (dss_graph, kretzer_graph)]
mgr = HistogramManager(ROOT.TFile(simuFile))
gluon = mgr.qg['96011']['plus']['pt_gluon_jet']
quark = mgr.qg['96011']['plus']['pt_quark_jet']
ratio = gluon.Clone()
both = gluon.Clone()
both.Add(quark)
ratio.Divide(both)
ratio.GetXaxis().SetRangeUser(2.00, 12.84)
ratio.GetYaxis().SetRangeUser(0.0, 1.0)
ratio.SetMarkerStyle(24)
ratio.SetTitle('Fraction of qg #pi+ from gluon jet')
ratio.SetXTitle('#pi p_{T}')
leg = ROOT.TLegend(0.7, 0.7, 0.88, 0.88)
leg.AddEntry(dss_graph, 'DSS', 'l')
leg.AddEntry(kretzer_graph, 'Kretzer', 'l')
leg.AddEntry(ratio, 'Pythia')
c = graphics.canvas1()
ratio.Draw()
dss_graph.Draw()
kretzer_graph.Draw()
leg.Draw()
graphics.maybe_save()
def qg_fragmentation_comparison():
simuFile = "/Users/kocolosk/work/2009-03-19-pi-fragmentation/merged.root"
import operator
from array import array
from analysis.plots import graphics
from analysis.histos2 import HistogramManager
bin_centers = array("d", [2.59, 3.87, 5.44, 7.56, 10.82])
dss_g = [2.5988e06, 1.3668e05, 1.0331e04, 7.4421e02, 3.4858e01]
dss_q = [7.9033e05, 6.0820e04, 6.3704e03, 6.3982e02, 4.3774e01]
kretzer_g = [1.6305e06, 7.3650e04, 4.8672e03, 3.0306e02, 1.1753e01]
kretzer_q = [1.4892e06, 1.1274e05, 1.1583e04, 1.1391e03, 7.5693e01]
dss_graph = ROOT.TGraph(5, bin_centers, array("d", map(operator.div, dss_g, map(operator.add, dss_q, dss_g))))
kretzer_graph = ROOT.TGraph(
5, bin_centers, array("d", map(operator.div, kretzer_g, map(operator.add, kretzer_q, kretzer_g)))
)
dss_graph.SetLineStyle(2)
[graph.SetLineWidth(2) for graph in (dss_graph, kretzer_graph)]
mgr = HistogramManager(ROOT.TFile(simuFile))
gluon = mgr.qg["96011"]["plus"]["pt_gluon_jet"]
quark = mgr.qg["96011"]["plus"]["pt_quark_jet"]
ratio = gluon.Clone()
both = gluon.Clone()
both.Add(quark)
ratio.Divide(both)
ratio.GetXaxis().SetRangeUser(2.00, 12.84)
ratio.GetYaxis().SetRangeUser(0.0, 1.0)
ratio.SetMarkerStyle(24)
ratio.SetTitle("Fraction of qg #pi+ from gluon jet")
ratio.SetXTitle("#pi p_{T}")
leg = ROOT.TLegend(0.7, 0.7, 0.88, 0.88)
leg.AddEntry(dss_graph, "DSS", "l")
leg.AddEntry(kretzer_graph, "Kretzer", "l")
leg.AddEntry(ratio, "Pythia")
c = graphics.canvas1()
ratio.Draw()
dss_graph.Draw()
kretzer_graph.Draw()
leg.Draw()
graphics.maybe_save()
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 pt_shift_uncertainty():
## this is my guess based on inverting Dave's fit
measured_pt = [10.31, 12.91, 15.58, 19.06, 23.20, 28.45]
## http://cyclotron.tamu.edu/star/2005n06Jets/PRDweb/
## "numbers are from the preliminary analysis"
total_uncertainty = [ 0.72, 0.90, 0.99, 1.09, 1.27, 1.52 ]
c = graphics.canvas1()
low = ROOT.TGraph(len(measured_pt))
for i,(pt,error) in enumerate(zip(measured_pt, total_uncertainty)):
low.SetPoint(i, pt, shifted(pt)-error)
lowfit = ROOT.TF1('lowfit', 'pol2', 10, 30)
lowfit.SetLineStyle(2)
low.Fit(lowfit)
low.GetXaxis().SetTitle('measured jet p_{T}')
low.GetXaxis().SetRangeUser(10,30)
low.GetYaxis().SetTitle('corrected jet p_{T}')
low.GetYaxis().SetRangeUser(9,26)
low.SetTitle('Uncertainty on p_{T} shift')
low.SetLineStyle(2)
low.Draw('ap2')
high = ROOT.TGraph(len(measured_pt))
for i,(pt,error) in enumerate(zip(measured_pt, total_uncertainty)):
high.SetPoint(i, pt, shifted(pt)+error)
highfit = ROOT.TF1('lowfit', 'pol2', measured_pt[0], measured_pt[-1])
highfit.SetLineStyle(2)
high.Fit(highfit)
high.SetLineStyle(2)
high.Draw('same')
mid = ROOT.TF1('mid', lambda x: shifted(x[0]), 10.31, 28.45)
mid.Draw('same')
raw_input('wait here:')
def subprocess_fraction_z(charge = 1):
import histos
histos.simu = True
ROOT.gStyle.SetErrorX()
f = ROOT.TFile(datamcFile)
mgr = HistogramManager(f, ['z_away2'])
mb = {
'all': mgr.anyspin['117001'].trackHistograms(charge)['z_away2'],
'gg': mgr.gg['117001'].trackHistograms(charge)['z_away2'],
'qg': mgr.qg['117001'].trackHistograms(charge)['z_away2'],
'qq': mgr.qq['117001'].trackHistograms(charge)['z_away2']
}
jp = {
'all': mgr.anyspin['jetpatch'].trackHistograms(charge)['z_away2'],
'gg': mgr.gg['jetpatch'].trackHistograms(charge)['z_away2'],
'qg': mgr.qg['jetpatch'].trackHistograms(charge)['z_away2'],
'qq': mgr.qq['jetpatch'].trackHistograms(charge)['z_away2']
}
for key in ('gg','qg','qq'):
mb[key].Divide(mb['all'])
jp[key].Divide(jp['all'])
for h in (mb,jp):
h['gg'].SetLineColor(ROOT.kRed)
h['qg'].SetLineColor(ROOT.kBlue)
h['qq'].SetLineColor(ROOT.kGreen)
[h[k].SetMarkerStyle(24) for k in ('gg','qg','qq')]
leg = ROOT.TLegend(0.65, 0.7, 0.85, 0.88)
leg.AddEntry(mb['gg'], 'gg')
leg.AddEntry(mb['qg'], 'qg')
leg.AddEntry(mb['qq'], 'qq')
c = graphics.canvas2()
c.cd(1)
mb['gg'].Draw('e1')
mb['gg'].SetTitle('MB for charge=%d' % charge)
mb['gg'].GetYaxis().SetRangeUser(0., 0.8)
mb['gg'].GetXaxis().SetRangeUser(zbins[0], zbins[-1])
mb['qg'].Draw('e1 same')
mb['qq'].Draw('e1 same')
leg.Draw()
c.cd(2)
jp['gg'].Draw('e1')
jp['gg'].SetTitle('JP for charge=%d' % charge)
jp['gg'].GetYaxis().SetRangeUser(0., 0.8)
jp['gg'].GetXaxis().SetRangeUser(zbins[0], zbins[-1])
jp['qg'].Draw('e1 same')
jp['qq'].Draw('e1 same')
c2 = graphics.canvas1()
delta = {
'gg': jp['gg'].Clone(),
'qg': jp['qg'].Clone(),
'qq': jp['qq'].Clone()
}
for k in delta.keys():
delta[k].Add(mb[k], -1)
delta[k].SetMarkerStyle(24)
delta['gg'].SetTitle('Subprocess fraction difference (JP - MB) : charge=%d'\
% charge)
delta['gg'].GetYaxis().SetRangeUser(-0.4, 0.4)
delta['gg'].Draw('e1')
delta['qg'].Draw('e1 same')
delta['qq'].Draw('e1 same')
leg2 = ROOT.TLegend(0.15, 0.7, 0.25, 0.88)
leg2.AddEntry(mb['gg'], 'gg')
leg2.AddEntry(mb['qg'], 'qg')
leg2.AddEntry(mb['qq'], 'qq')
leg2.Draw()
line = ROOT.TLine(zbins[0], 0.0, zbins[-1], 0.0)
line.SetLineStyle(2)
line.Draw()
raw_input('wait here:')
if charge == 1:
c.Print('subprocess_fraction_trigger_plus.png')
c2.Print('subprocess_fraction_difference_plus.png')
else:
c.Print('subprocess_fraction_trigger_minus.png')
c2.Print('subprocess_fraction_difference_minus.png')
def nlo_pion_fractions(ff='Kretzer'):
simuFile = '/Users/kocolosk/data/run5-simu/hist/merged.cphist.root'
import operator
from array import array
from analysis.plots import graphics
from analysis.histos2 import HistogramManager
bin_centers = array('d', [2.59, 3.87, 5.44, 7.56, 10.82])
if ff == 'DSS':
gg_gg = [4.4390E+06, 1.9967E+05, 1.2634E+04, 7.4998E+02, 2.8100E+01]
qg_qg = [3.3888E+06, 1.9764E+05, 1.6707E+04, 1.3828E+03, 7.8631E+01]
qq_qq = [2.5272E+05, 2.2866E+04, 2.9083E+03, 3.6410E+02, 3.2692E+01]
qb_qb = [1.4414E+05, 1.1077E+04, 1.1646E+03, 1.1523E+02, 7.4628E+00]
total = [8.3013E+06, 4.3322E+05, 3.3477E+04, 2.6092E+03, 1.4636E+02]
elif ff == 'Kretzer':
gg_gg = [2.0141E+06, 8.0377E+04, 4.5900E+03, 2.4401E+02, 7.9185E+00]
qg_qg = [3.0735E+06, 1.8293E+05, 1.6132E+04, 1.4147E+03, 8.5984E+01]
qq_qq = [4.8958E+05, 4.3564E+04, 5.4494E+03, 6.7049E+02, 5.8678E+01]
qb_qb = [1.3586E+05, 1.0378E+04, 1.0776E+03, 1.0510E+02, 6.6962E+00]
total = [5.9360E+06, 3.3000E+05, 2.8449E+04, 2.5438E+03, 1.6599E+02]
pythia_id = {11: qq_qq, 12: qb_qb, 28: qg_qg, 68: gg_gg}
graph = {
11:
ROOT.TGraph(5, bin_centers,
array('d', map(operator.div, pythia_id[11], total))),
12:
ROOT.TGraph(5, bin_centers,
array('d', map(operator.div, pythia_id[12], total))),
28:
ROOT.TGraph(5, bin_centers,
array('d', map(operator.div, pythia_id[28], total))),
68:
ROOT.TGraph(5, bin_centers,
array('d', map(operator.div, pythia_id[68], total)))
}
mgr = HistogramManager(ROOT.TFile(simuFile))
hist = {
11: mgr.qq['96011']['plus']['pt'],
28: mgr.qg['96011']['plus']['pt'],
68: mgr.gg['96011']['plus']['pt']
}
htotal = mgr.anyspin['96011']['plus']['pt']
[h.Divide(htotal) for h in hist.values()]
graph[68].SetLineColor(ROOT.kRed)
graph[28].SetLineColor(ROOT.kGreen)
graph[11].SetLineColor(ROOT.kBlue)
graph[12].SetLineColor(ROOT.kMagenta)
[g.SetLineWidth(2) for g in graph.values()]
hist[68].SetMarkerColor(ROOT.kRed)
hist[28].SetMarkerColor(ROOT.kGreen)
hist[11].SetMarkerColor(ROOT.kBlue)
[h.SetMarkerStyle(24) for h in hist.values()]
bg = ROOT.TH2D('bg', '', 1, 2.00, 12.84, 1, 0., 0.7)
bg.SetTitle('Subprocess Fractions (solid=NLO+%s, points=Pythia)' % ff)
bg.SetXTitle('#pi p_{T}')
leg = ROOT.TLegend(0.8, 0.2, 0.88, 0.8)
leg.AddEntry(graph[68], 'gg', 'l')
leg.AddEntry(graph[28], 'qg', 'l')
leg.AddEntry(graph[11], 'qq', 'l')
leg.AddEntry(graph[12], 'q#bar{q}', 'l')
c = graphics.canvas1()
bg.Draw()
[g.Draw() for g in graph.values()]
[h.Draw('same') for h in hist.values()]
leg.Draw()
graphics.maybe_save()
def nlo_pion_fractions(ff="Kretzer"):
simuFile = "/Users/kocolosk/data/run5-simu/hist/merged.cphist.root"
import operator
from array import array
from analysis.plots import graphics
from analysis.histos2 import HistogramManager
bin_centers = array("d", [2.59, 3.87, 5.44, 7.56, 10.82])
if ff == "DSS":
gg_gg = [4.4390e06, 1.9967e05, 1.2634e04, 7.4998e02, 2.8100e01]
qg_qg = [3.3888e06, 1.9764e05, 1.6707e04, 1.3828e03, 7.8631e01]
qq_qq = [2.5272e05, 2.2866e04, 2.9083e03, 3.6410e02, 3.2692e01]
qb_qb = [1.4414e05, 1.1077e04, 1.1646e03, 1.1523e02, 7.4628e00]
total = [8.3013e06, 4.3322e05, 3.3477e04, 2.6092e03, 1.4636e02]
elif ff == "Kretzer":
gg_gg = [2.0141e06, 8.0377e04, 4.5900e03, 2.4401e02, 7.9185e00]
qg_qg = [3.0735e06, 1.8293e05, 1.6132e04, 1.4147e03, 8.5984e01]
qq_qq = [4.8958e05, 4.3564e04, 5.4494e03, 6.7049e02, 5.8678e01]
qb_qb = [1.3586e05, 1.0378e04, 1.0776e03, 1.0510e02, 6.6962e00]
total = [5.9360e06, 3.3000e05, 2.8449e04, 2.5438e03, 1.6599e02]
pythia_id = {11: qq_qq, 12: qb_qb, 28: qg_qg, 68: gg_gg}
graph = {
11: ROOT.TGraph(5, bin_centers, array("d", map(operator.div, pythia_id[11], total))),
12: ROOT.TGraph(5, bin_centers, array("d", map(operator.div, pythia_id[12], total))),
28: ROOT.TGraph(5, bin_centers, array("d", map(operator.div, pythia_id[28], total))),
68: ROOT.TGraph(5, bin_centers, array("d", map(operator.div, pythia_id[68], total))),
}
mgr = HistogramManager(ROOT.TFile(simuFile))
hist = {11: mgr.qq["96011"]["plus"]["pt"], 28: mgr.qg["96011"]["plus"]["pt"], 68: mgr.gg["96011"]["plus"]["pt"]}
htotal = mgr.anyspin["96011"]["plus"]["pt"]
[h.Divide(htotal) for h in hist.values()]
graph[68].SetLineColor(ROOT.kRed)
graph[28].SetLineColor(ROOT.kGreen)
graph[11].SetLineColor(ROOT.kBlue)
graph[12].SetLineColor(ROOT.kMagenta)
[g.SetLineWidth(2) for g in graph.values()]
hist[68].SetMarkerColor(ROOT.kRed)
hist[28].SetMarkerColor(ROOT.kGreen)
hist[11].SetMarkerColor(ROOT.kBlue)
[h.SetMarkerStyle(24) for h in hist.values()]
bg = ROOT.TH2D("bg", "", 1, 2.00, 12.84, 1, 0.0, 0.7)
bg.SetTitle("Subprocess Fractions (solid=NLO+%s, points=Pythia)" % ff)
bg.SetXTitle("#pi p_{T}")
leg = ROOT.TLegend(0.8, 0.2, 0.88, 0.8)
leg.AddEntry(graph[68], "gg", "l")
leg.AddEntry(graph[28], "qg", "l")
leg.AddEntry(graph[11], "qq", "l")
leg.AddEntry(graph[12], "q#bar{q}", "l")
c = graphics.canvas1()
bg.Draw()
[g.Draw() for g in graph.values()]
[h.Draw("same") for h in hist.values()]
leg.Draw()
graphics.maybe_save()
