ze_api_generator.py

Go to the documentation of this file.

import re
import sys
 
 
def camel_to_snake(src):
    return re.sub(r"(?<!^)(?=[A-Z])", "_", src).lower()
 
 
def snake_to_camel(src):
    temp = src.split("_")
    return "".join(x.title() for x in temp)
 
 
def extract_ze_apis(header):
    """
    Emit file with contents of
    _ZE_API(api_name, api_domain, cb, param_type)
    """
    api = open("ze_api.def", "w")
 
    matches = re.finditer(
        r"typedef struct _ze_([_a-z]+)_callbacks_t\n\{\n([a-zA-Z_;\s\n]+)\n\} ze_([_a-z]+)_callbacks_t;",
        header,
    )
 
    for match in matches:
        api_domain = snake_to_camel(match.group(1))
        for l in match.group(2).splitlines():
            parts = l.split()
            api_match = re.match(r"ze_pfn([a-zA-Z]+)Cb_t", parts[0])
            api_name_tail = api_match.group(1)
            api_name = "ze" + api_name_tail
 
            param_type = "ze_" + camel_to_snake(api_name_tail) + "_params_t"
 
            cb = "pfn" + api_name_tail.replace(api_domain, "") + "Cb"
 
            api.write(
                "_ZE_API({}, {}, {}, {})\n".format(api_name, api_domain, cb, param_type)
            )
 
    api.close()
 
 
if __name__ == "__main__":
    with open(sys.argv[1], "r") as f:
        header = f.read()
        extract_ze_apis(header)