Added microscopic verification plot

Author: brownbaerchen

pySDC/projects/RayleighBenard/README.rst

@@ -59,3 +59,19 @@ for all the configurations to generate the data and then run
     python analysis_scripts/RBC3D_order.py
 
 to make the plot.
+
+
+Plotting microscopic verification
+---------------------------------
+After you have plotted the order of accuracy, you can make a plot for microscopic verification, which includes the order of accuracy plot and adds a plot for the spectrum.
+You need to run and analyse simulations with:
+ - ``--res=32 --dt=0.06 --config=RBC3DG4R4SDC23Ra1e5``
+ - ``--res=64 --dt=0.01 --config=RBC3DG4R4SDC23Ra1e6``
+ - ``--res=128 --dt=0.005 --config=RBC3DG4R4SDC23Ra1e7``
+
+The, just run
+
+.. code-block:: bash
+    python analysis_scripts/RBC3D_spectrum.py
+
+to make the plot.

pySDC/projects/RayleighBenard/analysis_scripts/RBC3D_order.py

@@ -71,10 +71,14 @@ def plot_error_all_components(args):  # pragma: no cover
     ax.set_ylabel(r'$e$')
 
 
-def compare_order(Ra):  # pragma: no cover
-    fig, ax = plt.subplots(figsize=figsize_by_journal('Nature_CS', 1, 0.6))
+def compare_order(Ra, ax=None):  # pragma: no cover
+    if ax is None:
+        fig, ax = plt.subplots(figsize=figsize_by_journal('Nature_CS', 1, 0.6))
+    else:
+        fig = None
+
     if Ra == 1e5:
-        names = ['RK', 'Euler', 'SDC22', 'SDC23', 'SDC34', 'SDC44'][::-1]
+        names = ['RK', 'Euler', 'SDC23', 'SDC44'][::-1]
         configs = [f'RBC3DG4R4{me}Ra1e5' for me in names]
         paths = [f'./data/RBC3DG4R4{me}Ra1e5-res-1-order.pickle' for me in names]
 
@@ -99,7 +103,8 @@ def compare_order(Ra):  # pragma: no cover
     ax.legend(frameon=False)
     ax.set_xlabel(r'$\Delta t$')
     ax.set_ylabel(r'$e$')
-    savefig(fig, 'RBC3D_order_Ra1e5')
+    if fig is not None:
+        savefig(fig, 'RBC3D_order_Ra1e5')
 
 
 def run(args, dt, Tend):

pySDC/projects/RayleighBenard/analysis_scripts/RBC3D_spectrum.py

@@ -50,8 +50,11 @@ def plot_spectra_Ra1e6():  # pragma: no cover
     savefig(fig, 'RBC3D_spectrum_Ra1e6')
 
 
-def plot_all_spectra():  # pragma: no cover
-    fig, ax = plt.subplots(figsize=figsize(scale=1, ratio=0.6))
+def plot_all_spectra(ax=None):  # pragma: no cover
+    if ax is None:
+        fig, ax = plt.subplots(figsize=figsize(scale=1, ratio=0.6))
+    else:
+        fig = None
 
     Ras = ['1e5', '1e6', '1e7']
     dts = [0.06, 0.01, 0.005]
@@ -68,12 +71,26 @@ def plot_all_spectra():  # pragma: no cover
         plot_spectrum(_res, dt, config, ax, label=f'$Ra$={Ra}', color=color, marker=marker)
 
     ax.legend(frameon=False)
-    savefig(fig, 'RBC3D_all_spectra')
+    if fig is not None:
+        savefig(fig, 'RBC3D_all_spectra')
+
+
+def plot_microscopic_verification():  # pragma: no cover
+    from pySDC.projects.RayleighBenard.analysis_scripts.RBC3D_order import compare_order
+
+    fig, axs = plt.subplots(1, 2, figsize=figsize(scale=1, ratio=0.45))
+    plot_all_spectra(axs[0])
+    compare_order(1e5, axs[1])
+    for ax in axs:
+        ax.set_box_aspect(1)
+    fig.tight_layout()
+    savefig(fig, 'RBC3D_microscopic_verification')
 
 
 if __name__ == '__main__':
-    plot_spectra_Ra1e5()
+    # plot_spectra_Ra1e5()
     # plot_spectra_Ra1e6()
     # plot_all_spectra()
+    plot_microscopic_verification()
 
     plt.show()