nvidia-rtx-godot / tests / test_builders.py

"""Functions used to generate source files during build time"""

import re

import methods

RE_PATH_SPLIT = re.compile(r"[^/\\]+?(?=\.)")

TEMPLATE = """\
#ifndef _WIN32
#define TEST_DLL_PRIVATE __attribute__((visibility("hidden")))
#else
#define TEST_DLL_PRIVATE
#endif // _WIN32

namespace ForceLink {{
	TEST_DLL_PRIVATE void force_link_tests();
	{TESTS_DECLARE}
}} // namespace ForceLink

void ForceLink::force_link_tests() {{
	{TESTS_CALL}
}}
"""


def force_link_builder(target, source, env):
    names = [RE_PATH_SPLIT.search(str(path)).group() for path in source[0].read()]
    declares = [f"TEST_DLL_PRIVATE void force_link_{name}();" for name in names]
    calls = [f"force_link_{name}();" for name in names]

    with methods.generated_wrapper(str(target[0])) as file:
        file.write(
            TEMPLATE.format(
                TESTS_DECLARE="\n\t".join(declares),
                TESTS_CALL="\n\t".join(calls),
            )
        )