164 lines
5.8 KiB
Python
164 lines
5.8 KiB
Python
"""
|
|
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)
|