1 | !-------------------------------------------------------------------------- |
---|
2 | !---------------------------- caldyn_solver ---------------------------------- |
---|
3 | IF (ll_begin==1) THEN |
---|
4 | !DIR$ SIMD |
---|
5 | DO ij=ij_begin_ext, ij_end_ext |
---|
6 | m_il(ij,1) = .5*rhodz(ij,1) |
---|
7 | END DO |
---|
8 | END IF |
---|
9 | DO l = ll_beginp1, ll_end |
---|
10 | !DIR$ SIMD |
---|
11 | DO ij=ij_begin_ext, ij_end_ext |
---|
12 | m_il(ij,l) = .5*(rhodz(ij,l)+rhodz(ij,l-1)) |
---|
13 | END DO |
---|
14 | END DO |
---|
15 | IF(ll_endp1==llm+1) THEN |
---|
16 | !DIR$ SIMD |
---|
17 | DO ij=ij_begin_ext, ij_end_ext |
---|
18 | m_il(ij,llm+1) = .5*rhodz(ij,llm+1 -1) |
---|
19 | END DO |
---|
20 | END IF |
---|
21 | ! |
---|
22 | IF(tau>0) THEN ! solve implicit problem for geopotential |
---|
23 | CALL compute_NH_geopot(tau,PHI_BOT_VAR, rhodz, m_il, theta, W, geopot) |
---|
24 | END IF |
---|
25 | ! |
---|
26 | ! Compute pressure (pres) and Exner function (pk) |
---|
27 | ! kappa = R/Cp |
---|
28 | ! 1-kappa = Cv/Cp |
---|
29 | ! Cp/Cv = 1/(1-kappa) |
---|
30 | gamma = 1./(1.-kappa) |
---|
31 | vreff = Rd*Treff/preff ! reference specific volume |
---|
32 | Cvd = cpp-Rd |
---|
33 | DO l = ll_begin, ll_end |
---|
34 | !DIR$ SIMD |
---|
35 | DO ij=ij_begin_ext, ij_end_ext |
---|
36 | SELECT CASE(caldyn_thermo) |
---|
37 | CASE(thermo_theta) |
---|
38 | rho_ij = g*rhodz(ij,l)/(geopot(ij,l+1)-geopot(ij,l)) |
---|
39 | X_ij = (cpp/preff)*kappa*theta(ij,l,1)*rho_ij |
---|
40 | ! kappa.theta.rho = p/exner |
---|
41 | ! => X = (p/p0)/(exner/Cp) |
---|
42 | ! = (p/p0)^(1-kappa) |
---|
43 | pres(ij,l) = preff*(X_ij**gamma) ! pressure |
---|
44 | ! Compute Exner function (needed by compute_caldyn_fast) |
---|
45 | ! other formulae possible if exponentiation is slow |
---|
46 | pk(ij,l) = cpp*((pres(ij,l)/preff)**kappa) ! Exner |
---|
47 | CASE(thermo_entropy) |
---|
48 | rho_ij = g*rhodz(ij,l)/(geopot(ij,l+1)-geopot(ij,l)) |
---|
49 | T_ij = Treff*exp( (theta(ij,l,1)+Rd*log(vreff*rho_ij))/Cvd ) |
---|
50 | pres(ij,l) = rho_ij*Rd*T_ij |
---|
51 | pk(ij,l) = T_ij |
---|
52 | CASE DEFAULT |
---|
53 | STOP |
---|
54 | END SELECT |
---|
55 | END DO |
---|
56 | END DO |
---|
57 | ! We need a barrier here because we compute pres above and do a vertical difference below |
---|
58 | !$OMP BARRIER |
---|
59 | IF (ll_begin==1) THEN |
---|
60 | !DIR$ SIMD |
---|
61 | DO ij=ij_begin_ext, ij_end_ext |
---|
62 | ! Lower BC |
---|
63 | dW(ij,1) = (1./g)*(pbot-rho_bot*(geopot(ij,1)-PHI_BOT(ij))-pres(ij,1)) - m_il(ij,1) |
---|
64 | W(ij,1) = W(ij,1)+tau*dW(ij,1) ! update W |
---|
65 | dPhi(ij,1) = g*g*W(ij,1)/m_il(ij,1) |
---|
66 | END DO |
---|
67 | END IF |
---|
68 | DO l = ll_beginp1, ll_end |
---|
69 | !DIR$ SIMD |
---|
70 | DO ij=ij_begin_ext, ij_end_ext |
---|
71 | dW(ij,l) = (1./g)*(pres(ij,l-1)-pres(ij,l)) - m_il(ij,l) |
---|
72 | W(ij,l) = W(ij,l)+tau*dW(ij,l) ! update W |
---|
73 | dPhi(ij,l) = g*g*W(ij,l)/m_il(ij,l) |
---|
74 | END DO |
---|
75 | END DO |
---|
76 | IF(ll_endp1==llm+1) THEN |
---|
77 | !DIR$ SIMD |
---|
78 | DO ij=ij_begin_ext, ij_end_ext |
---|
79 | ! Top BC |
---|
80 | dW(ij,llm+1) = (1./g)*(pres(ij,llm+1 -1)-ptop) - m_il(ij,llm+1) |
---|
81 | W(ij,llm+1) = W(ij,llm+1)+tau*dW(ij,llm+1) ! update W |
---|
82 | dPhi(ij,llm+1) = g*g*W(ij,llm+1)/m_il(ij,llm+1) |
---|
83 | END DO |
---|
84 | END IF |
---|
85 | ! We need a barrier here because we update W above and do a vertical average below |
---|
86 | !$OMP BARRIER |
---|
87 | DO l = ll_begin, ll_end |
---|
88 | !DIR$ SIMD |
---|
89 | DO ij=ij_begin_ext, ij_end_ext |
---|
90 | ! compute du = -0.5*g^2.grad(w^2) |
---|
91 | berni(ij,l) = (-.25*g*g)*((W(ij,l)/m_il(ij,l))**2 + (W(ij,l+1)/m_il(ij,l+1))**2 ) |
---|
92 | END DO |
---|
93 | END DO |
---|
94 | DO l = ll_begin, ll_end |
---|
95 | !DIR$ SIMD |
---|
96 | DO ij=ij_begin_ext, ij_end_ext |
---|
97 | du(ij+u_right,l) = ne_right*(berni(ij,l)-berni(ij+t_right,l)) |
---|
98 | du(ij+u_lup,l) = ne_lup*(berni(ij,l)-berni(ij+t_lup,l)) |
---|
99 | du(ij+u_ldown,l) = ne_ldown*(berni(ij,l)-berni(ij+t_ldown,l)) |
---|
100 | END DO |
---|
101 | END DO |
---|
102 | !---------------------------- caldyn_solver ---------------------------------- |
---|
103 | !-------------------------------------------------------------------------- |
---|