Leahnaya/pmd-sky
1
0
Fork 0
mirror of https://github.com/pret/pmd-sky.git synced 2026-04-25 15:42:34 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix some issue with extraction script

Browse Source
This commit is contained in:
marius david 2023-08-30 13:36:20 +02:00
parent 7cbd2abede
commit e7b6241d85
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tools/extract_function/extract_function.py
View File

@ -59,9 +59,9 @@ def get_line_address(line: str):
for i, line in enumerate(original_lines):
if first_function_start_line is None and line.startswith(ARM_FUNC_START):
first_function_start_line = i
if line == f'{ARM_FUNC_START}{function_name}\n':
if line.strip() == f'{ARM_FUNC_START}{function_name}'.strip():
function_start_line = i
elif line.startswith(f'\tarm_func_end {function_name}'):
elif line.strip() == f'arm_func_end {function_name}'.strip():
function_end_line = i
if function_start_line is not None and extract_function_address is None and ADDRESS_FIND in line:
@ -231,4 +231,4 @@ else:
src_file_path = os.path.join(SRC_FOLDER, f'{extract_file_name}.c')
print('Creating', src_file_path)
with open(src_file_path, 'w') as new_src_file:
new_src_file.write(f'#include "{extract_file_name}.h\n\n{function_body}\n')
new_src_file.write(f'#include "{extract_file_name}.h\"\n\n{function_body}\n')