trunk/dyn3d/leapfrog.f90

module leapfrog_m

  IMPLICIT NONE

contains

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

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

    ! Int\'egration temporelle du mod\`ele : Matsuno-leapfrog scheme.

    use addfi_m, only: addfi
    use bilan_dyn_m, only: bilan_dyn
    use caladvtrac_m, only: caladvtrac
    use caldyn_m, only: caldyn
    USE calfis_m, ONLY: calfis
    USE comconst, ONLY: dtvr
    USE comgeom, ONLY: aire_2d, apoln, apols
    use covcont_m, only: covcont
    USE disvert_m, ONLY: ap, bp
    USE conf_gcm_m, ONLY: day_step, iconser, iperiod, iphysiq, nday, iflag_phys
    USE conf_guide_m, ONLY: ok_guide
    USE dimensions, ONLY: iim, jjm, llm, nqmx
    use dissip_m, only: dissip
    USE dynetat0_m, ONLY: day_ini
    use dynredem1_m, only: dynredem1
    use enercin_m, only: enercin
    USE exner_hyb_m, ONLY: exner_hyb
    use filtreg_scal_m, only: filtreg_scal
    use geopot_m, only: geopot
    USE guide_m, ONLY: guide
    use inidissip_m, only: idissip
    use integrd_m, only: integrd
    use nr_util, only: assert
    USE temps, ONLY: itau_dyn
    use writehist_m, only: writehist

    ! Variables dynamiques:
    REAL, intent(inout):: ucov(:, :, :) ! (iim + 1, jjm + 1, llm) vent covariant
    REAL, intent(inout):: vcov(:, :, :) ! (iim + 1, jjm, llm) ! vent covariant

    REAL, intent(inout):: teta(:, :, :) ! (iim + 1, jjm + 1, llm)
    ! potential temperature

    REAL, intent(inout):: ps(:, :) ! (iim + 1, jjm + 1) pression au sol, en Pa
    REAL, intent(inout):: masse(:, :, :) ! (iim + 1, jjm + 1, llm) masse d'air
    REAL, intent(in):: phis(:, :) ! (iim + 1, jjm + 1) surface geopotential

    REAL, intent(inout):: q(:, :, :, :) ! (iim + 1, jjm + 1, llm, nqmx)
    ! mass fractions of advected fields

    ! Local:

    ! Variables dynamiques:

    REAL pks(iim + 1, jjm + 1) ! exner au sol
    REAL pk(iim + 1, jjm + 1, llm) ! exner au milieu des couches
    REAL pkf(iim + 1, jjm + 1, llm) ! exner filtr\'e au milieu des couches
    REAL phi(iim + 1, jjm + 1, llm) ! geopotential
    REAL w(iim + 1, jjm + 1, llm) ! vitesse verticale

    ! Variables dynamiques interm\'ediaires pour le transport
    ! Flux de masse :
    REAL pbaru(iim + 1, jjm + 1, llm), pbarv(iim + 1, jjm, llm)

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

    ! Tendances dynamiques
    REAL dv((iim + 1) * jjm, llm), du(iim + 1, jjm + 1, llm)
    REAL dteta(iim + 1, jjm + 1, llm)
    real dp(iim + 1, jjm + 1)

    ! Tendances de la dissipation :
    REAL dvdis(iim + 1, jjm, llm), dudis(iim + 1, jjm + 1, llm)
    REAL dtetadis(iim + 1, jjm + 1, llm)

    ! Tendances physiques
    REAL dvfi(iim + 1, jjm, llm), dufi(iim + 1, jjm + 1, llm)
    REAL dtetafi(iim + 1, jjm + 1, llm), dqfi(iim + 1, jjm + 1, llm, nqmx)

    ! Variables pour le fichier histoire
    INTEGER itau ! index of the time step of the dynamics, starts at 0
    INTEGER itaufin
    INTEGER l

    ! Variables test conservation \'energie
    REAL ecin(iim + 1, jjm + 1, llm), ecin0(iim + 1, jjm + 1, llm)

    REAL vcont((iim + 1) * jjm, llm), ucont((iim + 1) * (jjm + 1), llm)
    logical leapf
    real dt ! time step, in s

    REAL p3d(iim + 1, jjm + 1, llm + 1) ! pressure at layer interfaces, in Pa
    ! ("p3d(i, j, l)" is at longitude "rlonv(i)", latitude "rlatu(j)",
    ! for interface "l")

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

    print *, "Call sequence information: leapfrog"
    call assert(shape(ucov) == (/iim + 1, jjm + 1, llm/), "leapfrog")

    itaufin = nday * day_step
    ! "day_step" is a multiple of "iperiod", therefore so is "itaufin".

    ! On initialise la pression et la fonction d'Exner :
    forall (l = 1: llm + 1) p3d(:, :, l) = ap(l) + bp(l) * ps
    CALL exner_hyb(ps, p3d, pks, pk)
    pkf = pk
    CALL filtreg_scal(pkf, direct = .true., intensive = .true.)

    time_integration: do itau = 0, itaufin - 1
       leapf = mod(itau, iperiod) /= 0
       
       if (leapf) then
          dt = 2 * dtvr
       else
          ! Matsuno
          dt = dtvr
          if (ok_guide) call guide(itau, ucov, vcov, teta, q(:, :, :, 1), ps)
          vcovm1 = vcov
          ucovm1 = ucov
          tetam1 = teta
          massem1 = masse
          psm1 = ps
       end if

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

       CALL caladvtrac(q, pbaru, pbarv, p3d, masse, teta, pk)

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

       forall (l = 1: llm + 1) p3d(:, :, l) = ap(l) + bp(l) * ps
       CALL exner_hyb(ps, p3d, pks, pk)
       pkf = pk
       CALL filtreg_scal(pkf, direct = .true., intensive = .true.)

       if (.not. leapf) then
          ! Matsuno backward
          ! Calcul des tendances dynamiques:
          CALL geopot(teta, pk, pks, phis, phi)
          CALL caldyn(itau + 1, ucov, vcov, teta, ps, masse, pk, pkf, phis, &
               phi, du, dv, dteta, dp, w, pbaru, pbarv, conser = .false.)

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

          forall (l = 1: llm + 1) p3d(:, :, l) = ap(l) + bp(l) * ps
          CALL exner_hyb(ps, p3d, pks, pk)
          pkf = pk
          CALL filtreg_scal(pkf, direct = .true., intensive = .true.)
       end if

       IF (MOD(itau + 1, iphysiq) == 0 .AND. iflag_phys) THEN
          CALL calfis(ucov, vcov, teta, q, p3d, pk, phis, phi, w, dufi, dvfi, &
               dtetafi, dqfi, dayvrai = itau / day_step + day_ini, &
               time = REAL(mod(itau, day_step)) / day_step, &
               lafin = itau + 1 == itaufin)

          CALL addfi(ucov, vcov, teta, q, dufi, dvfi, dtetafi, dqfi)
       ENDIF

       IF (MOD(itau + 1, idissip) == 0) THEN
          ! Dissipation horizontale et verticale des petites \'echelles

          ! calcul de l'\'energie cin\'etique 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 ajoute la tendance due \`a la transformation \'energie
          ! cin\'etique en \'energie thermique par la dissipation
          call covcont(llm, ucov, vcov, ucont, vcont)
          call enercin(vcov, ucov, vcont, ucont, ecin)
          dtetadis = dtetadis + (ecin0 - ecin) / pk
          teta = teta + dtetadis

          ! Calcul de la valeur moyenne aux p\^oles :
          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
       END IF

       IF (MOD(itau + 1, iperiod) == 0) THEN
          call bilan_dyn(ps, masse, pk, pbaru, pbarv, teta, phi, ucov, vcov, &
               q(:, :, :, 1))
       ENDIF

       CALL geopot(teta, pk, pks, phis, phi)
       CALL writehist(vcov, ucov, teta, pk, phi, q, masse, ps, &
            itau_w = itau_dyn + itau + 1)
    end do time_integration

    CALL dynredem1(vcov, ucov, teta, q, masse, ps, itau = itau_dyn + itaufin)

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

  END SUBROUTINE leapfrog

end module leapfrog_m
1	guez	3	module leapfrog_m
2
3			IMPLICIT NONE
4
5			contains
6
7	guez	128	SUBROUTINE leapfrog(ucov, vcov, teta, ps, masse, phis, q)
8	guez	3
9	guez	67	! From dyn3d/leapfrog.F, version 1.6, 2005/04/13 08:58:34 revision 616
10	guez	27	! Authors: P. Le Van, L. Fairhead, F. Hourdin
11	guez	3
12	guez	266	! Int\'egration temporelle du mod\`ele : Matsuno-leapfrog scheme.
13	guez	129
14	guez	37	use addfi_m, only: addfi
15	guez	40	use bilan_dyn_m, only: bilan_dyn
16			use caladvtrac_m, only: caladvtrac
17	guez	43	use caldyn_m, only: caldyn
18	guez	26	USE calfis_m, ONLY: calfis
19	guez	178	USE comconst, ONLY: dtvr
20	guez	29	USE comgeom, ONLY: aire_2d, apoln, apols
21	guez	161	use covcont_m, only: covcont
22	guez	66	USE disvert_m, ONLY: ap, bp
23	guez	266	USE conf_gcm_m, ONLY: day_step, iconser, iperiod, iphysiq, nday, iflag_phys
24	guez	115	USE conf_guide_m, ONLY: ok_guide
25	guez	265	USE dimensions, ONLY: iim, jjm, llm, nqmx
26	guez	47	use dissip_m, only: dissip
27	guez	26	USE dynetat0_m, ONLY: day_ini
28	guez	27	use dynredem1_m, only: dynredem1
29	guez	207	use enercin_m, only: enercin
30	guez	26	USE exner_hyb_m, ONLY: exner_hyb
31	guez	137	use filtreg_scal_m, only: filtreg_scal
32	guez	43	use geopot_m, only: geopot
33	guez	26	USE guide_m, ONLY: guide
34			use inidissip_m, only: idissip
35	guez	32	use integrd_m, only: integrd
36	guez	55	use nr_util, only: assert
37	guez	28	USE temps, ONLY: itau_dyn
38	guez	69	use writehist_m, only: writehist
39	guez	3
40	guez	10	! Variables dynamiques:
41	guez	55	REAL, intent(inout):: ucov(:, :, :) ! (iim + 1, jjm + 1, llm) vent covariant
42			REAL, intent(inout):: vcov(:, :, :) ! (iim + 1, jjm, llm) ! vent covariant
43	guez	43
44			REAL, intent(inout):: teta(:, :, :) ! (iim + 1, jjm + 1, llm)
45			! potential temperature
46
47	guez	45	REAL, intent(inout):: ps(:, :) ! (iim + 1, jjm + 1) pression au sol, en Pa
48	guez	70	REAL, intent(inout):: masse(:, :, :) ! (iim + 1, jjm + 1, llm) masse d'air
49	guez	69	REAL, intent(in):: phis(:, :) ! (iim + 1, jjm + 1) surface geopotential
50	guez	40
51			REAL, intent(inout):: q(:, :, :, :) ! (iim + 1, jjm + 1, llm, nqmx)
52			! mass fractions of advected fields
53
54	guez	91	! Local:
55	guez	10
56			! Variables dynamiques:
57
58	guez	70	REAL pks(iim + 1, jjm + 1) ! exner au sol
59	guez	29	REAL pk(iim + 1, jjm + 1, llm) ! exner au milieu des couches
60	guez	90	REAL pkf(iim + 1, jjm + 1, llm) ! exner filtr\'e au milieu des couches
61	guez	47	REAL phi(iim + 1, jjm + 1, llm) ! geopotential
62	guez	91	REAL w(iim + 1, jjm + 1, llm) ! vitesse verticale
63	guez	3
64	guez	260	! Variables dynamiques interm\'ediaires pour le transport
65	guez	55	! Flux de masse :
66	guez	91	REAL pbaru(iim + 1, jjm + 1, llm), pbarv(iim + 1, jjm, llm)
67	guez	3
68	guez	56	! Variables dynamiques au pas - 1
69	guez	55	REAL vcovm1(iim + 1, jjm, llm), ucovm1(iim + 1, jjm + 1, llm)
70	guez	29	REAL tetam1(iim + 1, jjm + 1, llm), psm1(iim + 1, jjm + 1)
71	guez	67	REAL massem1(iim + 1, jjm + 1, llm)
72	guez	3
73	guez	56	! Tendances dynamiques
74	guez	260	REAL dv((iim + 1) * jjm, llm), du(iim + 1, jjm + 1, llm)
75	guez	71	REAL dteta(iim + 1, jjm + 1, llm)
76	guez	252	real dp(iim + 1, jjm + 1)
77	guez	3
78	guez	56	! Tendances de la dissipation :
79	guez	55	REAL dvdis(iim + 1, jjm, llm), dudis(iim + 1, jjm + 1, llm)
80	guez	29	REAL dtetadis(iim + 1, jjm + 1, llm)
81	guez	3
82	guez	56	! Tendances physiques
83	guez	91	REAL dvfi(iim + 1, jjm, llm), dufi(iim + 1, jjm + 1, llm)
84			REAL dtetafi(iim + 1, jjm + 1, llm), dqfi(iim + 1, jjm + 1, llm, nqmx)
85	guez	3
86	guez	56	! Variables pour le fichier histoire
87	guez	22	INTEGER itau ! index of the time step of the dynamics, starts at 0
88	guez	27	INTEGER itaufin
89	guez	33	INTEGER l
90	guez	3
91	guez	90	! Variables test conservation \'energie
92	guez	29	REAL ecin(iim + 1, jjm + 1, llm), ecin0(iim + 1, jjm + 1, llm)
93	guez	43
94	guez	55	REAL vcont((iim + 1) * jjm, llm), ucont((iim + 1) * (jjm + 1), llm)
95	guez	33	logical leapf
96	guez	91	real dt ! time step, in s
97	guez	3
98	guez	212	REAL p3d(iim + 1, jjm + 1, llm + 1) ! pressure at layer interfaces, in Pa
99	guez	162	! ("p3d(i, j, l)" is at longitude "rlonv(i)", latitude "rlatu(j)",
100			! for interface "l")
101
102	guez	3	!---------------------------------------------------
103
104			print *, "Call sequence information: leapfrog"
105	guez	55	call assert(shape(ucov) == (/iim + 1, jjm + 1, llm/), "leapfrog")
106	guez	3
107			itaufin = nday * day_step
108	guez	62	! "day_step" is a multiple of "iperiod", therefore so is "itaufin".
109	guez	30
110	guez	3	! On initialise la pression et la fonction d'Exner :
111	guez	37	forall (l = 1: llm + 1) p3d(:, :, l) = ap(l) + bp(l) * ps
112	guez	137	CALL exner_hyb(ps, p3d, pks, pk)
113			pkf = pk
114			CALL filtreg_scal(pkf, direct = .true., intensive = .true.)
115	guez	3
116	guez	40	time_integration: do itau = 0, itaufin - 1
117	guez	33	leapf = mod(itau, iperiod) /= 0
118	guez	266
119	guez	33	if (leapf) then
120			dt = 2 * dtvr
121			else
122			! Matsuno
123			dt = dtvr
124	guez	108	if (ok_guide) call guide(itau, ucov, vcov, teta, q(:, :, :, 1), ps)
125	guez	33	vcovm1 = vcov
126			ucovm1 = ucov
127			tetam1 = teta
128			massem1 = masse
129			psm1 = ps
130			end if
131	guez	30
132			! Calcul des tendances dynamiques:
133	guez	70	CALL geopot(teta, pk, pks, phis, phi)
134	guez	266	CALL caldyn(itau, ucov, vcov, teta, ps, masse, pk, pkf, phis, phi, du, &
135			dv, dteta, dp, w, pbaru, pbarv, conser = MOD(itau, iconser) == 0)
136	guez	30
137	guez	71	CALL caladvtrac(q, pbaru, pbarv, p3d, masse, teta, pk)
138	guez	33
139	guez	90	! Int\'egrations dynamique et traceurs:
140	guez	266	CALL integrd(vcovm1, ucovm1, tetam1, psm1, massem1, dv, du, dteta, dp, &
141			vcov, ucov, teta, q(:, :, :, :2), ps, masse, dt, leapf)
142	guez	30
143	guez	97	forall (l = 1: llm + 1) p3d(:, :, l) = ap(l) + bp(l) * ps
144	guez	137	CALL exner_hyb(ps, p3d, pks, pk)
145			pkf = pk
146			CALL filtreg_scal(pkf, direct = .true., intensive = .true.)
147	guez	97
148	guez	33	if (.not. leapf) then
149			! Matsuno backward
150	guez	27	! Calcul des tendances dynamiques:
151	guez	70	CALL geopot(teta, pk, pks, phis, phi)
152	guez	33	CALL caldyn(itau + 1, ucov, vcov, teta, ps, masse, pk, pkf, phis, &
153	guez	260	phi, du, dv, dteta, dp, w, pbaru, pbarv, conser = .false.)
154	guez	3
155			! integrations dynamique et traceurs:
156	guez	266	CALL integrd(vcovm1, ucovm1, tetam1, psm1, massem1, dv, du, dteta, &
157			dp, vcov, ucov, teta, q(:, :, :, :2), ps, masse, dtvr, &
158	guez	161	leapf=.false.)
159	guez	97
160			forall (l = 1: llm + 1) p3d(:, :, l) = ap(l) + bp(l) * ps
161	guez	137	CALL exner_hyb(ps, p3d, pks, pk)
162			pkf = pk
163			CALL filtreg_scal(pkf, direct = .true., intensive = .true.)
164	guez	33	end if
165	guez	3
166	guez	208	IF (MOD(itau + 1, iphysiq) == 0 .AND. iflag_phys) THEN
167	guez	162	CALL calfis(ucov, vcov, teta, q, p3d, pk, phis, phi, w, dufi, dvfi, &
168	guez	154	dtetafi, dqfi, dayvrai = itau / day_step + day_ini, &
169			time = REAL(mod(itau, day_step)) / day_step, &
170	guez	128	lafin = itau + 1 == itaufin)
171	guez	3
172	guez	91	CALL addfi(ucov, vcov, teta, q, dufi, dvfi, dtetafi, dqfi)
173	guez	33	ENDIF
174	guez	3
175	guez	33	IF (MOD(itau + 1, idissip) == 0) THEN
176	guez	90	! Dissipation horizontale et verticale des petites \'echelles
177	guez	3
178	guez	90	! calcul de l'\'energie cin\'etique avant dissipation
179	guez	33	call covcont(llm, ucov, vcov, ucont, vcont)
180			call enercin(vcov, ucov, vcont, ucont, ecin0)
181	guez	3
182	guez	33	! dissipation
183			CALL dissip(vcov, ucov, teta, p3d, dvdis, dudis, dtetadis)
184	guez	55	ucov = ucov + dudis
185			vcov = vcov + dvdis
186	guez	3
187	guez	90	! On ajoute la tendance due \`a la transformation \'energie
188			! cin\'etique en \'energie thermique par la dissipation
189	guez	33	call covcont(llm, ucov, vcov, ucont, vcont)
190			call enercin(vcov, ucov, vcont, ucont, ecin)
191	guez	56	dtetadis = dtetadis + (ecin0 - ecin) / pk
192	guez	55	teta = teta + dtetadis
193	guez	3
194	guez	90	! Calcul de la valeur moyenne aux p\^oles :
195	guez	33	forall (l = 1: llm)
196	guez	266	teta(:, 1, l) = SUM(aire_2d(:iim, 1) * teta(:iim, 1, l)) / apoln
197	guez	212	teta(:, jjm + 1, l) = SUM(aire_2d(:iim, jjm + 1) &
198	guez	33	* teta(:iim, jjm + 1, l)) / apols
199			END forall
200			END IF
201	guez	3
202	guez	33	IF (MOD(itau + 1, iperiod) == 0) THEN
203	guez	40	call bilan_dyn(ps, masse, pk, pbaru, pbarv, teta, phi, ucov, vcov, &
204	guez	57	q(:, :, :, 1))
205	guez	33	ENDIF
206	guez	68
207	guez	266	CALL geopot(teta, pk, pks, phis, phi)
208			CALL writehist(vcov, ucov, teta, pk, phi, q, masse, ps, &
209			itau_w = itau_dyn + itau + 1)
210	guez	40	end do time_integration
211	guez	3
212	guez	157	CALL dynredem1(vcov, ucov, teta, q, masse, ps, itau = itau_dyn + itaufin)
213	guez	30
214			! Calcul des tendances dynamiques:
215	guez	70	CALL geopot(teta, pk, pks, phis, phi)
216	guez	266	CALL caldyn(itaufin, ucov, vcov, teta, ps, masse, pk, pkf, phis, phi, du, &
217			dv, dteta, dp, w, pbaru, pbarv, conser = MOD(itaufin, iconser) == 0)
218	guez	56
219	guez	3	END SUBROUTINE leapfrog
220
221			end module leapfrog_m