plotClonesPyRoot.py

Go to the documentation of this file.

from ROOT import *
import sys
import re

"""
Prepare the clones plot.
Parses the end of the output file, printed by the destructors of the CPUCruncher.
A line with the pattern "Summary: name=" is looked for.
The number of Events in flight is parsed as well.
"""

# configure
Color = kBlue
MarkerStyle = kFullStar
MarkerSize = 3.8

NEventsInFlight = 0
NThreads = -1


def parseLog(logfilename):
    global NEventsInFlight
    global NThreads
    ifile = open(logfilename, "r")
    lines = ifile.readlines()
    vals = []
    for line in lines:
        if "Summary: name=" in line:
            runtime_nclones = map(float,
                                  re.match(".* avg_runtime= ([0-9]*.[0-9]*|[0-9]*.[0-9]*e-[0-9]*) n_clones= ([0-9]).*", line).groups())
            vals.append(runtime_nclones)
        elif "Running with" in line:
            NEventsInFlight, NThreads = map(int, re.match(
                ".* Running with ([0-9]*) parallel events.*algorithms, ([0-9]*) threads", line).groups())

    return vals


def createGraph(vals):
    graph = TGraph(len(vals))
    counter = 0
    for runtime, nclones in vals:
        graph.SetPoint(counter, runtime, nclones)
        counter += 1

    graph.SetMarkerStyle(MarkerStyle)
    graph.SetMarkerSize(MarkerSize)
    graph.SetMarkerColor(Color)
    graph.SetTitle(
        "GaudiHive Speedup (Brunel, 100 evts);Algorithm Runtime [s];Number of Clones")
    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))

    lat.SetNDC(NDC)
    if angle != 0.:
        lat.SetTextAngle(angle)
    return lat


def getCountLatexes(vals, xmax):
    # print vals
    def getNclones(runtime_nclones):
        return runtime_nclones[1]
    max_nclones = int(max(vals, key=getNclones)[1])

    latexes = []
    for i in xrange(1, max_nclones + 1):
        n_algos = len(
            filter(lambda runtime_nclones: runtime_nclones[1] == i, vals))
        latexes.append(getText(xmax * 1.01,
                               i,
                               n_algos,
                               .7,
                               0,
                               600,
                               12))

    label = getText(.95,
                    .55,
                    "Total",
                    .8,
                    270,
                    600,
                    12,
                    True)

    latexes.append(label)

    return latexes


def doPlot(logfilename):
    global NEventsInFlight
    global NThreads
    vals = parseLog(logfilename)
    graph = createGraph(vals)
    nalgorithms = len(vals)

    canvas = TCanvas("Clones", "Clones", 1024, 768)
    canvas.cd()
    canvas.SetGrid()
    graph.Draw("AP")

    # Latex

    countLatexes = getCountLatexes(vals, graph.GetXaxis().GetXmax())
    map(lambda latex: latex.Draw(), countLatexes)
    evtsIf = getText(.6, .365,
                     "#splitline{#splitline{%s Simultaneous Events}{%s Threads}}{%s Algorithms}" % (
                         NEventsInFlight, NThreads, nalgorithms),
                     .8, 0, 2, 12, True)
    evtsIf.Draw()

    a = raw_input("press enter to continue")
    canvas.Print("PlotClones.png")


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