"""
Make charts from ReJSONBenchmark's output
"""

import matplotlib.pyplot as plt
import numpy as np

data = np.genfromtxt('benchmark.csv', delimiter=',', names=True, dtype=None)

# Each JSON value has 4 operations: set root, get root, set path, get path
NOP = 4

# The number of JSON values
N = int(len(data) / NOP)

# The x locations
ind = np.arange(N)

# The bars' width
width = (1 - .3) / NOP

colors = ['r','g','b','y']

fig, ax = plt.subplots()

# Iterate each operation
for oidx in range(NOP):
    d = data[oidx::NOP]
    # ax1 = ax[oidx]

    # plt.subplot(NOP,1,oidx+1)

    # Plot the rate as bars
    plt.bar(ind+oidx*width, d['rate'], width, align='center', color=colors[oidx])

    # Plot the latency as line
    # tax = plt.twinx()
    # tax.set_yscale('log')
    # tax.plot(ind, d['avgLatency'], color='r')

    # plt.title('{} rate and average latency'.format(d['title'][0]))
    # ax1.set_ylabel('Rate (op/s)', color='b')
    # ax1.set_xlabel('Object size (bytes)')
    # for t in ax1.get_yticklabels():
    #     t.set_color('b')
    # ax2.set_ylabel('Average latency (msec)', color='r')
    # for t in ax2.get_yticklabels():
    #     t.set_color('r')
    # plt.xticks(ind, d['size']) 

plt.grid(True)
plt.xticks(ind + width, d['size'])
plt.show()