scripts/plot-pr-contourmap: Allow the use of folder names other than "pr-logs"

1 files changed, 9 insertions, 2 deletions

diff --git a/scripts/plot-pr-contourmap b/scripts/plot-pr-contourmap
index 8b8e60c3..39fedc97 100755
--- a/scripts/plot-pr-contourmap
+++ b/scripts/plot-pr-contourmap
@@ -44,7 +44,7 @@ def update_graph():
 
     global im, cnt, centre_marker
 
-    filename="pr-logs/grid-crystal%i-cycle%s-%s.dat" % (crystal,cycle,varpair)
+    filename="%s/grid-crystal%i-cycle%s-%s.dat" % (folder,crystal,cycle,varpair)
     print filename
 
     with open(filename, "r") as f:
@@ -99,11 +99,18 @@ def update_graph():
 fig = plt.figure(figsize=(10,5))
 fig.subplots_adjust(left=0.05, bottom=0.05, right=0.70, top=0.95)
 
+if len(sys.argv) == 2:
+    folder = sys.argv[1]
+else:
+    folder = "pr-logs"
+
+print("Using folder '%s'" % folder)
+
 # Find out what there is to plot
 crystals = []
 cycles = []
 varpairs = []
-for file in os.listdir("pr-logs"):
+for file in os.listdir(folder):
     if not fnmatch.fnmatch(file, "grid-*.dat"):
         continue
     sp = file.rstrip(".dat").split("-")