Sources/dyn3d/leapfrog.f

module leapfrog_m

  IMPLICIT NONE

contains

  SUBROUTINE leapfrog(ucov, vcov, teta, ps, masse, phis, q, time_0)

    ! From dyn3d/leapfrog.F, version 1.6, 2005/04/13 08:58:34
    ! Authors: P. Le Van, L. Fairhead, F. Hourdin
    ! schema matsuno + leapfrog

    USE calfis_m, ONLY: calfis
    USE com_io_dyn, ONLY: histaveid
    USE comconst, ONLY: daysec, dtphys, dtvr
    USE comgeom, ONLY: aire_2d, apoln, apols
    USE comvert, ONLY: ap, bp
    USE conf_gcm_m, ONLY: day_step, iconser, iperiod, iphysiq, nday, offline, &
         periodav
    USE dimens_m, ONLY: iim, jjm, llm, nqmx
    USE dynetat0_m, ONLY: day_ini
    use dynredem1_m, only: dynredem1
    USE exner_hyb_m, ONLY: exner_hyb
    use filtreg_m, only: filtreg
    USE guide_m, ONLY: guide
    use inidissip_m, only: idissip
    use integrd_m, only: integrd
    USE logic, ONLY: iflag_phys, ok_guide
    USE paramet_m, ONLY: ip1jmp1
    USE pression_m, ONLY: pression
    USE pressure_var, ONLY: p3d
    USE temps, ONLY: itau_dyn

    ! Variables dynamiques:
    REAL, intent(inout):: vcov((iim + 1) * jjm, llm) ! vent covariant
    REAL, intent(inout):: ucov(ip1jmp1, llm) ! vent covariant
    REAL, intent(inout):: teta(iim + 1, jjm + 1, llm) ! potential temperature
    REAL ps(iim + 1, jjm + 1) ! pression au sol, en Pa

    REAL masse(ip1jmp1, llm) ! masse d'air
    REAL phis(ip1jmp1) ! geopotentiel au sol
    REAL q(ip1jmp1, llm, nqmx) ! mass fractions of advected fields
    REAL, intent(in):: time_0

    ! Variables local to the procedure:

    ! Variables dynamiques:

    REAL pks(ip1jmp1) ! exner au sol
    REAL pk(iim + 1, jjm + 1, llm) ! exner au milieu des couches
    REAL pkf(ip1jmp1, llm) ! exner filt.au milieu des couches
    REAL phi(ip1jmp1, llm) ! geopotential
    REAL w(ip1jmp1, llm) ! vitesse verticale

    ! variables dynamiques intermediaire pour le transport
    REAL pbaru(ip1jmp1, llm), pbarv((iim + 1) * jjm, llm) !flux de masse

    ! variables dynamiques au pas - 1
    REAL vcovm1((iim + 1) * jjm, llm), ucovm1(ip1jmp1, llm)
    REAL tetam1(iim + 1, jjm + 1, llm), psm1(iim + 1, jjm + 1)
    REAL massem1(ip1jmp1, llm)

    ! tendances dynamiques
    REAL dv((iim + 1) * jjm, llm), du(ip1jmp1, llm)
    REAL dteta(ip1jmp1, llm), dq(ip1jmp1, llm, nqmx), dp(ip1jmp1)

    ! tendances de la dissipation
    REAL dvdis((iim + 1) * jjm, llm), dudis(ip1jmp1, llm)
    REAL dtetadis(iim + 1, jjm + 1, llm)

    ! tendances physiques
    REAL dvfi((iim + 1) * jjm, llm), dufi(ip1jmp1, llm)
    REAL dtetafi(ip1jmp1, llm), dqfi(ip1jmp1, llm, nqmx), dpfi(ip1jmp1)

    ! variables pour le fichier histoire

    INTEGER itau ! index of the time step of the dynamics, starts at 0
    INTEGER itaufin
    REAL time ! time of day, as a fraction of day length
    real finvmaold(ip1jmp1, llm)
    INTEGER l
    REAL rdayvrai, rdaym_ini

    ! Variables test conservation energie
    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(iim + 1, jjm + 1, llm)
    REAL vcont((iim + 1) * jjm, llm), ucont(ip1jmp1, llm)
    logical leapf
    real dt

    !---------------------------------------------------

    print *, "Call sequence information: leapfrog"

    itaufin = nday * day_step
    ! "day_step" is a multiple of "iperiod", therefore "itaufin" is one too

    dq = 0.

    ! On initialise la pression et la fonction d'Exner :
    CALL pression(ip1jmp1, ap, bp, ps, p3d)
    CALL exner_hyb(ps, p3d, pks, pk, pkf)

    ! Début de l'integration temporelle :
    do itau = 0, itaufin - 1
       leapf = mod(itau, iperiod) /= 0
       if (leapf) then
          dt = 2 * dtvr
       else
          ! Matsuno
          dt = dtvr
          if (ok_guide .and. (itaufin - itau - 1) * dtvr > 21600.) &
               call guide(itau, ucov, vcov, teta, q, masse, ps)
          vcovm1 = vcov
          ucovm1 = ucov
          tetam1 = teta
          massem1 = masse
          psm1 = ps
          finvmaold = masse
          CALL filtreg(finvmaold, jjm + 1, llm, - 2, 2, .TRUE., 1)
       end if

       ! Calcul des tendances dynamiques:
       CALL geopot(ip1jmp1, teta, pk, pks, phis, phi)
       CALL caldyn(itau, ucov, vcov, teta, ps, masse, pk, pkf, phis, phi, &
            MOD(itau, iconser) == 0, du, dv, dteta, dp, w, pbaru, pbarv, &
            time_0)

       ! Calcul des tendances advection des traceurs (dont l'humidité)
       CALL caladvtrac(q, pbaru, pbarv, p3d, masse, dq, teta, pk)

       ! Stokage du flux de masse pour traceurs offline:
       IF (offline) CALL fluxstokenc(pbaru, pbarv, masse, teta, phi, phis, &
            dtvr, itau)

       ! integrations dynamique et traceurs:
       CALL integrd(2, vcovm1, ucovm1, tetam1, psm1, massem1, dv, du, dteta, &
            dp, vcov, ucov, teta, q, ps, masse, finvmaold, leapf, dt)

       if (.not. leapf) then
          ! Matsuno backward
          CALL pression(ip1jmp1, ap, bp, ps, p3d)
          CALL exner_hyb(ps, p3d, pks, pk, pkf)

          ! Calcul des tendances dynamiques:
          CALL geopot(ip1jmp1, teta, pk, pks, phis, phi)
          CALL caldyn(itau + 1, ucov, vcov, teta, ps, masse, pk, pkf, phis, &
               phi, .false., du, dv, dteta, dp, w, pbaru, pbarv, time_0)

          ! integrations dynamique et traceurs:
          CALL integrd(2, vcovm1, ucovm1, tetam1, psm1, massem1, dv, du, &
               dteta, dp, vcov, ucov, teta, q, ps, masse, finvmaold, .false., &
               dtvr)
       end if

       IF (MOD(itau + 1, iphysiq) == 0 .AND. iflag_phys /= 0) THEN
          ! calcul des tendances physiques:

          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
          time = REAL(mod(itau, day_step)) / day_step + time_0
          IF (time > 1.) time = time - 1.

          CALL calfis(itau + 1 == itaufin, 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

       CALL pression(ip1jmp1, ap, bp, ps, p3d)
       CALL exner_hyb(ps, p3d, pks, pk, pkf)

       IF (MOD(itau + 1, idissip) == 0) THEN
          ! dissipation horizontale et verticale des petites echelles:

          ! calcul de l'energie cinetique avant dissipation
          call covcont(llm, ucov, vcov, ucont, vcont)
          call enercin(vcov, ucov, vcont, ucont, ecin0)

          ! dissipation
          CALL dissip(vcov, ucov, teta, p3d, dvdis, dudis, dtetadis)
          ucov=ucov + dudis
          vcov=vcov + dvdis

          ! On rajoute la tendance due à la transformation Ec -> E
          ! thermique créée lors de la dissipation
          call covcont(llm, ucov, vcov, ucont, vcont)
          call enercin(vcov, ucov, vcont, ucont, ecin)
          dtetaecdt= (ecin0 - ecin) / pk
          dtetadis=dtetadis + dtetaecdt
          teta=teta + dtetadis

          ! Calcul de la valeur moyenne aux pôles :
          forall (l = 1: llm)
             teta(:, 1, l) = SUM(aire_2d(:iim, 1) * teta(:iim, 1, l)) &
                  / apoln
             teta(:, jjm + 1, l) = SUM(aire_2d(:iim, jjm+1) &
                  * teta(:iim, jjm + 1, l)) / apols
          END forall

          ps(:, 1) = SUM(aire_2d(:iim, 1) * ps(:iim, 1)) / apoln
          ps(:, jjm + 1) = SUM(aire_2d(:iim, jjm+1) * ps(:iim, jjm + 1)) &
               / apols
       END IF

       IF (MOD(itau + 1, iperiod) == 0) THEN
          ! ecriture du fichier histoire moyenne:
          CALL writedynav(histaveid, nqmx, itau + 1, 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
    end do

    CALL dynredem1("restart.nc", vcov, ucov, teta, q, masse, ps, &
         itau=itau_dyn+itaufin)

    ! Calcul des tendances dynamiques:
    CALL geopot(ip1jmp1, teta, pk, pks, phis, phi)
    CALL caldyn(itaufin, ucov, vcov, teta, ps, masse, pk, pkf, phis, phi, &
         MOD(itaufin, iconser) == 0, du, dv, dteta, dp, w, pbaru, pbarv, &
         time_0)

  END SUBROUTINE leapfrog

end module leapfrog_m
1	module leapfrog_m
2
3	IMPLICIT NONE
4
5	contains
6
7	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
10	! Authors: P. Le Van, L. Fairhead, F. Hourdin
11	! schema matsuno + leapfrog
12
13	USE calfis_m, ONLY: calfis
14	USE com_io_dyn, ONLY: histaveid
15	USE comconst, ONLY: daysec, dtphys, dtvr
16	USE comgeom, ONLY: aire_2d, apoln, apols
17	USE comvert, ONLY: ap, bp
18	USE conf_gcm_m, ONLY: day_step, iconser, iperiod, iphysiq, nday, offline, &
19	periodav
20	USE dimens_m, ONLY: iim, jjm, llm, nqmx
21	USE dynetat0_m, ONLY: day_ini
22	use dynredem1_m, only: dynredem1
23	USE exner_hyb_m, ONLY: exner_hyb
24	use filtreg_m, only: filtreg
25	USE guide_m, ONLY: guide
26	use inidissip_m, only: idissip
27	use integrd_m, only: integrd
28	USE logic, ONLY: iflag_phys, ok_guide
29	USE paramet_m, ONLY: ip1jmp1
30	USE pression_m, ONLY: pression
31	USE pressure_var, ONLY: p3d
32	USE temps, ONLY: itau_dyn
33
34	! Variables dynamiques:
35	REAL, intent(inout):: vcov((iim + 1) * jjm, llm) ! vent covariant
36	REAL, intent(inout):: ucov(ip1jmp1, llm) ! vent covariant
37	REAL, intent(inout):: teta(iim + 1, jjm + 1, llm) ! potential temperature
38	REAL ps(iim + 1, jjm + 1) ! pression au sol, en Pa
39
40	REAL masse(ip1jmp1, llm) ! masse d'air
41	REAL phis(ip1jmp1) ! geopotentiel au sol
42	REAL q(ip1jmp1, llm, nqmx) ! mass fractions of advected fields
43	REAL, intent(in):: time_0
44
45	! Variables local to the procedure:
46
47	! Variables dynamiques:
48
49	REAL pks(ip1jmp1) ! exner au sol
50	REAL pk(iim + 1, jjm + 1, llm) ! exner au milieu des couches
51	REAL pkf(ip1jmp1, llm) ! exner filt.au milieu des couches
52	REAL phi(ip1jmp1, llm) ! geopotential
53	REAL w(ip1jmp1, llm) ! vitesse verticale
54
55	! variables dynamiques intermediaire pour le transport
56	REAL pbaru(ip1jmp1, llm), pbarv((iim + 1) * jjm, llm) !flux de masse
57
58	! variables dynamiques au pas - 1
59	REAL vcovm1((iim + 1) * jjm, llm), ucovm1(ip1jmp1, llm)
60	REAL tetam1(iim + 1, jjm + 1, llm), psm1(iim + 1, jjm + 1)
61	REAL massem1(ip1jmp1, llm)
62
63	! tendances dynamiques
64	REAL dv((iim + 1) * jjm, llm), du(ip1jmp1, llm)
65	REAL dteta(ip1jmp1, llm), dq(ip1jmp1, llm, nqmx), dp(ip1jmp1)
66
67	! tendances de la dissipation
68	REAL dvdis((iim + 1) * jjm, llm), dudis(ip1jmp1, llm)
69	REAL dtetadis(iim + 1, jjm + 1, llm)
70
71	! tendances physiques
72	REAL dvfi((iim + 1) * jjm, llm), dufi(ip1jmp1, llm)
73	REAL dtetafi(ip1jmp1, llm), dqfi(ip1jmp1, llm, nqmx), dpfi(ip1jmp1)
74
75	! variables pour le fichier histoire
76
77	INTEGER itau ! index of the time step of the dynamics, starts at 0
78	INTEGER itaufin
79	REAL time ! time of day, as a fraction of day length
80	real finvmaold(ip1jmp1, llm)
81	INTEGER l
82	REAL rdayvrai, rdaym_ini
83
84	! Variables test conservation energie
85	REAL ecin(iim + 1, jjm + 1, llm), ecin0(iim + 1, jjm + 1, llm)
86	! Tendance de la temp. potentiel d (theta) / d t due a la
87	! tansformation d'energie cinetique en energie thermique
88	! cree par la dissipation
89	REAL dtetaecdt(iim + 1, jjm + 1, llm)
90	REAL vcont((iim + 1) * jjm, llm), ucont(ip1jmp1, llm)
91	logical leapf
92	real dt
93
94	!---------------------------------------------------
95
96	print *, "Call sequence information: leapfrog"
97
98	itaufin = nday * day_step
99	! "day_step" is a multiple of "iperiod", therefore "itaufin" is one too
100
101	dq = 0.
102
103	! On initialise la pression et la fonction d'Exner :
104	CALL pression(ip1jmp1, ap, bp, ps, p3d)
105	CALL exner_hyb(ps, p3d, pks, pk, pkf)
106
107	! Début de l'integration temporelle :
108	do itau = 0, itaufin - 1
109	leapf = mod(itau, iperiod) /= 0
110	if (leapf) then
111	dt = 2 * dtvr
112	else
113	! Matsuno
114	dt = dtvr
115	if (ok_guide .and. (itaufin - itau - 1) * dtvr > 21600.) &
116	call guide(itau, ucov, vcov, teta, q, masse, ps)
117	vcovm1 = vcov
118	ucovm1 = ucov
119	tetam1 = teta
120	massem1 = masse
121	psm1 = ps
122	finvmaold = masse
123	CALL filtreg(finvmaold, jjm + 1, llm, - 2, 2, .TRUE., 1)
124	end if
125
126	! Calcul des tendances dynamiques:
127	CALL geopot(ip1jmp1, teta, pk, pks, phis, phi)
128	CALL caldyn(itau, ucov, vcov, teta, ps, masse, pk, pkf, phis, phi, &
129	MOD(itau, iconser) == 0, du, dv, dteta, dp, w, pbaru, pbarv, &
130	time_0)
131
132	! Calcul des tendances advection des traceurs (dont l'humidité)
133	CALL caladvtrac(q, pbaru, pbarv, p3d, masse, dq, teta, pk)
134
135	! Stokage du flux de masse pour traceurs offline:
136	IF (offline) CALL fluxstokenc(pbaru, pbarv, masse, teta, phi, phis, &
137	dtvr, itau)
138
139	! integrations dynamique et traceurs:
140	CALL integrd(2, vcovm1, ucovm1, tetam1, psm1, massem1, dv, du, dteta, &
141	dp, vcov, ucov, teta, q, ps, masse, finvmaold, leapf, dt)
142
143	if (.not. leapf) then
144	! Matsuno backward
145	CALL pression(ip1jmp1, ap, bp, ps, p3d)
146	CALL exner_hyb(ps, p3d, pks, pk, pkf)
147
148	! Calcul des tendances dynamiques:
149	CALL geopot(ip1jmp1, teta, pk, pks, phis, phi)
150	CALL caldyn(itau + 1, ucov, vcov, teta, ps, masse, pk, pkf, phis, &
151	phi, .false., du, dv, dteta, dp, w, pbaru, pbarv, time_0)
152
153	! integrations dynamique et traceurs:
154	CALL integrd(2, vcovm1, ucovm1, tetam1, psm1, massem1, dv, du, &
155	dteta, dp, vcov, ucov, teta, q, ps, masse, finvmaold, .false., &
156	dtvr)
157	end if
158
159	IF (MOD(itau + 1, iphysiq) == 0 .AND. iflag_phys /= 0) THEN
160	! calcul des tendances physiques:
161
162	CALL pression(ip1jmp1, ap, bp, ps, p3d)
163	CALL exner_hyb(ps, p3d, pks, pk, pkf)
164
165	rdaym_ini = itau * dtvr / daysec
166	rdayvrai = rdaym_ini + day_ini
167	time = REAL(mod(itau, day_step)) / day_step + time_0
168	IF (time > 1.) time = time - 1.
169
170	CALL calfis(itau + 1 == itaufin, rdayvrai, time, ucov, vcov, &
171	teta, q, masse, ps, pk, phis, phi, du, dv, dteta, dq, w, dufi, &
172	dvfi, dtetafi, dqfi, dpfi)
173
174	! ajout des tendances physiques:
175	CALL addfi(nqmx, dtphys, ucov, vcov, teta, q, ps, dufi, dvfi, &
176	dtetafi, dqfi, dpfi)
177	ENDIF
178
179	CALL pression(ip1jmp1, ap, bp, ps, p3d)
180	CALL exner_hyb(ps, p3d, pks, pk, pkf)
181
182	IF (MOD(itau + 1, idissip) == 0) THEN
183	! dissipation horizontale et verticale des petites echelles:
184
185	! calcul de l'energie cinetique avant dissipation
186	call covcont(llm, ucov, vcov, ucont, vcont)
187	call enercin(vcov, ucov, vcont, ucont, ecin0)
188
189	! dissipation
190	CALL dissip(vcov, ucov, teta, p3d, dvdis, dudis, dtetadis)
191	ucov=ucov + dudis
192	vcov=vcov + dvdis
193
194	! On rajoute la tendance due à la transformation Ec -> E
195	! thermique créée lors de la dissipation
196	call covcont(llm, ucov, vcov, ucont, vcont)
197	call enercin(vcov, ucov, vcont, ucont, ecin)
198	dtetaecdt= (ecin0 - ecin) / pk
199	dtetadis=dtetadis + dtetaecdt
200	teta=teta + dtetadis
201
202	! Calcul de la valeur moyenne aux pôles :
203	forall (l = 1: llm)
204	teta(:, 1, l) = SUM(aire_2d(:iim, 1) * teta(:iim, 1, l)) &
205	/ apoln
206	teta(:, jjm + 1, l) = SUM(aire_2d(:iim, jjm+1) &
207	* teta(:iim, jjm + 1, l)) / apols
208	END forall
209
210	ps(:, 1) = SUM(aire_2d(:iim, 1) * ps(:iim, 1)) / apoln
211	ps(:, jjm + 1) = SUM(aire_2d(:iim, jjm+1) * ps(:iim, jjm + 1)) &
212	/ apols
213	END IF
214
215	IF (MOD(itau + 1, iperiod) == 0) THEN
216	! ecriture du fichier histoire moyenne:
217	CALL writedynav(histaveid, nqmx, itau + 1, vcov, ucov, teta, pk, &
218	phi, q, masse, ps, phis)
219	call bilan_dyn(2, dtvr * iperiod, dtvr * day_step * periodav, ps, &
220	masse, pk, pbaru, pbarv, teta, phi, ucov, vcov, q)
221	ENDIF
222	end do
223
224	CALL dynredem1("restart.nc", vcov, ucov, teta, q, masse, ps, &
225	itau=itau_dyn+itaufin)
226
227	! Calcul des tendances dynamiques:
228	CALL geopot(ip1jmp1, teta, pk, pks, phis, phi)
229	CALL caldyn(itaufin, ucov, vcov, teta, ps, masse, pk, pkf, phis, phi, &
230	MOD(itaufin, iconser) == 0, du, dv, dteta, dp, w, pbaru, pbarv, &
231	time_0)
232
233	END SUBROUTINE leapfrog
234
235	end module leapfrog_m