add summarize option to specify a custom latex preamble

this is meant for some latex setup specifics like including common.tex

3 files changed, 14 insertions, 0 deletions

diff --git a/src/globalvars.py b/src/globalvars.py
index bcb722b..474c124 100644
--- a/src/globalvars.py
+++ b/src/globalvars.py
@@ -62,3 +62,5 @@ benchmarks = [e[:-3] for e in os.listdir(os.path.join(allocbenchdir, benchsrcdir
               if e[-3:] == ".py" and e != "__init__.py"]
 
 summary_file_ext = "svg"
+
+latex_custom_preamble = ""
diff --git a/src/plots.py b/src/plots.py
index 1be0d31..d7fb0ea 100644
--- a/src/plots.py
+++ b/src/plots.py
@@ -469,6 +469,9 @@ def pgfplot_legend(bench, sumdir=""):
         rgb = matplotlib.colors.to_rgb(alloc_dict["color"])
         tex += f"\\providecolor{{{alloc_name}-color}}{{rgb}}{{{rgb[0]},{rgb[1]},{rgb[2]}}}\n"
 
+    if src.globalvars.latex_custom_preamble:
+        tex += src.globalvars.latex_custom_preamble + "\n"
+
     tex +=\
 """
 \\begin{document}
@@ -517,6 +520,9 @@ def pgfplot_linear(bench, perms, xval, yval, ylabel="'y-label'", xlabel="'x-labe
         rgb = matplotlib.colors.to_rgb(alloc_dict["color"])
         tex += f"\\providecolor{{{alloc_name}-color}}{{rgb}}{{{rgb[0]},{rgb[1]},{rgb[2]}}}\n"
 
+    if src.globalvars.latex_custom_preamble:
+        tex += src.globalvars.latex_custom_preamble + "\n"
+
     tex +=\
 f"""
 \\begin{{document}}
diff --git a/summarize.py b/summarize.py
index 6bb0485..4ccd928 100755
--- a/summarize.py
+++ b/summarize.py
@@ -161,12 +161,18 @@ if __name__ == "__main__":
                         "--exclude-benchmarks",
                         help="benchmarks to exclude",
                         nargs='+')
+    parser.add_argument("--latex-preamble",
+                        help="latex code to include in the preamble of generated standalones",
+                        type=str)
 
     args = parser.parse_args()
 
     if args.file_ext:
         src.globalvars.summary_file_ext = args.file_ext
 
+    if args.latex_preamble:
+        src.globalvars.latex_custom_preamble = args.latex_preamble
+
     if not os.path.isdir(args.results):
         print_error(f"{args.results} is no directory")
         sys.exit(1)