pypy-effect-analysis / pypy / jit / tool / loopcounter.py

#!/usr/bin/env python
"""
Parse and display the traces produced by pypy-c-jit when PYPYLOG is set.
"""

import autopath
import py
import sys
import optparse
import re

def get_timestamp(line):
    match = re.match(r'\[([0-9a-f]*)\] .*', line)
    return int(match.group(1), 16)

def count_loops_and_bridges(log):
    loops = 0
    bridges = 0
    time0 = None
    lines = iter(log)
    for line in lines:
        if time0 is None and line.startswith('['):
            time0 = get_timestamp(line)
        if '{jit-mem-looptoken-' in line:
            time_now = get_timestamp(line) - time0
            text = lines.next()
            if text.startswith('allocating Loop #'):
                loops += 1
            elif text.startswith('allocating Bridge #'):
                bridges += 1
            elif text.startswith('freeing Loop #'):
                match = re.match('freeing Loop # .* with ([0-9]*) attached bridges\n', text)
                loops -=1
                bridges -= int(match.group(1))
            else:
                assert False, 'unknown line' % line
            total = loops+bridges
            yield (time_now, total, loops, bridges)

def main(logfile, options):
    print 'timestamp,total,loops,bridges'
    log = open(logfile)
    for timestamp, total, loops, bridges in count_loops_and_bridges(log):
        print '%d,%d,%d,%d' % (timestamp, total, loops, bridges)        

if __name__ == '__main__':
    parser = optparse.OptionParser(usage="%prog loopfile [options]")
    options, args = parser.parse_args()
    if len(args) != 1:
        parser.print_help()
        sys.exit(2)
    main(args[0], options)
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.