/[lmdze]/trunk/libf/dyn3d/leapfrog.f90
ViewVC logotype

Diff of /trunk/libf/dyn3d/leapfrog.f90

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

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

Legend:
Removed from v.25  
changed lines
  Added in v.66

  ViewVC Help
Powered by ViewVC 1.1.21