mirror of
https://github.com/open-goal/jak-project.git
synced 2024-10-20 11:26:18 -04:00
31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
|
import os
|
||
|
|
||
|
new_file = []
|
||
|
with open("./decompiler/config/all-types.gc") as f:
|
||
|
symbols_found = []
|
||
|
lines = f.readlines()
|
||
|
|
||
|
for line_num, line in enumerate(lines):
|
||
|
if "flag-assert" in line:
|
||
|
heap_base_amount = ""
|
||
|
flag_assert = line.split("#x")[1].strip()
|
||
|
heap_base_amount = flag_assert[len(flag_assert)-7:len(flag_assert)-4]
|
||
|
if len(flag_assert) > 1 and heap_base_amount != "000":
|
||
|
# First let's check to see if the heap-base already exists around this line
|
||
|
found_heapbase = False
|
||
|
for i in range(line_num - 4, line_num + 3):
|
||
|
temp_line = lines[i]
|
||
|
if "heap-base" in temp_line:
|
||
|
found_heapbase = True
|
||
|
break
|
||
|
if found_heapbase == False:
|
||
|
if line.startswith(";"):
|
||
|
new_file.append("; :heap-base #x{}\n".format(heap_base_amount.lstrip("0")))
|
||
|
else:
|
||
|
new_file.append(" :heap-base #x{}\n".format(heap_base_amount.lstrip("0")))
|
||
|
new_file.append(line)
|
||
|
|
||
|
os.remove("./decompiler/config/all-types.gc")
|
||
|
with open("./decompiler/config/all-types.gc", "w") as f:
|
||
|
f.writelines(new_file)
|