generate LaTeX tables from some of the benchmarks

1 files changed, 79 insertions, 0 deletions

diff --git a/measurements/tolatex.py b/measurements/tolatex.py
new file mode 100644
index 000000000..73ec7fa3a
--- /dev/null
+++ b/measurements/tolatex.py
@@ -0,0 +1,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)))
+