Gustavo J. A. M. Carneiro
106 lines
2.6 KiB
Python
106 lines
2.6 KiB
Python
|
|
import pylab
|
|
from xml.dom.minidom import parse, Node
|
|
import sys
|
|
import numpy as np
|
|
import shutil
|
|
import os
|
|
|
|
DPI = 75
|
|
|
|
def getText(nodelist):
|
|
rc = ""
|
|
for node in nodelist:
|
|
if node.nodeType == node.TEXT_NODE:
|
|
rc = rc + node.data
|
|
return rc
|
|
|
|
def main(argv):
|
|
input_fname = sys.argv[1]
|
|
dom = parse(input_fname)
|
|
outdir = sys.argv[2]
|
|
res = dom.getElementsByTagName("results")[0]
|
|
tools = [e for e in res.childNodes if e.nodeType == Node.ELEMENT_NODE]
|
|
|
|
# get the pybindgen revno
|
|
pbg_env = dom.getElementsByTagName("environment")[0].getElementsByTagName("pybindgen")[0]
|
|
pbg_txt = getText(pbg_env.childNodes).split("\n")
|
|
for l in pbg_txt:
|
|
k, s, v = l.partition(':')
|
|
k = k.strip()
|
|
if k == 'revno':
|
|
v = v.strip()
|
|
revno = v
|
|
break
|
|
|
|
num_tests = len(tools[0].getElementsByTagName("test"))
|
|
shutil.rmtree(outdir, True)
|
|
os.mkdir(outdir)
|
|
shutil.copy2(input_fname, outdir)
|
|
|
|
figures = []
|
|
|
|
|
|
sizes = [float(t.getAttribute('module-file-size')) for t in tools]
|
|
labels= [t.tagName for t in tools]
|
|
ind = range(len(sizes))
|
|
pylab.bar(ind, sizes)
|
|
pylab.xticks([0.5+x for x in ind], labels)
|
|
pylab.title("Extension module file size (B)")
|
|
fname = "sizes.png"
|
|
pylab.savefig(os.path.join(outdir, fname), dpi=DPI)
|
|
figures.append(fname)
|
|
|
|
|
|
for t in range(num_tests):
|
|
pylab.figure()
|
|
labels = []
|
|
values = []
|
|
for x, tool in enumerate(tools):
|
|
labels.append(tool.tagName)
|
|
values.append(float(tool.getElementsByTagName("test")[t].getAttribute('time')))
|
|
ind = range(len(values))
|
|
pylab.bar(ind, values)
|
|
pylab.xticks([0.5+x for x in ind], labels)
|
|
|
|
desc = tools[0].getElementsByTagName("test")[t].getAttribute("description")
|
|
pylab.title(desc)
|
|
|
|
fname = "%s.png" % (desc)
|
|
pylab.savefig(os.path.join(outdir, fname), dpi=DPI)
|
|
figures.append(fname)
|
|
|
|
index_html = file("%s/index.html" % outdir, "wt")
|
|
print >> index_html, """
|
|
<html>
|
|
<head>
|
|
<title> PyBindGen Benchmarks </title>
|
|
</head>
|
|
|
|
<body>
|
|
|
|
<div>Details in the <a href=\"%s\">Raw XML file</a>.
|
|
<a href=\"http://bazaar.launchpad.net/~gjc/pybindgen/trunk/files/%s/benchmarks/\">Source files for the benchmarks</a>.
|
|
</div>
|
|
|
|
""" % (os.path.basename(input_fname), revno)
|
|
|
|
for fig in figures:
|
|
print >> index_html, """
|
|
<div>
|
|
<img src=\"%s\"/>
|
|
</div>
|
|
""" % (fig,)
|
|
|
|
print >> index_html, """
|
|
</body>
|
|
"""
|
|
|
|
index_html.close()
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main(sys.argv)
|
|
|