-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathadvance_clubb_core_module.F90
5549 lines (4729 loc) · 269 KB
/
advance_clubb_core_module.F90
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
module advance_clubb_core_module
! Description:
! The module containing the `core' of the CLUBB parameterization.
! It advances CLUBB's equations one model time step.
!
! References:
! https://arxiv.org/pdf/1711.03675v1.pdf#nameddest=url:overview_clubb
!
! ``A PDF-Based Model for Boundary Layer Clouds. Part I:
! Method and Model Description'' Golaz, et al. (2002)
! JAS, Vol. 59, pp. 3540--3551.
!
! Copyright Notice:
!
! This code and the source code it references are (C) 2006-2020.
!
! The distribution of this code and derived works thereof
! should include this notice.
!
! Portions of this code derived from other sources (Hugh Morrison,
! ACM TOMS, Numerical Recipes, et cetera) are the intellectual
! property of their respective authors as noted and are also subject
! to copyright.
!
!
!
! Cloud Layers Unified By Binormals (CLUBB) user license
! agreement.
!
! Thank you for your interest in CLUBB. We work hard to create a
! code that implements the best software engineering practices,
! is supported to the extent allowed by our limited resources,
! and is available without cost to non-commercial users. You may
! use CLUBB if, in return, you abide by these conditions:
!
! 1. Please cite CLUBB in presentations and publications that
! contain results obtained using CLUBB.
!
! 2. You may not use any part of CLUBB to create or modify
! another single-column (1D) model that is not called CLUBB.
! However, you may modify or augment CLUBB or parts of CLUBB if
! you include "CLUBB" in the name of the resulting single-column
! model. For example, a user at MIT might modify CLUBB and call
! the modified version "CLUBB-MIT." Or, for example, a user of
! the CLM land-surface model might interface CLM to CLUBB and
! call it "CLM-CLUBB." This naming convention recognizes the
! contributions of both sets of developers.
!
! 3. You may implement CLUBB as a parameterization in a large-
! scale host model that has 2 or 3 spatial dimensions without
! including "CLUBB" in the combined model name, but please
! acknowledge in presentations and publications that CLUBB has
! been included as a parameterization.
!
! 4. You may not provide all or part of CLUBB to anyone without
! prior permission from Vincent Larson (vlarson@uwm.edu). If
! you wish to share CLUBB with your collaborators without
! seeking permission, please ask your collaborators to register
! as CLUBB users at https://carson.math.uwm.edu/larson-group/clubb_site/ and to
! download CLUBB from there.
!
! 5. You may not use CLUBB for commercial purposes unless you
! receive permission from Vincent Larson.
!
! 6. You may not re-license all or any part of CLUBB.
!
! 7. CLUBB is provided "as is" and without warranty.
!
! We hope that CLUBB will develop into a community resource. We
! encourage users to contribute their CLUBB modifications or
! extensions to the CLUBB development group. We will then
! consider them for inclusion in CLUBB. Such contributions will
! benefit all CLUBB users. We would be pleased to acknowledge
! contributors and list their CLUBB-related papers on our "About
! CLUBB" webpage (https://carson.math.uwm.edu/larson-group/clubb_site/about.html) for
! those contributors who so desire.
!
! Thanks so much and best wishes for your research!
!
! The CLUBB Development Group
! (Present and past contributors to the source code include
! Vincent Larson, Chris Golaz, David Schanen, Brian Griffin,
! Joshua Fasching, Adam Smith, and Michael Falk).
!-----------------------------------------------------------------------
! Options for the placement of the call to CLUBB's PDF.
use model_flags, only: &
ipdf_pre_advance_fields, & ! Call before advancing predictive fields
ipdf_post_advance_fields, & ! Call after advancing predictive fields
ipdf_pre_post_advance_fields ! Call both before and after advancing
! predictive fields
implicit none
public :: &
setup_clubb_core, &
advance_clubb_core, &
cleanup_clubb_core, &
set_Lscale_max, &
calculate_thlp2_rad
private ! Default Scope
! Advance subroutine ordering variables
integer, parameter, private :: &
order_xm_wpxp = 1, &
order_xp2_xpyp = 2, &
order_wp2_wp3 = 3, &
order_windm = 4
contains
!-----------------------------------------------------------------------
!#######################################################################
!#######################################################################
! If you change the argument list of advance_clubb_core you also have to
! change the calls to this function in the host models CAM, WRF, SAM
! and GFDL.
!#######################################################################
!#######################################################################
subroutine advance_clubb_core ( gr, nz, ngrdcol, & ! intent(in)
l_implemented, dt, fcor, sfc_elevation, hydromet_dim, & ! intent(in)
thlm_forcing, rtm_forcing, um_forcing, vm_forcing, & ! intent(in)
sclrm_forcing, edsclrm_forcing, wprtp_forcing, & ! intent(in)
wpthlp_forcing, rtp2_forcing, thlp2_forcing, & ! intent(in)
rtpthlp_forcing, wm_zm, wm_zt, & ! intent(in)
wpthlp_sfc, wprtp_sfc, upwp_sfc, vpwp_sfc, & ! intent(in)
wpsclrp_sfc, wpedsclrp_sfc, & ! intent(in)
upwp_sfc_pert, vpwp_sfc_pert, & ! intent(in)
rtm_ref, thlm_ref, um_ref, vm_ref, ug, vg, & ! Intent(in)
p_in_Pa, rho_zm, rho, exner, & ! intent(in)
rho_ds_zm, rho_ds_zt, invrs_rho_ds_zm, & ! intent(in)
invrs_rho_ds_zt, thv_ds_zm, thv_ds_zt, & ! intent(in)
hydromet, & ! Unused
rfrzm, radf, & ! intent(in)
#ifdef CLUBBND_CAM
varmu, & ! intent(in)
#endif
wphydrometp, wp2hmp, rtphmp_zt, thlphmp_zt, & ! intent(in)
host_dx, host_dy, & ! intent(in)
clubb_params, nu_vert_res_dep, lmin, & ! intent(in)
clubb_config_flags, & ! intent(in)
stats_metadata, & ! intent(in)
stats_zt, stats_zm, stats_sfc, & ! intent(inout)
um, vm, upwp, vpwp, up2, vp2, up3, vp3, & ! intent(inout)
thlm, rtm, wprtp, wpthlp, & ! intent(inout)
wp2, wp3, rtp2, rtp3, thlp2, thlp3, rtpthlp, & ! intent(inout)
sclrm, & ! intent(inout)
#ifdef GFDL
sclrm_trsport_only, & ! h1g, 2010-06-16 ! intent(inout)
#endif
sclrp2, sclrp3, sclrprtp, sclrpthlp, & ! intent(inout)
wpsclrp, edsclrm, & ! intent(inout)
rcm, cloud_frac, & ! intent(inout)
wpthvp, wp2thvp, rtpthvp, thlpthvp, & ! intent(inout)
sclrpthvp, & ! intent(inout)
wp2rtp, wp2thlp, uprcp, vprcp, rc_coef, wp4, & ! intent(inout)
wpup2, wpvp2, wp2up2, wp2vp2, ice_supersat_frac, & ! intent(inout)
um_pert, vm_pert, upwp_pert, vpwp_pert, & ! intent(inout)
pdf_params, pdf_params_zm, & ! intent(inout)
pdf_implicit_coefs_terms, & ! intent(inout)
#ifdef GFDL
RH_crit, & !h1g, 2010-06-16 ! intent(inout)
do_liquid_only_in_clubb, & ! intent(in)
#endif
Kh_zm, Kh_zt, & ! intent(out)
#ifdef CLUBB_CAM
qclvar, & ! intent(out)
#endif
thlprcp, wprcp, w_up_in_cloud, w_down_in_cloud, & ! intent(out)
cloudy_updraft_frac, cloudy_downdraft_frac, & ! intent(out)
rcm_in_layer, cloud_cover, invrs_tau_zm, & ! intent(out)
err_code_out ) ! intent(out)
! Description:
! Subroutine to advance CLUBB one timestep
! References:
! https://arxiv.org/pdf/1711.03675v1.pdf#nameddest=url:overview_clubb
!
! ``A PDF-Based Model for Boundary Layer Clouds. Part I:
! Method and Model Description'' Golaz, et al. (2002)
! JAS, Vol. 59, pp. 3540--3551.
!-----------------------------------------------------------------------
! Modules to be included
use constants_clubb, only: &
em_min, &
thl_tol, &
rt_tol, &
w_tol, &
w_tol_sqd, &
fstderr, &
zero_threshold, &
three_halves, &
one, &
two, &
zero, &
unused_var, &
grav, &
eps, &
num_hf_draw_points
use parameter_indices, only: &
nparams, & ! Variable(s)
itaumax, &
ic_K, &
ic_K10, &
ic_K10h, &
imu, &
igamma_coef, &
igamma_coefb, &
igamma_coefc, &
iC_wp2_splat, &
ixp3_coef_base, &
ixp3_coef_slope, &
ilambda0_stability_coef, &
ibeta, &
iSkw_denom_coef, &
iSkw_max_mag, &
iup2_sfc_coef, &
ia3_coef_min, &
ibv_efold
use parameters_tunable, only: &
nu_vertical_res_dep ! Type(s)
use parameters_model, only: &
sclr_dim, & ! Variable(s)
edsclr_dim, &
sclr_tol
use model_flags, only: &
clubb_config_flags_type, & ! Type
l_host_applies_sfc_fluxes, & ! Variable(s)
l_gamma_Skw, &
l_advance_xp3, &
iiPDF_ADG1
use grid_class, only: &
grid, & ! Type
zm2zt, & ! Procedure(s)
zt2zm, &
ddzm, &
ddzt, &
zm2zt2zm
use numerical_check, only: &
parameterization_check, & ! Procedure(s)
calculate_spurious_source
use pdf_parameter_module, only: &
pdf_parameter, &
implicit_coefs_terms
#ifdef GFDL
use advance_sclrm_Nd_module, only: & ! h1g, 2010-06-16 begin mod
advance_sclrm_Nd_diffusion_OG, &
advance_sclrm_Nd_upwind, &
advance_sclrm_Nd_semi_implicit ! h1g, 2010-06-16 end mod
#endif
use advance_xm_wpxp_module, only: &
advance_xm_wpxp ! Compute mean/flux terms
use advance_xp2_xpyp_module, only: &
advance_xp2_xpyp ! Computes variance terms
use sfc_varnce_module, only: &
calc_sfc_varnce ! Procedure
use mixing_length, only: &
compute_mixing_length, & ! Procedure
calc_Lscale_directly, & ! for Lscale
diagnose_Lscale_from_tau ! for Lscale from tau
use advance_windm_edsclrm_module, only: &
advance_windm_edsclrm ! Procedure(s)
use saturation, only: &
! Procedure
sat_mixrat_liq ! Saturation mixing ratio
use advance_wp2_wp3_module, only: &
advance_wp2_wp3 ! Procedure
use advance_xp3_module, only: &
advance_xp3 ! Procedure(s)
use calc_pressure, only: &
calculate_thvm
use clubb_precision, only: &
core_rknd ! Variable(s)
use error_code, only: &
clubb_at_least_debug_level, & ! Procedure
err_code, & ! Error Indicator
clubb_no_error, & ! Constant
clubb_fatal_error ! Constant
use Skx_module, only: &
Skx_func, & ! Procedure(s)
xp3_LG_2005_ansatz
use clip_explicit, only: &
clip_covars_denom ! Procedure(s)
use T_in_K_module, only: &
! Read values from namelist
thlm2T_in_K ! Procedure
use sigma_sqd_w_module, only: &
compute_sigma_sqd_w ! Procedure(s)
use stats_clubb_utilities, only: &
stats_accumulate ! Procedure
use stats_type_utilities, only: &
stat_update_var_pt, & ! Procedure(s)
stat_update_var, &
stat_begin_update, &
stat_begin_update_pt, &
stat_end_update, &
stat_end_update_pt
use fill_holes, only: &
fill_holes_vertical
use advance_helper_module, only: &
calc_stability_correction, & ! Procedure(s)
compute_Cx_fnc_Richardson, &
calc_brunt_vaisala_freq_sqd, &
wp2_term_splat_lhs, &
wp3_term_splat_lhs, &
vertical_integral, &
Lscale_width_vert_avg
use interpolation, only: &
pvertinterp
use stats_type, only: stats ! Type
use pdf_parameter_module, only: &
copy_single_pdf_params_to_multi, &
copy_multi_pdf_params_to_single, &
init_pdf_params
use stats_variables, only: &
stats_metadata_type
implicit none
!!! External
intrinsic :: sqrt, min, max, exp, mod, real
! Constant Parameters
real( kind = core_rknd ), parameter :: &
tau_const = 1000._core_rknd
!--------------------------- Input Variables ---------------------------
integer, intent(in) :: &
nz, & ! Number of vertical levels
ngrdcol ! Number of grid columns
type (grid), target, intent(in) :: gr
logical, intent(in) :: &
l_implemented ! True if CLUBB is being run within a large-scale host model,
! rather than a standalone single-column model.
real( kind = core_rknd ), intent(in) :: &
dt ! Current timestep duration [s]
real( kind = core_rknd ) :: &
dt_advance ! General timestep duration for advance_wp2_wp3,
! advance_xm_xpwp, and advance_xp2_xpyp.
! Only differs from dt if l_lmm_stepping is used [s]
real( kind = core_rknd ), intent(in), dimension(ngrdcol) :: &
fcor, & ! Coriolis forcing [s^-1]
sfc_elevation ! Elevation of ground level [m above MSL]
integer, intent(in) :: &
hydromet_dim ! Total number of hydrometeor species [#]
real( kind = core_rknd ), intent(in), dimension(ngrdcol,nz) :: &
thlm_forcing, & ! liquid potential temp. forcing (thermodynamic levels) [K/s]
rtm_forcing, & ! total water forcing (thermodynamic levels) [(kg/kg)/s]
um_forcing, & ! eastward wind forcing (thermodynamic levels) [m/s/s]
vm_forcing, & ! northward wind forcing (thermodynamic levels) [m/s/s]
wprtp_forcing, & ! total water turbulent flux forcing (momentum levels) [m*K/s^2]
wpthlp_forcing, & ! liq pot temp turb flux forcing (momentum levels) [m*(kg/kg)/s^2]
rtp2_forcing, & ! total water variance forcing (momentum levels) [(kg/kg)^2/s]
thlp2_forcing, & ! liq pot temp variance forcing (momentum levels) [K^2/s]
rtpthlp_forcing, & ! <r_t'th_l'> covariance forcing (momentum levels) [K*(kg/kg)/s]
wm_zm, & ! vertical mean wind component on momentum levels [m/s]
wm_zt, & ! vertical mean wind component on thermo. levels [m/s]
rho_zm, & ! Air density on momentum levels [kg/m^3]
rho, & ! Air density on thermodynamic levels [kg/m^3]
rho_ds_zm, & ! Dry, static density on momentum levels [kg/m^3]
rho_ds_zt, & ! Dry, static density on thermo. levels [kg/m^3]
invrs_rho_ds_zm, & ! Inverse dry, static density on momentum levs. [m^3/kg]
invrs_rho_ds_zt, & ! Inverse dry, static density on thermo levs. [m^3/kg]
thv_ds_zm, & ! Dry, base-state theta_v on momentum levs. [K]
thv_ds_zt, & ! Dry, base-state theta_v on thermo levs. [K]
rfrzm ! Total ice-phase water mixing ratio [kg/kg]
real( kind = core_rknd ), dimension(ngrdcol,nz,hydromet_dim), intent(in) :: &
hydromet ! Array of hydrometeors [units vary]
real( kind = core_rknd ), dimension(ngrdcol,nz), intent(in) :: &
radf ! Buoyancy production at cloud top due to longwave radiative cooling [m^2/s^3]
#ifdef CLUBBND_CAM
real( kind = core_rknd ), intent(in), dimension(ngrdcol) :: &
varmu
#endif
real( kind = core_rknd ), dimension(ngrdcol,nz,hydromet_dim), intent(in) :: &
wphydrometp, & ! Covariance of w and a hydrometeor [(m/s) <hm units>]
wp2hmp, & ! Third-order moment: < w'^2 hm' > (hm = hydrometeor) [(m/s)^2 <hm units>]
rtphmp_zt, & ! Covariance of rt and hm (on thermo levs.) [(kg/kg) <hm units>]
thlphmp_zt ! Covariance of thl and hm (on thermo levs.) [K <hm units>]
real( kind = core_rknd ), intent(in), dimension(ngrdcol) :: &
wpthlp_sfc, & ! w' theta_l' at surface [(m K)/s]
wprtp_sfc, & ! w' r_t' at surface [(kg m)/( kg s)]
upwp_sfc, & ! u'w' at surface [m^2/s^2]
vpwp_sfc ! v'w' at surface [m^2/s^2]
! Passive scalar variables
real( kind = core_rknd ), intent(in), dimension(ngrdcol,nz,sclr_dim) :: &
sclrm_forcing ! Passive scalar forcing [{units vary}/s]
real( kind = core_rknd ), intent(in), dimension(ngrdcol,sclr_dim) :: &
wpsclrp_sfc ! Passive scalar flux at surface [{units vary} m/s]
! Eddy passive scalar variables
real( kind = core_rknd ), intent(in), dimension(ngrdcol,nz,edsclr_dim) :: &
edsclrm_forcing ! Eddy-diffusion passive scalar forcing [{units vary}/s]
real( kind = core_rknd ), intent(in), dimension(ngrdcol,edsclr_dim) :: &
wpedsclrp_sfc ! Eddy-diffusion passive scalar flux at surface [{units vary} m/s
real( kind = core_rknd ), intent(in), dimension(ngrdcol) :: &
upwp_sfc_pert, & ! pertubed u'w' at surface [m^2/s^2]
vpwp_sfc_pert ! pertubed v'w' at surface [m^2/s^2]
! Reference profiles (used for nudging, sponge damping, and Coriolis effect)
real( kind = core_rknd ), dimension(ngrdcol,nz), intent(in) :: &
rtm_ref, & ! Initial total water mixing ratio [kg/kg]
thlm_ref, & ! Initial liquid water potential temperature [K]
um_ref, & ! Initial u wind; Michael Falk [m/s]
vm_ref, & ! Initial v wind; Michael Falk [m/s]
ug, & ! u geostrophic wind [m/s]
vg ! v geostrophic wind [m/s]
! Host model horizontal grid spacing, if part of host model.
real( kind = core_rknd ), intent(in), dimension(ngrdcol) :: &
host_dx, & ! East-west horizontal grid spacing [m]
host_dy ! North-south horizontal grid spacing [m]
real( kind = core_rknd ), dimension(nparams), intent(in) :: &
clubb_params ! Array of CLUBB's tunable parameters [units vary]
type(nu_vertical_res_dep), intent(in) :: &
nu_vert_res_dep ! Vertical resolution dependent nu values
real( kind = core_rknd ), intent(in) :: &
lmin ! Min. value for the length scale [m]
type( clubb_config_flags_type ), intent(in) :: &
clubb_config_flags ! Derived type holding all configurable CLUBB flags
type (stats_metadata_type), intent(in) :: &
stats_metadata
!--------------------------- Input/Output Variables ---------------------------
type (stats), target, intent(inout), dimension(ngrdcol) :: &
stats_zt, &
stats_zm, &
stats_sfc
! These are prognostic or are planned to be in the future
real( kind = core_rknd ), intent(inout), dimension(ngrdcol,nz) :: &
um, & ! eastward grid-mean wind component (thermodynamic levels) [m/s]
upwp, & ! u'w' (momentum levels) [m^2/s^2]
vm, & ! northward grid-mean wind component (thermodynamic levels) [m/s]
vpwp, & ! v'w' (momentum levels) [m^2/s^2]
up2, & ! u'^2 (momentum levels) [m^2/s^2]
vp2, & ! v'^2 (momentum levels) [m^2/s^2]
up3, & ! u'^3 (thermodynamic levels) [m^3/s^3]
vp3, & ! v'^3 (thermodynamic levels) [m^3/s^3]
rtm, & ! total water mixing ratio, r_t (thermo. levels) [kg/kg]
wprtp, & ! w' r_t' (momentum levels) [(kg/kg) m/s]
thlm, & ! liq. water pot. temp., th_l (thermo. levels) [K]
wpthlp, & ! w'th_l' (momentum levels) [(m/s) K]
rtp2, & ! r_t'^2 (momentum levels) [(kg/kg)^2]
rtp3, & ! r_t'^3 (thermodynamic levels) [(kg/kg)^3]
thlp2, & ! th_l'^2 (momentum levels) [K^2]
thlp3, & ! th_l'^3 (thermodynamic levels) [K^3]
rtpthlp, & ! r_t'th_l' (momentum levels) [(kg/kg) K]
wp2, & ! w'^2 (momentum levels) [m^2/s^2]
wp3 ! w'^3 (thermodynamic levels) [m^3/s^3]
! Passive scalar variables
real( kind = core_rknd ), intent(inout), dimension(ngrdcol,nz,sclr_dim) :: &
sclrm, & ! Passive scalar mean (thermo. levels) [units vary]
wpsclrp, & ! w'sclr' (momentum levels) [{units vary} m/s]
sclrp2, & ! sclr'^2 (momentum levels) [{units vary}^2]
sclrp3, & ! sclr'^3 (thermodynamic levels) [{units vary}^3]
sclrprtp, & ! sclr'rt' (momentum levels) [{units vary} (kg/kg)]
sclrpthlp ! sclr'thl' (momentum levels) [{units vary} K]
real( kind = core_rknd ), intent(inout), dimension(ngrdcol,nz) :: &
p_in_Pa, & ! Air pressure (thermodynamic levels) [Pa]
exner ! Exner function (thermodynamic levels) [-]
real( kind = core_rknd ), intent(inout), dimension(ngrdcol,nz) :: &
rcm, & ! cloud water mixing ratio, r_c (thermo. levels) [kg/kg]
cloud_frac, & ! cloud fraction (thermodynamic levels) [-]
wpthvp, & ! < w' th_v' > (momentum levels) [kg/kg K]
wp2thvp, & ! < w'^2 th_v' > (thermodynamic levels) [m^2/s^2 K]
rtpthvp, & ! < r_t' th_v' > (momentum levels) [kg/kg K]
thlpthvp ! < th_l' th_v' > (momentum levels) [K^2]
real( kind = core_rknd ), intent(inout), dimension(ngrdcol,nz,sclr_dim) :: &
sclrpthvp ! < sclr' th_v' > (momentum levels) [units vary]
real( kind = core_rknd ), intent(inout), dimension(ngrdcol,nz) :: &
wp2rtp, & ! w'^2 rt' (thermodynamic levels) [m^2/s^2 kg/kg]
wp2thlp, & ! w'^2 thl' (thermodynamic levels) [m^2/s^2 K]
uprcp, & ! < u' r_c' > (momentum levels) [(m/s)(kg/kg)]
vprcp, & ! < v' r_c' > (momentum levels) [(m/s)(kg/kg)]
rc_coef, & ! Coef of X'r_c' in Eq. (34) (t-levs.) [K/(kg/kg)]
wp4, & ! w'^4 (momentum levels) [m^4/s^4]
wpup2, & ! w'u'^2 (thermodynamic levels) [m^3/s^3]
wpvp2, & ! w'v'^2 (thermodynamic levels) [m^3/s^3]
wp2up2, & ! w'^2 u'^2 (momentum levels) [m^4/s^4]
wp2vp2, & ! w'^2 v'^2 (momentum levels) [m^4/s^4]
ice_supersat_frac ! ice cloud fraction (thermo. levels) [-]
! Variables used to track perturbed version of winds.
real( kind = core_rknd ), intent(inout), dimension(ngrdcol,nz) :: &
um_pert, & ! perturbed <u> [m/s]
vm_pert, & ! perturbed <v> [m/s]
upwp_pert, & ! perturbed <u'w'> [m^2/s^2]
vpwp_pert ! perturbed <v'w'> [m^2/s^2]
type(pdf_parameter), intent(inout) :: &
pdf_params, & ! Fortran structure of PDF parameters on thermodynamic levels [units vary]
pdf_params_zm ! Fortran structure of PDF parameters on momentum levels [units vary]
type(implicit_coefs_terms), intent(inout) :: &
pdf_implicit_coefs_terms ! Implicit coefs / explicit terms [units vary]
#ifdef GFDL
real( kind = core_rknd ), intent(inout), dimension(ngrdcol,nz,sclr_dim) :: & ! h1g, 2010-06-16
sclrm_trsport_only ! Passive scalar concentration due to pure transport [{units vary}/s]
#endif
! Eddy passive scalar variable
real( kind = core_rknd ), intent(inout), dimension(ngrdcol,nz,edsclr_dim) :: &
edsclrm ! Eddy passive scalar grid-mean (thermo. levels) [units vary]
! Variables that need to be output for use in other parts of the CLUBB
! code, such as microphysics (rcm, pdf_params), forcings (rcm), and/or
! BUGSrad (cloud_cover).
real( kind = core_rknd ), intent(out), dimension(ngrdcol,nz) :: &
rcm_in_layer, & ! rcm within cloud layer [kg/kg]
cloud_cover ! cloud cover [-]
! Variables that need to be output for use in host models
real( kind = core_rknd ), intent(out), dimension(ngrdcol,nz) :: &
wprcp, & ! w'r_c' (momentum levels) [(kg/kg) m/s]
w_up_in_cloud, & ! Average cloudy updraft velocity [m/s]
w_down_in_cloud, & ! Average cloudy downdraft velocity [m/s]
cloudy_updraft_frac, & ! cloudy updraft fraction [-]
cloudy_downdraft_frac, & ! cloudy downdraft fraction [-]
invrs_tau_zm ! One divided by tau on zm levels [1/s]
real( kind = core_rknd ), dimension(ngrdcol,nz), intent(out) :: &
Kh_zt, & ! Eddy diffusivity coefficient on thermodynamic levels [m^2/s]
Kh_zm ! Eddy diffusivity coefficient on momentum levels [m^2/s]
#ifdef CLUBB_CAM
real( kind = core_rknd), intent(out), dimension(ngrdcol,nz) :: &
qclvar ! cloud water variance
#endif
real( kind = core_rknd ), dimension(ngrdcol,nz), intent(out) :: &
thlprcp ! thl'rc' [K kg/kg]
#ifdef GFDL
! hlg, 2010-06-16
real( kind = core_rknd ), intent(inout), dimension(ngrdcol,nz, min(1,sclr_dim) , 2) :: &
RH_crit ! critical relative humidity for droplet and ice nucleation
! ---> h1g, 2012-06-14
logical, intent(in) :: do_liquid_only_in_clubb
! <--- h1g, 2012-06-14
#endif
!--------------------------- Local Variables ---------------------------
integer :: i, k, j
#ifdef CLUBB_CAM
integer :: ixind
#endif
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
Skw_zm, & ! Skewness of w on momentum levels [-]
Skw_zt, & ! Skewness of w on thermodynamic levels [-]
thvm, & ! Virtual potential temperature [K]
thvm_zm, & ! Virtual potential temperature on momentum levs. [K]
ddzm_thvm_zm ! d(thvm_zm)/dz, centered over thermodynamic levs. [K/m]
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
rsat ! Saturation mixing ratio ! Brian
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
rtprcp, & ! rt'rc' [kg^2/kg^2]
rcp2 ! rc'^2 [kg^2/kg^2]
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
wpthlp2, & ! w'thl'^2 [m K^2/s]
wprtp2, & ! w'rt'^2 [m kg^2/kg^2]
wprtpthlp, & ! w'rt'thl' [m kg K/kg s]
wp2rcp, & ! w'^2 rc' [m^2 kg/kg s^2]
wp3_zm ! w'^3 [m^3/s^3]
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
Lscale, & ! Length scale [m]
Lscale_up, & ! Length scale (upwards component) [m]
Lscale_down, & ! Length scale (downwards component) [m]
Lscale_zm ! Length scale on momentum levels [m]
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
em, & ! Turbulent Kinetic Energy (TKE) [m^2/s^2]
tau_zm, & ! Eddy dissipation time scale on momentum levels [s]
tau_zt ! Eddy dissipation time scale on thermodynamic levels [s]
real( kind = core_rknd ), dimension(ngrdcol,nz,edsclr_dim) :: &
wpedsclrp ! w'edsclr'
real( kind = core_rknd ), dimension(ngrdcol,nz,sclr_dim) :: &
sclrprcp, & ! sclr'rc'
wp2sclrp, & ! w'^2 sclr'
wpsclrp2, & ! w'sclr'^2
wpsclrprtp, & ! w'sclr'rt'
wpsclrpthlp ! w'sclr'thl'
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
wp2_zt, & ! w'^2 on thermo. grid [m^2/s^2]
thlp2_zt, & ! thl'^2 on thermo. grid [K^2]
wpthlp_zt, & ! w'thl' on thermo. grid [m K/s]
wprtp_zt, & ! w'rt' on thermo. grid [m kg/(kg s)]
rtp2_zt, & ! rt'^2 on therm. grid [(kg/kg)^2]
rtpthlp_zt, & ! rt'thl' on thermo. grid [kg K/kg]
up2_zt, & ! u'^2 on thermo. grid [m^2/s^2]
vp2_zt, & ! v'^2 on thermo. grid [m^2/s^2]
upwp_zt, & ! u'w' on thermo. grid [m^2/s^2]
vpwp_zt ! v'w' on thermo. grid [m^2/s^2]
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
Skw_velocity, & ! Skewness velocity [m/s]
a3_coef, & ! The a3 coefficient from CLUBB eqns [-]
a3_coef_zt ! The a3 coefficient interpolated to the zt grid [-]
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
wp3_on_wp2, & ! w'^3 / w'^2 on the zm grid [m/s]
wp3_on_wp2_zt ! w'^3 / w'^2 on the zt grid [m/s]
! Eric Raut declared this variable solely for output to disk
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
rc_coef_zm ! Coefficient of X'r_c' in Eq. (34) on m-levs. [K/(kg/kg)]
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
Km_zm, & ! Eddy diffusivity for momentum on zm grid levels [m^2/s]
Kmh_zm ! Eddy diffusivity for thermodynamic variables [m^2/s]
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
gamma_Skw_fnc, & ! Gamma as a function of skewness [-]
sigma_sqd_w, & ! PDF width parameter (momentum levels) [-]
sigma_sqd_w_tmp, &
sigma_sqd_w_zt, & ! PDF width parameter (thermodynamic levels) [-]
sqrt_em_zt, & ! sqrt( em ) on zt levels; where em is TKE [m/s]
xp3_coef_fnc ! Coefficient in simple xp3 equation [-]
!Lscale_weight Uncomment this if you need to use this vairable at some point.
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
w_1_zm, & ! Mean w (1st PDF component) [m/s]
w_2_zm, & ! Mean w (2nd PDF component) [m/s]
varnce_w_1_zm, & ! Variance of w (1st PDF component) [m^2/s^2]
varnce_w_2_zm, & ! Variance of w (2nd PDF component) [m^2/s^2]
mixt_frac_zm ! Weight of 1st PDF component (Sk_w dependent) [-]
integer :: &
wprtp_cl_num, & ! Instance of w'r_t' clipping (1st or 3rd).
wpthlp_cl_num, & ! Instance of w'th_l' clipping (1st or 3rd).
wpsclrp_cl_num, & ! Instance of w'sclr' clipping (1st or 3rd).
upwp_cl_num, & ! Instance of u'w' clipping (1st or 2nd).
vpwp_cl_num ! Instance of v'w' clipping (1st or 2nd).
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
rcp2_zt, & ! r_c'^2 (on thermo. grid) [kg^2/kg^2]
cloud_frac_zm, & ! Cloud Fraction on momentum grid [-]
ice_supersat_frac_zm, & ! Ice Cloud Fraction on momentum grid [-]
rtm_zm, & ! Total water mixing ratio [kg/kg]
thlm_zm, & ! Liquid potential temperature [kg/kg]
rcm_zm, & ! Liquid water mixing ratio on m-levs. [kg/kg]
wpsclrp_zt, & ! Scalar flux on thermo. levels [un. vary]
sclrp2_zt ! Scalar variance on thermo.levels [un. vary]
real( kind = core_rknd ), dimension(ngrdcol) :: &
rtm_integral_before, &
rtm_integral_after, &
rtm_integral_forcing, &
rtm_flux_top, &
rtm_flux_sfc, &
rtm_spur_src, &
thlm_integral_before, &
thlm_integral_after, &
thlm_integral_forcing, &
thlm_flux_top, &
thlm_flux_sfc, &
thlm_spur_src
real( kind = core_rknd ), dimension(ngrdcol) :: &
thlm1000, &
thlm700
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
rcm_supersat_adj, & ! Adjustment to rcm due to spurious supersaturation
rel_humidity ! Relative humidity after PDF closure [-]
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
stability_correction, & ! Stability correction factor
invrs_tau_N2_zm, & ! Inverse tau with static stability correction applied [1/s]
invrs_tau_C6_zm, & ! Inverse tau values used for C6 (pr1) term in wpxp [1/s]
invrs_tau_C1_zm, & ! Inverse tau values used for C1 (dp1) term in wp2 [1/s]
invrs_tau_xp2_zm, & ! Inverse tau values used for advance_xp2_wpxp [s^-1]
invrs_tau_N2_iso, & ! Inverse tau values used for C4 when
! l_use_invrs_tau_N2_iso = .true. [s^-1]
invrs_tau_C4_zm, & ! Inverse tau values used for C4 terms [s^-1]
invrs_tau_C14_zm, & ! Inverse tau valuse used for C14 terms [s^-1]
invrs_tau_wp2_zm, & ! Inverse tau values used for advance_wp2_wpxp [s^-1]
invrs_tau_wpxp_zm, & ! invrs_tau_C6_zm = invrs_tau_wpxp_zm
invrs_tau_wp3_zm, & ! Inverse tau values used for advance_wp3_wp2 [s^-1]
invrs_tau_no_N2_zm, & ! One divided by tau (without N2) on zm levels [s^-1]
invrs_tau_bkgnd, & ! One divided by tau_wp3 [s^-1]
invrs_tau_shear, & ! One divided by tau with stability effects [s^-1]
invrs_tau_sfc, & ! One divided by tau (without N2) on zm levels [s^-1]
invrs_tau_zt, & ! Inverse time-scale tau on thermodynamics levels [1/s]
invrs_tau_wp3_zt, & ! Inverse tau wp3 at zt levels
Cx_fnc_Richardson, & ! Cx_fnc computed from Richardson_num [-]
brunt_vaisala_freq_sqd, & ! Buoyancy frequency squared, N^2 [s^-2]
brunt_vaisala_freq_sqd_mixed, & ! A mixture of dry and moist N^2 [s^-2]
brunt_vaisala_freq_sqd_dry, & ! dry N^2 [s^-2]
brunt_vaisala_freq_sqd_moist, & ! moist N^2 [s^-2]
brunt_vaisala_freq_sqd_splat, & ! [s^-2]
brunt_vaisala_freq_sqd_zt, & ! Buoyancy frequency squared on t-levs. [s^-2]
Ri_zm ! Richardson number [-]
real( kind = core_rknd ), parameter :: &
ufmin = 0.01_core_rknd ! minimum value of friction velocity [m/s]
real( kind = core_rknd ), dimension(ngrdcol) :: &
Lscale_max ! Max. allowable mixing length (based on grid box size) [m]
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
tau_max_zm, & ! Max. allowable eddy dissipation time scale on m-levs [s]
tau_max_zt ! Max. allowable eddy dissipation time scale on t-levs [s]
real( kind = core_rknd ), dimension(ngrdcol) :: newmu
real( kind = core_rknd ) :: below_grnd_val = 0.01_core_rknd
real( kind = core_rknd ) :: &
taumax, & ! CLUBB tunable parameter taumax
c_K, & ! CLUBB tunable parameter c_K
gamma_coef, & ! CLUBB tunable parameter gamma_coef
gamma_coefb, & ! CLUBB tunable parameter gamma_coefb
gamma_coefc, & ! CLUBB tunable parameter gamma_coefc
xp3_coef_base, & ! CLUBB tunable parameter xp3_coef_base
xp3_coef_slope, & ! CLUBB tunable parameter xp3_coef_slope
beta, & ! CLUBB tunable parameter beta
Skw_denom_coef, & ! CLUBB tunable parameter Skw_denom_coef
Skw_max_mag, & ! CLUBB tunable parameter Skw_max_mag
mu, &
a3_coef_min, &
C_K10, &
C_K10h
! Flag to sample stats in a particular call to subroutine
! pdf_closure_driver.
logical :: l_samp_stats_in_pdf_call
! Flag to determine whether invrs_tau_N2_iso is used in C4 terms.
! Important! This flag is only in use when l_diag_Lscale_from_tau = true
! Setting l_use_invrs_tau_N2_iso = true will not change anything unless
! l_diag_Lscale_from_tau is also true
logical, parameter :: l_use_invrs_tau_N2_iso = .false.
real( kind = core_rknd ), dimension(ngrdcol,nz) :: &
lhs_splat_wp2, & ! LHS coefficient of wp2 splatting term [1/s]
lhs_splat_wp3 ! LHS coefficient of wp3 splatting term [1/s]
! Variables associated with upgradient momentum contributions due to cumuli
!real( kind = core_rknd ), dimension(nz) :: &
! Km_Skw_factor ! Factor, with value < 1, that reduces eddy diffusivity,
! Km_zm, in skewed layers
!real( kind = core_rknd ),parameter :: &
! Km_Skw_thresh = zero_threshold, & ! Value of Skw at which Skw correction kicks in
! Km_Skw_factor_efold = 0.5_core_rknd, & ! E-folding rate of exponential Skw correction
! Km_Skw_factor_min = 0.2_core_rknd ! Minimum value of Km_Skw_factor
integer, intent(out) :: &
err_code_out ! Error code indicator
type(pdf_parameter) :: pdf_params_single_col(ngrdcol), &
pdf_params_zm_single_col(ngrdcol)
integer :: advance_order_loop_iter
integer :: smth_type = 2 ! Used for Lscale_width_vert_avg
!----- Begin Code -----
!$acc data copyin( gr, gr%zm, gr%zt, gr%dzm, gr%dzt, gr%invrs_dzt, gr%invrs_dzm, &
!$acc gr%weights_zt2zm, gr%weights_zm2zt, &
!$acc nu_vert_res_dep, nu_vert_res_dep%nu2, nu_vert_res_dep%nu9, &
!$acc nu_vert_res_dep%nu1, nu_vert_res_dep%nu8, nu_vert_res_dep%nu10, &
!$acc nu_vert_res_dep%nu6, &
!$acc pdf_params, pdf_params_zm, &
!$acc fcor, sfc_elevation, thlm_forcing, rtm_forcing, um_forcing, &
!$acc vm_forcing, wprtp_forcing, wpthlp_forcing, rtp2_forcing, thlp2_forcing, &
!$acc rtpthlp_forcing, wm_zm, wm_zt, rho_zm, rho, rho_ds_zm, rho_ds_zt, &
!$acc invrs_rho_ds_zm, invrs_rho_ds_zt, thv_ds_zm, thv_ds_zt, rfrzm, &
!$acc radf, wpthlp_sfc, &
!$acc wprtp_sfc, upwp_sfc, vpwp_sfc, sclrm_forcing, wpsclrp_sfc, edsclrm_forcing, &
!$acc wpedsclrp_sfc, upwp_sfc_pert, vpwp_sfc_pert, rtm_ref, thlm_ref, um_ref, &
#ifdef CLUBBND_CAM
!$acc varmu, &
#endif
!$acc vm_ref, ug, vg, host_dx, host_dy ) &
!$acc copy( um, upwp, vm, vpwp, up2, vp2, up3, vp3, rtm, wprtp, thlm, wpthlp, rtp2, &
!$acc rtp3, thlp2, thlp3, rtpthlp, wp2, wp3, sclrm, wpsclrp, sclrp2, sclrp3, &
!$acc sclrprtp, sclrpthlp, p_in_Pa, exner, rcm, cloud_frac, wpthvp, wp2thvp, &
!$acc rtpthvp, thlpthvp, sclrpthvp, wp2rtp, wp2thlp, uprcp, vprcp, rc_coef, &
!$acc wp4, wpup2, wpvp2, wp2up2, wp2vp2, ice_supersat_frac, um_pert, &
!$acc vm_pert, upwp_pert, vpwp_pert, &
#ifdef GFDL
!$acc sclrm_trsport_only, &
#endif
!$acc edsclrm, &
!$acc pdf_params%w_1, pdf_params%w_2, &
!$acc pdf_params%varnce_w_1, pdf_params%varnce_w_2, &
!$acc pdf_params%rt_1, pdf_params%rt_2, &
!$acc pdf_params%varnce_rt_1, pdf_params%varnce_rt_2, &
!$acc pdf_params%thl_1, pdf_params%thl_2, &
!$acc pdf_params%varnce_thl_1, pdf_params%varnce_thl_2, &
!$acc pdf_params%corr_w_rt_1, pdf_params%corr_w_rt_2, &
!$acc pdf_params%corr_w_thl_1, pdf_params%corr_w_thl_2, &
!$acc pdf_params%corr_rt_thl_1, pdf_params%corr_rt_thl_2,&
!$acc pdf_params%alpha_thl, pdf_params%alpha_rt, &
!$acc pdf_params%crt_1, pdf_params%crt_2, pdf_params%cthl_1, &
!$acc pdf_params%cthl_2, pdf_params%chi_1, &
!$acc pdf_params%chi_2, pdf_params%stdev_chi_1, &
!$acc pdf_params%stdev_chi_2, pdf_params%stdev_eta_1, &
!$acc pdf_params%stdev_eta_2, pdf_params%covar_chi_eta_1, &
!$acc pdf_params%covar_chi_eta_2, pdf_params%corr_w_chi_1, &
!$acc pdf_params%corr_w_chi_2, pdf_params%corr_w_eta_1, &
!$acc pdf_params%corr_w_eta_2, pdf_params%corr_chi_eta_1, &
!$acc pdf_params%corr_chi_eta_2, pdf_params%rsatl_1, &
!$acc pdf_params%rsatl_2, pdf_params%rc_1, pdf_params%rc_2, &
!$acc pdf_params%cloud_frac_1, pdf_params%cloud_frac_2, &
!$acc pdf_params%mixt_frac, pdf_params%ice_supersat_frac_1, &
!$acc pdf_params%ice_supersat_frac_2, &
!$acc pdf_params_zm%w_1, pdf_params_zm%w_2, &
!$acc pdf_params_zm%varnce_w_1, pdf_params_zm%varnce_w_2, &
!$acc pdf_params_zm%rt_1, pdf_params_zm%rt_2, &
!$acc pdf_params_zm%varnce_rt_1, pdf_params_zm%varnce_rt_2, &
!$acc pdf_params_zm%thl_1, pdf_params_zm%thl_2, &
!$acc pdf_params_zm%varnce_thl_1, pdf_params_zm%varnce_thl_2, &
!$acc pdf_params_zm%corr_w_rt_1, pdf_params_zm%corr_w_rt_2, &
!$acc pdf_params_zm%corr_w_thl_1, pdf_params_zm%corr_w_thl_2, &
!$acc pdf_params_zm%corr_rt_thl_1, pdf_params_zm%corr_rt_thl_2,&
!$acc pdf_params_zm%alpha_thl, pdf_params_zm%alpha_rt, &
!$acc pdf_params_zm%crt_1, pdf_params_zm%crt_2, pdf_params_zm%cthl_1, &
!$acc pdf_params_zm%cthl_2, pdf_params_zm%chi_1, &
!$acc pdf_params_zm%chi_2, pdf_params_zm%stdev_chi_1, &
!$acc pdf_params_zm%stdev_chi_2, pdf_params_zm%stdev_eta_1, &
!$acc pdf_params_zm%stdev_eta_2, pdf_params_zm%covar_chi_eta_1, &
!$acc pdf_params_zm%covar_chi_eta_2, pdf_params_zm%corr_w_chi_1, &
!$acc pdf_params_zm%corr_w_chi_2, pdf_params_zm%corr_w_eta_1, &
!$acc pdf_params_zm%corr_w_eta_2, pdf_params_zm%corr_chi_eta_1, &
!$acc pdf_params_zm%corr_chi_eta_2, pdf_params_zm%rsatl_1, &
!$acc pdf_params_zm%rsatl_2, pdf_params_zm%rc_1, pdf_params_zm%rc_2, &
!$acc pdf_params_zm%cloud_frac_1, pdf_params_zm%cloud_frac_2, &
!$acc pdf_params_zm%mixt_frac, pdf_params_zm%ice_supersat_frac_1, &
!$acc pdf_params_zm%ice_supersat_frac_2 ) &
!$acc copyout( rcm_in_layer, cloud_cover, wprcp, w_up_in_cloud, w_down_in_cloud, &
!$acc cloudy_updraft_frac, cloudy_downdraft_frac, invrs_tau_zm, Kh_zt, &
!$acc Kh_zm, &
#ifdef CLUBB_CAM
!$acc qclvar, &
#endif
!$acc thlprcp )
!$acc enter data create( Skw_zm, Skw_zt, thvm, thvm_zm, ddzm_thvm_zm, rtprcp, rcp2, &
!$acc wpthlp2, wprtp2, wprtpthlp, wp2rcp, wp3_zm, Lscale, Lscale_up, Lscale_zm, &
!$acc Lscale_down, em, tau_zm, tau_zt, &
!$acc wp2_zt, thlp2_zt, wpthlp_zt, &
!$acc wprtp_zt, rtp2_zt, rtpthlp_zt, up2_zt, vp2_zt, upwp_zt, vpwp_zt, &
!$acc Skw_velocity, a3_coef, a3_coef_zt, wp3_on_wp2, wp3_on_wp2_zt, &
!$acc rc_coef_zm, Km_zm, Kmh_zm, gamma_Skw_fnc, sigma_sqd_w, sigma_sqd_w_tmp, sigma_sqd_w_zt, &
!$acc sqrt_em_zt, xp3_coef_fnc, w_1_zm, w_2_zm, varnce_w_1_zm, varnce_w_2_zm, &
!$acc mixt_frac_zm, rcp2_zt, cloud_frac_zm, ice_supersat_frac_zm, rtm_zm, &
!$acc thlm_zm, rcm_zm, thlm1000, thlm700, &
!$acc rcm_supersat_adj, stability_correction, invrs_tau_N2_zm, &
!$acc invrs_tau_C6_zm, invrs_tau_C1_zm, invrs_tau_xp2_zm, invrs_tau_N2_iso, &
!$acc invrs_tau_C4_zm, invrs_tau_C14_zm, invrs_tau_wp2_zm, invrs_tau_wpxp_zm, &
!$acc invrs_tau_wp3_zm, invrs_tau_no_N2_zm, invrs_tau_bkgnd, invrs_tau_shear, &
!$acc invrs_tau_sfc, invrs_tau_zt, invrs_tau_wp3_zt, Cx_fnc_Richardson, &
!$acc brunt_vaisala_freq_sqd, brunt_vaisala_freq_sqd_mixed, &
!$acc brunt_vaisala_freq_sqd_dry, brunt_vaisala_freq_sqd_moist, &
!$acc brunt_vaisala_freq_sqd_splat, &
!$acc brunt_vaisala_freq_sqd_zt, Ri_zm, Lscale_max, &
!$acc tau_max_zm, tau_max_zt, newmu, lhs_splat_wp2, lhs_splat_wp3 )
!$acc enter data if( sclr_dim > 0 ) &
!$acc create( wpedsclrp, sclrprcp, wp2sclrp, &
!$acc wpsclrp2, wpsclrprtp, wpsclrpthlp, wpsclrp_zt, sclrp2_zt )
!$acc enter data if( sclr_dim > 0 ) &
!$acc create( hydromet, wphydrometp, wp2hmp, rtphmp_zt, thlphmp_zt )
if ( clubb_config_flags%l_lmm_stepping ) then
dt_advance = two * dt
else
dt_advance = dt
end if
err_code_out = clubb_no_error ! Initialize to no error value
mu = clubb_params(imu)
a3_coef_min = clubb_params(ia3_coef_min)
C_K10 = clubb_params(ic_K10)
C_K10h = clubb_params(ic_K10h)
! Determine the maximum allowable value for Lscale (in meters).
call set_Lscale_max( ngrdcol, l_implemented, host_dx, host_dy, & ! intent(in)
Lscale_max ) ! intent(out)
if ( stats_metadata%l_stats .and. stats_metadata%l_stats_samp ) then
!$acc update host( wm_zt, wm_zm, rho_ds_zt, rtm, gr%dzt, &
!$acc rtm, thlm )
! Spurious source will only be calculated if rtm_ma and thlm_ma are zero.
! Therefore, wm must be zero or l_implemented must be true.
do i = 1, ngrdcol
if ( l_implemented .or. ( all( abs(wm_zt(i,:)) < eps ) .and. &
all( abs(wm_zm(i,:)) < eps ) ) ) then
! Get the vertical integral of rtm and thlm before this function begins
! so that spurious source can be calculated
rtm_integral_before(i) &
= vertical_integral( (nz - 2 + 1), rho_ds_zt(i,2:nz), &
rtm(i,2:nz), gr%dzt(i,2:nz) )
thlm_integral_before(i) &
= vertical_integral( (nz - 2 + 1), rho_ds_zt(i,2:nz), &
thlm(i,2:nz), gr%dzt(i,2:nz) )
end if
end do
end if
!----------------------------------------------------------------
! Test input variables
!----------------------------------------------------------------
if ( clubb_at_least_debug_level( 2 ) ) then
!$acc update host( thlm_forcing, rtm_forcing, um_forcing, vm_forcing, &
!$acc wm_zm, wm_zt, p_in_Pa, rho_zm, rho, exner, rho_ds_zm, &
!$acc rho_ds_zt, invrs_rho_ds_zm, invrs_rho_ds_zt, thv_ds_zm, &
!$acc thv_ds_zt, wpthlp_sfc, wprtp_sfc, upwp_sfc, vpwp_sfc, &
!$acc um, upwp, vm, vpwp, up2, vp2, rtm, wprtp, thlm, wpthlp, &
!$acc wp2, wp3, rtp2, thlp2, rtpthlp, wpsclrp_sfc, wpedsclrp_sfc, &
!$acc sclrm, wpsclrp, sclrp2, sclrprtp, sclrpthlp, sclrm_forcing, &
!$acc edsclrm, edsclrm_forcing )
do i = 1, ngrdcol