papermario/tools/substitute.py
2021-01-02 16:51:14 +09:00

58 lines
1.8 KiB
Python
Executable File

#!/usr/bin/python3
import argparse
import os
import re
from pathlib import Path
script_dir = os.path.dirname(os.path.realpath(__file__))
root_dir = script_dir + "/../"
src_dir = root_dir + "src/"
asm_dir = root_dir + "asm/"
parser = argparse.ArgumentParser(description="Replace many functions with one")
parser.add_argument("from_list", help="path to line-separated file of functions to be replaced. first line is the string to replace them with")
args = parser.parse_args()
from_funcs = []
with open(args.from_list) as f:
from_text = f.readlines()
to_line = from_text[0].strip()
func_name = to_line[to_line.rfind("/") + 1:to_line.find(".")]
for from_line in from_text[1:]:
if len(from_line.strip()) > 0:
from_funcs.append(from_line.rstrip().split(" ")[-1])
for root, dirs, files in os.walk(src_dir):
for f_name in files:
if f_name.endswith(".c"):
f_path = os.path.join(root, f_name)
with open(f_path) as f:
f_text_orig = f.read()
f_text = f_text_orig
for func in from_funcs:
search_pattern = re.compile("\n.*" + func + "\).*\n")
f_text = re.sub(search_pattern, "\n" + to_line + "\n", f_text)
if f_text != f_text_orig:
with open(f_path, "w", newline="\n") as f:
f.write(f_text)
for root, dirs, files in os.walk(asm_dir):
for f_name in files:
if f_name.endswith(".s"):
f_path = os.path.join(root, f_name)
with open(f_path) as f:
f_text_orig = f.read()
f_text = f_text_orig
for func in from_funcs:
f_text = f_text.replace(func, Path(f_path).parent.parent.name + "_" + func_name)
if f_text != f_text_orig:
with open(f_path, "w", newline="\n") as f:
f.write(f_text)