easydiskbench/plot.py at master · AlexZzz/easydiskbench · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
#!/usr/bin/env python3
import argparse
import sys
import os
import plotly.graph_objects as go
import pandas as pd

def work(args):
    fig = go.Figure()
    header = ["time","value","op","bs"]

    for input_file in args.input:
        try:
            df = pd.read_csv(input_file,names=header)
        except FileNotFoundError:
            print("No such file: {}. Skipping it...".format(input_file))
            continue
        plot_name = os.path.basename(os.path.dirname(input_file)) \
                + ": " + (os.path.basename(input_file)).split(".", 1)[0]
        df['time_round'] = round(df['time']/args.interval)*args.interval+args.interval
        df['value'] /= args.value_divider

        if not args.msec:
            df['time_round'] /= 1e3

        if args.per_second and (args.median or args.mean):
            df = df.groupby('time_round').sum().reset_index()
            df['value'] /= args.interval/1e3

        if not args.median and not args.mean:
            fig.add_trace(go.Box(x=df['time_round'], y=df['value'],
                                name=plot_name, marker=dict(size=1)))
        if args.median:
            median_df = df.groupby('time_round').median().reset_index()
            fig.add_trace(go.Scatter(x=median_df['time_round'], y=median_df['value'],
                                mode="lines+markers",name=plot_name+" median"))
        if args.mean:
            mean_df = df.groupby('time_round').mean().reset_index()
            fig.add_trace(go.Scatter(x=mean_df['time_round'], y=mean_df['value'],
                                mode="lines+markers",name=plot_name+" mean"))

    fig.update_layout(title=args.title,
                    xaxis_title=args.xlabel,
                    yaxis_title=args.ylabel,
                    showlegend=True,
                    )
    if (args.top_limit):
        fig.update_yaxes(range=[0,args.top_limit])
    fig.update_yaxes(rangemode="tozero")

    if args.output:
        if os.path.splitext(args.output)[1] == ".html":
            fig.write_html(args.output, include_plotlyjs=args.plotlyjs,
                            full_html=args.no_full_html)
        else:
            fig.update_layout(
                    legend=dict(
                        font=dict(
                            size=24
                        )))
            fig.write_image(args.output,width="1920",height="1080")
    else:
        fig.show()

def main():
    parser = argparse.ArgumentParser(description="Plot stuff")
    parser.add_argument('--input','-i',type=str,nargs='*',help="Input files")
    parser.add_argument('--output','-o',type=str,help="Write char to this output file")
    parser.add_argument('--interval',type=int,help="Plot boxplot on interval (msec)",default=30000)
    parser.add_argument('--title','-t',type=str,help="Plot title",default="FIO results")
    parser.add_argument('--per-second',action='store_true',help="Count per-second average value for the given interval (eg IO/s graph)")
    parser.add_argument('--median',action='store_true',help="Plot median values")
    parser.add_argument('--mean',action='store_true',help="Plot mean values")
    parser.add_argument('--msec',action='store_true',help="Show time in milliseconds instead of seconds")
    parser.add_argument('--ylabel',type=str,help="Set this Y-label",default="latency (msec)")
    parser.add_argument('--xlabel',type=str,help="Set this X-label",default="time (s)")
    parser.add_argument('--value-divider',type=int,help="Divide values on this value. Default is for nsec->msec convertion. Set to 1 for IOPS plot",default=1e6)
    parser.add_argument('--top-limit',type=float,help="Set Y axis top limit")
    parser.add_argument('--plotlyjs',type=str,help="Passes this argument to include_plotlyjs when output is set to HTML. Refer to plotly documentation",default=True)
    parser.add_argument('--no-full-html',action='store_false',help="Passes full_html=False to plotly when output is set to HTML. Refer to plotly documentation")
    args = parser.parse_args()
    work(args)

if __name__=='__main__':
    sys.exit(main())