TL: script update

Author: tlunet

pySDC/playgrounds/dedalus/scripts/.gitignore

@@ -0,0 +1 @@
+postData

pySDC/playgrounds/dedalus/scripts/checkDNS.py

@@ -5,12 +5,11 @@
 
 @author: cpf5546
 """
+import json
 import pandas as pd
 import matplotlib.pyplot as plt
 
-from pySDC.playgrounds.dedalus.problems.rbc import OutputFiles, checkDNS
-
-OutputFiles.VERBOSE = True
+from pySDC.playgrounds.dedalus.problems.rbc import checkDNS
 
 simDirs = [
     "run_3D_A4_M0.5_R1_Ra5e3",
@@ -29,30 +28,33 @@
     "run_3D_A4_M1_R1_Ra1e6",
     ]
 
-simDirs = [
+runs = [
     "run_3D_A4_M1_R2_Ra1.5e5",
+    "run_3D_A4_M1_R2_Ra3e5",
+    "run_3D_A4_M1_R2_Ra5e5",
+    "run_3D_A4_M1_R2_Ra8e5",
+    "run_3D_A4_M1_R2_Ra9.5e5",
     "run_3D_A4_M1_R2_Ra1e6",
+    "run_3D_A4_M1_R2_Ra2e6"
+    ]
+
+runs = [
+    "run_3D_A4_M1_R4_Ra2e6",
+    "run_3D_A4_M1_R4_Ra5e6",
     ]
 
 df = pd.DataFrame(
     columns=["Ra", "c_2[u]", "c_2[uv]", "c_2[uh]", "c_2[b]", "c_2[p]"])
 
-for i, dirName in enumerate(simDirs):
+for i, run in enumerate(runs):
 
-    output = OutputFiles(dirName)
-    df.loc[i, "Ra"] = float(dirName.split("_Ra")[-1])
+    with open(f"postData/{run}.json", "r") as f:
+        data = json.load(f)
 
-    # assert len(output.times) == 61, f"not 61 fields for {dirName}"
-    # if len(output.times) == 61:
-    #     start = 20
-    # else:
-    #     start = 60
-    start = 20
+    df.loc[i, "Ra"] = data["infos"]["Ra"]
 
-    spectrum = output.getSpectrum(
-            which="all", zVal="all",
-            start=start, batchSize=None)
-    kappa = output.kappa
+    spectrum = data["spectrum"]
+    kappa = data["spectrum"]["kappa"]
 
     for name in ["u", "uv", "uh", "b", "p"]:
         check = checkDNS(spectrum[name], kappa)
@@ -70,4 +72,5 @@
 plt.ylabel("quadratic coefficient $c_2$")
 plt.tight_layout()
 
-# df.columns = ["$"+label+"$" for label in df.columns]
+df.columns = ["$"+label+"$" for label in df.columns]
+print(df.to_markdown(index=False))

pySDC/playgrounds/dedalus/scripts/plotSpectrum.py

@@ -0,0 +1,34 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Created on Tue Oct 21 14:26:58 2025
+
+@author: cpf5546
+"""
+import json
+import numpy as np
+import matplotlib.pyplot as plt
+
+
+data = [
+    ("run_3D_A4_M1_R4_Ra2e6", "$Ra=2e6$"),
+    ("run_3D_A4_M1_R4_Ra5e6", "$Ra=5e6$"),
+]
+
+plt.figure("spectrum")
+for run, label in data:
+
+    with open(f"postData/{run}.json", "r") as f:
+        data = json.load(f)
+
+    spectrum = data["spectrum"]
+    kappa = np.array(spectrum["kappa"])
+
+    var = "uh"
+    plt.loglog(kappa[1:], spectrum[var][1:], label=f"$E({var})$ ({label})")
+
+plt.loglog(kappa[1:], kappa[1:]**(-5/3), '--k')
+plt.text(10, 0.1, r"$\kappa^{-5/3}$", fontsize=16)
+plt.legend(); plt.grid(True)
+plt.ylabel("spectrum"); plt.xlabel(r"wavenumber $\kappa$")
+plt.tight_layout()