"""Simple 1d solver (:mod:`fluidsim.solvers.square1d.solver`)
=============================================================
Provides:
.. autoclass:: Simul
:members:
:private-members:
"""
import numpy as np
from fluidsim.base.setofvariables import SetOfVariables
from fluidsim.base.solvers.pseudo_spect import (
SimulBasePseudoSpectral,
InfoSolverPseudoSpectral,
)
class InfoSolverSquare1DPseudoSpect(InfoSolverPseudoSpectral):
def _init_root(self):
super()._init_root()
package = "fluidsim.solvers.nl1d"
self.module_name = package + ".solver"
self.class_name = "Simul"
self.short_name = "nl1d"
classes = self.classes
classes.Operators.module_name = "fluidsim.operators.operators1d"
classes.Operators.class_name = "OperatorsPseudoSpectral1D"
classes.State.module_name = "fluidsim.solvers.ad1d.pseudo_spect.state"
classes.State.class_name = "StateAD1DPseudoSpectral"
base = "fluidsim.solvers.ad1d"
classes.Output.module_name = base + ".output"
classes.Output.class_name = "Output"
classes.InitFields.module_name = base + ".init_fields"
classes.InitFields.class_name = "InitFieldsAD1D"
[docs]
class Simul(SimulBasePseudoSpectral):
InfoSolver = InfoSolverSquare1DPseudoSpect
[docs]
@staticmethod
def _complete_params_with_default(params):
SimulBasePseudoSpectral._complete_params_with_default(params)
params._set_attrib("sigma", 1.0)
[docs]
def tendencies_nonlin(self, state_spect=None, old=None):
if state_spect is None:
signal = self.state.state_phys.get_var("s")
else:
s_fft = state_spect.get_var("s_fft")
signal = self.oper.ifft(s_fft)
f_signal = -np.sign(signal) * self.params.sigma * signal**2
if old is None:
tendencies_fft = SetOfVariables(
like=self.state.state_spect, info="tendencies_nonlin"
)
else:
tendencies_fft = old
f_fft = tendencies_fft.get_var("s_fft")
self.oper.fft_as_arg(f_signal, f_fft)
self.oper.dealiasing(tendencies_fft)
return tendencies_fft
if __name__ == "__main__":
from scipy.signal import gausspulse
params = Simul.create_default_params()
params.output.sub_directory = "examples"
params.output.periods_save.phys_fields = 1.0
params.short_name_type_run = "test"
params.oper.Lx = 10
params.oper.nx = 256
params.nu_2 = 0
params.time_stepping.USE_CFL = False
params.time_stepping.deltat0 = 1e-3
params.time_stepping.t_end = 10.0
params.time_stepping.type_time_scheme = "RK2"
params.init_fields.type = "in_script"
sim = Simul(params)
# Initialize
x = sim.oper.x
Lx = sim.oper.Lx
s = gausspulse(x - Lx / 2, fc=2)
s_fft = sim.oper.fft(s)
sim.state.init_statephys_from(s=s)
sim.state.init_statespect_from(s_fft=s_fft)
# Plot once
sim.output.init_with_initialized_state()
sim.output.phys_fields.plot(field="s")
sim.time_stepping.start()
sim.output.phys_fields.plot(field="s")