Update task runner to conditionally compile / clean benchmarks

Author: jacob-hughes

tasks.py

@@ -9,10 +9,10 @@
 @task
 def build_alloy(c):
     """Build all alloy configurations"""
-    exps = Experiments()
+    exps = Experiments.new()
     cfgs = exps.configurations(only_missing=True)
     alloy_cfgs_needed = set(cfg.alloy for cfg in cfgs if not cfg.alloy.installed)
-    alloy_build_steps = sum(a.build_steps for a in alloy_cfgs_needed)
+    alloy_build_steps = sum(a.steps for a in alloy_cfgs_needed)
     with timer("Building missing alloy configurations", alloy_build_steps):
         for alloy in alloy_cfgs_needed:
             alloy.build()
@@ -22,14 +22,17 @@ def build_alloy(c):
 def build_benchmarks(c, suites=None):
     """Build all benchmarks for all configurations"""
 
-    exps = Experiments()
+    exps = Experiments.new()
     if suites:
         exps = exps.filter_suites(suites)
 
     cfgs = exps.configurations(only_missing=True)
 
     alloy_cfgs_needed = set(cfg.alloy for cfg in cfgs if not cfg.alloy.installed)
-    alloy_build_steps = sum(a.build_steps for a in alloy_cfgs_needed)
+    alloy_build_steps = sum(a.steps for a in alloy_cfgs_needed)
+
+    print(f"Found {len(alloy_cfgs_needed)} Alloy configuration(s):")
+    [print(f"  {os.path.relpath(a.path)}") for a in alloy_cfgs_needed]
 
     with timer("Building missing alloy configurations", alloy_build_steps):
         for alloy in alloy_cfgs_needed:
@@ -43,7 +46,7 @@ def build_benchmarks(c, suites=None):
 @task
 def run_benchmarks(c, pexecs, experiments=None, suites=None, metric=None):
     pexecs = int(pexecs)
-    exps = Experiments(pexecs)
+    exps = Experiments.new(pexecs)
     if suites:
         exps = exps.filter_suites(suites)
 
@@ -117,13 +120,17 @@ def compare(c, experiment_name, cfg, name, suite=None):
 
 @task
 def clean_alloy(c, experiments=None):
-    exps = Experiments()
+    exps = Experiments.new()
 
     if experiments:
         exps = exps.filter_experiments(experiments)
 
     to_remove = set(cfg.alloy for cfg in exps.configurations() if cfg.alloy.installed)
 
+    if not to_remove:
+        print("No Alloy configurations installed")
+        return
+
     print(f"Found {len(to_remove)} Alloy configuration(s):")
     [print(f"  {os.path.relpath(a.bin)}") for a in to_remove]
 
@@ -146,21 +153,37 @@ def clean_results(c, suites=None):
 
 
 @task
-def clean_benchmarks(c, suites=None, reclone=False):
+def clean_benchmarks(c, suites=None, including_src=False):
     """Clean all benchmarks for all configurations"""
 
-    exps = Experiments()
+    exps = Experiments.new()
     if suites:
         exps = exps.filter_suites(suites)
 
     cfgs = exps.configurations(only_installed=True)
+
+    if not cfgs:
+        print("No benchmark programs installed")
+        return
     print(f"Found {len(cfgs)} configurations:")
     [print(f"  {os.path.relpath(cfg.bin)}") for cfg in cfgs]
 
     response = input(f"Are you sure you want to remove them? (y/n): ").strip().lower()
     if response == "y":
         for cfg in cfgs:
-            cfg.remove()
+            cfg.remove(including_src)
         print(f"{len(cfgs)} benchmark configurations removed.")
     else:
         print("cancelled.")
+
+
+@task
+def clean(c):
+    response = (
+        input("Are you sure you want to remove everything? (y/n): ").strip().lower()
+    )
+    if response == "y":
+        clean_alloy(c)
+        clean_benchmarks(c)
+    else:
+        print("cancelled.")