#!/usr/bin/env python3

import subprocess

# Plot the dat files generated by parselogs.py using gnuplot

if __name__ == '__main__':

    # The analytical functions come from analytical.py
    # Replace 'R' in the output of that program by 'x' and replace
    # 'logR' by 'ceil(log(x)/log(2))'.

    relay_perclient_analyticals = {
        'singlepass_merkle': '0.020524244*x + 1459.6*ceil(log(x)/log(2)) + 43404.140896',
        'singlepass_threshsig': '0.9048*x + 45628.912096',
        'telescoping_merkle': '0.017528004*x + 1459.6*ceil(log(x)/log(2)) + 36977.500696',
        'telescoping_threshsig': '0.7982*x + 39135.071896',
        'vanilla_none': '38.551644414*x + 25316.281696',
    }

    relay_analyticals = {
        k: '(2500000/6500)*(%s)' % v for k, v in relay_perclient_analyticals.items()
    }

    client_analyticals = {
        'singlepass_merkle': '729.8*ceil(log(x)/log(2)) + 16211.1',
        'singlepass_threshsig': '17261.3000000000',
        'telescoping_merkle': '729.8*ceil(log(x)/log(2)) + 15090.1',
        'telescoping_threshsig': '16126.5000000000',
        'vanilla_none': '38.53524*x + 8735.66',
    }

    plots = [
        ('dirauth', 'Directory authorities total bytes each', 2, False, None),
        ('relay', 'Relay total bytes each', 4, False, None),
        ('relay_bf', 'Bootstrapping fallback relays bytes per bw', 6, True, None),
        ('relay_f', 'Non-bootstrapping fallback relays bytes per bw', 8, True, None),
        ('relay_b', 'Bootstrapping normal relays bytes per bw', 10, True, None),
        ('relay_n', 'Non-bootstrapping normal relays bytes per bw', 12, True, None),
        ('client', 'Client total bytes each', 14, False, None),
        ('client_b', 'Bootstrapping client total bytes', 16, True, None),
        ('client_n', 'Non-bootstrapping client total bytes', 18, True, None),
        ('dirauth_ss', 'Directory authority total bytes each', 20, True, None),
        ('relay_ss', 'Relay total bytes each', 22, True, relay_analyticals),
        ('client_ss', 'Client total bytes each', 24, True, client_analyticals),
        ('relay_perclient_ss', 'Relay total bytes per client', 26, True, relay_perclient_analyticals),
        ('relay_ss_wide', 'Relay total bytes each', 22, True, relay_analyticals),
        ('client_ss_wide', 'Client total bytes each', 24, True, client_analyticals),
        ('relay_perclient_ss_wide', 'Relay total bytes per client', 26, True, relay_perclient_analyticals),
    ]

    dats = [
        ('vanilla_none', 'Vanilla', 1),
        ('singlepass_merkle', 'Sing(M)', 2),
        ('telescoping_merkle', 'Tele(M)', 3),
        ('singlepass_threshsig', 'Sing(T)', 4),
        ('telescoping_threshsig', 'Tele(T)', 5),
    ]

    for filename, title, col, errbars, analyticals in plots:
        if analyticals is None:
            analyticals = dict()
        if filename == 'relay_ss_wide':
            ranges = "set xrange [300:650000]\nset logscale xy\nset yrange [10000000:]"
        elif filename[-5:] == '_wide':
            ranges = "set xrange [300:650000]\nset logscale xy\nset yrange [10000:]"
        else:
            ranges = "set xrange [0:2200]\nset yrange [0:]"
        gpcode = """set terminal pdf font "DejaVuSans,14" size 5,2.25
set output '%s.pdf'
set title '%s'
%s
set key out
set rmargin 17
set arrow from 6500, graph 0 to 6500, graph 1 nohead lc 0 lw 2
set xlabel "Number of relays"
set style line 1 lw 2 lc 1 pt 1
set style line 2 lw 2 lc 2 pt 1
set style line 3 lw 2 lc 3 pt 1
set style line 4 lw 2 lc 4 pt 1
set style line 5 lw 2 lc 5 pt 1
set style line 10 lw 2 lc 0 dt 2
set style line 11 lw 2 lc 1 dt 2
set style line 12 lw 2 lc 2 dt 2
set style line 13 lw 2 lc 3 dt 2
set style line 14 lw 2 lc 4 dt 2
set style line 15 lw 2 lc 5 dt 2
plot """ % (filename, title, ranges)
        firstplot = True
        for datname, title, style in dats:
            if firstplot is False:
                gpcode += ", "
            else:
                firstplot = False
            gpcode += "'%s.dat' using 1:%d with lines ls %d title '%s'" % \
                (datname, col, style, title)
            if errbars:
                gpcode += ", '%s.dat' using 1:%d:%d with errorbars ls %d notitle" % \
                    (datname, col, col+1, style)
            if datname in analyticals:
                gpcode += ", %s ls %d notitle" % \
                    (analyticals[datname], style+10)
        if analyticals:
            gpcode += ", -100 ls 10 title 'Analytical'"

        gp = subprocess.Popen(['gnuplot', '-'], stdin=subprocess.PIPE) 
        gp.communicate(gpcode.encode('ascii'))