diff --git a/fluidsim/solvers/ns3d/output/spectra.py b/fluidsim/solvers/ns3d/output/spectra.py index 9b80c0e27161e8837da34aa74225c264fb0cb54d_Zmx1aWRzaW0vc29sdmVycy9uczNkL291dHB1dC9zcGVjdHJhLnB5..ae0dedfbdaa3efadc31b8bbf17bd5e17d5a93e53_Zmx1aWRzaW0vc29sdmVycy9uczNkL291dHB1dC9zcGVjdHJhLnB5 100644 --- a/fluidsim/solvers/ns3d/output/spectra.py +++ b/fluidsim/solvers/ns3d/output/spectra.py @@ -225,7 +225,7 @@ if ylim is not None: ax.set_ylim(ylim) - def plot_kzkh(self, tmin=0, tmax=None, key="K", ax=None): + def plot_kzkh(self, tmin=0, tmax=None, key="K", ax=None, vmin=None, vmax=None): data = self.load_kzkh_mean(tmin, tmax, key) spectrum = np.log10(data[key]) kz = data["kz"] @@ -233,6 +233,8 @@ if ax is None: fig, ax = self.output.figure_axe() + else: + fig = ax.figure ax.set_xlabel(r"$\kappa_h$") ax.set_ylabel("$k_z$") @@ -236,5 +238,5 @@ ax.set_xlabel(r"$\kappa_h$") ax.set_ylabel("$k_z$") - ax.set_title("log 3D spectra\n" + self.output.summary_simul) + ax.set_title("log10 spectra\n" + self.output.summary_simul) @@ -240,5 +242,6 @@ - ax.pcolormesh(kh, kz, spectrum) + qmesh = ax.pcolormesh(kh, kz, spectrum, vmin=vmin, vmax=vmax) + fig.colorbar(qmesh) def plot1d( self, @@ -324,5 +327,5 @@ ks_no0[ks == 0] = np.nan coef_norm = ks_no0 ** (coef_compensate) - style_line = "" + style_line = "r" if direction == "z": @@ -328,6 +331,6 @@ if direction == "z": - style_line = ":" + style_line = "g" ax.plot( ks, spectrum * coef_norm, @@ -330,8 +333,8 @@ ax.plot( ks, spectrum * coef_norm, - "k" + style_line, + style_line, linewidth=2, label=f"$E(k_{direction})$", )