plt_cpu_agents_line.py

#!/usr/bin/env python3

'''Plot and save WIMA vs Baseline cpu line graph (agents included).'''

import sys

import matplotlib.pyplot as plt
import matplotlib.ticker as ticker
import matplotlib.lines as mlines
import requests


NUM_TICKS = 12
NUM_SUBTICKS = NUM_TICKS * 5
MAJ_ALPHA = 1.5
MIN_ALPHA = 0.2

X_LABEL = 'Time (s)'
Y_LABEL = 'CPU usage (%)'

DATA_TYPE = 'cpu-agents'
GRAPH_TYPE = 'line'
FILE_EXTENSION = '.png'

URL = 'http://10.7.229.175:19999/api/v1/data?chart=cgroup_container_name.cpu_limit&format=csv'

def format_axis(axis):
    '''Uses axis to configure the graph.'''
    locator = ticker.MaxNLocator
    axis.xaxis.set_ticks_position('both')
    axis.xaxis.set_major_locator(locator(NUM_TICKS))
    axis.xaxis.set_minor_locator(locator(NUM_SUBTICKS))
    axis.xaxis.grid(which='major', linestyle='-', alpha=MAJ_ALPHA)
    axis.xaxis.grid(which='minor', linestyle='-', alpha=MIN_ALPHA)
    axis.yaxis.grid(which='major', linestyle='-', alpha=MAJ_ALPHA)
    axis.yaxis.grid(which='minor', linestyle='-', alpha=MIN_ALPHA)
    axis.set_axisbelow(True)
    for iter_axis in (axis, axis.twinx()):
        iter_axis.yaxis.set_major_locator(locator(NUM_TICKS))
        iter_axis.yaxis.set_minor_locator(locator(NUM_SUBTICKS))
        iter_axis.set_ylim(axis.get_ylim())


def format_plot():
    '''Uses pyplot to configure the graph.'''
    plt.xlabel(X_LABEL, fontweight='bold')
    plt.ylabel(Y_LABEL, fontweight='bold')

    # Legend configuration
    black_line = mlines.Line2D([], [], color='black',
                               marker='', label='WIMA')

    red_line = mlines.Line2D([], [], color='r',
                             marker='', label='Baseline')

    leg = plt.legend(handles=[black_line, red_line],
                     loc='upper left', fancybox=False, framealpha=1,
                     shadow=False, borderpad=1)

    leg.get_frame().set_edgecolor('black')


def get_filename(conn):
    '''Return filename of the figure.'''
    filename = DATA_TYPE + '-'
    filename += GRAPH_TYPE + '-'
    filename += str(conn) + FILE_EXTENSION
    return filename


def open_file(filename):
    '''Return file content given filename.'''
    with open(filename) as file_:
        return file_.read()


def plot_graphs(graphs_dir):
    '''Plot the graphs and save the figures in graphs_dir.'''
    beg = 0
    end = 0
    inc = 0
    data = []
    wima = 1
    _, axis = plt.subplots()

    if graphs_dir[-1] != '/':
        graphs_dir += '/'

    summ = open_file('summary.csv').split(',')
    beg = int(summ[4])
    end = int(summ[5])
    inc = int(summ[6])
    conn = beg

    for line in open_file('timestamps.txt').splitlines():
        url = URL.replace('container_name', 'wima_master')
        resp = requests.get(url + line, timeout=15).text
        data = resp.splitlines()[1:]
        data = [i.split(',')[1] for i in data]
        data = [float(i) for i in data]
        if wima:
            axis.plot(data, '-', color='black')
            wima = 0
        else:
            containers = (
                'wima_agent_ryu',
                'wima_agent_onos',
                'wima_agent_odl')
            baseline_data = [data]
            for container in containers:
                url = URL.replace('container_name', container)
                resp = requests.get(url + line, timeout=15).text
                agent_data = resp.splitlines()[1:]
                agent_data = [i.split(',')[1] for i in agent_data]
                agent_data = [float(i) for i in agent_data]
                baseline_data.append(agent_data)
            data = [sum(i) for i in zip(*baseline_data)]
            axis.plot(data, '-', color='r')
            format_plot()
            format_axis(axis)
            #plt.show()
            plt.savefig(graphs_dir + get_filename(conn), dpi=600)
            _, axis = plt.subplots()
            wima = 1
            conn += inc
            if conn > end:
                break


if __name__ == '__main__':
    plot_graphs(sys.argv[1])