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.

traldf_iso.F90 in branches/2011/DEV_r2739_STFC_dCSE/NEMOGCM/NEMO/OPA_SRC/TRA – NEMO

Context Navigation

source: branches/2011/DEV_r2739_STFC_dCSE/NEMOGCM/NEMO/OPA_SRC/TRA/traldf_iso.F90 @ 4401

Last change on this file since 4401 was 4401, checked in by trackstand2, 10 years ago
Put jpkf into traldf_iso and traadv_tvd (inc nonosc)
Property svn:keywords set to `Id`
File size: 25.6 KB

Rev	Line
[3]	1	MODULE traldf_iso
[503]	2	!!======================================================================
[457]	3	!! * MODULE traldf_iso *
[2528]	4	!! Ocean tracers: horizontal component of the lateral tracer mixing trend
[503]	5	!!======================================================================
[2528]	6	!! History : OPA ! 1994-08 (G. Madec, M. Imbard)
	7	!! 8.0 ! 1997-05 (G. Madec) split into traldf and trazdf
	8	!! NEMO ! 2002-08 (G. Madec) Free form, F90
	9	!! 1.0 ! 2005-11 (G. Madec) merge traldf and trazdf :-)
	10	!! 3.3 ! 2010-09 (C. Ethe, G. Madec) Merge TRA-TRC
[503]	11	!!----------------------------------------------------------------------
[457]	12	#if defined key_ldfslp \|\| defined key_esopa
[3]	13	!!----------------------------------------------------------------------
[457]	14	!! 'key_ldfslp' slope of the lateral diffusive direction
[3]	15	!!----------------------------------------------------------------------
[457]	16	!! tra_ldf_iso : update the tracer trend with the horizontal
	17	!! component of a iso-neutral laplacian operator
	18	!! and with the vertical part of
	19	!! the isopycnal or geopotential s-coord. operator
[3]	20	!!----------------------------------------------------------------------
[457]	21	USE oce ! ocean dynamics and active tracers
	22	USE dom_oce ! ocean space and time domain
[2528]	23	USE trc_oce ! share passive tracers/Ocean variables
	24	USE zdf_oce ! ocean vertical physics
[74]	25	USE ldftra_oce ! ocean active tracers: lateral physics
[3]	26	USE ldfslp ! iso-neutral slopes
[132]	27	USE diaptr ! poleward transport diagnostics
[2528]	28	USE in_out_manager ! I/O manager
	29	USE iom ! I/O library
[1756]	30	#if defined key_diaar5
	31	USE phycst ! physical constants
	32	USE lbclnk ! ocean lateral boundary conditions (or mpp link)
	33	#endif
[3]	34
	35	IMPLICIT NONE
	36	PRIVATE
	37
[503]	38	PUBLIC tra_ldf_iso ! routine called by step.F90
[3]	39
[3211]	40	!! * Control permutation of array indices
	41	# include "oce_ftrans.h90"
	42	# include "dom_oce_ftrans.h90"
	43	# include "trc_oce_ftrans.h90"
	44	# include "zdf_oce_ftrans.h90"
	45	# include "ldftra_oce_ftrans.h90"
	46	# include "ldfslp_ftrans.h90"
	47
[3]	48	!! * Substitutions
	49	# include "domzgr_substitute.h90"
	50	# include "ldftra_substitute.h90"
	51	# include "vectopt_loop_substitute.h90"
	52	!!----------------------------------------------------------------------
[2528]	53	!! NEMO/OPA 3.3 , NEMO Consortium (2010)
	54	!! $Id$
	55	!! Software governed by the CeCILL licence (NEMOGCM/NEMO_CeCILL.txt)
[247]	56	!!----------------------------------------------------------------------
[3]	57	CONTAINS
	58
[2528]	59	SUBROUTINE tra_ldf_iso( kt, cdtype, pgu, pgv, &
	60	& ptb, pta, kjpt, pahtb0 )
[3]	61	!!----------------------------------------------------------------------
	62	!! * ROUTINE tra_ldf_iso *
[457]	63	!!
[3]	64	!! ** Purpose : Compute the before horizontal tracer (t & s) diffusive
[457]	65	!! trend for a laplacian tensor (ezxcept the dz[ dz[.] ] term) and
	66	!! add it to the general trend of tracer equation.
[3]	67	!!
	68	!! ** Method : The horizontal component of the lateral diffusive trends
	69	!! is provided by a 2nd order operator rotated along neural or geopo-
	70	!! tential surfaces to which an eddy induced advection can be added
	71	!! It is computed using before fields (forward in time) and isopyc-
	72	!! nal or geopotential slopes computed in routine ldfslp.
	73	!!
[2528]	74	!! 1st part : masked horizontal derivative of T ( di[ t ] )
[457]	75	!! ======== with partial cell update if ln_zps=T.
	76	!!
	77	!! 2nd part : horizontal fluxes of the lateral mixing operator
	78	!! ========
[3]	79	!! zftu = (aht+ahtb0) e2u*e3u/e1u di[ tb ]
	80	!! - aht e2u*uslp dk[ mi(mk(tb)) ]
	81	!! zftv = (aht+ahtb0) e1v*e3v/e2v dj[ tb ]
	82	!! - aht e2u*vslp dk[ mj(mk(tb)) ]
	83	!! take the horizontal divergence of the fluxes:
	84	!! difft = 1/(e1te2te3t) { di-1[ zftu ] + dj-1[ zftv ] }
	85	!! Add this trend to the general trend (ta,sa):
	86	!! ta = ta + difft
	87	!!
[457]	88	!! 3rd part: vertical trends of the lateral mixing operator
	89	!! ======== (excluding the vertical flux proportional to dk[t] )
	90	!! vertical fluxes associated with the rotated lateral mixing:
	91	!! zftw =-aht { e2t*wslpi di[ mi(mk(tb)) ]
	92	!! + e1t*wslpj dj[ mj(mk(tb)) ] }
	93	!! take the horizontal divergence of the fluxes:
	94	!! difft = 1/(e1te2te3t) dk[ zftw ]
	95	!! Add this trend to the general trend (ta,sa):
[2528]	96	!! pta = pta + difft
[3]	97	!!
[2528]	98	!! ** Action : Update pta arrays with the before rotated diffusion
[503]	99	!!----------------------------------------------------------------------
[3837]	100	! USE arpdebugging, ONLY: dump_array
[3432]	101	USE timing, ONLY: timing_start, timing_stop
[2715]	102	USE wrk_nemo, ONLY: wrk_in_use, wrk_not_released
	103	USE oce , ONLY: zftu => ua , zftv => va ! (ua,va) used as workspace
[3211]	104	!! DCSE_NEMO: need additional directives for renamed module variables
	105	!FTRANS zftu zftv :I :I :z
	106	#if defined key_z_first
[3432]	107	! USE wrk_nemo, ONLY: wdkt => wrk_3d_9 , wdk1t => wrk_3d_10 ! 3D workspace
[3211]	108	!FTRANS wdkt wdk1t :I :I :z
	109	#else
	110	USE wrk_nemo, ONLY: zdkt => wrk_2d_1 , zdk1t => wrk_2d_2
	111	#endif
	112	USE wrk_nemo, ONLY: z2d => wrk_2d_3 ! 2D workspace
[2715]	113	USE wrk_nemo, ONLY: zdit => wrk_3d_6 , zdjt => wrk_3d_7 , ztfw => wrk_3d_8 ! 3D workspace
[3211]	114	!FTRANS zdit zdjt ztfw :I :I :z
	115
[2715]	116	!
[2528]	117	INTEGER , INTENT(in ) :: kt ! ocean time-step index
	118	CHARACTER(len=3) , INTENT(in ) :: cdtype ! =TRA or TRC (tracer indicator)
	119	INTEGER , INTENT(in ) :: kjpt ! number of tracers
	120	REAL(wp), DIMENSION(jpi,jpj ,kjpt), INTENT(in ) :: pgu, pgv ! tracer gradient at pstep levels
[3211]	121
	122	!! DCSE_NEMO: This style defeats ftrans
	123	! REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(in ) :: ptb ! before and now tracer fields
	124	! REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(inout) :: pta ! tracer trend
	125	!FTRANS ptb pta :I :I :z :
	126	REAL(wp), INTENT(in ) :: ptb(jpi,jpj,jpk,kjpt) ! before and now tracer fields
	127	REAL(wp), INTENT(inout) :: pta(jpi,jpj,jpk,kjpt) ! tracer trend
	128
[2528]	129	REAL(wp) , INTENT(in ) :: pahtb0 ! background diffusion coef
[2715]	130	!
[2528]	131	INTEGER :: ji, jj, jk, jn ! dummy loop indices
	132	REAL(wp) :: zmsku, zabe1, zcof1, zcoef3 ! local scalars
	133	REAL(wp) :: zmskv, zabe2, zcof2, zcoef4 ! - -
	134	REAL(wp) :: zcoef0, zbtr, ztra ! - -
[3432]	135	#if defined key_z_first
	136	REAL(wp) :: wdkt , wdki1t , wdkim1t , wdkj1t , wdkjm1t
	137	REAL(wp) :: wdk1t, wdk1i1t, wdk1im1t, wdk1j1t, wdk1jm1t
	138	#endif
	139
[1756]	140	#if defined key_diaar5
[2528]	141	REAL(wp) :: zztmp ! local scalar
[1756]	142	#endif
[3]	143	!!----------------------------------------------------------------------
	144
[3432]	145	CALL timing_start('tra_ldf_iso')
	146
[3211]	147	#if defined key_z_first
	148	IF( wrk_in_use(3, 6,7,8,9,10) .OR. wrk_in_use(2, 3) ) THEN
	149	#else
[2715]	150	IF( wrk_in_use(3, 6,7,8) .OR. wrk_in_use(2, 1,2,3) ) THEN
[3211]	151	#endif
[2715]	152	CALL ctl_stop('tra_ldf_iso : requested workspace array unavailable') ; RETURN
	153	ENDIF
	154
[2528]	155	IF( kt == nit000 ) THEN
[3]	156	IF(lwp) WRITE(numout,*)
[2528]	157	IF(lwp) WRITE(numout,*) 'tra_ldf_iso : rotated laplacian diffusion operator on ', cdtype
[457]	158	IF(lwp) WRITE(numout,*) '~~~~~~~~~~~'
[3]	159	ENDIF
[2528]	160	!
[3837]	161	! CALL dump_array(kt, 'ptb', ptb(:,:,1,1), withHalos=.TRUE.)
[2528]	162	! ! ===========
[3432]	163	!DIR$ SHORTLOOP
[2528]	164	DO jn = 1, kjpt ! tracer loop
	165	! ! ===========
	166	!
	167	!!----------------------------------------------------------------------
	168	!! I - masked horizontal derivative
	169	!!----------------------------------------------------------------------
[4401]	170	!CALL timing_start('traldf_iso_I')
[2528]	171	!!bug ajout.... why? ( 1,jpj,:) and (jpi,1,:) should be sufficient....
[3432]	172	#if defined key_z_first
	173	DO jj=1,jpj,1
[4401]	174	DO jk=1,jpkf,1
[3432]	175	zdit(1 ,jj,jk) = 0.0_wp
	176	zdit(jpi,jj,jk) = 0.0_wp
	177	zdjt(1 ,jj,jk) = 0.0_wp
	178	zdjt(jpi,jj,jk) = 0.0_wp
	179	END DO
	180	END DO
	181	#else
[4401]	182	zdit (1,:,1:jpkf) = 0.e0 ; zdit (jpi,:,1:jpkf) = 0.e0
	183	zdjt (1,:,1:jpkf) = 0.e0 ; zdjt (jpi,:,1:jpkf) = 0.e0
[3432]	184	#endif
[2528]	185	!!end
[3]	186
[2528]	187	! Horizontal tracer gradient
[3211]	188	#if defined key_z_first
	189	DO jj = 1, jpjm1
	190	DO ji = 1, jpim1
[4401]	191	DO jk = 1, jpkfm1 ! jpkm1
[3211]	192	#else
[4401]	193	DO jk = 1, jpkfm1 ! jpkm1
[2528]	194	DO jj = 1, jpjm1
	195	DO ji = 1, fs_jpim1 ! vector opt.
[3211]	196	#endif
[2528]	197	zdit(ji,jj,jk) = ( ptb(ji+1,jj ,jk,jn) - ptb(ji,jj,jk,jn) ) * umask(ji,jj,jk)
	198	zdjt(ji,jj,jk) = ( ptb(ji ,jj+1,jk,jn) - ptb(ji,jj,jk,jn) ) * vmask(ji,jj,jk)
	199	END DO
[457]	200	END DO
	201	END DO
[2528]	202	IF( ln_zps ) THEN ! partial steps correction at the last ocean level
	203	DO jj = 1, jpjm1
	204	DO ji = 1, fs_jpim1 ! vector opt.
	205	zdit(ji,jj,mbku(ji,jj)) = pgu(ji,jj,jn)
	206	zdjt(ji,jj,mbkv(ji,jj)) = pgv(ji,jj,jn)
	207	END DO
[457]	208	END DO
[3]	209	ENDIF
[3432]	210	!
[4401]	211	!CALL timing_stop('traldf_iso_I','section')
[3]	212
[2528]	213	!!----------------------------------------------------------------------
	214	!! II - horizontal trend (full)
	215	!!----------------------------------------------------------------------
[4401]	216	!CALL timing_start('traldf_iso_II')
[3211]	217	#if defined key_z_first
	218	! 1. Vertical tracer gradient at level jk and jk+1
	219	! ------------------------------------------------
	220	! surface boundary condition: wdkt(jk=1)=wdkt(jk=2)
	221
[3432]	222	!!$ DO jj = 1, jpj
	223	!!$ DO ji = 1, jpi
	224	!!$ DO jk = 1, jpkm1
	225	!!$ wdk1t(ji,jj,jk) = ( ptb(ji,jj,jk,jn) - ptb(ji,jj,jk+1,jn) ) * tmask(ji,jj,jk+1)
	226	!!$ END DO
	227	!!$ wdkt(ji,jj,1) = wdk1t(ji,jj,1)
	228	!!$ DO jk = 2, jpkm1
	229	!!$ wdkt(ji,jj,jk) = ( ptb(ji,jj,jk-1,jn) - ptb(ji,jj,jk,jn) ) * tmask(ji,jj,jk)
	230	!!$ END DO
	231	!!$ END DO
	232	!!$ END DO
[3211]	233
	234	! 2. Horizontal fluxes
	235	! --------------------
[3432]	236	!!$ DO jj = 1 , jpjm1
	237	!!$ DO ji = 1, jpim1
	238	!!$ DO jk = 1, jpkm1
	239	!!$ zabe1 = ( fsahtu(ji,jj,jk) + pahtb0 ) * e2u(ji,jj) * fse3u(ji,jj,jk) / e1u(ji,jj)
	240	!!$ zabe2 = ( fsahtv(ji,jj,jk) + pahtb0 ) * e1v(ji,jj) * fse3v(ji,jj,jk) / e2v(ji,jj)
	241	!!$ zmsku = 1. / MAX( tmask(ji+1,jj,jk ) + tmask(ji,jj,jk+1) &
	242	!!$ & + tmask(ji+1,jj,jk+1) + tmask(ji,jj,jk ), 1. )
	243	!!$ zmskv = 1. / MAX( tmask(ji,jj+1,jk ) + tmask(ji,jj,jk+1) &
	244	!!$ & + tmask(ji,jj+1,jk+1) + tmask(ji,jj,jk ), 1. )
	245	!!$ zcof1 = - fsahtu(ji,jj,jk) * e2u(ji,jj) * uslp(ji,jj,jk) * zmsku
	246	!!$ zcof2 = - fsahtv(ji,jj,jk) * e1v(ji,jj) * vslp(ji,jj,jk) * zmskv
	247	!!$ zftu(ji,jj,jk ) = ( zabe1 * zdit(ji,jj,jk) &
	248	!!$ & + zcof1 * ( wdkt (ji+1,jj,jk) + wdk1t(ji,jj,jk) &
	249	!!$ & + wdk1t(ji+1,jj,jk) + wdkt (ji,jj,jk) ) ) * umask(ji,jj,jk)
	250	!!$ zftv(ji,jj,jk) = ( zabe2 * zdjt(ji,jj,jk) &
	251	!!$ & + zcof2 * ( wdkt (ji,jj+1,jk) + wdk1t(ji,jj,jk) &
	252	!!$ & + wdk1t(ji,jj+1,jk) + wdkt (ji,jj,jk) ) ) * vmask(ji,jj,jk)
	253	!!$ END DO
	254	!!$ END DO
	255	!!$ END DO
[3211]	256
	257	DO jj = 2 , jpjm1
	258	DO ji = 2, jpim1
[4401]	259	DO jk = 1, jpkfm1 ! jpkm1
[3432]	260
	261	! 1. Vertical tracer gradient at level jk and jk+1
	262	! ------------------------------------------------
	263	! surface boundary condition: wdkt(jk=1)=wdkt(jk=2)
	264
	265	wdk1t = ( ptb(ji,jj,jk,jn) - ptb(ji,jj,jk+1,jn) ) * tmask(ji,jj,jk+1)
	266	wdk1i1t = ( ptb(ji+1,jj,jk,jn) - ptb(ji+1,jj,jk+1,jn) ) * tmask(ji+1,jj,jk+1)
	267	wdk1im1t = ( ptb(ji-1,jj,jk,jn) - ptb(ji-1,jj,jk+1,jn) ) * tmask(ji-1,jj,jk+1)
	268	wdk1j1t = ( ptb(ji,jj+1,jk,jn) - ptb(ji,jj+1,jk+1,jn) ) * tmask(ji,jj+1,jk+1)
	269	wdk1jm1t = ( ptb(ji,jj-1,jk,jn) - ptb(ji,jj-1,jk+1,jn) ) * tmask(ji,jj-1,jk+1)
	270
	271	IF(jk > 1)THEN
	272	wdkt = ( ptb(ji,jj,jk-1,jn) - ptb(ji,jj,jk,jn) ) * tmask(ji,jj,jk)
	273	wdki1t = ( ptb(ji+1,jj,jk-1,jn) - ptb(ji+1,jj,jk,jn) ) * tmask(ji+1,jj,jk)
	274	wdkim1t = ( ptb(ji-1,jj,jk-1,jn) - ptb(ji-1,jj,jk,jn) ) * tmask(ji-1,jj,jk)
	275	wdkj1t = ( ptb(ji,jj+1,jk-1,jn) - ptb(ji,jj+1,jk,jn) ) * tmask(ji,jj+1,jk)
	276	wdkjm1t = ( ptb(ji,jj-1,jk-1,jn) - ptb(ji,jj-1,jk,jn) ) * tmask(ji,jj-1,jk)
	277	ELSE
	278	wdkt = wdk1t
	279	wdki1t = wdk1i1t
	280	wdkim1t= wdk1im1t
	281	wdkj1t = wdk1j1t
	282	wdkjm1t= wdk1jm1t
	283	END IF
	284
	285	! II.4 Second derivative (divergence) and add to the general trend
	286	! ----------------------------------------------------------------
	287	zbtr = 1._wp / ( e1t(ji,jj) * e2t(ji,jj) * fse3t(ji,jj,jk) )
	288
	289	ztra = zbtr * ( &
	290
	291	! zftu(ji,jj,jk) -
	292	( ((fsahtu(ji,jj,jk) + pahtb0) * e2u(ji,jj) * fse3u(ji,jj,jk) / e1u(ji,jj)) * zdit(ji,jj,jk) &
	293	- ( fsahtu(ji,jj,jk) * e2u(ji,jj) * uslp(ji,jj,jk) / &
	294	MAX( tmask(ji+1,jj,jk ) + tmask(ji,jj,jk+1) &
	295	+ tmask(ji+1,jj,jk+1) + tmask(ji,jj,jk ), 1.) ) * &
	296	(wdki1t + wdk1t + wdk1i1t + wdkt) ) * umask(ji,jj,jk) - &
	297
	298	! zftu(ji-1,jj,jk) +
	299	( ((fsahtu(ji-1,jj,jk) + pahtb0) * e2u(ji-1,jj) * fse3u(ji-1,jj,jk) / e1u(ji-1,jj)) * zdit(ji-1,jj,jk) &
	300	- ( fsahtu(ji-1,jj,jk) * e2u(ji-1,jj) * uslp(ji-1,jj,jk) / &
	301	MAX( tmask(ji,jj,jk ) + tmask(ji-1,jj,jk+1) &
	302	+ tmask(ji,jj,jk+1) + tmask(ji-1,jj,jk ), 1.) ) * &
	303	(wdkt + wdk1im1t + wdk1t + wdkim1t) ) * umask(ji-1,jj,jk) + &
	304
	305
	306	! zftv(ji,jj,jk) -
	307	( ((fsahtv(ji,jj,jk) + pahtb0) * e1v(ji,jj) * fse3v(ji,jj,jk) / e2v(ji,jj)) * zdjt(ji,jj,jk) &
	308	& - ( fsahtv(ji,jj,jk) * e1v(ji,jj) * vslp(ji,jj,jk) / &
	309	MAX( tmask(ji,jj+1,jk ) + tmask(ji,jj,jk+1) &
	310	& + tmask(ji,jj+1,jk+1) + tmask(ji,jj,jk ), 1. )) * &
	311	(wdkj1t + wdk1t + wdk1j1t + wdkt) ) * vmask(ji,jj,jk) - &
	312	! zftv(ji,jj-1,jk) &
	313	( ((fsahtv(ji,jj-1,jk) + pahtb0) * e1v(ji,jj-1) * fse3v(ji,jj-1,jk) / e2v(ji,jj-1)) * zdjt(ji,jj-1,jk) &
	314	& - ( fsahtv(ji,jj-1,jk) * e1v(ji,jj-1) * vslp(ji,jj-1,jk) / &
	315	MAX( tmask(ji,jj,jk ) + tmask(ji,jj-1,jk+1) &
	316	& + tmask(ji,jj,jk+1) + tmask(ji,jj-1,jk ), 1. )) * &
	317	(wdkt + wdk1jm1t + wdk1t + wdkjm1t) ) * vmask(ji,jj-1,jk) &
	318
	319	)
[3211]	320	pta(ji,jj,jk,jn) = pta(ji,jj,jk,jn) + ztra
	321	END DO
	322	END DO
	323	END DO
	324	#else
[2528]	325	!CDIR PARALLEL DO PRIVATE( zdk1t )
	326	! ! ===============
[4401]	327	DO jk = 1, jpkfm1 ! jpkm1 ! Horizontal slab
[2528]	328	! ! ===============
	329	! 1. Vertical tracer gradient at level jk and jk+1
	330	! ------------------------------------------------
	331	! surface boundary condition: zdkt(jk=1)=zdkt(jk=2)
	332	zdk1t(:,:) = ( ptb(:,:,jk,jn) - ptb(:,:,jk+1,jn) ) * tmask(:,:,jk+1)
	333	!
	334	IF( jk == 1 ) THEN ; zdkt(:,:) = zdk1t(:,:)
	335	ELSE ; zdkt(:,:) = ( ptb(:,:,jk-1,jn) - ptb(:,:,jk,jn) ) * tmask(:,:,jk)
	336	ENDIF
[3]	337
[2528]	338	! 2. Horizontal fluxes
	339	! --------------------
	340	DO jj = 1 , jpjm1
	341	DO ji = 1, fs_jpim1 ! vector opt.
	342	zabe1 = ( fsahtu(ji,jj,jk) + pahtb0 ) * e2u(ji,jj) * fse3u(ji,jj,jk) / e1u(ji,jj)
	343	zabe2 = ( fsahtv(ji,jj,jk) + pahtb0 ) * e1v(ji,jj) * fse3v(ji,jj,jk) / e2v(ji,jj)
	344	!
	345	zmsku = 1. / MAX( tmask(ji+1,jj,jk ) + tmask(ji,jj,jk+1) &
	346	& + tmask(ji+1,jj,jk+1) + tmask(ji,jj,jk ), 1. )
	347	!
	348	zmskv = 1. / MAX( tmask(ji,jj+1,jk ) + tmask(ji,jj,jk+1) &
	349	& + tmask(ji,jj+1,jk+1) + tmask(ji,jj,jk ), 1. )
	350	!
	351	zcof1 = - fsahtu(ji,jj,jk) * e2u(ji,jj) * uslp(ji,jj,jk) * zmsku
	352	zcof2 = - fsahtv(ji,jj,jk) * e1v(ji,jj) * vslp(ji,jj,jk) * zmskv
	353	!
	354	zftu(ji,jj,jk ) = ( zabe1 * zdit(ji,jj,jk) &
	355	& + zcof1 * ( zdkt (ji+1,jj) + zdk1t(ji,jj) &
	356	& + zdk1t(ji+1,jj) + zdkt (ji,jj) ) ) * umask(ji,jj,jk)
	357	zftv(ji,jj,jk) = ( zabe2 * zdjt(ji,jj,jk) &
	358	& + zcof2 * ( zdkt (ji,jj+1) + zdk1t(ji,jj) &
	359	& + zdk1t(ji,jj+1) + zdkt (ji,jj) ) ) * vmask(ji,jj,jk)
	360	END DO
[3]	361	END DO
	362
[2528]	363	! II.4 Second derivative (divergence) and add to the general trend
	364	! ----------------------------------------------------------------
	365	DO jj = 2 , jpjm1
	366	DO ji = fs_2, fs_jpim1 ! vector opt.
	367	zbtr = 1.0 / ( e1t(ji,jj) * e2t(ji,jj) * fse3t(ji,jj,jk) )
	368	ztra = zbtr * ( zftu(ji,jj,jk) - zftu(ji-1,jj,jk) + zftv(ji,jj,jk) - zftv(ji,jj-1,jk) )
	369	pta(ji,jj,jk,jn) = pta(ji,jj,jk,jn) + ztra
	370	END DO
[3]	371	END DO
[2528]	372	! ! ===============
	373	END DO ! End of slab
	374	! ! ===============
[3211]	375	#endif
[2528]	376	!
	377	! "Poleward" diffusive heat or salt transports (T-S case only)
	378	IF( cdtype == 'TRA' .AND. ln_diaptr .AND. ( MOD( kt, nn_fptr ) == 0 ) ) THEN
[3432]	379	IF( jn == jp_tem)THEN
	380	htr_ldf = ptr_vj( zftv )
	381	END IF
	382	IF( jn == jp_sal)THEN
	383	str_ldf = ptr_vj( zftv )
	384	END IF
[2528]	385	ENDIF
	386
[1756]	387	#if defined key_diaar5
[2528]	388	IF( cdtype == 'TRA' .AND. jn == jp_tem ) THEN
	389	z2d(:,:) = 0._wp
	390	zztmp = rau0 * rcp
[3211]	391	#if defined key_z_first
	392	DO jj = 2, jpjm1
	393	DO ji = 2, jpim1
[4401]	394	DO jk = 1, jpkfm1 ! jpkm1
[3211]	395	#else
[4401]	396	DO jk = 1, jpkfm1 ! jpkm1
[2528]	397	DO jj = 2, jpjm1
	398	DO ji = fs_2, fs_jpim1 ! vector opt.
[3211]	399	#endif
[2528]	400	z2d(ji,jj) = z2d(ji,jj) + zftu(ji,jj,jk)
	401	END DO
	402	END DO
[1756]	403	END DO
[2528]	404	z2d(:,:) = zztmp * z2d(:,:)
	405	CALL lbc_lnk( z2d, 'U', -1. )
	406	CALL iom_put( "udiff_heattr", z2d ) ! heat transport in i-direction
	407	z2d(:,:) = 0._wp
[3211]	408	#if defined key_z_first
	409	DO jj = 2, jpjm1
	410	DO ji = 2, jpim1
[4401]	411	DO jk = 1, jpkfm1 ! jpkm1
[3211]	412	#else
[4401]	413	DO jk = 1, jpkfm1 ! jpkm1
[2528]	414	DO jj = 2, jpjm1
	415	DO ji = fs_2, fs_jpim1 ! vector opt.
[3211]	416	#endif
[2528]	417	z2d(ji,jj) = z2d(ji,jj) + zftv(ji,jj,jk)
	418	END DO
	419	END DO
[1756]	420	END DO
[2528]	421	z2d(:,:) = zztmp * z2d(:,:)
	422	CALL lbc_lnk( z2d, 'V', -1. )
	423	CALL iom_put( "vdiff_heattr", z2d ) ! heat transport in i-direction
	424	END IF
[1756]	425	#endif
[4401]	426	!CALL timing_stop('traldf_iso_II','section')
[3]	427
[4401]	428	!!--------------------------------------------------------------------
[2528]	429	!! III - vertical trend of T & S (extra diagonal terms only)
[4401]	430	!!--------------------------------------------------------------------
	431	!CALL timing_start('traldf_iso_III')
[2528]	432
	433	! Local constant initialization
	434	! -----------------------------
[3432]	435	#if defined key_z_first
	436	DO jj=1,jpj,1
[4401]	437	DO jk=1,jpkf,1
[3432]	438	ztfw(1 ,jj,jk) = 0.0_wp
	439	ztfw(jpi,jj,jk) = 0.0_wp
	440	END DO
	441	END DO
	442	#else
[4401]	443	ztfw(1,:,1:jpkf) = 0.e0 ; ztfw(jpi,:,1:jpkf) = 0.e0
[3432]	444	#endif
[2528]	445	! Vertical fluxes
	446	! ---------------
	447
	448	! Surface and bottom vertical fluxes set to zero
[3432]	449	#if defined key_z_first
	450	DO ji=1,jpi,1
	451	DO jj=1,jpj,1
	452	ztfw(ji,jj,1 ) = 0.0_wp
[4401]	453	ztfw(ji,jj,jpkf) = 0.0_wp ! ARPDBG - should this be jpk anyway
	454	! since may be below ocean floor?
[3432]	455	END DO
	456	END DO
	457	#else
[4401]	458	ztfw(:,:, 1 ) = 0.e0 ; ztfw(:,:,jpkf) = 0.e0
[3432]	459	#endif
	460
[2528]	461	! interior (2=<jk=<jpk-1)
[3211]	462	#if defined key_z_first
	463	DO jj = 2, jpjm1
	464	DO ji = 2, jpim1
[4401]	465	DO jk = 2, jpkfm1
[3211]	466	#else
[4401]	467	DO jk = 2, jpkfm1
[2528]	468	DO jj = 2, jpjm1
	469	DO ji = fs_2, fs_jpim1 ! vector opt.
[3211]	470	#endif
[2528]	471	zcoef0 = - fsahtw(ji,jj,jk) * tmask(ji,jj,jk)
	472	!
	473	zmsku = 1./MAX( umask(ji ,jj,jk-1) + umask(ji-1,jj,jk) &
	474	& + umask(ji-1,jj,jk-1) + umask(ji ,jj,jk), 1. )
	475	zmskv = 1./MAX( vmask(ji,jj ,jk-1) + vmask(ji,jj-1,jk) &
	476	& + vmask(ji,jj-1,jk-1) + vmask(ji,jj ,jk), 1. )
	477	!
	478	zcoef3 = zcoef0 * e2t(ji,jj) * zmsku * wslpi (ji,jj,jk)
	479	zcoef4 = zcoef0 * e1t(ji,jj) * zmskv * wslpj (ji,jj,jk)
	480	!
	481	ztfw(ji,jj,jk) = zcoef3 * ( zdit(ji ,jj ,jk-1) + zdit(ji-1,jj ,jk) &
	482	& + zdit(ji-1,jj ,jk-1) + zdit(ji ,jj ,jk) ) &
	483	& + zcoef4 * ( zdjt(ji ,jj ,jk-1) + zdjt(ji ,jj-1,jk) &
	484	& + zdjt(ji ,jj-1,jk-1) + zdjt(ji ,jj ,jk) )
	485	END DO
[457]	486	END DO
	487	END DO
[2528]	488
	489
	490	! I.5 Divergence of vertical fluxes added to the general tracer trend
	491	! -------------------------------------------------------------------
[3211]	492	#if defined key_z_first
	493	DO jj = 2, jpjm1
	494	DO ji = 2, jpim1
[4401]	495	DO jk = 1, jpkfm1
[3211]	496	#else
[4401]	497	DO jk = 1, jpkfm1
[2528]	498	DO jj = 2, jpjm1
	499	DO ji = fs_2, fs_jpim1 ! vector opt.
[3211]	500	#endif
[2528]	501	zbtr = 1.0 / ( e1t(ji,jj) * e2t(ji,jj) * fse3t(ji,jj,jk) )
	502	ztra = ( ztfw(ji,jj,jk) - ztfw(ji,jj,jk+1) ) * zbtr
	503	pta(ji,jj,jk,jn) = pta(ji,jj,jk,jn) + ztra
	504	END DO
[457]	505	END DO
	506	END DO
[2528]	507	!
[3432]	508
[4401]	509	!CALL timing_stop('traldf_iso_III','section')
[3432]	510
[457]	511	END DO
[503]	512	!
[3211]	513	#if defined key_z_first
	514	IF( wrk_not_released(3, 6,7,8,9,10) .OR. &
	515	wrk_not_released(2, 3) ) CALL ctl_stop('tra_ldf_iso: failed to release workspace arrays')
	516	#else
[2715]	517	IF( wrk_not_released(3, 6,7,8) .OR. &
	518	wrk_not_released(2, 1,2,3) ) CALL ctl_stop('tra_ldf_iso: failed to release workspace arrays')
[3211]	519	#endif
[2715]	520	!
[3432]	521	CALL timing_stop('tra_ldf_iso','section')
	522	!
[3]	523	END SUBROUTINE tra_ldf_iso
	524
	525	#else
	526	!!----------------------------------------------------------------------
[457]	527	!! default option : Dummy code NO rotation of the diffusive tensor
[3]	528	!!----------------------------------------------------------------------
	529	CONTAINS
[2528]	530	SUBROUTINE tra_ldf_iso( kt, cdtype, pgu, pgv, ptb, pta, kjpt, pahtb0 ) ! Empty routine
	531	CHARACTER(len=3) :: cdtype
	532	REAL, DIMENSION(:,:,:) :: pgu, pgv ! tracer gradient at pstep levels
	533	REAL, DIMENSION(:,:,:,:) :: ptb, pta
	534	WRITE(,) 'tra_ldf_iso: You should not have seen this print! error?', kt, cdtype, pgu(1,1,1), pgv(1,1,1), &
	535	& ptb(1,1,1,1), pta(1,1,1,1), kjpt, pahtb0
[3]	536	END SUBROUTINE tra_ldf_iso
	537	#endif
	538
	539	!!==============================================================================
	540	END MODULE traldf_iso

Note: See TracBrowser for help on using the repository browser.

Download in other formats: