- Notifications
You must be signed in to change notification settings - Fork 7
/
Copy pathmakeplots.py
114 lines (96 loc) · 3.68 KB
/
makeplots.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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
importmatplotlib.pyplotasplt
importmatplotlib.tickerasticker
importsys, os, glob, re, math
results= {}
defAddRes(name, mem, N, eltime):
globalresults
results.setdefault(mem, {}).setdefault(name, {})[N] =eltime
defReadRes(fn):
withopen(fn, "rt") asf:
data=f.read()
match=re.search(r"Arrays: (\d*) x (\d*)", data)
N=int(match.group(2))
match=re.search(r"Memory: (\d*)", data)
mem=int(match.group(1))
formatchinre.finditer(r"\s*([0-9.]+)\s*ns\s*:\s*(\S+)", data):
eltime=float(match.group(1))
name=match.group(2)
AddRes(name, mem, N, eltime)
forfninglob.glob("res/*.log"):
ReadRes(fn)
# plt.loglog([1,2,3,4], [1,4,9,16], 'bo', [1,2,3,4], [16,9,9,10], 'ro', basex=2, basey=2, linestyle='-')
# plt.show()
styles= ['yx', 'rx', 'r+', 'mx', 'm+', 'k.', 'ko', 'bo', 'bs', 'yo', 'g*', 'gP', 'gd', 'm*', 'c*']
dpi=150
formem, graphsinresults.items():
args= []
names= []
argsPE= []
argsLog= []
idx=0
forname, graphingraphs.items():
if ('linear'innameand'scalar'inname):
continue
X= []
Y= []
Z= []
W= []
forN, eltimeingraph.items():
X.append(N)
Y.append(eltime)
Z.append(eltime/N)
W.append(eltime/math.log(N, 2.0))
args+= [X, Y, styles[idx]]
argsPE+= [X, Z, styles[idx]]
argsLog+= [X, W, styles[idx]]
names.append(name)
idx+=1
print("%s: %s"% (name, args[-1]))
title="(memory = %dB)"%mem
iflen(sys.argv) >1:
title=sys.argv[1] +" "+title
ax=plt.axes()
ax.set_title(title)
ax.loglog(*args, basex=2, basey=2, linestyle='-')
ax.set_xlabel("Array length (N)")
ax.set_ylabel("Time per search, ns")
ax.grid(True, which="major")
ax.grid(True, which="minor", color='0.8', linestyle=':')
ax.legend(names, loc=2, prop={'size': 6})
ax.get_yaxis().get_minor_locator().subs([1.25, 1.5, 1.75])
ax.get_yaxis().set_minor_formatter(ticker.FuncFormatter(lambdax,p: str(int(x))))
ax.get_yaxis().set_major_formatter(ticker.ScalarFormatter())
ax.get_xaxis().set_major_formatter(ticker.ScalarFormatter())
#plt.show()
plt.savefig('res/plot_search_%d.png'%mem, bbox_inches='tight', dpi=dpi)
plt.gcf().clear()
ax=plt.axes()
ax.set_title(title)
ax.semilogx(*argsPE, basex=2, linestyle='-')
ax.set_xlabel("Array length (N)")
ax.set_ylabel("Time per element, ns")
ax.grid(True, which="major")
ax.grid(True, which="minor", color='0.8', linestyle=':')
ax.legend(names, loc=1, prop={'size': 6})
ax.set_ylim(0.0, 0.5)
ax.get_yaxis().set_minor_locator(ticker.MultipleLocator(0.01))
ax.get_yaxis().tick_right()
ax.get_xaxis().set_major_formatter(ticker.ScalarFormatter())
#plt.show()
plt.savefig('res/plot_elem_%d.png'%mem, bbox_inches='tight', dpi=dpi)
plt.gcf().clear()
ax=plt.axes()
ax.set_title(title)
ax.semilogx(*argsLog, basex=2, linestyle='-')
ax.set_xlabel("Array length (N)")
ax.set_ylabel("Time per one bin.search comparison, ns")
ax.grid(True, which="major")
ax.grid(True, which="minor", color='0.8', linestyle=':')
ax.legend(names, loc=2, prop={'size': 6})
ax.set_ylim(1.0, 7.0)
ax.get_yaxis().set_minor_locator(ticker.MultipleLocator(0.5))
ax.get_yaxis().tick_right()
ax.get_xaxis().set_major_formatter(ticker.ScalarFormatter())
#plt.show()
plt.savefig('res/plot_log_%d.png'%mem, bbox_inches='tight', dpi=dpi)
plt.gcf().clear()
