EGGS/plot.py at master · txstc55/EGGS · 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
86
87
88
89
90
91
92
93
94
95
96
97
98
import json
import matplotlib.pyplot as plt
import numpy as np
import sys
from math import ceil, log

json_file_name = "result_datas/all_result.json"
plt.rc('ytick', labelsize=28)    # fontsize of the tick labels
plt.rc('xtick', labelsize=21)    # fontsize of the tick labels
plt.rc('legend', fontsize=14)    # fontsize of the tick labels

results = []
if len(sys.argv) == 2:
    with open(sys.argv[1], 'r') as j:
        results = json.load(j)
        json_file_name = sys.argv[1]
else:
    with open('result_datas/all_result.json', 'r') as j:
        results = json.load(j)

assemble = [float(x["ASSEMBLE"])/100.0/1000.0 for x in results]
compute = [float(x["COMPUTE"])/100.0/1000.0 for x in results]
solve = [float(x["SOLVE"])/100.0/1000.0 for x in results]

sum = []
for i in range(len(assemble)):
    sum.append(assemble[i]+compute[i]+solve[i])

# total_time =[assemble[i]+compute[i]+solve[i] for i in range(5)]

bars = np.add(solve, compute).tolist()

N = len(results)

ind = np.arange(N)    # the x locations for the groups
width = 0.6       # the width of the bars: can also be len(x) sequence

if (max(solve) != 0):
    p1 = plt.bar(ind, solve, width, color="#9b59b6")
p2 = plt.bar(ind, compute, width, bottom=solve, color="#2ecc71")
if (max(assemble) != 0):
    p3 = plt.bar(ind, assemble, width, bottom=bars, color="#3498db")


best_log_10 = ceil(log(max(sum), 10))
div = max(sum)/(10**(best_log_10-1))+1
max_y = div*(10**(best_log_10-1))
gap = 10**(best_log_10-1)


plt.xticks(ind, [x["name"] for x in results])
plt.yticks(np.arange(0, max_y, gap))


if (max(solve) != 0):
    for r1, r2, r3 in zip(p1, p2, p3):
        h1 = r1.get_height()
        h2 = r2.get_height()
        h3 = r3.get_height()
        #plt.text(r1.get_x() + r1.get_width() / 2., h1 / 2., "%.2E" % h1,
        #         ha="center", va="center", color="white", fontsize=11, fontweight="bold")
        #plt.text(r2.get_x() + r2.get_width() / 2., h1 + h2 / 2., "%.2E" % h2,
        #         ha="center", va="center", color="white", fontsize=11, fontweight="bold")
        #plt.text(r3.get_x() + r3.get_width() / 2., h1 + h2 + h3 / 2., "%.2E" %
        #         h3, ha="center", va="center", color="white", fontsize=11, fontweight="bold")
        plt.legend((p1[0], p2[0], p3[0]), ('Solve', 'Compute', 'Assemble'), fontsize=22)
else:
    for r2 in p2:
        h2 = r2.get_height()
        #plt.text(r2.get_x() + r2.get_width() / 2.,  h2 / 2., "%.2E" % h2,
        #         ha="center", va="center", color="white", fontsize=11, fontweight="bold")
        # plt.legend((p2[0],), ('Compute',), fontsize=22)


save_plot_name = json_file_name.split(".")[0].split("/")[-1]


# plt.gca().set_axis_off()
plt.subplots_adjust(top=1, bottom=0, right=1, left=0,
                    hspace=0, wspace=0)
ax = plt.axes()
ax.spines['top'].set_visible(False)
ax.spines['right'].set_visible(False)
# ax.spines['bottom'].set_visible(False)
# ax.spines['left'].set_visible(False)
# plt.margins(0, 0)
# plt.gca().xaxis.set_major_locator(plt.NullLocator())
# plt.gca().yaxis.set_major_locator(plt.NullLocator())
plt.savefig("test_result_graphs/"+save_plot_name+".pdf", bbox_inches='tight',
            pad_inches=0, dpi=200)


# plt.savefig("test_result_graphs/"+save_plot_name+".png", dpi=200)


print(results)