#!/usr/bin/python

# http://test.kernel.org/perf/kernbench.elm3b6.png

import cgi, cgitb, os, sys, re, subprocess
cgitb.enable()
Popen = subprocess.Popen

import common
from autotest_lib.tko import db, display, frontend, plotgraph
from autotest_lib.client.common_lib import kernel_versions

db = db.db()

def main():
    form = cgi.FieldStorage()
    machine_idx = form["machine"].value
    benchmark = form["benchmark"].value
    key = form["key"].value

    machine = frontend.machine.select(db, {'machine_idx' : machine_idx})[0]

    data = {}
    where = { 'subdir' : benchmark, 'machine_idx' : machine.idx }
    for test in frontend.test.select(db, where):
        iterations = test.iterations()
        if iterations.has_key(key):
            data[test.kernel().printable] = iterations[key]

    # for kernel in sort_kernels(data.keys()):
    #    print "%s %s" % (kernel, str(data[kernel]))
    title = "%s on %s" % (benchmark, machine.hostname)
    graph = plotgraph.gnuplot(title, 'Kernel', key, xsort = sort_kernels)
    graph.add_dataset('all kernels', data)
    graph.plot(cgi_header = True)


def sort_kernels(kernels):
    return sorted(kernels, key = kernel_versions.version_encode)

main()