#!/usr/bin/env python

import sys

import time

import os

from optparse import OptionParser

def get_svn_rev():

    try:

        f = os.popen("svn info", "r")

        lines = f.readlines()

        f.close()

    except:

        return "unk"

    svn_rev = "unk"

    for l in lines:

        if l.startswith('Revision:'):

            t = l.rstrip()

            svn_rev = t.split()[-1]

    return svn_rev

def is_ps3():

    try:

        f = open("/proc/cpuinfo")

        s = f.read()

    except:

        return False

    return s.find('PS3') != -1

def main():

    def make_fname(suffix):

        return basename + '.' + suffix

    max_spes_default = 16

    if is_ps3():

        max_spes_default = 6

    parser = OptionParser()

    parser.add_option("-m", "--max-spes", type="int", default=max_spes_default,

                      metavar="NSPES",

                      help="maximum number of SPEs to use [default=%default]")

    parser.add_option("", "--min-spes", type="int", default=1,

                      metavar="NSPES",

                      help="minimum number of SPEs to use [default=%default]")

    parser.add_option("-p", "--oprofile", action="store_true", default=False,

                      help="emit oprofile commands")

    parser.add_option("-t", "--tag", default=None,

                      help="additional goodie in generated filenames")

    (options, args) = parser.parse_args()

    if len(args) != 0:

        parser.print_help()

        sys.exit(1)

    svn_rev = get_svn_rev()

    os.environ['TZ'] = 'PST8PDT'        # always pacific

    time.tzset()

    tag = ''

    if options.tag:

        tag = '-' + options.tag

    basename = 'R-%s%s-%s' % (svn_rev, tag, time.strftime('%Y%m%d-%H%M'))

    base_njobs = int(500e3)

    njobs = {

         1 : base_njobs,

        10 : base_njobs,

        50 : base_njobs,

       100 : base_njobs, 

       200 : int(base_njobs/2),

       250 : int(base_njobs/2.5),

       300 : int(base_njobs/3),

       400 : int(base_njobs/4),

       500 : int(base_njobs/5)

        }

    f_results = make_fname('results')

    f_opreport = make_fname('opreport')

    f_avg = make_fname('avg')

    f_png = make_fname('png')

    f = sys.stdout

    f.write("#!/bin/bash\n")

    if options.oprofile:

        f.write("opcontrol --stop\n")

        f.write("opcontrol --reset\n")

        f.write("opcontrol --start\n")

    f.write("(\n")

    for udelay in (10, 50, 100, 200, 300):

        for nspes in range(options.min_spes, options.max_spes+1):

            cmd = "./benchmark_nop -n %d -u %d -N %d\n" % (nspes, udelay, njobs[udelay])

            f.write(cmd)

            f.write(cmd)

    f.write(") | tee %s\n" % (f_results,))

    if options.oprofile:

        f.write("opcontrol --dump\n")

        f.write("opcontrol --stop\n")

        f.write("opreport -l | head -100 > %s\n" % (f_opreport,))

    f.write("./split_and_avg_results.py %s > %s\n" % (f_results, f_avg))

    f.write("./plot_speedup.py %s -o %s\n" % (f_avg, f_png))

if __name__ == '__main__':

    main()