papermario/coverage.py

91 lines
3.4 KiB
Python
Raw Normal View History

2020-08-16 07:11:40 +02:00
#!/usr/bin/env python3
from os import path
2020-08-18 02:19:21 +02:00
from os import remove
2020-08-19 02:52:52 +02:00
from sys import argv
2020-08-16 07:11:40 +02:00
import re
from glob import glob
2020-09-18 03:28:34 +02:00
from pathlib import Path
2020-08-16 07:11:40 +02:00
DIR = path.dirname(__file__)
2020-09-18 03:28:34 +02:00
NONMATCHINGS_DIR = Path(path.join(DIR, "asm", "nonmatchings"))
C_FILES = Path(path.join(DIR, "src")).rglob("*.c")
ASM_FILES = NONMATCHINGS_DIR.rglob("*.s")
2020-08-16 07:11:40 +02:00
def strip_c_comments(text):
def replacer(match):
s = match.group(0)
if s.startswith('/'):
return " "
else:
return s
pattern = re.compile(
r'//.*?$|/\*.*?\*/|\'(?:\\.|[^\\\'])*\'|"(?:\\.|[^\\"])*"',
re.DOTALL | re.MULTILINE
)
return re.sub(pattern, replacer, text)
c_func_pattern = re.compile(
r"^(static\s+)?[^\s]+\s+([^\s(]+)\(([^\n)]*)\)\s+{",
2020-08-16 07:11:40 +02:00
re.MULTILINE
)
def funcs_in_c(text):
return (match.group(2) for match in c_func_pattern.finditer(text))
2020-08-19 03:45:26 +02:00
asm_func_pattern = re.compile(
2020-09-26 03:51:54 +02:00
r"INCLUDE_ASM\([^,]+, [^,]+, ([^,)]+)",
2020-08-19 03:45:26 +02:00
re.MULTILINE
)
def include_asms_in_c(text):
return (match.group(1) for match in asm_func_pattern.finditer(text))
2020-08-16 07:11:40 +02:00
matched = []
2020-08-19 03:45:26 +02:00
asm = []
2020-08-16 07:11:40 +02:00
for filename in C_FILES:
with open(filename, "r") as file:
2020-08-19 03:45:26 +02:00
text = strip_c_comments(file.read())
matched.extend((m for m in funcs_in_c(text) if not m in matched))
asm.extend((m for m in include_asms_in_c(text) if not m in asm))
2020-08-16 07:11:40 +02:00
non_matched = [path.splitext(path.basename(filename))[0] for filename in ASM_FILES]
2020-08-19 03:45:26 +02:00
partial_matched = [f for f in matched if f in asm]
matched = [f for f in matched if not f in partial_matched]
matched_but_undeleted_asm = [f for f in matched if f in non_matched and not f in partial_matched]
2020-08-16 07:11:40 +02:00
2020-08-19 02:52:52 +02:00
if __name__ == "__main__":
if "--help" in argv:
print("--fail-matched-undeleted exit with error code 1 if matched function(s) exist in asm/nonmatchings/")
print("--fail-unincluded exit with error code 2 if unincluded assembly files exist")
2020-08-19 02:52:52 +02:00
print("--delete-matched delete matched function(s) from asm/nonmatchings/ without asking")
print("--delete-unincluded delete unincluded, unmatched assembly files")
2020-08-19 02:52:52 +02:00
exit()
2020-08-18 02:19:21 +02:00
2020-08-19 02:52:52 +02:00
total = len(matched) + len(non_matched)
2020-08-19 03:45:26 +02:00
print(f"{len(matched)}+{len(partial_matched)} / {total} functions ({(len(matched) / total) * 100:.2f}% matched)")
2020-08-19 02:52:52 +02:00
if len(matched_but_undeleted_asm) > 0:
print(f"The following functions have been matched but still exist in asm/nonmatchings/: {' '.join(matched_but_undeleted_asm)}")
if "--fail-matched-undeleted" in argv:
exit(1)
2020-08-19 03:47:02 +02:00
elif "--delete-matched" in argv or input("Delete them [y/N]? ").upper() == "Y":
for func in matched_but_undeleted_asm:
2020-09-18 03:28:34 +02:00
f = next(NONMATCHINGS_DIR.rglob(func + ".s"))
remove(f)
2020-08-19 03:45:26 +02:00
elif len(asm) != len(non_matched):
#print(f"warning: number of INCLUDE_ASM macros ({len(asm)}) != number of asm files ({len(non_matched)})")
if len(non_matched) > len(asm):
print(f"The following functions are unmatched but are also unINCLUDEd: {set(non_matched) - set(asm)}")
if "--fail-unincluded" in argv:
exit(2)
elif "--delete-unincluded" in argv or input("Delete them [y/N]? ").upper() == "Y":
for func in set(non_matched) - set(asm):
f = next(NONMATCHINGS_DIR.rglob(func + ".s"))
remove(f)
else:
print(f"warning: The following .s files are INCLUDEd but don't exist: {set(asm) - set(non_matched)}")