Home

Context Navigation

source: palm/trunk/SOURCE/prognostic_equations.f90 @ 4234

Last change on this file since 4234 was 4182, checked in by scharf, 5 years ago
corrected "Former revisions" section minor formatting in "Former revisions" section added "Author" section
Property svn:keywords set to `Id`
File size: 59.0 KB

Rev	Line
[1873]	1	!> @file prognostic_equations.f90
[2000]	2	!------------------------------------------------------------------------------!
[2696]	3	! This file is part of the PALM model system.
[1875]	4	!
[2000]	5	! PALM is free software: you can redistribute it and/or modify it under the
	6	! terms of the GNU General Public License as published by the Free Software
	7	! Foundation, either version 3 of the License, or (at your option) any later
	8	! version.
[1875]	9	!
	10	! PALM is distributed in the hope that it will be useful, but WITHOUT ANY
	11	! WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
	12	! A PARTICULAR PURPOSE. See the GNU General Public License for more details.
	13	!
	14	! You should have received a copy of the GNU General Public License along with
	15	! PALM. If not, see <http://www.gnu.org/licenses/>.
	16	!
[3655]	17	! Copyright 1997-2019 Leibniz Universitaet Hannover
[2000]	18	!------------------------------------------------------------------------------!
[1875]	19	!
	20	! Current revisions:
	21	! ------------------
[4110]	22	!
	23	!
	24	! Former revisions:
	25	! -----------------
	26	! $Id: prognostic_equations.f90 4182 2019-08-22 15:20:23Z gronemeier $
[4182]	27	! Corrected "Former revisions" section
	28	!
	29	! 4110 2019-07-22 17:05:21Z suehring
[4109]	30	! pass integer flag array to WS scalar advection routine which is now necessary
	31	! as the flags may differ for scalars, e.g. pt can be cyclic while chemical
	32	! species may be non-cyclic. Further, pass boundary flags.
[2156]	33	!
[4110]	34	! 4109 2019-07-22 17:00:34Z suehring
[4079]	35	! Application of monotonic flux limiter for the vertical scalar advection
	36	! up to the topography top (only for the cache-optimized version at the
	37	! moment). Please note, at the moment the limiter is only applied for passive
	38	! scalars.
	39	!
	40	! 4048 2019-06-21 21:00:21Z knoop
[4048]	41	! Moved tcm_prognostic_equations to module_interface
	42	!
	43	! 3987 2019-05-22 09:52:13Z kanani
[3987]	44	! Introduce alternative switch for debug output during timestepping
	45	!
	46	! 3956 2019-05-07 12:32:52Z monakurppa
[3956]	47	! Removed salsa calls.
	48	!
	49	! 3931 2019-04-24 16:34:28Z schwenkel
[3929]	50	! Correct/complete module_interface introduction for chemistry model
	51	!
	52	! 3899 2019-04-16 14:05:27Z monakurppa
[3899]	53	! Corrections in the OpenMP version of salsa
[3929]	54	!
	55	! 3887 2019 -04-12 08:47:41Z schwenkel
[3887]	56	! Implicit Bugfix for chemistry model, loop for non_transport_physics over
	57	! ghost points is avoided. Instead introducing module_interface_exchange_horiz.
	58	!
	59	! 3885 2019-04-11 11:29:34Z kanani
[3885]	60	! Changes related to global restructuring of location messages and introduction
	61	! of additional debug messages
	62	!
	63	! 3881 2019-04-10 09:31:22Z suehring
[3881]	64	! Bugfix in OpenMP directive
	65	!
	66	! 3880 2019-04-08 21:43:02Z knoop
[3875]	67	! Moved wtm_tendencies to module_interface_actions
	68	!
	69	! 3874 2019-04-08 16:53:48Z knoop
[3874]	70	! Added non_transport_physics module interfaces and moved bcm code into it
	71	!
	72	! 3872 2019-04-08 15:03:06Z knoop
[3870]	73	! Moving prognostic equations of bcm into bulk_cloud_model_mod
	74	!
	75	! 3864 2019-04-05 09:01:56Z monakurppa
[3864]	76	! Modifications made for salsa:
	77	! - salsa_prognostic_equations moved to salsa_mod (and the call to
	78	! module_interface_mod)
	79	! - Renamed nbins --> nbins_aerosol, ncc_tot --> ncomponents_mass and
	80	! ngast --> ngases_salsa and loop indices b, c and sg to ib, ic and ig
	81	!
	82	! 3840 2019-03-29 10:35:52Z knoop
[3879]	83	! added USE chem_gasphase_mod for nspec, nspec and spc_names
[3833]	84	!
	85	! 3820 2019-03-27 11:53:41Z forkel
[3820]	86	! renamed do_depo to deposition_dry (ecc)
	87	!
	88	! 3797 2019-03-15 11:15:38Z forkel
[3797]	89	! Call chem_integegrate in OpenMP loop (ketelsen)
	90	!
	91	!
	92	! 3771 2019-02-28 12:19:33Z raasch
[3771]	93	! preprocessor directivs fro rrtmg added
	94	!
	95	! 3761 2019-02-25 15:31:42Z raasch
[3761]	96	! unused variable removed
	97	!
	98	! 3719 2019-02-06 13:10:18Z kanani
[3719]	99	! Cleaned up chemistry cpu measurements
	100	!
	101	! 3684 2019-01-20 20:20:58Z knoop
[3634]	102	! OpenACC port for SPEC
[3458]	103	!
[4182]	104	! Revision 1.1 2000/04/13 14:56:27 schroeter
	105	! Initial revision
	106	!
	107	!
[1875]	108	! Description:
	109	! ------------
	110	!> Solving the prognostic equations.
	111	!------------------------------------------------------------------------------!
	112	MODULE prognostic_equations_mod
	113
[3294]	114	USE advec_s_bc_mod, &
	115	ONLY: advec_s_bc
[2155]	116
[3294]	117	USE advec_s_pw_mod, &
	118	ONLY: advec_s_pw
	119
	120	USE advec_s_up_mod, &
	121	ONLY: advec_s_up
	122
	123	USE advec_u_pw_mod, &
	124	ONLY: advec_u_pw
	125
	126	USE advec_u_up_mod, &
	127	ONLY: advec_u_up
	128
	129	USE advec_v_pw_mod, &
	130	ONLY: advec_v_pw
	131
	132	USE advec_v_up_mod, &
	133	ONLY: advec_v_up
	134
	135	USE advec_w_pw_mod, &
	136	ONLY: advec_w_pw
	137
	138	USE advec_w_up_mod, &
	139	ONLY: advec_w_up
	140
	141	USE advec_ws, &
	142	ONLY: advec_s_ws, advec_u_ws, advec_v_ws, advec_w_ws
	143
[1875]	144	USE arrays_3d, &
[3870]	145	ONLY: diss_l_e, diss_l_pt, diss_l_q, &
	146	diss_l_s, diss_l_sa, diss_s_e, &
	147	diss_s_pt, diss_s_q, diss_s_s, diss_s_sa, &
	148	e, e_p, flux_s_e, flux_s_pt, flux_s_q, &
	149	flux_s_s, flux_s_sa, flux_l_e, &
	150	flux_l_pt, flux_l_q, flux_l_s, &
	151	flux_l_sa, pt, ptdf_x, ptdf_y, pt_init, &
	152	pt_p, prho, q, q_init, q_p, rdf, rdf_sc, &
	153	ref_state, rho_ocean, s, s_init, s_p, tend, te_m, &
	154	tpt_m, tq_m, ts_m, tu_m, tv_m, tw_m, u, &
[3294]	155	ug, u_init, u_p, v, vg, vpt, v_init, v_p, w, w_p
[2155]	156
[3294]	157	USE buoyancy_mod, &
	158	ONLY: buoyancy
[3864]	159
[1875]	160	USE control_parameters, &
[4109]	161	ONLY: bc_dirichlet_l, &
	162	bc_dirichlet_n, &
	163	bc_dirichlet_r, &
	164	bc_dirichlet_s, &
	165	bc_radiation_l, &
	166	bc_radiation_n, &
	167	bc_radiation_r, &
	168	bc_radiation_s, &
	169	constant_diffusion, &
[3987]	170	debug_output_timestep, &
[2696]	171	dp_external, dp_level_ind_b, dp_smooth_factor, dpdxy, dt_3d, &
[3182]	172	humidity, intermediate_timestep_count, &
[1875]	173	intermediate_timestep_count_max, large_scale_forcing, &
[4079]	174	large_scale_subsidence, &
	175	monotonic_limiter_z, &
	176	neutral, nudging, &
[3294]	177	ocean_mode, passive_scalar, plant_canopy, pt_reference, &
[1875]	178	scalar_advec, scalar_advec, simulated_time, sloping_surface, &
[2232]	179	timestep_scheme, tsc, use_subsidence_tendencies, &
[2563]	180	use_upstream_for_tke, wind_turbine, ws_scheme_mom, &
[3467]	181	ws_scheme_sca, urban_surface, land_surface, &
[3582]	182	time_since_reference_point, salsa
[1875]	183
[3294]	184	USE coriolis_mod, &
	185	ONLY: coriolis
	186
[1875]	187	USE cpulog, &
[2696]	188	ONLY: cpu_log, log_point, log_point_s
[1875]	189
	190	USE diffusion_s_mod, &
[2118]	191	ONLY: diffusion_s
[1875]	192
	193	USE diffusion_u_mod, &
[2118]	194	ONLY: diffusion_u
[1875]	195
	196	USE diffusion_v_mod, &
[2118]	197	ONLY: diffusion_v
[1875]	198
	199	USE diffusion_w_mod, &
[2118]	200	ONLY: diffusion_w
[1875]	201
[3294]	202	USE indices, &
[4109]	203	ONLY: advc_flags_s, &
	204	nbgp, nxl, nxlg, nxlu, nxr, nxrg, nyn, nyng, nys, nysg, nysv, &
[3294]	205	nzb, nzt, wall_flags_0
	206
[1875]	207	USE kinds
	208
[2320]	209	USE lsf_nudging_mod, &
	210	ONLY: ls_advec, nudge
[1875]	211
[3684]	212	USE module_interface, &
[3837]	213	ONLY: module_interface_actions, &
[3931]	214	module_interface_non_advective_processes, &
[3887]	215	module_interface_exchange_horiz, &
[3837]	216	module_interface_prognostic_equations
[3684]	217
[3294]	218	USE ocean_mod, &
[3840]	219	ONLY: stokes_drift_terms, stokes_force, &
[3302]	220	wave_breaking, wave_breaking_term
[3294]	221
[1875]	222	USE plant_canopy_model_mod, &
[2746]	223	ONLY: cthf, pcm_tendency
[1875]	224
[3771]	225	#if defined( __rrtmg )
[1875]	226	USE radiation_model_mod, &
[1976]	227	ONLY: radiation, radiation_tendency, &
[1875]	228	skip_time_do_radiation
[3771]	229	#endif
[3864]	230
[1875]	231	USE statistics, &
	232	ONLY: hom
	233
	234	USE subsidence_mod, &
	235	ONLY: subsidence
	236
[3294]	237	USE surface_mod, &
	238	ONLY : surf_def_h, surf_def_v, surf_lsm_h, surf_lsm_v, surf_usm_h, &
	239	surf_usm_v
	240
[3874]	241	IMPLICIT NONE
[1914]	242
[1875]	243	PRIVATE
[2118]	244	PUBLIC prognostic_equations_cache, prognostic_equations_vector
[1875]	245
	246	INTERFACE prognostic_equations_cache
	247	MODULE PROCEDURE prognostic_equations_cache
	248	END INTERFACE prognostic_equations_cache
	249
	250	INTERFACE prognostic_equations_vector
	251	MODULE PROCEDURE prognostic_equations_vector
	252	END INTERFACE prognostic_equations_vector
	253
	254
	255	CONTAINS
	256
	257
	258	!------------------------------------------------------------------------------!
	259	! Description:
	260	! ------------
	261	!> Version with one optimized loop over all equations. It is only allowed to
	262	!> be called for the Wicker and Skamarock or Piascek-Williams advection scheme.
	263	!>
	264	!> Here the calls of most subroutines are embedded in two DO loops over i and j,
	265	!> so communication between CPUs is not allowed (does not make sense) within
	266	!> these loops.
	267	!>
	268	!> (Optimized to avoid cache missings, i.e. for Power4/5-architectures.)
	269	!------------------------------------------------------------------------------!
[2155]	270
[1875]	271	SUBROUTINE prognostic_equations_cache
	272
	273
	274	INTEGER(iwp) :: i !<
	275	INTEGER(iwp) :: i_omp_start !<
	276	INTEGER(iwp) :: j !<
	277	INTEGER(iwp) :: k !<
[3241]	278	!$ INTEGER(iwp) :: omp_get_thread_num !<
[1875]	279	INTEGER(iwp) :: tn = 0 !<
[2155]	280
[1875]	281	LOGICAL :: loop_start !<
	282
	283
[3987]	284	IF ( debug_output_timestep ) CALL debug_message( 'prognostic_equations_cache', 'start' )
[1875]	285	!
	286	!-- Time measurement can only be performed for the whole set of equations
	287	CALL cpu_log( log_point(32), 'all progn.equations', 'start' )
	288
[3878]	289	!$OMP PARALLEL PRIVATE (i,j)
	290	!$OMP DO
[3887]	291	DO i = nxl, nxr
	292	DO j = nys, nyn
[1875]	293	!
[3931]	294	!-- Calculate non advective processes for all other modules
	295	CALL module_interface_non_advective_processes( i, j )
[3878]	296	ENDDO
	297	ENDDO
[3887]	298	!
[3931]	299	!-- Module Inferface for exchange horiz after non_advective_processes but before
[3956]	300	!-- advection. Therefore, non_advective_processes must not run for ghost points.
	301	!$OMP END PARALLEL
[3887]	302	CALL module_interface_exchange_horiz()
[2696]	303	!
[2192]	304	!-- Loop over all prognostic equations
[3881]	305	!$OMP PARALLEL PRIVATE (i,i_omp_start,j,k,loop_start,tn)
[2192]	306
	307	!$ tn = omp_get_thread_num()
	308	loop_start = .TRUE.
	309
	310	!$OMP DO
[1875]	311	DO i = nxl, nxr
	312
	313	!
	314	!-- Store the first loop index. It differs for each thread and is required
	315	!-- later in advec_ws
	316	IF ( loop_start ) THEN
	317	loop_start = .FALSE.
[2155]	318	i_omp_start = i
[1875]	319	ENDIF
	320
	321	DO j = nys, nyn
	322	!
[3022]	323	!-- Tendency terms for u-velocity component. Please note, in case of
	324	!-- non-cyclic boundary conditions the grid point i=0 is excluded from
[3899]	325	!-- the prognostic equations for the u-component.
[3022]	326	IF ( i >= nxlu ) THEN
[1875]	327
	328	tend(:,j,i) = 0.0_wp
	329	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	330	IF ( ws_scheme_mom ) THEN
	331	CALL advec_u_ws( i, j, i_omp_start, tn )
[2155]	332	ELSE
[1875]	333	CALL advec_u_pw( i, j )
[2155]	334	ENDIF
[1875]	335	ELSE
	336	CALL advec_u_up( i, j )
	337	ENDIF
	338	CALL diffusion_u( i, j )
	339	CALL coriolis( i, j, 1 )
	340	IF ( sloping_surface .AND. .NOT. neutral ) THEN
	341	CALL buoyancy( i, j, pt, 1 )
	342	ENDIF
	343
	344	!
	345	!-- Drag by plant canopy
	346	IF ( plant_canopy ) CALL pcm_tendency( i, j, 1 )
	347
	348	!
	349	!-- External pressure gradient
	350	IF ( dp_external ) THEN
	351	DO k = dp_level_ind_b+1, nzt
	352	tend(k,j,i) = tend(k,j,i) - dpdxy(1) * dp_smooth_factor(k)
	353	ENDDO
	354	ENDIF
	355
	356	!
	357	!-- Nudging
	358	IF ( nudging ) CALL nudge( i, j, simulated_time, 'u' )
	359
[1914]	360	!
[3302]	361	!-- Effect of Stokes drift (in ocean mode only)
	362	IF ( stokes_force ) CALL stokes_drift_terms( i, j, 1 )
	363
[3684]	364	CALL module_interface_actions( i, j, 'u-tendency' )
[1875]	365	!
	366	!-- Prognostic equation for u-velocity component
[2232]	367	DO k = nzb+1, nzt
	368
	369	u_p(k,j,i) = u(k,j,i) + ( dt_3d * &
	370	( tsc(2) * tend(k,j,i) + &
	371	tsc(3) * tu_m(k,j,i) ) &
	372	- tsc(5) * rdf(k) &
	373	* ( u(k,j,i) - u_init(k) ) &
	374	) * MERGE( 1.0_wp, 0.0_wp, &
	375	BTEST( wall_flags_0(k,j,i), 1 )&
	376	)
[1875]	377	ENDDO
	378
	379	!
[3302]	380	!-- Add turbulence generated by wave breaking (in ocean mode only)
	381	IF ( wave_breaking .AND. &
	382	intermediate_timestep_count == intermediate_timestep_count_max )&
	383	THEN
	384	CALL wave_breaking_term( i, j, 1 )
	385	ENDIF
	386
	387	!
[1875]	388	!-- Calculate tendencies for the next Runge-Kutta step
	389	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	390	IF ( intermediate_timestep_count == 1 ) THEN
[2232]	391	DO k = nzb+1, nzt
[1875]	392	tu_m(k,j,i) = tend(k,j,i)
	393	ENDDO
	394	ELSEIF ( intermediate_timestep_count < &
	395	intermediate_timestep_count_max ) THEN
[2232]	396	DO k = nzb+1, nzt
	397	tu_m(k,j,i) = -9.5625_wp * tend(k,j,i) &
	398	+ 5.3125_wp * tu_m(k,j,i)
[1875]	399	ENDDO
	400	ENDIF
	401	ENDIF
	402
	403	ENDIF
	404	!
[3022]	405	!-- Tendency terms for v-velocity component. Please note, in case of
	406	!-- non-cyclic boundary conditions the grid point j=0 is excluded from
	407	!-- the prognostic equations for the v-component. !--
	408	IF ( j >= nysv ) THEN
[1875]	409
	410	tend(:,j,i) = 0.0_wp
	411	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	412	IF ( ws_scheme_mom ) THEN
	413	CALL advec_v_ws( i, j, i_omp_start, tn )
[2155]	414	ELSE
[1875]	415	CALL advec_v_pw( i, j )
	416	ENDIF
	417	ELSE
	418	CALL advec_v_up( i, j )
	419	ENDIF
	420	CALL diffusion_v( i, j )
	421	CALL coriolis( i, j, 2 )
	422
	423	!
	424	!-- Drag by plant canopy
[2155]	425	IF ( plant_canopy ) CALL pcm_tendency( i, j, 2 )
[1875]	426
	427	!
	428	!-- External pressure gradient
	429	IF ( dp_external ) THEN
	430	DO k = dp_level_ind_b+1, nzt
	431	tend(k,j,i) = tend(k,j,i) - dpdxy(2) * dp_smooth_factor(k)
	432	ENDDO
	433	ENDIF
	434
	435	!
	436	!-- Nudging
	437	IF ( nudging ) CALL nudge( i, j, simulated_time, 'v' )
	438
[1914]	439	!
[3302]	440	!-- Effect of Stokes drift (in ocean mode only)
	441	IF ( stokes_force ) CALL stokes_drift_terms( i, j, 2 )
	442
[3684]	443	CALL module_interface_actions( i, j, 'v-tendency' )
[1875]	444	!
	445	!-- Prognostic equation for v-velocity component
[2232]	446	DO k = nzb+1, nzt
	447	v_p(k,j,i) = v(k,j,i) + ( dt_3d * &
	448	( tsc(2) * tend(k,j,i) + &
	449	tsc(3) * tv_m(k,j,i) ) &
	450	- tsc(5) * rdf(k) &
	451	* ( v(k,j,i) - v_init(k) )&
	452	) * MERGE( 1.0_wp, 0.0_wp, &
	453	BTEST( wall_flags_0(k,j,i), 2 )&
	454	)
[1875]	455	ENDDO
	456
	457	!
[3302]	458	!-- Add turbulence generated by wave breaking (in ocean mode only)
	459	IF ( wave_breaking .AND. &
	460	intermediate_timestep_count == intermediate_timestep_count_max )&
	461	THEN
	462	CALL wave_breaking_term( i, j, 2 )
	463	ENDIF
	464
	465	!
[1875]	466	!-- Calculate tendencies for the next Runge-Kutta step
	467	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	468	IF ( intermediate_timestep_count == 1 ) THEN
[2232]	469	DO k = nzb+1, nzt
[1875]	470	tv_m(k,j,i) = tend(k,j,i)
	471	ENDDO
	472	ELSEIF ( intermediate_timestep_count < &
	473	intermediate_timestep_count_max ) THEN
[2232]	474	DO k = nzb+1, nzt
	475	tv_m(k,j,i) = -9.5625_wp * tend(k,j,i) &
	476	+ 5.3125_wp * tv_m(k,j,i)
[1875]	477	ENDDO
	478	ENDIF
	479	ENDIF
	480
	481	ENDIF
	482
	483	!
	484	!-- Tendency terms for w-velocity component
	485	tend(:,j,i) = 0.0_wp
	486	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	487	IF ( ws_scheme_mom ) THEN
	488	CALL advec_w_ws( i, j, i_omp_start, tn )
[2155]	489	ELSE
[1875]	490	CALL advec_w_pw( i, j )
	491	END IF
	492	ELSE
	493	CALL advec_w_up( i, j )
	494	ENDIF
	495	CALL diffusion_w( i, j )
	496	CALL coriolis( i, j, 3 )
	497
	498	IF ( .NOT. neutral ) THEN
[3294]	499	IF ( ocean_mode ) THEN
[2031]	500	CALL buoyancy( i, j, rho_ocean, 3 )
[1875]	501	ELSE
	502	IF ( .NOT. humidity ) THEN
	503	CALL buoyancy( i, j, pt, 3 )
	504	ELSE
	505	CALL buoyancy( i, j, vpt, 3 )
	506	ENDIF
	507	ENDIF
	508	ENDIF
	509
	510	!
	511	!-- Drag by plant canopy
	512	IF ( plant_canopy ) CALL pcm_tendency( i, j, 3 )
	513
[1914]	514	!
[3302]	515	!-- Effect of Stokes drift (in ocean mode only)
	516	IF ( stokes_force ) CALL stokes_drift_terms( i, j, 3 )
	517
[3684]	518	CALL module_interface_actions( i, j, 'w-tendency' )
[1875]	519	!
	520	!-- Prognostic equation for w-velocity component
[2232]	521	DO k = nzb+1, nzt-1
	522	w_p(k,j,i) = w(k,j,i) + ( dt_3d * &
	523	( tsc(2) * tend(k,j,i) + &
[1875]	524	tsc(3) * tw_m(k,j,i) ) &
[2232]	525	- tsc(5) * rdf(k) * w(k,j,i) &
	526	) * MERGE( 1.0_wp, 0.0_wp, &
	527	BTEST( wall_flags_0(k,j,i), 3 )&
	528	)
[1875]	529	ENDDO
	530
	531	!
	532	!-- Calculate tendencies for the next Runge-Kutta step
	533	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	534	IF ( intermediate_timestep_count == 1 ) THEN
[2232]	535	DO k = nzb+1, nzt-1
[1875]	536	tw_m(k,j,i) = tend(k,j,i)
	537	ENDDO
	538	ELSEIF ( intermediate_timestep_count < &
	539	intermediate_timestep_count_max ) THEN
[2232]	540	DO k = nzb+1, nzt-1
	541	tw_m(k,j,i) = -9.5625_wp * tend(k,j,i) &
	542	+ 5.3125_wp * tw_m(k,j,i)
[1875]	543	ENDDO
	544	ENDIF
	545	ENDIF
	546
	547	!
	548	!-- If required, compute prognostic equation for potential temperature
	549	IF ( .NOT. neutral ) THEN
	550	!
	551	!-- Tendency terms for potential temperature
	552	tend(:,j,i) = 0.0_wp
	553	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	554	IF ( ws_scheme_sca ) THEN
[4109]	555	CALL advec_s_ws( advc_flags_s, &
	556	i, j, pt, 'pt', flux_s_pt, diss_s_pt, &
	557	flux_l_pt, diss_l_pt, i_omp_start, tn, &
	558	bc_dirichlet_l .OR. bc_radiation_l, &
	559	bc_dirichlet_n .OR. bc_radiation_n, &
	560	bc_dirichlet_r .OR. bc_radiation_r, &
	561	bc_dirichlet_s .OR. bc_radiation_s )
[1875]	562	ELSE
	563	CALL advec_s_pw( i, j, pt )
	564	ENDIF
	565	ELSE
	566	CALL advec_s_up( i, j, pt )
	567	ENDIF
[2232]	568	CALL diffusion_s( i, j, pt, &
	569	surf_def_h(0)%shf, surf_def_h(1)%shf, &
	570	surf_def_h(2)%shf, &
	571	surf_lsm_h%shf, surf_usm_h%shf, &
	572	surf_def_v(0)%shf, surf_def_v(1)%shf, &
	573	surf_def_v(2)%shf, surf_def_v(3)%shf, &
	574	surf_lsm_v(0)%shf, surf_lsm_v(1)%shf, &
	575	surf_lsm_v(2)%shf, surf_lsm_v(3)%shf, &
	576	surf_usm_v(0)%shf, surf_usm_v(1)%shf, &
	577	surf_usm_v(2)%shf, surf_usm_v(3)%shf )
[1875]	578
	579	!
	580	!-- Consideration of heat sources within the plant canopy
[3014]	581	IF ( plant_canopy .AND. &
	582	(cthf /= 0.0_wp .OR. urban_surface .OR. land_surface) ) THEN
[1875]	583	CALL pcm_tendency( i, j, 4 )
	584	ENDIF
	585
	586	!
	587	!-- Large scale advection
	588	IF ( large_scale_forcing ) THEN
	589	CALL ls_advec( i, j, simulated_time, 'pt' )
[2155]	590	ENDIF
[1875]	591
	592	!
	593	!-- Nudging
[2155]	594	IF ( nudging ) CALL nudge( i, j, simulated_time, 'pt' )
[1875]	595
	596	!
	597	!-- If required, compute effect of large-scale subsidence/ascent
	598	IF ( large_scale_subsidence .AND. &
	599	.NOT. use_subsidence_tendencies ) THEN
	600	CALL subsidence( i, j, tend, pt, pt_init, 2 )
	601	ENDIF
	602
[3771]	603	#if defined( __rrtmg )
[1875]	604	!
	605	!-- If required, add tendency due to radiative heating/cooling
[1976]	606	IF ( radiation .AND. &
[1875]	607	simulated_time > skip_time_do_radiation ) THEN
	608	CALL radiation_tendency ( i, j, tend )
	609	ENDIF
[3771]	610	#endif
[1875]	611
[3684]	612	CALL module_interface_actions( i, j, 'pt-tendency' )
[1875]	613	!
	614	!-- Prognostic equation for potential temperature
[2232]	615	DO k = nzb+1, nzt
	616	pt_p(k,j,i) = pt(k,j,i) + ( dt_3d * &
	617	( tsc(2) * tend(k,j,i) + &
[1875]	618	tsc(3) * tpt_m(k,j,i) ) &
[2232]	619	- tsc(5) &
	620	* ( pt(k,j,i) - pt_init(k) ) &
	621	* ( rdf_sc(k) + ptdf_x(i) &
	622	+ ptdf_y(j) ) &
	623	) &
	624	* MERGE( 1.0_wp, 0.0_wp, &
	625	BTEST( wall_flags_0(k,j,i), 0 )&
	626	)
[1875]	627	ENDDO
	628
	629	!
	630	!-- Calculate tendencies for the next Runge-Kutta step
	631	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	632	IF ( intermediate_timestep_count == 1 ) THEN
[2232]	633	DO k = nzb+1, nzt
[1875]	634	tpt_m(k,j,i) = tend(k,j,i)
	635	ENDDO
	636	ELSEIF ( intermediate_timestep_count < &
	637	intermediate_timestep_count_max ) THEN
[2232]	638	DO k = nzb+1, nzt
	639	tpt_m(k,j,i) = -9.5625_wp * tend(k,j,i) + &
	640	5.3125_wp * tpt_m(k,j,i)
[1875]	641	ENDDO
	642	ENDIF
	643	ENDIF
	644
	645	ENDIF
	646
	647	!
[1960]	648	!-- If required, compute prognostic equation for total water content
	649	IF ( humidity ) THEN
[1875]	650
	651	!
	652	!-- Tendency-terms for total water content / scalar
	653	tend(:,j,i) = 0.0_wp
[4109]	654	IF ( timestep_scheme(1:5) == 'runge' ) &
[1875]	655	THEN
	656	IF ( ws_scheme_sca ) THEN
[4109]	657	CALL advec_s_ws( advc_flags_s, &
	658	i, j, q, 'q', flux_s_q, &
	659	diss_s_q, flux_l_q, diss_l_q, &
	660	i_omp_start, tn, &
	661	bc_dirichlet_l .OR. bc_radiation_l, &
	662	bc_dirichlet_n .OR. bc_radiation_n, &
	663	bc_dirichlet_r .OR. bc_radiation_r, &
	664	bc_dirichlet_s .OR. bc_radiation_s )
[2155]	665	ELSE
[1875]	666	CALL advec_s_pw( i, j, q )
	667	ENDIF
	668	ELSE
	669	CALL advec_s_up( i, j, q )
	670	ENDIF
[2232]	671	CALL diffusion_s( i, j, q, &
	672	surf_def_h(0)%qsws, surf_def_h(1)%qsws, &
	673	surf_def_h(2)%qsws, &
	674	surf_lsm_h%qsws, surf_usm_h%qsws, &
	675	surf_def_v(0)%qsws, surf_def_v(1)%qsws, &
	676	surf_def_v(2)%qsws, surf_def_v(3)%qsws, &
	677	surf_lsm_v(0)%qsws, surf_lsm_v(1)%qsws, &
	678	surf_lsm_v(2)%qsws, surf_lsm_v(3)%qsws, &
	679	surf_usm_v(0)%qsws, surf_usm_v(1)%qsws, &
	680	surf_usm_v(2)%qsws, surf_usm_v(3)%qsws )
[1875]	681
	682	!
[1960]	683	!-- Sink or source of humidity due to canopy elements
[1875]	684	IF ( plant_canopy ) CALL pcm_tendency( i, j, 5 )
	685
	686	!
	687	!-- Large scale advection
	688	IF ( large_scale_forcing ) THEN
	689	CALL ls_advec( i, j, simulated_time, 'q' )
	690	ENDIF
	691
	692	!
	693	!-- Nudging
[2155]	694	IF ( nudging ) CALL nudge( i, j, simulated_time, 'q' )
[1875]	695
	696	!
	697	!-- If required compute influence of large-scale subsidence/ascent
	698	IF ( large_scale_subsidence .AND. &
	699	.NOT. use_subsidence_tendencies ) THEN
	700	CALL subsidence( i, j, tend, q, q_init, 3 )
	701	ENDIF
	702
[3684]	703	CALL module_interface_actions( i, j, 'q-tendency' )
[1875]	704
	705	!
	706	!-- Prognostic equation for total water content / scalar
[2232]	707	DO k = nzb+1, nzt
	708	q_p(k,j,i) = q(k,j,i) + ( dt_3d * &
	709	( tsc(2) * tend(k,j,i) + &
[1875]	710	tsc(3) * tq_m(k,j,i) ) &
[2232]	711	- tsc(5) * rdf_sc(k) * &
	712	( q(k,j,i) - q_init(k) ) &
	713	) &
	714	* MERGE( 1.0_wp, 0.0_wp, &
	715	BTEST( wall_flags_0(k,j,i), 0 )&
	716	)
[1875]	717	IF ( q_p(k,j,i) < 0.0_wp ) q_p(k,j,i) = 0.1_wp * q(k,j,i)
	718	ENDDO
	719
	720	!
	721	!-- Calculate tendencies for the next Runge-Kutta step
	722	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	723	IF ( intermediate_timestep_count == 1 ) THEN
[2232]	724	DO k = nzb+1, nzt
[1875]	725	tq_m(k,j,i) = tend(k,j,i)
	726	ENDDO
	727	ELSEIF ( intermediate_timestep_count < &
	728	intermediate_timestep_count_max ) THEN
[2232]	729	DO k = nzb+1, nzt
	730	tq_m(k,j,i) = -9.5625_wp * tend(k,j,i) + &
	731	5.3125_wp * tq_m(k,j,i)
[1875]	732	ENDDO
	733	ENDIF
	734	ENDIF
	735
	736	ENDIF
[2155]	737
[1960]	738	!
	739	!-- If required, compute prognostic equation for scalar
	740	IF ( passive_scalar ) THEN
	741	!
	742	!-- Tendency-terms for total water content / scalar
	743	tend(:,j,i) = 0.0_wp
[4109]	744	IF ( timestep_scheme(1:5) == 'runge' ) &
[1960]	745	THEN
	746	IF ( ws_scheme_sca ) THEN
[4079]	747	!
	748	!-- For scalar advection apply monotonic flux limiter near
	749	!-- topography.
[4109]	750	CALL advec_s_ws( advc_flags_s, &
	751	i, j, s, 's', flux_s_s, &
[4079]	752	diss_s_s, flux_l_s, diss_l_s, i_omp_start, &
[4109]	753	tn, &
	754	bc_dirichlet_l .OR. bc_radiation_l, &
	755	bc_dirichlet_n .OR. bc_radiation_n, &
	756	bc_dirichlet_r .OR. bc_radiation_r, &
	757	bc_dirichlet_s .OR. bc_radiation_s, &
	758	monotonic_limiter_z )
[2155]	759	ELSE
[1960]	760	CALL advec_s_pw( i, j, s )
	761	ENDIF
	762	ELSE
	763	CALL advec_s_up( i, j, s )
	764	ENDIF
[2232]	765	CALL diffusion_s( i, j, s, &
	766	surf_def_h(0)%ssws, surf_def_h(1)%ssws, &
	767	surf_def_h(2)%ssws, &
	768	surf_lsm_h%ssws, surf_usm_h%ssws, &
	769	surf_def_v(0)%ssws, surf_def_v(1)%ssws, &
	770	surf_def_v(2)%ssws, surf_def_v(3)%ssws, &
	771	surf_lsm_v(0)%ssws, surf_lsm_v(1)%ssws, &
	772	surf_lsm_v(2)%ssws, surf_lsm_v(3)%ssws, &
	773	surf_usm_v(0)%ssws, surf_usm_v(1)%ssws, &
	774	surf_usm_v(2)%ssws, surf_usm_v(3)%ssws )
[1875]	775
	776	!
[1960]	777	!-- Sink or source of scalar concentration due to canopy elements
	778	IF ( plant_canopy ) CALL pcm_tendency( i, j, 7 )
	779
	780	!
	781	!-- Large scale advection, still need to be extended for scalars
	782	! IF ( large_scale_forcing ) THEN
	783	! CALL ls_advec( i, j, simulated_time, 's' )
	784	! ENDIF
	785
	786	!
	787	!-- Nudging, still need to be extended for scalars
[2155]	788	! IF ( nudging ) CALL nudge( i, j, simulated_time, 's' )
[1960]	789
	790	!
	791	!-- If required compute influence of large-scale subsidence/ascent.
[2155]	792	!-- Note, the last argument is of no meaning in this case, as it is
	793	!-- only used in conjunction with large_scale_forcing, which is to
[1960]	794	!-- date not implemented for scalars.
	795	IF ( large_scale_subsidence .AND. &
	796	.NOT. use_subsidence_tendencies .AND. &
	797	.NOT. large_scale_forcing ) THEN
	798	CALL subsidence( i, j, tend, s, s_init, 3 )
	799	ENDIF
	800
[3684]	801	CALL module_interface_actions( i, j, 's-tendency' )
[1960]	802
	803	!
	804	!-- Prognostic equation for scalar
[2232]	805	DO k = nzb+1, nzt
	806	s_p(k,j,i) = s(k,j,i) + ( dt_3d * &
	807	( tsc(2) * tend(k,j,i) + &
	808	tsc(3) * ts_m(k,j,i) ) &
	809	- tsc(5) * rdf_sc(k) &
	810	* ( s(k,j,i) - s_init(k) )&
	811	) &
	812	* MERGE( 1.0_wp, 0.0_wp, &
	813	BTEST( wall_flags_0(k,j,i), 0 )&
	814	)
[1960]	815	IF ( s_p(k,j,i) < 0.0_wp ) s_p(k,j,i) = 0.1_wp * s(k,j,i)
	816	ENDDO
	817
	818	!
	819	!-- Calculate tendencies for the next Runge-Kutta step
	820	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	821	IF ( intermediate_timestep_count == 1 ) THEN
[2232]	822	DO k = nzb+1, nzt
[1960]	823	ts_m(k,j,i) = tend(k,j,i)
	824	ENDDO
	825	ELSEIF ( intermediate_timestep_count < &
	826	intermediate_timestep_count_max ) THEN
[2232]	827	DO k = nzb+1, nzt
	828	ts_m(k,j,i) = -9.5625_wp * tend(k,j,i) + &
	829	5.3125_wp * ts_m(k,j,i)
[1960]	830	ENDDO
	831	ENDIF
	832	ENDIF
	833
[2155]	834	ENDIF
[1960]	835	!
[3837]	836	!-- Calculate prognostic equations for all other modules
	837	CALL module_interface_prognostic_equations( i, j, i_omp_start, tn )
[1875]	838
[3294]	839	ENDDO ! loop over j
	840	ENDDO ! loop over i
[2192]	841	!$OMP END PARALLEL
[1875]	842
[2232]	843
[1875]	844	CALL cpu_log( log_point(32), 'all progn.equations', 'stop' )
	845
[3987]	846	IF ( debug_output_timestep ) CALL debug_message( 'prognostic_equations_cache', 'end' )
[1875]	847
	848	END SUBROUTINE prognostic_equations_cache
	849
	850
	851	!------------------------------------------------------------------------------!
	852	! Description:
	853	! ------------
	854	!> Version for vector machines
	855	!------------------------------------------------------------------------------!
[2155]	856
[1875]	857	SUBROUTINE prognostic_equations_vector
	858
	859
[2815]	860	INTEGER(iwp) :: i !<
	861	INTEGER(iwp) :: j !<
	862	INTEGER(iwp) :: k !<
[1875]	863
	864	REAL(wp) :: sbt !<
	865
[3885]	866
[3987]	867	IF ( debug_output_timestep ) CALL debug_message( 'prognostic_equations_vector', 'start' )
[3467]	868	!
[3931]	869	!-- Calculate non advective processes for all other modules
	870	CALL module_interface_non_advective_processes
[1875]	871	!
[3931]	872	!-- Module Inferface for exchange horiz after non_advective_processes but before
[3956]	873	!-- advection. Therefore, non_advective_processes must not run for ghost points.
[3887]	874	CALL module_interface_exchange_horiz()
	875	!
[1875]	876	!-- u-velocity component
	877	CALL cpu_log( log_point(5), 'u-equation', 'start' )
	878
[3634]	879	!$ACC KERNELS PRESENT(tend)
[1875]	880	tend = 0.0_wp
[3634]	881	!$ACC END KERNELS
[1875]	882	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	883	IF ( ws_scheme_mom ) THEN
	884	CALL advec_u_ws
	885	ELSE
	886	CALL advec_u_pw
	887	ENDIF
	888	ELSE
	889	CALL advec_u_up
	890	ENDIF
	891	CALL diffusion_u
	892	CALL coriolis( 1 )
	893	IF ( sloping_surface .AND. .NOT. neutral ) THEN
	894	CALL buoyancy( pt, 1 )
	895	ENDIF
	896
	897	!
	898	!-- Drag by plant canopy
	899	IF ( plant_canopy ) CALL pcm_tendency( 1 )
	900
	901	!
	902	!-- External pressure gradient
	903	IF ( dp_external ) THEN
	904	DO i = nxlu, nxr
	905	DO j = nys, nyn
	906	DO k = dp_level_ind_b+1, nzt
	907	tend(k,j,i) = tend(k,j,i) - dpdxy(1) * dp_smooth_factor(k)
	908	ENDDO
	909	ENDDO
	910	ENDDO
	911	ENDIF
	912
	913	!
	914	!-- Nudging
	915	IF ( nudging ) CALL nudge( simulated_time, 'u' )
	916
[1914]	917	!
[3302]	918	!-- Effect of Stokes drift (in ocean mode only)
	919	IF ( stokes_force ) CALL stokes_drift_terms( 1 )
	920
[3684]	921	CALL module_interface_actions( 'u-tendency' )
[1875]	922
	923	!
	924	!-- Prognostic equation for u-velocity component
[3634]	925	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
	926	!$ACC PRESENT(u, tend, tu_m, u_init, rdf, wall_flags_0) &
	927	!$ACC PRESENT(tsc(2:5)) &
	928	!$ACC PRESENT(u_p)
[1875]	929	DO i = nxlu, nxr
	930	DO j = nys, nyn
[2232]	931	DO k = nzb+1, nzt
	932	u_p(k,j,i) = u(k,j,i) + ( dt_3d * ( tsc(2) * tend(k,j,i) + &
	933	tsc(3) * tu_m(k,j,i) ) &
	934	- tsc(5) * rdf(k) * &
	935	( u(k,j,i) - u_init(k) ) &
	936	) * MERGE( 1.0_wp, 0.0_wp, &
	937	BTEST( wall_flags_0(k,j,i), 1 ) &
	938	)
[1875]	939	ENDDO
	940	ENDDO
	941	ENDDO
	942
	943	!
[3302]	944	!-- Add turbulence generated by wave breaking (in ocean mode only)
	945	IF ( wave_breaking .AND. &
	946	intermediate_timestep_count == intermediate_timestep_count_max ) &
	947	THEN
	948	CALL wave_breaking_term( 1 )
	949	ENDIF
	950
	951	!
[1875]	952	!-- Calculate tendencies for the next Runge-Kutta step
	953	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	954	IF ( intermediate_timestep_count == 1 ) THEN
[3634]	955	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
	956	!$ACC PRESENT(tend, tu_m)
[1875]	957	DO i = nxlu, nxr
	958	DO j = nys, nyn
[2232]	959	DO k = nzb+1, nzt
[1875]	960	tu_m(k,j,i) = tend(k,j,i)
	961	ENDDO
	962	ENDDO
	963	ENDDO
	964	ELSEIF ( intermediate_timestep_count < &
	965	intermediate_timestep_count_max ) THEN
[3634]	966	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
	967	!$ACC PRESENT(tend, tu_m)
[1875]	968	DO i = nxlu, nxr
	969	DO j = nys, nyn
[2232]	970	DO k = nzb+1, nzt
	971	tu_m(k,j,i) = -9.5625_wp * tend(k,j,i) &
	972	+ 5.3125_wp * tu_m(k,j,i)
[1875]	973	ENDDO
	974	ENDDO
	975	ENDDO
	976	ENDIF
	977	ENDIF
	978
	979	CALL cpu_log( log_point(5), 'u-equation', 'stop' )
	980
	981	!
	982	!-- v-velocity component
	983	CALL cpu_log( log_point(6), 'v-equation', 'start' )
	984
[3634]	985	!$ACC KERNELS PRESENT(tend)
[1875]	986	tend = 0.0_wp
[3634]	987	!$ACC END KERNELS
[1875]	988	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	989	IF ( ws_scheme_mom ) THEN
	990	CALL advec_v_ws
[2155]	991	ELSE
[1875]	992	CALL advec_v_pw
	993	END IF
	994	ELSE
	995	CALL advec_v_up
	996	ENDIF
	997	CALL diffusion_v
	998	CALL coriolis( 2 )
	999
	1000	!
	1001	!-- Drag by plant canopy
	1002	IF ( plant_canopy ) CALL pcm_tendency( 2 )
	1003
	1004	!
	1005	!-- External pressure gradient
	1006	IF ( dp_external ) THEN
	1007	DO i = nxl, nxr
	1008	DO j = nysv, nyn
	1009	DO k = dp_level_ind_b+1, nzt
	1010	tend(k,j,i) = tend(k,j,i) - dpdxy(2) * dp_smooth_factor(k)
	1011	ENDDO
	1012	ENDDO
	1013	ENDDO
	1014	ENDIF
	1015
	1016	!
	1017	!-- Nudging
	1018	IF ( nudging ) CALL nudge( simulated_time, 'v' )
	1019
[1914]	1020	!
[3302]	1021	!-- Effect of Stokes drift (in ocean mode only)
	1022	IF ( stokes_force ) CALL stokes_drift_terms( 2 )
	1023
[3684]	1024	CALL module_interface_actions( 'v-tendency' )
[1875]	1025
	1026	!
	1027	!-- Prognostic equation for v-velocity component
[3634]	1028	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
	1029	!$ACC PRESENT(v, tend, tv_m, v_init, rdf, wall_flags_0) &
	1030	!$ACC PRESENT(tsc(2:5)) &
	1031	!$ACC PRESENT(v_p)
[1875]	1032	DO i = nxl, nxr
	1033	DO j = nysv, nyn
[2232]	1034	DO k = nzb+1, nzt
	1035	v_p(k,j,i) = v(k,j,i) + ( dt_3d * ( tsc(2) * tend(k,j,i) + &
	1036	tsc(3) * tv_m(k,j,i) ) &
	1037	- tsc(5) * rdf(k) * &
	1038	( v(k,j,i) - v_init(k) ) &
	1039	) * MERGE( 1.0_wp, 0.0_wp, &
	1040	BTEST( wall_flags_0(k,j,i), 2 )&
	1041	)
[1875]	1042	ENDDO
	1043	ENDDO
	1044	ENDDO
	1045
	1046	!
[3302]	1047	!-- Add turbulence generated by wave breaking (in ocean mode only)
	1048	IF ( wave_breaking .AND. &
	1049	intermediate_timestep_count == intermediate_timestep_count_max ) &
	1050	THEN
	1051	CALL wave_breaking_term( 2 )
	1052	ENDIF
	1053
	1054	!
[1875]	1055	!-- Calculate tendencies for the next Runge-Kutta step
	1056	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	1057	IF ( intermediate_timestep_count == 1 ) THEN
[3634]	1058	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
	1059	!$ACC PRESENT(tend, tv_m)
[1875]	1060	DO i = nxl, nxr
	1061	DO j = nysv, nyn
[2232]	1062	DO k = nzb+1, nzt
[1875]	1063	tv_m(k,j,i) = tend(k,j,i)
	1064	ENDDO
	1065	ENDDO
	1066	ENDDO
	1067	ELSEIF ( intermediate_timestep_count < &
	1068	intermediate_timestep_count_max ) THEN
[3634]	1069	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
	1070	!$ACC PRESENT(tend, tv_m)
[1875]	1071	DO i = nxl, nxr
	1072	DO j = nysv, nyn
[2232]	1073	DO k = nzb+1, nzt
	1074	tv_m(k,j,i) = -9.5625_wp * tend(k,j,i) &
	1075	+ 5.3125_wp * tv_m(k,j,i)
[1875]	1076	ENDDO
	1077	ENDDO
	1078	ENDDO
	1079	ENDIF
	1080	ENDIF
	1081
	1082	CALL cpu_log( log_point(6), 'v-equation', 'stop' )
	1083
	1084	!
	1085	!-- w-velocity component
	1086	CALL cpu_log( log_point(7), 'w-equation', 'start' )
	1087
[3634]	1088	!$ACC KERNELS PRESENT(tend)
[1875]	1089	tend = 0.0_wp
[3634]	1090	!$ACC END KERNELS
[1875]	1091	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	1092	IF ( ws_scheme_mom ) THEN
	1093	CALL advec_w_ws
	1094	ELSE
	1095	CALL advec_w_pw
	1096	ENDIF
	1097	ELSE
	1098	CALL advec_w_up
	1099	ENDIF
	1100	CALL diffusion_w
	1101	CALL coriolis( 3 )
	1102
	1103	IF ( .NOT. neutral ) THEN
[3294]	1104	IF ( ocean_mode ) THEN
[2031]	1105	CALL buoyancy( rho_ocean, 3 )
[1875]	1106	ELSE
	1107	IF ( .NOT. humidity ) THEN
	1108	CALL buoyancy( pt, 3 )
	1109	ELSE
	1110	CALL buoyancy( vpt, 3 )
	1111	ENDIF
	1112	ENDIF
	1113	ENDIF
	1114
	1115	!
	1116	!-- Drag by plant canopy
	1117	IF ( plant_canopy ) CALL pcm_tendency( 3 )
	1118
[1914]	1119	!
[3302]	1120	!-- Effect of Stokes drift (in ocean mode only)
	1121	IF ( stokes_force ) CALL stokes_drift_terms( 3 )
	1122
[3684]	1123	CALL module_interface_actions( 'w-tendency' )
[1875]	1124
	1125	!
	1126	!-- Prognostic equation for w-velocity component
[3634]	1127	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
	1128	!$ACC PRESENT(w, tend, tw_m, v_init, rdf, wall_flags_0) &
	1129	!$ACC PRESENT(tsc(2:5)) &
	1130	!$ACC PRESENT(w_p)
[1875]	1131	DO i = nxl, nxr
	1132	DO j = nys, nyn
[2232]	1133	DO k = nzb+1, nzt-1
	1134	w_p(k,j,i) = w(k,j,i) + ( dt_3d * ( tsc(2) * tend(k,j,i) + &
	1135	tsc(3) * tw_m(k,j,i) ) &
	1136	- tsc(5) * rdf(k) * w(k,j,i) &
	1137	) * MERGE( 1.0_wp, 0.0_wp, &
	1138	BTEST( wall_flags_0(k,j,i), 3 )&
	1139	)
[1875]	1140	ENDDO
	1141	ENDDO
	1142	ENDDO
	1143
	1144	!
	1145	!-- Calculate tendencies for the next Runge-Kutta step
	1146	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	1147	IF ( intermediate_timestep_count == 1 ) THEN
[3634]	1148	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
	1149	!$ACC PRESENT(tend, tw_m)
[1875]	1150	DO i = nxl, nxr
	1151	DO j = nys, nyn
[2232]	1152	DO k = nzb+1, nzt-1
[1875]	1153	tw_m(k,j,i) = tend(k,j,i)
	1154	ENDDO
	1155	ENDDO
	1156	ENDDO
	1157	ELSEIF ( intermediate_timestep_count < &
	1158	intermediate_timestep_count_max ) THEN
[3634]	1159	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
	1160	!$ACC PRESENT(tend, tw_m)
[1875]	1161	DO i = nxl, nxr
	1162	DO j = nys, nyn
[2232]	1163	DO k = nzb+1, nzt-1
	1164	tw_m(k,j,i) = -9.5625_wp * tend(k,j,i) &
	1165	+ 5.3125_wp * tw_m(k,j,i)
[1875]	1166	ENDDO
	1167	ENDDO
	1168	ENDDO
	1169	ENDIF
	1170	ENDIF
	1171
	1172	CALL cpu_log( log_point(7), 'w-equation', 'stop' )
	1173
	1174
	1175	!
	1176	!-- If required, compute prognostic equation for potential temperature
	1177	IF ( .NOT. neutral ) THEN
	1178
	1179	CALL cpu_log( log_point(13), 'pt-equation', 'start' )
	1180
	1181	!
	1182	!-- pt-tendency terms with communication
	1183	sbt = tsc(2)
	1184	IF ( scalar_advec == 'bc-scheme' ) THEN
	1185
	1186	IF ( timestep_scheme(1:5) /= 'runge' ) THEN
	1187	!
	1188	!-- Bott-Chlond scheme always uses Euler time step. Thus:
	1189	sbt = 1.0_wp
	1190	ENDIF
	1191	tend = 0.0_wp
	1192	CALL advec_s_bc( pt, 'pt' )
	1193
	1194	ENDIF
	1195
	1196	!
	1197	!-- pt-tendency terms with no communication
	1198	IF ( scalar_advec /= 'bc-scheme' ) THEN
[3634]	1199	!$ACC KERNELS PRESENT(tend)
[1875]	1200	tend = 0.0_wp
[3634]	1201	!$ACC END KERNELS
[1875]	1202	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	1203	IF ( ws_scheme_sca ) THEN
[4109]	1204	CALL advec_s_ws( advc_flags_s, pt, 'pt', &
	1205	bc_dirichlet_l .OR. bc_radiation_l, &
	1206	bc_dirichlet_n .OR. bc_radiation_n, &
	1207	bc_dirichlet_r .OR. bc_radiation_r, &
	1208	bc_dirichlet_s .OR. bc_radiation_s )
[1875]	1209	ELSE
	1210	CALL advec_s_pw( pt )
	1211	ENDIF
	1212	ELSE
	1213	CALL advec_s_up( pt )
	1214	ENDIF
	1215	ENDIF
	1216
[2232]	1217	CALL diffusion_s( pt, &
	1218	surf_def_h(0)%shf, surf_def_h(1)%shf, &
	1219	surf_def_h(2)%shf, &
	1220	surf_lsm_h%shf, surf_usm_h%shf, &
	1221	surf_def_v(0)%shf, surf_def_v(1)%shf, &
	1222	surf_def_v(2)%shf, surf_def_v(3)%shf, &
	1223	surf_lsm_v(0)%shf, surf_lsm_v(1)%shf, &
	1224	surf_lsm_v(2)%shf, surf_lsm_v(3)%shf, &
	1225	surf_usm_v(0)%shf, surf_usm_v(1)%shf, &
	1226	surf_usm_v(2)%shf, surf_usm_v(3)%shf )
[1875]	1227
	1228	!
	1229	!-- Consideration of heat sources within the plant canopy
[3014]	1230	IF ( plant_canopy .AND. &
	1231	(cthf /= 0.0_wp .OR. urban_surface .OR. land_surface) ) THEN
[1875]	1232	CALL pcm_tendency( 4 )
	1233	ENDIF
	1234
	1235	!
	1236	!-- Large scale advection
	1237	IF ( large_scale_forcing ) THEN
	1238	CALL ls_advec( simulated_time, 'pt' )
	1239	ENDIF
	1240
	1241	!
	1242	!-- Nudging
[2155]	1243	IF ( nudging ) CALL nudge( simulated_time, 'pt' )
[1875]	1244
	1245	!
	1246	!-- If required compute influence of large-scale subsidence/ascent
	1247	IF ( large_scale_subsidence .AND. &
	1248	.NOT. use_subsidence_tendencies ) THEN
	1249	CALL subsidence( tend, pt, pt_init, 2 )
	1250	ENDIF
	1251
[3771]	1252	#if defined( __rrtmg )
[1875]	1253	!
	1254	!-- If required, add tendency due to radiative heating/cooling
[1976]	1255	IF ( radiation .AND. &
[1875]	1256	simulated_time > skip_time_do_radiation ) THEN
	1257	CALL radiation_tendency ( tend )
	1258	ENDIF
[3771]	1259	#endif
[1875]	1260
[3684]	1261	CALL module_interface_actions( 'pt-tendency' )
[1875]	1262
	1263	!
	1264	!-- Prognostic equation for potential temperature
[3634]	1265	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
	1266	!$ACC PRESENT(pt, tend, tpt_m, wall_flags_0) &
	1267	!$ACC PRESENT(pt_init, rdf_sc, ptdf_x, ptdf_y) &
	1268	!$ACC PRESENT(tsc(3:5)) &
	1269	!$ACC PRESENT(pt_p)
[1875]	1270	DO i = nxl, nxr
	1271	DO j = nys, nyn
[2232]	1272	DO k = nzb+1, nzt
	1273	pt_p(k,j,i) = pt(k,j,i) + ( dt_3d * ( sbt * tend(k,j,i) + &
	1274	tsc(3) * tpt_m(k,j,i) ) &
	1275	- tsc(5) * &
	1276	( pt(k,j,i) - pt_init(k) ) *&
	1277	( rdf_sc(k) + ptdf_x(i) + ptdf_y(j) )&
	1278	) &
	1279	* MERGE( 1.0_wp, 0.0_wp, &
	1280	BTEST( wall_flags_0(k,j,i), 0 ) &
	1281	)
[1875]	1282	ENDDO
	1283	ENDDO
	1284	ENDDO
	1285	!
	1286	!-- Calculate tendencies for the next Runge-Kutta step
	1287	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	1288	IF ( intermediate_timestep_count == 1 ) THEN
[3634]	1289	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
	1290	!$ACC PRESENT(tend, tpt_m)
[1875]	1291	DO i = nxl, nxr
	1292	DO j = nys, nyn
[2232]	1293	DO k = nzb+1, nzt
[1875]	1294	tpt_m(k,j,i) = tend(k,j,i)
	1295	ENDDO
	1296	ENDDO
	1297	ENDDO
	1298	ELSEIF ( intermediate_timestep_count < &
	1299	intermediate_timestep_count_max ) THEN
[3634]	1300	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
	1301	!$ACC PRESENT(tend, tpt_m)
[1875]	1302	DO i = nxl, nxr
	1303	DO j = nys, nyn
[2232]	1304	DO k = nzb+1, nzt
	1305	tpt_m(k,j,i) = -9.5625_wp * tend(k,j,i) + &
	1306	5.3125_wp * tpt_m(k,j,i)
[1875]	1307	ENDDO
	1308	ENDDO
	1309	ENDDO
	1310	ENDIF
	1311	ENDIF
	1312
	1313	CALL cpu_log( log_point(13), 'pt-equation', 'stop' )
	1314
	1315	ENDIF
	1316
	1317	!
[1960]	1318	!-- If required, compute prognostic equation for total water content
	1319	IF ( humidity ) THEN
[1875]	1320
[1960]	1321	CALL cpu_log( log_point(29), 'q-equation', 'start' )
[1875]	1322
	1323	!
	1324	!-- Scalar/q-tendency terms with communication
	1325	sbt = tsc(2)
	1326	IF ( scalar_advec == 'bc-scheme' ) THEN
	1327
	1328	IF ( timestep_scheme(1:5) /= 'runge' ) THEN
	1329	!
	1330	!-- Bott-Chlond scheme always uses Euler time step. Thus:
	1331	sbt = 1.0_wp
	1332	ENDIF
	1333	tend = 0.0_wp
	1334	CALL advec_s_bc( q, 'q' )
	1335
	1336	ENDIF
	1337
	1338	!
	1339	!-- Scalar/q-tendency terms with no communication
	1340	IF ( scalar_advec /= 'bc-scheme' ) THEN
	1341	tend = 0.0_wp
	1342	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	1343	IF ( ws_scheme_sca ) THEN
[4109]	1344	CALL advec_s_ws( advc_flags_s, q, 'q', &
	1345	bc_dirichlet_l .OR. bc_radiation_l, &
	1346	bc_dirichlet_n .OR. bc_radiation_n, &
	1347	bc_dirichlet_r .OR. bc_radiation_r, &
	1348	bc_dirichlet_s .OR. bc_radiation_s )
[1875]	1349	ELSE
	1350	CALL advec_s_pw( q )
	1351	ENDIF
	1352	ELSE
	1353	CALL advec_s_up( q )
	1354	ENDIF
	1355	ENDIF
	1356
[2232]	1357	CALL diffusion_s( q, &
	1358	surf_def_h(0)%qsws, surf_def_h(1)%qsws, &
	1359	surf_def_h(2)%qsws, &
	1360	surf_lsm_h%qsws, surf_usm_h%qsws, &
	1361	surf_def_v(0)%qsws, surf_def_v(1)%qsws, &
	1362	surf_def_v(2)%qsws, surf_def_v(3)%qsws, &
	1363	surf_lsm_v(0)%qsws, surf_lsm_v(1)%qsws, &
	1364	surf_lsm_v(2)%qsws, surf_lsm_v(3)%qsws, &
	1365	surf_usm_v(0)%qsws, surf_usm_v(1)%qsws, &
	1366	surf_usm_v(2)%qsws, surf_usm_v(3)%qsws )
[2155]	1367
[1875]	1368	!
[1960]	1369	!-- Sink or source of humidity due to canopy elements
[1875]	1370	IF ( plant_canopy ) CALL pcm_tendency( 5 )
	1371
	1372	!
	1373	!-- Large scale advection
	1374	IF ( large_scale_forcing ) THEN
	1375	CALL ls_advec( simulated_time, 'q' )
	1376	ENDIF
	1377
	1378	!
	1379	!-- Nudging
[2155]	1380	IF ( nudging ) CALL nudge( simulated_time, 'q' )
[1875]	1381
	1382	!
	1383	!-- If required compute influence of large-scale subsidence/ascent
	1384	IF ( large_scale_subsidence .AND. &
	1385	.NOT. use_subsidence_tendencies ) THEN
	1386	CALL subsidence( tend, q, q_init, 3 )
	1387	ENDIF
	1388
[3684]	1389	CALL module_interface_actions( 'q-tendency' )
[1875]	1390
	1391	!
[1960]	1392	!-- Prognostic equation for total water content
[1875]	1393	DO i = nxl, nxr
	1394	DO j = nys, nyn
[2232]	1395	DO k = nzb+1, nzt
	1396	q_p(k,j,i) = q(k,j,i) + ( dt_3d * ( sbt * tend(k,j,i) + &
	1397	tsc(3) * tq_m(k,j,i) ) &
	1398	- tsc(5) * rdf_sc(k) * &
	1399	( q(k,j,i) - q_init(k) ) &
	1400	) * MERGE( 1.0_wp, 0.0_wp, &
	1401	BTEST( wall_flags_0(k,j,i), 0 ) &
	1402	)
[1875]	1403	IF ( q_p(k,j,i) < 0.0_wp ) q_p(k,j,i) = 0.1_wp * q(k,j,i)
	1404	ENDDO
	1405	ENDDO
	1406	ENDDO
	1407
	1408	!
	1409	!-- Calculate tendencies for the next Runge-Kutta step
	1410	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	1411	IF ( intermediate_timestep_count == 1 ) THEN
	1412	DO i = nxl, nxr
	1413	DO j = nys, nyn
[2232]	1414	DO k = nzb+1, nzt
[1875]	1415	tq_m(k,j,i) = tend(k,j,i)
	1416	ENDDO
	1417	ENDDO
	1418	ENDDO
	1419	ELSEIF ( intermediate_timestep_count < &
	1420	intermediate_timestep_count_max ) THEN
	1421	DO i = nxl, nxr
	1422	DO j = nys, nyn
[2232]	1423	DO k = nzb+1, nzt
	1424	tq_m(k,j,i) = -9.5625_wp * tend(k,j,i) &
	1425	+ 5.3125_wp * tq_m(k,j,i)
[1875]	1426	ENDDO
	1427	ENDDO
	1428	ENDDO
	1429	ENDIF
	1430	ENDIF
	1431
[1960]	1432	CALL cpu_log( log_point(29), 'q-equation', 'stop' )
[1875]	1433
	1434	ENDIF
[1960]	1435	!
	1436	!-- If required, compute prognostic equation for scalar
	1437	IF ( passive_scalar ) THEN
[1875]	1438
[1960]	1439	CALL cpu_log( log_point(66), 's-equation', 'start' )
	1440
[1875]	1441	!
[1960]	1442	!-- Scalar/q-tendency terms with communication
	1443	sbt = tsc(2)
	1444	IF ( scalar_advec == 'bc-scheme' ) THEN
	1445
	1446	IF ( timestep_scheme(1:5) /= 'runge' ) THEN
	1447	!
	1448	!-- Bott-Chlond scheme always uses Euler time step. Thus:
	1449	sbt = 1.0_wp
	1450	ENDIF
	1451	tend = 0.0_wp
	1452	CALL advec_s_bc( s, 's' )
	1453
	1454	ENDIF
	1455
	1456	!
	1457	!-- Scalar/q-tendency terms with no communication
	1458	IF ( scalar_advec /= 'bc-scheme' ) THEN
	1459	tend = 0.0_wp
	1460	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	1461	IF ( ws_scheme_sca ) THEN
[4109]	1462	CALL advec_s_ws( advc_flags_s, s, 's', &
	1463	bc_dirichlet_l .OR. bc_radiation_l, &
	1464	bc_dirichlet_n .OR. bc_radiation_n, &
	1465	bc_dirichlet_r .OR. bc_radiation_r, &
	1466	bc_dirichlet_s .OR. bc_radiation_s )
[1960]	1467	ELSE
	1468	CALL advec_s_pw( s )
	1469	ENDIF
	1470	ELSE
	1471	CALL advec_s_up( s )
	1472	ENDIF
	1473	ENDIF
	1474
[2232]	1475	CALL diffusion_s( s, &
	1476	surf_def_h(0)%ssws, surf_def_h(1)%ssws, &
	1477	surf_def_h(2)%ssws, &
	1478	surf_lsm_h%ssws, surf_usm_h%ssws, &
	1479	surf_def_v(0)%ssws, surf_def_v(1)%ssws, &
	1480	surf_def_v(2)%ssws, surf_def_v(3)%ssws, &
	1481	surf_lsm_v(0)%ssws, surf_lsm_v(1)%ssws, &
	1482	surf_lsm_v(2)%ssws, surf_lsm_v(3)%ssws, &
	1483	surf_usm_v(0)%ssws, surf_usm_v(1)%ssws, &
	1484	surf_usm_v(2)%ssws, surf_usm_v(3)%ssws )
[2155]	1485
[1960]	1486	!
	1487	!-- Sink or source of humidity due to canopy elements
	1488	IF ( plant_canopy ) CALL pcm_tendency( 7 )
	1489
	1490	!
	1491	!-- Large scale advection. Not implemented for scalars so far.
	1492	! IF ( large_scale_forcing ) THEN
	1493	! CALL ls_advec( simulated_time, 'q' )
	1494	! ENDIF
	1495
	1496	!
	1497	!-- Nudging. Not implemented for scalars so far.
[2155]	1498	! IF ( nudging ) CALL nudge( simulated_time, 'q' )
[1960]	1499
	1500	!
	1501	!-- If required compute influence of large-scale subsidence/ascent.
	1502	!-- Not implemented for scalars so far.
	1503	IF ( large_scale_subsidence .AND. &
	1504	.NOT. use_subsidence_tendencies .AND. &
	1505	.NOT. large_scale_forcing ) THEN
	1506	CALL subsidence( tend, s, s_init, 3 )
	1507	ENDIF
	1508
[3684]	1509	CALL module_interface_actions( 's-tendency' )
[1960]	1510
	1511	!
	1512	!-- Prognostic equation for total water content
	1513	DO i = nxl, nxr
	1514	DO j = nys, nyn
[2232]	1515	DO k = nzb+1, nzt
	1516	s_p(k,j,i) = s(k,j,i) + ( dt_3d * ( sbt * tend(k,j,i) + &
	1517	tsc(3) * ts_m(k,j,i) ) &
	1518	- tsc(5) * rdf_sc(k) * &
	1519	( s(k,j,i) - s_init(k) ) &
	1520	) &
	1521	* MERGE( 1.0_wp, 0.0_wp, &
	1522	BTEST( wall_flags_0(k,j,i), 0 ) &
	1523	)
[1960]	1524	IF ( s_p(k,j,i) < 0.0_wp ) s_p(k,j,i) = 0.1_wp * s(k,j,i)
	1525	ENDDO
	1526	ENDDO
	1527	ENDDO
	1528
	1529	!
	1530	!-- Calculate tendencies for the next Runge-Kutta step
	1531	IF ( timestep_scheme(1:5) == 'runge' ) THEN
	1532	IF ( intermediate_timestep_count == 1 ) THEN
	1533	DO i = nxl, nxr
	1534	DO j = nys, nyn
[2232]	1535	DO k = nzb+1, nzt
[1960]	1536	ts_m(k,j,i) = tend(k,j,i)
	1537	ENDDO
	1538	ENDDO
	1539	ENDDO
	1540	ELSEIF ( intermediate_timestep_count < &
	1541	intermediate_timestep_count_max ) THEN
	1542	DO i = nxl, nxr
	1543	DO j = nys, nyn
[2232]	1544	DO k = nzb+1, nzt
	1545	ts_m(k,j,i) = -9.5625_wp * tend(k,j,i) &
	1546	+ 5.3125_wp * ts_m(k,j,i)
[1960]	1547	ENDDO
	1548	ENDDO
	1549	ENDDO
	1550	ENDIF
	1551	ENDIF
	1552
	1553	CALL cpu_log( log_point(66), 's-equation', 'stop' )
	1554
	1555	ENDIF
[3294]	1556	!
[3837]	1557	!-- Calculate prognostic equations for all other modules
	1558	CALL module_interface_prognostic_equations()
[1875]	1559
[3987]	1560	IF ( debug_output_timestep ) CALL debug_message( 'prognostic_equations_vector', 'end' )
[3885]	1561
[1875]	1562	END SUBROUTINE prognostic_equations_vector
	1563
	1564
	1565	END MODULE prognostic_equations_mod

Note: See TracBrowser for help on using the repository browser.

Download in other formats:

| Impressum | ©Leibniz Universität Hannover |