mirror of
https://github.com/sharkdp/hyperfine.git
synced 2026-10-28 16:18:57 +00:00
Benjamin Rosemann
b04a9ec6c4
Add `imgname` parameter to be able to save a plot to a file instead of showing it. Use-case: platforms without GUI like docker containers. Add `labels` parameter to histogram plot. Use-case: replacing long commands with human friendly labels.
51 lines
1.4 KiB
Python
Executable File
51 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
"""This program shows `hyperfine` benchmark results as a histogram."""
|
|
|
|
import argparse
|
|
import json
|
|
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument("file", help="JSON file with benchmark results")
|
|
parser.add_argument("--title", help="Plot title")
|
|
parser.add_argument(
|
|
"--labels", help="Comma-separated list of entries for the plot legend"
|
|
)
|
|
parser.add_argument("--bins", help="Number of bins (default: auto)")
|
|
parser.add_argument(
|
|
"--type", help="Type of histogram (*bar*, barstacked, step, stepfilled)"
|
|
)
|
|
parser.add_argument("--imgname", help="Save image to the given filename.")
|
|
args = parser.parse_args()
|
|
|
|
with open(args.file) as f:
|
|
results = json.load(f)["results"]
|
|
|
|
if args.labels:
|
|
labels = args.labels.split(",")
|
|
else:
|
|
labels = [b["command"] for b in results]
|
|
all_times = [b["times"] for b in results]
|
|
|
|
t_min = np.min(list(map(np.min, all_times)))
|
|
t_max = np.max(list(map(np.max, all_times)))
|
|
|
|
bins = int(args.bins) if args.bins else "auto"
|
|
histtype = args.type if args.type else "bar"
|
|
|
|
plt.hist(
|
|
all_times, label=labels, bins=bins, histtype=histtype, range=(t_min, t_max),
|
|
)
|
|
plt.legend(prop={"family": ["Source Code Pro", "Fira Mono", "Courier New"]})
|
|
|
|
plt.xlabel("Time [s]")
|
|
if args.title:
|
|
plt.title(args.title)
|
|
|
|
if args.imgname:
|
|
plt.savefig(args.imgname)
|
|
else:
|
|
plt.show()
|