/[lmdze]/trunk/dyn3d/leapfrog.f
ViewVC logotype

Diff of /trunk/dyn3d/leapfrog.f

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

trunk/libf/dyn3d/leapfrog.f90 revision 28 by guez, Fri Mar 26 18:33:04 2010 UTC trunk/dyn3d/leapfrog.f90 revision 76 by guez, Fri Nov 15 18:45:49 2013 UTC
# Line 6  contains Line 6  contains
6    
7    SUBROUTINE leapfrog(ucov, vcov, teta, ps, masse, phis, q, time_0)    SUBROUTINE leapfrog(ucov, vcov, teta, ps, masse, phis, q, time_0)
8    
9      ! From dyn3d/leapfrog.F, version 1.6, 2005/04/13 08:58:34      ! From dyn3d/leapfrog.F, version 1.6, 2005/04/13 08:58:34 revision 616
10      ! Authors: P. Le Van, L. Fairhead, F. Hourdin      ! Authors: P. Le Van, L. Fairhead, F. Hourdin
11        ! Matsuno-leapfrog scheme.
12    
13        use addfi_m, only: addfi
14        use bilan_dyn_m, only: bilan_dyn
15        use caladvtrac_m, only: caladvtrac
16        use caldyn_m, only: caldyn
17      USE calfis_m, ONLY: calfis      USE calfis_m, ONLY: calfis
     USE com_io_dyn, ONLY: histaveid  
18      USE comconst, ONLY: daysec, dtphys, dtvr      USE comconst, ONLY: daysec, dtphys, dtvr
19      USE comgeom, ONLY: aire, apoln, apols      USE comgeom, ONLY: aire_2d, apoln, apols
20      USE comvert, ONLY: ap, bp      USE disvert_m, ONLY: ap, bp
21      USE conf_gcm_m, ONLY: day_step, iconser, iperiod, iphysiq, nday, offline, &      USE conf_gcm_m, ONLY: day_step, iconser, iperiod, iphysiq, nday, offline, &
22           periodav           iflag_phys, ok_guide, iecri
23      USE dimens_m, ONLY: iim, llm, nqmx      USE dimens_m, ONLY: iim, jjm, llm, nqmx
24        use dissip_m, only: dissip
25      USE dynetat0_m, ONLY: day_ini      USE dynetat0_m, ONLY: day_ini
26      use dynredem1_m, only: dynredem1      use dynredem1_m, only: dynredem1
27      USE exner_hyb_m, ONLY: exner_hyb      USE exner_hyb_m, ONLY: exner_hyb
28      use filtreg_m, only: filtreg      use filtreg_m, only: filtreg
29        use fluxstokenc_m, only: fluxstokenc
30        use geopot_m, only: geopot
31      USE guide_m, ONLY: guide      USE guide_m, ONLY: guide
32      use inidissip_m, only: idissip      use inidissip_m, only: idissip
33      USE logic, ONLY: iflag_phys, ok_guide      use integrd_m, only: integrd
34      USE paramet_m, ONLY: iip1, ip1jm, ip1jmp1, jjp1      use nr_util, only: assert
     USE pression_m, ONLY: pression  
35      USE pressure_var, ONLY: p3d      USE pressure_var, ONLY: p3d
36      USE temps, ONLY: itau_dyn      USE temps, ONLY: itau_dyn
37        use writedynav_m, only: writedynav
38        use writehist_m, only: writehist
39    
40      ! Variables dynamiques:      ! Variables dynamiques:
41      REAL vcov(ip1jm, llm), ucov(ip1jmp1, llm) ! vents covariants      REAL, intent(inout):: ucov(:, :, :) ! (iim + 1, jjm + 1, llm) vent covariant
42      REAL teta(ip1jmp1, llm) ! temperature potentielle      REAL, intent(inout):: vcov(:, :, :) ! (iim + 1, jjm, llm) ! vent covariant
43      REAL ps(ip1jmp1) ! pression au sol, en Pa  
44        REAL, intent(inout):: teta(:, :, :) ! (iim + 1, jjm + 1, llm)
45      REAL masse(ip1jmp1, llm) ! masse d'air      ! potential temperature
46      REAL phis(ip1jmp1) ! geopotentiel au sol  
47      REAL q(ip1jmp1, llm, nqmx) ! mass fractions of advected fields      REAL, intent(inout):: ps(:, :) ! (iim + 1, jjm + 1) pression au sol, en Pa
48        REAL, intent(inout):: masse(:, :, :) ! (iim + 1, jjm + 1, llm) masse d'air
49        REAL, intent(in):: phis(:, :) ! (iim + 1, jjm + 1) surface geopotential
50    
51        REAL, intent(inout):: q(:, :, :, :) ! (iim + 1, jjm + 1, llm, nqmx)
52        ! mass fractions of advected fields
53    
54      REAL, intent(in):: time_0      REAL, intent(in):: time_0
55    
56      ! Variables local to the procedure:      ! Variables local to the procedure:
57    
58      ! Variables dynamiques:      ! Variables dynamiques:
59    
60      REAL pks(ip1jmp1) ! exner au sol      REAL pks(iim + 1, jjm + 1) ! exner au sol
61      REAL pk(ip1jmp1, llm) ! exner au milieu des couches      REAL pk(iim + 1, jjm + 1, llm) ! exner au milieu des couches
62      REAL pkf(ip1jmp1, llm) ! exner filt.au milieu des couches      REAL pkf(iim + 1, jjm + 1, llm) ! exner filtré au milieu des couches
63      REAL phi(ip1jmp1, llm) ! geopotential      REAL phi(iim + 1, jjm + 1, llm) ! geopotential
64      REAL w(ip1jmp1, llm) ! vitesse verticale      REAL w((iim + 1) * (jjm + 1), llm) ! vitesse verticale
65    
66      ! variables dynamiques intermediaire pour le transport      ! Variables dynamiques intermediaire pour le transport
67      REAL pbaru(ip1jmp1, llm), pbarv(ip1jm, llm) !flux de masse      ! Flux de masse :
68        REAL pbaru((iim + 1) * (jjm + 1), llm), pbarv((iim + 1) * jjm, llm)
69      ! variables dynamiques au pas - 1  
70      REAL vcovm1(ip1jm, llm), ucovm1(ip1jmp1, llm)      ! Variables dynamiques au pas - 1
71      REAL tetam1(ip1jmp1, llm), psm1(ip1jmp1)      REAL vcovm1(iim + 1, jjm, llm), ucovm1(iim + 1, jjm + 1, llm)
72      REAL massem1(ip1jmp1, llm)      REAL tetam1(iim + 1, jjm + 1, llm), psm1(iim + 1, jjm + 1)
73        REAL massem1(iim + 1, jjm + 1, llm)
74      ! tendances dynamiques  
75      REAL dv(ip1jm, llm), du(ip1jmp1, llm)      ! Tendances dynamiques
76      REAL dteta(ip1jmp1, llm), dq(ip1jmp1, llm, nqmx), dp(ip1jmp1)      REAL dv((iim + 1) * jjm, llm), dudyn((iim + 1) * (jjm + 1), llm)
77        REAL dteta(iim + 1, jjm + 1, llm)
78      ! tendances de la dissipation      real dp((iim + 1) * (jjm + 1))
79      REAL dvdis(ip1jm, llm), dudis(ip1jmp1, llm)  
80      REAL dtetadis(ip1jmp1, llm)      ! Tendances de la dissipation :
81        REAL dvdis(iim + 1, jjm, llm), dudis(iim + 1, jjm + 1, llm)
82      ! tendances physiques      REAL dtetadis(iim + 1, jjm + 1, llm)
83      REAL dvfi(ip1jm, llm), dufi(ip1jmp1, llm)  
84      REAL dtetafi(ip1jmp1, llm), dqfi(ip1jmp1, llm, nqmx), dpfi(ip1jmp1)      ! Tendances physiques
85        REAL dvfi((iim + 1) * jjm, llm), dufi((iim + 1) * (jjm + 1), llm)
86      ! variables pour le fichier histoire      REAL dtetafi(iim + 1, jjm + 1, llm), dqfi((iim + 1) * (jjm + 1), llm, nqmx)
87        real dpfi((iim + 1) * (jjm + 1))
88    
89      REAL tppn(iim), tpps(iim), tpn, tps      ! Variables pour le fichier histoire
90    
91      INTEGER itau ! index of the time step of the dynamics, starts at 0      INTEGER itau ! index of the time step of the dynamics, starts at 0
92      INTEGER itaufin      INTEGER itaufin
     INTEGER iday ! jour julien  
93      REAL time ! time of day, as a fraction of day length      REAL time ! time of day, as a fraction of day length
94      real finvmaold(ip1jmp1, llm)      real finvmaold(iim + 1, jjm + 1, llm)
95      LOGICAL:: lafin=.false.      INTEGER l
     INTEGER ij, l  
   
96      REAL rdayvrai, rdaym_ini      REAL rdayvrai, rdaym_ini
97    
98      ! Variables test conservation energie      ! Variables test conservation énergie
99      REAL ecin(ip1jmp1, llm), ecin0(ip1jmp1, llm)      REAL ecin(iim + 1, jjm + 1, llm), ecin0(iim + 1, jjm + 1, llm)
100      ! Tendance de la temp. potentiel d (theta) / d t due a la  
101      ! tansformation d'energie cinetique en energie thermique      REAL vcont((iim + 1) * jjm, llm), ucont((iim + 1) * (jjm + 1), llm)
102      ! cree par la dissipation      logical leapf
103      REAL dtetaecdt(ip1jmp1, llm)      real dt
     REAL vcont(ip1jm, llm), ucont(ip1jmp1, llm)  
     logical forward, leapf  
     REAL dt  
104    
105      !---------------------------------------------------      !---------------------------------------------------
106    
107      print *, "Call sequence information: leapfrog"      print *, "Call sequence information: leapfrog"
108        call assert(shape(ucov) == (/iim + 1, jjm + 1, llm/), "leapfrog")
109    
110      itaufin = nday * day_step      itaufin = nday * day_step
111      itau = 0      ! "day_step" is a multiple of "iperiod", therefore so is "itaufin".
112      iday = day_ini  
     time = time_0  
     dq = 0.  
113      ! On initialise la pression et la fonction d'Exner :      ! On initialise la pression et la fonction d'Exner :
114      CALL pression(ip1jmp1, ap, bp, ps, p3d)      forall (l = 1: llm + 1) p3d(:, :, l) = ap(l) + bp(l) * ps
115      CALL exner_hyb(ps, p3d, pks, pk, pkf)      CALL exner_hyb(ps, p3d, pks, pk, pkf)
116    
117      ! Début de l'integration temporelle :      time_integration: do itau = 0, itaufin - 1
118      outer_loop:do         leapf = mod(itau, iperiod) /= 0
119         if (ok_guide .and. (itaufin - itau - 1) * dtvr > 21600.) &         if (leapf) then
120              call guide(itau, ucov, vcov, teta, q, masse, ps)            dt = 2 * dtvr
121         vcovm1 = vcov         else
122         ucovm1 = ucov            ! Matsuno
123         tetam1 = teta            dt = dtvr
124         massem1 = masse            if (ok_guide .and. (itaufin - itau - 1) * dtvr > 21600.) &
125         psm1 = ps                 call guide(itau, ucov, vcov, teta, q, masse, ps)
126         forward = .TRUE.            vcovm1 = vcov
127         leapf = .FALSE.            ucovm1 = ucov
128         dt = dtvr            tetam1 = teta
129         finvmaold = masse            massem1 = masse
130         CALL filtreg(finvmaold, jjp1, llm, - 2, 2, .TRUE., 1)            psm1 = ps
131              finvmaold = masse
132              CALL filtreg(finvmaold, jjm + 1, llm, - 2, 2, .TRUE.)
133           end if
134    
135           ! Calcul des tendances dynamiques:
136           CALL geopot(teta, pk, pks, phis, phi)
137           CALL caldyn(itau, ucov, vcov, teta, ps, masse, pk, pkf, phis, phi, &
138                dudyn, dv, dteta, dp, w, pbaru, pbarv, time_0, &
139                conser=MOD(itau, iconser)==0)
140    
141           CALL caladvtrac(q, pbaru, pbarv, p3d, masse, teta, pk)
142    
143           ! Stokage du flux de masse pour traceurs offline:
144           IF (offline) CALL fluxstokenc(pbaru, pbarv, masse, teta, phi, phis, &
145                dtvr, itau)
146    
147           ! Intégrations dynamique et traceurs:
148           CALL integrd(vcovm1, ucovm1, tetam1, psm1, massem1, dv, dudyn, dteta, &
149                dp, vcov, ucov, teta, q(:, :, :, :2), ps, masse, finvmaold, dt, &
150                leapf)
151    
152           if (.not. leapf) then
153              ! Matsuno backward
154              forall (l = 1: llm + 1) p3d(:, :, l) = ap(l) + bp(l) * ps
155              CALL exner_hyb(ps, p3d, pks, pk, pkf)
156    
        do  
157            ! Calcul des tendances dynamiques:            ! Calcul des tendances dynamiques:
158            CALL geopot(ip1jmp1, teta, pk, pks, phis, phi)            CALL geopot(teta, pk, pks, phis, phi)
159            CALL caldyn(itau, ucov, vcov, teta, ps, masse, pk, pkf, phis, phi, &            CALL caldyn(itau + 1, ucov, vcov, teta, ps, masse, pk, pkf, phis, &
160                 MOD(itau, iconser) == 0, du, dv, dteta, dp, w, pbaru, pbarv, &                 phi, dudyn, dv, dteta, dp, w, pbaru, pbarv, time_0, &
161                 time + iday - day_ini)                 conser=.false.)
   
           IF (forward .OR. leapf) THEN  
              ! Calcul des tendances advection des traceurs (dont l'humidité)  
              CALL caladvtrac(q, pbaru, pbarv, p3d, masse, dq, teta, pk)  
              IF (offline) THEN  
                 ! Stokage du flux de masse pour traceurs off-line  
                 CALL fluxstokenc(pbaru, pbarv, masse, teta, phi, phis, dtvr, &  
                      itau)  
              ENDIF  
           ENDIF  
162    
163            ! integrations dynamique et traceurs:            ! integrations dynamique et traceurs:
164            CALL integrd(2, vcovm1, ucovm1, tetam1, psm1, massem1, dv, du, &            CALL integrd(vcovm1, ucovm1, tetam1, psm1, massem1, dv, dudyn, &
165                 dteta, dq, dp, vcov, ucov, teta, q, ps, masse, phis, &                 dteta, dp, vcov, ucov, teta, q(:, :, :, :2), ps, masse, &
166                 finvmaold, leapf, dt)                 finvmaold, dtvr, leapf=.false.)
167           end if
           IF (MOD(itau + 1, iphysiq) == 0 .AND. iflag_phys /= 0) THEN  
              ! calcul des tendances physiques:  
              IF (itau + 1 == itaufin) lafin = .TRUE.  
   
              CALL pression(ip1jmp1, ap, bp, ps, p3d)  
              CALL exner_hyb(ps, p3d, pks, pk, pkf)  
   
              rdaym_ini = itau * dtvr / daysec  
              rdayvrai = rdaym_ini + day_ini  
   
              CALL calfis(nqmx, lafin, rdayvrai, time, ucov, vcov, teta, q, &  
                   masse, ps, pk, phis, phi, du, dv, dteta, dq, w, &  
                   dufi, dvfi, dtetafi, dqfi, dpfi)  
   
              ! ajout des tendances physiques:  
              CALL addfi(nqmx, dtphys, &  
                   ucov, vcov, teta, q, ps, &  
                   dufi, dvfi, dtetafi, dqfi, dpfi)  
           ENDIF  
168    
169            CALL pression(ip1jmp1, ap, bp, ps, p3d)         IF (MOD(itau + 1, iphysiq) == 0 .AND. iflag_phys /= 0) THEN
170            CALL exner_hyb(ps, p3d, pks, pk, pkf)            ! Calcul des tendances physiques:
171    
172            IF (MOD(itau + 1, idissip) == 0) THEN            forall (l = 1: llm + 1) p3d(:, :, l) = ap(l) + bp(l) * ps
173               ! dissipation horizontale et verticale des petites echelles:            CALL exner_hyb(ps, p3d, pks, pk, pkf)
174    
175               ! calcul de l'energie cinetique avant dissipation            rdaym_ini = itau * dtvr / daysec
176               call covcont(llm, ucov, vcov, ucont, vcont)            rdayvrai = rdaym_ini + day_ini
177               call enercin(vcov, ucov, vcont, ucont, ecin0)            time = REAL(mod(itau, day_step)) / day_step + time_0
178              IF (time > 1.) time = time - 1.
179               ! dissipation  
180               CALL dissip(vcov, ucov, teta, p3d, dvdis, dudis, dtetadis)            CALL calfis(rdayvrai, time, ucov, vcov, teta, q, ps, pk, phis, phi, &
181               ucov=ucov + dudis                 dudyn, dv, w, dufi, dvfi, dtetafi, dqfi, dpfi, &
182               vcov=vcov + dvdis                 lafin = itau + 1 == itaufin)
183    
184               ! On rajoute la tendance due à la transformation Ec -> E            ! Ajout des tendances physiques:
185               ! thermique créée lors de la dissipation            CALL addfi(ucov, vcov, teta, q, ps, dufi, dvfi, dtetafi, dqfi, dpfi)
186               call covcont(llm, ucov, vcov, ucont, vcont)         ENDIF
187               call enercin(vcov, ucov, vcont, ucont, ecin)  
188               dtetaecdt= (ecin0 - ecin) / pk         forall (l = 1: llm + 1) p3d(:, :, l) = ap(l) + bp(l) * ps
189               dtetadis=dtetadis + dtetaecdt         CALL exner_hyb(ps, p3d, pks, pk, pkf)
190               teta=teta + dtetadis  
191           IF (MOD(itau + 1, idissip) == 0) THEN
192               ! Calcul de la valeur moyenne unique de h aux pôles            ! Dissipation horizontale et verticale des petites échelles
193               DO l = 1, llm  
194                  DO ij = 1, iim            ! calcul de l'énergie cinétique avant dissipation
195                     tppn(ij) = aire(ij) * teta(ij, l)            call covcont(llm, ucov, vcov, ucont, vcont)
196                     tpps(ij) = aire(ij + ip1jm) * teta(ij + ip1jm, l)            call enercin(vcov, ucov, vcont, ucont, ecin0)
197                  ENDDO  
198                  tpn = SUM(tppn) / apoln            ! dissipation
199                  tps = SUM(tpps) / apols            CALL dissip(vcov, ucov, teta, p3d, dvdis, dudis, dtetadis)
200              ucov = ucov + dudis
201                  DO ij = 1, iip1            vcov = vcov + dvdis
202                     teta(ij, l) = tpn  
203                     teta(ij + ip1jm, l) = tps            ! On ajoute la tendance due à la transformation énergie
204                  ENDDO            ! cinétique en énergie thermique par la dissipation
205               ENDDO            call covcont(llm, ucov, vcov, ucont, vcont)
206              call enercin(vcov, ucov, vcont, ucont, ecin)
207               DO ij = 1, iim            dtetadis = dtetadis + (ecin0 - ecin) / pk
208                  tppn(ij) = aire(ij) * ps(ij)            teta = teta + dtetadis
209                  tpps(ij) = aire(ij + ip1jm) * ps(ij + ip1jm)  
210               ENDDO            ! Calcul de la valeur moyenne aux pôles :
211               tpn = SUM(tppn) / apoln            forall (l = 1: llm)
212               tps = SUM(tpps) / apols               teta(:, 1, l) = SUM(aire_2d(:iim, 1) * teta(:iim, 1, l)) &
213                      / apoln
214               DO ij = 1, iip1               teta(:, jjm + 1, l) = SUM(aire_2d(:iim, jjm+1) &
215                  ps(ij) = tpn                    * teta(:iim, jjm + 1, l)) / apols
216                  ps(ij + ip1jm) = tps            END forall
217               ENDDO  
218            END IF            ps(:, 1) = SUM(aire_2d(:iim, 1) * ps(:iim, 1)) / apoln
219              ps(:, jjm + 1) = SUM(aire_2d(:iim, jjm+1) * ps(:iim, jjm + 1)) &
220            ! fin de l'intégration dynamique et physique pour le pas "itau"                 / apols
221            ! préparation du pas d'intégration suivant         END IF
222    
223            ! schema matsuno + leapfrog         IF (MOD(itau + 1, iperiod) == 0) THEN
224            IF (forward .OR. leapf) THEN            ! Écriture du fichier histoire moyenne:
225               itau = itau + 1            CALL writedynav(vcov, ucov, teta, pk, phi, q, masse, ps, phis, &
226               iday = day_ini + itau / day_step                 time = itau + 1)
227               time = REAL(itau - (iday - day_ini) * day_step) / day_step &            call bilan_dyn(ps, masse, pk, pbaru, pbarv, teta, phi, ucov, vcov, &
228                    + time_0                 q(:, :, :, 1))
229               IF (time > 1.) THEN         ENDIF
230                  time = time - 1.  
231                  iday = iday + 1         IF (MOD(itau + 1, iecri * day_step) == 0) THEN
232               ENDIF            CALL geopot(teta, pk, pks, phis, phi)
233            ENDIF            CALL writehist(itau, vcov, ucov, teta, phi, q, masse, ps)
234           END IF
235            IF (itau == itaufin + 1) exit outer_loop      end do time_integration
236    
237            IF (MOD(itau, iperiod) == 0 .OR. itau == itaufin) THEN      CALL dynredem1("restart.nc", vcov, ucov, teta, q, masse, ps, &
238               ! ecriture du fichier histoire moyenne:           itau = itau_dyn + itaufin)
239               CALL writedynav(histaveid, nqmx, itau, vcov, &  
240                    ucov, teta, pk, phi, q, masse, ps, phis)      ! Calcul des tendances dynamiques:
241               call bilan_dyn(2, dtvr * iperiod, dtvr * day_step * periodav, &      CALL geopot(teta, pk, pks, phis, phi)
242                    ps, masse, pk, pbaru, pbarv, teta, phi, ucov, vcov, q)      CALL caldyn(itaufin, ucov, vcov, teta, ps, masse, pk, pkf, phis, phi, &
243            ENDIF           dudyn, dv, dteta, dp, w, pbaru, pbarv, time_0, &
244             conser = MOD(itaufin, iconser) == 0)
           IF (itau == itaufin) THEN  
              CALL dynredem1("restart.nc", vcov, ucov, teta, q, masse, ps, &  
                   itau=itau_dyn+itaufin)  
           ENDIF  
   
           ! gestion de l'integration temporelle:  
           IF (MOD(itau, iperiod) == 0) exit  
           IF (MOD(itau - 1, iperiod) == 0) THEN  
              IF (forward) THEN  
                 ! fin du pas forward et debut du pas backward  
                 forward = .FALSE.  
                 leapf = .FALSE.  
              ELSE  
                 ! fin du pas backward et debut du premier pas leapfrog  
                 leapf = .TRUE.  
                 dt = 2. * dtvr  
              END IF  
           ELSE  
              ! pas leapfrog  
              leapf = .TRUE.  
              dt = 2. * dtvr  
           END IF  
        end do  
     end do outer_loop  
245    
246    END SUBROUTINE leapfrog    END SUBROUTINE leapfrog
247    
Legend:
Removed from v.28  
changed lines
  Added in v.76
  ViewVC Help
Powered by ViewVC 1.1.21