117 lines
4.1 KiB
Python
Executable File
117 lines
4.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import find_clang
|
|
import sys
|
|
import clang.cindex
|
|
import time
|
|
import os
|
|
|
|
|
|
structs = []
|
|
extrastructs = []
|
|
|
|
def valid_spelling(spelling):
|
|
return spelling and not spelling.startswith("(")
|
|
|
|
def build_struct(cursor, anonymousUnion=False):
|
|
if not anonymousUnion:
|
|
structs.append(cursor.spelling)
|
|
print("template <class A, class B>\nvoid copy%s(A * dest, B * src)\n{" % cursor.spelling)
|
|
|
|
for c in cursor.get_children():
|
|
if c.kind == clang.cindex.CursorKind.UNION_DECL:
|
|
if valid_spelling(c.spelling):
|
|
raise Exception("Cannot handle non anonymous unions")
|
|
|
|
copied_union_member = False
|
|
for uc in c.get_children():
|
|
if not valid_spelling(uc.spelling) or uc.kind == clang.cindex.CursorKind.PACKED_ATTR:
|
|
# Ignore
|
|
pass
|
|
else:
|
|
# per default we copy only the first member of a union and warn if there are more
|
|
# members (declare the other members NOBACKUP)
|
|
if copied_union_member:
|
|
print("Warning more than one union member (%s) in anynomous union inside struct %s, consider NOBACKUP statements" % (uc.spelling, cursor.spelling), file=sys.stderr)
|
|
else:
|
|
copy_decl(uc, uc.spelling)
|
|
copied_union_member = True
|
|
|
|
elif c.kind == clang.cindex.CursorKind.FIELD_DECL:
|
|
copy_decl(c, c.spelling)
|
|
|
|
if not anonymousUnion:
|
|
print("}\n")
|
|
|
|
|
|
def build(cursor):
|
|
result = []
|
|
for c in cursor.get_children():
|
|
if c.location.file.name == sys.argv[1]:
|
|
if c.kind == clang.cindex.CursorKind.STRUCT_DECL:
|
|
build_struct(c)
|
|
for c, spelling in extrastructs:
|
|
print("template <class A, class B>\nvoid copy%s(A * dest, B * src)\n{" % spelling)
|
|
build_struct(c, True)
|
|
print("}\n")
|
|
|
|
return result
|
|
|
|
|
|
def copy_decl(c, spelling):
|
|
children = [ch for ch in c.get_children()]
|
|
if c.type.get_array_size() > 0:
|
|
if c.type.get_array_element_type().spelling in structs:
|
|
print(" for (int i=0; i<%d; i++) {" % c.type.get_array_size())
|
|
print(" copy%s(&dest->%s[i], &src->%s[i]);" % (c.type.get_array_element_type().spelling, spelling, spelling))
|
|
print(" }")
|
|
else:
|
|
print(" memcpy(dest->%s, src->%s, sizeof(dest->%s));" % (spelling, spelling, spelling))
|
|
elif len(children) == 1 and children[0].kind == clang.cindex.CursorKind.STRUCT_DECL and not valid_spelling(children[0].spelling):
|
|
# inline declared structs
|
|
if valid_spelling(c.semantic_parent.spelling):
|
|
spelling_func = c.semantic_parent.spelling + "_" + spelling
|
|
else:
|
|
spelling_func = c.semantic_parent.semantic_parent.spelling + "_" + spelling
|
|
|
|
extrastructs.append((children[0], spelling_func))
|
|
print(" copy%s(&dest->%s, &src->%s);" % (spelling_func, spelling, spelling))
|
|
elif c.type.get_declaration().spelling in structs:
|
|
print(" copy%s(&dest->%s, &src->%s);" % (c.type.get_declaration().spelling, spelling, spelling))
|
|
else:
|
|
print(" dest->%s = src->%s;" % (spelling, spelling))
|
|
|
|
|
|
def header():
|
|
print("// This file was auto-generated by %s script on %s. Do not edit this file!\n\n\n" % (os.path.basename(sys.argv[0]), time.asctime()))
|
|
|
|
|
|
def print_translation_unit_diags(diags, prefix=''):
|
|
for diag in diags:
|
|
print(prefix + str(diag), file=sys.stderr)
|
|
print_translation_unit_diags(diag.children, ' ' + prefix)
|
|
|
|
|
|
def main():
|
|
|
|
if not find_clang.initLibClang():
|
|
sys.exit(-1)
|
|
|
|
index = find_clang.index
|
|
args = ['-x', 'c++', '-std=c++11'] + sys.argv[2:]
|
|
if find_clang.builtin_hdr_path:
|
|
args.append("-I" + find_clang.builtin_hdr_path)
|
|
|
|
translation_unit = index.parse(sys.argv[1], args)
|
|
|
|
if translation_unit.diagnostics:
|
|
print_translation_unit_diags(translation_unit.diagnostics)
|
|
sys.exit(-1)
|
|
|
|
header()
|
|
build(translation_unit.cursor)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|