def produceFamilyPlot(family, surname):
canvas = TCanvas(surname, "", 2400, 800)
legend = TLegend(0.1, 0.5, 0.4, 0.87)
legend.SetBorderSize(0)
legend.SetFillStyle(0)
counter = 0
markers = []
for person in family:
if len(person) < 2:
print "error"
continue
name = person[0]
sex = person[1]
if len(person) >= 3:
birth = person[2]
else:
birth = 0
plot = makePlot(name, sex)
normalizePlot(plot)
plot.SetTitle("")
plot.GetYaxis().SetLabelSize(0)
plot.GetXaxis().SetLabelSize(0.07)
plot.GetYaxis().SetTickSize(0)
plot.GetYaxis().SetTitle("normalized popularity")
plot.GetYaxis().SetTitleSize(0.07)
plot.GetYaxis().SetTitleOffset(0.22)
plot.SetLineColor(colors[counter])
legend.AddEntry(plot, name, "L")
plot.Draw("same C")
if birth > 0:
marker = TMarker(birth+0.5,
plot.GetBinContent(plot.FindBin(birth)), 20)
marker.SetMarkerColor(colors[counter])
marker.SetMarkerSize(5)
markers.append(marker)
markers[counter].Draw("same")
counter += 1
legend.Draw()
canvas.SaveAs(surname + ".pdf")
def draw_potential():
markers = []
c1 = TCanvas("c1", "Prova")
h2 = TH2F("h2", "h2", 1000, -100, 1200, 1000, 0, 1000)
m = re.compile("(\d+)\s(\d+)\s([\-0-9\.]+)\s([\-0-9]+)")
file = open("mappa_piazza.dat")
while 1:
line = file.readline()
if (not line):
break
n = m.match(line)
if (n):
po = TMarker(float(n.group(1)), float(n.group(2)), 6)
po.SetMarkerColor(abs(int(n.group(4))) + 1)
markers.append(po)
h2.Draw()
for i in markers:
i.Draw("SAME")
c1.SaveAs("potenziale_piazza.png")
sys.exit(1)
def myMarkerTextSmall2(x, y, lcolor, lstyle, mcolor, mstyle, msize, size,
text):
l = TLatex()
l.SetTextAlign(12)
l.SetTextSize(size / 2.5)
l.SetNDC()
l.DrawLatex(x, y, text)
yb1 = y - 0.15 * size
yb2 = y + 0.15 * size
xb1 = x - 0.28 * size - 0.45 * size
xb2 = x + 0.28 * size - 0.45 * size
print "box = ", xb1, yb1, xb2, yb2
#print "line = ",xl1,yl1,xl2,yl2
mline1 = TLine(xb1, yb1, xb2, yb2)
mline1.SetLineColor(lcolor)
mline1.SetLineStyle(lstyle)
mline1.SetLineWidth(2)
y_new = (yb1 + yb2) / 2.
mline1.DrawLineNDC(xb1, y_new, xb2, y_new)
mline2 = TLine(xb1, yb1, xb2, yb2)
mline2.SetLineColor(lcolor)
mline2.SetLineStyle(lstyle)
mline2.SetLineWidth(2)
x_new = (xb1 + xb2) / 2.
mline2.DrawLineNDC(x_new, yb1, x_new, yb2)
marker = TMarker((xb1 + xb2) / 2.0, y, mstyle)
marker.SetNDC()
marker.SetMarkerColor(mcolor)
marker.SetMarkerStyle(mstyle)
marker.SetMarkerSize(msize)
marker.Draw()
return marker
def plotFailedHoMatchesNoTrg():
c = TCanvas("cFailedHoMatchesNoTrg","cFailedHoMatchesNoTrg",1200,1200)
c.cd().SetRightMargin(0.25)
c.cd().SetLeftMargin(0.08)
#Graph for in events not in geometric acceptance
grNotInGaNC = file.Get("hoMuonAnalyzer/graphs/NoTrgTdmiNotInGA")
grNotInGa = PlotStyle.convertToHcalCoords(grNotInGaNC)
grNotInGa.GetYaxis().SetTitle("i#phi / a.u.")
grNotInGa.GetXaxis().SetTitle("i#eta / a.u.")
grNotInGa.GetYaxis().SetTitleFont(62)
grNotInGa.GetYaxis().SetLabelFont(62)
grNotInGa.SetMarkerStyle(6)
grNotInGa.SetMarkerColor(PlotStyle.colorRwthDarkBlue)
grNotInGa.SetTitle("#eta #phi plot failed HO matches in no Single #mu Trg. events")
grNotInGa.Draw("AP")
#Graph of events with HO match below threshold
grHoBelowThrNC = file.Get("hoMuonAnalyzer/graphs/NoTrgTdmiBelowThr")
grHoBelowThr = PlotStyle.convertToHcalCoords(grHoBelowThrNC)
grHoBelowThr.SetMarkerStyle(20)
grHoBelowThr.SetMarkerSize(1.2)
grHoBelowThr.SetMarkerColor( PlotStyle.colorRwthGruen )
grHoBelowThr.Draw("samep")
#Graph for events where HO matching failed
grHoMatchFailNC = file.Get("hoMuonAnalyzer/graphs/NoTrgHoMatchFail")
grHoMatchFail = PlotStyle.convertToHcalCoords(grHoMatchFailNC)
grHoMatchFail.SetMarkerStyle(21)
grHoMatchFail.SetMarkerSize(1)
grHoMatchFail.SetMarkerColor( PlotStyle.colorRwthRot )
grHoMatchFail.Draw("samep")
#Draw chimneys
chimney1Converted = PlotStyle.convertToHcalCoords(PlotStyle.chimney1)
chimney2Converted = PlotStyle.convertToHcalCoords(PlotStyle.chimney2)
chimney1Converted.SetLineColor(PlotStyle.colorRwthMagenta)
chimney2Converted.SetLineColor(PlotStyle.colorRwthMagenta)
chimney1Converted.Draw('same')
chimney2Converted.Draw('same')
#cms private label
label = TPaveText(PlotStyle.getLabelCmsPrivateSimulation(x1ndc=0.5,x2ndc=0.75))
label.Draw()
#create extra marker for the legend
marker = TMarker(1,1,2)
marker.SetMarkerSize(3)
marker.SetMarkerColor(PlotStyle.colorRwthDarkBlue)
#legend
legend = TLegend(0.75,0.8,0.99,0.9)
legend.AddEntry(chimney2Converted,"chimney","l")
legend.AddEntry(marker,'Not in GA','p')
legend.AddEntry(grHoMatchFail,'HO match fail','p')
legend.AddEntry(grHoBelowThr,'HO match < 0.2 GeV','p')
legend.Draw()
nNotMatching = grHoMatchFail.GetN()
nNotInGa = grNotInGa.GetN()
nBelowThr = grHoBelowThr.GetN()
nTotal = nNotMatching + nNotInGa + nBelowThr
print 80*'#'
print 'Not Matching:\t%5d/%d\t=> %5.2f%% +- %f%%' % (nNotMatching,nTotal,nNotMatching/float(nTotal)*100,PlotStyle.calcSigma(nNotMatching,float(nTotal)))
print 'Not in GA:\t%5d/%d\t=> %5.2f%% +- %f%%' % (nNotInGa,nTotal,nNotInGa/float(nTotal)*100,PlotStyle.calcSigma(nNotInGa,float(nTotal)))
print 'Below Thr:\t%5d/%d\t=> %5.2f%% +- %f%%' % (nBelowThr,nTotal,nBelowThr/float(nTotal)*100,PlotStyle.calcSigma(nBelowThr,float(nTotal)))
print 80*'#'
c.Update()
c.SaveAs('plots/graphsEtaPhi/gNoTrgHoMatchingFailed.png')
c.SaveAs('plots/graphsEtaPhi/gNoTrgHoMatchingFailed.pdf')
return c,grNotInGa,label,chimney1Converted,chimney2Converted,legend,grHoMatchFail,grHoBelowThr
ROCwmass.SetLineColor(2)
ROCwmass.SetLineWidth(2)
ROCwmass.SetTitle(";signal efficiency;background rejection")
c2 = TCanvas("c2")
ptlimstrs = [str(int(2000.0*ptlims[0])),str(int(2000.0*ptlims[1]))]
ROCwmass.Draw()
prelim.DrawLatex( 0.15, 0.85, ptlimstrs[0]+" GeV < p_{T} < "+ptlimstrs[1]+" GeV" )
if (cust=="w" or cust=="top") and Drawmarker:
if cust=="w":
prelim.DrawLatex( 0.44, 0.55, "N_{2}^{DDT}+msd" )
TM = TMarker(0.4, 0.01, 23)
if cust=="top":
prelim.DrawLatex( 0.44, 0.60, "#tau_{32}+msd+subjetb" )
TM = TMarker(0.4, 0.017, 23)
TM.SetMarkerSize(2)
TM.SetMarkerColor(4)
TM.Draw()
c2.SetLogy()
c2.Write("rocwmass")
ROC.Draw()
c2.Write("roccomp")
dnnsig.Write()
dnnbkg.Write()
output.Write()
output.Close()
g_width.Draw('AC')
g_width.SetLineWidth(2)
g_width.SetTitle('hMSSM, m_{{A}} = {mass} GeV'.format(mass=int(mass)))
l_5percent.SetLineColor(2)
l_5percent.SetLineWidth(3)
l_10percent.SetLineColor(2)
l_10percent.SetLineWidth(3)
l_10percent.SetLineStyle(2)
l_25percent.SetLineColor(2)
l_25percent.SetLineStyle(3)
l_25percent.SetLineWidth(3)
p_5percent.SetMarkerStyle(20)
p_5percent.SetMarkerColor(2)
p_10percent.SetMarkerStyle(21)
p_10percent.SetMarkerColor(2)
p_25percent.SetMarkerStyle(22)
p_25percent.SetMarkerColor(2)
p_5percent.Draw()
l_5percent.Draw()
p_10percent.Draw()
l_10percent.Draw()
p_25percent.Draw()
l_25percent.Draw()
g_width.GetXaxis().SetTitle('1/tan(#beta)^{2}') # #prop coupling^{2}
g_width.GetYaxis().SetTitle('Width (GeV)')
g_width.GetYaxis().SetTitleSize(0.05)