Support all exported metrics in plotting scripts (#981)

scripts/plot_utils.py

@@ -0,0 +1,27 @@
+"""Shared metric options for the plotting scripts."""
+
+METRICS = {
+    "time_wall_clock": ("Time [s]", 1),
+    "time_user": ("User CPU time [s]", 1),
+    "time_system": ("System CPU time [s]", 1),
+    "memory_peak_resident": ("Peak resident memory [MiB]", 1024**2),
+}
+
+
+def add_metric_argument(parser):
+    parser.add_argument(
+        "--metric",
+        choices=METRICS,
+        default="time_wall_clock",
+        help="Metric to plot (default: time_wall_clock). "
+        "Times are in seconds; memory is in MiB.",
+    )
+
+
+def validate_metric(parser, results, metric):
+    for result in results:
+        if metric not in result["summary"]:
+            parser.error(
+                f"metric {metric!r} is unavailable for "
+                f"{result.get('name', result['command'])!r}"
+            )