SCAMPy/EDMF_Updrafts.pxd at master · ilopezgp/SCAMPy · 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
cimport Grid
cimport ReferenceState
from Variables cimport GridMeanVariables
from NetCDFIO cimport NetCDFIO_Stats
from EDMF_Environment cimport EnvironmentVariables
from EDMF_Rain cimport RainVariables

cdef class UpdraftVariable:
    cdef:
        double [:,:] values
        double [:,:] new
        double [:,:] old
        double [:,:] tendencies
        double [:,:] flux
        double [:] bulkvalues
        str loc
        str kind
        str name
        str units
    cpdef set_bcs(self, Grid.Grid Gr)

cdef class UpdraftVariables:
    cdef:
        Grid.Grid Gr

        UpdraftVariable W
        UpdraftVariable Area
        UpdraftVariable QT
        UpdraftVariable QL
        UpdraftVariable H
        UpdraftVariable RH
        UpdraftVariable THL
        UpdraftVariable T
        UpdraftVariable B

        Py_ssize_t n_updrafts
        bint prognostic

        double [:] cloud_fraction
        double [:] cloud_base
        double [:] cloud_top
        double [:] updraft_top
        double [:] cloud_cover

        double updraft_fraction
        double lwp

    cpdef initialize(self, GridMeanVariables GMV)
    cpdef initialize_DryBubble(self, GridMeanVariables GMV, ReferenceState.ReferenceState Ref)
    cpdef initialize_io(self, NetCDFIO_Stats Stats)
    cpdef io(self, NetCDFIO_Stats Stats, ReferenceState.ReferenceState Ref)
    cpdef set_means(self, GridMeanVariables GMV)
    cpdef set_new_with_values(self)
    cpdef set_old_with_values(self)
    cpdef set_values_with_new(self)
    cpdef upd_cloud_diagnostics(self, ReferenceState.ReferenceState Ref)

cdef class UpdraftThermodynamics:
    cdef:
        double (*t_to_prog_fp)(double p0, double T, double qt, double ql, double qi) nogil
        double (*prog_to_t_fp)(double H, double pd, double pv, double qt ) nogil

        Grid.Grid Gr
        ReferenceState.ReferenceState Ref
        Py_ssize_t n_updraft

        double [:,:] prec_source_h
        double [:,:] prec_source_qt
        double [:] prec_source_h_tot
        double [:] prec_source_qt_tot

    cpdef buoyancy(
        self, UpdraftVariables UpdVar, EnvironmentVariables EnvVar,
        GridMeanVariables GMV, bint extrap
    )

    # helper functions to calculate autoconversion source terms to THL and QT
    cpdef clear_precip_sources(self)
    cpdef update_total_precip_sources(self)

    cpdef microphysics(self, UpdraftVariables UpdVar, RainVariables Rain, double dt)