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)