Newer
Older
PROGRAM MAIN_ICE_ADJUST
USE XRD_GETOPTIONS
USE GETDATA_ICE_ADJUST_MOD
USE MODI_ICE_ADJUST
USE MODD_DIMPHYEX, ONLY: DIMPHYEX_t
RODIER Quentin
committed
USE MODD_IO, ONLY: TFILEDATA
USE MODD_PHYEX, ONLY: PHYEX_t
USE STACK_MOD
USE OMP_LIB
USE YOMHOOK, ONLY : LHOOK, DR_HOOK
USE PARKIND1, ONLY : JPRB, JPIM
IMPLICIT NONE
INTEGER :: KLON
INTEGER :: KLEV
INTEGER :: KRR
REAL, ALLOCATABLE :: PRHODJ (:,:,:,:)
REAL, ALLOCATABLE :: PEXNREF (:,:,:,:)
REAL, ALLOCATABLE :: PRHODREF (:,:,:,:)
REAL, ALLOCATABLE :: PPABSM (:,:,:,:)
REAL, ALLOCATABLE :: PTHT (:,:,:,:)
REAL, ALLOCATABLE :: PSIGS (:,:,:,:)
REAL, ALLOCATABLE :: PMFCONV (:,:,:,:)
REAL, ALLOCATABLE :: PRC_MF (:,:,:,:)
REAL, ALLOCATABLE :: PRI_MF (:,:,:,:)
REAL, ALLOCATABLE :: PCF_MF (:,:,:,:)
REAL, ALLOCATABLE :: PTHS (:,:,:,:)
REAL, ALLOCATABLE :: PRS (:,:,:,:,:)
REAL, ALLOCATABLE :: PSRCS (:,:,:,:)
REAL, ALLOCATABLE :: PCLDFR (:,:,:,:)
REAL, ALLOCATABLE :: PHLC_HRC (:,:,:,:)
REAL, ALLOCATABLE :: PHLC_HCF (:,:,:,:)
REAL, ALLOCATABLE :: PHLI_HRI (:,:,:,:)
REAL, ALLOCATABLE :: PHLI_HCF (:,:,:,:)
REAL, ALLOCATABLE :: ZRS (:,:,:,:,:)
REAL, ALLOCATABLE :: ZZZ (:,:,:,:)
REAL, ALLOCATABLE :: ZSIGQSAT (:,:,:)
REAL, ALLOCATABLE :: ZICE_CLD_WGT (:,:,:)
REAL, ALLOCATABLE :: ZDUM1 (:,:,:,:)
REAL, ALLOCATABLE :: ZDUM2 (:,:,:,:)
REAL, ALLOCATABLE :: ZDUM3 (:,:,:,:)
REAL, ALLOCATABLE :: ZDUM4 (:,:,:,:)
REAL, ALLOCATABLE :: ZDUM5 (:,:,:,:)
REAL, ALLOCATABLE :: PRS_OUT (:,:,:,:,:)
REAL, ALLOCATABLE :: PSRCS_OUT (:,:,:,:)
REAL, ALLOCATABLE :: PCLDFR_OUT (:,:,:,:)
REAL, ALLOCATABLE :: PHLC_HRC_OUT (:,:,:,:)
REAL, ALLOCATABLE :: PHLC_HCF_OUT (:,:,:,:)
REAL, ALLOCATABLE :: PHLI_HRI_OUT (:,:,:,:)
REAL, ALLOCATABLE :: PHLI_HCF_OUT (:,:,:,:)
INTEGER :: NPROMA, NGPBLKS, NFLEVG
INTEGER :: IBL, JLON, JLEV
TYPE(DIMPHYEX_t) :: D, D0
TYPE(PHYEX_t) :: PHYEX
RODIER Quentin
committed
TYPE(TFILEDATA) :: TPFILE
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
LOGICAL :: LLCHECK
LOGICAL :: LLCHECKDIFF
LOGICAL :: LLDIFF
INTEGER :: IBLOCK1, IBLOCK2
INTEGER :: ISTSZ, JBLK1, JBLK2
INTEGER :: NTID, ITID
REAL, ALLOCATABLE :: PSTACK(:,:)
TYPE (STACK) :: YLSTACK
REAL(KIND=8) :: TS,TE
REAL(KIND=8) :: TSC, TEC, TSD, TED, ZTC, ZTD
INTEGER :: ITIME, NTIME
INTEGER :: IRANK, ISIZE
LOGICAL :: LLVERBOSE, LLSTAT, LLBIND
REAL (KIND=JPRB) :: ZHOOK_HANDLE
CALL INITOPTIONS ()
NGPBLKS = 296
CALL GETOPTION ("--blocks", NGPBLKS)
NPROMA = 32
CALL GETOPTION ("--nproma", NPROMA)
NFLEVG = -1
CALL GETOPTION ("--nflevg", NFLEVG)
CALL GETOPTION ("--check", LLCHECK)
CALL GETOPTION ("--checkdiff", LLCHECKDIFF)
IBLOCK1 = 1
CALL GETOPTION ("--check-block-1", IBLOCK1)
IBLOCK2 = NGPBLKS
CALL GETOPTION ("--check-block-2", IBLOCK2)
CALL GETOPTION ("--stat", LLSTAT)
NTIME = 1
CALL GETOPTION ("--times", NTIME)
CALL GETOPTION ("--verbose", LLVERBOSE)
CALL GETOPTION ("--bind", LLBIND)
CALL CHECKOPTIONS ()
LLDIFF = .FALSE.
IRANK = 0
ISIZE = 1
IF (LLBIND) THEN
CALL LINUX_BIND (IRANK, ISIZE)
CALL LINUX_BIND_DUMP (IRANK, ISIZE)
ENDIF
CALL GETDATA_ICE_ADJUST (NPROMA, NGPBLKS, NFLEVG, PRHODJ, PEXNREF, PRHODREF, PPABSM, PTHT, ZICE_CLD_WGT, &
& ZSIGQSAT, PSIGS, PMFCONV, PRC_MF, PRI_MF, PCF_MF, ZDUM1, ZDUM2, ZDUM3, ZDUM4, ZDUM5, &
& PTHS, PRS, PSRCS, PCLDFR, PHLC_HRC, PHLC_HCF, &
& PHLI_HRI, PHLI_HCF, ZRS, ZZZ, PRS_OUT, PSRCS_OUT, PCLDFR_OUT, PHLC_HRC_OUT, PHLC_HCF_OUT, &
& PHLI_HRI_OUT, PHLI_HCF_OUT, LLVERBOSE)
KLEV = SIZE (PRS, 3)
KRR = SIZE (PRS, 4)
IF (LLVERBOSE) PRINT *, " KLEV = ", KLEV, " KRR = ", KRR
PRINT *, " NPROMA = ", NPROMA, " KLEV = ", KLEV, " NGPBLKS = ", NGPBLKS
CALL INIT_PHYEX(KRR, PHYEX)
D0%NIT = NPROMA
D0%NIB = 1
D0%NIE = NPROMA
D0%NJT = 1
D0%NJB = 1
D0%NJE = 1
D0%NIJT = D0%NIT * D0%NJT
D0%NIJB = 1
D0%NIJE = NPROMA
D0%NKL = -1
D0%NKT = KLEV
D0%NKA = KLEV
D0%NKU = 1
D0%NKB = KLEV
D0%NKE = 1
D0%NKTB = 1
D0%NKTE = KLEV
ISTSZ = NPROMA * 20 * KLEV
ALLOCATE (PSTACK (ISTSZ, NGPBLKS))
TS = OMP_GET_WTIME ()
ZTD = 0.
ZTC = 0.
IF (LHOOK) CALL DR_HOOK ('MAIN',0,ZHOOK_HANDLE)
DO ITIME = 1, NTIME
TSD = OMP_GET_WTIME ()
!$acc data &
!$acc & copyin (D0, PHYEX, &
!$acc & ZSIGQSAT, PRHODJ, PEXNREF, PRHODREF, PSIGS, PMFCONV, PPABSM, ZZZ, PCF_MF, PRC_MF, PRI_MF, ZDUM1, ZDUM2, ZDUM3, ZDUM4, ZDUM5, ZRS, ZICE_CLD_WGT) &
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
!$acc & copy (PRS, PTHS), &
!$acc & copyout (PSRCS, PCLDFR, PHLC_HRC, PHLC_HCF, PHLI_HRI, PHLI_HCF) &
!$acc & create (PSTACK)
TSC = OMP_GET_WTIME ()
#ifdef USE_OPENMP
!$OMP PARALLEL PRIVATE (D, ITID, JBLK1, JBLK2)
#endif
#ifdef _OPENACC
JBLK1 = 1
JBLK2 = NGPBLKS
#endif
#ifdef USE_OPENMP
NTID = OMP_GET_MAX_THREADS ()
ITID = OMP_GET_THREAD_NUM ()
JBLK1 = 1 + (NGPBLKS * (ITID+0)) / NTID
JBLK2 = (NGPBLKS * (ITID+1)) / NTID
!PRINT *, ITID, JBLK1, JBLK2
#endif
!$acc parallel loop gang vector private (YLSTACK, IBL, JLON, D) collapse (2)
DO IBL = JBLK1, JBLK2
#ifdef _OPENACC
DO JLON = 1, NPROMA
D = D0
D%NIB = JLON
D%NIE = JLON
D%NIJB = JLON
D%NIJE = JLON
#endif
#ifdef USE_OPENMP
D = D0
#endif
#ifdef USE_STACK
YLSTACK%L = LOC (PSTACK (1, IBL))
YLSTACK%U = YLSTACK%L + ISTSZ * KIND (PSTACK)
#else
YLSTACK%L = 0
YLSTACK%U = 0
#endif
CALL ICE_ADJUST (D, PHYEX%CST, PHYEX%RAIN_ICE_PARAMN, PHYEX%NEBN, PHYEX%TURBN, PHYEX%PARAM_ICEN, &
& PHYEX%MISC%TBUCONF, PHYEX%MISC%KRR, PHYEX%MISC%HBUNAME, &
& PHYEX%MISC%PTSTEP, ZSIGQSAT (:, :, IBL), PRHODJ=PRHODJ (:, :, :, IBL), &
RODIER Quentin
committed
& PEXNREF=PEXNREF (:, :, :, IBL), &
& PRHODREF=PRHODREF (:, :, :, IBL), PSIGS=PSIGS (:, :, :, IBL), LMFCONV=PHYEX%MISC%LMFCONV, PMFCONV=PMFCONV (:, :, :, IBL), &
& PPABST=PPABSM (:, :, :, IBL), PZZ=ZZZ (:, :, :, IBL), PEXN=PEXNREF (:, :, :, IBL), PCF_MF=PCF_MF (:, :, :, IBL), &
& PRC_MF=PRC_MF (:, :, :, IBL), PRI_MF=PRI_MF (:, :, :, IBL), &
& PICLDFR=ZDUM1(:, :, :, IBL), PWCLDFR=ZDUM2(:, :, :, IBL), PSSIO=ZDUM3(:, :, :, IBL), &
RODIER Quentin
committed
& PSSIU=ZDUM4(:, :, :, IBL), PIFR=ZDUM5(:, :, :, IBL), &
& PRV=ZRS(:, :, :, 1, IBL), PRC=ZRS(:, :, :, 2, IBL), &
& PRVS=PRS(:, :, :, 1, IBL), PRCS=PRS(:, :, :, 2, IBL), PTH=ZRS(:, :, :, 0, IBL), PTHS=PTHS (:, :, :, IBL), &
& OCOMPUTE_SRC=PHYEX%MISC%OCOMPUTE_SRC, &
& PSRCS=PSRCS (:, :, :, IBL), PCLDFR=PCLDFR (:, :, :, IBL), PRR=ZRS(:, :, :, 3, IBL), PRI=ZRS(:, :, :, 4, IBL), &
& PRIS=PRS(:, :, :, 4, IBL), PRS=ZRS(:, :, :, 5, IBL), PRG=ZRS(:, :, :, 6, IBL), &
& TBUDGETS=PHYEX%MISC%YLBUDGET, KBUDGETS=PHYEX%MISC%NBUDGET, &
& PICE_CLD_WGT=ZICE_CLD_WGT(:, :, IBL), &
& PHLC_HRC=PHLC_HRC(:, :, :, IBL), PHLC_HCF=PHLC_HCF(:, :, :, IBL), &
& PHLI_HRI=PHLI_HRI(:, :, :, IBL), PHLI_HCF=PHLI_HCF(:, :, :, IBL) &
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
#ifdef USE_STACK
& , YDSTACK=YLSTACK &
#endif
& )
#ifdef _OPENACC
ENDDO
#endif
ENDDO
#ifdef USE_OPENMP
!$OMP END PARALLEL
#endif
!$acc end parallel loop
TEC = OMP_GET_WTIME ()
!$acc end data
TED = OMP_GET_WTIME ()
ZTC = ZTC + (TEC - TSC)
ZTD = ZTD + (TED - TSD)
ENDDO
IF (LHOOK) CALL DR_HOOK ('MAIN',1,ZHOOK_HANDLE)
TE = OMP_GET_WTIME()
WRITE (*,'(A,F8.2,A)') 'elapsed time : ',TE-TS,' s'
WRITE (*,'(A,F8.4,A)') ' i.e. ',1000.*(TE-TS)/(NPROMA*NGPBLKS)/NTIME,' ms/gp'
PRINT *, " ZTD = ", ZTD, ZTD / REAL (NPROMA*NGPBLKS*NTIME)
PRINT *, " ZTC = ", ZTC, ZTC / REAL (NPROMA*NGPBLKS*NTIME)
IF (LLCHECK .OR. LLSTAT .OR. LLCHECKDIFF) THEN
DO IBL = IBLOCK1, IBLOCK2
PRINT *, " IBL = ", IBL
CALL DIFF ("PSRCS", PSRCS_OUT (:,:,:,IBL), PSRCS (:,:,:,IBL), LLSTAT, LLCHECK, NPROMA, LLCHECKDIFF, LLDIFF)
CALL DIFF ("PCLDFR", PCLDFR_OUT (:,:,:,IBL), PCLDFR (:,:,:,IBL), LLSTAT, LLCHECK, NPROMA, LLCHECKDIFF, LLDIFF)
CALL DIFF ("PHLC_HRC", PHLC_HRC_OUT (:,:,:,IBL), PHLC_HRC (:,:,:,IBL), LLSTAT, LLCHECK, NPROMA, LLCHECKDIFF, LLDIFF)
CALL DIFF ("PHLC_HCF", PHLC_HCF_OUT (:,:,:,IBL), PHLC_HCF (:,:,:,IBL), LLSTAT, LLCHECK, NPROMA, LLCHECKDIFF, LLDIFF)
CALL DIFF ("PHLI_HRI", PHLI_HRI_OUT (:,:,:,IBL), PHLI_HRI (:,:,:,IBL), LLSTAT, LLCHECK, NPROMA, LLCHECKDIFF, LLDIFF)
CALL DIFF ("PHLI_HCF", PHLI_HCF_OUT (:,:,:,IBL), PHLI_HCF (:,:,:,IBL), LLSTAT, LLCHECK, NPROMA, LLCHECKDIFF, LLDIFF)
ENDDO
ENDIF
IF (LLCHECKDIFF) THEN
IF (LLDIFF) THEN
PRINT*, "THERE ARE DIFF SOMEWHERE"
ELSE
PRINT*, "THERE IS NO DIFF AT ALL"
ENDIF
ENDIF
STOP
SUBROUTINE INIT_PHYEX(KRR, PHYEX)
USE MODD_BUDGET, ONLY: TBUCONF_ASSOCIATE, NBUDGET_RI, TBUCONF, LBU_ENABLE, LBUDGET_U, LBUDGET_V, LBUDGET_W, LBUDGET_TH, &
LBUDGET_TKE, LBUDGET_RV, LBUDGET_RC, LBUDGET_RR, LBUDGET_RI, LBUDGET_RS, LBUDGET_RG, LBUDGET_RH, LBUDGET_SV
USE MODD_PHYEX, ONLY: PHYEX_t
USE MODI_INI_PHYEX, ONLY: INI_PHYEX
IMPLICIT NONE
! -----------------------------------------------------------------------
! DUMMY VARIABLES
INTEGER, INTENT(IN) :: KRR
TYPE(PHYEX_t), INTENT(OUT) :: PHYEX
!-----------------------------------------------------------------------
! LOCAL VARIABLES
INTEGER :: IULOUT, JRR
REAL :: ZDZMIN
CHARACTER(LEN=6) :: CPROGRAM
CHARACTER(LEN=4) :: CMICRO, CSCONV, CTURB
REAL :: PTSTEP
! -----------------------------------------------------------------------
IULOUT=20
CMICRO='ICE3'
CSCONV='NONE'
PTSTEP = 50.000000000000000
RODIER Quentin
committed
TPFILE%NLU=0
RODIER Quentin
committed
CALL INI_PHYEX(CPROGRAM, TPFILE, .TRUE., IULOUT, 0, 1, &
&CMICRO, CSCONV, CTURB, &
&LDDEFAULTVAL=.TRUE., LDREADNAM=.FALSE., LDCHECK=.FALSE., KPRINT=0, LDINIT=.FALSE., &
&PHYEX_OUT=PHYEX)
!Control parameters
PHYEX%MISC%PTSTEP = PTSTEP
PHYEX%MISC%HBUNAME = 'DEPI'
PHYEX%MISC%LMFCONV = .TRUE.
PHYEX%MISC%KRR = KRR
PHYEX%MISC%OCOMPUTE_SRC = .TRUE.
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
PHYEX%PARAM_ICEN%LCRIAUTI=.TRUE.
PHYEX%PARAM_ICEN%XCRIAUTI_NAM=0.2E-3
PHYEX%PARAM_ICEN%XT0CRIAUTI_NAM=-5.
PHYEX%PARAM_ICEN%XCRIAUTC_NAM=0.1E-2
PHYEX%PARAM_ICEN%LOCND2=.FALSE.
PHYEX%PARAM_ICEN%CSEDIM='STAT'
PHYEX%PARAM_ICEN%LWARM=.TRUE.
PHYEX%PARAM_ICEN%LSEDIC=.TRUE.
PHYEX%PARAM_ICEN%CSNOWRIMING='M90 '
PHYEX%PARAM_ICEN%XFRACM90=0.1 ! Fraction used for the Murakami 1990 formulation
PHYEX%PARAM_ICEN%LCONVHG=.TRUE. ! TRUE to allow the conversion from hail to graupel
PHYEX%PARAM_ICEN%LCRFLIMIT=.TRUE. !True to limit rain contact freezing to possible heat exchange
PHYEX%PARAM_ICEN%LFEEDBACKT=.TRUE. ! When .TRUE. feed back on temperature is taken into account
PHYEX%PARAM_ICEN%LEVLIMIT=.TRUE. ! When .TRUE. water vapour pressure is limited by saturation
PHYEX%PARAM_ICEN%LNULLWETG=.TRUE. ! When .TRUE. graupel wet growth is activated with null rate (to allow water shedding)
PHYEX%PARAM_ICEN%LWETGPOST=.TRUE. ! When .TRUE. graupel wet growth is activated with positive temperature (to allow water shedding)
PHYEX%PARAM_ICEN%LNULLWETH=.TRUE. ! Same as LNULLWETG but for hail
PHYEX%PARAM_ICEN%LWETHPOST=.TRUE. ! Same as LWETGPOST but for hail
PHYEX%PARAM_ICEN%LSEDIM_AFTER=.FALSE. ! Sedimentation done after microphysics
PHYEX%PARAM_ICEN%XSPLIT_MAXCFL=0.8
PHYEX%PARAM_ICEN%LDEPOSC=.FALSE. ! water deposition on vegetation
PHYEX%PARAM_ICEN%XVDEPOSC=0.02 ! deposition speed (2 cm.s-1)
PHYEX%PARAM_ICEN%CSUBG_RC_RR_ACCR='NONE'
PHYEX%PARAM_ICEN%CSUBG_RR_EVAP='NONE'
PHYEX%PARAM_ICEN%CSUBG_PR_PDF='SIGM'
PHYEX%NEBN%LSUBG_COND = .TRUE.
PHYEX%NEBN%LSIGMAS = .TRUE.
PHYEX%NEBN%CFRAC_ICE_ADJUST='S' ! Ice/liquid partition rule to use in adjustment
PHYEX%NEBN%CFRAC_ICE_SHALLOW_MF='S' ! Ice/liquid partition rule to use in shallow_mf
RODIER Quentin
committed
CALL INI_PHYEX(CPROGRAM, TPFILE, .TRUE., IULOUT, 0, 1, &
&CMICRO, CSCONV, CTURB, &
&LDDEFAULTVAL=.FALSE., LDREADNAM=.FALSE., LDCHECK=.TRUE., KPRINT=2, LDINIT=.TRUE., &
&PHYEX_IN=PHYEX, PHYEX_OUT=PHYEX)
!Budgets
PHYEX%MISC%NBUDGET=NBUDGET_RI
DO JRR=1, PHYEX%MISC%NBUDGET
PHYEX%MISC%YLBUDGET(JRR)%NBUDGET=JRR
ENDDO
LBU_ENABLE=.FALSE.
LBUDGET_U=.FALSE.
LBUDGET_V=.FALSE.
LBUDGET_W=.FALSE.
LBUDGET_TH=.FALSE.
LBUDGET_TKE=.FALSE.
LBUDGET_RV=.FALSE.
LBUDGET_RC=.FALSE.
LBUDGET_RR=.FALSE.
LBUDGET_RI=.FALSE.
LBUDGET_RS=.FALSE.
LBUDGET_RG=.FALSE.
LBUDGET_RH=.FALSE.
LBUDGET_SV=.FALSE.
PHYEX%MISC%TBUCONF=TBUCONF
END SUBROUTINE INIT_PHYEX
END PROGRAM