forked from lanl/CLAMR
-
Notifications
You must be signed in to change notification settings - Fork 1
/
plotLineCount.py
77 lines (61 loc) · 2.73 KB
/
plotLineCount.py
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
import numpy as np
import matplotlib.pyplot as plt
# Collect the data from the file, ignore empty lines
data = open('linecount.txt', 'r')
name = []
totalCPUtime = []
AllRuntime = dict()
FaceLineCount = dict()
CellLineCount = dict()
OrigLineCount = dict()
InPlaceLineCount = dict()
RegGridLineCount = dict()
for line in data:
if 'State' in line:
continue
if '^calc' in line:
continue
if 'finite_difference' in line:
line = line.strip('\n')
#print(len(line.split()))
if len(line.split()) == 4:
string,dummy,dummy,value = line.split()
if 'face' in string:
if 'via_faces' in string:
FaceLineCount["origface"] = float(value)
if 'face_in_place' in string:
FaceLineCount["faceinplace"] = float(value)
if 'regular_cells_by_faces' in string:
FaceLineCount["reggridbyfaces"] = float(value)
else:
if 'difference.cpp' in string:
CellLineCount["origcell"] = float(value)
if 'cell_in_place' in string:
CellLineCount["cellinplace"] = float(value)
if 'regular_cells.cpp' in string:
CellLineCount["reggridbycell"] = float(value)
fig, ax = plt.subplots()
bar_width = 0.35
#y1 = [ CellLineCount["origcell"]/CellLineCount["origcell"],CellLineCount["cellinplace"]/CellLineCount["origcell"],0 ]
#y2 = [ FaceLineCount["origface"]/CellLineCount["origcell"],FaceLineCount["faceinplace"]/CellLineCount["origcell"],0 ]
#y1 = [ CellLineCount["origcell"]/CellLineCount["origcell"],CellLineCount["cellinplace"]/CellLineCount["origcell"],CellLineCount["reggridbycell"]/CellLineCount["origcell"] ]
#y2 = [ FaceLineCount["origface"]/CellLineCount["origcell"],FaceLineCount["faceinplace"]/CellLineCount["origcell"],FaceLineCount["reggridbyfaces"]/CellLineCount["origcell"] ]
y1 = [ CellLineCount["origcell"],CellLineCount["cellinplace"],CellLineCount["reggridbycell"] ]
y2 = [ FaceLineCount["origface"],FaceLineCount["faceinplace"],FaceLineCount["reggridbyfaces"] ]
x1 = np.arange(len(y1))
x2 = [x + bar_width for x in x1]
plt.bar(x1, y1, width=bar_width, color='b', edgecolor='white', label='Cells')
plt.bar(x2, y2, width=bar_width, color='r', edgecolor='white', label='Faces')
axes = plt.gca() # get current axes
axes.set_ylim([0,500])
ax.set_xlabel('Mesh Data Structure',fontsize=18)
ax.set_ylabel('Linecount',fontsize=18)
ax.tick_params(axis = 'both', labelsize = 14)
ax.set_xticks(x1 + bar_width)
ax.set_xticklabels(('Original AMR','InPlace','RegGrid'))
ax.legend(loc=1)
plt.setp(ax.get_legend().get_texts(), fontsize='16') # for legend text
fig.tight_layout()
plt.savefig("plotLineCount.pdf")
plt.savefig("plotLineCount.svg")
plt.show()