New URL for NEMO forge!   http://forge.nemo-ocean.eu

Since March 2022 along with NEMO 4.2 release, the code development moved to a self-hosted GitLab.
This present forge is now archived and remained online for history.
agrif_top_update.F90 in NEMO/trunk/src/NST – NEMO

source: NEMO/trunk/src/NST/agrif_top_update.F90 @ 14086

Last change on this file since 14086 was 14086, checked in by cetlod, 3 years ago

Adding AGRIF branches into the trunk

  • Property svn:keywords set to Id
File size: 8.1 KB
Line 
1#undef DECAL_FEEDBACK
2
3MODULE agrif_top_update
4   !!======================================================================
5   !!                ***  MODULE agrif_top_update  ***
6   !! AGRIF :   update package for passive tracers (TOP)
7   !!======================================================================
8   !! History : 
9   !!----------------------------------------------------------------------
10#if defined key_agrif && defined key_top
11   !!----------------------------------------------------------------------
12   !!   'key_agrif'                                              AGRIF zoom
13   !!   'key_TOP'                                           on-line tracers
14   !!----------------------------------------------------------------------
15   USE par_oce
16   USE oce
17   USE dom_oce
18   USE agrif_oce
19   USE par_trc
20   USE trc
21   USE vremap
22
23   IMPLICIT NONE
24   PRIVATE
25
26   PUBLIC Agrif_Update_Trc
27
28   !!----------------------------------------------------------------------
29   !! NEMO/NST 4.0 , NEMO Consortium (2018)
30   !! $Id$
31   !! Software governed by the CeCILL license (see ./LICENSE)
32   !!----------------------------------------------------------------------
33CONTAINS
34
35   SUBROUTINE Agrif_Update_Trc( )
36      !!----------------------------------------------------------------------
37      !!                   *** ROUTINE Agrif_Update_Trc ***
38      !!----------------------------------------------------------------------
39      !
40      IF (Agrif_Root()) RETURN 
41      !
42      l_vremap                      = ln_vert_remap
43      Agrif_UseSpecialValueInUpdate = .NOT.l_vremap
44      Agrif_SpecialValueFineGrid    = 0._wp
45
46      !
47# if ! defined DECAL_FEEDBACK
48      CALL Agrif_Update_Variable(trn_id, procname=updateTRC )
49!      CALL Agrif_Update_Variable( trn_id, locupdate=(/0,2/), procname=updateTRC )
50# else
51      CALL Agrif_Update_Variable(trn_id, locupdate=(/1,0/),procname=updateTRC )
52!      CALL Agrif_Update_Variable( trn_id, locupdate=(/1,2/), procname=updateTRC )
53# endif
54      !
55      Agrif_UseSpecialValueInUpdate = .FALSE.
56      l_vremap                      = .FALSE.
57      !
58   END SUBROUTINE Agrif_Update_Trc
59
60   SUBROUTINE updateTRC( tabres, i1, i2, j1, j2, k1, k2, n1, n2, before )
61
62      INTEGER, INTENT(in) :: i1,i2,j1,j2,k1,k2,n1,n2
63      REAL(wp),DIMENSION(i1:i2,j1:j2,k1:k2,n1:n2), INTENT(inout) :: tabres
64      LOGICAL, INTENT(in) :: before
65      !!
66      INTEGER :: ji,jj,jk,jn
67      REAL(wp) :: ztb, ztnu, ztno
68      REAL(wp) :: h_in(k1:k2)
69      REAL(wp) :: h_out(1:jpk)
70      INTEGER  :: N_in, N_out
71      REAL(wp) :: h_diff
72      REAL(wp) :: tabin(k1:k2,1:jptra)
73      REAL(wp), DIMENSION(i1:i2,j1:j2,1:jpk,1:jptra) :: tabres_child
74
75      IF (before) THEN
76         IF ( l_vremap ) THEN
77            DO jn = n1,n2-1
78               DO jk=k1,k2
79                  DO jj=j1,j2
80                     DO ji=i1,i2
81                        tabres(ji,jj,jk,jn) = tr(ji,jj,jk,jn,Kmm_a) * e3t(ji,jj,jk,Kmm_a)
82                     END DO
83                  END DO
84               END DO
85            END DO
86            DO jk=k1,k2
87               DO jj=j1,j2
88                  DO ji=i1,i2
89                     tabres(ji,jj,jk,n2) = tmask(ji,jj,jk) * e3t(ji,jj,jk,Kmm_a)
90                  END DO
91               END DO
92            END DO
93         ELSE
94            DO jn = 1,jptra
95               DO jk=k1,k2
96                  DO jj=j1,j2
97                     DO ji=i1,i2
98                        tabres(ji,jj,jk,jn) = tr(ji,jj,jk,jn,Kmm_a)  * e3t(ji,jj,jk,Kmm_a) / e3t_0(ji,jj,jk)
99                     END DO
100                  END DO
101               END DO
102            END DO
103
104         ENDIF
105      ELSE
106         IF ( l_vremap ) THEN
107            tabres_child(:,:,:,:) = 0._wp
108            AGRIF_SpecialValue = 0._wp
109            DO jj=j1,j2
110               DO ji=i1,i2
111                  N_in = 0
112                  DO jk=k1,k2 !k2 = jpk of child grid
113                     IF (tabres(ji,jj,jk,n2) <= 1.e-6_wp  ) EXIT
114                     N_in = N_in + 1
115                     tabin(jk,:) = tabres(ji,jj,jk,n1:n2-1)/tabres(ji,jj,jk,n2)
116                     h_in(N_in) = tabres(ji,jj,jk,n2)
117                  ENDDO
118                  N_out = 0
119                  DO jk=1,jpk ! jpk of parent grid
120                     IF (tmask(ji,jj,jk) == 0 ) EXIT ! TODO: Will not work with ISF
121                     N_out = N_out + 1
122                     h_out(N_out) = e3t(ji,jj,jk,Kmm_a)
123                  ENDDO
124                  IF (N_in*N_out > 0) THEN !Remove this?
125                     CALL reconstructandremap(tabin(1:N_in,1:jptra),h_in(1:N_in),tabres_child(ji,jj,1:N_out,1:jptra),h_out(1:N_out),N_in,N_out,jptra)
126                  ENDIF
127               ENDDO
128            ENDDO
129
130            IF (.NOT.(lk_agrif_fstep.AND.(l_1st_euler))) THEN
131               ! Add asselin part
132               DO jn = 1,jptra
133                  DO jk = 1, jpkm1
134                     DO jj = j1, j2
135                        DO ji = i1, i2
136                           IF( tabres_child(ji,jj,jk,jn) /= 0._wp ) THEN
137                              ztb  = tr(ji,jj,jk,jn,Kbb_a) * e3t(ji,jj,jk,Kbb_a) ! fse3t_b prior update should be used
138                              ztnu = tabres_child(ji,jj,jk,jn) * e3t(ji,jj,jk,Kmm_a)
139                              ztno = tr(ji,jj,jk,jn,Kmm_a) * e3t(ji,jj,jk,Krhs_a)
140                              tr(ji,jj,jk,jn,Kbb_a) = ( ztb + rn_atfp * ( ztnu - ztno) )  &
141                                        &        * tmask(ji,jj,jk) / e3t(ji,jj,jk,Kbb_a)
142                           ENDIF
143                        END DO
144                     END DO
145                  END DO
146               END DO
147            ENDIF
148            DO jn = 1,jptra
149               DO jk = 1, jpkm1
150                  DO jj = j1, j2
151                     DO ji = i1, i2
152                        IF( tabres_child(ji,jj,jk,jn) /= 0._wp ) THEN
153                           tr(ji,jj,jk,jn,Kmm_a) = tabres_child(ji,jj,jk,jn)
154                        END IF
155                     END DO
156                  END DO
157               END DO
158            END DO
159         ELSE
160            DO jn = 1,jptra
161               tabres(i1:i2,j1:j2,k1:k2,jn) =  tabres(i1:i2,j1:j2,k1:k2,jn) * e3t_0(i1:i2,j1:j2,k1:k2) &
162                                            & * tmask(i1:i2,j1:j2,k1:k2)
163            ENDDO
164            IF (.NOT.(lk_agrif_fstep.AND.(l_1st_euler))) THEN
165               ! Add asselin part
166               DO jn = 1,jptra
167                  DO jk = k1, k2
168                     DO jj = j1, j2
169                        DO ji = i1, i2
170                           IF( tabres(ji,jj,jk,jn) /= 0._wp ) THEN
171                              ztb  = tr(ji,jj,jk,jn,Kbb_a) * e3t(ji,jj,jk,Kbb_a) ! fse3t_b prior update should be used
172                              ztnu = tabres(ji,jj,jk,jn)
173                              ztno = tr(ji,jj,jk,jn,Kmm_a) * e3t(ji,jj,jk,Krhs_a)
174                              tr(ji,jj,jk,jn,Kbb_a) = ( ztb + rn_atfp * ( ztnu - ztno) )  &
175                                        &        * tmask(ji,jj,jk) / e3t(ji,jj,jk,Kbb_a)
176                           ENDIF
177                        END DO
178                     END DO
179                  END DO
180               END DO
181            ENDIF
182            DO jn = 1,jptra
183               DO jk=k1,k2
184                  DO jj=j1,j2
185                     DO ji=i1,i2
186                        IF( tabres(ji,jj,jk,jn) /= 0._wp ) THEN
187                           tr(ji,jj,jk,jn,Kmm_a) = tabres(ji,jj,jk,jn) / e3t(ji,jj,jk,Kmm_a)
188                        END IF
189                     END DO
190                  END DO
191               END DO
192            END DO
193            !
194         ENDIF
195         IF  ((l_1st_euler).AND.(Agrif_Nb_Step()==0) ) THEN
196            tr(i1:i2,j1:j2,1:jpkm1,1:jptra,Kbb_a)  = tr(i1:i2,j1:j2,1:jpkm1,1:jptra,Kmm_a)
197         ENDIF
198      ENDIF
199      !
200   END SUBROUTINE updateTRC
201
202#else
203   !!----------------------------------------------------------------------
204   !!   Empty module                                           no TOP AGRIF
205   !!----------------------------------------------------------------------
206CONTAINS
207   SUBROUTINE agrif_top_update_empty
208      WRITE(*,*)  'agrif_top_update : You should not have seen this print! error?'
209   END SUBROUTINE agrif_top_update_empty
210#endif
211
212   !!======================================================================
213END MODULE agrif_top_update
Note: See TracBrowser for help on using the repository browser.