support output filtering

summarize.py

@@ -105,37 +105,37 @@ def summarize(avgs=None, stats=None, desc_stats=None) -> int:
     return 0
 
 
-def collect_and_summarize(result_dir=None, desc_stats=None) -> int:
-    """Collect data and print a summary of the collected data"""
-    data = collect(result_dir)
-    stats = calc_stats(data)
-
-    if not stats:
-        return 1
-
-    if not summarize(stats=stats, desc_stats=desc_stats):
-        return 1
-
-    return 0
-
-
-if __name__ == '__main__':
+def main():
+    """program entry point"""
     parser = argparse.ArgumentParser()
     parser.add_argument('-s',
                         '--desc-stats',
                         help='print all stats not only means',
                         nargs='*')
+    parser.add_argument("-i",
+                        "--implementations",
+                        help="implementations to plot",
+                        nargs='+')
     parser.add_argument('--yaml',
                         help='dump statistics as yaml',
                         action='store_true')
     parser.add_argument('result_dir',
                         help='directory containing the results to summarize')
 
-    _args = parser.parse_args()
+    args = parser.parse_args()
+
+    data = collect(args.result_dir)
+    stats = calc_stats(data)
+    if args.implementations:
+        stats = {v: s for v, s in stats.items() if v in args.implementations}
 
-    if _args.yaml:
-        print(yaml.safe_dump(calc_stats(collect(_args.result_dir))))
+    if args.yaml:
+        print(yaml.safe_dump(stats))
         sys.exit(0)
 
     print('### Summary ###')
-    sys.exit(collect_and_summarize(**vars(_args)))
+    sys.exit(summarize(data, desc_stats=args.desc_stats))
+
+
+if __name__ == '__main__':
+    main()