sm64coopdx/autogen/extract_functions.py

104 lines
2.5 KiB
Python
Raw Normal View History

2022-01-26 03:28:10 +00:00
import os
import re
import sys
2022-03-03 09:04:15 +00:00
replacements = {
'BAD_RETURN(s8)': 'void',
'BAD_RETURN(s16)': 'void',
'BAD_RETURN(s32)': 'void',
'BAD_RETURN(s64)': 'void',
'BAD_RETURN(u8)': 'void',
'BAD_RETURN(u16)': 'void',
'BAD_RETURN(u32)': 'void',
'BAD_RETURN(u64)': 'void',
'BAD_RETURN(f32)': 'void',
'BAD_RETURN(f64)': 'void',
}
def extract_functions(filename):
with open(filename) as file:
lines = file.readlines()
2022-01-26 03:28:10 +00:00
2022-02-23 02:34:51 +00:00
# deal with certain ifdefs
txt = ''
gobbling = False
for line in lines:
if line.strip() == '#ifdef AVOID_UB':
gobbling = True
if line.strip() == '#else':
gobbling = False
if line.strip() == '#endif':
gobbling = False
if not gobbling:
txt += line + '\n'
2022-03-03 09:04:15 +00:00
for replacement in replacements:
txt = txt.replace(replacement, replacements[replacement])
# strip directives and comments
in_directive = False
2022-02-23 02:34:51 +00:00
tmp = txt
txt = ''
2022-02-23 02:34:51 +00:00
for line in tmp.splitlines():
if line.strip().startswith('#') or in_directive:
in_directive = line.strip().endswith('\\')
continue
if '//' in line:
line = line.split('//', 1)[0]
txt += line
2022-01-26 03:28:10 +00:00
while ('/*' in txt):
s1 = txt.split('/*', 1)
s2 = s1[1].split('*/', 1)
txt = s1[0] + s2[-1]
2022-01-26 03:28:10 +00:00
# normalize newlines
txt = txt.replace('\n', ' ')
txt = txt.replace(';', ';\n')
txt = txt.replace('{', '{\n')
while (' ' in txt):
txt = txt.replace(' ', ' ')
2022-01-26 03:28:10 +00:00
# strip macros
txt = re.sub(r'[^a-zA-Z0-9_][A-Z0-9_]+\(.*\)', '', txt)
2022-01-26 03:28:10 +00:00
# strip blocks
tmp = txt
txt = ''
inside = 0
for character in tmp:
if inside == 0:
txt += character
2022-01-26 03:28:10 +00:00
if character == '{':
txt += '\n'
inside += 1
2022-01-26 03:28:10 +00:00
if character == '}':
inside -= 1
2022-01-26 03:28:10 +00:00
# cull obvious non-functions, statics, and externs
tmp = txt
txt = ''
for line in tmp.splitlines():
line = line.strip()
if '(' not in line:
continue
if ')' not in line:
continue
if '=' in line:
continue
#if '{' not in line:
# continue
if line.startswith('static '):
continue
if line.startswith('extern '):
continue
txt += line + '\n'
2022-01-26 03:28:10 +00:00
# normalize function ending
txt = txt.replace(' {', ';')
return txt
2022-02-23 02:34:51 +00:00
if __name__ == "__main__":
print(extract_functions(sys.argv[1]))