blob: 46db3db4e518ac7d8531a25cc7b4d20d600fd2b3 [file] [log] [blame]
"""
Utility script to generate headers for the following macros
- Z_LISTIFY
- Z_UTIL_INC
- Z_UTIL_DEC
- Z_UTIL_X2
- Z_IS_EQ
.. note::
The script will simply create the header files in the current working directory,
they should be copied manually to $ZEPHYR_BASE/include/zephyr/sys/ accordingly.
Usage::
python $ZEPHYR_BASE/scripts/utils/gen_util_macros.py -l 4095
Copyright (c) 2023, Meta
SPDX-License-Identifier: Apache-2.0
"""
import argparse
def write_hidden_start(file):
file.write("/**\n")
file.write(" * @cond INTERNAL_HIDDEN\n")
file.write(" */\n")
def write_hidden_stop(file):
file.write("/**\n")
file.write(" * INTERNAL_HIDDEN @endcond\n")
file.write(" */\n")
def gen_util_listify(limit:int):
with open("util_listify.h", "w") as file:
write_hidden_start(file)
file.write("\n")
file.write("#ifndef ZEPHYR_INCLUDE_SYS_UTIL_LOOPS_H_\n")
file.write("#error \"This header should not be used directly, please include util_loops.h instead\"\n")
file.write("#endif /* ZEPHYR_INCLUDE_SYS_UTIL_LOOPS_H_ */\n")
file.write("\n")
file.write("#ifndef ZEPHYR_INCLUDE_SYS_UTIL_LISTIFY_H_\n")
file.write("#define ZEPHYR_INCLUDE_SYS_UTIL_LISTIFY_H_\n")
file.write("\n")
file.write("/* Set of UTIL_LISTIFY particles */\n")
file.write("#define Z_UTIL_LISTIFY_0(F, sep, ...)\n\n")
file.write("#define Z_UTIL_LISTIFY_1(F, sep, ...) \\\n")
file.write(" F(0, __VA_ARGS__)\n\n")
for i in range(2, limit + 3):
file.write(f"#define Z_UTIL_LISTIFY_{i}(F, sep, ...) \\\n")
file.write(f" Z_UTIL_LISTIFY_{i - 1}(F, sep, __VA_ARGS__) __DEBRACKET sep \\\n")
file.write(f" F({i - 1}, __VA_ARGS__)\n")
file.write("\n")
file.write("#endif /* ZEPHYR_INCLUDE_SYS_UTIL_LISTIFY_H_ */\n")
file.write("\n")
write_hidden_stop(file)
def gen_util_internal_is_eq(limit):
with open("util_internal_is_eq.h", "w") as file:
write_hidden_start(file)
file.write("\n")
file.write("#ifndef ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_H_\n")
file.write("#error \"This header should not be used directly, \
please include util_internal.h instead\"\n")
file.write("#endif /* ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_H_ */\n")
file.write("\n")
file.write("#ifndef ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_IS_EQ_H_\n")
file.write("#define ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_IS_EQ_H_\n")
file.write("\n")
for i in range(0, limit + 1):
file.write(f"#define Z_IS_{i}_EQ_{i}(...) \\,\n")
file.write("\n")
file.write("#endif /* ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_IS_EQ_H_ */\n")
file.write("\n")
write_hidden_stop(file)
def gen_util_internal_util_inc(limit):
with open("util_internal_util_inc.h", "w") as file:
write_hidden_start(file)
file.write("\n")
file.write("#ifndef ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_H_\n")
file.write("#error \"This header should not be used directly, \
please include util_internal.h instead\"\n")
file.write("#endif /* ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_H_ */\n")
file.write("\n")
file.write("#ifndef ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_UTIL_INC_H_\n")
file.write("#define ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_UTIL_INC_H_\n")
file.write("\n")
for i in range(0, limit + 2):
file.write(f"#define Z_UTIL_INC_{i} {i + 1}\n")
file.write("\n")
file.write("#endif /* ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_UTIL_INC_H_ */\n")
file.write("\n")
write_hidden_stop(file)
def gen_util_internal_util_dec(limit):
with open("util_internal_util_dec.h", "w") as file:
write_hidden_start(file)
file.write("\n")
file.write("#ifndef ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_H_\n")
file.write("#error \"This header should not be used directly, \
please include util_internal.h instead\"\n")
file.write("#endif /* ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_H_ */\n")
file.write("\n")
file.write("#ifndef ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_UTIL_DEC_H_\n")
file.write("#define ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_UTIL_DEC_H_\n")
file.write("\n")
file.write(f"#define Z_UTIL_DEC_0 0\n")
for i in range(1, limit + 2):
file.write(f"#define Z_UTIL_DEC_{i} {i - 1}\n")
file.write("\n")
file.write("#endif /* ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_UTIL_DEC_H_ */\n")
file.write("\n")
write_hidden_stop(file)
def gen_util_internal_util_x2(limit):
with open("util_internal_util_x2.h", "w") as file:
write_hidden_start(file)
file.write("\n")
file.write("#ifndef ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_H_\n")
file.write("#error \"This header should not be used directly, \
please include util_internal.h instead\"\n")
file.write("#endif /* ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_H_ */\n")
file.write("\n")
file.write("#ifndef ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_UTIL_X2_H_\n")
file.write("#define ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_UTIL_X2_H_\n")
file.write("\n")
for i in range(0, limit + 1):
file.write(f"#define Z_UTIL_X2_{i} {i *2}\n")
file.write("\n")
file.write("#endif /* ZEPHYR_INCLUDE_SYS_UTIL_INTERNAL_UTIL_X2_H_ */\n")
file.write("\n")
write_hidden_stop(file)
if __name__ == "__main__":
parser = argparse.ArgumentParser(allow_abbrev=False)
parser.add_argument(
"-l", "--limit", type=int, required=True, help="Limit of macros"
)
args = parser.parse_args()
gen_util_listify(args.limit)
gen_util_internal_is_eq(args.limit)
gen_util_internal_util_inc(args.limit)
gen_util_internal_util_dec(args.limit)
gen_util_internal_util_x2(args.limit)