1 files changed, 9 insertions, 2 deletions

diff --git a/util/plot_gen.py b/util/plot_gen.py
index f24fd2c..c37d7c2 100755
--- a/util/plot_gen.py
+++ b/util/plot_gen.py
@@ -27,6 +27,7 @@ if __name__ == "__main__":
     parser.add_argument("--hist", dest="hist", action="store_true", help="Show histogram.")
     parser.add_argument("--hw-hist", dest="hw_hist", action="store_true", help="Show Hamming weight 2D histogram (private key Hamming weight and generation time).")
     parser.add_argument("--skip-first", dest="skip_first", action="store_true", help="Skip first entry, as it's usually a large outlier.")
+    parser.add_argument("-t", "--title", dest="title", type=str, nargs="?", default="", help="What title to give the figure.")
     parser.add_argument("file", type=str, help="The file to plot(csv).")
 
     opts = parser.parse_args()
@@ -58,8 +59,14 @@ if __name__ == "__main__":
 
     plt.style.use("ggplot")
     fig = plt.figure()
-    fig.tight_layout(rect=[0, 0.02, 1, 0.98])
-    fig.suptitle(opts.file)
+    layout_kwargs = {}
+    if opts.title is None:
+        fig.suptitle(opts.file)
+        layout_kwargs["rect"] = [0, 0.02, 1, 0.98]
+    elif opts.title:
+        fig.suptitle(opts.title)
+        layout_kwargs["rect"] = [0, 0.02, 1, 0.98]
+    fig.tight_layout(**layout_kwargs)        
 
     plot_i = 1
     if plots[0]: