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.

traadv_mus.F90 in NEMO/branches/2019/dev_r11514_HPC-02_single-core-extrahalo/src/OCE/TRA – NEMO

Context Navigation

source: NEMO/branches/2019/dev_r11514_HPC-02_single-core-extrahalo/src/OCE/TRA/traadv_mus.F90 @ 11719

Last change on this file since 11719 was 11719, checked in by francesca, 5 years ago
add extra halo support- ticket #2009
Property svn:keywords set to `Id`
File size: 21.5 KB

Rev	Line
[5770]	1	MODULE traadv_mus
[503]	2	!!======================================================================
[5770]	3	!! * MODULE traadv_mus *
[2528]	4	!! Ocean tracers: horizontal & vertical advective trend
[503]	5	!!======================================================================
[2528]	6	!! History : ! 2000-06 (A.Estublier) for passive tracers
	7	!! ! 2001-08 (E.Durand, G.Madec) adapted for T & S
	8	!! NEMO 1.0 ! 2002-06 (G. Madec) F90: Free form and module
	9	!! 3.2 ! 2010-05 (C. Ethe, G. Madec) merge TRC-TRA + switch from velocity to transport
[3680]	10	!! 3.4 ! 2012-06 (P. Oddo, M. Vichi) include the upstream where needed
[5770]	11	!! 3.7 ! 2015-09 (G. Madec) add the ice-shelf cavities boundary condition
[503]	12	!!----------------------------------------------------------------------
[3]	13
	14	!!----------------------------------------------------------------------
[5770]	15	!! tra_adv_mus : update the tracer trend with the horizontal
[3]	16	!! and vertical advection trends using MUSCL scheme
	17	!!----------------------------------------------------------------------
[3625]	18	USE oce ! ocean dynamics and active tracers
[4990]	19	USE trc_oce ! share passive tracers/Ocean variables
[3625]	20	USE dom_oce ! ocean space and time domain
[4990]	21	USE trd_oce ! trends: ocean variables
	22	USE trdtra ! tracers trends manager
[5147]	23	USE sbcrnf ! river runoffs
[3625]	24	USE diaptr ! poleward transport diagnostics
[7646]	25	USE diaar5 ! AR5 diagnostics
	26
[4990]	27	!
[9019]	28	USE iom ! XIOS library
[4990]	29	USE in_out_manager ! I/O manager
	30	USE lib_mpp ! distribued memory computing
	31	USE lbclnk ! ocean lateral boundary condition (or mpp link)
[11719]	32	USE lib_fortran ! Fortran utilities (allows no signed zero when 'key_nosignedzero' defined)
	33	USE halo_mng
[3]	34
	35	IMPLICIT NONE
	36	PRIVATE
	37
[5770]	38	PUBLIC tra_adv_mus ! routine called by traadv.F90
[4990]	39
[11719]	40	REAL(wp), ALLOCATABLE, DIMENSION(:,: ) :: r1_e1e2t_exh2, r1_e1e2u_exh2, r1_e1e2v_exh2
	41	REAL(wp), ALLOCATABLE, DIMENSION(:,: ) :: rnfmsk_exh2, upsmsk_exh2, mikt_exh2
	42	REAL(wp), ALLOCATABLE, DIMENSION(:,:,: ) :: tmask_exh2, wmask_exh2, umask_exh2, vmask_exh2
	43	REAL(wp), ALLOCATABLE, DIMENSION(:,:,: ) :: e3u_n_exh2, e3v_n_exh2, e3t_n_exh2, e3w_n_exh2
	44	REAL(wp), ALLOCATABLE, DIMENSION(:,:,: ) :: pun_exh2, pvn_exh2, pwn_exh2 ! 3 ocean velocity components
	45	REAL(wp), ALLOCATABLE, DIMENSION(:,:,:,:) :: ptb_exh2, pta_exh2 ! before and now tracer fields
	46
	47
[4990]	48	REAL(wp), ALLOCATABLE, SAVE, DIMENSION(:,:) :: upsmsk !: mixed upstream/centered scheme near some straits
[7646]	49	! ! and in closed seas (orca 2 and 1 configurations)
[4990]	50	REAL(wp), ALLOCATABLE, SAVE, DIMENSION(:,:,:) :: xind !: mixed upstream/centered index
	51
[7646]	52	LOGICAL :: l_trd ! flag to compute trends
	53	LOGICAL :: l_ptr ! flag to compute poleward transport
	54	LOGICAL :: l_hst ! flag to compute heat/salt transport
[11719]	55
	56	INTEGER :: jphls = 2
[7646]	57
[3]	58	!! * Substitutions
	59	# include "vectopt_loop_substitute.h90"
	60	!!----------------------------------------------------------------------
[9598]	61	!! NEMO/OCE 4.0 , NEMO Consortium (2018)
[1152]	62	!! $Id$
[10068]	63	!! Software governed by the CeCILL license (see ./LICENSE)
[3]	64	!!----------------------------------------------------------------------
	65	CONTAINS
	66
[5770]	67	SUBROUTINE tra_adv_mus( kt, kit000, cdtype, p2dt, pun, pvn, pwn, &
	68	& ptb, pta, kjpt, ld_msc_ups )
[3]	69	!!----------------------------------------------------------------------
[5770]	70	!! * ROUTINE tra_adv_mus *
[216]	71	!!
[5770]	72	!! ** Purpose : Compute the now trend due to total advection of tracers
	73	!! using a MUSCL scheme (Monotone Upstream-centered Scheme for
	74	!! Conservation Laws) and add it to the general tracer trend.
[3]	75	!!
[216]	76	!! ** Method : MUSCL scheme plus centered scheme at ocean boundaries
[5770]	77	!! ld_msc_ups=T :
[3]	78	!!
[6140]	79	!! ** Action : - update pta with the now advective tracer trends
	80	!! - send trends to trdtra module for further diagnostcs (l_trdtra=T)
	81	!! - htr_adv, str_adv : poleward advective heat and salt transport (ln_diaptr=T)
[3]	82	!!
[503]	83	!! References : Estubier, A., and M. Levy, Notes Techn. Pole de Modelisation
	84	!! IPSL, Sept. 2000 (http://www.lodyc.jussieu.fr/opa)
	85	!!----------------------------------------------------------------------
[2528]	86	INTEGER , INTENT(in ) :: kt ! ocean time-step index
[3294]	87	INTEGER , INTENT(in ) :: kit000 ! first time step index
[2528]	88	CHARACTER(len=3) , INTENT(in ) :: cdtype ! =TRA or TRC (tracer indicator)
	89	INTEGER , INTENT(in ) :: kjpt ! number of tracers
[3718]	90	LOGICAL , INTENT(in ) :: ld_msc_ups ! use upstream scheme within muscl
[6140]	91	REAL(wp) , INTENT(in ) :: p2dt ! tracer time-step
[2528]	92	REAL(wp), DIMENSION(jpi,jpj,jpk ), INTENT(in ) :: pun, pvn, pwn ! 3 ocean velocity components
	93	REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(in ) :: ptb ! before tracer field
	94	REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(inout) :: pta ! tracer trend
[2715]	95	!
[9019]	96	INTEGER :: ji, jj, jk, jn ! dummy loop indices
[11719]	97	INTEGER :: last_khls, ierr ! local integer
[9019]	98	REAL(wp) :: zu, z0u, zzwx, zw , zalpha ! local scalars
	99	REAL(wp) :: zv, z0v, zzwy, z0w ! - -
[11719]	100	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE :: zwx, zslpx ! 3D workspace
	101	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE :: zwy, zslpy ! - -
[3]	102	!!----------------------------------------------------------------------
[3294]	103	!
[11719]	104
	105	CALL halo_mng_set(jphls)
	106
	107	ALLOCATE(zwx(jplbi:jpi,jplbj:jpj,jpk))
	108	ALLOCATE(zwy(jplbi:jpi,jplbj:jpj,jpk))
	109	ALLOCATE(zslpx(jplbi:jpi,jplbj:jpj,jpk))
	110	ALLOCATE(zslpy(jplbi:jpi,jplbj:jpj,jpk))
	111
	112	IF (kt==kit000) THEN
	113	if (.not. allocated(pun_exh2)) ALLOCATE(pun_exh2(jplbi:jpi,jplbj:jpj,jpk))
	114	if (.not. allocated(pvn_exh2)) ALLOCATE(pvn_exh2(jplbi:jpi,jplbj:jpj,jpk))
	115	if (.not. allocated(pwn_exh2)) ALLOCATE(pwn_exh2(jplbi:jpi,jplbj:jpj,jpk))
	116	if (.not. allocated(ptb_exh2)) ALLOCATE(ptb_exh2(jplbi:jpi,jplbj:jpj,jpk,kjpt))
	117	if (.not. allocated(pta_exh2)) ALLOCATE(pta_exh2(jplbi:jpi,jplbj:jpj,jpk,kjpt))
	118	if (.not. allocated(r1_e1e2t_exh2)) ALLOCATE(r1_e1e2t_exh2(jplbi:jpi,jplbj:jpj))
	119	if (.not. allocated(r1_e1e2u_exh2)) ALLOCATE(r1_e1e2u_exh2(jplbi:jpi,jplbj:jpj))
	120	if (.not. allocated(r1_e1e2v_exh2)) ALLOCATE(r1_e1e2v_exh2(jplbi:jpi,jplbj:jpj))
	121	if (.not. allocated(tmask_exh2)) ALLOCATE(tmask_exh2(jplbi:jpi,jplbj:jpj,jpk))
	122	if (.not. allocated(wmask_exh2)) ALLOCATE(wmask_exh2(jplbi:jpi,jplbj:jpj,jpk))
	123	if (.not. allocated(umask_exh2)) ALLOCATE(umask_exh2(jplbi:jpi,jplbj:jpj,jpk))
	124	if (.not. allocated(vmask_exh2)) ALLOCATE(vmask_exh2(jplbi:jpi,jplbj:jpj,jpk))
	125	if (.not. allocated(e3u_n_exh2)) ALLOCATE(e3u_n_exh2(jplbi:jpi,jplbj:jpj,jpk))
	126	if (.not. allocated(e3v_n_exh2)) ALLOCATE(e3v_n_exh2(jplbi:jpi,jplbj:jpj,jpk))
	127	if (.not. allocated(e3t_n_exh2)) ALLOCATE(e3t_n_exh2(jplbi:jpi,jplbj:jpj,jpk))
	128	if (.not. allocated(e3w_n_exh2)) ALLOCATE(e3w_n_exh2(jplbi:jpi,jplbj:jpj,jpk))
	129	IF( ln_isfcav.and..not.allocated(mikt_exh2)) ALLOCATE(mikt_exh2(jplbi:jpi,jplbj:jpj))
	130	IF( ld_msc_ups.and..not.allocated(rnfmsk_exh2)) ALLOCATE(rnfmsk_exh2(jplbi:jpi,jplbj:jpj))
	131	IF( ld_msc_ups.and..not.allocated(upsmsk_exh2)) ALLOCATE(upsmsk_exh2(jplbi:jpi,jplbj:jpj))
	132
	133	CALL halo_mng_copy(r1_e1e2t, r1_e1e2t_exh2)
	134	CALL halo_mng_copy(r1_e1e2u, r1_e1e2u_exh2)
	135	CALL halo_mng_copy(r1_e1e2v, r1_e1e2v_exh2)
	136	CALL halo_mng_copy(tmask, tmask_exh2)
	137	CALL halo_mng_copy(wmask, wmask_exh2)
	138	CALL halo_mng_copy(umask, umask_exh2)
	139	CALL halo_mng_copy(vmask, vmask_exh2)
	140
	141	CALL lbc_lnk( 'traadv_mus', r1_e1e2u_exh2, 'U', -1.)
	142	CALL lbc_lnk( 'traadv_mus', r1_e1e2v_exh2, 'V', -1.)
	143	CALL lbc_lnk( 'traadv_mus', r1_e1e2t_exh2, 'T', 1.)
	144	CALL lbc_lnk( 'traadv_mus', tmask_exh2, 'T', 1. )
	145	CALL lbc_lnk( 'traadv_mus', wmask_exh2, 'W', 1.)
	146	CALL lbc_lnk( 'traadv_mus', umask_exh2, 'U', 1. )
	147	CALL lbc_lnk( 'traadv_mus', vmask_exh2, 'V', 1.)
	148	ENDIF
	149
	150	IF( ln_isfcav ) THEN ; CALL halo_mng_copy(REAL(mikt), mikt_exh2) ; CALL lbc_lnk( 'traadv_mus', mikt_exh2, 'T', 1.) ; ENDIF
	151	IF( ld_msc_ups) THEN ; CALL halo_mng_copy(rnfmsk, rnfmsk_exh2) ; CALL lbc_lnk( 'traadv_mus', rnfmsk_exh2, 'T', 1.) ; ENDIF
	152	IF( ld_msc_ups) THEN ; CALL halo_mng_copy(upsmsk, upsmsk_exh2) ; CALL lbc_lnk( 'traadv_mus', upsmsk_exh2, 'T', 1.) ; ENDIF
	153
	154	CALL halo_mng_copy(e3u_n, e3u_n_exh2)
	155	CALL halo_mng_copy(e3v_n, e3v_n_exh2)
	156	CALL halo_mng_copy(e3t_n, e3t_n_exh2)
	157	CALL halo_mng_copy(e3w_n, e3w_n_exh2)
	158	CALL halo_mng_copy(pun, pun_exh2)
	159	CALL halo_mng_copy(pvn, pvn_exh2)
	160	CALL halo_mng_copy(pwn, pwn_exh2)
	161	CALL halo_mng_copy(ptb, ptb_exh2)
	162	CALL halo_mng_copy(pta, pta_exh2)
	163
	164	CALL lbc_lnk( 'traadv_mus', e3u_n_exh2, 'U', -1., pfillval = 1.0_wp )
	165	CALL lbc_lnk( 'traadv_mus', e3v_n_exh2, 'V', -1., pfillval = 1.0_wp )
	166	CALL lbc_lnk( 'traadv_mus', e3t_n_exh2, 'T', 1., pfillval = 1.0_wp )
	167	CALL lbc_lnk( 'traadv_mus', e3w_n_exh2, 'W', 1., pfillval = 1.0_wp )
	168	CALL lbc_lnk( 'traadv_mus', pun_exh2, 'U', -1.)
	169	CALL lbc_lnk( 'traadv_mus', pvn_exh2, 'V', -1.)
	170	CALL lbc_lnk( 'traadv_mus', pwn_exh2, 'W', 1.)
	171	CALL lbc_lnk( 'traadv_mus', pta_exh2, 'T', 1.)
	172	CALL lbc_lnk( 'traadv_mus', ptb_exh2, 'T', 1.)
	173
	174	# define pun pun_exh2
	175	# define pvn pvn_exh2
	176	# define pwn pwn_exh2
	177	# define ptb ptb_exh2
	178	# define pta pta_exh2
	179	# define r1_e1e2t r1_e1e2t_exh2
	180	# define r1_e1e2u r1_e1e2u_exh2
	181	# define r1_e1e2v r1_e1e2v_exh2
	182	# define tmask tmask_exh2
	183	# define wmask wmask_exh2
	184	# define umask umask_exh2
	185	# define vmask vmask_exh2
	186	# define e3u_n e3u_n_exh2
	187	# define e3v_n e3v_n_exh2
	188	# define e3t_n e3t_n_exh2
	189	# define e3w_n e3w_n_exh2
	190	# define mikt mikt_exh2
	191	# define rnfmsk rnfmsk_exh2
	192	# define upsmsk upsmsk_exh2
	193
[3294]	194	IF( kt == kit000 ) THEN
[2528]	195	IF(lwp) WRITE(numout,*)
	196	IF(lwp) WRITE(numout,*) 'tra_adv : MUSCL advection scheme on ', cdtype
[3718]	197	IF(lwp) WRITE(numout,*) ' : mixed up-stream ', ld_msc_ups
[2528]	198	IF(lwp) WRITE(numout,*) '~~~~~~~'
[3680]	199	IF(lwp) WRITE(numout,*)
[2528]	200	!
[5770]	201	! Upstream / MUSCL scheme indicator
[3680]	202	!
[11719]	203	ALLOCATE( xind(jplbi:jpi,jplbj:jpj,jpk), STAT=ierr )
[7753]	204	xind(:,:,:) = 1._wp ! set equal to 1 where up-stream is not needed
[5770]	205	!
	206	IF( ld_msc_ups ) THEN ! define the upstream indicator (if asked)
[11719]	207	ALLOCATE( upsmsk(jplbi:jpi,jplbj:jpj), STAT=ierr )
[7753]	208	upsmsk(:,:) = 0._wp ! not upstream by default
[5770]	209	!
[4990]	210	DO jk = 1, jpkm1
[7753]	211	xind(:,:,jk) = 1._wp & ! =>1 where up-stream is not needed
	212	& - MAX ( rnfmsk(:,:) * rnfmsk_z(jk), & ! =>0 near runoff mouths (& closed sea outflows)
	213	& upsmsk(:,:) ) * tmask(:,:,jk) ! =>0 in some user defined area
[3718]	214	END DO
[3680]	215	ENDIF
[3718]	216	!
	217	ENDIF
[4990]	218	!
[7646]	219	l_trd = .FALSE.
	220	l_hst = .FALSE.
	221	l_ptr = .FALSE.
	222	IF( ( cdtype == 'TRA' .AND. l_trdtra ) .OR. ( cdtype == 'TRC' .AND. l_trdtrc ) ) l_trd = .TRUE.
	223	IF( cdtype == 'TRA' .AND. ln_diaptr ) l_ptr = .TRUE.
	224	IF( cdtype == 'TRA' .AND. ( iom_use("uadv_heattr") .OR. iom_use("vadv_heattr") .OR. &
	225	& iom_use("uadv_salttr") .OR. iom_use("vadv_salttr") ) ) l_hst = .TRUE.
	226	!
[6140]	227	DO jn = 1, kjpt !== loop over the tracers ==!
	228	!
	229	! !* Horizontal advective fluxes
	230	!
	231	! !-- first guess of the slopes
[7753]	232	zwx(:,:,jpk) = 0._wp ! bottom values
	233	zwy(:,:,jpk) = 0._wp
[6140]	234	DO jk = 1, jpkm1 ! interior values
[11719]	235	DO jj = jplbj, jpj-1
	236	DO ji = jplbi, jpi-1 ! vector opt.
[2528]	237	zwx(ji,jj,jk) = umask(ji,jj,jk) * ( ptb(ji+1,jj,jk,jn) - ptb(ji,jj,jk,jn) )
	238	zwy(ji,jj,jk) = vmask(ji,jj,jk) * ( ptb(ji,jj+1,jk,jn) - ptb(ji,jj,jk,jn) )
	239	END DO
	240	END DO
[3]	241	END DO
[9094]	242	! lateral boundary conditions (changed sign)
[11719]	243	!CALL lbc_lnk_multi( 'traadv_mus', zwx, 'U', -1. , zwy, 'V', -1. )
[6140]	244	! !-- Slopes of tracer
[7753]	245	zslpx(:,:,jpk) = 0._wp ! bottom values
	246	zslpy(:,:,jpk) = 0._wp
[6140]	247	DO jk = 1, jpkm1 ! interior values
[11719]	248	DO jj = jplbj+1, jpj
	249	DO ji = jplbi+1, jpi ! vector opt.
[2528]	250	zslpx(ji,jj,jk) = ( zwx(ji,jj,jk) + zwx(ji-1,jj ,jk) ) &
	251	& * ( 0.25 + SIGN( 0.25, zwx(ji,jj,jk) * zwx(ji-1,jj ,jk) ) )
	252	zslpy(ji,jj,jk) = ( zwy(ji,jj,jk) + zwy(ji ,jj-1,jk) ) &
	253	& * ( 0.25 + SIGN( 0.25, zwy(ji,jj,jk) * zwy(ji ,jj-1,jk) ) )
	254	END DO
[3]	255	END DO
	256	END DO
[503]	257	!
[6140]	258	DO jk = 1, jpkm1 !-- Slopes limitation
[11719]	259	DO jj = jplbj+1, jpj
	260	DO ji = jplbi+1, jpi ! vector opt.
[2528]	261	zslpx(ji,jj,jk) = SIGN( 1., zslpx(ji,jj,jk) ) * MIN( ABS( zslpx(ji ,jj,jk) ), &
	262	& 2.*ABS( zwx (ji-1,jj,jk) ), &
	263	& 2.*ABS( zwx (ji ,jj,jk) ) )
	264	zslpy(ji,jj,jk) = SIGN( 1., zslpy(ji,jj,jk) ) * MIN( ABS( zslpy(ji,jj ,jk) ), &
	265	& 2.*ABS( zwy (ji,jj-1,jk) ), &
	266	& 2.*ABS( zwy (ji,jj ,jk) ) )
[503]	267	END DO
[2528]	268	END DO
[5770]	269	END DO
	270	!
[6140]	271	DO jk = 1, jpkm1 !-- MUSCL horizontal advective fluxes
[11719]	272	DO jj = jplbj+1, jpj-1
	273	DO ji = jplbi+1, jpi-1 ! vector opt.
[2528]	274	! MUSCL fluxes
	275	z0u = SIGN( 0.5, pun(ji,jj,jk) )
	276	zalpha = 0.5 - z0u
[6140]	277	zu = z0u - 0.5 * pun(ji,jj,jk) * p2dt * r1_e1e2u(ji,jj) / e3u_n(ji,jj,jk)
[4990]	278	zzwx = ptb(ji+1,jj,jk,jn) + xind(ji,jj,jk) * zu * zslpx(ji+1,jj,jk)
	279	zzwy = ptb(ji ,jj,jk,jn) + xind(ji,jj,jk) * zu * zslpx(ji ,jj,jk)
[2528]	280	zwx(ji,jj,jk) = pun(ji,jj,jk) * ( zalpha * zzwx + (1.-zalpha) * zzwy )
	281	!
	282	z0v = SIGN( 0.5, pvn(ji,jj,jk) )
	283	zalpha = 0.5 - z0v
[6140]	284	zv = z0v - 0.5 * pvn(ji,jj,jk) * p2dt * r1_e1e2v(ji,jj) / e3v_n(ji,jj,jk)
[4990]	285	zzwx = ptb(ji,jj+1,jk,jn) + xind(ji,jj,jk) * zv * zslpy(ji,jj+1,jk)
	286	zzwy = ptb(ji,jj ,jk,jn) + xind(ji,jj,jk) * zv * zslpy(ji,jj ,jk)
[2528]	287	zwy(ji,jj,jk) = pvn(ji,jj,jk) * ( zalpha * zzwx + (1.-zalpha) * zzwy )
[503]	288	END DO
	289	END DO
	290	END DO
[11719]	291	!CALL lbc_lnk_multi( 'traadv_mus', zwx, 'U', -1. , zwy, 'V', -1. ) ! lateral boundary conditions (changed sign)
[503]	292	!
[6140]	293	DO jk = 1, jpkm1 !-- Tracer advective trend
[11719]	294	DO jj = jplbj+1, jpj-1
	295	DO ji = jplbi+1, jpi-1 ! vector opt.
[5770]	296	pta(ji,jj,jk,jn) = pta(ji,jj,jk,jn) - ( zwx(ji,jj,jk) - zwx(ji-1,jj ,jk ) &
	297	& + zwy(ji,jj,jk) - zwy(ji ,jj-1,jk ) ) &
[6140]	298	& * r1_e1e2t(ji,jj) / e3t_n(ji,jj,jk)
[3]	299	END DO
[2528]	300	END DO
	301	END DO
[6140]	302	! ! trend diagnostics
[7646]	303	IF( l_trd ) THEN
[4990]	304	CALL trd_tra( kt, cdtype, jn, jptra_xad, zwx, pun, ptb(:,:,:,jn) )
	305	CALL trd_tra( kt, cdtype, jn, jptra_yad, zwy, pvn, ptb(:,:,:,jn) )
[2528]	306	END IF
[7646]	307	! ! "Poleward" heat and salt transports
	308	IF( l_ptr ) CALL dia_ptr_hst( jn, 'adv', zwy(:,:,:) )
	309	! ! heat transport
	310	IF( l_hst ) CALL dia_ar5_hst( jn, 'adv', zwx(:,:,:), zwy(:,:,:) )
[6140]	311	!
	312	! !* Vertical advective fluxes
	313	!
[5770]	314	! !-- first guess of the slopes
[7753]	315	zwx(:,:, 1 ) = 0._wp ! surface & bottom boundary conditions
	316	zwx(:,:,jpk) = 0._wp
[6140]	317	DO jk = 2, jpkm1 ! interior values
[7753]	318	zwx(:,:,jk) = tmask(:,:,jk) * ( ptb(:,:,jk-1,jn) - ptb(:,:,jk,jn) )
[3]	319	END DO
[5770]	320	! !-- Slopes of tracer
[7753]	321	zslpx(:,:,1) = 0._wp ! surface values
[5770]	322	DO jk = 2, jpkm1 ! interior value
[11719]	323	DO jj = jplbj, jpj
	324	DO ji = jplbi, jpi
[6140]	325	zslpx(ji,jj,jk) = ( zwx(ji,jj,jk) + zwx(ji,jj,jk+1) ) &
	326	& * ( 0.25 + SIGN( 0.25, zwx(ji,jj,jk) * zwx(ji,jj,jk+1) ) )
[2528]	327	END DO
[3]	328	END DO
	329	END DO
[6140]	330	DO jk = 2, jpkm1 !-- Slopes limitation
[11719]	331	DO jj = jplbj, jpj ! interior values
	332	DO ji = jplbi, jpi
[2528]	333	zslpx(ji,jj,jk) = SIGN( 1., zslpx(ji,jj,jk) ) * MIN( ABS( zslpx(ji,jj,jk ) ), &
	334	& 2.*ABS( zwx (ji,jj,jk+1) ), &
	335	& 2.*ABS( zwx (ji,jj,jk ) ) )
	336	END DO
[3]	337	END DO
	338	END DO
[6140]	339	DO jk = 1, jpk-2 !-- vertical advective flux
[11719]	340	DO jj = jplbj+1, jpj-1
	341	DO ji = jplbi+1, jpi-1 ! vector opt.
[2528]	342	z0w = SIGN( 0.5, pwn(ji,jj,jk+1) )
	343	zalpha = 0.5 + z0w
[6140]	344	zw = z0w - 0.5 * pwn(ji,jj,jk+1) * p2dt * r1_e1e2t(ji,jj) / e3w_n(ji,jj,jk+1)
[4990]	345	zzwx = ptb(ji,jj,jk+1,jn) + xind(ji,jj,jk) * zw * zslpx(ji,jj,jk+1)
	346	zzwy = ptb(ji,jj,jk ,jn) + xind(ji,jj,jk) * zw * zslpx(ji,jj,jk )
[5770]	347	zwx(ji,jj,jk+1) = pwn(ji,jj,jk+1) * ( zalpha * zzwx + (1.-zalpha) * zzwy ) * wmask(ji,jj,jk)
[2528]	348	END DO
[3]	349	END DO
	350	END DO
[6140]	351	IF( ln_linssh ) THEN ! top values, linear free surface only
	352	IF( ln_isfcav ) THEN ! ice-shelf cavities (top of the ocean)
[11719]	353	DO jj = jplbj, jpj
	354	DO ji = jplbi, jpi
[5770]	355	zwx(ji,jj, mikt(ji,jj) ) = pwn(ji,jj,mikt(ji,jj)) * ptb(ji,jj,mikt(ji,jj),jn)
	356	END DO
	357	END DO
[6140]	358	ELSE ! no cavities: only at the ocean surface
[7753]	359	zwx(:,:,1) = pwn(:,:,1) * ptb(:,:,1,jn)
[5770]	360	ENDIF
	361	ENDIF
	362	!
[6140]	363	DO jk = 1, jpkm1 !-- vertical advective trend
[11719]	364	DO jj = jplbj+1, jpj-1
	365	DO ji = jplbi+1, jpi-1 ! vector opt.
[6140]	366	pta(ji,jj,jk,jn) = pta(ji,jj,jk,jn) - ( zwx(ji,jj,jk) - zwx(ji,jj,jk+1) ) * r1_e1e2t(ji,jj) / e3t_n(ji,jj,jk)
[503]	367	END DO
	368	END DO
	369	END DO
[6140]	370	! ! send trends for diagnostic
[7646]	371	IF( l_trd ) CALL trd_tra( kt, cdtype, jn, jptra_zad, zwx, pwn, ptb(:,:,:,jn) )
[503]	372	!
[6140]	373	END DO ! end of tracer loop
[503]	374	!
[11719]	375	# undef pun
	376	# undef pvn
	377	# undef pwn
	378	# undef ptb
	379	# undef pta
	380	# undef r1_e1e2t
	381	# undef r1_e1e2u
	382	# undef r1_e1e2v
	383	# undef tmask
	384	# undef wmask
	385	# undef umask
	386	# undef vmask
	387	# undef e3u_n
	388	# undef e3v_n
	389	# undef e3t_n
	390	# undef e3w_n
	391	# undef mikt
	392	# undef rnfmsk
	393	# undef upsmsk
	394
	395	CALL halo_mng_copy(pta_exh2, pta)
	396
	397	last_khls = jphls - ((SIZE(pta_exh2, 1) - SIZE(pta, 1))/2)
	398
	399	CALL halo_mng_set(last_khls)
	400
	401	CALL lbc_lnk( 'traadv_mus', pta, 'T', 1. )
	402
	403	IF( kt==nitend ) THEN
	404	if (allocated(pun_exh2)) DEALLOCATE(pun_exh2)
	405	if (allocated(pvn_exh2)) DEALLOCATE(pvn_exh2)
	406	if (allocated(pwn_exh2)) DEALLOCATE(pwn_exh2)
	407	if (allocated(ptb_exh2)) DEALLOCATE(ptb_exh2)
	408	if (allocated(pta_exh2)) DEALLOCATE(pta_exh2)
	409	if (allocated(r1_e1e2t_exh2)) DEALLOCATE(r1_e1e2t_exh2)
	410	if (allocated(r1_e1e2u_exh2)) DEALLOCATE(r1_e1e2u_exh2)
	411	if (allocated(r1_e1e2v_exh2)) DEALLOCATE(r1_e1e2v_exh2)
	412	if (allocated(tmask_exh2)) DEALLOCATE(tmask_exh2)
	413	if (allocated(wmask_exh2)) DEALLOCATE(wmask_exh2)
	414	if (allocated(umask_exh2)) DEALLOCATE(umask_exh2)
	415	if (allocated(vmask_exh2)) DEALLOCATE(vmask_exh2)
	416	if (allocated(e3u_n_exh2)) DEALLOCATE(e3u_n_exh2)
	417	if (allocated(e3v_n_exh2)) DEALLOCATE(e3v_n_exh2)
	418	if (allocated(e3t_n_exh2)) DEALLOCATE(e3t_n_exh2)
	419	if (allocated(e3w_n_exh2)) DEALLOCATE(e3w_n_exh2)
	420	IF (ln_isfcav.and.allocated(mikt_exh2)) DEALLOCATE(mikt_exh2)
	421	IF( ld_msc_ups.and.allocated(rnfmsk_exh2)) DEALLOCATE(rnfmsk_exh2)
	422	IF( ld_msc_ups.and.allocated(upsmsk_exh2)) DEALLOCATE(upsmsk_exh2)
	423
	424	ENDIF
	425
	426	DEALLOCATE(zwx,zwy)
	427	DEALLOCATE(zslpx,zslpy)
[5770]	428	END SUBROUTINE tra_adv_mus
[3]	429
	430	!!======================================================================
[5770]	431	END MODULE traadv_mus

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

Download in other formats: