plotBacklogPyRoot.py

Go to the documentation of this file.

from ROOT import *
import sys
import re

"""
Produces the backlog plot, parsing the output of the EventLoopManager.
Lines with the pattern "Event backlog" are looked for.
Events in flight are looked for as well.
"""

#configuration

NEventsInFlight=-1
NThreads=-1

LineStyles=[1,2]
LineWidth=3
Colors=[kBlue,kGreen+2]
MarkerStyles=[kOpenCircle,kFullCircle]
MarkesSize=1.5

graphCounter=0
maxY=-1

LegendDrawOpts="lp"


def parseLog(logfilename):
  # a line looks like 
  #"HiveSlimEventLoopMgr  SUCCESS Event backlog (max= 3, min= 0 ) = 3"
  global NEventsInFlight
  global NThreads
  ifile = open(logfilename,"r")
  lines=ifile.readlines()
  ifile.close()
  content=[]
  for line in lines:      
    if "Event backlog" in line:
      content.append( re.match(".* \(max= ([0-9]*), min= ([0-9]*) \) = ([0-9]*).*",line).groups())
    elif "Running with" in line:
      NEventsInFlight,NThreads=map(int, re.match(".*Running with ([0-9]*).* ([0-9]*) threads",line).groups())      
  
  return content

def createGraph(max_min_blog_vals):  
  global graphCounter
  global maxY
  graph = TGraph(len(max_min_blog_vals))
  counter=0
  for maxn,minn,blog in max_min_blog_vals:
    blog=float(blog)
    graph.SetPoint(counter,counter+1,float(blog))
    if maxY < blog:
      maxY = blog
    counter+=1
    
  graph.SetMarkerSize(MarkesSize)
  graph.SetMarkerStyle(MarkerStyles[graphCounter])
  graph.SetMarkerColor(Colors[graphCounter])
  graph.SetLineWidth(LineWidth)
  graph.SetLineColor(Colors[graphCounter])
  graph.SetLineStyle(LineStyles[graphCounter])
  
  graphCounter+=1
  
  return graph
  
def createInFlightGraph(nevts):
  global NEventsInFlight
  graph = TGraph(2)
  graph.SetPoint(0,0.,float(NEventsInFlight))
  graph.SetPoint(1,float(nevts)+1,float(NEventsInFlight))
  graph.SetLineWidth(3 )
  graph.SetLineColor(kRed)
  graph.SetLineStyle(2)
  graph.SetTitle("GaudiHive Backlog (Brunel, 100 evts);Events Finished;Event Backlog")
  print NEventsInFlight
  return graph

def getText(x,y,text,scale,angle,colour,font,NDC=False):
    lat = TLatex(float(x),float(y),
                 "#scale[%s]{#color[%s]{#font[%s]{%s}}}"%(scale,colour,font,text))
    if (NDC):
      lat.SetNDC()
    if angle!=0.:
       lat.SetTextAngle(angle)
    return lat 

def doPlot(logfilename,logfilename_copy):
  global NEventsInFlight
  global maxY
  global NThreads
  vals=parseLog(logfilename)
  vals_c=parseLog(logfilename_copy)
  n_vals=len(vals)
  inFlightgraph=createInFlightGraph(n_vals)
  graph = createGraph( vals )
  graph_c = createGraph( vals_c )
  
  canvas= TCanvas ("Backlog","Backlog",1100,900)
  canvas.cd()
  canvas.SetGrid()  
  inFlightgraph.Draw("APL")
  inFlightgraph.GetYaxis().SetRangeUser(0.,maxY*1.2)
  inFlightgraph.GetXaxis().SetRangeUser(0.,float(n_vals+1))
  graph.Draw("PLSame")
  graph_c.Draw("PLSame")
  
  # Labels
  eventInFlightLabel=getText(float(n_vals+1)*1.03,NEventsInFlight,"#splitline{# Simultaneous}{       Events}",.6,270,2,12)
  eventInFlightLabel.Draw()
  nThreadsLabel=getText(.15,.7,"%s Threads"%NThreads,.6,0,2,12,True)
  nThreadsLabel.Draw()
  
  # Build a Legend
  legend = TLegend(.7,.75,.9,.9)
  legend.SetFillColor(kWhite)
  legend.SetHeader("Algo Management")
  legend.AddEntry(graph,"No Cloning",LegendDrawOpts)
  legend.AddEntry(graph_c,"Cloning enabled",LegendDrawOpts)
  legend.Draw()  
    
  a= raw_input("press enter to continue") 
  
  canvas.Print("EventBacklog.png")

if __name__ == "__main__":
  argc=len(sys.argv)
  if argc!=3:
    print "Usage: plotBacklogPyRoot.py logfilename logfilename_copy"
    sys.exit(1)
  doPlot(sys.argv[1],sys.argv[2])