plotBacklogPyRoot.py

Go to the documentation of this file.

 
from __future__ import print_function
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])