Tools: add parameters to test_build_options.py script

filtering by define name is the most useful bit

Tools/autotest/test_build_options.py

@@ -8,15 +8,33 @@ AP_FLAKE8_CLEAN
 
 from __future__ import print_function
 
-
+import fnmatch
+import optparse
 import os
 
 from pysim import util
 
 
 class TestBuildOptions(object):
-    def __init__(self):
+    def __init__(self,
+                 match_glob=None,
+                 do_step_disable_all=True,
+                 do_step_disable_none=False,
+                 build_targets=None):
         self.sizes_nothing_disabled = None
+        self.match_glob = match_glob
+        self.do_step_disable_all = do_step_disable_all
+        self.do_step_disable_none = do_step_disable_none
+        self.build_targets = build_targets
+        if self.build_targets is None:
+            self.build_targets = self.all_targets()
+
+    @staticmethod
+    def all_targets():
+        return ['copter', 'plane', 'rover', 'antennatracker', 'sub', 'blimp']
+
+    def progress(self, message):
+        print("###### %s" % message)
 
     # swiped from app.py:
     def get_build_options_from_ardupilot_tree(self):
@@ -63,13 +81,9 @@ class TestBuildOptions(object):
         }
         self.test_compile_with_defines(defines)
 
-    def build_targets(self):
-        '''return a list of build targets'''
-        return ['copter', 'plane', 'rover', 'antennatracker', 'sub', 'blimp']
-
     def board(self):
         '''returns board to build for'''
-        return "CubeOrange"
+        return "BeastH7v2"
 
     def test_compile_with_defines(self, defines):
         extra_hwdef_filepath = "/tmp/extra.hwdef"
@@ -78,7 +92,7 @@ class TestBuildOptions(object):
             self.board(),
             extra_hwdef=extra_hwdef_filepath,
         )
-        for t in self.build_targets():
+        for t in self.build_targets:
             try:
                 util.run_cmd([util.relwaf(), t])
             except Exception:
@@ -97,24 +111,29 @@ class TestBuildOptions(object):
             "sub": "ardusub",
             "blimp": "blimp",
         }
-        for target in self.build_targets():
+        for target in self.build_targets:
             path = os.path.join("build", self.board(), "bin", "%s.bin" % target_to_binpath[target])
             ret[target] = os.path.getsize(path)
         return ret
 
     def disable_in_turn_check_sizes(self, feature, sizes_nothing_disabled):
+        if not self.do_step_disable_none:
+            self.progress("disable-none skipped, size comparison not available")
+            return
         current_sizes = self.find_build_sizes()
         for (build, new_size) in current_sizes.items():
             old_size = sizes_nothing_disabled[build]
-            print("Disabling %s(%s) on %s saves %u bytes" %
-                  (feature.label, feature.define, build, old_size - new_size))
+            self.progress("Disabling %s(%s) on %s saves %u bytes" %
+                          (feature.label, feature.define, build, old_size - new_size))
 
     def run_disable_in_turn(self):
         options = self.get_build_options_from_ardupilot_tree()
+        if self.match_glob is not None:
+            options = list(filter(lambda x : fnmatch.fnmatch(x.define, self.match_glob), options))
         count = 1
         for feature in options:
-            print("##### Disabling feature %s(%s) (%u/%u)" %
-                  (feature.label, feature.define, count, len(options)))
+            self.progress("Disabling feature %s(%s) (%u/%u)" %
+                          (feature.label, feature.define, count, len(options)))
             self.test_feature(feature, options)
             count += 1
             self.disable_in_turn_check_sizes(feature, self.sizes_nothing_disabled)
@@ -123,6 +142,9 @@ class TestBuildOptions(object):
         options = self.get_build_options_from_ardupilot_tree()
         defines = {}
         for feature in options:
+            if self.match_glob is not None:
+                if not fnmatch.fnmatch(feature.define, self.match_glob):
+                    continue
             defines[feature.define] = 0
         self.test_compile_with_defines(defines)
 
@@ -131,11 +153,40 @@ class TestBuildOptions(object):
         self.sizes_nothing_disabled = self.find_build_sizes()
 
     def run(self):
-        self.run_disable_all()
-        self.run_disable_none()
+        if self.do_step_disable_all:
+            self.progress("Running disable-all step")
+            self.run_disable_all()
+        if self.do_step_disable_none:
+            self.progress("Running disable-none step")
+            self.run_disable_none()
         self.run_disable_in_turn()
 
 
 if __name__ == '__main__':
-    tbo = TestBuildOptions()
+
+    parser = optparse.OptionParser()
+    parser.add_option("--define-match-glob",
+                      type='string',
+                      default=None,
+                      help='feature define must match this glob to be tested')
+    parser.add_option("--no-disable-all",
+                      action='store_true',
+                      help='Do not run the disable-all step')
+    parser.add_option("--no-disable-none",
+                      action='store_true',
+                      help='Do not run the disable-none step')
+    parser.add_option("--build-targets",
+                      type='choice',
+                      choices=TestBuildOptions.all_targets(),
+                      action='append',
+                      help='vehicle targets to build')
+
+    opts, args = parser.parse_args()
+
+    tbo = TestBuildOptions(
+        match_glob=opts.define_match_glob,
+        do_step_disable_all=not opts.no_disable_all,
+        do_step_disable_none=not opts.no_disable_none,
+        build_targets=opts.build_targets,
+    )
     tbo.run()