Script diffing two sets of GBA basic block traces

covdiff.py

import sys

def getvals(path):
    lines = open(path, "r").readlines()
    lst = []

    for line in lines:
        line = line.strip()

        if len(line) >= 10:
            addr, cnt = line.split(" ")
            lst.append((int(addr, 16), int(cnt)))

    resmap = {}
    resset = set()

    for addr, cnt in lst:
        resmap[addr] = cnt
        resset.add(addr)
    
    return resset, resmap

if __name__ == '__main__':
    if len(sys.argv) < 3:
        print("covdiff.py <before.cov...> <after.cov>")
    else:
        diffset, diffmap = getvals(sys.argv[-1])

        for e in sys.argv[1:-1]:
            eset, emap = getvals(e)
            diffset = diffset - eset
        
        for e in diffset:
            print("0x{:08x} {}".format(e, diffmap[e]))