-
Notifications
You must be signed in to change notification settings - Fork 237
/
thsice_main.F
289 lines (262 loc) · 10 KB
/
thsice_main.F
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
#include "THSICE_OPTIONS.h"
#ifdef ALLOW_AUTODIFF
# include "AUTODIFF_OPTIONS.h"
# ifdef ALLOW_EXF
# include "EXF_OPTIONS.h"
# endif
#endif
CBOP
C !ROUTINE: THSICE_MAIN
C !INTERFACE:
SUBROUTINE THSICE_MAIN(
I myTime, myIter, myThid )
C !DESCRIPTION: \bv
C *==========================================================*
C | S/R THSICE_MAIN
C | o Therm_SeaIce main routine.
C | step forward Thermodynamic_SeaIce variables and modify
C | ocean surface forcing accordingly.
C *==========================================================*
C !USES:
IMPLICIT NONE
C === Global variables ===
#include "SIZE.h"
#include "EEPARAMS.h"
#include "PARAMS.h"
#include "FFIELDS.h"
#include "THSICE_PARAMS.h"
#include "THSICE_SIZE.h"
#include "THSICE_VARS.h"
#ifdef ALLOW_AUTODIFF
# include "THSICE_COST.h"
# include "DYNVARS.h"
# ifdef ALLOW_EXF
# include "EXF_PARAM.h"
# include "EXF_CONSTANTS.h"
# include "EXF_FIELDS.h"
# endif /* ALLOW_EXF */
#endif /* ALLOW_AUTODIFF */
#ifdef ALLOW_AUTODIFF_TAMC
# include "tamc.h"
#endif
C !INPUT/OUTPUT PARAMETERS:
C === Routine arguments ===
C myTime :: Current time in simulation (s)
C myIter :: Current iteration number
C myThid :: My Thread Id. number
_RL myTime
INTEGER myIter
INTEGER myThid
CEOP
#ifdef ALLOW_THSICE
C !LOCAL VARIABLES:
C === Local variables ===
C prcAtm :: total precip from the atmosphere [kg/m2/s]
C snowPr :: snow precipitation [kg/m2/s]
C qPrcRn :: Energy content of Precip+RunOff (+=down) [W/m2]
INTEGER i,j
INTEGER bi,bj
INTEGER iMin, iMax
INTEGER jMin, jMax
_RL prcAtm(1-OLx:sNx+OLx,1-OLy:sNy+OLy,nSx,nSy)
_RL snowPr(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
_RL qPrcRn(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
c _RL evpAtm(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
c _RL flxAtm(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
c _RL flxSW (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
_RL tauFac
#ifdef ALLOW_EXF
INTEGER grpDiag
#endif
#ifdef ALLOW_AUTODIFF_TAMC
C tkey :: tape key (depends on tiles)
INTEGER tkey
#endif
C---+----1----+----2----+----3----+----4----+----5----+----6----+----7-|--+----|
#ifdef ALLOW_EXF
IF ( useEXF .AND. useDiagnostics ) THEN
C- Fill-in EXF surface flux diags, weighted by open-ocean fraction
grpDiag = 2
IF ( thSIce_skipThermo ) grpDiag = -2
CALL EXF_WEIGHT_SFX_DIAGS(
I iceMask, grpDiag, myTime, myIter, myThid )
IF ( .NOT.useSEAICE ) CALL EXF_WEIGHT_SFX_DIAGS(
I iceMask, -1, myTime, myIter, myThid )
ENDIF
#endif /* ALLOW_EXF */
C- Only compute/update seaice fields over the interior
C (excluding overlap) and apply exchanges when needed
iMin = 1
iMax = sNx
jMin = 1
jMax = sNy
DO bj=myByLo(myThid),myByHi(myThid)
DO bi=myBxLo(myThid),myBxHi(myThid)
#ifdef ALLOW_AUTODIFF_TAMC
tkey = bi + (bj-1)*nSx + (ikey_dynamics-1)*nSx*nSy
CADJ STORE uvel (:,:,1,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
CADJ STORE vvel (:,:,1,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
CADJ STORE qsw (:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
CADJ STORE Qice1 (:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
CADJ STORE Qice2 (:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
CADJ STORE sHeating(:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
CADJ STORE Tsrf (:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
#endif
DO j=1-OLy,sNy+OLy
DO i=1-OLx,sNx+OLx
prcAtm (i,j,bi,bj) = 0. _d 0
snowPr (i,j) = 0. _d 0
qPrcRn (i,j) = 0. _d 0
ENDDO
ENDDO
#ifndef ALLOW_AUTODIFF
IF ( .NOT.useCheapAML ) THEN
#endif
CALL THSICE_GET_OCEAN(
I bi, bj, myTime, myIter, myThid )
#ifdef ALLOW_AUTODIFF_TAMC
CADJ STORE iceMask (:,:,bi,bj) = comlev1_bibj,key=tkey,kind=isbyte
CADJ STORE iceHeight (:,:,bi,bj) = comlev1_bibj,key=tkey,kind=isbyte
CADJ STORE snowHeight(:,:,bi,bj) = comlev1_bibj,key=tkey,kind=isbyte
CADJ STORE snowAge (:,:,bi,bj) = comlev1_bibj,key=tkey,kind=isbyte
CADJ STORE hOceMxL (:,:,bi,bj) = comlev1_bibj,key=tkey,kind=isbyte
CADJ STORE tOceMxL (:,:,bi,bj) = comlev1_bibj,key=tkey,kind=isbyte
CADJ STORE sOceMxL (:,:,bi,bj) = comlev1_bibj,key=tkey,kind=isbyte
CADJ STORE v2ocMxL (:,:,bi,bj) = comlev1_bibj,key=tkey,kind=isbyte
#endif /* ALLOW_AUTODIFF_TAMC */
#ifndef ALLOW_AUTODIFF
C- end if not useCheapAML
ENDIF
#endif
#ifndef ALLOW_AUTODIFF
IF ( useBulkforce .OR. useCheapAML ) THEN
CALL THSICE_GET_PRECIP(
I iceMask, tOceMxL,
O prcAtm(1-OLx,1-OLy,bi,bj),
O snowPr, qPrcRn,
O icFlxSW(1-OLx,1-OLy,bi,bj),
I iMin,iMax,jMin,jMax, bi,bj, myThid )
ENDIF
#endif
IF ( useEXF ) THEN
CALL THSICE_MAP_EXF(
I iceMask, tOceMxL,
O prcAtm(1-OLx,1-OLy,bi,bj),
O snowPr, qPrcRn,
O icFlxSW(1-OLx,1-OLy,bi,bj),
I iMin,iMax,jMin,jMax, bi,bj, myThid )
ENDIF
#ifndef ALLOW_AUTODIFF
IF ( .NOT.( useCheapAML .OR. thSIce_skipThermo ) ) THEN
#endif
CALL THSICE_STEP_TEMP(
I bi, bj, iMin, iMax, jMin, jMax,
I myTime, myIter, myThid )
#ifdef ALLOW_AUTODIFF_TAMC
C These store directives avoid recomputation of this routine in ad-code.
CADJ STORE flxCndBt(:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
CADJ STORE EmPmR (:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
CADJ STORE Qnet (:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
CADJ STORE snowAge (:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
CADJ STORE sHeating(:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
CADJ STORE snowPr = comlev1_bibj, key=tkey, kind=isbyte
CADJ STORE Qice1 (:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
CADJ STORE Qice2 (:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
CADJ STORE icFrwAtm(:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
C This part of the state is generally not necessary, but you never
C know so I leave it, but commented out.
cCADJ STORE Qsw (:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
cCADJ STORE icFlxSW (:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
cCADJ STORE Tice1 (:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
cCADJ STORE Tice2 (:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
cCADJ STORE sHeating(:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
cCADJ STORE Tsrf (:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
cCADJ STORE icFlxAtm(:,:,bi,bj) = comlev1_bibj, key=tkey, kind=isbyte
#endif
#ifndef ALLOW_AUTODIFF
C- end if not skipThermo / useCheapAML
ENDIF
IF ( .NOT.thSIce_skipThermo ) THEN
#endif
CALL THSICE_STEP_FWD(
I bi, bj, iMin, iMax, jMin, jMax,
I prcAtm(1-OLx,1-OLy,bi,bj),
I snowPr, qPrcRn,
I myTime, myIter, myThid )
#ifndef ALLOW_AUTODIFF
ELSE
C- Compute sIceLoad (if not previously done)
DO j=1,sNy
DO i=1,sNx
sIceLoad(i,j,bi,bj) = ( snowHeight(i,j,bi,bj)*rhos
& + iceHeight(i,j,bi,bj)*rhoi
& )*iceMask(i,j,bi,bj)
ENDDO
ENDDO
ENDIF
#endif
C-- end bi,bj loop
ENDDO
ENDDO
#ifdef ALLOW_BALANCE_FLUXES
C-- Balance net Fresh-Water flux from Atm+Land
IF ( thSIceBalanceAtmFW.NE.0 ) THEN
CALL THSICE_BALANCE_FRW(
I iMin, iMax, jMin, jMax,
I prcAtm, myTime, myIter, myThid )
ENDIF
#endif /* ALLOW_BALANCE_FLUXES */
C add a small piece of code to check AddFluid implementation:
c#include "thsice_test_addfluid.h"
C-- Exchange fields that are advected by seaice dynamics
IF ( useSEAICE .OR. thSIceAdvScheme.GT.0
& .OR. stressReduction.GT.zeroRL ) THEN
CALL THSICE_DO_EXCH( myThid )
ENDIF
IF ( thSIceAdvScheme.GT.0 .AND. .NOT.useSEAICE ) THEN
C- when useSEAICE=.true., this S/R is called from SEAICE_MODEL;
C otherwise, call it from here, after thsice-thermodynamics is done
CALL THSICE_DO_ADVECT(
I 0, 0, myTime, myIter, myThid )
ELSEIF ( thSIceAdvScheme.LE.0 .AND. useRealFreshWaterFlux ) THEN
_EXCH_XY_RS( sIceLoad, myThid )
ENDIF
DO bj=myByLo(myThid),myByHi(myThid)
DO bi=myBxLo(myThid),myBxHi(myThid)
#ifdef ALLOW_AUTODIFF_TAMC
tkey = bi + (bj-1)*nSx + (ikey_dynamics-1)*nSx*nSy
CADJ STORE iceMask(:,:,bi,bj) = comlev1_bibj,key=tkey,kind=isbyte
CADJ STORE fu (:,:,bi,bj) = comlev1_bibj,key=tkey,kind=isbyte
CADJ STORE fv (:,:,bi,bj) = comlev1_bibj,key=tkey,kind=isbyte
#endif /* ALLOW_AUTODIFF_TAMC */
C-- Cumulate time-averaged fields and also fill-up flux diagnostics
C (if not done in THSICE_DO_ADVECT call)
IF ( thSIceAdvScheme.LE.0 ) THEN
CALL THSICE_AVE(
I bi,bj, myTime, myIter, myThid )
ENDIF
C-- note: If useSEAICE=.true., the stress is computed in seaice_model,
C-- and stressReduction is always set to zero
IF ( stressReduction.GT. 0. _d 0 ) THEN
DO j = 1-OLy,sNy+OLy-1
DO i = 2-OLx,sNx+OLx-1
tauFac = stressReduction
& *(iceMask(i-1,j,bi,bj)+iceMask(i,j,bi,bj))*0.5 _d 0
fu(i,j,bi,bj) = (1. _d 0 - tauFac)*fu(i,j,bi,bj)
ENDDO
ENDDO
DO j = 2-OLy,sNy+OLy-1
DO i = 1-OLx,sNx+OLx-1
tauFac = stressReduction
& *(iceMask(i,j-1,bi,bj)+iceMask(i,j,bi,bj))*0.5 _d 0
fv(i,j,bi,bj) = (1. _d 0 - tauFac)*fv(i,j,bi,bj)
ENDDO
ENDDO
ENDIF
C-- end bi,bj loop
ENDDO
ENDDO
C---+----1----+----2----+----3----+----4----+----5----+----6----+----7-|--+----|
#endif /* ALLOW_THSICE */
RETURN
END