!-------------------------------------- LICENCE BEGIN ------------------------------------
!Environment Canada - Atmospheric Science and Technology License/Disclaimer,
! version 3; Last Modified: May 7, 2008.
!This is free but copyrighted software; you can use/redistribute/modify it under the terms
!of the Environment Canada - Atmospheric Science and Technology License/Disclaimer
!version 3 or (at your option) any later version that should be found at:
!http://collaboration.cmc.ec.gc.ca/science/rpn.comm/license.html
!
!This software is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
!without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
!See the above mentioned License/Disclaimer for more details.
!You should have received a copy of the License/Disclaimer along with this software;
!if not, you can write to: EC-RPN COMM Group, 2121 TransCanada, suite 500, Dorval (Quebec),
!CANADA, H9P 1J3; or send e-mail to service.rpn@ec.gc.ca
!-------------------------------------- LICENCE END --------------------------------------
***s/r adw_main_2_pos_tl - TLM of adw_main_2_pos when Adw_nosetint_L=.TRUE.
*
#include "model_macros_f.h"
*
subroutine adw_main_2_pos_tl ( F_it, F_u, F_v, F_w, F_um, F_vm, F_wm ) 1,23
*
implicit none
*
integer F_it
real F_u (*),F_v (*),F_w (*)
real F_um(*),F_vm(*),F_wm(*)
*
*author
* monique tanguay
*
*revision
* v2_31 - Tanguay M. - initial MPI version
* v3_00 - Tanguay M. - adapt to restructured adw_main
* v3_11 - Tanguay M. - AIXport+Opti+OpenMP for TLM-ADJ
* v3_20 - Tanguay M. - Optimized SETINT/TRILIN
* - TLM of Change test a lower and upper boundaries
* v3_21 - Tanguay M. - Call adw_main_2_pos_noset_tl based on Adw_nosetint_L
* v3_30 - Tanguay M. - correct calculation for LAM when Glb_pil gt 7
* - correct parameters adw_trilin_turbo_tl
* v3_31 - Tanguay M. - new scope for operator + adw_cliptraj (LAM)
* v3_31 - Tanguay M. - Do adjoint of outsiders in advection
* v3_31 - Tanguay M. - SETTLS option
*
*language
* fortran 77
*
*object
* see id section
*
*arguments
*______________________________________________________________________
* | | |
* NAME | DESCRIPTION | I/O |
*--------|-------------------------------------------------------|-----|
* F_it | total number of iterations for trajectory | i |
* | | |
* F_u,F_v| input: 3 components of wind on advection grid | io |
* F_w | output: 3 components of upstream positions at t1 | |
*________|_______________________________________________________|_____|
*
*implicits
#include "glb_ld.cdk"
#include "glb_pil.cdk"
#include "lun.cdk"
#include "geomg.cdk"
#include "adw.cdk"
#include "dcst.cdk"
#include "cstv.cdk"
#include "vth.cdk"
#include "vt1.cdk"
#include "v4dg.cdk"
#include "lctl.cdk"
#include "vthm.cdk"
#include "vt1m.cdk"
#include "schm.cdk"
#include "orh.cdk"
*
*modules
integer vmmlod, vmmget, vmmuld
external vmmlod, vmmget, vmmuld
*
logical doh_L
*
integer pnerr, pnlkey1(30), pnlod
*
integer i, j, k, n, ij, ijk, nij, nijk, it
integer i1,j1,k1,nn
*
real, dimension(l_ni*l_nj*l_nk) :: wrkx1,wrky1,wrkz1,wrkc1,wrk_yth
*
real, dimension(l_ni*l_nj*l_nk) :: capx1m,capy1m,capz1m
real, dimension(l_ni*l_nj*l_nk) :: wrkx1m,wrky1m,wrkz1m,wrkc1m,wrk_ythm
*
integer, dimension(:), allocatable :: n2m
real, dimension(:), allocatable :: capx2m,capy2m,capz2m
real, dimension(:), allocatable :: xpos2m,ypos2m,zpos2m
*
integer, dimension(:), allocatable :: n2
real, dimension(:), allocatable :: capx2,capy2,capz2
real, dimension(:), allocatable :: xpos2,ypos2,zpos2
*
real dummy, dth
real*8 r2pi_8,TWO_8,HALF_8,pdp_8,pdm_8
parameter (TWO_8 = 2.0,HALF_8=0.5)
*
integer i0,in,j0,jn
*
logical step2_settls_L
*
************************************************************************
if ( .not.Adw_nosetint_L ) call gem_stop ('ABORT adw_main_2_pos_tl',-1)
************************************************************************
*
if (Lun_debug_L) write (Lun_out,1000)
*
step2_settls_L = Schm_settls_L.and.Orh_crank_L.and.Orh_icn.eq.Schm_itcn
*
nij = l_ni *l_nj
nijk = l_ni *l_nj *l_nk
*
r2pi_8 = TWO_8 * Dcst_pi_8
dth = Cstv_dt_8/2.
pdp_8 = 1.d0 + 1.d-6
pdm_8 = 1.d0 - 1.d-6
*
pnlkey1(1) = VMM_KEY(xth)
pnlkey1(2) = VMM_KEY(yth)
pnlkey1(3) = VMM_KEY(zth)
pnlkey1(4) = VMM_KEY(xcth)
pnlkey1(5) = VMM_KEY(ycth)
pnlkey1(6) = VMM_KEY(zcth)
pnlkey1(7) = VMM_KEY(xct1)
pnlkey1(8) = VMM_KEY(yct1)
pnlkey1(9) = VMM_KEY(zct1)
pnlod = 9
*
if (step2_settls_L) then
pnlkey1(pnlod+1) = VMM_KEY(xt1)
pnlkey1(pnlod+2) = VMM_KEY(yt1)
pnlkey1(pnlod+3) = VMM_KEY(zt1)
pnlod = pnlod+3
endif
*
pnlkey1(1+pnlod) = VMM_KEY(xthm)
pnlkey1(2+pnlod) = VMM_KEY(ythm)
pnlkey1(3+pnlod) = VMM_KEY(zthm)
pnlkey1(4+pnlod) = VMM_KEY(xcthm)
pnlkey1(5+pnlod) = VMM_KEY(ycthm)
pnlkey1(6+pnlod) = VMM_KEY(zcthm)
pnlkey1(7+pnlod) = VMM_KEY(xct1m)
pnlkey1(8+pnlod) = VMM_KEY(yct1m)
pnlkey1(9+pnlod) = VMM_KEY(zct1m)
pnlod = 9+pnlod
*
if (step2_settls_L) then
pnlkey1(pnlod+1) = VMM_KEY(xt1m)
pnlkey1(pnlod+2) = VMM_KEY(yt1m)
pnlkey1(pnlod+3) = VMM_KEY(zt1m)
pnlod = pnlod+3
endif
*
pnerr = vmmlod(pnlkey1,pnlod)
*
pnerr = VMM_GET_VAR(xth)
pnerr = VMM_GET_VAR(yth)
pnerr = VMM_GET_VAR(zth)
pnerr = VMM_GET_VAR(xcth)
pnerr = VMM_GET_VAR(ycth)
pnerr = VMM_GET_VAR(zcth)
pnerr = VMM_GET_VAR(xct1)
pnerr = VMM_GET_VAR(yct1)
pnerr = VMM_GET_VAR(zct1)
if (step2_settls_L) then
pnerr = VMM_GET_VAR(xt1)
pnerr = VMM_GET_VAR(yt1)
pnerr = VMM_GET_VAR(zt1)
endif
*
pnerr = VMM_GET_VAR(xthm)
pnerr = VMM_GET_VAR(ythm)
pnerr = VMM_GET_VAR(zthm)
pnerr = VMM_GET_VAR(xcthm)
pnerr = VMM_GET_VAR(ycthm)
pnerr = VMM_GET_VAR(zcthm)
pnerr = VMM_GET_VAR(xct1m)
pnerr = VMM_GET_VAR(yct1m)
pnerr = VMM_GET_VAR(zct1m)
if (step2_settls_L) then
pnerr = VMM_GET_VAR(xt1m)
pnerr = VMM_GET_VAR(yt1m)
pnerr = VMM_GET_VAR(zt1m)
endif
*
i0=1
in=l_ni
j0=1
jn=l_nj
if (G_lam) then
if (l_west) i0=pil_w
if (l_east) in=l_niu - pil_e + 2
if (l_south) j0=pil_s
if (l_north) jn=l_njv - pil_n + 2
endif
************************************************************************
do it=1,F_it
************************************************************************
doh_L = .false.
if (it .eq. 1) doh_L = .true.
*
do n = 1,nijk
*
* TRAJ
* ----
wrk_ythm(n)= ythm(n)
*
* TLM
* ---
wrk_yth(n) = yth(n)
*
enddo
*
if (G_lam) then
*
call adw_cliptraj_tl ( xth, wrk_yth, xthm, wrk_ythm, i0, in, j0, jn, 'IN1_TL')
*
else
*
call adw_exch_1_tl ( wrkx1, wrky1, wrkz1, wrkc1,
$ xth, wrk_yth, zth,
$ wrkx1m,wrky1m,wrkz1m,wrkc1m,
$ xthm,wrk_ythm,zthm)
*
allocate(capx2m(max(1,Adw_fro_a)),
% capy2m(max(1,Adw_fro_a)),
% capz2m(max(1,Adw_fro_a)),
% xpos2m(max(1,Adw_fro_a)),
% ypos2m(max(1,Adw_fro_a)),
% zpos2m(max(1,Adw_fro_a)),
% n2m(max(1,Adw_fro_a)))
*
allocate(capx2(max(1,Adw_fro_a)),
% capy2(max(1,Adw_fro_a)),
% capz2(max(1,Adw_fro_a)),
% xpos2(max(1,Adw_fro_a)),
% ypos2(max(1,Adw_fro_a)),
% zpos2(max(1,Adw_fro_a)),
% n2(max(1,Adw_fro_a)))
*
call adw_exch_2_tl ( xpos2, ypos2, zpos2,
% wrkx1, wrky1, wrkz1,
% xpos2m,ypos2m,zpos2m,
% wrkx1m,wrky1m,wrkz1m,
% Adw_fro_n, Adw_fro_s, Adw_fro_a,
% Adw_for_n, Adw_for_s, Adw_for_a, 3)
*
endif
*
Adw_hor_L = doh_L
Adw_ver_L = .true.
call adw_trilin_turbo_tl (wrkx1, F_u, 1.0,xth, wrk_yth, zth,
% wrkx1m,F_um, xthm,wrk_ythm,zthm,
c % Adw_Fn_I,capx1m,capy1m,capz1m,
% capx1m,capy1m,capz1m,Adw_Fn_I,
% nijk,i0,in,j0,jn,l_nk)
*
*
call adw_trilin_turbo_tl (wrky1, F_v, 1.0,xth, wrk_yth, zth,
% wrky1m,F_vm, xthm,wrk_ythm,zthm,
c % Adw_Fn_I,capx1m,capy1m,capz1m,
% capx1m,capy1m,capz1m,Adw_Fn_I,
% nijk,i0,in,j0,jn,l_nk)
*
if (.not.G_lam) then
*
if ( Adw_fro_a .gt. 0 ) then
*
if ( Adw_ckbd_L ) call adw_ckbd ( ypos2m )
*
call adw_setint_tl ( n2, capx2, dummy,dummy, capy2, dummy,
% dummy,capz2, dummy,xpos2, ypos2, zpos2,
% n2m, capx2m,dummy,dummy, capy2m,dummy,
% dummy,capz2m,dummy,xpos2m,ypos2m,zpos2m,
% .true.,.true.,.true.,Adw_fro_a,1,Adw_fro_a,1,1,1)
*
call adw_trilin_tl ( xpos2, F_u, 1.0, n2, capx2, capy2, capz2,
% xpos2m,F_um, n2m,capx2m,capy2m,capz2m,
% Adw_fro_a,1,Adw_fro_a,1,1,1)
*
call adw_trilin_tl ( ypos2, F_v, 1.0, n2, capx2, capy2, capz2,
% ypos2m,F_vm, n2m,capx2m,capy2m,capz2m,
% Adw_fro_a,1,Adw_fro_a,1,1,1)
*
endif
*
call adw_exch_2_tl ( wrkz1, wrk_yth, dummy,
% xpos2, ypos2, dummy,
% wrkz1m,wrk_ythm,dummy,
% xpos2m,ypos2m, dummy,
% Adw_for_n, Adw_for_s, Adw_for_a,
% Adw_fro_n, Adw_fro_s, Adw_fro_a, 2)
*
if ( Adw_for_a .gt. 0 )
% call adw_exch_3_tl ( wrkx1, wrky1, wrkz1, wrk_yth, wrkc1,
% wrkx1m,wrky1m,wrkz1m,wrk_ythm,wrkc1m, 2)
*
deallocate(capx2m,capy2m,capz2m,xpos2m,ypos2m,zpos2m,n2m)
deallocate(capx2, capy2, capz2, xpos2, ypos2, zpos2, n2)
*
endif
*
************************************************************************
call adw_trajsp_tl ( xth, yth, xcth, ycth, zcth, wrkx1, wrky1,
% xthm, ythm, xcthm, ycthm, zcthm, wrkx1m, wrky1m, dth,
% i0,in,j0,jn)
************************************************************************
*
do n = 1,nijk
*
* TRAJ
* ----
wrk_ythm(n)= ythm(n)
*
* TLM
* ---
wrk_yth(n) = yth(n)
*
enddo
*
if (G_lam) then
*
call adw_cliptraj_tl ( xth, wrk_yth, xthm, wrk_ythm, i0, in, j0, jn, 'IN2_TL')
*
else
*
call adw_exch_1_tl ( wrkx1, wrky1, wrkz1, wrkc1,
$ xth, wrk_yth, zth,
$ wrkx1m,wrky1m,wrkz1m,wrkc1m,
$ xthm,wrk_ythm,zthm)
*
allocate(capx2m(max(1,Adw_fro_a)),
% capy2m(max(1,Adw_fro_a)),
% capz2m(max(1,Adw_fro_a)),
% xpos2m(max(1,Adw_fro_a)),
% ypos2m(max(1,Adw_fro_a)),
% zpos2m(max(1,Adw_fro_a)),
% n2m(max(1,Adw_fro_a)))
*
allocate(capx2(max(1,Adw_fro_a)),
% capy2(max(1,Adw_fro_a)),
% capz2(max(1,Adw_fro_a)),
% xpos2(max(1,Adw_fro_a)),
% ypos2(max(1,Adw_fro_a)),
% zpos2(max(1,Adw_fro_a)),
% n2(max(1,Adw_fro_a)))
*
call adw_exch_2_tl ( xpos2, ypos2, zpos2,
% wrkx1, wrky1, wrkz1,
% xpos2m,ypos2m,zpos2m,
% wrkx1m,wrky1m,wrkz1m,
% Adw_fro_n, Adw_fro_s, Adw_fro_a,
% Adw_for_n, Adw_for_s, Adw_for_a, 3)
*
endif
*
Adw_hor_L = .true.
Adw_ver_L = .false.
call adw_trilin_turbo_tl (wrkx1, F_w, -dth,xth, wrk_yth, zth,
% wrkx1m,F_wm, xthm,wrk_ythm,zthm,
c % Adw_Fn_I,capx1m,capy1m,capz1m,
% capx1m,capy1m,capz1m,Adw_Fn_I,
% nijk,i0,in,j0,jn,l_nk)
*
if (.not.G_lam) then
*
if ( Adw_fro_a .gt. 0 ) then
*
if ( Adw_ckbd_L ) call adw_ckbd ( ypos2m )
*
call adw_setint_tl ( n2, capx2, dummy,dummy, capy2, dummy,
% dummy,capz2, dummy,xpos2, ypos2, zpos2,
% n2m, capx2m,dummy,dummy, capy2m,dummy,
% dummy,capz2m,dummy,xpos2m,ypos2m,zpos2m,
% .true.,.true.,.true.,Adw_fro_a,1,Adw_fro_a,1,1,1)
*
call adw_trilin_tl ( xpos2, F_w, -dth,n2, capx2, capy2, capz2,
% xpos2m,F_wm, n2m,capx2m,capy2m,capz2m,
% Adw_fro_a,1,Adw_fro_a,1,1,1)
endif
*
call adw_exch_2_tl ( wrkz1, dummy, dummy,
% xpos2, dummy, dummy,
% wrkz1m,dummy, dummy,
% xpos2m,dummy, dummy,
% Adw_for_n, Adw_for_s, Adw_for_a,
% Adw_fro_n, Adw_fro_s, Adw_fro_a, 1)
*
if ( Adw_for_a .gt. 0 )
% call adw_exch_3_tl ( wrkx1, dummy,wrkz1, dummy,wrkc1,
% wrkx1m,dummy,wrkz1m,dummy,wrkc1m, 1)
*
deallocate(capx2m,capy2m,capz2m,xpos2m,ypos2m,zpos2m,n2m)
deallocate(capx2, capy2, capz2, xpos2, ypos2, zpos2, n2)
*
endif
*
************************************************************************
!$omp parallel private(n)
!$omp do
do k = 2,l_nk-1
do j = j0,jn
do i = i0,in
n = (k-1)*nij + ((j-1)*l_ni) + i
*
* TRAJECTORY
* ----------
zthm(n) = Geomg_z_8(k) + TWO_8*wrkx1m(n)
*
* TLM
* ---
zth(n) = TWO_8*wrkx1(n)
*
* TRAJECTORY and TLM
* ------------------
* The following min statement is expanded as two IF blocks:
* zthm(n) = min( pdm_8*Geomg_z_8(l_nk), max( 1.0d0*zthm(n), pdp_8*Geomg_z_8(1) )
*
if (1.0d0*zthm(n).lt.pdp_8*Geomg_z_8(1)) then
zthm(n) = pdp_8*Geomg_z_8(1)
zth (n) = 0.
elseif (1.0d0*zthm(n).gt.pdm_8*Geomg_z_8(l_nk)) then
zthm(n) = pdm_8*Geomg_z_8(l_nk)
zth (n) = 0.
endif
*
* TRAJECTORY and TLM
* ------------------
zthm(n) = HALF_8*(zthm(n) + Geomg_z_8(k))
zth (n) = HALF_8*(zth (n) )
*
enddo
enddo
enddo
!$omp enddo
!$omp end parallel
************************************************************************
enddo ! end of iterations loop
************************************************************************
call adw_trajex_tl (F_u, F_v, xct1, yct1, zct1, xcth, ycth, zcth,
% F_um,F_vm,xct1m,yct1m,zct1m,xcthm,ycthm,zcthm,
% i0,in,j0,jn)
*
* Keep xt1,yt1,zt1 positions (used in adw_main_2_int_settls_sw at next timesteps)
* -------------------------------------------------------------------------------
if (step2_settls_L) then
*
do i=1,l_ni*l_nj*l_nk
*
* TRAJ
* ----
xt1m(i) = F_um(i)
yt1m(i) = F_vm(i)
*
* TLM
* ---
xt1(i) = F_u(i)
yt1(i) = F_v(i)
*
enddo
*
k = 1
do j = 1, l_nj
do i = 1, l_ni
ijk=(k-1)*nij+(j-1)*l_ni+i
*
* TRAJ
* ----
zt1m(ijk) = Geomg_z_8(k)
*
* TLM
* ---
zt1(ijk) = 0.
*
enddo
enddo
*
k = l_nk
do j = 1, l_nj
do i = 1, l_ni
ijk=(k-1)*nij+(j-1)*l_ni+i
*
* TRAJ
* ----
zt1m(ijk) = Geomg_z_8(k)
*
* TLM
* ---
zt1(ijk) = 0.
*
enddo
enddo
*
* Recover TRAJ ZT1 mid-calculation (NOT REQUESTED)
* ------------------------------------------------
*
!$omp parallel private(n)
!$omp do
do k = 2,l_nk-1
do j = j0,jn
do i = i0,in
n = (k-1)*nij + ((j-1)*l_ni) + i
*
* TRAJ
* ----
zt1m(n) = Geomg_z_8(k) + TWO_8*wrkx1m(n)
*
* TLM
* ---
zt1 (n) = TWO_8*wrkx1(n)
*
* TRAJ and TLM
* ------------
if (1.0d0*zt1m(n).lt.pdp_8*Geomg_z_8(1)) then
zt1m(n) = pdp_8*Geomg_z_8(1)
zt1 (n) = 0.
elseif (1.0d0*zt1m(n).gt.pdm_8*Geomg_z_8(l_nk)) then
zt1m(n) = pdm_8*Geomg_z_8(l_nk)
zt1 (n) = 0.
endif
*
enddo
enddo
enddo
!$omp enddo
!$omp end parallel
*
endif
*
!$omp parallel private(n)
!$omp do
do j = j0,jn
do i = i0,in
n = ((j-1)*l_ni) + i !for k=1
*
* TRAJECTORY
* ----------
F_wm(n) = Geomg_z_8(1)
*
* TLM
* ---
F_w(n) = 0.
*
n = (l_nk-1)*nij+((j-1)*l_ni) + i !for k=l_nk
*
* TRAJECTORY
* ----------
F_wm(n) = Geomg_z_8(l_nk)
*
* TLM
* ---
F_w(n) = 0.
*
enddo
enddo
!$omp enddo
!$omp do
do k = 2,l_nk-1
do j = j0,jn
do i = i0,in
n=(k-1)*nij+((j-1)*l_ni) + i
*
* TRAJECTORY
* ----------
F_wm(n) = zthm(n) - Geomg_z_8(k)
F_wm(n) = Geomg_z_8(k) + 2.0 * F_wm(n)
*
* TLM
* ---
F_w(n) = zth(n)
F_w(n) = 2.0 * F_w(n)
*
enddo
enddo
enddo
!$omp enddo
!$omp end parallel
*
pnerr = vmmuld(-1,0)
*
1000 format(3X,'TLM of CALC UPSTREAM POSITIONS: (S/R ADW_MAIN_2_POS_TL)')
return
end