Home

Context Navigation

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

Last change on this file since 4214 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

Line
1	!> @file prognostic_equations.f90
2	!------------------------------------------------------------------------------!
3	! This file is part of the PALM model system.
4	!
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.
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	!
17	! Copyright 1997-2019 Leibniz Universitaet Hannover
18	!------------------------------------------------------------------------------!
19	!
20	! Current revisions:
21	! ------------------
22	!
23	!
24	! Former revisions:
25	! -----------------
26	! $Id: prognostic_equations.f90 4182 2019-08-22 15:20:23Z suehring $
27	! Corrected "Former revisions" section
28	!
29	! 4110 2019-07-22 17:05:21Z suehring
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.
33	!
34	! 4109 2019-07-22 17:00:34Z suehring
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
41	! Moved tcm_prognostic_equations to module_interface
42	!
43	! 3987 2019-05-22 09:52:13Z kanani
44	! Introduce alternative switch for debug output during timestepping
45	!
46	! 3956 2019-05-07 12:32:52Z monakurppa
47	! Removed salsa calls.
48	!
49	! 3931 2019-04-24 16:34:28Z schwenkel
50	! Correct/complete module_interface introduction for chemistry model
51	!
52	! 3899 2019-04-16 14:05:27Z monakurppa
53	! Corrections in the OpenMP version of salsa
54	!
55	! 3887 2019 -04-12 08:47:41Z schwenkel
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
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
64	! Bugfix in OpenMP directive
65	!
66	! 3880 2019-04-08 21:43:02Z knoop
67	! Moved wtm_tendencies to module_interface_actions
68	!
69	! 3874 2019-04-08 16:53:48Z knoop
70	! Added non_transport_physics module interfaces and moved bcm code into it
71	!
72	! 3872 2019-04-08 15:03:06Z knoop
73	! Moving prognostic equations of bcm into bulk_cloud_model_mod
74	!
75	! 3864 2019-04-05 09:01:56Z monakurppa
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
83	! added USE chem_gasphase_mod for nspec, nspec and spc_names
84	!
85	! 3820 2019-03-27 11:53:41Z forkel
86	! renamed do_depo to deposition_dry (ecc)
87	!
88	! 3797 2019-03-15 11:15:38Z forkel
89	! Call chem_integegrate in OpenMP loop (ketelsen)
90	!
91	!
92	! 3771 2019-02-28 12:19:33Z raasch
93	! preprocessor directivs fro rrtmg added
94	!
95	! 3761 2019-02-25 15:31:42Z raasch
96	! unused variable removed
97	!
98	! 3719 2019-02-06 13:10:18Z kanani
99	! Cleaned up chemistry cpu measurements
100	!
101	! 3684 2019-01-20 20:20:58Z knoop
102	! OpenACC port for SPEC
103	!
104	! Revision 1.1 2000/04/13 14:56:27 schroeter
105	! Initial revision
106	!
107	!
108	! Description:
109	! ------------
110	!> Solving the prognostic equations.
111	!------------------------------------------------------------------------------!
112	MODULE prognostic_equations_mod
113
114	USE advec_s_bc_mod, &
115	ONLY: advec_s_bc
116
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
144	USE arrays_3d, &
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, &
155	ug, u_init, u_p, v, vg, vpt, v_init, v_p, w, w_p
156
157	USE buoyancy_mod, &
158	ONLY: buoyancy
159
160	USE control_parameters, &
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, &
170	debug_output_timestep, &
171	dp_external, dp_level_ind_b, dp_smooth_factor, dpdxy, dt_3d, &
172	humidity, intermediate_timestep_count, &
173	intermediate_timestep_count_max, large_scale_forcing, &
174	large_scale_subsidence, &
175	monotonic_limiter_z, &
176	neutral, nudging, &
177	ocean_mode, passive_scalar, plant_canopy, pt_reference, &
178	scalar_advec, scalar_advec, simulated_time, sloping_surface, &
179	timestep_scheme, tsc, use_subsidence_tendencies, &
180	use_upstream_for_tke, wind_turbine, ws_scheme_mom, &
181	ws_scheme_sca, urban_surface, land_surface, &
182	time_since_reference_point, salsa
183
184	USE coriolis_mod, &
185	ONLY: coriolis
186
187	USE cpulog, &
188	ONLY: cpu_log, log_point, log_point_s
189
190	USE diffusion_s_mod, &
191	ONLY: diffusion_s
192
193	USE diffusion_u_mod, &
194	ONLY: diffusion_u
195
196	USE diffusion_v_mod, &
197	ONLY: diffusion_v
198
199	USE diffusion_w_mod, &
200	ONLY: diffusion_w
201
202	USE indices, &
203	ONLY: advc_flags_s, &
204	nbgp, nxl, nxlg, nxlu, nxr, nxrg, nyn, nyng, nys, nysg, nysv, &
205	nzb, nzt, wall_flags_0
206
207	USE kinds
208
209	USE lsf_nudging_mod, &
210	ONLY: ls_advec, nudge
211
212	USE module_interface, &
213	ONLY: module_interface_actions, &
214	module_interface_non_advective_processes, &
215	module_interface_exchange_horiz, &
216	module_interface_prognostic_equations
217
218	USE ocean_mod, &
219	ONLY: stokes_drift_terms, stokes_force, &
220	wave_breaking, wave_breaking_term
221
222	USE plant_canopy_model_mod, &
223	ONLY: cthf, pcm_tendency
224
225	#if defined( __rrtmg )
226	USE radiation_model_mod, &
227	ONLY: radiation, radiation_tendency, &
228	skip_time_do_radiation
229	#endif
230
231	USE statistics, &
232	ONLY: hom
233
234	USE subsidence_mod, &
235	ONLY: subsidence
236
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
241	IMPLICIT NONE
242
243	PRIVATE
244	PUBLIC prognostic_equations_cache, prognostic_equations_vector
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	!------------------------------------------------------------------------------!
270
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 !<
278	!$ INTEGER(iwp) :: omp_get_thread_num !<
279	INTEGER(iwp) :: tn = 0 !<
280
281	LOGICAL :: loop_start !<
282
283
284	IF ( debug_output_timestep ) CALL debug_message( 'prognostic_equations_cache', 'start' )
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
289	!$OMP PARALLEL PRIVATE (i,j)
290	!$OMP DO
291	DO i = nxl, nxr
292	DO j = nys, nyn
293	!
294	!-- Calculate non advective processes for all other modules
295	CALL module_interface_non_advective_processes( i, j )
296	ENDDO
297	ENDDO
298	!
299	!-- Module Inferface for exchange horiz after non_advective_processes but before
300	!-- advection. Therefore, non_advective_processes must not run for ghost points.
301	!$OMP END PARALLEL
302	CALL module_interface_exchange_horiz()
303	!
304	!-- Loop over all prognostic equations
305	!$OMP PARALLEL PRIVATE (i,i_omp_start,j,k,loop_start,tn)
306
307	!$ tn = omp_get_thread_num()
308	loop_start = .TRUE.
309
310	!$OMP DO
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.
318	i_omp_start = i
319	ENDIF
320
321	DO j = nys, nyn
322	!
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
325	!-- the prognostic equations for the u-component.
326	IF ( i >= nxlu ) THEN
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 )
332	ELSE
333	CALL advec_u_pw( i, j )
334	ENDIF
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
360	!
361	!-- Effect of Stokes drift (in ocean mode only)
362	IF ( stokes_force ) CALL stokes_drift_terms( i, j, 1 )
363
364	CALL module_interface_actions( i, j, 'u-tendency' )
365	!
366	!-- Prognostic equation for u-velocity component
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	)
377	ENDDO
378
379	!
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	!
388	!-- Calculate tendencies for the next Runge-Kutta step
389	IF ( timestep_scheme(1:5) == 'runge' ) THEN
390	IF ( intermediate_timestep_count == 1 ) THEN
391	DO k = nzb+1, nzt
392	tu_m(k,j,i) = tend(k,j,i)
393	ENDDO
394	ELSEIF ( intermediate_timestep_count < &
395	intermediate_timestep_count_max ) THEN
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)
399	ENDDO
400	ENDIF
401	ENDIF
402
403	ENDIF
404	!
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
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 )
414	ELSE
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
425	IF ( plant_canopy ) CALL pcm_tendency( i, j, 2 )
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
439	!
440	!-- Effect of Stokes drift (in ocean mode only)
441	IF ( stokes_force ) CALL stokes_drift_terms( i, j, 2 )
442
443	CALL module_interface_actions( i, j, 'v-tendency' )
444	!
445	!-- Prognostic equation for v-velocity component
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	)
455	ENDDO
456
457	!
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	!
466	!-- Calculate tendencies for the next Runge-Kutta step
467	IF ( timestep_scheme(1:5) == 'runge' ) THEN
468	IF ( intermediate_timestep_count == 1 ) THEN
469	DO k = nzb+1, nzt
470	tv_m(k,j,i) = tend(k,j,i)
471	ENDDO
472	ELSEIF ( intermediate_timestep_count < &
473	intermediate_timestep_count_max ) THEN
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)
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 )
489	ELSE
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
499	IF ( ocean_mode ) THEN
500	CALL buoyancy( i, j, rho_ocean, 3 )
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
514	!
515	!-- Effect of Stokes drift (in ocean mode only)
516	IF ( stokes_force ) CALL stokes_drift_terms( i, j, 3 )
517
518	CALL module_interface_actions( i, j, 'w-tendency' )
519	!
520	!-- Prognostic equation for w-velocity component
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) + &
524	tsc(3) * tw_m(k,j,i) ) &
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	)
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
535	DO k = nzb+1, nzt-1
536	tw_m(k,j,i) = tend(k,j,i)
537	ENDDO
538	ELSEIF ( intermediate_timestep_count < &
539	intermediate_timestep_count_max ) THEN
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)
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
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 )
562	ELSE
563	CALL advec_s_pw( i, j, pt )
564	ENDIF
565	ELSE
566	CALL advec_s_up( i, j, pt )
567	ENDIF
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 )
578
579	!
580	!-- Consideration of heat sources within the plant canopy
581	IF ( plant_canopy .AND. &
582	(cthf /= 0.0_wp .OR. urban_surface .OR. land_surface) ) THEN
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' )
590	ENDIF
591
592	!
593	!-- Nudging
594	IF ( nudging ) CALL nudge( i, j, simulated_time, 'pt' )
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
603	#if defined( __rrtmg )
604	!
605	!-- If required, add tendency due to radiative heating/cooling
606	IF ( radiation .AND. &
607	simulated_time > skip_time_do_radiation ) THEN
608	CALL radiation_tendency ( i, j, tend )
609	ENDIF
610	#endif
611
612	CALL module_interface_actions( i, j, 'pt-tendency' )
613	!
614	!-- Prognostic equation for potential temperature
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) + &
618	tsc(3) * tpt_m(k,j,i) ) &
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	)
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
633	DO k = nzb+1, nzt
634	tpt_m(k,j,i) = tend(k,j,i)
635	ENDDO
636	ELSEIF ( intermediate_timestep_count < &
637	intermediate_timestep_count_max ) THEN
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)
641	ENDDO
642	ENDIF
643	ENDIF
644
645	ENDIF
646
647	!
648	!-- If required, compute prognostic equation for total water content
649	IF ( humidity ) THEN
650
651	!
652	!-- Tendency-terms for total water content / scalar
653	tend(:,j,i) = 0.0_wp
654	IF ( timestep_scheme(1:5) == 'runge' ) &
655	THEN
656	IF ( ws_scheme_sca ) THEN
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 )
665	ELSE
666	CALL advec_s_pw( i, j, q )
667	ENDIF
668	ELSE
669	CALL advec_s_up( i, j, q )
670	ENDIF
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 )
681
682	!
683	!-- Sink or source of humidity due to canopy elements
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
694	IF ( nudging ) CALL nudge( i, j, simulated_time, 'q' )
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
703	CALL module_interface_actions( i, j, 'q-tendency' )
704
705	!
706	!-- Prognostic equation for total water content / scalar
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) + &
710	tsc(3) * tq_m(k,j,i) ) &
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	)
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
724	DO k = nzb+1, nzt
725	tq_m(k,j,i) = tend(k,j,i)
726	ENDDO
727	ELSEIF ( intermediate_timestep_count < &
728	intermediate_timestep_count_max ) THEN
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)
732	ENDDO
733	ENDIF
734	ENDIF
735
736	ENDIF
737
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
744	IF ( timestep_scheme(1:5) == 'runge' ) &
745	THEN
746	IF ( ws_scheme_sca ) THEN
747	!
748	!-- For scalar advection apply monotonic flux limiter near
749	!-- topography.
750	CALL advec_s_ws( advc_flags_s, &
751	i, j, s, 's', flux_s_s, &
752	diss_s_s, flux_l_s, diss_l_s, i_omp_start, &
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 )
759	ELSE
760	CALL advec_s_pw( i, j, s )
761	ENDIF
762	ELSE
763	CALL advec_s_up( i, j, s )
764	ENDIF
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 )
775
776	!
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
788	! IF ( nudging ) CALL nudge( i, j, simulated_time, 's' )
789
790	!
791	!-- If required compute influence of large-scale subsidence/ascent.
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
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
801	CALL module_interface_actions( i, j, 's-tendency' )
802
803	!
804	!-- Prognostic equation for scalar
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	)
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
822	DO k = nzb+1, nzt
823	ts_m(k,j,i) = tend(k,j,i)
824	ENDDO
825	ELSEIF ( intermediate_timestep_count < &
826	intermediate_timestep_count_max ) THEN
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)
830	ENDDO
831	ENDIF
832	ENDIF
833
834	ENDIF
835	!
836	!-- Calculate prognostic equations for all other modules
837	CALL module_interface_prognostic_equations( i, j, i_omp_start, tn )
838
839	ENDDO ! loop over j
840	ENDDO ! loop over i
841	!$OMP END PARALLEL
842
843
844	CALL cpu_log( log_point(32), 'all progn.equations', 'stop' )
845
846	IF ( debug_output_timestep ) CALL debug_message( 'prognostic_equations_cache', 'end' )
847
848	END SUBROUTINE prognostic_equations_cache
849
850
851	!------------------------------------------------------------------------------!
852	! Description:
853	! ------------
854	!> Version for vector machines
855	!------------------------------------------------------------------------------!
856
857	SUBROUTINE prognostic_equations_vector
858
859
860	INTEGER(iwp) :: i !<
861	INTEGER(iwp) :: j !<
862	INTEGER(iwp) :: k !<
863
864	REAL(wp) :: sbt !<
865
866
867	IF ( debug_output_timestep ) CALL debug_message( 'prognostic_equations_vector', 'start' )
868	!
869	!-- Calculate non advective processes for all other modules
870	CALL module_interface_non_advective_processes
871	!
872	!-- Module Inferface for exchange horiz after non_advective_processes but before
873	!-- advection. Therefore, non_advective_processes must not run for ghost points.
874	CALL module_interface_exchange_horiz()
875	!
876	!-- u-velocity component
877	CALL cpu_log( log_point(5), 'u-equation', 'start' )
878
879	!$ACC KERNELS PRESENT(tend)
880	tend = 0.0_wp
881	!$ACC END KERNELS
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
917	!
918	!-- Effect of Stokes drift (in ocean mode only)
919	IF ( stokes_force ) CALL stokes_drift_terms( 1 )
920
921	CALL module_interface_actions( 'u-tendency' )
922
923	!
924	!-- Prognostic equation for u-velocity component
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)
929	DO i = nxlu, nxr
930	DO j = nys, nyn
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	)
939	ENDDO
940	ENDDO
941	ENDDO
942
943	!
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	!
952	!-- Calculate tendencies for the next Runge-Kutta step
953	IF ( timestep_scheme(1:5) == 'runge' ) THEN
954	IF ( intermediate_timestep_count == 1 ) THEN
955	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
956	!$ACC PRESENT(tend, tu_m)
957	DO i = nxlu, nxr
958	DO j = nys, nyn
959	DO k = nzb+1, nzt
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
966	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
967	!$ACC PRESENT(tend, tu_m)
968	DO i = nxlu, nxr
969	DO j = nys, nyn
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)
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
985	!$ACC KERNELS PRESENT(tend)
986	tend = 0.0_wp
987	!$ACC END KERNELS
988	IF ( timestep_scheme(1:5) == 'runge' ) THEN
989	IF ( ws_scheme_mom ) THEN
990	CALL advec_v_ws
991	ELSE
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
1020	!
1021	!-- Effect of Stokes drift (in ocean mode only)
1022	IF ( stokes_force ) CALL stokes_drift_terms( 2 )
1023
1024	CALL module_interface_actions( 'v-tendency' )
1025
1026	!
1027	!-- Prognostic equation for v-velocity component
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)
1032	DO i = nxl, nxr
1033	DO j = nysv, nyn
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	)
1042	ENDDO
1043	ENDDO
1044	ENDDO
1045
1046	!
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	!
1055	!-- Calculate tendencies for the next Runge-Kutta step
1056	IF ( timestep_scheme(1:5) == 'runge' ) THEN
1057	IF ( intermediate_timestep_count == 1 ) THEN
1058	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
1059	!$ACC PRESENT(tend, tv_m)
1060	DO i = nxl, nxr
1061	DO j = nysv, nyn
1062	DO k = nzb+1, nzt
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
1069	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
1070	!$ACC PRESENT(tend, tv_m)
1071	DO i = nxl, nxr
1072	DO j = nysv, nyn
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)
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
1088	!$ACC KERNELS PRESENT(tend)
1089	tend = 0.0_wp
1090	!$ACC END KERNELS
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
1104	IF ( ocean_mode ) THEN
1105	CALL buoyancy( rho_ocean, 3 )
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
1119	!
1120	!-- Effect of Stokes drift (in ocean mode only)
1121	IF ( stokes_force ) CALL stokes_drift_terms( 3 )
1122
1123	CALL module_interface_actions( 'w-tendency' )
1124
1125	!
1126	!-- Prognostic equation for w-velocity component
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)
1131	DO i = nxl, nxr
1132	DO j = nys, nyn
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	)
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
1148	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
1149	!$ACC PRESENT(tend, tw_m)
1150	DO i = nxl, nxr
1151	DO j = nys, nyn
1152	DO k = nzb+1, nzt-1
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
1159	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
1160	!$ACC PRESENT(tend, tw_m)
1161	DO i = nxl, nxr
1162	DO j = nys, nyn
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)
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
1199	!$ACC KERNELS PRESENT(tend)
1200	tend = 0.0_wp
1201	!$ACC END KERNELS
1202	IF ( timestep_scheme(1:5) == 'runge' ) THEN
1203	IF ( ws_scheme_sca ) THEN
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 )
1209	ELSE
1210	CALL advec_s_pw( pt )
1211	ENDIF
1212	ELSE
1213	CALL advec_s_up( pt )
1214	ENDIF
1215	ENDIF
1216
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 )
1227
1228	!
1229	!-- Consideration of heat sources within the plant canopy
1230	IF ( plant_canopy .AND. &
1231	(cthf /= 0.0_wp .OR. urban_surface .OR. land_surface) ) THEN
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
1243	IF ( nudging ) CALL nudge( simulated_time, 'pt' )
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
1252	#if defined( __rrtmg )
1253	!
1254	!-- If required, add tendency due to radiative heating/cooling
1255	IF ( radiation .AND. &
1256	simulated_time > skip_time_do_radiation ) THEN
1257	CALL radiation_tendency ( tend )
1258	ENDIF
1259	#endif
1260
1261	CALL module_interface_actions( 'pt-tendency' )
1262
1263	!
1264	!-- Prognostic equation for potential temperature
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)
1270	DO i = nxl, nxr
1271	DO j = nys, nyn
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	)
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
1289	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
1290	!$ACC PRESENT(tend, tpt_m)
1291	DO i = nxl, nxr
1292	DO j = nys, nyn
1293	DO k = nzb+1, nzt
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
1300	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
1301	!$ACC PRESENT(tend, tpt_m)
1302	DO i = nxl, nxr
1303	DO j = nys, nyn
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)
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	!
1318	!-- If required, compute prognostic equation for total water content
1319	IF ( humidity ) THEN
1320
1321	CALL cpu_log( log_point(29), 'q-equation', 'start' )
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
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 )
1349	ELSE
1350	CALL advec_s_pw( q )
1351	ENDIF
1352	ELSE
1353	CALL advec_s_up( q )
1354	ENDIF
1355	ENDIF
1356
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 )
1367
1368	!
1369	!-- Sink or source of humidity due to canopy elements
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
1380	IF ( nudging ) CALL nudge( simulated_time, 'q' )
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
1389	CALL module_interface_actions( 'q-tendency' )
1390
1391	!
1392	!-- Prognostic equation for total water content
1393	DO i = nxl, nxr
1394	DO j = nys, nyn
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	)
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
1414	DO k = nzb+1, nzt
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
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)
1426	ENDDO
1427	ENDDO
1428	ENDDO
1429	ENDIF
1430	ENDIF
1431
1432	CALL cpu_log( log_point(29), 'q-equation', 'stop' )
1433
1434	ENDIF
1435	!
1436	!-- If required, compute prognostic equation for scalar
1437	IF ( passive_scalar ) THEN
1438
1439	CALL cpu_log( log_point(66), 's-equation', 'start' )
1440
1441	!
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
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 )
1467	ELSE
1468	CALL advec_s_pw( s )
1469	ENDIF
1470	ELSE
1471	CALL advec_s_up( s )
1472	ENDIF
1473	ENDIF
1474
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 )
1485
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.
1498	! IF ( nudging ) CALL nudge( simulated_time, 'q' )
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
1509	CALL module_interface_actions( 's-tendency' )
1510
1511	!
1512	!-- Prognostic equation for total water content
1513	DO i = nxl, nxr
1514	DO j = nys, nyn
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	)
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
1535	DO k = nzb+1, nzt
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
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)
1547	ENDDO
1548	ENDDO
1549	ENDDO
1550	ENDIF
1551	ENDIF
1552
1553	CALL cpu_log( log_point(66), 's-equation', 'stop' )
1554
1555	ENDIF
1556	!
1557	!-- Calculate prognostic equations for all other modules
1558	CALL module_interface_prognostic_equations()
1559
1560	IF ( debug_output_timestep ) CALL debug_message( 'prognostic_equations_vector', 'end' )
1561
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 |