masters-thesis/op-summarizer/op-summarizer.py

59 lines
1.8 KiB
Python

from dataclasses import dataclass
from typing import Dict, List, Optional
from gcovreader import GCovFile, GCovLine
from opfinderreader import OperationLogReader, OperationLog
@dataclass(frozen=True)
class UniqueOperation:
name: str
type_lhs: Optional[str]
type_rhs: Optional[str]
type_expr: Optional[str]
@staticmethod
def from_operation_log(op_log: OperationLog) -> "UniqueOperation":
return UniqueOperation(
op_log.operation,
op_log.operand_lhs,
op_log.operand_rhs,
op_log.operand_result
)
if __name__ == "__main__":
gcov = GCovFile("./data/gcov.json")
ops = OperationLogReader("./data/opfinder.json")
gcov.read()
ops.read()
for file_name in gcov.files:
op_counter: Dict[UniqueOperation, int] = {}
if file_name not in ops.files:
print(f"Couldn't find {file_name} in op-finder output. Skipping.")
continue
loop_stack: List[int] = []
for gcov_line in gcov.files[file_name]:
op_lines = ops.get_lines(file_name, gcov_line.line_number)
for opfinder_line in op_lines:
# TODO: revise this. Need a special case for for-loop clauses
# or branching in general.
if opfinder_line.is_fallthrough != gcov_line.is_fallthrough:
continue
unique_op = UniqueOperation.from_operation_log(opfinder_line)
if unique_op in op_counter:
op_counter[unique_op] += gcov_line.count
else:
op_counter[unique_op] = gcov_line.count
print(f"Unique operations for file {file_name}:")
for uop, count in op_counter.items():
print(f"\t{count}: {uop}")