import matplotlib.pyplot as plt
from config import markers, colors
from memcached_qps_scan import extract_qps_latency
import os

experiments = [
  "memcached_qps_scan.5.15.79-peter.first4.t8.c20", # vanilla
  "memcached_qps_scan.5.15.79-peter.kernel_polling.first4.t8.c20", # kernel polling
]

experiments_uma = [
  "memcached_qps_scan.5.15.79-peter.t8.c20", # vanilla
  "memcached_qps_scan.5.15.79-peter.kernel_polling.t8.c20", # kernel polling
]

names = [
  "vanilla",
  "kernel polling (patched)",
]

def show_plot():
    plt.cla()

    data = extract_qps_latency(experiments)
    data_uma = extract_qps_latency(experiments_uma)

    for i in range(len(experiments)):
        k = experiments[i]
        display_name = names[i]
        plt.errorbar(data[k][0], data[k][1], yerr = data[k][2], capsize = 4, label = display_name, marker = markers[display_name], color = colors[display_name])
        k = experiments_uma[i]
        plt.plot(data_uma[k][0], data_uma[k][1], linestyle = 'dotted', color = colors[display_name])

    plt.xlabel('QPS')
    plt.ylabel('Latency (99th percentile)')
    plt.yscale('log')
    plt.xlim(xmin = 0)

    plt.legend()
    if os.getenv('SAVE_FIGURE') != 'true':
        plt.rcParams.update({'font.size': 16})
        plt.show()
    else:
        plt.savefig('../data/figs/memcached_qps_scan_numa.png', dpi = 192)

show_plot()