Make perf2tex.py -e create a tikzpicture environment

performance/perf2tex.py

@@ -22,11 +22,13 @@ LATEX_HEADER = r"""\documentclass[12pt,a4paper,onecolumn,openright]{report}
 % Define bar chart colors
 %"""
 
-LATEX_DOCUMENT_HEADER = r"""\begin{document}
-\begin{tikzpicture}"""
+LATEX_DOCUMENT_HEADER = r"""\begin{document}"""
 
-LATEX_DOCUMENT_FOOTER = r"""\end{tikzpicture}
-\end{document}"""
+TIKZ_PICTURE_HEADER = r"""\begin{tikzpicture}"""
+
+TIKZ_PICTURE_FOOTER = r"""\end{tikzpicture}"""
+
+LATEX_DOCUMENT_FOOTER = r"""\end{document}"""
 
 
 def encode_latex_string(value):
@@ -49,8 +51,8 @@ def assemble_latex_chart(optimization_levels,
             lines.append(r'\definecolor{%s}{HTML}{%02X%02X%02X}' %
                          (color_name, red, green, blue))
         lines.append(LATEX_DOCUMENT_HEADER)
-    lines.append(r"""
-    \begin{axis}[
+    lines.append(TIKZ_PICTURE_HEADER)
+    lines.append(r"""    \begin{axis}[
         width = 0.85*\textwidth,
         height = 8cm,
         major x tick style = transparent,
@@ -82,6 +84,7 @@ def assemble_latex_chart(optimization_levels,
                  ','.join(map(encode_latex_string, optimization_levels)))
     lines.append(r"""
     \end{axis}""")
+    lines.append(TIKZ_PICTURE_FOOTER)
     if not embed:
         lines.append(LATEX_DOCUMENT_FOOTER)
     return '\n'.join(lines)