def draw_box(boxList,xl,xr,yl,yr,opacity=1,color=1,style=1001,lstyle=1,lw=3):
from ROOT import TBox
b=TBox(xl,yl,xr,yr)
b.SetBit(ROOT.kCanDelete)
b.SetFillStyle(style)
b.SetFillColorAlpha(color, opacity)
b.SetLineColor(color)
b.SetLineWidth(lw)
b.SetLineStyle(lstyle)
b.Draw()
boxList.append(b)
def drawHoBoxes(canvas):
canvas.cd()
boxes = []
for i in range(1, 6):
for j in range(1, 6):
box = TBox(-2.5 + i - 1, -2.5 + j - 1, -2.5 + i, -2.5 + j)
box.SetFillStyle(0)
box.SetLineColor(ROOT.kBlack)
box.SetLineWidth(2)
box.Draw()
boxes.append(box)
return boxes
def drawHoBoxes(canvas): canvas.cd() boxes = [] for i in range(-1,2): for j in range(-1,2): box = TBox(i*0.087 - 0.0435,j*0.087 - 0.0435,(i*0.087+0.0435),(j*0.087+0.0435)) box.SetFillStyle(0) box.SetLineColor(colorRwthMagenta) box.SetLineWidth(3) box.Draw() boxes.append(box) return boxes
def drawHcalBoxesHcalCoords(canvas): canvas.cd() boxes = [] for i in range(-15,15): for j in range(0,72): box = TBox(i,j,(i+1),(j+1)) box.SetFillStyle(0) box.SetLineColor(colorRwthMagenta) box.SetLineWidth(2) box.Draw() boxes.append(box) return boxes
def drawHoBoxes(canvas):
canvas.cd()
boxes = []
for i in range(1, 8):
for j in range(1, 8):
box = TBox(-0.3045 + (i - 1) * 0.087, -0.3045 + (j - 1) * 0.087,
-0.3045 + i * 0.087, -0.3045 + j * 0.087)
box.SetFillStyle(0)
box.SetLineColor(ROOT.kBlack)
box.SetLineWidth(2)
box.Draw()
boxes.append(box)
return boxes
def makeTimeVsEtaPlot(self, source, title=""):
canvas = TCanvas(source)
if title == "":
title = source
hist = TH2D(source, title + ";i#eta;Time / ns;#", 33, -16.5, 16.5, 201,
-100.5, 100.5)
graph = self.fileHandler.getGraph('graphs/timingSupport_' + source)
fillGraphIn2DHist(graph, hist)
hist.SetStats(0)
hist.Draw('colz')
canvas.Update()
setupAxes(hist)
label = self.drawLabel()
canvas.Update()
fractionGraph, counterDict = self.printFractionsPerIEta(graph)
self.debug('Integral of plot %20s: %d' % (source, hist.Integral()))
medianTofZero = (counterDict[9]['median'] +
counterDict[11]['median']) / 2.
tofFunction = TF1(
'f',
'4*sqrt(1+ 1/(tan(2*atan(exp(-x*0.087/2.)))**2))/300000000.*1e9 - 13.3 + [0]',
-10, 10)
tofFunction.SetParameter(0, medianTofZero)
tofFunction.Draw('same')
###
# Draw Boxes for the Intervals
###
intervalBoxes = []
for index, item in enumerate(counterDict):
iEta = index - 10
# Skip iEta 0
if iEta == 0:
continue
box = TBox(iEta - 0.5, item['median'] - 12.5, iEta + 0.5,
item['median'] + 12.5)
box.SetFillStyle(0)
box.SetLineColor(colorRwthMagenta)
box.SetLineWidth(2)
box.Draw()
intervalBoxes.append(box)
return canvas, hist, label, fractionGraph, intervalBoxes, tofFunction
def plot3x3MatchQualityCodesVsPt(self):
c = TCanvas('cMatchQC3x3', 'Match QC 3x3', 0, 0, 900, 700)
c.SetLogz()
hist = self.fileHandler.getHistogram(
'qualityCode/L1Muon3x3Match_QcVsPt')
hist.SetStats(0)
hist.Scale(1, 'width')
hist.Draw('colz')
c.Update()
setupPalette(hist)
c.Update()
label = self.drawLabel()
c.Update()
box = TBox(0, 6.5, 180, 7.5)
box.SetLineColor(3)
box.SetLineWidth(2)
box.Draw()
c.Update()
return hist, c, label, box
def pullsVertical_noBonly(fileName):
content = filterPullFile(fileName)
nbins, off = len(content), 0.10
# Graphs
h_pulls = TH2F("pulls", "", 6, -3., 3., nbins, 0, nbins)
S_pulls = TGraphAsymmErrors(nbins)
boxes = []
canvas = TCanvas("canvas", "Pulls", 720, 300 + nbins * 18) #nbins*20)
canvas.cd()
canvas.SetGrid(0, 1)
canvas.SetTopMargin(0.01)
canvas.SetRightMargin(0.01)
canvas.SetBottomMargin(0.10)
canvas.SetLeftMargin(0.40)
canvas.SetTicks(1, 1)
for i, s in enumerate(content):
l = s.split()
h_pulls.GetYaxis().SetBinLabel(i + 1, l[0])
S_pulls.SetPoint(i, float(l[3]), float(i + 1) - 0.5)
S_pulls.SetPointError(i, float(l[4]), float(l[4]), 0., 0.)
h_pulls.GetXaxis().SetTitle("(#hat{#theta} - #theta_{0}) / #Delta#theta")
h_pulls.GetXaxis().SetLabelOffset(0.0)
h_pulls.GetXaxis().SetTitleOffset(0.8)
h_pulls.GetXaxis().SetLabelSize(0.045)
h_pulls.GetXaxis().SetTitleSize(0.050)
h_pulls.GetYaxis().SetLabelSize(0.046)
h_pulls.GetYaxis().SetNdivisions(nbins, 0, 0)
S_pulls.SetFillColor(kBlack)
S_pulls.SetLineColor(kBlack)
S_pulls.SetMarkerColor(kBlack)
S_pulls.SetLineWidth(2)
S_pulls.SetMarkerStyle(20)
S_pulls.SetMarkerSize(1)
box1 = TBox(-1., 0., 1., nbins)
#box1.SetFillStyle(3001) # 3001 checkered
#box1.SetFillStyle(0)
box1.SetFillColor(kGreen + 1) # 417
box1.SetLineWidth(2)
box1.SetLineStyle(2)
box1.SetLineColor(kGreen + 1) # 417
box2 = TBox(-2., 0., 2., nbins)
#box2.SetFillStyle(3001) # 3001 checkered
#box2.SetFillStyle(0)
box2.SetFillColor(kOrange) # 800
box2.SetLineWidth(2)
box2.SetLineStyle(2)
box2.SetLineColor(kOrange) # 800
leg = TLegend(0.01, 0.01, 0.3, 0.15)
leg.SetTextSize(0.05)
leg.SetBorderSize(0)
leg.SetFillStyle(0)
leg.SetFillColor(0)
#leg.SetNColumns(2)
leg.AddEntry(S_pulls, "S+B fit", "lp")
if text: leg.AddEntry(0, text, "")
h_pulls.Draw("")
box2.Draw()
box1.Draw()
S_pulls.Draw("P6SAME")
leg.Draw()
canvas.RedrawAxis()
canvas.Print(outName + ".png")
canvas.Print(outName + ".pdf")
if not gROOT.IsBatch(): raw_input("Press Enter to continue...")
def pullsVertical(fileName):
content = filterPullFile(fileName)
nbins, off = len(content), 0.10
b_pulls = TH1F("b_pulls", ";;Pulls", nbins, 0. - off, nbins - off)
s_pulls = TH1F("s_pulls", ";;Pulls", nbins, 0. + off, nbins + off) #
for i, s in enumerate(content):
l = s.split()
b_pulls.GetXaxis().SetBinLabel(i + 1, l[0])
s_pulls.GetXaxis().SetBinLabel(i + 1, l[0])
b_pulls.SetBinContent(i + 1, float(l[1]))
b_pulls.SetBinError(i + 1, float(l[2]))
s_pulls.SetBinContent(i + 1, float(l[3]))
s_pulls.SetBinError(i + 1, float(l[4]))
b_pulls.SetFillStyle(3005)
b_pulls.SetFillColor(923)
b_pulls.SetLineColor(923)
b_pulls.SetLineWidth(1)
b_pulls.SetMarkerStyle(20)
b_pulls.SetMarkerSize(1.25)
s_pulls.SetLineColor(602)
s_pulls.SetMarkerColor(602)
s_pulls.SetMarkerStyle(24) #24
s_pulls.SetLineWidth(1)
b_pulls.GetYaxis().SetRangeUser(-2.5, 2.5)
# Graphs
h_pulls = TH2F("pulls", "", 6, -3., 3., nbins, 0, nbins)
B_pulls = TGraphAsymmErrors(nbins)
S_pulls = TGraphAsymmErrors(nbins)
boxes = []
canvas = TCanvas("canvas", "Pulls", 600, 150 + nbins * 10) #nbins*20)
canvas.cd()
canvas.SetGrid(0, 1)
canvas.GetPad(0).SetTopMargin(0.01)
canvas.GetPad(0).SetRightMargin(0.01)
canvas.GetPad(0).SetBottomMargin(0.05)
canvas.GetPad(0).SetLeftMargin(0.25) #(0.25)#(0.065)
canvas.GetPad(0).SetTicks(1, 1)
for i, s in enumerate(content):
l = s.split()
if "1034h" in l[0]: l[0] = "CMS_PDF_13TeV"
h_pulls.GetYaxis().SetBinLabel(i + 1, l[0].replace('CMS2016_', '')) #C
#y1 = gStyle.GetPadBottomMargin()
#y2 = 1. - gStyle.GetPadTopMargin()
#h = (y2 - y1) / float(nbins)
#y1 = y1 + float(i) * h
#y2 = y1 + h
#box = TPaveText(0, y1, 1, y2, 'NDC')
#box.SetFillColor(0)
#box.SetTextSize(0.02)
#box.SetBorderSize(0)
#box.SetTextAlign(12)
#box.SetMargin(0.005)
#if i % 2 == 0:
# box.SetFillColor(18)
#box.Draw()
#boxes.append(box)
B_pulls.SetPoint(i + 1, float(l[1]), float(i + 1) - 0.3) #C
B_pulls.SetPointError(i + 1, float(l[2]), float(l[2]), 0., 0.) #C
for i, s in enumerate(content):
l = s.split()
S_pulls.SetPoint(i + 1, float(l[3]), float(i + 1) - 0.7) #C
S_pulls.SetPointError(i + 1, float(l[4]), float(l[4]), 0., 0.) #C
h_pulls.GetXaxis().SetTitle("(#hat{#theta} - #theta_{0}) / #Delta#theta")
h_pulls.GetXaxis().SetLabelOffset(-0.01)
h_pulls.GetXaxis().SetTitleOffset(.6)
h_pulls.GetYaxis().SetNdivisions(nbins, 0, 0)
B_pulls.SetFillColor(1)
B_pulls.SetLineColor(1)
B_pulls.SetLineStyle(1)
B_pulls.SetLineWidth(2)
B_pulls.SetMarkerColor(1)
B_pulls.SetMarkerStyle(20)
B_pulls.SetMarkerSize(1) #(0.75)
S_pulls.SetFillColor(629)
S_pulls.SetLineColor(629)
S_pulls.SetMarkerColor(629)
S_pulls.SetLineWidth(2)
S_pulls.SetMarkerStyle(20)
S_pulls.SetMarkerSize(1)
box1 = TBox(-1., 0., 1., nbins)
box1.SetFillStyle(3001)
#box1.SetFillStyle(0)
box1.SetFillColor(417)
box1.SetLineWidth(2)
box1.SetLineStyle(2)
box1.SetLineColor(417)
box2 = TBox(-2., 0., 2., nbins)
box2.SetFillStyle(3001)
#box2.SetFillStyle(0)
box2.SetFillColor(800)
box2.SetLineWidth(2)
box2.SetLineStyle(2)
box2.SetLineColor(800)
leg = TLegend(0.1, -0.05, 0.7, 0.08)
leg.SetBorderSize(0)
leg.SetFillStyle(0)
leg.SetFillColor(0)
leg.SetNColumns(2)
leg.AddEntry(B_pulls, "B-only fit", "lp")
leg.AddEntry(S_pulls, "S+B fit", "lp")
if text: leg.AddEntry(0, text, "")
h_pulls.Draw("")
box2.Draw()
box1.Draw()
B_pulls.Draw("P6SAME")
S_pulls.Draw("P6SAME")
leg.Draw()
# drawCMS(35867, "Preliminary")
# drawAnalysis("VH")
# drawRegion(outName)
canvas.Print(outName + ".png")
canvas.Print(outName + ".pdf")
if not gROOT.IsBatch(): raw_input("Press Enter to continue...")
c_xy = TCanvas("c_xy")
h_xy.Draw()
h_xy_mc.Draw("same")
h_xy_corr.Draw("same")
h_xy_corr.SetMarkerColor(kBlue + 1)
h_xy_mc.SetMarkerColor(kRed + 1)
h_xy.GetYaxis().SetTitleOffset(1.37)
c_xy.Update()
c_spacecharge.cd()
tpc_xy = TBox(x_start, y_start, x_end, y_end)
tpc_xy.SetLineWidth(3)
tpc_xy.SetLineStyle(2)
tpc_xy.SetFillStyle(0)
tpc_xy.SetFillStyle(0)
tpc_xy.SetLineWidth(3)
h_xy.Draw("col")
h_xy.GetYaxis().SetTitleOffset(1.3)
#h_xy_corr.Draw("col")
h_xy_corr.GetYaxis().SetTitleOffset(1.3)
preliminary = TPaveText(0.48, 0.74, 0.81, 0.84, "NDC")
preliminary.SetShadowColor(0)
preliminary.SetBorderSize(0)
preliminary.SetFillStyle(0)
preliminary.AddText("MicroBooNE Preliminary")
#preliminary.Draw()
def main():
if True:
mf=TGMainFrame(gClient.GetRoot(),1500,475)
gvf=TGVerticalFrame(mf,1500,475)
rec=TRootEmbeddedCanvas("ccc",gvf,1500,450)
rec2=TRootEmbeddedCanvas("ccc2",gvf,1500,25)
gvf.AddFrame(rec,TGLayoutHints(ROOT.kLHintsExpandX|ROOT.kLHintsTop))
gvf.AddFrame(rec2,TGLayoutHints(ROOT.kLHintsExpandX|ROOT.kLHintsBottom))
mf.AddFrame(gvf,TGLayoutHints(ROOT.kLHintsExpandX))
cc=rec.GetCanvas()
cc2=rec2.GetCanvas()
mf.SetEditable(0)
mf.SetWindowName('HPS ECAL FADC SCALERS')
mf.MapSubwindows()
mf.Resize(1501,476)# resize to get proper frame placement
mf.MapWindow()
else:
cc=TCanvas('cc','',1500,450)
cc.cd()
cc.SetBorderMode(0)
cc.SetFixedAspectRatio(1)
cc.FeedbackMode(1)
gStyle.SetOptStat(0)
gStyle.SetGridStyle(1)
gStyle.SetGridColor(11)
hh=TH2D('hh',';X;Y',46,-22,24,11,-5,6)
hi=TH2I('hi',';X;Y',46,-22,24,11,-5,6)
setupHists([hh,hi])
xax,yax=hh.GetXaxis(),hh.GetYaxis()
hh.Draw('COLZ')
hi.Draw('TEXTSAME')
gPad.SetLogz()
gPad.SetGrid(1,1)
gPad.SetLeftMargin(0.05)
tt1=TPaveText(0.1,0.9,0.3,1.0,'NDC')
tt2=TPaveText(0.7,0.91,0.9,0.99,'NDC')
ttT=TPaveText(-22+13+0.05,6-5,-22+22,7-5-0.05)
ttB=TPaveText(-22+13+0.05,4-5+0.05,-22+22,5-5)
ttM=TPaveText(-22+0+0.05,5-5+0.05,-22+13,6-5.01)
ttime=TPaveText(-10,-6.5,10,-5.8)
tchan=TPaveText(0,0,0.9,1)
setupPaveTexts([tt1,tt2,ttT,ttB,ttM,ttime,tchan])
ttM.SetTextColor(2)
bb=TBox()
bb.SetFillStyle(1001)
bb.SetFillColor(0)
bb.SetLineWidth(1)
bb.SetLineColor(1)
bb.DrawBox(-9+0.05,-1,0,1.97)
bb.DrawBox(-24,0,24.05,0.97)
tarrow=TText(14.5,0.3,'Beam Left')
arrow=TArrow(19,0.5,23,0.5,0.02,'|>')
arrow.SetAngle(40)
arrow.SetFillColor(1)
arrow.SetLineWidth(2)
tt=TText()
tt.SetTextColor(1)
tt.SetTextAngle(90)
tt.SetTextSize(0.08)
tt.DrawText(25.4,0,'kHz')
tt.SetTextAngle(0)
tt.SetTextColor(2)
tt.DrawTextNDC(0.3,0.92,'ECAL FADC SCALERS')
cc.cd()
for xx in [tt2,ttT,ttB,ttM,arrow,tarrow,ttime]: xx.Draw()
cc2.cd()
tchan.Draw('NDC')
cc.cd()
ll=TLine()
ll.DrawLine(xax.GetXmin(),yax.GetXmin(),xax.GetXmax(),yax.GetXmin())
ll.DrawLine(xax.GetXmin(),yax.GetXmax(),xax.GetXmax(),yax.GetXmax())
ll.DrawLine(xax.GetXmin(),yax.GetXmin(),xax.GetXmin(),0)
ll.DrawLine(xax.GetXmax(),yax.GetXmin(),xax.GetXmax(),0)
ll.DrawLine(xax.GetXmin(),yax.GetXmax(),xax.GetXmin(),1)
ll.DrawLine(xax.GetXmax(),yax.GetXmax(),xax.GetXmax(),1)
ll.DrawLine(xax.GetXmax(),0,0,0)
ll.DrawLine(xax.GetXmax(),1,0,1)
ll.DrawLine(xax.GetXmin(),0,-9,0)
ll.DrawLine(xax.GetXmin(),1,-9,1)
ll.DrawLine(-9,-1,0,-1)
ll.DrawLine(-9,2,0,2)
ll.DrawLine(-9,1,-9,2)
ll.DrawLine(-9,-1,-9,0)
ll.DrawLine(0,-1,0,0)
ll.DrawLine(0,1,0,2)
gPad.SetEditable(0)
while True:
# try:
zvals=getPVS()
for ii in range(len(zvals)):
hh.SetBinContent(xax.FindBin(XVALS[ii]),yax.FindBin(YVALS[ii]),zvals[ii])
hi.SetBinContent(xax.FindBin(XVALS[ii]),yax.FindBin(YVALS[ii]),zvals[ii])
for xx in [ttime,tt2,ttT,ttB,ttM]: xx.Clear()
[top,bottom,maximum]=calcRates(zvals)
tt2.AddText('Total: %.1f MHz'%((top+bottom)/1000))
ttT.AddText('%.1f MHz'%(top/1000))
ttB.AddText('%.1f MHz'%(bottom/1000))
ttM.AddText('MAX SINGLE CRYSTAL = %.0f kHz'%(maximum))
ttime.AddText(makeTime())
if gPad.GetEvent()==11:
xy=pix2xy(gPad)
ee=ECAL.findChannelXY(xy[0],xy[1])
if ee:
tchan.Clear()
tchan.AddText(printChannel(ee))
cc2.Modified()
cc2.Update()
elif gPad.GetEvent()==12:
tchan.Clear()
cc2.Modified()
cc2.Update()
cc.Modified()
cc.Update()
time.sleep(1)
def main():
cc.cd()
cc.SetBorderMode(0)
cc.SetFixedAspectRatio(1)
cc.FeedbackMode(1)
gStyle.SetOptStat(0)
gStyle.SetGridStyle(1)
gStyle.SetGridColor(11)
hh=TH2D('hh',';X;Y',22,-10.5,11.5,22,-10.5,11.5)
hi=TH2I('hi',';X;Y',22,-10.5,11.5,22,-10.5,11.5)
setupHists([hh,hi])
xax,yax=hh.GetXaxis(),hh.GetYaxis()
hh.Draw('COLZ')
hi.Draw('TEXTSAME')
gPad.SetLogz()
gPad.SetGrid(1,1)
gPad.SetLeftMargin(0.09)
gPad.SetRightMargin(0.11)
#tt2=TPaveText(0.7,0.96,0.9,0.99,'NDC')
ttM=TPaveText(-3+0.05, 7-4.45, 4.0, 8-4.51)
tt2=TPaveText(-3+0.05, 7-5.45, 4.0, 8-5.51)
ttX=TPaveText(-2, 7-8.00, 3, 8-8.00)
ttY=TPaveText(-2, 7-9.00, 3, 8-9.00)
ttZ=TPaveText(-2, 6-8.80, 3, 8-9.30)
ttZ.AddText("positive = beam top/right")
ttime=TPaveText(-10,-12.5,10,-11.8)
tchan=TPaveText(0,0,0.9,1)
setupPaveTexts([tt2,ttM,ttime,tchan,ttX,ttY,ttZ])
ttM.SetTextColor(2)
ttM.SetFillStyle(0)
ttZ.SetFillStyle(0)
tt2.SetFillStyle(0)
tarrow=TText(-0.9,0.7,'Beam Right')
tarrow.SetTextSizePixels(15)
arrow=TArrow(-1.4,0.5,2.4,0.5,0.02,'|>')
arrow.SetAngle(40)
arrow.SetFillColor(1)
arrow.SetLineWidth(2)
tt=TText()
tt.SetTextColor(1)
tt.SetTextAngle(90)
tt.SetTextSize(0.04)
tt.DrawText(12.4,0,'kHz')
tt.SetTextAngle(0)
tt.SetTextColor(1)
tt.DrawTextNDC(0.3,0.92,'FTC FADC SCALERS')
bb=TBox()
bb.SetFillStyle(1001)
bb.SetFillColor(0)
bb.SetLineWidth(1)
bb.SetLineColor(1)
bb.DrawBox(-3.47,-1.47,4.47,2.46)
bb.DrawBox(-1.47,-3.47,2.49,4.47)
bb.DrawBox(-2.47,-2.47,3.49,3.47)
cc.cd()
for xx in [ttM,tt2,ttime,arrow,tarrow,ttX,ttY,ttZ]: xx.Draw()
cc2.cd()
tchan.Draw('NDC')
cc.cd()
gPad.SetEditable(0)
while True:
for ch in ECAL.chans:
loadPV(ch)
ch=ch.vals
xx,yy=ch['X'],ch['Y']
#if (ch['PVVAL']>10):
# print xx,yy,ch['PVVAL']
# swap x to make it downstream view:
xx=-xx
#after, fix the fact x=0 / y=0 don't exists
if xx<0: xx+=1
if yy<0: yy+=1
hh.SetBinContent(xax.FindBin(xx),yax.FindBin(yy),ch['PVVAL'])
hi.SetBinContent(xax.FindBin(xx),yax.FindBin(yy),ch['PVVAL'])
for xx in [ttime,tt2,ttM,ttX,ttY]: xx.Clear()
[total,maximum,top,bottom,left,right]=calcRates(ECAL.chans)
tt2.AddText('Total: %.1f MHz'%(total/1000))
ttM.AddText('Max: %.0f kHz'%(maximum))
if total>1e2:
xasy = (right-left)/total
yasy = (top-bottom)/total
ttX.AddText('X-Asy: %+.1f%%'%(100*xasy))
ttY.AddText('Y-Asy: %+.1f%%'%(100*yasy))
else:
ttX.AddText('X-Asy: N/A')
ttY.AddText('Y-Asy: N/A')
ttime.AddText(makeTime())
if not gPad: sys.exit()
if gPad.GetEvent()==11:
xy=pix2xy(gPad)
ee=ECAL.findChannelXY(xy[0],xy[1])
if ee:
tchan.Clear()
tchan.AddText(printChannel(ee))
cc2.Modified()
cc2.Update()
elif gPad.GetEvent()==12:
tchan.Clear()
cc2.Modified()
cc2.Update()
cc.Modified()
cc.Update()
time.sleep(2)
def drawROIGeometry(geometry, is2016):
outfn = "ROILayout%s.pdf" % ("2016" if is2016 else "2015")
global box, c, h, leg
gROOT.Reset()
gStyle.SetOptStat(0)
c = TCanvas('c', "MuCTPi Geometry %s" % "2016" if is2016 else "2015", 1400,
950)
c.Draw()
#h = TH2F("h","Muon Geometry",10,-2.6,2.6,10,-6.4,6.4)
h = TH2F("h", "Muon Geometry %s" % "2016" if is2016 else "2015", 10, -2.6,
2.6, 10, -0.15, 6.4)
h.SetXTitle("#eta")
h.SetYTitle("#phi")
h.Draw()
box = TBox()
box.SetFillStyle(0)
box.SetLineColor(3)
box.SetLineColor(3)
text = TLatex()
text.SetTextSize(0.005)
text.SetTextFont(42)
text.SetTextAlign(22)
secLabel = TLatex()
secLabel.SetTextSize(0.008)
secLabel.SetTextFont(42)
secLabel.SetTextAlign(22)
leg = TLegend(0.7, 0.1, 0.9, 0.4)
leg.SetEntrySeparation(0.05)
leg.SetNColumns(2)
#for MIOCT in geometry.getMIOCTs():
for colorIndex, MioctID in enumerate(drawOrder):
MIOCT = geometry.getMIOCT(MioctID)
firstBox = True
color = colorMap[colorIndex % len(colorMap)]
#print "Using color ",color
box.SetLineColor(color)
box.SetLineWidth(1)
for Sector in MIOCT.Sectors:
ymin = 10
ymax = -10
for ROI in Sector.ROIs:
c1_x = float(ROI["etamin"])
c1_y = float(ROI["phimin"])
c2_x = float(ROI["etamax"])
c2_y = float(ROI["phimax"])
ymin = min(ymin, c1_y)
ymax = max(ymax, c2_y)
#print "eta [%f - %f], phi [%f - %f]" % (c1_x,c2_x,c1_y,c2_y)
b = box.DrawBox(c1_x, c1_y, c2_x, c2_y)
text.DrawText((c1_x + c2_x) / 2, (c1_y + c2_y) / 2,
ROI["roiid"])
if firstBox:
firstBox = False
leg.AddEntry(b, "Slot %s" % MIOCT["slot"], "l")
if Sector["name"].startswith("B"):
if int(Sector["name"][1:]) < 32:
xpos = -0.02
ypos = (ymin + ymax) / 2 - 0.05
else:
xpos = 0.02
ypos = (ymin + ymax) / 2 + 0.03
secLabel.DrawText(xpos, ypos, Sector["name"])
leg.Draw()
c.Update()
c.SaveAs(outfn)
