Added spectrum plot

Author: brownbaerchen

pySDC/projects/GPU/analysis_scripts/RBC3D_order.py

@@ -6,7 +6,8 @@
 from pySDC.implementations.problem_classes.RayleighBenard3D import RayleighBenard3D
 from mpi4py import MPI
 import matplotlib.pyplot as plt
-from pySDC.helpers.plot_helper import setup_mpl, figsize_by_journal
+from pySDC.helpers.plot_helper import figsize_by_journal
+from pySDC.projects.GPU.analysis_scripts.RBC3D_plotting_utils import get_plotting_style, savefig
 
 step_sizes = {
     'RBC3DG4R4Ra1e5': [8e-2, 4e-2, 2e-2, 1e-2, 5e-3],
@@ -53,7 +54,6 @@ def compute_errors(args, dts, Tend):
 
 
 def plot_error_all_components(args):  # pragma: no cover
-    setup_mpl()
     fig, ax = plt.subplots()
     with open(get_path(args), 'rb') as file:
         errors = pickle.load(file)
@@ -72,27 +72,24 @@ def plot_error_all_components(args):  # pragma: no cover
 
 
 def compare_order(Ra):  # pragma: no cover
-    setup_mpl()
     fig, ax = plt.subplots(figsize=figsize_by_journal('Nature_CS', 1, 0.6))
-    ls = {'SD': '-', 'RK': '--', 'Eu': '-.'}
     if Ra == 1e5:
-        labels = ['RK443', 'Euler', 'SDC23', 'SDC34', 'SDC44'][::-1]
         names = ['RK', 'Euler', 'SDC23', 'SDC34', 'SDC44'][::-1]
-        markers = ['>', '.', 'o', '<', 'x'][::-1]
+        configs = [f'RBC3DG4R4{me}Ra1e5' for me in names]
         paths = [f'./data/RBC3DG4R4{me}Ra1e5-res-1-order.pickle' for me in names]
 
     else:
         raise NotImplementedError
 
-    for path, label, marker in zip(paths, labels, markers, strict=True):
+    for path, config in zip(paths, configs, strict=True):
         with open(path, 'rb') as file:
             errors = pickle.load(file)
 
         e = np.array(errors['T'])
         dt = np.array(errors['dt'])
         order = np.log(e[1:] / e[:-1]) / np.log(dt[1:] / dt[:-1])
-        print(f'{label}: order: mean={np.mean(order):.1f} / median={np.median(order):.1f}')
-        ax.loglog(dt, e, label=f'{label}', ls=ls[label[:2]], marker=marker, markersize=6)
+        print(f'{config}: order: mean={np.mean(order):.1f} / median={np.median(order):.1f}')
+        ax.loglog(dt, e, **get_plotting_style(config))
 
     for _dt in dt:
         for i in [1, 3, 4]:
@@ -102,7 +99,7 @@ def compare_order(Ra):  # pragma: no cover
     ax.legend(frameon=False)
     ax.set_xlabel(r'$\Delta t$')
     ax.set_ylabel(r'$e$')
-    fig.savefig('plots/SDC_order_Ra1e5.pdf', bbox_inches='tight')
+    savefig(fig, 'RBC3D_order_Ra1e5')
 
 
 def run(args, dt, Tend):
@@ -140,13 +137,15 @@ def run(args, dt, Tend):
     from pySDC.projects.GPU.run_experiment import parse_args
 
     args = parse_args()
-    config = get_config(args)
 
-    dts = step_sizes[type(config).__name__]
     if args['mode'] == 'run':
+        config = get_config(args)
+        dts = step_sizes[type(config).__name__]
         compute_errors(args, dts, max(dts))
 
-    plot_error_all_components(args)
+    if args['config'] is not None:
+        plot_error_all_components(args)
+
     compare_order(1e5)
     if MPI.COMM_WORLD.rank == 0:
         plt.show()

pySDC/projects/GPU/analysis_scripts/RBC3D_plotting_utils.py

@@ -0,0 +1,45 @@
+from pySDC.helpers.plot_helper import figsize_by_journal, setup_mpl
+import warnings
+
+setup_mpl()
+
+
+def get_plotting_style(config):
+
+    args = {'color': None, 'ls': None, 'marker': None, 'markersize': 6, 'label': None}
+
+    if config == 'RBC3DG4R4SDC23Ra1e5':
+        args['color'] = 'tab:blue'
+        args['ls'] = '-'
+        args['marker'] = 'o'
+        args['label'] = 'SDC23'
+    elif config == 'RBC3DG4R4SDC34Ra1e5':
+        args['color'] = 'tab:orange'
+        args['ls'] = '-'
+        args['marker'] = '<'
+        args['label'] = 'SDC34'
+    elif config in ['RBC3DG4R4SDC44Ra1e5', 'RBC3DG4R4Ra1e5']:
+        args['color'] = 'tab:green'
+        args['ls'] = '-'
+        args['marker'] = 'x'
+        args['label'] = 'SDC44'
+    elif config == 'RBC3DG4R4EulerRa1e5':
+        args['color'] = 'tab:purple'
+        args['ls'] = '--'
+        args['marker'] = '.'
+        args['label'] = 'Euler'
+    elif config == 'RBC3DG4R4RKRa1e5':
+        args['color'] = 'tab:red'
+        args['ls'] = '--'
+        args['marker'] = '>'
+        args['label'] = 'RK444'
+    else:
+        warnings.warn(f'No plotting style for {config=!r}')
+
+    return args
+
+
+def savefig(fig, name, format='pdf', base_path='./plots', **kwargs):
+    path = f'{base_path}/{name}.{format}'
+    fig.savefig(path, bbox_inches='tight', **kwargs)
+    print(f'Saved figure {path!r}')

pySDC/projects/GPU/analysis_scripts/RBC3D_spectrum.py

@@ -0,0 +1,33 @@
+from pySDC.projects.GPU.analysis_scripts.process_RBC3D_data import get_pySDC_data
+from pySDC.projects.GPU.analysis_scripts.RBC3D_plotting_utils import figsize_by_journal, get_plotting_style, savefig
+import matplotlib.pyplot as plt
+
+
+def plot_spectrum(res, dt, config_name, ax):  # pragma: no cover
+    data = get_pySDC_data(res=res, dt=dt, config_name=config_name)
+
+    spectrum = data['avg_spectrum']
+    k = data['k']
+    ax.loglog(k[spectrum > 1e-16], spectrum[spectrum > 1e-16], **get_plotting_style(config_name), markevery=5)
+    ax.set_xlabel('$k$')
+    ax.set_ylabel(r'$\|\hat{u}_x\|$')
+
+
+def plot_spectra_Ra1e5():  # pragma: no cover
+    fig, ax = plt.subplots(figsize=figsize_by_journal('Nature_CS', 1, 0.6))
+
+    configs = [f'RBC3DG4R4{name}Ra1e5' for name in ['SDC34', 'SDC23', '', 'Euler', 'RK']]
+    dts = [0.06, 0.06, 0.06, 0.02, 0.04]
+    res = 32
+
+    for config, dt in zip(configs, dts):
+        plot_spectrum(res, dt, config, ax)
+
+    ax.legend(frameon=False)
+    savefig(fig, 'RBC3D_spectrum_Ra1e5')
+
+
+if __name__ == '__main__':
+    plot_spectra_Ra1e5()
+
+    plt.show()

pySDC/projects/GPU/analysis_scripts/plot_Nu.py

@@ -3,18 +3,11 @@
 import numpy as np
 from scipy import integrate
 from pySDC.helpers.plot_helper import figsize_by_journal, setup_mpl
+from pySDC.projects.GPU.analysis_scripts.process_RBC3D_data import get_pySDC_data
 
 setup_mpl()
 
 
-def get_pySDC_data(res=-1, dt=-1, config_name='RBC3DG4', base_path='data/RBC_time_averaged'):
-    path = f'{base_path}/{config_name}-res{res}-dt{dt:.0e}.pickle'
-    with open(path, 'rb') as file:
-        data = pickle.load(file)
-
-    return data
-
-
 def interpolate_NuV_to_reference_times(data, reference_data, order=12):
     from qmat.lagrange import getSparseInterpolationMatrix
 

pySDC/projects/GPU/analysis_scripts/process_RBC3D_data.py

@@ -204,5 +204,13 @@ def process_RBC3D_data(base_path='./data/RBC_time_averaged', plot=True, args=Non
     return path
 
 
+def get_pySDC_data(res=-1, dt=-1, config_name='RBC3DG4', base_path='data/RBC_time_averaged'):
+    path = f'{base_path}/{config_name}-res{res}-dt{dt:.0e}.pickle'
+    with open(path, 'rb') as file:
+        data = pickle.load(file)
+
+    return data
+
+
 if __name__ == '__main__':
     process_RBC3D_data()

pySDC/projects/GPU/tests/test_RBC_3D_analysis.py

@@ -80,7 +80,7 @@ def test_processing(tmp_processed_data):
 
 
 def test_get_pySDC_data(tmp_processed_data, tmp_path):
-    from pySDC.projects.GPU.analysis_scripts.plot_Nu import get_pySDC_data
+    from pySDC.projects.GPU.analysis_scripts.process_RBC3D_data import get_pySDC_data
 
     args = get_args(tmp_path)
     data = get_pySDC_data(res=args['res'], dt=args['dt'], config_name=args['config'], base_path=tmp_path)