CreateStatusGraph.py 文件源码

def main():
    if len(sys.argv) != 2:
        print "python %s <filename>" % sys.argv[0]
        sys.exit(1)

    try:
        filename = sys.argv[1]
        with open(filename, 'rb') as f:
            saved_state = pickle.load(f)

        bitmap = saved_state['bitmap']

    except:
        print "[!] Could not load bitmap"
        sys.exit(1)

    # Get rough code coverage value
    coverage = get_coverage(bitmap)
    print "[*] Code coverage (basic block calls): %.2f" % coverage

    if HAS_PYPNG:       
        # Create PNG image from bitmap values
        p = populate_array(bitmap)
        img = png.from_array(p, 'RGB')
        img.save('status_graph.png')

        print "[*] Created PNG file"