TL: minor improvements on plotScaling script

Author: tlunet

pySDC/playgrounds/dedalus/scripts/plotScaling.py

@@ -3,13 +3,22 @@
 """
 Plot string scaling results stored in a given folder
 """
+import os
 import json
 import glob
 
 import numpy as np
 import matplotlib.pyplot as plt
 
-folder = "_benchJureca"
+folder = (
+    "_benchJureca"
+    # "_benchJusuf_64tpc"
+    # "_benchJusuf"
+)
+assert os.path.isdir(folder), f"{folder} is not a folder"
+
+nNodes = 4
+nSweeps = 4
 
 R = 2
 if R == 2:
@@ -51,13 +60,13 @@
 
 symbols = ["o", "^", "s", "p", "*"]
 colors = plt.rcParams['axes.prop_cycle'].by_key()['color']
-ls = "--" if "64tpc" in folder else "-"
+ls = "--" if "128tpc" in folder else "-"
 
-plt.figure("scaling"+"-nSpS"*useNSpS+f"-R{R}")
+plt.figure("scaling"+"-nSpS"*useNSpS+f"-R{R}-{folder}")
 for scheme, sym, col in zip(results.keys(), symbols, colors):
     res = np.array(results[scheme]).T
     plt.loglog(*res, sym+ls, label=scheme, c=col)
-    # plt.loglog(res[0], np.prod(res[:, 0])/res[0], "--", c="gray")
+    plt.loglog(res[0], np.prod(res[:, 0])/res[0], "--", c="gray")
 plt.legend()
 plt.grid(True)
 plt.xlabel("$N_{p}$")
@@ -68,13 +77,19 @@
 plt.tight_layout()
 
 
-plt.figure(f"PinT-speedup-R{R}")
+plt.figure(f"PinT-efficiency-R{R}-{folder}")
 nProcSpace, tSDC = np.array(results["SDC"]).T
+spdIdeal = (1+(nSweeps-1)*nNodes)/(1+(nSweeps-1))
+effIdeal = spdIdeal/nNodes
 for scheme, sym, col in zip(schemes[2:], symbols, colors):
     _, tSDCPinT = np.array(results[scheme]).T
     speedup = tSDC[:len(tSDCPinT)]/tSDCPinT
-    plt.semilogx(nProcSpace[:len(tSDCPinT)], speedup, sym+ls, c=col, label=scheme)
+    efficiency = speedup/nNodes
+    nPS = nProcSpace[:len(tSDCPinT)]
+    plt.semilogx(nPS, efficiency, sym+ls, c=col, label=scheme)
+plt.semilogx(nPS, 0*nPS+effIdeal, "--", c="gray")
+plt.ylim(0, 1)
 plt.legend()
 plt.grid(True)
-plt.xlabel("$N_{p,S}$"), plt.ylabel("PinT Speedup")
+plt.xlabel("$N_{p,Space}$"), plt.ylabel("PinT-Efficiency")
 plt.tight_layout()