Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
"""State of the variables (:mod:`fluidsim.base.dedalus.state`)
===============================================================
Provides:
.. autoclass:: StateDedalus
:members:
:private-members:
"""
import numpy as np
from ..state import StateBase
class StatePhysDedalus:
def __init__(self, dedalus_solver, keys):
self.dedalus_solver = dedalus_solver
self.keys = keys
def get_var(self, key):
if self.dedalus_solver is None:
return np.zeros(1)
field = self.dedalus_solver.state[key]
field.set_scales(1.0)
return field["g"].transpose()
def set_var(self, key, value):
field = self.dedalus_solver.state[key]
field.set_scales(1.0)
field["g"] = value.transpose()
def initialize(self, value):
for key in self.keys:
self.get_var(key).fill(value)
@property
def nbytes(self):
return 1
@property
def info(self):
return "Dedalus state"
class StateDedalus(StateBase):
@staticmethod
def _complete_info_solver(info_solver):
"""Complete the ParamContainer info_solver.
This is a static method!
"""
info_solver.classes.State._set_attribs(
{
"keys_state_phys": ["b", "vx", "vz"],
"keys_computable": ["dz_b", "dz_vx", "dz_vz"],
"keys_phys_needed": ["b", "vx", "vz"],
}
)
def compute(self, key):
"""Compute scalar fields such a component of the velocity or vorticity."""
return self.state_phys.get_var(key)
def __init__(self, sim, oper=None):
super().__init__(sim, oper)
sim.init_dedalus()
if self.sim.params.ONLY_COARSE_OPER:
solver = None
else:
solver = self.sim.dedalus_solver
self.state_phys = StatePhysDedalus(solver, self.keys_state_phys)