#
#    Copyright (c) 2007-2009 Corey Goldberg (corey@goldb.org)
#    License: GNU GPLv3
#
#    This file is part of Pylot.
#    
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation, either version 3 of the License, or
#    (at your option) any later version.  See the GNU General Public License 
#    for more details.
#


import sys

try:
    from pylab import *  # Matplotlib for graphing.  Only used on systems that have it installed.
except ImportError, e:
    sys.stderr.write('Matplotlib ImportError: %s\n' % e)
    


# response time graph
def resp_graph(nested_resp_list, dir='./'):
    fig = figure(figsize=(8, 3))  # image dimensions  
    ax = fig.add_subplot(111)
    ax.set_xlabel('Elapsed Time In Test (secs)', size='x-small')
    ax.set_ylabel('Response Time (secs)' , size='x-small')
    ax.grid(True, color='#666666')
    xticks(size='x-small')
    yticks(size='x-small')
    axis(xmin=0)
    x_seq = [item[0] for item in nested_resp_list] 
    y_seq = [item[1] for item in nested_resp_list] 
    ax.plot(x_seq, y_seq, 
        color='blue', linestyle='-', linewidth=1.0, marker='o', 
        markeredgecolor='blue', markerfacecolor='yellow', markersize=2.0)
    savefig(dir + 'response_time_graph.png') 
    
    

# throughput graph
def tp_graph(throughputs_dict, dir='./'):
    fig = figure(figsize=(8, 3))  # image dimensions  
    ax = fig.add_subplot(111)
    ax.set_xlabel('Elapsed Time In Test (secs)', size='x-small')
    ax.set_ylabel('Requests Per Second (count)' , size='x-small')
    ax.grid(True, color='#666666')
    xticks(size='x-small')
    yticks(size='x-small')
    axis(xmin=0)
    keys = throughputs_dict.keys()
    keys.sort()
    values = []
    for key in keys:
        values.append(throughputs_dict[key])
    x_seq = keys
    y_seq = values
    ax.plot(x_seq, y_seq, 
        color='red', linestyle='-', linewidth=1.0, marker='o', 
        markeredgecolor='red', markerfacecolor='yellow', markersize=2.0)
    savefig(dir + 'throughput_graph.png')