Github/littlefs
1
Fork 0
Code Actions Activity

scripts: test.py/bench.py: Added ifndef attribute for tests/benches

Browse Source 
As you might expect, this is the inverse of ifdef, and is useful for
supporting opt-out flags.

I don't think ifdef + ifndef is powerful enough to handle _all_
compile-time corner cases, but they at least provide convenient handling
for the most common flags. Worst case, tests/benches can always include
explicit #if/#ifdef/#ifndef statements in the code itself.
This commit is contained in:
Christopher Haster
2025-06-24 13:23:43 -05:00
parent db1f941e90
commit 213dba6f6d
2 changed files with 64 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/bench.py
+32 -6
View File
@@ -87,6 +87,9 @@ class BenchCase:
self.ifdef = config.pop('ifdef', [])
if not isinstance(self.ifdef, list):
self.ifdef = [self.ifdef]
self.ifndef = config.pop('ifndef', [])
if not isinstance(self.ifndef, list):
self.ifndef = [self.ifndef]
self.code = config.pop('code')
self.code_lineno = config.pop('code_lineno', None)
self.in_ = config.pop('in',
@@ -224,6 +227,9 @@ class BenchSuite:
self.ifdef = config.pop('ifdef', [])
if not isinstance(self.ifdef, list):
self.ifdef = [self.ifdef]
self.ifndef = config.pop('ifndef', [])
if not isinstance(self.ifndef, list):
self.ifndef = [self.ifndef]
self.code = config.pop('code', None)
self.code_lineno = min(
@@ -385,9 +391,11 @@ def compile(bench_paths, **args):
# the bench defines
def write_case_functions(f, suite, case):
# write any ifdef prologues
if case.ifdef:
if case.ifdef or case.ifndef:
for ifdef in case.ifdef:
f.writeln('#ifdef %s' % ifdef)
for ifndef in case.ifndef:
f.writeln('#ifndef %s' % ifndef)
f.writeln()
# create case define functions
@@ -446,16 +454,20 @@ def compile(bench_paths, **args):
f.writeln()
# write any ifdef epilogues
if case.ifdef:
if case.ifdef or case.ifndef:
for ifdef in case.ifdef:
f.writeln('#endif')
for ifndef in case.ifndef:
f.writeln('#endif')
f.writeln()
if not args.get('source'):
# write any ifdef prologues
if suite.ifdef:
if suite.ifdef or suite.ifndef:
for ifdef in suite.ifdef:
f.writeln('#ifdef %s' % ifdef)
for ifndef in suite.ifndef:
f.writeln('#ifndef %s' % ifndef)
f.writeln()
# write any suite defines
@@ -496,9 +508,11 @@ def compile(bench_paths, **args):
f.writeln()
# write any ifdef epilogues
if suite.ifdef:
if suite.ifdef or suite.ifndef:
for ifdef in suite.ifdef:
f.writeln('#endif')
for ifndef in suite.ifndef:
f.writeln('#endif')
f.writeln()
# create suite struct
@@ -512,6 +526,8 @@ def compile(bench_paths, **args):
or 0))
for ifdef in suite.ifdef:
f.writeln(4*' '+'#ifdef %s' % ifdef)
for ifndef in suite.ifndef:
f.writeln(4*' '+'#ifndef %s' % ifndef)
# create suite defines
if suite.defines:
f.writeln(4*' '+'.defines = (const bench_define_t[]){')
@@ -522,6 +538,8 @@ def compile(bench_paths, **args):
f.writeln(4*' '+'.define_count = %d,' % len(suite.defines))
for ifdef in suite.ifdef:
f.writeln(4*' '+'#endif')
for ifndef in suite.ifndef:
f.writeln(4*' '+'#endif')
if suite.cases:
f.writeln(4*' '+'.cases = (const struct bench_case[]){')
for case in suite.cases:
@@ -536,6 +554,8 @@ def compile(bench_paths, **args):
or 0))
for ifdef in it.chain(suite.ifdef, case.ifdef):
f.writeln(12*' '+'#ifdef %s' % ifdef)
for ifndef in it.chain(suite.ifndef, case.ifndef):
f.writeln(12*' '+'#ifndef %s' % ifndef)
# create case defines
if case.defines:
f.writeln(12*' '+'.defines'
@@ -567,6 +587,8 @@ def compile(bench_paths, **args):
case.name))
for ifdef in it.chain(suite.ifdef, case.ifdef):
f.writeln(12*' '+'#endif')
for ifndef in it.chain(suite.ifndef, case.ifndef):
f.writeln(12*' '+'#endif')
f.writeln(8*' '+'},')
f.writeln(4*' '+'},')
f.writeln(4*' '+'.case_count = %d,' % len(suite.cases))
@@ -600,9 +622,11 @@ def compile(bench_paths, **args):
# write any internal benches
for suite in suites:
# any ifdef prologues
if suite.ifdef:
if suite.ifdef or suite.ifndef:
for ifdef in suite.ifdef:
f.writeln('#ifdef %s' % ifdef)
for ifndef in suite.ifndef:
f.writeln('#ifndef %s' % ifndef)
f.writeln()
# any suite code
@@ -622,9 +646,11 @@ def compile(bench_paths, **args):
write_case_functions(f, suite, case)
# any ifdef epilogues
if suite.ifdef:
if suite.ifdef or suite.ifndef:
for ifdef in suite.ifdef:
f.writeln('#endif')
for ifndef in suite.ifndef:
f.writeln('#endif')
f.writeln()
# declare our bench suites