81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
# coding=utf-8
|
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import matplotlib as mpl
|
|
mpl.use('Agg') # do not require X server
|
|
|
|
import matplotlib.pyplot as plt
|
|
|
|
|
|
def draw_area(plot, area, color, label):
|
|
for i, c in enumerate(area):
|
|
plot.axvspan(c.start, c.end, color=color, label=label)
|
|
label = None # show label only once
|
|
|
|
|
|
def draw_plot(run_result, show_etalon=True, show_errors=True,
|
|
show_anomalies=False, show_degradation=True):
|
|
table = run_result.data
|
|
x = [p.time for p in table]
|
|
y = [p.duration for p in table]
|
|
|
|
x2 = [p.time for p in table if p.error]
|
|
y2 = [p.duration for p in table if p.error]
|
|
|
|
figure = plt.figure()
|
|
plot = figure.add_subplot(111)
|
|
plot.plot(x, y, 'b.', label='Successful operations')
|
|
plot.plot(x2, y2, 'r.', label='Failed operations')
|
|
plot.set_ylim(0)
|
|
|
|
plot.axhline(run_result.etalon_threshold, color='violet',
|
|
label='Degradation threshold')
|
|
|
|
# highlight etalon
|
|
if show_etalon:
|
|
plot.axvspan(run_result.etalon_interval.inf,
|
|
run_result.etalon_interval.sup,
|
|
color='#b0efa0', label='Baseline')
|
|
|
|
# highlight anomalies
|
|
if show_anomalies:
|
|
draw_area(plot, run_result.anomaly_area,
|
|
color='#f0f0f0', label='Anomaly')
|
|
|
|
# highlight degradation
|
|
if show_degradation:
|
|
draw_area(plot, run_result.degradation_area,
|
|
color='#f8efa8', label='Degradation')
|
|
|
|
# highlight errors
|
|
if show_errors:
|
|
draw_area(plot, run_result.error_area,
|
|
color='#ffc0a7', label='Downtime')
|
|
|
|
# draw mean
|
|
plot.plot([p.time for p in run_result.smooth_data],
|
|
[p.duration for p in run_result.smooth_data],
|
|
color='cyan', label='Mean duration')
|
|
|
|
plot.grid(True)
|
|
plot.set_xlabel('time, s')
|
|
plot.set_ylabel('operation duration, s')
|
|
|
|
# add legend
|
|
legend = plot.legend(loc='right', shadow=True)
|
|
for label in legend.get_texts():
|
|
label.set_fontsize('small')
|
|
|
|
return figure
|