def DrawScatYLine(graph, title, fname, xcoord): #, grid):
c = TCanvas("c", "c", 800, 600)
graph.SetTitle(title)
graph.GetXaxis().SetTitleSize(.04)
graph.GetYaxis().SetTitleSize(.04)
graph.GetXaxis().SetTitleOffset(1.2)
graph.GetYaxis().SetTitleOffset(1.25)
graph.GetXaxis().CenterTitle(1)
graph.GetYaxis().CenterTitle(1)
graph.GetYaxis().SetMaxDigits(4)
graph.SetMarkerStyle(20) # Full circle
graph.Draw()
gPad.Update()
# line = TLine(gPad.GetUxmin(),ycoord,gPad.GetUxmax(),ycoord)
line = TLine(xcoord, gPad.GetUymin(), xcoord, gPad.GetUymax())
# print(gPad.GetUymin(),gPad.GetUymax())
line.SetLineStyle(2)
line.SetLineColor(2)
line.SetLineWidth(3)
graph.Draw("AP")
line.Draw("same")
c.SaveAs(fname + ".C")
c.SaveAs(fname + ".pdf")
return
def __call__(self):
h = gPad.GetSelected()
if not h:
return
if not isinstance(h, TH2):
return
gPad.GetCanvas().FeedbackMode(kTRUE)
# erase old position and draw a line at current position
px = gPad.GetEventX()
py = gPad.GetEventY()
uxmin, uxmax = gPad.GetUxmin(), gPad.GetUxmax()
uymin, uymax = gPad.GetUymin(), gPad.GetUymax()
pxmin, pxmax = gPad.XtoAbsPixel(uxmin), gPad.XtoAbsPixel(uxmax)
pymin, pymax = gPad.YtoAbsPixel(uymin), gPad.YtoAbsPixel(uymax)
if self._old != None:
gVirtualX.DrawLine(pxmin, self._old[1], pxmax, self._old[1])
gVirtualX.DrawLine(self._old[0], pymin, self._old[0], pymax)
gVirtualX.DrawLine(pxmin, py, pxmax, py)
gVirtualX.DrawLine(px, pymin, px, pymax)
self._old = px, py
upx = gPad.AbsPixeltoX(px)
x = gPad.PadtoX(upx)
upy = gPad.AbsPixeltoY(py)
y = gPad.PadtoY(upy)
padsav = gPad
# create or set the display canvases
if not self._cX:
self._cX = TCanvas('c2', 'Projection Canvas in X', 730, 10, 700,
500)
else:
self._DestroyPrimitive('X')
if not self._cY:
self._cY = TCanvas('c3', 'Projection Canvas in Y', 10, 550, 700,
500)
else:
self._DestroyPrimitive('Y')
self.DrawSlice(h, y, 'Y')
self.DrawSlice(h, x, 'X')
padsav.cd()
def get_rank_section(directory):
# do Rank histo png
imgname = "RankSummary.png"
gStyle.SetPadTickY(0)
c = TCanvas("ranks", "ranks", 500, 400)
#gStyle.SetOptStat(0)
c.cd()
h = directory.rank_histo
rank_histof = TH1F(h.GetName(), "", h.GetNbinsX(),
h.GetXaxis().GetXmin(),
h.GetXaxis().GetXmax())
rank_histof.SetLineWidth(2)
for i in xrange(0, h.GetNbinsX() + 1):
rank_histof.SetBinContent(i, h.GetBinContent(i))
h.SetTitle("Ranks Summary;Rank;Frequency")
h.Draw("Hist")
c.Update()
rank_histof.ComputeIntegral()
integral = rank_histof.GetIntegral()
rank_histof.SetContent(integral)
rightmax = 1.1 * rank_histof.GetMaximum()
scale = gPad.GetUymax() / rightmax
rank_histof.SetLineColor(kRed)
rank_histof.Scale(scale)
rank_histof.Draw("same")
#draw an axis on the right side
axis = TGaxis(gPad.GetUxmax(), gPad.GetUymin(), gPad.GetUxmax(),
gPad.GetUymax(), 0, rightmax, 510, "+L")
axis.SetTitle("Cumulative")
axis.SetTitleColor(kRed)
axis.SetLineColor(kRed)
axis.SetLabelColor(kRed)
axis.Draw()
rank_histof.Draw("Same")
c.Print(imgname)
page_html = '<div class="span-20"><h2 class="alt"><a name="rank_summary">Ranks Summary</a></h2>'
page_html += '<div class="span-19"><img src="%s"></div>' % imgname
page_html += '</div> <a href="#top">Top...</a><hr>'
return page_html
corMatrix.SetBinContent(i + 1, j + 1,
fit_result.correlation(inuis, jnuis))
corMatrix.SetStats(0)
corMatrix.SetTitle("")
corMatrix.GetXaxis().SetLabelOffset(99)
corMatrix.GetYaxis().SetLabelOffset(99)
corMatrix.Draw("colz")
xlabels = TLatex()
xlabels.SetTextSize(0.018)
xlabels.SetTextAngle(90)
xlabels.SetTextAlign(32)
for i, nuis in enumerate(allpars):
y = gPad.GetUymin() - 0.5 * corMatrix.GetYaxis().GetBinWidth(i + 1)
x = corMatrix.GetXaxis().GetBinCenter(i + 1)
xlabels.DrawLatex(x, y, nuis)
ylabels = TLatex()
ylabels.SetTextSize(0.018)
ylabels.SetTextAngle(0)
ylabels.SetTextAlign(32)
for j, nuis in enumerate(allpars):
x = gPad.GetUxmin() - 0.5 * corMatrix.GetXaxis().GetBinWidth(j + 1)
y = corMatrix.GetXaxis().GetBinCenter(j + 1)
ylabels.DrawLatex(x, y, nuis)
filename = filename.replace('multidimfit', 'correlation_matrix')
filename = filename.replace('fitDiagnostics', 'correlation_matrix')
def __call__( self ):
h = self.hname
if not h:
return
if not isinstance( h, TH2 ):
return
gPad.GetCanvas().FeedbackMode( kTRUE )
event = gPad.GetEvent()
if (event==1): # left mouse click
self.projection = not self.projection
elif (event==12): # middle mouse click
self.logy = not self.logy
# erase old position and draw a line at current position
#(gPad coordinate system)
px = gPad.GetEventX()
py = gPad.GetEventY()
# min/max x values visible on the pad (xmin and xmax of the x-axis)
uxmin, uxmax = gPad.GetUxmin(), gPad.GetUxmax()
uymin, uymax = gPad.GetUymin(), gPad.GetUymax()
# same in pixels
pxmin, pxmax = gPad.XtoAbsPixel( uxmin ), gPad.XtoAbsPixel( uxmax )
pymin, pymax = gPad.YtoAbsPixel( uymin ), gPad.YtoAbsPixel( uymax )
# if self.projection:
# axis = h.GetYaxis()
# else:
# axis = h.GetXaxis()
scaleY = abs((pymax-pymin) / (uymax-uymin))
scaleX = abs((pxmax-pxmin) / (uxmax-uxmin))
width = 0.0 # arbitrary default value [cm]
if self._old:
width = self.range[1] - self.range[0]
ywidth = int(width * scaleY)
xwidth = int(width * scaleX)
if self._old != None:
if self.projection:
# gVirtualX.DrawBox( pxmin, self._old[1]-ywidth, pxmax, self._old[1], kSolid)
gVirtualX.DrawLine( pxmin, self._old[1], pxmax, self._old[1])
gVirtualX.DrawLine( pxmin, self._old[1]-ywidth, pxmax, self._old[1]-ywidth)
else:
# gVirtualX.DrawBox( self._old[0], pymin, self._old[0]+ywidth, pymax, kSolid )
gVirtualX.DrawLine( self._old[0], pymin, self._old[0], pymax)
gVirtualX.DrawLine( self._old[0]+xwidth, pymin, self._old[0]+xwidth, pymax)
# Normally these calls remove old lines, but we do not need them since we update the pad in the end of DrawSlice
# BUG: the lines disappear when the mouse does not move. This happens if this gPad.Update() is called. TODO: How to fix it?
# if self.projection:
# gVirtualX.DrawLine( pxmin, py, pxmax, py )
# gVirtualX.DrawLine( pxmin, py-ywidth, pxmax, py-ywidth )
# else:
# gVirtualX.DrawLine( px, pymin, px, pymax)
# gVirtualX.DrawLine( px+xwidth, pymin, px+xwidth, pymax)
if self.projection:
gPad.SetUniqueID(py)
else:
gPad.SetUniqueID(px)
self._old = px, py
upx = gPad.AbsPixeltoX( px )
x = gPad.PadtoX( upx )
upy = gPad.AbsPixeltoY( py )
y = gPad.PadtoY( upy )
padsav = gPad
# create or set the display canvases
if not self._cX:
self._cX = gPad.GetCanvas().GetPad(2)
else:
self._DestroyPrimitive( 'X' )
if not self._cY:
self._cY = gPad.GetCanvas().GetPad(2)
else:
self._DestroyPrimitive( 'Y' )
if self.projection:
self.range = self.DrawSlice( h, y, 'Y' )
else:
self.range = self.DrawSlice( h, x, 'X' )
padsav.cd()
def PlotNpv(inputTXT, lumiInp):
canvas = makeCMSCanvas(str(random.random()),"Npv",900,700)
canvas.Divide(1,2)
canvas.cd(2)
Npv = []
error = []
lumi = []
lumi_err = []
noerr = []
with open(inputTXT, "r") as Inp:
print("Opening " + inputTXT + ".")
for line in Inp:
cleared_line = line.split(" ")
Npv.append(float(cleared_line[0]))
error.append(float(cleared_line[1]))
sumLumi = 0
for i in range(0,len(lumiInp)):
sumLumi += float(lumiInp[i])
lumi.append(sumLumi - float(lumiInp[i])/2)
lumi_err.append(float(lumiInp[i])/2)
noerr.append(0.)
graph1 = TGraphErrors(len(lumiInp),array('d',lumi),array('d',Npv),array('d',lumi_err),array('d',error))
graph2 = TGraphErrors(len(lumiInp),array('d',lumi),array('d',Npv),array('d',lumi_err),array('d',noerr))
graph1.SetTitle("")
graph1.GetXaxis().SetTitle("Lumi [fb^{-1}]")
graph1.GetXaxis().SetLimits(0.,40.)
graph1.GetYaxis().SetTitle("Npv")
graph1.SetMarkerStyle(20)
graph1.SetMarkerSize(1)
graph1.Draw("AP")
canvas.cd(1)
graph2.SetTitle("")
graph2.GetXaxis().SetTitle("Lumi [fb^{-1}]")
graph2.GetXaxis().SetLimits(0.,35.)
graph2.GetYaxis().SetTitle("Npv")
graph2.SetMarkerStyle(20)
graph2.SetMarkerSize(1)
graph2.Draw("AP")
canvas.Update()
canvas.cd(2)
down = gPad.GetUymin()
up = gPad.GetUymax()
lineB = TLine(5.57,down,5.57, up)
lineB.SetLineColor(kBlack)
lineB.SetLineStyle(2)
lineB.Draw()
lineC = TLine(8.58,down,8.58,up)
lineC.SetLineColor(kBlack)
lineC.SetLineStyle(2)
lineC.Draw()
lineD = TLine(12.9,down,12.9,up)
lineD.SetLineColor(kBlack)
lineD.SetLineStyle(2)
lineD.Draw()
lineE = TLine(16.57,down,16.57,up)
lineE.SetLineColor(kBlack)
lineE.SetLineStyle(2)
lineE.Draw()
lineF = TLine(19.7,down,19.7,up)
lineF.SetLineColor(kBlack)
lineF.SetLineStyle(2)
lineF.Draw()
lineG = TLine(26.9,down,26.9,up)
lineG.SetLineColor(kBlack)
lineG.SetLineStyle(2)
lineG.Draw()
canvas.SaveAs("Npv.pdf")
return;
def TwoFileSAME2VsLumi(F1graph1, F1graph2, F2graph1, F2graph2, title, type):
canvas = makeCMSCanvas(str(random.random()),"canvas",900,700)
canvas.cd()
F1graph1.SetMarkerColor(kBlue)#electrons
F1graph2.SetMarkerColor(kRed)#muons
F2graph1.SetMarkerColor(kBlue)#electrons
F2graph2.SetMarkerColor(kRed)#muons
F2graph1.SetMarkerStyle(kOpenStar)
F2graph2.SetMarkerStyle(kOpenStar)
multigraph = TMultiGraph()
multigraph.Add(F1graph1,"AP")
multigraph.Add(F1graph2,"AP")
multigraph.Add(F2graph1,"AP")
multigraph.Add(F2graph2,"AP")
multigraph.Draw("AP")
multigraph.GetXaxis().SetLimits(0.,40.)
#TGaxis.SetMaxDigits(2)
#TGaxis.SetExponentOffset(-0.06, 0.02, "y")
multigraph.GetXaxis().SetTitle("L [fb^{-1}]")
multigraph.GetYaxis().SetTitleOffset(1.4)
if(type == "ISO"):
multigraph.GetYaxis().SetTitle("Isolation")
gPad.Modified()
multigraph.SetMinimum(0.5*gPad.GetUymin())
multigraph.SetMaximum(1.5*gPad.GetUymax())
elif(type == "SIP"):
multigraph.GetYaxis().SetTitle("SIP")
printLumiPrelOut(canvas)
canvas.Update()
down = gPad.GetUymin()
up = gPad.GetUymax()
lineB = TLine(5.57,down,5.57, up)
lineB.SetLineColor(kBlack)
lineB.SetLineStyle(2)
lineB.Draw()
lineC = TLine(8.58,down,8.58,up)
lineC.SetLineColor(kBlack)
lineC.SetLineStyle(2)
lineC.Draw()
lineD = TLine(12.9,down,12.9,up)
lineD.SetLineColor(kBlack)
lineD.SetLineStyle(2)
lineD.Draw()
lineE = TLine(16.57,down,16.57,up)
lineE.SetLineColor(kBlack)
lineE.SetLineStyle(2)
lineE.Draw()
lineF = TLine(19.7,down,19.7,up)
lineF.SetLineColor(kBlack)
lineF.SetLineStyle(2)
lineF.Draw()
lineG = TLine(26.9,down,26.9,up)
lineG.SetLineColor(kBlack)
lineG.SetLineStyle(2)
lineG.Draw()
legend = TLegend(0.80,0.75,0.965,0.93)
legend.AddEntry(F1graph1,"e^{+}e^{-}","P")
legend.AddEntry(F1graph2,"#mu^{+}#mu^{-}","P")
legend.AddEntry(F2graph1,"e^{+}e^{-} ICHEP","P")
legend.AddEntry(F2graph2,"#mu^{+}#mu^{-} ICHEP","P")
legend.SetTextFont(32)
legend.Draw()
canvas.SaveAs(title + ".pdf")
canvas.SaveAs(title + ".png")
return;
def TwoFileSAME3VsLumi(F1graph1, F1graph2, F1graph3, F2graph1, F2graph2, F2graph3, title, type, DoInclusive):
canvas = makeCMSCanvas(str(random.random()),"canvas",900,700)
canvas.cd()
F1graph1.SetMarkerColor(kBlack)#electrons
F1graph2.SetMarkerColor(kBlue)#electrons
F1graph3.SetMarkerColor(kRed)#muons
F2graph1.SetMarkerColor(kBlack)#electrons
F2graph2.SetMarkerColor(kBlue)#electrons
F2graph3.SetMarkerColor(kRed)#muons
F2graph1.SetMarkerStyle(kOpenStar)#inclusive
F2graph2.SetMarkerStyle(kOpenStar)#electrons
F2graph3.SetMarkerStyle(kOpenStar)#muons
multigraph = TMultiGraph()
if(DoInclusive or type == "ZmassBARELL" or type == "ZwidthBARELL" or type == "ZmultBARELL"):
multigraph.Add(F1graph1,"AP")
multigraph.Add(F2graph1,"AP")
multigraph.Add(F1graph2,"AP")
multigraph.Add(F1graph3,"AP")
multigraph.Add(F2graph2,"AP")
multigraph.Add(F2graph3,"AP")
multigraph.Draw("AP")
multigraph.GetXaxis().SetTitle("L [fb^{-1}]")
multigraph.GetXaxis().SetLimits(0.,40.)
multigraph.GetYaxis().SetTitleOffset(1.4)
gPad.Modified()
if(type == "Zmass" or type == "ZmassBARELL"):
multigraph.SetMaximum(1.008*gPad.GetUymax())
multigraph.GetYaxis().SetTitle("M_{l^{+}l^{-}} [GeV]")
elif(type == "Zwidth" or type == "ZwidthBARELL"):
multigraph.SetMaximum(1.1*gPad.GetUymax())
multigraph.GetYaxis().SetTitle("#Gamma [GeV]")
elif(type == "Zmult" or type == "ZmultBARELL"):
multigraph.SetMaximum(1.2*gPad.GetUymax())
multigraph.GetYaxis().SetTitle("#Z")
elif(type == "SIP"):
multigraph.GetYaxis().SetTitle("SIP")
multigraph.SetMaximum(1.1*gPad.GetUymax())
printLumiPrelOut(canvas)
canvas.Update()
down = gPad.GetUymin()
up = gPad.GetUymax()
MC_muMass = 90.92
MC_eleMass = 90.63
Data_muMass = 90.95
Data_eleMass = 90.68
lineB = TLine(5.57,down,5.57, up)
lineB.SetLineColor(kBlack)
lineB.SetLineStyle(2)
lineB.Draw()
lineC = TLine(8.58,down,8.58,up)
lineC.SetLineColor(kBlack)
lineC.SetLineStyle(2)
lineC.Draw()
lineD = TLine(12.9,down,12.9,up)
lineD.SetLineColor(kBlack)
lineD.SetLineStyle(2)
lineD.Draw()
lineE = TLine(16.57,down,16.57,up)
lineE.SetLineColor(kBlack)
lineE.SetLineStyle(2)
lineE.Draw()
lineF = TLine(19.7,down,19.7,up)
lineF.SetLineColor(kBlack)
lineF.SetLineStyle(2)
lineF.Draw()
lineG = TLine(26.9,down,26.9,up)
lineG.SetLineColor(kBlack)
lineG.SetLineStyle(2)
lineG.Draw()
if(type == "Zmass"):
lineMC_ele = TLine(0.,MC_eleMass,40., MC_eleMass)
lineMC_ele.SetLineColor(kBlue)
lineMC_ele.SetLineStyle(1)
lineMC_ele.Draw()
lineMC_mu = TLine(0.,MC_muMass,40., MC_muMass)
lineMC_mu.SetLineColor(kRed)
lineMC_mu.SetLineStyle(1)
lineMC_mu.Draw()
lineData_ele = TLine(0.,Data_eleMass,40., Data_eleMass)
lineData_ele.SetLineColor(kBlue)
lineData_ele.SetLineStyle(2)
lineData_ele.Draw()
lineData_mu = TLine(0.,Data_muMass,40., Data_muMass)
lineData_mu.SetLineColor(kRed)
lineData_mu.SetLineStyle(2)
lineData_mu.Draw()
legend = TLegend(0.80,0.75,0.965,0.93)
if(type == "Zmass" or type == "Zwidth" or type == "Zmult"):
legend.AddEntry(F1graph2,"e^{+}e^{-}","P")
legend.AddEntry(F1graph3,"#mu^{+}#mu^{-}","P")
legend.AddEntry(F2graph2,"e^{+}e^{-} ICHEP","P")
legend.AddEntry(F2graph3,"#mu^{+}#mu^{-} ICHEP","P")
else:
legend.AddEntry(F1graph1,"EBEB","P")
legend.AddEntry(F2graph1,"EBEB ICHEP","P")
legend.AddEntry(F1graph2,"EBEE","P")
legend.AddEntry(F1graph3,"EEEE","P")
legend.AddEntry(F2graph2,"EBEE ICHEP","P")
legend.AddEntry(F2graph3,"EEEE ICHEP","P")
legend.SetTextFont(32)
legend.Draw()
canvas.SaveAs(title + ".pdf")
canvas.SaveAs(title + ".png")
return;
def SAME2VsLumi(g1, g2,title, ptype, dataPeriod):
canvas = makeCMSCanvas(str(random.random()),"canvas",900,700)
canvas.cd()
graph1=copy.deepcopy(g1)
graph2=copy.deepcopy(g2)
graph1.SetMarkerColor(kBlue)#electrons
graph2.SetMarkerColor(kRed)#muons
multigraph = TMultiGraph()
multigraph.Add(graph1,"AP")
multigraph.Add(graph2,"AP")
multigraph.Draw("AP")
TGaxis.SetMaxDigits(2)
TGaxis.SetExponentOffset(-0.06, 0.02, "y")
multigraph.GetXaxis().SetTitle("L [fb^{-1}]")
multigraph.GetYaxis().SetTitleOffset(1.4)
if(ptype == "ISO"):
multigraph.GetYaxis().SetTitle("Isolation")
gPad.Modified()
multigraph.SetMinimum(0.5*gPad.GetUymin())
multigraph.SetMaximum(1.5*gPad.GetUymax())
elif(ptype == "SIP"):
multigraph.GetYaxis().SetTitle("SIP")
multigraph.SetMinimum(min(multigraph.GetHistogram().GetMinimum(),1.))
multigraph.SetMaximum(max(multigraph.GetHistogram().GetMinimum(),2.2))
min(multigraph.GetHistogram().GetMinimum(),1.)
printLumiPrelOut(canvas)
# Draw legend
legend = TLegend(0.93,0.84,0.99,0.93)
legend.AddEntry(graph1,"e^{+}e^{-}","P")
legend.AddEntry(graph2,"#mu^{+}#mu^{-}","P")
legend.SetFillColor(kWhite)
legend.SetLineColor(kBlack)
legend.SetTextFont(43)
legend.SetTextSize(20)
legend.Draw()
canvas.Update()
# Draw letters for data-taking periods
if(dataPeriod == "data2017"):
textLetters = TLatex()
textLetters.SetTextColor(kGray+1)
textLetters.SetTextSize(0.03)
if(ptype == "ISO"):
textLetters.DrawLatex(2., 0.8*gPad.GetUymax(),"B")
textLetters.DrawLatex(9.5, 0.8*gPad.GetUymax(),"C")
textLetters.DrawLatex(16., 0.8*gPad.GetUymax(),"D")
textLetters.DrawLatex(23., 0.8*gPad.GetUymax(),"E")
textLetters.DrawLatex(36., 0.8*gPad.GetUymax(),"F")
elif(ptype == "SIP"):
textLetters.DrawLatex(2., 1.5,"B")
textLetters.DrawLatex(9.5, 1.5,"C")
textLetters.DrawLatex(16., 1.5,"D")
textLetters.DrawLatex(23., 1.5,"E")
textLetters.DrawLatex(36., 1.5,"F")
if(dataPeriod == "data2018"):
textLetters = TLatex()
textLetters.SetTextColor(kGray+1)
textLetters.SetTextSize(0.03)
if(ptype == "ISO"):
textLetters.DrawLatex(6., 0.8*gPad.GetUymax(), "A")
textLetters.DrawLatex(16., 0.8*gPad.GetUymax(), "B")
textLetters.DrawLatex(23., 0.8*gPad.GetUymax(), "C")
textLetters.DrawLatex(43., 0.8*gPad.GetUymax(), "D")
elif(ptype == "SIP"):
textLetters.DrawLatex(6., 1.5, "A")
textLetters.DrawLatex(16., 1.5, "B")
textLetters.DrawLatex(23., 1.5, "C")
textLetters.DrawLatex(43., 1.5, "D")
# ****
if(dataPeriod == "data2018"):
# draw vertical lines that divide different data taking periods
down = gPad.GetUymin()
up = gPad.GetUymax()
lineA = TLine(13.48, down, 13.48, up) # Run2018A up to 13.48 fb-1
lineA.SetLineColor(kBlack)
lineA.SetLineStyle(2)
lineA.Draw()
lineB = TLine(20.265, down, 20.265, up) # Run2018B up to 20.265 fb-1
lineB.SetLineColor(kBlack)
lineB.SetLineStyle(2)
lineB.Draw()
lineC = TLine(26.877, down, 26.877, up) # Run2018C up to 26.877 fb-1
lineC.SetLineColor(kBlack)
lineC.SetLineStyle(2)
lineC.Draw()
# ****
if(dataPeriod == "data2017"):
# draw vertical lines that divide different data taking periods
down = gPad.GetUymin()
up = gPad.GetUymax()
lineB = TLine(4.793, down, 4.793, up) # Run2017B up to 4.793 fb-1
lineB.SetLineColor(kBlack)
lineB.SetLineStyle(2)
lineB.Draw()
lineC = TLine(14.549, down, 14.549, up) # Run2017C up to 14.549 fb-1
lineC.SetLineColor(kBlack)
lineC.SetLineStyle(2)
lineC.Draw()
lineD = TLine(18.868, down, 18.868, up) # Run2017D up to 18.868 fb-1
lineD.SetLineColor(kBlack)
lineD.SetLineStyle(2)
lineD.Draw()
lineE = TLine(28.293, down, 28.293, up) # Run2017E up to 28.293 fb-1
lineE.SetLineColor(kBlack)
lineE.SetLineStyle(2)
lineE.Draw()
# ****
if(dataPeriod == "data2016"):
# draw vertical lines that divide different data taking periods
down = gPad.GetUymin()
up = gPad.GetUymax()
lineB = TLine(5.789, down, 5.789, up) # Run2016B up to 5.789 fb-1
lineB.SetLineColor(kBlack)
lineB.SetLineStyle(2)
lineB.Draw()
lineC = TLine(8.366, down, 8.366, up) # Run2016C up to 8.366 fb-1
lineC.SetLineColor(kBlack)
lineC.SetLineStyle(2)
lineC.Draw()
lineD = TLine(12.616, down, 12.616, up) # Run2016D up to 12.616 fb-1
lineD.SetLineColor(kBlack)
lineD.SetLineStyle(2)
lineD.Draw()
lineE = TLine(16.624, down, 16.624, up) # Run2016E up to 16.624 fb-1
lineE.SetLineColor(kBlack)
lineE.SetLineStyle(2)
lineE.Draw()
lineF = TLine(19.725, down, 19.725, up) # Run2016F up to 19.725 fb-1
lineF.SetLineColor(kBlack)
lineF.SetLineStyle(2)
lineF.Draw()
lineG = TLine(27.268, down, 27.268, up) # Run2016G up to 27.268 fb-1
lineG.SetLineColor(kBlack)
lineG.SetLineStyle(2)
lineG.Draw()
# ****
canvas.SaveAs(title + ".root")
canvas.SaveAs(title + ".pdf")
canvas.SaveAs(title + ".png")
return;
def SAME3VsLumi(g1, g2, g3, title, ptype, lineMC1, lineDATA1, lineMC2, lineDATA2, lineMC3, lineDATA3, DoInclusive, dataPeriod):
canvas = makeCMSCanvas(str(random.random()),"canvas",900,700)
canvas.cd()
graph2=copy.deepcopy(g2)
graph3=copy.deepcopy(g3)
graph2.SetMarkerColor(kBlue)#electrons
graph3.SetMarkerColor(kRed)#muons
multigraph = TMultiGraph()
if(DoInclusive):
graph1=copy.deepcopy(g1)
multigraph.Add(graph1,"AP")
multigraph.Add(graph2,"AP")
multigraph.Add(graph3,"AP")
multigraph.Draw("AP")
TGaxis.SetMaxDigits(2)
TGaxis.SetExponentOffset(-0.05, 0.02, "y")
multigraph.GetXaxis().SetTitle("L [fb^{-1}]")
multigraph.GetYaxis().SetTitleOffset(1.4)
if(ptype == "Zmass"):
multigraph.GetYaxis().SetTitle("M_{Z} [GeV]")
# multigraph.GetYaxis().SetTitle("M_{l^{+}l^{-}} [GeV]")
multigraph.SetMaximum(max(multigraph.GetHistogram().GetMaximum(),91.4))
multigraph.SetMinimum(min(multigraph.GetHistogram().GetMinimum(),89.6))
elif(ptype == "Zwidth"):
multigraph.GetYaxis().SetTitle("#Gamma_{Z} [GeV]")
elif(ptype == "Zmult"):
multigraph.GetYaxis().SetTitle("#Z / fb^{-1}")
if(not DoInclusive) :
multigraph.SetMaximum(max(multigraph.GetHistogram().GetMaximum(),60000.)) # set y axis minimum at 60000.
multigraph.SetMinimum(0.) # set y axis minimum at 0.
# multigraph.SetMaximum(60000.) #second type: vs 2016 plots
# multigraph.SetMinimum(25000.)
printLumiPrelOut(canvas)
# Draw legend
legend = TLegend(0.93,0.84,0.99,0.93)
if(DoInclusive):
#legend.AddEntry(graph1,"inclusive","P")
legend.AddEntry(graph1,"BB","P")
legend.AddEntry(graph2,"BE","P")
legend.AddEntry(graph3,"EE","P")
else :
legend.AddEntry(graph2,"e^{+}e^{-}","P")
legend.AddEntry(graph3,"#mu^{+}#mu^{-}","P")
legend.SetFillColor(kWhite)
legend.SetLineColor(kBlack)
legend.SetTextFont(43)
legend.SetTextSize(20)
legend.Draw()
canvas.Update()
# Draw letters for data-taking periods
if(dataPeriod == "data2017"):
textLetters = TLatex()
textLetters.SetTextColor(kGray+1)
textLetters.SetTextSize(0.03)
if(ptype == "Zmass"):
textLetters.DrawLatex(2., gPad.GetUymin()+0.2,"B")
textLetters.DrawLatex(9.5, gPad.GetUymin()+0.2,"C")
textLetters.DrawLatex(16., gPad.GetUymin()+0.2,"D")
textLetters.DrawLatex(23., gPad.GetUymin()+0.2,"E")
textLetters.DrawLatex(36., gPad.GetUymin()+0.2,"F")
elif(ptype == "Zwidth"):
textLetters.DrawLatex(2., gPad.GetUymin()+0.3,"B")
textLetters.DrawLatex(9.5, gPad.GetUymin()+0.3,"C")
textLetters.DrawLatex(16., gPad.GetUymin()+0.3,"D")
textLetters.DrawLatex(23., gPad.GetUymin()+0.3,"E")
textLetters.DrawLatex(36., gPad.GetUymin()+0.3,"F")
elif(ptype == "Zmult") :
textLetters.DrawLatex(2., 260000,"B")
textLetters.DrawLatex(9.5, 260000,"C")
textLetters.DrawLatex(16., 260000,"D")
textLetters.DrawLatex(23., 260000,"E")
textLetters.DrawLatex(36., 260000,"F")
if(dataPeriod == "data2018"):
textLetters = TLatex()
textLetters.SetTextColor(kGray+1)
textLetters.SetTextSize(0.03)
if(ptype == "Zmass"):
textLetters.DrawLatex(6., gPad.GetUymin() + 0.6,"A")
textLetters.DrawLatex(16., gPad.GetUymin() + 0.6,"B")
textLetters.DrawLatex(23., gPad.GetUymin() + 0.6,"C")
textLetters.DrawLatex(43., gPad.GetUymin() + 0.6,"D")
elif(ptype == "Zwidth"):
textLetters.DrawLatex(6., gPad.GetUymin() +0.3,"A")
textLetters.DrawLatex(16., gPad.GetUymin() +0.3,"B")
textLetters.DrawLatex(23., gPad.GetUymin() +0.3,"C")
textLetters.DrawLatex(43., gPad.GetUymin() +0.3,"D")
elif(ptype == "Zmult") :
textLetters.DrawLatex(6., 260000,"A")
textLetters.DrawLatex(16., 260000,"B")
textLetters.DrawLatex(23., 260000,"C")
textLetters.DrawLatex(43., 260000,"D")
# ****
if(dataPeriod == "data2018"):
# draw vertical lines that divide different data taking periods
down = gPad.GetUymin()
up = gPad.GetUymax()
lineA = TLine(13.48, down, 13.48, up) # Run2018A up to 13.48 fb-1
lineA.SetLineColor(kBlack)
lineA.SetLineStyle(2)
lineA.Draw()
lineB = TLine(20.265, down, 20.265, up) # Run2018B up to 20.265 fb-1
lineB.SetLineColor(kBlack)
lineB.SetLineStyle(2)
lineB.Draw()
lineC = TLine(26.877, down, 26.877, up) # Run2018C up to 26.877 fb-1
lineC.SetLineColor(kBlack)
lineC.SetLineStyle(2)
lineC.Draw()
if(dataPeriod == "data2017"):
# draw vertical lines that divide different data taking periods
down = gPad.GetUymin()
up = gPad.GetUymax()
lineB = TLine(4.793, down, 4.793, up) # Run2017B up to 4.793 fb-1
lineB.SetLineColor(kBlack)
lineB.SetLineStyle(2)
lineB.Draw()
lineC = TLine(14.549, down, 14.549, up) # Run2017C up to 14.549 fb-1
lineC.SetLineColor(kBlack)
lineC.SetLineStyle(2)
lineC.Draw()
lineD = TLine(18.868, down, 18.868, up) # Run2017D up to 18.868 fb-1
lineD.SetLineColor(kBlack)
lineD.SetLineStyle(2)
lineD.Draw()
lineE = TLine(28.293, down, 28.293, up) # Run2017E up to 28.293 fb-1
lineE.SetLineColor(kBlack)
lineE.SetLineStyle(2)
lineE.Draw()
if(dataPeriod == "data2016"):
# draw vertical lines that divide different data taking periods
down = gPad.GetUymin()
up = gPad.GetUymax()
lineB = TLine(5.789, down, 5.789, up) # Run2016B up to 5.789 fb-1
lineB.SetLineColor(kBlack)
lineB.SetLineStyle(2)
lineB.Draw()
lineC = TLine(8.366, down, 8.366, up) # Run2016C up to 8.366 fb-1
lineC.SetLineColor(kBlack)
lineC.SetLineStyle(2)
lineC.Draw()
lineD = TLine(12.616, down, 12.616, up) # Run2016D up to 12.616 fb-1
lineD.SetLineColor(kBlack)
lineD.SetLineStyle(2)
lineD.Draw()
lineE = TLine(16.624, down, 16.624, up) # Run2016E up to 16.624 fb-1
lineE.SetLineColor(kBlack)
lineE.SetLineStyle(2)
lineE.Draw()
lineF = TLine(19.725, down, 19.725, up) # Run2016F up to 19.725 fb-1
lineF.SetLineColor(kBlack)
lineF.SetLineStyle(2)
lineF.Draw()
lineG = TLine(27.268, down, 27.268, up) # Run2016G up to 27.268 fb-1
lineG.SetLineColor(kBlack)
lineG.SetLineStyle(2)
lineG.Draw()
# ****
# draw orizontal lines for MC and DATA fit
if(ptype == "Zmass" or ptype == "Zwidth") :
leftEnd = gPad.GetUxmin()
rightEnd = gPad.GetUxmax()
if(DoInclusive):
line1 = TLine(leftEnd,lineMC1,rightEnd,lineMC1)
line1.SetLineColor(kBlack)
line1.SetLineStyle(1)
line1.Draw()
line2 = TLine(leftEnd,lineDATA1,rightEnd,lineDATA1)
line2.SetLineColor(kBlack)
line2.SetLineStyle(2)
line2.Draw()
# line for graph 2: color blue
line3 = TLine(leftEnd,lineMC2,rightEnd,lineMC2)
line3.SetLineColor(kBlue)
line3.SetLineStyle(1)
line3.Draw()
line4 = TLine(leftEnd,lineDATA2,rightEnd,lineDATA2)
line4.SetLineColor(kBlue)
line4.SetLineStyle(2)
line4.Draw()
# line for graph 3: color red
line5 = TLine(leftEnd,lineMC3,rightEnd,lineMC3)
line5.SetLineColor(kRed)
line5.SetLineStyle(1)
line5.Draw()
line6 = TLine(leftEnd,lineDATA3,rightEnd,lineDATA3)
line6.SetLineColor(kRed)
line6.SetLineStyle(2)
line6.Draw()
# ***
canvas.SaveAs(title + ".root")
canvas.SaveAs(title + ".pdf")
canvas.SaveAs(title + ".png")
return;
all_graphs[0].SetMaximum(calo_init.args.axisMax)
if calo_init.args.axisMin:
all_graphs[0].SetMinimum(calo_init.args.axisMin)
canv.Update()
lines = []
for iLine, line in enumerate(avgSF):
lines.append(TLine(0, avgSF[iLine], Nslices * sliceWidth, avgSF[iLine]))
lines[iLine].SetLineColor(iLine + 1)
lines[iLine].Draw('same')
# add second axis
canv.SetRightMargin(0.1)
all_graphs[0].GetXaxis().SetRangeUser(0, 68)
all_graphs[0].GetXaxis().SetLabelOffset(0.02)
gPad.RangeAxis(0, gPad.GetUymin(), 68, gPad.GetUxmax())
axis = TGaxis(gPad.GetUxmin(), gPad.GetUymin(), gPad.GetUxmax(),
gPad.GetUymin(),
gPad.GetUxmin() * calo_init.args.X0density,
gPad.GetUxmax() * calo_init.args.X0density, 506, "-")
axis.SetLabelSize(0.05)
axis.SetTitleSize(0.07)
axis.SetTitleOffset(0.9)
axis.SetLabelOffset(0.02)
axis.Draw()
axis.SetLabelFont(42)
unit1 = draw_text(['(cm)'], [0.91, 0.09, 1, 0.14], 1, 0)
unit1.SetTextSize(0.05)
unit1.SetTextFont(42)
unit2 = draw_text(['(X_{0})'], [0.91, 0.16, 1, 0.21], 1, 0)
unit2.SetTextSize(0.05)
def __call__(self):
h = self.hname
if not h:
return
if not isinstance(h, TH2):
return
gPad.GetCanvas().FeedbackMode(kTRUE)
event = gPad.GetEvent()
if (event == 1): # left mouse click
self.projection = not self.projection
elif (event == 12): # middle mouse click
self.logy = not self.logy
# erase old position and draw a line at current position
#(gPad coordinate system)
px = gPad.GetEventX()
py = gPad.GetEventY()
uxmin, uxmax = gPad.GetUxmin(), gPad.GetUxmax()
uymin, uymax = gPad.GetUymin(), gPad.GetUymax()
pxmin, pxmax = gPad.XtoAbsPixel(uxmin), gPad.XtoAbsPixel(uxmax)
pymin, pymax = gPad.YtoAbsPixel(uymin), gPad.YtoAbsPixel(uymax)
# if self.projection:
# axis = h.GetYaxis()
# else:
# axis = h.GetXaxis()
width = 1
if self._old != None:
if self.projection:
gVirtualX.DrawBox(pxmin, self._old[1] - width, pxmax,
self._old[1], kSolid)
else:
gVirtualX.DrawBox(self._old[0], pymin, self._old[0] + width,
pymax, kSolid)
if self.projection:
gVirtualX.DrawBox(pxmin, py, pxmax, py + width, kSolid)
else:
gVirtualX.DrawBox(px, pymin, px, pymax + width, kSolid)
self._old = px, py
upx = gPad.AbsPixeltoX(px)
x = gPad.PadtoX(upx)
upy = gPad.AbsPixeltoY(py)
y = gPad.PadtoY(upy)
padsav = gPad
# create or set the display canvases
if not self._cX:
self._cX = gPad.GetCanvas().GetPad(2)
else:
self._DestroyPrimitive('X')
if not self._cY:
self._cY = gPad.GetCanvas().GetPad(2)
else:
self._DestroyPrimitive('Y')
if self.projection:
self.DrawSlice(h, y, 'Y')
else:
self.DrawSlice(h, x, 'X')
padsav.cd()
hRawYieldDistr.append(
inFileMT.Get(f'hRawYield_pT_{ptMin*10:.0f}-{ptMax*10:.0f}'))
rms = hRawYieldDistr[-1].GetRMS() / hRawYieldDistr[-1].GetMean() * 100
shift = TMath.Abs(
hRawYields.GetBinContent(iPt) -
hRawYieldDistr[-1].GetMean()) / hRawYieldDistr[-1].GetMean() * 100
syst = TMath.Sqrt(rms**2 + shift**2)
hRMS.SetBinContent(iPt, rms)
hMeanShift.SetBinContent(iPt, shift)
hSyst.SetBinContent(iPt, syst)
cRMS = TCanvas('cRMS', '', 800, 800)
cRMS.DrawFrame(hRMS.GetBinLowEdge(1), 0.1, ptMax, 20.,
';#it{p}_{T} (GeV/#it{c});RMS (%)')
hRMS.DrawCopy('same')
axisSoverB = TGaxis(gPad.GetUxmax(), gPad.GetUymin(), gPad.GetUxmax(),
gPad.GetUymax(), 0.01, 20., 510, "+LG")
axisSoverB.SetLineColor(kRed + 1)
axisSoverB.SetLabelColor(kRed + 1)
axisSoverB.SetLabelFont(42)
axisSoverB.SetLabelSize(0.045)
axisSoverB.SetTitle('S/B (3#sigma)')
axisSoverB.SetTitleOffset(1.2)
axisSoverB.SetLabelOffset(0.012)
axisSoverB.SetTitleColor(kRed + 1)
axisSoverB.SetTitleFont(42)
axisSoverB.SetTitleSize(0.05)
axisSoverB.SetMaxDigits(3)
axisSoverB.Draw()
hSoverB.DrawCopy('same')
cRMS.Update()
