@@ -29,6 +29,9 @@ def plotter(args):
29
29
30
30
df = pd .read_csv (args .csv_filepath )
31
31
telemetry_modes_y_data = {}
32
+ if not args .telemetry_modes :
33
+ args .telemetry_modes = df ["Labels" ].unique ()
34
+
32
35
metric_name = get_metric_name (args )
33
36
constructed_query_str = get_constructed_query_str (args )
34
37
@@ -41,6 +44,8 @@ def plotter(args):
41
44
fig = plt .figure (figsize = (1138 / dpi , 871 / dpi ), dpi = dpi )
42
45
ax = fig .add_subplot (111 )
43
46
ax .set_ylim (0 , 1.0 )
47
+ if args .title :
48
+ ax .set_title (args .title )
44
49
for key , val in telemetry_modes_y_data .items ():
45
50
plot_key = key
46
51
match key :
@@ -107,15 +112,15 @@ def get_data_helper(df, query_list, query_str, telemetry_mode, metric_name):
107
112
try :
108
113
data [metric_name ].head ().empty
109
114
except KeyError as e :
110
- y_series_data .append (None )
115
+ y_series_data .append (0 )
111
116
else :
112
117
if not data [metric_name ].head ().empty :
113
118
if metric_name .startswith ('cpu' ) or metric_name .startswith ('mem' ):
114
119
y_series_data .append (data [metric_name ].head (1 ).values [0 ])
115
120
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 )
117
122
else :
118
- y_series_data .append (None )
123
+ y_series_data .append (0 )
119
124
120
125
return y_series_data
121
126
@@ -174,15 +179,20 @@ def get_parser():
174
179
parser .add_argument (
175
180
"--query_str" ,
176
181
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 = ""
178
184
)
179
185
parser .add_argument (
180
186
"--csv_filepath" ,
181
187
help = "The path of the CSV file."
182
188
)
183
189
parser .add_argument (
184
190
"--graph_title" ,
185
- help = "The graph title."
191
+ help = "Output path."
192
+ )
193
+ parser .add_argument (
194
+ "--title" ,
195
+ help = "Visual title of graph."
186
196
)
187
197
return parser
188
198
0 commit comments