1 files changed, 60 insertions, 0 deletions
diff --git a/src/boost/libs/numeric/ublas/benchmarks/plot.py b/src/boost/libs/numeric/ublas/benchmarks/plot.py
new file mode 100755
index 000000000..1ca5e5a7f
--- /dev/null
+++ b/src/boost/libs/numeric/ublas/benchmarks/plot.py
@@ -0,0 +1,60 @@
+#!/usr/bin/env python
+#
+# Copyright (c) 2018 Stefan Seefeld
+# All rights reserved.
+#
+# This file is part of Boost.uBLAS. It is made available under the
+# Boost Software License, Version 1.0.
+# (Consult LICENSE or http://www.boost.org/LICENSE_1_0.txt)
+
+import argparse
+import matplotlib.pyplot as plt
+import numpy as np
+
+
+class plot(object):
+
+    def __init__(self, label, data):
+        self.label = label
+        self.data = data
+
+
+def load_file(filename):
+
+    lines = open(filename, 'r').readlines()
+    label = lines[0][1:-1].strip()
+    lines = [l.strip() for l in lines]
+    lines = [l.split('#', 1)[0] for l in lines]
+    lines = [l for l in lines if l]
+    data = [l.split() for l in lines]
+    return plot(label, list(zip(*data)))
+
+
+def main(argv):
+
+    parser = argparse.ArgumentParser(prog=argv[0], description='benchmark plotter')
+    parser.add_argument('data', nargs='+', help='benchmark data to plot')
+    parser.add_argument('--log', choices=['no', 'all', 'x', 'y'], help='use a logarithmic scale')
+    args = parser.parse_args(argv[1:])
+    runs = [load_file(d) for d in args.data]
+    plt.title('Benchmark plot')
+    plt.xlabel('size')
+    plt.ylabel('time (s)')
+    if args.log == 'all':
+        plot = plt.loglog
+    elif args.log == 'x':
+        plot = plt.semilogx
+    elif args.log == 'y':
+        plot = plt.semilogy
+    else:
+        plot = plt.plot
+    plots = [plot(r.data[0], r.data[1], label=r.label) for r in runs]
+    plt.legend()
+    plt.show()
+    return True
+
+    
+if __name__ == '__main__':
+
+    import sys
+    sys.exit(0 if main(sys.argv) else 1)