diff options
Diffstat (limited to 'src/Derivatives_4_3.F90')
| -rw-r--r-- | src/Derivatives_4_3.F90 | 233 |
1 files changed, 120 insertions, 113 deletions
diff --git a/src/Derivatives_4_3.F90 b/src/Derivatives_4_3.F90 index 1bc9734..dd88de7 100644 --- a/src/Derivatives_4_3.F90 +++ b/src/Derivatives_4_3.F90 @@ -2,7 +2,7 @@ #include "cctk_Functions.h" #include "cctk_Parameters.h" -subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) +subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, offset, delta, dvar ) use All_Coeffs_mod @@ -16,6 +16,7 @@ subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) CCTK_INT, intent(IN) :: dir CCTK_INT, intent(IN) :: bb(2) CCTK_INT, intent(IN) :: gsize + CCTK_INT, intent(IN) :: offset(2) CCTK_REAL, intent(IN) :: delta CCTK_REAL, dimension(ni,nj,nk), intent(OUT) :: dvar @@ -23,7 +24,7 @@ subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) CCTK_REAL, dimension(7,5), save :: q CCTK_REAL :: idel - CCTK_INT :: il, ir, jl, jr, kl, kr + CCTK_INT :: il, ir, jl, jr, kl, kr, ol, or logical, save :: first = .true. @@ -41,51 +42,53 @@ subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) if ( bb(1) == 0 ) then il = 1 + gsize else + ol = offset(1) !$omp parallel workshare - dvar(1,:,:) = ( q(1,1) * var(1,:,:) + q(2,1) * var(2,:,:) + & - q(3,1) * var(3,:,:) + q(4,1) * var(4,:,:) ) * idel - dvar(2,:,:) = ( q(1,2) * var(1,:,:) + q(2,2) * var(2,:,:) + & - q(3,2) * var(3,:,:) + q(4,2) * var(4,:,:) + & - q(5,2) * var(5,:,:) + q(6,2) * var(6,:,:) ) * idel - dvar(3,:,:) = ( q(1,3) * var(1,:,:) + q(2,3) * var(2,:,:) + & - q(3,3) * var(3,:,:) + q(4,3) * var(4,:,:) + & - q(5,3) * var(5,:,:) + q(6,3) * var(6,:,:) ) * idel - dvar(4,:,:) = ( q(1,4) * var(1,:,:) + q(2,4) * var(2,:,:) + & - q(3,4) * var(3,:,:) + q(4,4) * var(4,:,:) + & - q(5,4) * var(5,:,:) + q(6,4) * var(6,:,:) + & - q(7,4) * var(7,:,:) ) * idel - dvar(5,:,:) = ( q(1,5) * var(1,:,:) + q(2,5) * var(2,:,:) + & - q(3,5) * var(3,:,:) + q(4,5) * var(4,:,:) + & - q(5,5) * var(5,:,:) + q(6,5) * var(6,:,:) + & - q(7,5) * var(7,:,:) ) * idel + dvar(1+ol,:,:) = ( q(1,1) * var(1+ol,:,:) + q(2,1) * var(2+ol,:,:) + & + q(3,1) * var(3+ol,:,:) + q(4,1) * var(4+ol,:,:) ) * idel + dvar(2+ol,:,:) = ( q(1,2) * var(1+ol,:,:) + q(2,2) * var(2+ol,:,:) + & + q(3,2) * var(3+ol,:,:) + q(4,2) * var(4+ol,:,:) + & + q(5,2) * var(5+ol,:,:) + q(6,2) * var(6+ol,:,:) ) * idel + dvar(3+ol,:,:) = ( q(1,3) * var(1+ol,:,:) + q(2,3) * var(2+ol,:,:) + & + q(3,3) * var(3+ol,:,:) + q(4,3) * var(4+ol,:,:) + & + q(5,3) * var(5+ol,:,:) + q(6,3) * var(6+ol,:,:) ) * idel + dvar(4+ol,:,:) = ( q(1,4) * var(1+ol,:,:) + q(2,4) * var(2+ol,:,:) + & + q(3,4) * var(3+ol,:,:) + q(4,4) * var(4+ol,:,:) + & + q(5,4) * var(5+ol,:,:) + q(6,4) * var(6+ol,:,:) + & + q(7,4) * var(7+ol,:,:) ) * idel + dvar(5+ol,:,:) = ( q(1,5) * var(1+ol,:,:) + q(2,5) * var(2+ol,:,:) + & + q(3,5) * var(3+ol,:,:) + q(4,5) * var(4+ol,:,:) + & + q(5,5) * var(5+ol,:,:) + q(6,5) * var(6+ol,:,:) + & + q(7,5) * var(7+ol,:,:) ) * idel !$omp end parallel workshare - il = 6 + il = 6 + ol end if if ( bb(2) == 0 ) then ir = ni - gsize else + or = ni - offset(2) !$omp parallel workshare - dvar(ni-4,:,:) = - ( q(1,5) * var(ni,:,:) + q(2,5) * var(ni-1,:,:) + & - q(3,5) * var(ni-2,:,:) + q(4,5) * var(ni-3,:,:) + & - q(5,5) * var(ni-4,:,:) + q(6,5) * var(ni-5,:,:) + & - q(7,5) * var(ni-6,:,:) ) * idel - dvar(ni-3,:,:) = - ( q(1,4) * var(ni,:,:) + q(2,4) * var(ni-1,:,:) + & - q(3,4) * var(ni-2,:,:) + q(4,4) * var(ni-3,:,:) + & - q(5,4) * var(ni-4,:,:) + q(6,4) * var(ni-5,:,:) + & - q(7,4) * var(ni-6,:,:) ) * idel - dvar(ni-2,:,:) = - ( q(1,3) * var(ni,:,:) + q(2,3) * var(ni-1,:,:) + & - q(3,3) * var(ni-2,:,:) + q(4,3) * var(ni-3,:,:) + & - q(5,3) * var(ni-4,:,:) + & - q(6,3) * var(ni-5,:,:) ) * idel - dvar(ni-1,:,:) = - ( q(1,2) * var(ni,:,:) + q(2,2) * var(ni-1,:,:) + & - q(3,2) * var(ni-2,:,:) + q(4,2) * var(ni-3,:,:) + & - q(5,2) * var(ni-4,:,:) + & - q(6,2) * var(ni-5,:,:) ) * idel - dvar(ni,:,:) = - ( q(1,1) * var(ni,:,:) + q(2,1) * var(ni-1,:,:) + & - q(3,1) * var(ni-2,:,:) + & - q(4,1) * var(ni-3,:,:) ) * idel + dvar(or-4,:,:) = - ( q(1,5) * var(or,:,:) + q(2,5) * var(or-1,:,:) + & + q(3,5) * var(or-2,:,:) + q(4,5) * var(or-3,:,:) + & + q(5,5) * var(or-4,:,:) + q(6,5) * var(or-5,:,:) + & + q(7,5) * var(or-6,:,:) ) * idel + dvar(or-3,:,:) = - ( q(1,4) * var(or,:,:) + q(2,4) * var(or-1,:,:) + & + q(3,4) * var(or-2,:,:) + q(4,4) * var(or-3,:,:) + & + q(5,4) * var(or-4,:,:) + q(6,4) * var(or-5,:,:) + & + q(7,4) * var(or-6,:,:) ) * idel + dvar(or-2,:,:) = - ( q(1,3) * var(or,:,:) + q(2,3) * var(or-1,:,:) + & + q(3,3) * var(or-2,:,:) + q(4,3) * var(or-3,:,:) + & + q(5,3) * var(or-4,:,:) + & + q(6,3) * var(or-5,:,:) ) * idel + dvar(or-1,:,:) = - ( q(1,2) * var(or,:,:) + q(2,2) * var(or-1,:,:) + & + q(3,2) * var(or-2,:,:) + q(4,2) * var(or-3,:,:) + & + q(5,2) * var(or-4,:,:) + & + q(6,2) * var(or-5,:,:) ) * idel + dvar(or,:,:) = - ( q(1,1) * var(or,:,:) + q(2,1) * var(or-1,:,:) + & + q(3,1) * var(or-2,:,:) + & + q(4,1) * var(or-3,:,:) ) * idel !$omp end parallel workshare - ir = ni - 5 + ir = or - 5 end if if (il > ir+1) call CCTK_WARN (0, "domain too small") !$omp parallel workshare @@ -101,51 +104,53 @@ subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) if ( bb(1) == 0 ) then jl = 1 + gsize else + ol = offset(1) !$omp parallel workshare - dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & - q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) ) * idel - dvar(:,2,:) = ( q(1,2) * var(:,1,:) + q(2,2) * var(:,2,:) + & - q(3,2) * var(:,3,:) + q(4,2) * var(:,4,:) + & - q(5,2) * var(:,5,:) + q(6,2) * var(:,6,:) ) * idel - dvar(:,3,:) = ( q(1,3) * var(:,1,:) + q(2,3) * var(:,2,:) + & - q(3,3) * var(:,3,:) + q(4,3) * var(:,4,:) + & - q(5,3) * var(:,5,:) + q(6,3) * var(:,6,:) ) * idel - dvar(:,4,:) = ( q(1,4) * var(:,1,:) + q(2,4) * var(:,2,:) + & - q(3,4) * var(:,3,:) + q(4,4) * var(:,4,:) + & - q(5,4) * var(:,5,:) + q(6,4) * var(:,6,:) + & - q(7,4) * var(:,7,:) ) * idel - dvar(:,5,:) = ( q(1,5) * var(:,1,:) + q(2,5) * var(:,2,:) + & - q(3,5) * var(:,3,:) + q(4,5) * var(:,4,:) + & - q(5,5) * var(:,5,:) + q(6,5) * var(:,6,:) + & - q(7,5) * var(:,7,:) ) * idel + dvar(:,1+ol,:) = ( q(1,1) * var(:,1+ol,:) + q(2,1) * var(:,2+ol,:) + & + q(3,1) * var(:,3+ol,:) + q(4,1) * var(:,4+ol,:) ) * idel + dvar(:,2+ol,:) = ( q(1,2) * var(:,1+ol,:) + q(2,2) * var(:,2+ol,:) + & + q(3,2) * var(:,3+ol,:) + q(4,2) * var(:,4+ol,:) + & + q(5,2) * var(:,5+ol,:) + q(6,2) * var(:,6+ol,:) ) * idel + dvar(:,3+ol,:) = ( q(1,3) * var(:,1+ol,:) + q(2,3) * var(:,2+ol,:) + & + q(3,3) * var(:,3+ol,:) + q(4,3) * var(:,4+ol,:) + & + q(5,3) * var(:,5+ol,:) + q(6,3) * var(:,6+ol,:) ) * idel + dvar(:,4+ol,:) = ( q(1,4) * var(:,1+ol,:) + q(2,4) * var(:,2+ol,:) + & + q(3,4) * var(:,3+ol,:) + q(4,4) * var(:,4+ol,:) + & + q(5,4) * var(:,5+ol,:) + q(6,4) * var(:,6+ol,:) + & + q(7,4) * var(:,7+ol,:) ) * idel + dvar(:,5+ol,:) = ( q(1,5) * var(:,1+ol,:) + q(2,5) * var(:,2+ol,:) + & + q(3,5) * var(:,3+ol,:) + q(4,5) * var(:,4+ol,:) + & + q(5,5) * var(:,5+ol,:) + q(6,5) * var(:,6+ol,:) + & + q(7,5) * var(:,7+ol,:) ) * idel !$omp end parallel workshare - jl = 6 + jl = 6 + ol end if if ( bb(2) == 0 ) then jr = nj - gsize else + or = nj - offset(2) !$omp parallel workshare - dvar(:,nj-4,:) = - ( q(1,5) * var(:,nj,:) + q(2,5) * var(:,nj-1,:) + & - q(3,5) * var(:,nj-2,:) + q(4,5) * var(:,nj-3,:) + & - q(5,5) * var(:,nj-4,:) + q(6,5) * var(:,nj-5,:) + & - q(7,5) * var(:,nj-6,:) ) * idel - dvar(:,nj-3,:) = - ( q(1,4) * var(:,nj,:) + q(2,4) * var(:,nj-1,:) + & - q(3,4) * var(:,nj-2,:) + q(4,4) * var(:,nj-3,:) + & - q(5,4) * var(:,nj-4,:) + q(6,4) * var(:,nj-5,:) + & - q(7,4) * var(:,nj-6,:) ) * idel - dvar(:,nj-2,:) = - ( q(1,3) * var(:,nj,:) + q(2,3) * var(:,nj-1,:) + & - q(3,3) * var(:,nj-2,:) + q(4,3) * var(:,nj-3,:) + & - q(5,3) * var(:,nj-4,:) + & - q(6,3) * var(:,nj-5,:) ) * idel - dvar(:,nj-1,:) = - ( q(1,2) * var(:,nj,:) + q(2,2) * var(:,nj-1,:) + & - q(3,2) * var(:,nj-2,:) + q(4,2) * var(:,nj-3,:) + & - q(5,2) * var(:,nj-4,:) + & - q(6,2) * var(:,nj-5,:) ) * idel - dvar(:,nj,:) = - ( q(1,1) * var(:,nj,:) + q(2,1) * var(:,nj-1,:) + & - q(3,1) * var(:,nj-2,:) + & - q(4,1) * var(:,nj-3,:) ) * idel + dvar(:,or-4,:) = - ( q(1,5) * var(:,or,:) + q(2,5) * var(:,or-1,:) + & + q(3,5) * var(:,or-2,:) + q(4,5) * var(:,or-3,:) + & + q(5,5) * var(:,or-4,:) + q(6,5) * var(:,or-5,:) + & + q(7,5) * var(:,or-6,:) ) * idel + dvar(:,or-3,:) = - ( q(1,4) * var(:,or,:) + q(2,4) * var(:,or-1,:) + & + q(3,4) * var(:,or-2,:) + q(4,4) * var(:,or-3,:) + & + q(5,4) * var(:,or-4,:) + q(6,4) * var(:,or-5,:) + & + q(7,4) * var(:,or-6,:) ) * idel + dvar(:,or-2,:) = - ( q(1,3) * var(:,or,:) + q(2,3) * var(:,or-1,:) + & + q(3,3) * var(:,or-2,:) + q(4,3) * var(:,or-3,:) + & + q(5,3) * var(:,or-4,:) + & + q(6,3) * var(:,or-5,:) ) * idel + dvar(:,or-1,:) = - ( q(1,2) * var(:,or,:) + q(2,2) * var(:,or-1,:) + & + q(3,2) * var(:,or-2,:) + q(4,2) * var(:,or-3,:) + & + q(5,2) * var(:,or-4,:) + & + q(6,2) * var(:,or-5,:) ) * idel + dvar(:,or,:) = - ( q(1,1) * var(:,or,:) + q(2,1) * var(:,or-1,:) + & + q(3,1) * var(:,or-2,:) + & + q(4,1) * var(:,or-3,:) ) * idel !$omp end parallel workshare - jr = nj - 5 + jr = or - 5 end if if (jl > jr+1) call CCTK_WARN (0, "domain too small") !$omp parallel workshare @@ -162,51 +167,53 @@ subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) if ( bb(1) == 0 ) then kl = 1 + gsize else + ol = offset(1) !$omp parallel workshare - dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & - q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) ) * idel - dvar(:,:,2) = ( q(1,2) * var(:,:,1) + q(2,2) * var(:,:,2) + & - q(3,2) * var(:,:,3) + q(4,2) * var(:,:,4) + & - q(5,2) * var(:,:,5) + q(6,2) * var(:,:,6) ) * idel - dvar(:,:,3) = ( q(1,3) * var(:,:,1) + q(2,3) * var(:,:,2) + & - q(3,3) * var(:,:,3) + q(4,3) * var(:,:,4) + & - q(5,3) * var(:,:,5) + q(6,3) * var(:,:,6) ) * idel - dvar(:,:,4) = ( q(1,4) * var(:,:,1) + q(2,4) * var(:,:,2) + & - q(3,4) * var(:,:,3) + q(4,4) * var(:,:,4) + & - q(5,4) * var(:,:,5) + q(6,4) * var(:,:,6) + & - q(7,4) * var(:,:,7) ) * idel - dvar(:,:,5) = ( q(1,5) * var(:,:,1) + q(2,5) * var(:,:,2) + & - q(3,5) * var(:,:,3) + q(4,5) * var(:,:,4) + & - q(5,5) * var(:,:,5) + q(6,5) * var(:,:,6) + & - q(7,5) * var(:,:,7) ) * idel + dvar(:,:,1+ol) = ( q(1,1) * var(:,:,1+ol) + q(2,1) * var(:,:,2+ol) + & + q(3,1) * var(:,:,3+ol) + q(4,1) * var(:,:,4+ol) ) * idel + dvar(:,:,2+ol) = ( q(1,2) * var(:,:,1+ol) + q(2,2) * var(:,:,2+ol) + & + q(3,2) * var(:,:,3+ol) + q(4,2) * var(:,:,4+ol) + & + q(5,2) * var(:,:,5+ol) + q(6,2) * var(:,:,6+ol) ) * idel + dvar(:,:,3+ol) = ( q(1,3) * var(:,:,1+ol) + q(2,3) * var(:,:,2+ol) + & + q(3,3) * var(:,:,3+ol) + q(4,3) * var(:,:,4+ol) + & + q(5,3) * var(:,:,5+ol) + q(6,3) * var(:,:,6+ol) ) * idel + dvar(:,:,4+ol) = ( q(1,4) * var(:,:,1+ol) + q(2,4) * var(:,:,2+ol) + & + q(3,4) * var(:,:,3+ol) + q(4,4) * var(:,:,4+ol) + & + q(5,4) * var(:,:,5+ol) + q(6,4) * var(:,:,6+ol) + & + q(7,4) * var(:,:,7+ol) ) * idel + dvar(:,:,5+ol) = ( q(1,5) * var(:,:,1+ol) + q(2,5) * var(:,:,2+ol) + & + q(3,5) * var(:,:,3+ol) + q(4,5) * var(:,:,4+ol) + & + q(5,5) * var(:,:,5+ol) + q(6,5) * var(:,:,6+ol) + & + q(7,5) * var(:,:,7+ol) ) * idel !$omp end parallel workshare - kl = 6 + kl = 6 + ol end if if ( bb(2) == 0 ) then kr = nk - gsize else + or = nk - offset(2) !$omp parallel workshare - dvar(:,:,nk-4) = - ( q(1,5) * var(:,:,nk) + q(2,5) * var(:,:,nk-1) + & - q(3,5) * var(:,:,nk-2) + q(4,5) * var(:,:,nk-3) + & - q(5,5) * var(:,:,nk-4) + q(6,5) * var(:,:,nk-5) + & - q(7,5) * var(:,:,nk-6) ) * idel - dvar(:,:,nk-3) = - ( q(1,4) * var(:,:,nk) + q(2,4) * var(:,:,nk-1) + & - q(3,4) * var(:,:,nk-2) + q(4,4) * var(:,:,nk-3) + & - q(5,4) * var(:,:,nk-4) + q(6,4) * var(:,:,nk-5) + & - q(7,4) * var(:,:,nk-6) ) * idel - dvar(:,:,nk-2) = - ( q(1,3) * var(:,:,nk) + q(2,3) * var(:,:,nk-1) + & - q(3,3) * var(:,:,nk-2) + q(4,3) * var(:,:,nk-3) + & - q(5,3) * var(:,:,nk-4) + & - q(6,3) * var(:,:,nk-5) ) * idel - dvar(:,:,nk-1) = - ( q(1,2) * var(:,:,nk) + q(2,2) * var(:,:,nk-1) + & - q(3,2) * var(:,:,nk-2) + q(4,2) * var(:,:,nk-3) + & - q(5,2) * var(:,:,nk-4) + & - q(6,2) * var(:,:,nk-5) ) * idel - dvar(:,:,nk) = - ( q(1,1) * var(:,:,nk) + q(2,1) * var(:,:,nk-1) + & - q(3,1) * var(:,:,nk-2) + & - q(4,1) * var(:,:,nk-3) ) * idel + dvar(:,:,or-4) = - ( q(1,5) * var(:,:,or) + q(2,5) * var(:,:,or-1) + & + q(3,5) * var(:,:,or-2) + q(4,5) * var(:,:,or-3) + & + q(5,5) * var(:,:,or-4) + q(6,5) * var(:,:,or-5) + & + q(7,5) * var(:,:,or-6) ) * idel + dvar(:,:,or-3) = - ( q(1,4) * var(:,:,or) + q(2,4) * var(:,:,or-1) + & + q(3,4) * var(:,:,or-2) + q(4,4) * var(:,:,or-3) + & + q(5,4) * var(:,:,or-4) + q(6,4) * var(:,:,or-5) + & + q(7,4) * var(:,:,or-6) ) * idel + dvar(:,:,or-2) = - ( q(1,3) * var(:,:,or) + q(2,3) * var(:,:,or-1) + & + q(3,3) * var(:,:,or-2) + q(4,3) * var(:,:,or-3) + & + q(5,3) * var(:,:,or-4) + & + q(6,3) * var(:,:,or-5) ) * idel + dvar(:,:,or-1) = - ( q(1,2) * var(:,:,or) + q(2,2) * var(:,:,or-1) + & + q(3,2) * var(:,:,or-2) + q(4,2) * var(:,:,or-3) + & + q(5,2) * var(:,:,or-4) + & + q(6,2) * var(:,:,or-5) ) * idel + dvar(:,:,or) = - ( q(1,1) * var(:,:,or) + q(2,1) * var(:,:,or-1) + & + q(3,1) * var(:,:,or-2) + & + q(4,1) * var(:,:,or-3) ) * idel !$omp end parallel workshare - kr = nk - 5 + kr = or - 5 end if if (kl > kr+1) call CCTK_WARN (0, "domain too small") !$omp parallel workshare |