mirror of
https://github.com/izzy2lost/Diddy-Kong-Racing.git
synced 2026-06-19 01:16:26 -07:00
55 lines
2.5 KiB
Python
55 lines
2.5 KiB
Python
import re
|
|
|
|
from file_util import FileUtil
|
|
|
|
LINE_START_REGEX = r'\n(/\* ([0-9A-F]{6}) [0-9A-F]{8} [0-9A-F]{8} \*/\s+|\.L80[0-9A-F]{6}:)'
|
|
REG_REGEX = r'(\$(at|v[0-1]|a[0-3]|t[0-9]|s[0-7]|f[0-9]{1,2}|k[0-1]|gp|sp|fp|ra|zero))'
|
|
GLABEL_UPPER_REGEX = '(' + LINE_START_REGEX + r'lui\s+' + REG_REGEX + r',\s+(0x80[0-9a-f]{2}))'
|
|
GLABEL_LOWER_REGEX = '(' + LINE_START_REGEX + r'[sl][bhw](u|c1)?\s+' + REG_REGEX + r',\s+(-?0x[0-9a-f]{4})\(\4\))'
|
|
GLABEL_REGEX = GLABEL_UPPER_REGEX + '(' + LINE_START_REGEX + '[^\n]*){0,10}' + GLABEL_LOWER_REGEX
|
|
|
|
UPPER_INSTR_REGEX_TMPL = r'(/\* %s[^\n]*lui\s+)([^\n]*)'
|
|
LOWER_INSTR_REGEX_TMPL = r'(/\* %s[^\n]*[ls][bhw](u|c1)?\s+' + REG_REGEX + r',\s+)-?0x[0-9a-f]{4}([^\n]*)'
|
|
|
|
IGNORE_GLABELS = ['D_800E389E', 'D_801264A1', 'D_800E1B84']
|
|
|
|
def _find_glabels(asm):
|
|
glabels = []
|
|
for file in FileUtil.get_filenames_from_directory_recursive('.', '.s'):
|
|
contents = FileUtil.get_text_from_file(file)
|
|
matches = re.findall(GLABEL_REGEX, contents)
|
|
for match in matches:
|
|
reg = match[3]
|
|
offsets = (match[2], match[11]) # ROM offsets of the lui and [sl][bhw]
|
|
glabel_upper = int(match[5], 16) # upper immediate of glabel
|
|
glabel_lower = int(match[15], 16) # lower immediate of the glabel
|
|
if glabel_lower < 0:
|
|
glabel_lower = (-glabel_lower ^ 0xFFFF) + 1
|
|
if glabel_lower & 0x8000:
|
|
glabel_upper -= 1
|
|
glabel = 'D_%08X' % (glabel_upper << 16 | glabel_lower)
|
|
if glabel not in IGNORE_GLABELS:
|
|
glabels.append((glabel, *offsets))
|
|
# replace upper instruction
|
|
contents = re.sub(UPPER_INSTR_REGEX_TMPL % offsets[0],
|
|
r'\1' + '%s, %%hi(%s) #' % (reg, glabel) + r'\2',
|
|
contents)
|
|
# replace lower instruction
|
|
contents = re.sub(LOWER_INSTR_REGEX_TMPL % offsets[1],
|
|
r'\1' + '%%lo(%s)' % (glabel) + r'\5',
|
|
contents)
|
|
if len(matches):
|
|
FileUtil.write_text_to_file(file, contents)
|
|
return glabels
|
|
|
|
def main():
|
|
FileUtil.set_working_dir_to_project_base()
|
|
contents = FileUtil.get_text_from_file('asm/non_matchings/unknown_005740/func_80005254.s')
|
|
glabels = _find_glabels(contents)
|
|
print('%d undeclared labels found:' % len(glabels))
|
|
for glabel in glabels:
|
|
print(glabel)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|