@@ -29,6 +29,9 @@ def plotter(args):
2929
3030 df = pd .read_csv (args .csv_filepath )
3131 telemetry_modes_y_data = {}
32+ if not args .telemetry_modes :
33+ args .telemetry_modes = df ["Labels" ].unique ()
34+
3235 metric_name = get_metric_name (args )
3336 constructed_query_str = get_constructed_query_str (args )
3437
@@ -41,6 +44,8 @@ def plotter(args):
4144 fig = plt .figure (figsize = (1138 / dpi , 871 / dpi ), dpi = dpi )
4245 ax = fig .add_subplot (111 )
4346 ax .set_ylim (0 , 1.0 )
47+ if args .title :
48+ ax .set_title (args .title )
4449 for key , val in telemetry_modes_y_data .items ():
4550 plot_key = key
4651 match key :
@@ -107,15 +112,15 @@ def get_data_helper(df, query_list, query_str, telemetry_mode, metric_name):
107112 try :
108113 data [metric_name ].head ().empty
109114 except KeyError as e :
110- y_series_data .append (None )
115+ y_series_data .append (0 )
111116 else :
112117 if not data [metric_name ].head ().empty :
113118 if metric_name .startswith ('cpu' ) or metric_name .startswith ('mem' ):
114119 y_series_data .append (data [metric_name ].head (1 ).values [0 ])
115120 else :
116- y_series_data .append (data [metric_name ].head (1 ).values [0 ] / data [ "ActualQPS" ]. head ( 1 ). values [ 0 ] )
121+ y_series_data .append (data [metric_name ].head (1 ).values [0 ] / 1000 )
117122 else :
118- y_series_data .append (None )
123+ y_series_data .append (0 )
119124
120125 return y_series_data
121126
@@ -174,15 +179,20 @@ def get_parser():
174179 parser .add_argument (
175180 "--query_str" ,
176181 help = "Specify the qps or conn query_str that will be used to query your y-axis data based on the CSV file."
177- "For example: conn_query_str=ActualQPS==1000, qps_query_str=NumThreads==16."
182+ "For example: conn_query_str=ActualQPS==1000, qps_query_str=NumThreads==16." ,
183+ default = ""
178184 )
179185 parser .add_argument (
180186 "--csv_filepath" ,
181187 help = "The path of the CSV file."
182188 )
183189 parser .add_argument (
184190 "--graph_title" ,
185- help = "The graph title."
191+ help = "Output path."
192+ )
193+ parser .add_argument (
194+ "--title" ,
195+ help = "Visual title of graph."
186196 )
187197 return parser
188198
0 commit comments