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
|
# Generates benchmark graphs in LaTex (following format from the pgfplots
# package)
#
# This ignores duplicate entries, including different primes with the same
# number of bits.
import sys
USAGE = "USAGE: python tolatex.py [input file]"
SETUPS = {
"fiat_montgomery32": "color=blue,mark=triangle*",
"fiat_montgomery64": "color=blue,mark=square*",
"fiat_solinas32": "color=blue,mark=triangle",
"fiat_solinas64": "color=blue,mark=square",
"gmpvar": "color=red,mark=ball*",
"gmpxx": "color=red,mark=o",
"gmpsec" : "color=red,mark=*"
}
class ParseException(Exception): pass
def parse_line(line):
data = line.strip().split("\t")
if len(data) != 3 or (data[1] not in SETUPS) or ("2e" not in data[0]) :
raise ParseException("Could not parse line %s" %line)
return {
"prime" : data[0],
"setup" : data[1],
"time" : data[2] }
# remove duplicates, reorganize, and parse number of bits from primes
def clean_data(parsed_lines):
out = {s:{} for s in SETUPS}
for ln in parsed_lines:
nbits = ln["prime"].split("2e")[1].split("m")[0].split("p")[0]
# if some measurement is duplicated, ignore the repeats
if nbits not in out[ln["setup"]]:
out[ln["setup"]][nbits] = ln["time"]
return out
def makeplot(data):
out = """
\\begin{figure*}
\\begin{tikzpicture}
\t\\begin{axis}[
\t\theight=11cm,
\t\twidth=\\textwidth,
\t\tgrid=major,
\t\tlegend pos= north west,
\t\txlabel=Prime Size (bits),
\t\tylabel=Time (seconds)]
"""
for s in SETUPS:
out +="\t\t\\addplot[%s] coordinates {\n" %SETUPS[s]
for nbits in data[s]:
out += "\t\t\t(%s, %s) \n" %(nbits, data[s][nbits])
out += "\t\t};\n"
out += "\t\t\\addlegendentry{%s}\n\n" %s.replace("_", "\_")
out += """
\t\end{axis}
\\end{tikzpicture}
\\end{figure*}
"""
return out
if __name__ == "__main__":
if len(sys.argv) != 2:
print(USAGE)
sys.exit()
f = open(sys.argv[1])
parsed_lines = []
for line in f:
try:
parsed_lines.append(parse_line(line))
except ParseException:
print("WARNING: Could not parse line %s, skipping" %line)
f.close()
print(makeplot(clean_data(parsed_lines)))
|
