FE-Project
Loading...
Searching...
No Matches
Functions/Subroutines
scale_atm_dyn_dgm_nonhydro3d_etot_hevi Module Reference

module FElib / Fluid dyn solver / Atmosphere / Regional nonhydrostatic model / HEVI More...

Functions/Subroutines

subroutine, public atm_dyn_dgm_nonhydro3d_etot_hevi_init (mesh)
subroutine, public atm_dyn_dgm_nonhydro3d_etot_hevi_final ()
subroutine, public atm_dyn_dgm_nonhydro3d_etot_hevi_cal_tend (dens_dt, momx_dt, momy_dt, momz_dt, etot_dt, ddens_, momx_, momy_, momz_, etot_, dpres_, dens_hyd, pres_hyd, pres_hyd_ref, therm_hyd, coriolis, rtot, cvtot, cptot, dphyddx, dphyddy, element3d_operation, dx, dy, dz, sx, sy, sz, lift, lmesh, elem, lmesh2d, elem2d)
subroutine, public atm_dyn_dgm_nonhydro3d_etot_hevi_cal_vi (dens_dt, momx_dt, momy_dt, momz_dt, etot_dt, ddens_, momx_, momy_, momz_, etot_, dens_hyd, pres_hyd, ddens0_, momx0_, momy0_, momz0_, etot0_, rtot, cvtot, cptot, element3d_operation, dz, lift, impl_fac, dt, lmesh, elem, lmesh2d, elem2d)

Detailed Description

module FElib / Fluid dyn solver / Atmosphere / Regional nonhydrostatic model / HEVI

Description
HEVI DGM scheme for Atmospheric dynamical process The governing equations is a fully compressible nonhydrostatic equations, which consist of mass, momentum, and thermodynamics (total energy conservation) equations.
Author
Yuta Kawai, Team SCALE

Function/Subroutine Documentation

◆ atm_dyn_dgm_nonhydro3d_etot_hevi_init()

subroutine, public scale_atm_dyn_dgm_nonhydro3d_etot_hevi::atm_dyn_dgm_nonhydro3d_etot_hevi_init ( class(meshbase3d), intent(in) mesh)

Definition at line 73 of file scale_atm_dyn_dgm_nonhydro3d_etot_hevi.F90.

74
75 implicit none
76 class(MeshBase3D), intent(in) :: mesh
77 !--------------------------------------------
78
79 call atm_dyn_dgm_nonhydro3d_common_init( mesh )
80 return

References scale_atm_dyn_dgm_nonhydro3d_common::atm_dyn_dgm_nonhydro3d_common_init().

◆ atm_dyn_dgm_nonhydro3d_etot_hevi_final()

subroutine, public scale_atm_dyn_dgm_nonhydro3d_etot_hevi::atm_dyn_dgm_nonhydro3d_etot_hevi_final

Definition at line 84 of file scale_atm_dyn_dgm_nonhydro3d_etot_hevi.F90.

85 implicit none
86 !--------------------------------------------
87
88 call atm_dyn_dgm_nonhydro3d_common_final()
89 return

References scale_atm_dyn_dgm_nonhydro3d_common::atm_dyn_dgm_nonhydro3d_common_final().

◆ atm_dyn_dgm_nonhydro3d_etot_hevi_cal_tend()

subroutine, public scale_atm_dyn_dgm_nonhydro3d_etot_hevi::atm_dyn_dgm_nonhydro3d_etot_hevi_cal_tend ( real(rp), dimension(elem%np,lmesh%nea), intent(out) dens_dt,
real(rp), dimension(elem%np,lmesh%nea), intent(out) momx_dt,
real(rp), dimension(elem%np,lmesh%nea), intent(out) momy_dt,
real(rp), dimension(elem%np,lmesh%nea), intent(out) momz_dt,
real(rp), dimension(elem%np,lmesh%nea), intent(out) etot_dt,
real(rp), dimension(elem%np,lmesh%nea), intent(in) ddens_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) momx_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) momy_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) momz_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) etot_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) dpres_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) dens_hyd,
real(rp), dimension(elem%np,lmesh%nea), intent(in) pres_hyd,
real(rp), dimension(elem%np,lmesh%nea), intent(in) pres_hyd_ref,
real(rp), dimension(elem%np,lmesh%nea), intent(in) therm_hyd,
real(rp), dimension(elem2d%np,lmesh2d%nea), intent(in) coriolis,
real(rp), dimension(elem%np,lmesh%nea), intent(in) rtot,
real(rp), dimension(elem%np,lmesh%nea), intent(in) cvtot,
real(rp), dimension(elem%np,lmesh%nea), intent(in) cptot,
real(rp), dimension(elem%np,lmesh%nea), intent(in) dphyddx,
real(rp), dimension(elem%np,lmesh%nea), intent(in) dphyddy,
class(elementoperationbase3d), intent(in) element3d_operation,
type(sparsemat), intent(in) dx,
type(sparsemat), intent(in) dy,
type(sparsemat), intent(in) dz,
type(sparsemat), intent(in) sx,
type(sparsemat), intent(in) sy,
type(sparsemat), intent(in) sz,
type(sparsemat), intent(in) lift,
class(localmesh3d), intent(in) lmesh,
class(elementbase3d), intent(in) elem,
class(localmesh2d), intent(in) lmesh2d,
class(elementbase2d), intent(in) elem2d )

Definition at line 95 of file scale_atm_dyn_dgm_nonhydro3d_etot_hevi.F90.

102
103 use scale_atm_dyn_dgm_nonhydro3d_etot_hevi_numflux, only: &
104 get_ebnd_flux => atm_dyn_dgm_nonhydro3d_etot_hevi_numflux_get_generalvc
105 use, intrinsic :: ieee_arithmetic, only: isnan => ieee_is_nan
106 implicit none
107
108 class(LocalMesh3D), intent(in) :: lmesh
109 class(ElementBase3D), intent(in) :: elem
110 class(LocalMesh2D), intent(in) :: lmesh2D
111 class(ElementBase2D), intent(in) :: elem2D
112 class(ElementOperationBase3D), intent(in) :: element3D_operation
113 type(SparseMat), intent(in) :: Dx, Dy, Dz, Sx, Sy, Sz, Lift
114 real(RP), intent(out) :: DENS_dt(elem%Np,lmesh%NeA)
115 real(RP), intent(out) :: MOMX_dt(elem%Np,lmesh%NeA)
116 real(RP), intent(out) :: MOMY_dt(elem%Np,lmesh%NeA)
117 real(RP), intent(out) :: MOMZ_dt(elem%Np,lmesh%NeA)
118 real(RP), intent(out) :: ETOT_dt(elem%Np,lmesh%NeA)
119 real(RP), intent(in) :: DDENS_(elem%Np,lmesh%NeA)
120 real(RP), intent(in) :: MOMX_(elem%Np,lmesh%NeA)
121 real(RP), intent(in) :: MOMY_(elem%Np,lmesh%NeA)
122 real(RP), intent(in) :: MOMZ_(elem%Np,lmesh%NeA)
123 real(RP), intent(in) :: ETOT_(elem%Np,lmesh%NeA)
124 real(RP), intent(in) :: DPRES_(elem%Np,lmesh%NeA)
125 real(RP), intent(in) :: DENS_hyd(elem%Np,lmesh%NeA)
126 real(RP), intent(in) :: PRES_hyd(elem%Np,lmesh%NeA)
127 real(RP), intent(in) :: PRES_hyd_ref(elem%Np,lmesh%NeA)
128 real(RP), intent(in) :: THERM_hyd(elem%Np,lmesh%NeA)
129 real(RP), intent(in) :: CORIOLIS(elem2D%Np,lmesh2D%NeA)
130 real(RP), intent(in) :: Rtot(elem%Np,lmesh%NeA)
131 real(RP), intent(in) :: CVtot(elem%Np,lmesh%NeA)
132 real(RP), intent(in) :: CPtot(elem%Np,lmesh%NeA)
133 real(RP), intent(in) :: DPhydDx(elem%Np,lmesh%NeA)
134 real(RP), intent(in) :: DPhydDy(elem%Np,lmesh%NeA)
135
136 real(RP) :: Fx(elem%Np), Fy(elem%Np), Fz(elem%Np), LiftDelFlx(elem%Np)
137 real(RP) :: DPRES_hyd(elem%Np), GradPhyd_x(elem%Np), GradPhyd_y(elem%Np)
138 real(RP) :: del_flux(elem%NfpTot,lmesh%Ne,PRGVAR_NUM)
139 real(RP) :: del_flux_hyd(elem%NfpTot,lmesh%Ne,2)
140 real(RP) :: rdens_(elem%Np), u_(elem%Np), v_(elem%Np), w_(elem%Np), wt_(elem%Np)
141 real(RP) :: GsqrtV(elem%Np), RGsqrtV(elem%Np)
142 real(RP) :: Cori(elem%Np)
143
144 integer :: ke, ke2d
145
146 real(RP) :: gamm, rgamm
147 real(RP) :: rP0
148 real(RP) :: RovP0, P0ovR
149
150 real(RP) :: enthalpy_(elem%Np)
151
152 integer :: p
153 !------------------------------------------------------------------------
154
155 call prof_rapstart( 'cal_dyn_tend_bndflux', 3)
156 call get_ebnd_flux( &
157 del_flux, del_flux_hyd, & ! (out)
158 ddens_, momx_, momy_, momz_, etot_, dpres_, dens_hyd, pres_hyd, & ! (in)
159 rtot, cvtot, cptot, & ! (in)
160 lmesh%Gsqrt, lmesh%GI3(:,:,1), lmesh%GI3(:,:,2), lmesh%zlev(:,:), & ! (in)
161 lmesh%normal_fn(:,:,1), lmesh%normal_fn(:,:,2), lmesh%normal_fn(:,:,3), & ! (in)
162 lmesh%vmapM, lmesh%vmapP, & ! (in)
163 lmesh, elem, lmesh2d, elem2d ) ! (in)
164 call prof_rapend( 'cal_dyn_tend_bndflux', 3)
165
166 !-----
167 call prof_rapstart( 'cal_dyn_tend_interior', 3)
168 gamm = cpdry / cvdry
169 rgamm = cvdry / cpdry
170 rp0 = 1.0_rp / pres00
171 rovp0 = rdry * rp0
172 p0ovr = pres00 / rdry
173
174 !$omp parallel do private( ke2d, Cori, &
175 !$omp rdens_, u_, v_, w_, wt_, &
176 !$omp enthalpy_, &
177 !$omp DPRES_hyd, GradPhyd_x, GradPhyd_y, &
178 !$omp GsqrtV, RGsqrtV, &
179 !$omp Fx, Fy, Fz, LiftDelFlx )
180 do ke = lmesh%NeS, lmesh%NeE
181 !--
182 ke2d = lmesh%EMap3Dto2D(ke)
183 cori(:) = coriolis(elem%IndexH2Dto3D(:),ke2d)
184
185 gsqrtv(:) = lmesh%Gsqrt(:,ke) / lmesh%GsqrtH(elem%IndexH2Dto3D,ke2d)
186 rgsqrtv(:) = 1.0_rp / gsqrtv(:)
187
188 !--
189 rdens_(:) = 1.0_rp / (ddens_(:,ke) + dens_hyd(:,ke))
190 u_(:) = momx_(:,ke) * rdens_(:)
191 v_(:) = momy_(:,ke) * rdens_(:)
192 w_(:) = momz_(:,ke) * rdens_(:)
193 wt_(:) = w_(:) * rgsqrtv(:) + lmesh%GI3(:,ke,1) * u_(:) + lmesh%GI3(:,ke,2) * v_(:)
194
195 ! DPRES_(:) = ( CPtot(:,ke) / CVtot(:,ke) - 1.0_RP ) &
196 ! * ( ETOT_(:,ke) - Grav * ( DDENS_(:,ke) + DENS_hyd(:,ke) ) * lmesh%zlev(:,ke) &
197 ! - 0.5_RP * ( MOMX_(:,ke) * u_(:) + MOMY_(:,ke) * v_(:) + MOMZ_(:,ke) * w_(:) ) ) &
198 ! - PRES_hyd(:,ke)
199
200 enthalpy_(:) = etot_(:,ke) + pres_hyd(:,ke) + dpres_(:,ke)
201
202 !-- Gradient hydrostatic pressure
203
204 dpres_hyd(:) = pres_hyd(:,ke) - pres_hyd_ref(:,ke)
205
206 call sparsemat_matmul(dx, gsqrtv(:) * dpres_hyd(:), fx)
207 call sparsemat_matmul(dz, gsqrtv(:) * lmesh%GI3(:,ke,1) * dpres_hyd(:), fz)
208 call sparsemat_matmul(lift, lmesh%Fscale(:,ke) * del_flux_hyd(:,ke,1), liftdelflx)
209 gradphyd_x(:) = lmesh%Escale(:,ke,1,1) * fx(:) &
210 + lmesh%Escale(:,ke,3,3) * fz(:) &
211 + liftdelflx(:)
212
213 call sparsemat_matmul(dy, gsqrtv(:) * dpres_hyd(:), fy)
214 call sparsemat_matmul(dz, gsqrtv(:) * lmesh%GI3(:,ke,2) * dpres_hyd(:), fz)
215 call sparsemat_matmul(lift, lmesh%Fscale(:,ke) * del_flux_hyd(:,ke,2), liftdelflx)
216 gradphyd_y(:) = lmesh%Escale(:,ke,2,2) * fy(:) &
217 + lmesh%Escale(:,ke,3,3) * fz(:) &
218 + liftdelflx(:)
219
220 !-- DENS
221 call sparsemat_matmul(dx, lmesh%Gsqrt(:,ke) * momx_(:,ke), fx)
222 call sparsemat_matmul(dy, lmesh%Gsqrt(:,ke) * momy_(:,ke), fy)
223 call sparsemat_matmul(lift, lmesh%Fscale(:,ke) * del_flux(:,ke,dens_vid), liftdelflx)
224
225 dens_dt(:,ke) = - ( &
226 lmesh%Escale(:,ke,1,1) * fx(:) &
227 + lmesh%Escale(:,ke,2,2) * fy(:) &
228 + liftdelflx(:) ) / lmesh%Gsqrt(:,ke)
229
230 !-- MOMX
231 call sparsemat_matmul(dx, lmesh%Gsqrt(:,ke) * ( u_(:) * momx_(:,ke) + dpres_(:,ke) ), fx)
232 call sparsemat_matmul(dy, lmesh%Gsqrt(:,ke) * v_(:) * momx_(:,ke) , fy)
233 call sparsemat_matmul(dz, lmesh%Gsqrt(:,ke) * ( wt_(:) * momx_(:,ke) &
234 + lmesh%GI3(:,ke,1) * dpres_(:,ke) ), fz)
235 call sparsemat_matmul(lift, lmesh%Fscale(:,ke) * del_flux(:,ke,momx_vid), liftdelflx)
236
237 momx_dt(:,ke) = &
238 - ( lmesh%Escale(:,ke,1,1) * fx(:) &
239 + lmesh%Escale(:,ke,2,2) * fy(:) &
240 + lmesh%Escale(:,ke,3,3) * fz(:) &
241 + liftdelflx(:) ) / lmesh%Gsqrt(:,ke) &
242 - gradphyd_x(:) * rgsqrtv(:) &
243 + cori(:) * momy_(:,ke)
244
245 !-- MOMY
246 call sparsemat_matmul(dx, lmesh%Gsqrt(:,ke) * u_(:) * momy_(:,ke) , fx)
247 call sparsemat_matmul(dy, lmesh%Gsqrt(:,ke) * ( v_(:) * momy_(:,ke) + dpres_(:,ke) ), fy)
248 call sparsemat_matmul(dz, lmesh%Gsqrt(:,ke) * ( wt_(:) * momy_(:,ke) &
249 + lmesh%GI3(:,ke,2) * dpres_(:,ke) ), fz)
250 call sparsemat_matmul(lift, lmesh%Fscale(:,ke) * del_flux(:,ke,momy_vid), liftdelflx)
251
252 momy_dt(:,ke) = &
253 - ( lmesh%Escale(:,ke,1,1) * fx(:) &
254 + lmesh%Escale(:,ke,2,2) * fy(:) &
255 + lmesh%Escale(:,ke,3,3) * fz(:) &
256 + liftdelflx(:) ) / lmesh%Gsqrt(:,ke) &
257 - gradphyd_y(:) * rgsqrtv(:) &
258 - cori(:) * momx_(:,ke)
259
260 !-- MOMZ
261 call sparsemat_matmul(dx, lmesh%Gsqrt(:,ke) * u_(:) * momz_(:,ke), fx)
262 call sparsemat_matmul(dy, lmesh%Gsqrt(:,ke) * v_(:) * momz_(:,ke), fy)
263 call sparsemat_matmul(dz, lmesh%Gsqrt(:,ke) * wt_(:) * momz_(:,ke), fz)
264 call sparsemat_matmul(lift, lmesh%Fscale(:,ke) * del_flux(:,ke,momz_vid), liftdelflx)
265
266 momz_dt(:,ke) = &
267 - ( lmesh%Escale(:,ke,1,1) * fx(:) &
268 + lmesh%Escale(:,ke,2,2) * fy(:) &
269 + lmesh%Escale(:,ke,3,3) * fz(:) &
270 + liftdelflx(:) ) / lmesh%Gsqrt(:,ke)
271
272 !-- ETOT
273 call sparsemat_matmul(dx, lmesh%Gsqrt(:,ke) * u_(:) * enthalpy_(:), fx)
274 call sparsemat_matmul(dy, lmesh%Gsqrt(:,ke) * v_(:) * enthalpy_(:), fy)
275 call sparsemat_matmul(lift, lmesh%Fscale(:,ke) * del_flux(:,ke,etot_vid), liftdelflx)
276
277 etot_dt(:,ke) = &
278 - ( lmesh%Escale(:,ke,1,1) * fx(:) &
279 + lmesh%Escale(:,ke,2,2) * fy(:) &
280 + liftdelflx(:) ) / lmesh%Gsqrt(:,ke)
281
282 end do
283 call prof_rapend( 'cal_dyn_tend_interior', 3)
284
285 return
scale_atm_dyn_dgm_nonhydro3d_etot_hevi_numflux
module FElib / Fluid dyn solver / Atmosphere / Nonhydrostatic model / HEVI / Numflux
Definition scale_atm_dyn_dgm_nonhydro3d_etot_hevi_numflux.F90:11
scale_atm_dyn_dgm_nonhydro3d_etot_hevi_numflux::atm_dyn_dgm_nonhydro3d_etot_hevi_numflux_get_generalvc
subroutine, public atm_dyn_dgm_nonhydro3d_etot_hevi_numflux_get_generalvc(del_flux, del_flux_hyd, ddens_, momx_, momy_, momz_, etot_, dpres_, dens_hyd, pres_hyd, rtot, cvtot, cptot, gsqrt, g13, g23, zlev, nx, ny, nz, vmapm, vmapp, lmesh, elem, lmesh2d, elem2d)
Definition scale_atm_dyn_dgm_nonhydro3d_etot_hevi_numflux.F90:73

References scale_atm_dyn_dgm_nonhydro3d_etot_hevi_numflux::atm_dyn_dgm_nonhydro3d_etot_hevi_numflux_get_generalvc().

◆ atm_dyn_dgm_nonhydro3d_etot_hevi_cal_vi()

subroutine, public scale_atm_dyn_dgm_nonhydro3d_etot_hevi::atm_dyn_dgm_nonhydro3d_etot_hevi_cal_vi ( real(rp), dimension(elem%np,lmesh%nea), intent(out) dens_dt,
real(rp), dimension(elem%np,lmesh%nea), intent(out) momx_dt,
real(rp), dimension(elem%np,lmesh%nea), intent(out) momy_dt,
real(rp), dimension(elem%np,lmesh%nea), intent(out) momz_dt,
real(rp), dimension(elem%np,lmesh%nea), intent(out) etot_dt,
real(rp), dimension(elem%np,lmesh%nea), intent(in) ddens_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) momx_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) momy_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) momz_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) etot_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) dens_hyd,
real(rp), dimension(elem%np,lmesh%nea), intent(in) pres_hyd,
real(rp), dimension(elem%np,lmesh%nea), intent(in) ddens0_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) momx0_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) momy0_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) momz0_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) etot0_,
real(rp), dimension(elem%np,lmesh%nea), intent(in) rtot,
real(rp), dimension(elem%np,lmesh%nea), intent(in) cvtot,
real(rp), dimension(elem%np,lmesh%nea), intent(in) cptot,
class(elementoperationbase3d), intent(in) element3d_operation,
class(sparsemat), intent(in) dz,
class(sparsemat), intent(in) lift,
real(rp), intent(in) impl_fac,
real(rp), intent(in) dt,
class(localmesh3d), intent(in) lmesh,
class(elementbase3d), intent(in) elem,
class(localmesh2d), intent(in) lmesh2d,
class(elementbase2d), intent(in) elem2d )

Definition at line 291 of file scale_atm_dyn_dgm_nonhydro3d_etot_hevi.F90.

299
300 use scale_atm_dyn_dgm_nonhydro3d_etot_hevi_common, only: &
301 vi_eval_ax => atm_dyn_dgm_nonhydro3d_etot_hevi_common_eval_ax, &
302 vi_eval_ax_uv => atm_dyn_dgm_nonhydro3d_etot_hevi_common_eval_ax_uv, &
303 vi_construct_matbnd => atm_dyn_dgm_nonhydro3d_etot_hevi_common_construct_matbnd, &
304 vi_construct_matbnd_uv => atm_dyn_dgm_nonhydro3d_etot_hevi_common_construct_matbnd_uv
305
306 implicit none
307
308 class(LocalMesh3D), intent(in) :: lmesh
309 class(ElementBase3D), intent(in) :: elem
310 class(LocalMesh2D), intent(in) :: lmesh2D
311 class(ElementBase2D), intent(in) :: elem2D
312 real(RP), intent(out) :: DENS_dt(elem%Np,lmesh%NeA)
313 real(RP), intent(out) :: MOMX_dt(elem%Np,lmesh%NeA)
314 real(RP), intent(out) :: MOMY_dt(elem%Np,lmesh%NeA)
315 real(RP), intent(out) :: MOMZ_dt(elem%Np,lmesh%NeA)
316 real(RP), intent(out) :: ETOT_dt(elem%Np,lmesh%NeA)
317 real(RP), intent(in) :: DDENS_(elem%Np,lmesh%NeA)
318 real(RP), intent(in) :: MOMX_(elem%Np,lmesh%NeA)
319 real(RP), intent(in) :: MOMY_(elem%Np,lmesh%NeA)
320 real(RP), intent(in) :: MOMZ_(elem%Np,lmesh%NeA)
321 real(RP), intent(in) :: ETOT_(elem%Np,lmesh%NeA)
322 real(RP), intent(in) :: DENS_hyd(elem%Np,lmesh%NeA)
323 real(RP), intent(in) :: PRES_hyd(elem%Np,lmesh%NeA)
324 real(RP), intent(in) :: DDENS0_(elem%Np,lmesh%NeA)
325 real(RP), intent(in) :: MOMX0_(elem%Np,lmesh%NeA)
326 real(RP), intent(in) :: MOMY0_(elem%Np,lmesh%NeA)
327 real(RP), intent(in) :: MOMZ0_(elem%Np,lmesh%NeA)
328 real(RP), intent(in) :: ETOT0_(elem%Np,lmesh%NeA)
329 real(RP), intent(in) :: Rtot(elem%Np,lmesh%NeA)
330 real(RP), intent(in) :: CVtot(elem%Np,lmesh%NeA)
331 real(RP), intent(in) :: CPtot(elem%Np,lmesh%NeA)
332 class(ElementOperationBase3D), intent(in) :: element3D_operation
333 class(SparseMat), intent(in) :: Dz, Lift
334 real(RP), intent(in) :: impl_fac
335 real(RP), intent(in) :: dt
336
337 real(RP) :: PROG_VARS (elem%Np,lmesh%NeZ,PRGVAR_NUM,lmesh%NeX*lmesh%NeY)
338 real(RP) :: DPRES (elem%Np,lmesh%NeZ,lmesh%NeX*lmesh%NeY)
339 real(RP) :: PROG_VARS0(elem%Np,lmesh%NeZ,PRGVAR_NUM,lmesh%NeX*lmesh%NeY)
340 real(RP) :: DPRES0 (elem%Np,lmesh%NeZ,lmesh%NeX*lmesh%NeY)
341 real(RP) :: b1D(3,elem%Nnode_v,lmesh%NeZ,elem%Nnode_h1D**2,lmesh%NeX*lmesh%NeY)
342 real(RP) :: GeoPot (elem%Np,lmesh%NeZ,lmesh%NeX*lmesh%NeY)
343 real(RP) :: KinHovDENS(elem%Np,lmesh%NeZ,lmesh%NeX*lmesh%NeY)
344 integer :: ipiv(elem%Nnode_v*3*lmesh%NeZ,elem%Nnode_h1D**2)
345 real(RP) :: b1D_uv(elem%Nnode_v,lmesh%NeZ,2,elem%Nnode_h1D**2,lmesh%NeX*lmesh%NeY)
346 integer :: ipiv_uv(elem%Nnode_v*1*lmesh%NeZ,elem%Nnode_h1D**2)
347 real(RP) :: alph(elem%NfpTot,lmesh%NeZ,lmesh%NeX*lmesh%NeY)
348 real(RP) :: Rtot_z(elem%Np,lmesh%NeZ,lmesh%NeX*lmesh%NeY)
349 real(RP) :: CPtot_ov_CVtot(elem%Np,lmesh%NeZ,lmesh%NeX*lmesh%NeY)
350 real(RP) :: DENS_hyd_z(elem%Np,lmesh%NeZ,lmesh%NeX*lmesh%NeY)
351 real(RP) :: PRES_hyd_z(elem%Np,lmesh%NeZ,lmesh%NeX*lmesh%NeY)
352 real(RP) :: GnnM_z(elem%Np,lmesh%NeZ,lmesh%NeX*lmesh%NeY)
353 real(RP) :: G13_z(elem%Np,lmesh%NeZ,lmesh%NeX*lmesh%NeY)
354 real(RP) :: G23_z(elem%Np,lmesh%NeZ,lmesh%NeX*lmesh%NeY)
355 real(RP) :: GsqrtV_z(elem%Np,lmesh%NeZ,lmesh%NeX*lmesh%NeY)
356 real(RP) :: nz(elem%NfpTot,lmesh%NeZ,lmesh%NeX*lmesh%NeY)
357 integer :: vmapM(elem%NfpTot,lmesh%NeZ)
358 integer :: vmapP(elem%NfpTot,lmesh%NeZ)
359 integer :: ColMask(elem%Nnode_v)
360 integer :: ke_xy, ke_z, ke, ke2D, v
361 integer :: itr_nlin
362 integer :: kl, ku, nz_1D
363 integer :: kl_uv, ku_uv, nz_1D_uv
364 integer :: ij, info
365 logical :: is_converged
366
367 real(RP), allocatable :: PmatBnd(:,:,:)
368 real(RP), allocatable :: PmatBnd_uv(:,:,:)
369
370 real(RP) :: DENS_(elem%Np)
371 !------------------------------------------------------------------------
372
373 call prof_rapstart( 'hevi_cal_vi_prep', 3)
374
375 nz_1d = elem%Nnode_v * 3 * lmesh%NeZ
376 kl = ( elem%Nnode_v + 1 ) * 3 - 1
377 ku = kl
378 nz_1d_uv = elem%Nnode_v * 1 * lmesh%NeZ
379 kl_uv = elem%Nnode_v
380 ku_uv = kl_uv
381 allocate( pmatbnd(2*kl+ku+1,nz_1d,elem%Nnode_h1D**2) )
382 allocate( pmatbnd_uv(2*kl_uv+ku_uv+1,nz_1d_uv,elem%Nnode_h1D**2) )
383
384 call lmesh%GetVmapZ1D( vmapm, vmapp ) ! (out)
385
386 !-
387
388 !$omp parallel private( ke_xy, ke_z, ke, ke2D, DENS_ )
389 !$omp do collapse(2)
390 do ke_xy=1, lmesh%NeX*lmesh%NeY
391 do ke_z=1, lmesh%NeZ
392 ke = ke_xy + (ke_z-1)*lmesh%NeX*lmesh%NeY
393 ke2d = lmesh%EMap3Dto2D(ke)
394
395 prog_vars(:,ke_z,dens_vid,ke_xy) = ddens0_(:,ke)
396 prog_vars(:,ke_z,momx_vid,ke_xy) = momx0_(:,ke)
397 prog_vars(:,ke_z,momy_vid,ke_xy) = momy0_(:,ke)
398 prog_vars(:,ke_z,momz_vid,ke_xy) = momz0_(:,ke)
399 prog_vars(:,ke_z,etot_vid,ke_xy) = etot0_(:,ke)
400
401 dens_hyd_z(:,ke_z,ke_xy) = dens_hyd(:,ke)
402 pres_hyd_z(:,ke_z,ke_xy) = pres_hyd(:,ke)
403 geopot(:,ke_z,ke_xy) = grav * lmesh%zlev(:,ke)
404
405 dens_(:) = dens_hyd(:,ke) + ddens0_(:,ke)
406 kinhovdens(:,ke_z,ke_xy) = 0.5_rp * ( &
407 momx0_(:,ke) * ( lmesh%G_ij(elem%IndexH2Dto3D,ke2d,1,1) * momx0_(:,ke) + lmesh%G_ij(elem%IndexH2Dto3D,ke2d,2,1) * momy0_(:,ke) ) &
408 + momy0_(:,ke) * ( lmesh%G_ij(elem%IndexH2Dto3D,ke2d,2,1) * momx0_(:,ke) + lmesh%G_ij(elem%IndexH2Dto3D,ke2d,2,2) * momy0_(:,ke) ) &
409 ) / dens_(:)**2
410
411
412 rtot_z(:,ke_z,ke_xy) = rtot(:,ke)
413 cptot_ov_cvtot(:,ke_z,ke_xy) = cptot(:,ke) / cvtot(:,ke)
414
415 dpres(:,ke_z,ke_xy) = &
416 ( cptot_ov_cvtot(:,ke_z,ke_xy) - 1.0_rp ) &
417 * ( etot0_(:,ke) - ( dens_(:) * ( kinhovdens(:,ke_z,ke_xy) + geopot(:,ke_z,ke_xy) ) + 0.5_rp * momz0_(:,ke)**2 / dens_(:) ) ) &
418 - pres_hyd(:,ke)
419
420 nz(:,ke_z,ke_xy) = lmesh%normal_fn(:,ke,3)
421 g13_z(:,ke_z,ke_xy) = lmesh%GI3(:,ke,1)
422 g23_z(:,ke_z,ke_xy) = lmesh%GI3(:,ke,2)
423 gsqrtv_z(:,ke_z,ke_xy) = lmesh%Gsqrt(:,ke) / lmesh%GsqrtH(elem%IndexH2Dto3D,ke2d)
424
425 gnnm_z(:,ke_z,ke_xy) = ( 1.0_rp / gsqrtv_z(:,ke_z,ke_xy)**2 &
426 + g13_z(:,ke_z,ke_xy)**2 + g23_z(:,ke_z,ke_xy) )
427 end do
428 end do
429 !$omp end do
430 !$omp workshare
431 prog_vars0(:,:,:,:) = prog_vars(:,:,:,:)
432 dpres0(:,:,:) = dpres(:,:,:)
433 !$omp end workshare
434 !$omp end parallel
435
436 call prof_rapend( 'hevi_cal_vi_prep', 3)
437
438 !--
439
440 if ( abs(impl_fac) > 0.0_rp ) then
441 call prof_rapstart( 'hevi_cal_vi_itr', 3)
442
443 ! G = (q^n+1 - q^n*) + impl_fac * A(q^n+1) = 0
444 ! dG/dq^n+1 del[q] = - G(q^n*)
445 do itr_nlin = 1, 1
446 call prof_rapstart( 'hevi_cal_vi_ax', 3)
447
448 call vi_eval_ax_uv( &
449 momx_dt(:,:), momy_dt(:,:), alph(:,:,:), & ! (out, dummy)
450 prog_vars, dpres, prog_vars0, dpres0, & ! (in)
451 ddens_, momx_, momy_, momz_, etot_, & ! (in)
452 dens_hyd_z, pres_hyd_z, & ! (in)
453 rtot_z, cptot_ov_cvtot, & ! (in)
454 dz, lift, intrpmat_vpordm1, & ! (in)
455 gnnm_z, g13_z, g23_z, gsqrtv_z, & ! (in)
456 impl_fac, dt, & ! (in)
457 lmesh, elem, nz, vmapm, vmapp, & ! (in)
458 b1d_uv(:,:,:,:,:) ) ! (out)
459
460 call prof_rapend( 'hevi_cal_vi_ax', 3)
461
462 do ke_xy=1, lmesh%NeX * lmesh%NeY
463 call prof_rapstart( 'hevi_cal_vi_matbnd', 3)
464 call vi_construct_matbnd_uv( pmatbnd_uv(:,:,:), & ! (out)
465 kl_uv, ku_uv, nz_1d_uv, & ! (in)
466 prog_vars(:,:,:,ke_xy), kinhovdens(:,:,ke_xy), & ! (in)
467 dens_hyd_z(:,:,ke_xy), pres_hyd_z(:,:,ke_xy), & ! (in)
468 g13_z(:,:,ke_xy), g23_z(:,:,ke_xy), gsqrtv_z(:,:,ke_xy), & ! (in)
469 alph(:,:,ke_xy), & ! (in)
470 rtot_z(:,:,ke_xy), cptot_ov_cvtot(:,:,ke_xy), & ! (in)
471 geopot(:,:,ke_xy), & ! (in)
472 dz, lift, intrpmat_vpordm1, & ! (in)
473 impl_fac, dt, & ! (in)
474 lmesh, elem, nz(:,:,ke_xy), vmapm, vmapp, ke_xy, 1 ) ! (in)
475 call prof_rapend( 'hevi_cal_vi_matbnd', 3)
476
477 call prof_rapstart( 'hevi_cal_vi_lin', 3)
478 !$omp parallel private(ij, v, ke_z, info, ColMask)
479 !$omp do
480 do ij=1, elem%Nnode_h1D**2
481 call dgbsv( nz_1d_uv, kl_uv, ku_uv, 2, pmatbnd_uv(:,:,ij), 2*kl_uv+ku_uv+1, ipiv_uv(:,ij), b1d_uv(:,:,:,ij,ke_xy), nz_1d_uv, info)
482
483 colmask(:) = elem%Colmask(:,ij)
484 do ke_z=1, lmesh%NeZ
485 prog_vars(colmask(:),ke_z,momx_vid,ke_xy) = prog_vars(colmask(:),ke_z,momx_vid,ke_xy) + b1d_uv(:,ke_z,1,ij,ke_xy)
486 prog_vars(colmask(:),ke_z,momy_vid,ke_xy) = prog_vars(colmask(:),ke_z,momy_vid,ke_xy) + b1d_uv(:,ke_z,2,ij,ke_xy)
487 end do
488 end do ! for ij
489 !$omp end do
490 !$omp end parallel
491 call prof_rapend( 'hevi_cal_vi_lin', 3)
492
493 end do ! for ke_xy
494
495 call prof_rapstart( 'hevi_cal_vi_ax', 3)
496 call vi_eval_ax( &
497 dens_dt(:,:), momz_dt(:,:), etot_dt(:,:), & ! (out, dummy)
498 alph(:,:,:), & ! (in)
499 prog_vars, dpres, prog_vars0, dpres0, & ! (in)
500 ddens_, momx_, momy_, momz_, etot_, & ! (in)
501 dens_hyd_z, pres_hyd_z, & ! (in)
502 rtot_z, cptot_ov_cvtot, & ! (in)
503 dz, lift, intrpmat_vpordm1, & ! (in)
504 gnnm_z, g13_z, g23_z, gsqrtv_z, & ! (in)
505 impl_fac, dt, & ! (in)
506 lmesh, elem, nz, vmapm, vmapp, & ! (in)
507 b1d(:,:,:,:,:) ) ! (out)
508 call prof_rapend( 'hevi_cal_vi_ax', 3)
509
510 do ke_xy=1, lmesh%NeX * lmesh%NeY
511 call prof_rapstart( 'hevi_cal_vi_matbnd', 3)
512 call vi_construct_matbnd( pmatbnd(:,:,:), & ! (out)
513 kl, ku, nz_1d, & ! (in)
514 prog_vars(:,:,:,ke_xy), kinhovdens(:,:,ke_xy), & ! (in)
515 dens_hyd_z(:,:,ke_xy), pres_hyd_z(:,:,ke_xy), & ! (in)
516 g13_z(:,:,ke_xy), g23_z(:,:,ke_xy), gsqrtv_z(:,:,ke_xy), & ! (in)
517 alph(:,:,ke_xy), & ! (in)
518 rtot_z(:,:,ke_xy), cptot_ov_cvtot(:,:,ke_xy), & ! (in)
519 geopot(:,:,ke_xy), & ! (in)
520 dz, lift, intrpmat_vpordm1, & ! (in)
521 impl_fac, dt, & ! (in)
522 lmesh, elem, nz(:,:,ke_xy), vmapm, vmapp, ke_xy, 1 ) ! (in)
523 call prof_rapend( 'hevi_cal_vi_matbnd', 3)
524
525 call prof_rapstart( 'hevi_cal_vi_lin', 3)
526 !$omp parallel private(ij, v, ke_z, info, ColMask, DENS_)
527 !$omp do
528 do ij=1, elem%Nnode_h1D**2
529 call dgbsv( nz_1d, kl, ku, 1, pmatbnd(:,:,ij), 2*kl+ku+1, ipiv(:,ij), b1d(:,:,:,ij,ke_xy), nz_1d, info)
530
531 colmask(:) = elem%Colmask(:,ij)
532 do ke_z=1, lmesh%NeZ
533 prog_vars(colmask(:),ke_z,dens_vid,ke_xy) = prog_vars(colmask(:),ke_z,dens_vid,ke_xy) + b1d(1,:,ke_z,ij,ke_xy)
534 prog_vars(colmask(:),ke_z,momz_vid,ke_xy) = prog_vars(colmask(:),ke_z,momz_vid,ke_xy) + b1d(2,:,ke_z,ij,ke_xy)
535 prog_vars(colmask(:),ke_z,etot_vid,ke_xy) = prog_vars(colmask(:),ke_z,etot_vid,ke_xy) + b1d(3,:,ke_z,ij,ke_xy)
536 end do
537 end do ! for ij
538 !$omp end do
539 !$omp do
540 do ke_z=1, lmesh%NeZ
541 dens_(:) = dens_hyd_z(:,ke_z,ke_xy) + prog_vars(:,ke_z,dens_vid,ke_xy)
542 dpres(:,ke_z,ke_xy) = &
543 ( cptot_ov_cvtot(:,ke_z,ke_xy) - 1.0_rp ) &
544 * ( prog_vars(:,ke_z,etot_vid,ke_xy) &
545 - ( dens_(:) * ( kinhovdens(:,ke_z,ke_xy) + geopot(:,ke_z,ke_xy) ) + 0.5_rp * prog_vars(:,ke_z,momz_vid,ke_xy)**2 / dens_(:) ) ) &
546 - pres_hyd_z(:,ke_z,ke_xy)
547 end do
548 !$omp end parallel
549 call prof_rapend( 'hevi_cal_vi_lin', 3)
550
551 end do ! for ke_xy
552 end do ! itr nlin
553
554 call prof_rapend( 'hevi_cal_vi_itr', 3)
555 end if
556
557 call prof_rapstart( 'hevi_cal_vi_retrun_var', 3)
558 if ( abs(impl_fac) > 0.0_rp) then
559 !$omp parallel do collapse(2) private(ke_xy, ke_z, ke)
560 do ke_xy=1, lmesh%NeX * lmesh%NeY
561 do ke_z=1, lmesh%NeZ
562 ke = ke_xy + (ke_z-1)*lmesh%NeX*lmesh%NeY
563 dens_dt(:,ke) = ( prog_vars(:,ke_z,dens_vid,ke_xy) - ddens_(:,ke) ) / impl_fac
564 momx_dt(:,ke) = ( prog_vars(:,ke_z,momx_vid,ke_xy) - momx_(:,ke) ) / impl_fac
565 momy_dt(:,ke) = ( prog_vars(:,ke_z,momy_vid,ke_xy) - momy_(:,ke) ) / impl_fac
566 momz_dt(:,ke) = ( prog_vars(:,ke_z,momz_vid,ke_xy) - momz_(:,ke) ) / impl_fac
567 etot_dt(:,ke) = ( prog_vars(:,ke_z,etot_vid,ke_xy) - etot_(:,ke) ) / impl_fac
568 end do
569 end do
570 else
571 call vi_eval_ax_uv( &
572 momx_dt(:,:), momy_dt(:,:), alph(:,:,:), & ! (out)
573 prog_vars, dpres, prog_vars0, dpres0, & ! (in)
574 ddens_, momx_, momy_, momz_, etot_, & ! (in)
575 dens_hyd_z, pres_hyd_z, & ! (in)
576 rtot_z, cptot_ov_cvtot, & ! (in)
577 dz, lift, intrpmat_vpordm1, & ! (in)
578 gnnm_z, g13_z, g23_z, gsqrtv_z, & ! (in)
579 impl_fac, dt, & ! (in)
580 lmesh, elem, nz, vmapm, vmapp ) ! (in)
581
582 call vi_eval_ax( &
583 dens_dt(:,:), momz_dt(:,:), etot_dt(:,:), & ! (out)
584 alph(:,:,:), & ! (in)
585 prog_vars, dpres, prog_vars0, dpres0, & ! (in)
586 ddens_, momx_, momy_, momz_, etot_, & ! (in)
587 dens_hyd_z, pres_hyd_z, & ! (in)
588 rtot_z, cptot_ov_cvtot, & ! (in)
589 dz, lift, intrpmat_vpordm1, & ! (in)
590 gnnm_z, g13_z, g23_z, gsqrtv_z, & ! (in)
591 impl_fac, dt, & ! (in)
592 lmesh, elem, nz, vmapm, vmapp ) ! (in)
593 end if
594 call prof_rapend( 'hevi_cal_vi_retrun_var', 3)
595
596 return
scale_atm_dyn_dgm_nonhydro3d_etot_hevi_common
module FElib / Fluid dyn solver / Atmosphere / Nonhydrostatic model / HEVI / Common
Definition scale_atm_dyn_dgm_nonhydro3d_etot_hevi_common.F90:12
scale_atm_dyn_dgm_nonhydro3d_etot_hevi_common::atm_dyn_dgm_nonhydro3d_etot_hevi_common_construct_matbnd
subroutine, public atm_dyn_dgm_nonhydro3d_etot_hevi_common_construct_matbnd(pmatbnd, kl, ku, nz_1d, prog_vars0, kinhovdens00, dens_hyd, pres_hyd, g13, g23, gsqrtv, alph, rtot, cptot_ov_cvtot, geopot, dz, lift, intrpmat_vpordm1, impl_fac, dt, lmesh, elem, nz, vmapm, vmapp, ke_x, ke_y)
Definition scale_atm_dyn_dgm_nonhydro3d_etot_hevi_common.F90:355
scale_atm_dyn_dgm_nonhydro3d_etot_hevi_common::atm_dyn_dgm_nonhydro3d_etot_hevi_common_eval_ax
subroutine, public atm_dyn_dgm_nonhydro3d_etot_hevi_common_eval_ax(dens_t, momz_t, etot_t, alph, prog_vars, dpres, prog_vars0, dpres0, ddens00, momx00, momy00, momz00, entot00, dens_hyd, pres_hyd, rtot, cptot_ov_cvtot, dz, lift, intrpmat_vpordm1, gnnm, g13, g23, gsqrtv, impl_fac, dt, lmesh, elem, nz, vmapm, vmapp, b1d_ij)
Definition scale_atm_dyn_dgm_nonhydro3d_etot_hevi_common.F90:88
scale_atm_dyn_dgm_nonhydro3d_etot_hevi_common::atm_dyn_dgm_nonhydro3d_etot_hevi_common_eval_ax_uv
subroutine, public atm_dyn_dgm_nonhydro3d_etot_hevi_common_eval_ax_uv(momx_t, momy_t, alph, prog_vars, dpres, prog_vars0, dpres0, ddens00, momx00, momy00, momz00, entot00, dens_hyd, pres_hyd, rtot, cptot_ov_cvtot, dz, lift, intrpmat_vpordm1, gnnm, g13, g23, gsqrtv, impl_fac, dt, lmesh, elem, nz, vmapm, vmapp, b1d_ij_uv)
Definition scale_atm_dyn_dgm_nonhydro3d_etot_hevi_common.F90:237
scale_atm_dyn_dgm_nonhydro3d_etot_hevi_common::atm_dyn_dgm_nonhydro3d_etot_hevi_common_construct_matbnd_uv
subroutine, public atm_dyn_dgm_nonhydro3d_etot_hevi_common_construct_matbnd_uv(pmatbnd_uv, kl_uv, ku_uv, nz_1d_uv, prog_vars0, kinhovdens00, dens_hyd, pres_hyd, g13, g23, gsqrtv, alph, rtot, cptot_ov_cvtot, geopot, dz, lift, intrpmat_vpordm1, impl_fac, dt, lmesh, elem, nz, vmapm, vmapp, ke_x, ke_y)
Definition scale_atm_dyn_dgm_nonhydro3d_etot_hevi_common.F90:635

References scale_atm_dyn_dgm_nonhydro3d_etot_hevi_common::atm_dyn_dgm_nonhydro3d_etot_hevi_common_construct_matbnd(), scale_atm_dyn_dgm_nonhydro3d_etot_hevi_common::atm_dyn_dgm_nonhydro3d_etot_hevi_common_construct_matbnd_uv(), scale_atm_dyn_dgm_nonhydro3d_etot_hevi_common::atm_dyn_dgm_nonhydro3d_etot_hevi_common_eval_ax(), scale_atm_dyn_dgm_nonhydro3d_etot_hevi_common::atm_dyn_dgm_nonhydro3d_etot_hevi_common_eval_ax_uv(), and scale_atm_dyn_dgm_nonhydro3d_common::intrpmat_vpordm1.

Generated by doxygen 1.14.0