Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
fluiddyn
fluidsim
Commits
e531c0a89e4c
Commit
df6526ca
authored
Apr 12, 2021
by
Pierre Augier
Browse files
Try to fix no_vz_kz0 option
parent
a7f967c7c78e
Pipeline
#20602
passed with stage
in 15 minutes and 56 seconds
Changes
2
Pipelines
2
Show whitespace changes
Inline
Side-by-side
fluidsim/solvers/ns3d/strat/solver.py
View file @
e531c0a8
...
...
@@ -199,9 +199,6 @@ class Simul(SimulNS3D):
oper
.
project_perpk3d
(
fx_fft
,
fy_fft
,
fz_fft
)
if
self
.
no_vz_kz0
:
dealiasing_variable
(
fz_fft
,
self
.
where_kz_0
)
if
state_spect
is
None
:
b
=
self
.
state
.
state_phys
.
get_var
(
"b"
)
else
:
...
...
@@ -211,6 +208,10 @@ class Simul(SimulNS3D):
div_vb_fft
=
oper
.
div_vb_fft_from_vb
(
vx
,
vy
,
vz
,
b
)
fb_fft
=
compute_fb_fft
(
div_vb_fft
,
self
.
params
.
N
,
vz_fft
)
if
self
.
no_vz_kz0
:
dealiasing_variable
(
fz_fft
,
self
.
where_kz_0
)
dealiasing_variable
(
fb_fft
,
self
.
where_kz_0
)
tendencies_fft
.
set_var
(
"b_fft"
,
fb_fft
)
if
self
.
is_forcing_enabled
:
...
...
fluidsim/solvers/ns3d/time_stepping.py
View file @
e531c0a8
...
...
@@ -19,6 +19,11 @@ class TimeSteppingPseudoSpectralNS3D(TimeSteppingPseudoSpectral):
self
.
sim
.
oper
.
project_perpk3d
(
vx_fft
,
vy_fft
,
vz_fft
)
if
self
.
sim
.
no_vz_kz0
:
dealiasing_variable
(
vz_fft
,
self
.
sim
.
where_kz_0
)
if
"b_fft"
in
state_spect
.
keys
:
dealiasing_variable
(
state_spect
.
get_var
(
"b_fft"
),
self
.
sim
.
where_kz_0
)
self
.
sim
.
state
.
statephys_from_statespect
()
# np.isnan(np.sum seems to be really fast
if
np
.
isnan
(
np
.
sum
(
state_spect
[
0
])):
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment