SCM_pycles_v2/PrognosticVariables.pxd at master · meyerbe/SCM_pycles_v2 · GitHub

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
75
76
77
78
79
80
81
from NetCDFIO cimport NetCDFIO_Stats
from Grid cimport Grid
from TimeStepping cimport TimeStepping

cdef class PrognosticVariables:
    cdef:
        dict name_index
        list index_name
        dict units
        Py_ssize_t nv
        Py_ssize_t nv_scalars
        Py_ssize_t nv_velocities
        long [:] var_type
        double [:] bc_type
        # double [:] values
        # double [:] tendencies
        double [:,:] values
        double [:,:] tendencies

    cpdef add_variable(self,name,units,var_type)
    cpdef initialize(self, Grid Gr, NetCDFIO_Stats NS)
    cpdef update(self, Grid Gr, TimeStepping TS)
    cdef inline Py_ssize_t get_nv(self, str variable_name):
        return self.name_index[variable_name]


cdef class MeanVariables:
    cdef:
        dict name_index
        list index_name
        dict units
        Py_ssize_t nv
        Py_ssize_t nv_scalars
        Py_ssize_t nv_velocities
        long [:] var_type
        double [:] bc_type
        long [:] velocity_directions

        double [:,:] values
        double [:,:] tendencies
    cpdef add_variable(self,name,units,bc_type,var_type)
    cpdef initialize(self, Grid Gr, NetCDFIO_Stats NS)
    cpdef update(self, Grid Gr, TimeStepping TS)
    cpdef update_boundary_conditions(self, Grid Gr)
    # cpdef update_boundary_conditions_tendencies(self, Grid Gr)
    cpdef plot(self, str message, Grid Gr, TimeStepping TS)
    cpdef plot_tendencies(self, str message, Grid Gr, TimeStepping TS)
    cpdef get_variable_array(self,name,Grid Gr)
    cpdef get_tendency_array(self,name,Grid Gr)
    cdef inline Py_ssize_t get_nv(self, str variable_name):
        return self.name_index[variable_name]


cdef class SecondOrderMomenta:
    cdef:
        dict name_index
        list index_name
        dict var_index
        dict units
        Py_ssize_t nv
        Py_ssize_t nv_scalars
        Py_ssize_t nv_velocities
        long [:] var_type
        double [:,:] bc_type

        double [:,:,:] values
        double [:,:,:] tendencies
    cpdef add_variable(self,name,units,bc_type,var_type,n,m)
    # cpdef add_variables(self)
    cpdef initialize(self, Grid Gr, MeanVariables M1, NetCDFIO_Stats NS)
    cpdef update(self, Grid Gr, TimeStepping TS)
    cpdef update_boundary_conditions(self, Grid Gr)
    cpdef plot(self, str message, Grid Gr, TimeStepping TS)
    cpdef plot_nogw(self, str message, Grid Gr, TimeStepping TS)
    cpdef plot_tendencies(self, str message, Grid Gr, TimeStepping TS)
    cpdef plot_nogw_tendencies(self, str message, Grid Gr, TimeStepping TS)
    cpdef get_variable_array(self,name,Grid Gr)
    cpdef get_tendency_array(self,name,Grid Gr)
    cdef inline Py_ssize_t get_nv(self, str variable_name):
        return self.name_index[variable_name]