plotBacklogPyRoot.py

Go to the documentation of this file.

import re
import sys
 
from ROOT import (
    TCanvas,
    TGraph,
    TLatex,
    TLegend,
    kBlue,
    kFullCircle,
    kGreen,
    kOpenCircle,
    kRed,
    kWhite,
)
 
"""
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)
    lines = ifile.readlines()
    ifile.close()
    content = []
    for line in lines:
        if "Event backlog" in line:
            content.append(
                re.match(
                    r".* \‍(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.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),
        f"#scale[{scale}]{{#color[{colour}]{{#font[{font}]{{{text}}}}}}}",
    )
    if NDC:
        lat.SetNDC()
    if angle != 0.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.0, maxY * 1.2)
    inFlightgraph.GetXaxis().SetRangeUser(0.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}",
        0.6,
        270,
        2,
        12,
    )
    eventInFlightLabel.Draw()
    nThreadsLabel = getText(0.15, 0.7, f"{NThreads} Threads", 0.6, 0, 2, 12, True)
    nThreadsLabel.Draw()
 
    # Build a Legend
    legend = TLegend(0.7, 0.75, 0.9, 0.9)
    legend.SetFillColor(kWhite)
    legend.SetHeader("Algo Management")
    legend.AddEntry(graph, "No Cloning", LegendDrawOpts)
    legend.AddEntry(graph_c, "Cloning enabled", LegendDrawOpts)
    legend.Draw()
 
    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])