#!/usr/bin/env python

# WARNING WARNING WARNING
# This is a very fragile script.
# Don't change read_test2.cu and expect this to continue to work!

import sys, re

# --- Constants from read_test2 ---

# Memory access patterns in the order they are performed
access_patterns = ['random', 'broadcast', 'linear']

calls = 10
blocks = 500
threads = 256 # per block
iterations = 1000 # per thread
reads = 10 # per iteration

memory_read = calls * blocks * threads * iterations * reads * 4 # bytes

# --- Parse the profile log ---
access_times = []
counts = { }
inside_chunk = False
time_regex = re.compile(r'(?P<method>[a-zA-Z]+)_speedtest.+gputime=\[ (?P<time>[0-9.]+) \]')
for line in open(sys.argv[1]):
    if 'memcopy' in line:
        if inside_chunk == True:
            # Finished all tests for one access pattern
            access_times.append(counts)
            counts = {}
        inside_chunk = False
    elif 'speedtest' in line:
        # Extract gputime and access method from line
        match = time_regex.search(line)
        assert match
        method = match.group('method')
        time = match.group('time')
        counts[method] = counts.get(method, 0.0) + float(time)
        inside_chunk = True

# Take care of final chunk of runs
if inside_chunk:
    access_times.append(counts)
    
# --- Generate the output table ---

memory_types = access_times[0].keys()
memory_types.sort()
col_width = 10
table_width = (col_width + 2)*(len(access_patterns)+1) + 1

# Header
print '(All values in GB/sec)'
print '-' * table_width
print '|%*s' % (col_width, 'memtype'),
for access_pattern in access_patterns:
    print '|%*s' % (col_width, access_pattern),
print '|'
print '-' * table_width

# Body
for memory_type in memory_types:
    print '|%*s' % (col_width, memory_type),
    for access_time in access_times:
        # GB/sec
        print '|%*.1f' % (col_width,
                          memory_read / 1e9 / (access_time[memory_type]*1e-6)),
    print '|'
    
# Footer
print '-' * table_width
print 'Total Read: %d GB' % (memory_read/1e9)