build: meson benchmarks autogen

Author: mochaaP

benchmarks/meson.build

@@ -10,11 +10,23 @@ benchmarks_exe = executable(
   build_by_default: false,
 )
 
-# manually written, list all benchmarks with:
-# builddir/benchmarks/msft_proxy_benchmarks --benchmark_list_tests
+autogen_benchmarks = custom_target(
+  command: [
+    autogen_benchmarks_exe,
+    custom_target(
+      command: [benchmarks_exe, '--benchmark_list_tests'],
+      capture: true,
+      output: 'tests.txt',
+    ),
+    meson.current_source_dir() / 'meson.build',
+  ],
+  output: 'autogen.stamp',
+  build_by_default: true,
+)
+
+#pragma autogen push
 benchmarks = {
   'ObjectCreation': [
-    'BM_SmallObjectCreationWithProxy',
     'BM_SmallObjectCreationWithProxy_Shared',
     'BM_SmallObjectCreationWithProxy_SharedPooled',
     'BM_SmallObjectCreationWithUniquePtr',
@@ -31,7 +43,6 @@ benchmarks = {
     'BM_LargeObjectCreationWithAny',
   ],
   'ObjectInvocation': [
-    'BM_SmallObjectInvocationViaProxy',
     'BM_SmallObjectInvocationViaProxy_Shared',
     'BM_SmallObjectInvocationViaProxyView',
     'BM_SmallObjectInvocationViaVirtualFunction',
@@ -44,8 +55,7 @@ benchmarks = {
     'BM_LargeObjectInvocationViaVirtualFunction_Shared',
     'BM_LargeObjectInvocationViaVirtualFunction_RawPtr',
   ],
-  'ObjectOperation': [
-    'BM_SmallObjectRelocationViaProxy',
+  'ObjectRelocation': [
     'BM_SmallObjectRelocationViaProxy_NothrowRelocatable',
     'BM_SmallObjectRelocationViaUniquePtr',
     'BM_SmallObjectRelocationViaAny',
@@ -55,6 +65,7 @@ benchmarks = {
     'BM_LargeObjectRelocationViaAny',
   ],
 }
+#pragma autogen pop
 
 foreach suite : benchmarks.keys()
   foreach name : benchmarks[suite]
@@ -63,6 +74,7 @@ foreach suite : benchmarks.keys()
       benchmarks_exe,
       args: [f'--benchmark_filter=^@name@$'],
       suite: suite,
+      depends: autogen_benchmarks,
     )
   endforeach
 endforeach

tools/meson.build

@@ -1,2 +1,3 @@
 autogen_tests_exe = find_program('./meson_autogen_tests.py')
 autogen_doctest_exe = find_program('./meson_autogen_doctest.py')
+autogen_benchmarks_exe = find_program('./meson_autogen_benchmarks.py')

tools/meson_autogen_benchmarks.py

@@ -0,0 +1,52 @@
+#!/usr/bin/env python3
+# pyright: strict
+
+import re
+import sys
+from pathlib import Path
+
+from mesonbuild import mformat
+
+PATTERN = re.compile(r"(?=[A-Z])")
+
+benchmarkfile = Path(sys.argv[1])
+mesonfile = Path(sys.argv[2])
+
+with benchmarkfile.open(encoding="utf-8") as f:
+    bench = [line.rstrip() for line in f.readlines()]
+
+with mesonfile.open(encoding="utf-8") as f:
+    build = f.readlines()
+
+begin = -1
+end = -1
+
+for i, line in enumerate(build):
+    if begin == -1 and line.strip() == "#pragma autogen push":
+        begin = i + 1
+    elif end == -1 and line.strip() == "#pragma autogen pop":
+        end = i
+
+if begin == -1 or end == -1:
+    raise ValueError
+
+benchmarks: "dict[str, list[str]]" = {}
+
+for name in bench:
+    suite = "".join(PATTERN.split(name.split('_', 1)[1])[2:4])
+    if suite not in benchmarks:
+        benchmarks[suite] = []
+    else:
+        benchmarks[suite].append(name)
+
+config = mformat.get_meson_format([mesonfile])
+formatter = mformat.Formatter(config, False, False)
+pretty = formatter.format(f"{benchmarks = }", mesonfile)
+
+if pretty == "".join(build[begin:end]):
+    exit()
+
+with mesonfile.open("w", encoding="utf-8") as f:
+    f.writelines(build[:begin])
+    _ = f.write(pretty)
+    f.writelines(build[end:])