1#!/usr/bin/env python3
2# coding=utf-8
3# Copyright (c) 2023 Huawei Device Co., Ltd.
4# Licensed under the Apache License, Version 2.0 (the "License");
5# you may not use this file except in compliance with the License.
6# You may obtain a copy of the License at
7#
8#     http://www.apache.org/licenses/LICENSE-2.0
9#
10# Unless required by applicable law or agreed to in writing, software
11# distributed under the License is distributed on an "AS IS" BASIS,
12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13# See the License for the specific language governing permissions and
14# limitations under the License.
15
16import os
17import sys
18import pandas
19from matplotlib import pyplot
20
21stall = [
22    0, 5, 10, 15, 20, 25, 30, 35, 40, 50,
23    60, 70, 80, 90, 100, 120, 140, 160, 180, 200
24]
25
26
27def draw_fig(ax, base, data, title):
28    label = title[:-4]
29    thread = title[-3:]
30
31    # get optimization ratio
32    avg = sum(data) / len(data)
33    avg_base = base['_'.join([label, 'base', thread])].mean()
34    ratio = str(round((avg_base - avg) / avg_base, 3))
35
36    data_base = [float(base['_'.join([label, 'base', thread])][i])
37                 for i in range(len(stall))]
38    ax.plot(stall, data_base, 's-', label='_'.join([label, 'base']))
39    ax.plot(stall, data, 'o-', label=label)
40    ax.legend(loc='best')
41    ax.set_title(''.join([title, '[R:', ratio, ']']))
42
43
44def main(argv):
45    path = argv[1]
46    out = argv[2]
47
48    # read csv
49    base = pandas.read_csv(os.path.join(path, 'base.csv'))
50    t1 = pandas.read_csv(os.path.join(path, 'perf_thread1.csv'))
51    t8 = pandas.read_csv(os.path.join(path, 'perf_thread8.csv'))
52
53    # get actual data
54    t1_fork_join = [(t1[' fork_join'][i] / t1['repeat'][i] *
55                     1 - stall[i] * 10000) / 10000 for i in range(len(stall))]
56    t1_fork_join_worker_submit = [(t1['fork_join_worker_submit'][i] / t1['repeat']
57                                   [i] * 1 - stall[i] * 10000) / 10000 for i in range(len(stall))]
58    t1_airaw = [(t1['airaw'][i] / t1['repeat'][i] * 1 - stall[i]
59                 * 2400 * 3) / 2400 / 3 for i in range(len(stall))]
60    t1_airaw_worker_submit = [(t1['airaw_worker_submit'][i] / t1['repeat']
61                               [i] * 1 - stall[i] * 2400 * 3) / 2400 / 3 for i in range(len(stall))]
62    t1_fib_data_wait = [(t1['fib_data_wait'][i] / t1['repeat'][i]
63                         * 1 - stall[i] * 21890) / 21890 for i in range(len(stall))]
64    t1_fib_child_wait = [(t1['fib_child_wait'][i] / t1['repeat'][i]
65                          * 1 - stall[i] * 21890) / 21890 for i in range(len(stall))]
66    t1_fib_no_wait = [(t1['fib_no_wait'][i] / t1['repeat'][i]
67                       * 1 - stall[i] * 21890) / 21890 for i in range(len(stall))]
68    t8_fork_join = [(t8[' fork_join'][i] / t8['repeat'][i] *
69                     8 - stall[i] * 10000) / 10000 for i in range(len(stall))]
70    t8_fork_join_worker_submit = [(t8['fork_join_worker_submit'][i] / t8['repeat']
71                                   [i] * 8 - stall[i] * 10000) / 10000 for i in range(len(stall))]
72    t8_airaw = [(t8['airaw'][i] / t8['repeat'][i] * 3 - stall[i]
73                 * 2400 * 3) / 2400 / 3 for i in range(len(stall))]
74    t8_airaw_worker_submit = [(t8['airaw_worker_submit'][i] / t8['repeat']
75                               [i] * 3 - stall[i] * 2400 * 3) / 2400 / 3 for i in range(len(stall))]
76    t8_fib_data_wait = [(t8['fib_data_wait'][i] / t8['repeat'][i]
77                         * 8 - stall[i] * 21890) / 21890 for i in range(len(stall))]
78    t8_fib_child_wait = [(t8['fib_child_wait'][i] / t8['repeat'][i]
79                          * 8 - stall[i] * 21890) / 21890 for i in range(len(stall))]
80    t8_fib_no_wait = [(t8['fib_no_wait'][i] / t8['repeat'][i]
81                       * 8 - stall[i] * 21890) / 21890 for i in range(len(stall))]
82
83    # draw figure
84    fig, ax = pyplot.subplots(2, 7, figsize=(30, 8))
85
86    draw_fig(ax[0, 0], base, t1_fork_join, 'fork_join_1th')
87    draw_fig(ax[0, 1], base, t1_fork_join_worker_submit,
88             'fork_join_worker_submit_1th')
89    draw_fig(ax[0, 2], base, t1_airaw, 'airaw_1th')
90    draw_fig(ax[0, 3], base, t1_airaw_worker_submit, 'airaw_worker_submit_1th')
91    draw_fig(ax[0, 4], base, t1_fib_data_wait, 'fib20_data_wait_1th')
92    draw_fig(ax[0, 5], base, t1_fib_child_wait, 'fib20_child_wait_1th')
93    draw_fig(ax[0, 6], base, t1_fib_no_wait, 'fib20_no_wait_1th')
94    draw_fig(ax[1, 0], base, t8_fork_join, 'fork_join_8th')
95    draw_fig(ax[1, 1], base, t8_fork_join_worker_submit,
96             'fork_join_worker_submit_8th')
97    draw_fig(ax[1, 2], base, t8_airaw, 'airaw_8th')
98    draw_fig(ax[1, 3], base, t8_airaw_worker_submit, 'airaw_worker_submit_8th')
99    draw_fig(ax[1, 4], base, t8_fib_data_wait, 'fib20_data_wait_8th')
100    draw_fig(ax[1, 5], base, t8_fib_child_wait, 'fib20_child_wait_8th')
101    draw_fig(ax[1, 6], base, t8_fib_no_wait, 'fib20_no_wait_8th')
102
103    fig.savefig(os.path.join(path, out))
104
105
106if __name__ == '__main__':
107    main(sys.argv)
108