1 | MODULE euler_scheme_mod |
---|
2 | USE field_mod |
---|
3 | IMPLICIT NONE |
---|
4 | PRIVATE |
---|
5 | |
---|
6 | TYPE(t_field),POINTER,SAVE,PUBLIC :: f_geopot(:), f_q(:), & |
---|
7 | f_rhodz(:), f_mass(:), f_massm1(:), f_massm2(:), f_dmass(:), & ! current and previous time steps + tendency of mass, |
---|
8 | f_phis(:), f_ps(:),f_psm1(:), f_psm2(:), f_dps(:), & ! column-integrated mass |
---|
9 | f_u(:),f_um1(:),f_um2(:), f_du(:), f_W(:), & ! 'horizontal' and vertical (NH) momentum |
---|
10 | f_theta_rhodz(:),f_theta_rhodzm1(:),f_theta_rhodzm2(:), f_dtheta_rhodz(:), & ! mass-weighted theta/entropy |
---|
11 | f_hflux(:), f_wflux(:), f_hfluxt(:), f_wfluxt(:), & ! accumulated mass fluxes |
---|
12 | f_du_slow(:,:), f_du_fast(:,:), & ! slow/fast trend arrays |
---|
13 | f_dmass_slow(:,:), f_dps_slow(:,:), f_dtheta_rhodz_slow(:,:), &! for HEVI time scheme |
---|
14 | f_dPhi_slow(:,:), f_dPhi_fast(:,:), & ! geopotential tendencies (NH) |
---|
15 | f_dW_slow(:,:), f_dW_fast(:,:) ! vertical momentum tendencies (NH) |
---|
16 | |
---|
17 | INTEGER, PARAMETER, PUBLIC :: explicit=1, hevi=2, euler=1, rk4=2, mlf=3, rk25=4, ark23=6, ark33=7 |
---|
18 | |
---|
19 | INTEGER,SAVE, PUBLIC :: nb_stage, matsuno_period, scheme, scheme_family |
---|
20 | !$OMP THREADPRIVATE(nb_stage, matsuno_period, scheme, scheme_family) |
---|
21 | |
---|
22 | PUBLIC :: euler_scheme, accumulate_fluxes, legacy_to_DEC, DEC_to_legacy |
---|
23 | CONTAINS |
---|
24 | |
---|
25 | SUBROUTINE Euler_scheme(with_dps,fluxt_zero) |
---|
26 | USE icosa |
---|
27 | USE disvert_mod |
---|
28 | USE omp_para |
---|
29 | USE trace |
---|
30 | LOGICAL :: with_dps |
---|
31 | LOGICAL, OPTIONAL :: fluxt_zero(ndomain) ! set to .TRUE. to start accumulating fluxes in time |
---|
32 | REAL(rstd),POINTER :: ps(:), dps(:) |
---|
33 | REAL(rstd),POINTER :: u(:,:) , du(:,:) |
---|
34 | REAL(rstd),POINTER :: mass(:,:), dmass(:,:) |
---|
35 | REAL(rstd),POINTER :: theta_rhodz(:,:,:), dtheta_rhodz(:,:,:) |
---|
36 | REAL(rstd),POINTER :: hflux(:,:),wflux(:,:),hfluxt(:,:),wfluxt(:,:) |
---|
37 | INTEGER :: ind |
---|
38 | INTEGER :: i,j,ij,l |
---|
39 | CALL trace_start("Euler_scheme") |
---|
40 | |
---|
41 | DO ind=1,ndomain |
---|
42 | IF (.NOT. assigned_domain(ind)) CYCLE |
---|
43 | CALL swap_dimensions(ind) |
---|
44 | CALL swap_geometry(ind) |
---|
45 | |
---|
46 | IF(with_dps) THEN ! update ps/mass |
---|
47 | IF(caldyn_eta==eta_mass) THEN ! update ps |
---|
48 | ps=f_ps(ind) ; dps=f_dps(ind) ; |
---|
49 | IF (is_omp_first_level) THEN |
---|
50 | !DIR$ SIMD |
---|
51 | DO ij=ij_begin,ij_end |
---|
52 | ps(ij)=ps(ij)+dt*dps(ij) |
---|
53 | ENDDO |
---|
54 | ENDIF |
---|
55 | ELSE ! update mass |
---|
56 | mass=f_mass(ind) ; dmass=f_dmass(ind) ; |
---|
57 | DO l=ll_begin,ll_end |
---|
58 | !DIR$ SIMD |
---|
59 | DO ij=ij_begin,ij_end |
---|
60 | mass(ij,l)=mass(ij,l)+dt*dmass(ij,l) |
---|
61 | ENDDO |
---|
62 | END DO |
---|
63 | END IF |
---|
64 | |
---|
65 | hflux=f_hflux(ind); hfluxt=f_hfluxt(ind) |
---|
66 | wflux=f_wflux(ind); wfluxt=f_wfluxt(ind) |
---|
67 | CALL accumulate_fluxes(hflux,wflux,hfluxt,wfluxt,dt,fluxt_zero(ind)) |
---|
68 | END IF ! update ps/mass |
---|
69 | |
---|
70 | u=f_u(ind) ; theta_rhodz=f_theta_rhodz(ind) |
---|
71 | du=f_du(ind) ; dtheta_rhodz=f_dtheta_rhodz(ind) |
---|
72 | |
---|
73 | DO l=ll_begin,ll_end |
---|
74 | !DIR$ SIMD |
---|
75 | DO ij=ij_begin,ij_end |
---|
76 | u(ij+u_right,l)=u(ij+u_right,l)+dt*du(ij+u_right,l) |
---|
77 | u(ij+u_lup,l)=u(ij+u_lup,l)+dt*du(ij+u_lup,l) |
---|
78 | u(ij+u_ldown,l)=u(ij+u_ldown,l)+dt*du(ij+u_ldown,l) |
---|
79 | theta_rhodz(ij,l,1)=theta_rhodz(ij,l,1)+dt*dtheta_rhodz(ij,l,1) |
---|
80 | ENDDO |
---|
81 | ENDDO |
---|
82 | ENDDO |
---|
83 | |
---|
84 | CALL trace_end("Euler_scheme") |
---|
85 | |
---|
86 | END SUBROUTINE Euler_scheme |
---|
87 | |
---|
88 | SUBROUTINE accumulate_fluxes(hflux,wflux, hfluxt,wfluxt, tau,fluxt_zero) |
---|
89 | USE dimensions |
---|
90 | USE omp_para |
---|
91 | USE disvert_mod |
---|
92 | IMPLICIT NONE |
---|
93 | REAL(rstd), INTENT(IN) :: hflux(3*iim*jjm,llm), wflux(iim*jjm,llm+1) |
---|
94 | REAL(rstd), INTENT(INOUT) :: hfluxt(3*iim*jjm,llm), wfluxt(iim*jjm,llm+1) |
---|
95 | REAL(rstd), INTENT(IN) :: tau |
---|
96 | LOGICAL, INTENT(INOUT) :: fluxt_zero |
---|
97 | INTEGER :: l,i,j,ij |
---|
98 | |
---|
99 | IF(fluxt_zero) THEN |
---|
100 | |
---|
101 | fluxt_zero=.FALSE. |
---|
102 | |
---|
103 | DO l=ll_begin,ll_end |
---|
104 | !DIR$ SIMD |
---|
105 | DO ij=ij_begin_ext,ij_end_ext |
---|
106 | hfluxt(ij+u_right,l) = tau*hflux(ij+u_right,l) |
---|
107 | hfluxt(ij+u_lup,l) = tau*hflux(ij+u_lup,l) |
---|
108 | hfluxt(ij+u_ldown,l) = tau*hflux(ij+u_ldown,l) |
---|
109 | ENDDO |
---|
110 | ENDDO |
---|
111 | |
---|
112 | IF(caldyn_eta==eta_mass) THEN ! no need for vertical fluxes if eta_lag |
---|
113 | DO l=ll_begin,ll_endp1 |
---|
114 | !DIR$ SIMD |
---|
115 | DO ij=ij_begin,ij_end |
---|
116 | wfluxt(ij,l) = tau*wflux(ij,l) |
---|
117 | ENDDO |
---|
118 | ENDDO |
---|
119 | END IF |
---|
120 | |
---|
121 | ELSE |
---|
122 | |
---|
123 | DO l=ll_begin,ll_end |
---|
124 | !DIR$ SIMD |
---|
125 | DO ij=ij_begin_ext,ij_end_ext |
---|
126 | hfluxt(ij+u_right,l) = hfluxt(ij+u_right,l)+tau*hflux(ij+u_right,l) |
---|
127 | hfluxt(ij+u_lup,l) = hfluxt(ij+u_lup,l)+tau*hflux(ij+u_lup,l) |
---|
128 | hfluxt(ij+u_ldown,l) = hfluxt(ij+u_ldown,l)+tau*hflux(ij+u_ldown,l) |
---|
129 | ENDDO |
---|
130 | ENDDO |
---|
131 | |
---|
132 | IF(caldyn_eta==eta_mass) THEN ! no need for vertical fluxes if eta_lag |
---|
133 | DO l=ll_begin,ll_endp1 |
---|
134 | !DIR$ SIMD |
---|
135 | DO ij=ij_begin,ij_end |
---|
136 | wfluxt(ij,l) = wfluxt(ij,l)+tau*wflux(ij,l) |
---|
137 | ENDDO |
---|
138 | ENDDO |
---|
139 | END IF |
---|
140 | |
---|
141 | END IF |
---|
142 | |
---|
143 | END SUBROUTINE accumulate_fluxes |
---|
144 | |
---|
145 | SUBROUTINE legacy_to_DEC(f_ps, f_u) |
---|
146 | USE icosa |
---|
147 | USE disvert_mod |
---|
148 | USE omp_para |
---|
149 | USE trace |
---|
150 | TYPE(t_field),POINTER :: f_ps(:), f_u(:) |
---|
151 | REAL(rstd), POINTER :: ps(:), u(:,:) |
---|
152 | INTEGER :: ind,ij,l |
---|
153 | CALL trace_start("legacy_to_DEC") |
---|
154 | |
---|
155 | DO ind=1,ndomain |
---|
156 | IF (.NOT. assigned_domain(ind)) CYCLE |
---|
157 | CALL swap_dimensions(ind) |
---|
158 | CALL swap_geometry(ind) |
---|
159 | |
---|
160 | IF(caldyn_eta==eta_mass .AND. is_omp_first_level) THEN ! update ps |
---|
161 | ps=f_ps(ind) |
---|
162 | !DIR$ SIMD |
---|
163 | DO ij=ij_begin,ij_end |
---|
164 | ps(ij)=(ps(ij)-ptop)/g ! convert ps to column-integrated mass |
---|
165 | ENDDO |
---|
166 | END IF |
---|
167 | |
---|
168 | u=f_u(ind) |
---|
169 | DO l=ll_begin,ll_end |
---|
170 | !DIR$ SIMD |
---|
171 | DO ij=ij_begin,ij_end |
---|
172 | u(ij+u_right,l)=u(ij+u_right,l)*de(ij+u_right) |
---|
173 | u(ij+u_lup,l)=u(ij+u_lup,l)*de(ij+u_lup) |
---|
174 | u(ij+u_ldown,l)=u(ij+u_ldown,l)*de(ij+u_ldown) |
---|
175 | ENDDO |
---|
176 | ENDDO |
---|
177 | ENDDO |
---|
178 | |
---|
179 | CALL trace_end("legacy_to_DEC") |
---|
180 | END SUBROUTINE Legacy_to_DEC |
---|
181 | |
---|
182 | SUBROUTINE DEC_to_legacy(f_ps, f_u) |
---|
183 | USE icosa |
---|
184 | USE disvert_mod |
---|
185 | USE omp_para |
---|
186 | USE trace |
---|
187 | TYPE(t_field),POINTER :: f_ps(:), f_u(:) |
---|
188 | REAL(rstd), POINTER :: ps(:), u(:,:) |
---|
189 | INTEGER :: ind,ij,l |
---|
190 | CALL trace_start("legacy_to_DEC") |
---|
191 | |
---|
192 | DO ind=1,ndomain |
---|
193 | IF (.NOT. assigned_domain(ind)) CYCLE |
---|
194 | CALL swap_dimensions(ind) |
---|
195 | CALL swap_geometry(ind) |
---|
196 | |
---|
197 | IF(caldyn_eta==eta_mass .AND. is_omp_first_level) THEN |
---|
198 | ps=f_ps(ind) |
---|
199 | !DIR$ SIMD |
---|
200 | DO ij=ij_begin,ij_end |
---|
201 | ps(ij)=ptop+ps(ij)*g ! convert column-integrated mass to ps |
---|
202 | ENDDO |
---|
203 | ENDIF |
---|
204 | |
---|
205 | u=f_u(ind) |
---|
206 | DO l=ll_begin,ll_end |
---|
207 | !DIR$ SIMD |
---|
208 | DO ij=ij_begin,ij_end |
---|
209 | u(ij+u_right,l)=u(ij+u_right,l)/de(ij+u_right) |
---|
210 | u(ij+u_lup,l)=u(ij+u_lup,l)/de(ij+u_lup) |
---|
211 | u(ij+u_ldown,l)=u(ij+u_ldown,l)/de(ij+u_ldown) |
---|
212 | ENDDO |
---|
213 | ENDDO |
---|
214 | ENDDO |
---|
215 | |
---|
216 | CALL trace_end("DEC_to_legacy") |
---|
217 | END SUBROUTINE DEC_to_legacy |
---|
218 | |
---|
219 | END MODULE euler_scheme_mod |
---|