Add kconfig state space exploration script (random + neighbourhood)

1 files changed, 83 insertions, 0 deletions

diff --git a/bin/explore-kconfig.py b/bin/explore-kconfig.py
new file mode 100755
index 0000000..596faff
--- /dev/null
+++ b/bin/explore-kconfig.py
@@ -0,0 +1,83 @@
+#!/usr/bin/env python3
+
+import argparse
+import logging
+import os
+import sys
+
+from dfatool import kconfig
+
+from versuchung.experiment import Experiment
+from versuchung.types import String, Bool, Integer
+from versuchung.files import File, Directory
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        formatter_class=argparse.RawDescriptionHelpFormatter, description=__doc__
+    )
+    parser.add_argument(
+        "--neighbourhood",
+        type=str,
+        help="Explore neighbourhood of provided .config file(s)",
+    )
+    parser.add_argument(
+        "--log-level",
+        default=logging.INFO,
+        type=lambda level: getattr(logging, level.upper()),
+        help="Set log level",
+    )
+    parser.add_argument(
+        "--random",
+        type=int,
+        help="Explore a number of random configurations (make randconfig)",
+    )
+    parser.add_argument(
+        "--clean-command", type=str, help="Clean command", default="make clean"
+    )
+    parser.add_argument(
+        "--build-command", type=str, help="Build command", default="make"
+    )
+    parser.add_argument(
+        "--attribute-command",
+        type=str,
+        help="Attribute extraction command",
+        default="make attributes",
+    )
+    parser.add_argument("project_root", type=str, help="Project root directory")
+
+    args = parser.parse_args()
+
+    if isinstance(args.log_level, int):
+        logging.basicConfig(level=args.log_level)
+    else:
+        print(f"Invalid log level. Setting log level to INFO.", file=sys.stderr)
+
+    kconf = kconfig.KConfig(args.project_root)
+
+    if args.clean_command:
+        kconf.clean_command = args.clean_command
+    if args.build_command:
+        kconf.build_command = args.build_command
+    if args.attribute_command:
+        kconf.attribute_command = args.attribute_command
+
+    if args.random:
+        for i in range(args.random):
+            logging.info(f"Running experiment {i+1} of {args.random}")
+            kconf.run_randconfig()
+
+    if args.neighbourhood:
+        if os.path.isfile(args.neighbourhood):
+            kconf.run_exploration_from_file(args.neighbourhood)
+        elif os.path.isdir(args.neighbourhood):
+            pass
+        else:
+            print(
+                f"--neighbourhod: Error: {args.neighbourhood} must be a file or directory, but is neither",
+                file=sys.stderr,
+            )
+
+
+if __name__ == "__main__":
+    main()