Commit 88cd2021 authored by Pierre Augier's avatar Pierre Augier
Browse files

Hide save_urud for spatiotemp-spectra-2d

parent f878e338e245
......@@ -798,7 +798,10 @@ class SpatioTemporalSpectraNS:
vmax=None,
):
"""plot the spatiotemporal spectra, with a cylindrical average in k-space"""
keys_plot = self.keys_fields + ["Khd", "Khr", "Kp"]
keys_plot = self.keys_fields
if self.nb_dim == 3:
keys_plot.extend(["Khd", "Khr", "Kp"])
if key_field is None:
key_field = keys_plot[0]
if key_field not in keys_plot:
......@@ -821,9 +824,12 @@ class SpatioTemporalSpectraNS:
# compute and save spectra if needed
if not path_file.exists():
self.save_spectra_kzkhomega(
tmin=tmin, tmax=tmax, dtype=dtype, save_urud=save_urud
)
if self.nb_dim == 3:
self.save_spectra_kzkhomega(
tmin=tmin, tmax=tmax, dtype=dtype, save_urud=save_urud
)
else:
self.save_spectra_kzkhomega(tmin=tmin, tmax=tmax, dtype=dtype)
# load spectrum
with h5py.File(path_file, "r") as file:
......
......@@ -91,3 +91,6 @@ class SpatioTemporalSpectraNS2D(SpatioTemporalSpectraNS, SpatioTemporalSpectra2D
compute_spectrum_kzkhomega = staticmethod(compute_spectrum_kzkhomega)
_sum_wavenumber = staticmethod(_sum_wavenumber2D)
def save_spectra_kzkhomega(self, tmin=0, tmax=None, dtype=None):
return super().save_spectra_kzkhomega(tmin, tmax, dtype)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment