diff options
Diffstat (limited to 'src/Dissipation_4_2_alt.F90')
-rw-r--r-- | src/Dissipation_4_2_alt.F90 | 219 |
1 files changed, 113 insertions, 106 deletions
diff --git a/src/Dissipation_4_2_alt.F90 b/src/Dissipation_4_2_alt.F90 index feef10b..577ed39 100644 --- a/src/Dissipation_4_2_alt.F90 +++ b/src/Dissipation_4_2_alt.F90 @@ -3,7 +3,7 @@ #include "cctk.h" #include "cctk_Parameters.h" -subroutine dissipation_4_2_alt (var, ni, nj, nk, bb, gsize, delta, epsilon, rhs) +subroutine dissipation_4_2_alt (var, ni, nj, nk, bb, gsize, offset, delta, epsilon, rhs) implicit none @@ -14,6 +14,7 @@ subroutine dissipation_4_2_alt (var, ni, nj, nk, bb, gsize, delta, epsilon, rhs) CCTK_REAL, dimension(ni,nj,nk), intent(inout) :: rhs CCTK_INT, dimension(6), intent(in) :: bb CCTK_INT, dimension(3), intent(in) :: gsize + CCTK_INT, dimension(6), intent(in) :: offset CCTK_REAL, dimension(3), intent(in) :: delta CCTK_REAL, intent(in) :: epsilon @@ -22,7 +23,7 @@ subroutine dissipation_4_2_alt (var, ni, nj, nk, bb, gsize, delta, epsilon, rhs) CCTK_REAL, dimension(7,4) :: a CCTK_REAL :: idel - CCTK_INT :: il, ir, jl, jr, kl, kr + CCTK_INT :: il, ir, jl, jr, kl, kr, ol, or call set_coeff ( a ) @@ -31,46 +32,48 @@ subroutine dissipation_4_2_alt (var, ni, nj, nk, bb, gsize, delta, epsilon, rhs) if ( bb(1) == 0 ) then il = 1 + gsize(1) else - rhs(1,:,:) = rhs(1,:,:) + & - ( a(1,1) * var(1,:,:) + a(2,1) * var(2,:,:) + & - a(3,1) * var(3,:,:) + a(4,1) * var(4,:,:) ) * idel - rhs(2,:,:) = rhs(2,:,:) + & - ( a(1,2) * var(1,:,:) + a(2,2) * var(2,:,:) + & - a(3,2) * var(3,:,:) + a(4,2) * var(4,:,:) + & - a(5,2) * var(5,:,:) ) * idel - rhs(3,:,:) = rhs(3,:,:) + & - ( a(1,3) * var(1,:,:) + a(2,3) * var(2,:,:) + & - a(3,3) * var(3,:,:) + a(4,3) * var(4,:,:) + & - a(5,3) * var(5,:,:) + a(6,3) * var(6,:,:) ) * idel - rhs(4,:,:) = rhs(4,:,:) + & - ( a(1,4) * var(1,:,:) + a(2,4) * var(2,:,:) + & - a(3,4) * var(3,:,:) + a(4,4) * var(4,:,:) + & - a(5,4) * var(5,:,:) + a(6,4) * var(6,:,:) + & - a(7,4) * var(7,:,:) ) * idel - - il = 5 + ol = offset(1) + rhs(1+ol,:,:) = rhs(1+ol,:,:) + & + ( a(1,1) * var(1+ol,:,:) + a(2,1) * var(2+ol,:,:) + & + a(3,1) * var(3+ol,:,:) + a(4,1) * var(4+ol,:,:) ) * idel + rhs(2+ol,:,:) = rhs(2+ol,:,:) + & + ( a(1,2) * var(1+ol,:,:) + a(2,2) * var(2+ol,:,:) + & + a(3,2) * var(3+ol,:,:) + a(4,2) * var(4+ol,:,:) + & + a(5,2) * var(5+ol,:,:) ) * idel + rhs(3+ol,:,:) = rhs(3+ol,:,:) + & + ( a(1,3) * var(1+ol,:,:) + a(2,3) * var(2+ol,:,:) + & + a(3,3) * var(3+ol,:,:) + a(4,3) * var(4+ol,:,:) + & + a(5,3) * var(5+ol,:,:) + a(6,3) * var(6+ol,:,:) ) * idel + rhs(4+ol,:,:) = rhs(4+ol,:,:) + & + ( a(1,4) * var(1+ol,:,:) + a(2,4) * var(2+ol,:,:) + & + a(3,4) * var(3+ol,:,:) + a(4,4) * var(4+ol,:,:) + & + a(5,4) * var(5+ol,:,:) + a(6,4) * var(6+ol,:,:) + & + a(7,4) * var(7+ol,:,:) ) * idel + + il = 5 + ol end if if ( bb(2) == 0 ) then ir = ni - gsize(1) else - rhs(ni-3,:,:) = rhs(ni-3,:,:) + & - ( a(1,4) * var(ni,:,:) + a(2,4) * var(ni-1,:,:) + & - a(3,4) * var(ni-2,:,:) + a(4,4) * var(ni-3,:,:) + & - a(5,4) * var(ni-4,:,:) + a(6,4) * var(ni-5,:,:) + & - a(7,4) * var(ni-6,:,:) ) * idel - rhs(ni-2,:,:) = rhs(ni-2,:,:) + & - ( a(1,3) * var(ni,:,:) + a(2,3) * var(ni-1,:,:) + & - a(3,3) * var(ni-2,:,:) + a(4,3) * var(ni-3,:,:) + & - a(5,3) * var(ni-4,:,:) + a(6,3) * var(ni-5,:,:) ) * idel - rhs(ni-1,:,:) = rhs(ni-1,:,:) + & - ( a(1,2) * var(ni,:,:) + a(2,2) * var(ni-1,:,:) + & - a(3,2) * var(ni-2,:,:) + a(4,2) * var(ni-3,:,:) + & - a(5,2) * var(ni-4,:,:) ) * idel - rhs(ni,:,:) = rhs(ni,:,:) + & - ( a(1,1) * var(ni,:,:) + a(2,1) * var(ni-1,:,:) + & - a(3,1) * var(ni-2,:,:) + a(4,1) * var(ni-3,:,:) ) * idel - - ir = ni - 4 + or = ni - offset(2) + rhs(or-3,:,:) = rhs(or-3,:,:) + & + ( a(1,4) * var(or,:,:) + a(2,4) * var(or-1,:,:) + & + a(3,4) * var(or-2,:,:) + a(4,4) * var(or-3,:,:) + & + a(5,4) * var(or-4,:,:) + a(6,4) * var(or-5,:,:) + & + a(7,4) * var(or-6,:,:) ) * idel + rhs(or-2,:,:) = rhs(or-2,:,:) + & + ( a(1,3) * var(or,:,:) + a(2,3) * var(or-1,:,:) + & + a(3,3) * var(or-2,:,:) + a(4,3) * var(or-3,:,:) + & + a(5,3) * var(or-4,:,:) + a(6,3) * var(or-5,:,:) ) * idel + rhs(or-1,:,:) = rhs(or-1,:,:) + & + ( a(1,2) * var(or,:,:) + a(2,2) * var(or-1,:,:) + & + a(3,2) * var(or-2,:,:) + a(4,2) * var(or-3,:,:) + & + a(5,2) * var(or-4,:,:) ) * idel + rhs(or,:,:) = rhs(or,:,:) + & + ( a(1,1) * var(or,:,:) + a(2,1) * var(or-1,:,:) + & + a(3,1) * var(or-2,:,:) + a(4,1) * var(or-3,:,:) ) * idel + + ir = or - 4 end if rhs(il:ir,:,:) = rhs(il:ir,:,:) + & ( -20.0_wp * var(il:ir,:,:) + & @@ -89,46 +92,48 @@ subroutine dissipation_4_2_alt (var, ni, nj, nk, bb, gsize, delta, epsilon, rhs) if ( bb(3) == 0 ) then jl = 1 + gsize(2) else - rhs(:,1,:) = rhs(:,1,:) + & - ( a(1,1) * var(:,1,:) + a(2,1) * var(:,2,:) + & - a(3,1) * var(:,3,:) + a(4,1) * var(:,4,:) ) * idel - rhs(:,2,:) = rhs(:,2,:) + & - ( a(1,2) * var(:,1,:) + a(2,2) * var(:,2,:) + & - a(3,2) * var(:,3,:) + a(4,2) * var(:,4,:) + & - a(5,2) * var(:,5,:) ) * idel - rhs(:,3,:) = rhs(:,3,:) + & - ( a(1,3) * var(:,1,:) + a(2,3) * var(:,2,:) + & - a(3,3) * var(:,3,:) + a(4,3) * var(:,4,:) + & - a(5,3) * var(:,5,:) + a(6,3) * var(:,6,:) ) * idel - rhs(:,4,:) = rhs(:,4,:) + & - ( a(1,4) * var(:,1,:) + a(2,4) * var(:,2,:) + & - a(3,4) * var(:,3,:) + a(4,4) * var(:,4,:) + & - a(5,4) * var(:,5,:) + a(6,4) * var(:,6,:) + & - a(7,4) * var(:,7,:) ) * idel + ol = offset(3) + rhs(:,1+ol,:) = rhs(:,1+ol,:) + & + ( a(1,1) * var(:,1+ol,:) + a(2,1) * var(:,2+ol,:) + & + a(3,1) * var(:,3+ol,:) + a(4,1) * var(:,4+ol,:) ) * idel + rhs(:,2+ol,:) = rhs(:,2+ol,:) + & + ( a(1,2) * var(:,1+ol,:) + a(2,2) * var(:,2+ol,:) + & + a(3,2) * var(:,3+ol,:) + a(4,2) * var(:,4+ol,:) + & + a(5,2) * var(:,5+ol,:) ) * idel + rhs(:,3+ol,:) = rhs(:,3+ol,:) + & + ( a(1,3) * var(:,1+ol,:) + a(2,3) * var(:,2+ol,:) + & + a(3,3) * var(:,3+ol,:) + a(4,3) * var(:,4+ol,:) + & + a(5,3) * var(:,5+ol,:) + a(6,3) * var(:,6+ol,:) ) * idel + rhs(:,4+ol,:) = rhs(:,4+ol,:) + & + ( a(1,4) * var(:,1+ol,:) + a(2,4) * var(:,2+ol,:) + & + a(3,4) * var(:,3+ol,:) + a(4,4) * var(:,4+ol,:) + & + a(5,4) * var(:,5+ol,:) + a(6,4) * var(:,6+ol,:) + & + a(7,4) * var(:,7+ol,:) ) * idel - jl = 5 + jl = 5 + ol end if if ( bb(4) == 0 ) then jr = nj - gsize(2) else - rhs(:,nj-3,:) = rhs(:,nj-3,:) + & - ( a(1,4) * var(:,nj,:) + a(2,4) * var(:,nj-1,:) + & - a(3,4) * var(:,nj-2,:) + a(4,4) * var(:,nj-3,:) + & - a(5,4) * var(:,nj-4,:) + a(6,4) * var(:,nj-5,:) + & - a(7,4) * var(:,nj-6,:) ) * idel - rhs(:,nj-2,:) = rhs(:,nj-2,:) + & - ( a(1,3) * var(:,nj,:) + a(2,3) * var(:,nj-1,:) + & - a(3,3) * var(:,nj-2,:) + a(4,3) * var(:,nj-3,:) + & - a(5,3) * var(:,nj-4,:) + a(6,3) * var(:,nj-5,:) ) * idel - rhs(:,nj-1,:) = rhs(:,nj-1,:) + & - ( a(1,2) * var(:,nj,:) + a(2,2) * var(:,nj-1,:) + & - a(3,2) * var(:,nj-2,:) + a(4,2) * var(:,nj-3,:) + & - a(5,2) * var(:,nj-4,:) ) * idel - rhs(:,nj,:) = rhs(:,nj,:) + & - ( a(1,1) * var(:,nj,:) + a(2,1) * var(:,nj-1,:) + & - a(3,1) * var(:,nj-2,:) + a(4,1) * var(:,nj-3,:) ) * idel + or = nj - offset(4) + rhs(:,or-3,:) = rhs(:,or-3,:) + & + ( a(1,4) * var(:,or,:) + a(2,4) * var(:,or-1,:) + & + a(3,4) * var(:,or-2,:) + a(4,4) * var(:,or-3,:) + & + a(5,4) * var(:,or-4,:) + a(6,4) * var(:,or-5,:) + & + a(7,4) * var(:,or-6,:) ) * idel + rhs(:,or-2,:) = rhs(:,or-2,:) + & + ( a(1,3) * var(:,or,:) + a(2,3) * var(:,or-1,:) + & + a(3,3) * var(:,or-2,:) + a(4,3) * var(:,or-3,:) + & + a(5,3) * var(:,or-4,:) + a(6,3) * var(:,or-5,:) ) * idel + rhs(:,or-1,:) = rhs(:,or-1,:) + & + ( a(1,2) * var(:,or,:) + a(2,2) * var(:,or-1,:) + & + a(3,2) * var(:,or-2,:) + a(4,2) * var(:,or-3,:) + & + a(5,2) * var(:,or-4,:) ) * idel + rhs(:,or,:) = rhs(:,or,:) + & + ( a(1,1) * var(:,or,:) + a(2,1) * var(:,or-1,:) + & + a(3,1) * var(:,or-2,:) + a(4,1) * var(:,or-3,:) ) * idel - jr = nj - 4 + jr = or - 4 end if rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & ( -20.0_wp * var(:,jl:jr,:) + & @@ -148,46 +153,48 @@ subroutine dissipation_4_2_alt (var, ni, nj, nk, bb, gsize, delta, epsilon, rhs) if ( bb(5) == 0 ) then kl = 1 + gsize(3) else - rhs(:,:,1) = rhs(:,:,1) + & - ( a(1,1) * var(:,:,1) + a(2,1) * var(:,:,2) + & - a(3,1) * var(:,:,3) + a(4,1) * var(:,:,4) ) * idel - rhs(:,:,2) = rhs(:,:,2) + & - ( a(1,2) * var(:,:,1) + a(2,2) * var(:,:,2) + & - a(3,2) * var(:,:,3) + a(4,2) * var(:,:,4) + & - a(5,2) * var(:,:,5) ) * idel - rhs(:,:,3) = rhs(:,:,3) + & - ( a(1,3) * var(:,:,1) + a(2,3) * var(:,:,2) + & - a(3,3) * var(:,:,3) + a(4,3) * var(:,:,4) + & - a(5,3) * var(:,:,5) + a(6,3) * var(:,:,6) ) * idel - rhs(:,:,4) = rhs(:,:,4) + & - ( a(1,4) * var(:,:,1) + a(2,4) * var(:,:,2) + & - a(3,4) * var(:,:,3) + a(4,4) * var(:,:,4) + & - a(5,4) * var(:,:,5) + a(6,4) * var(:,:,6) + & - a(7,4) * var(:,:,7) ) * idel + ol = offset(5) + rhs(:,:,1+ol) = rhs(:,:,1+ol) + & + ( a(1,1) * var(:,:,1+ol) + a(2,1) * var(:,:,2+ol) + & + a(3,1) * var(:,:,3+ol) + a(4,1) * var(:,:,4+ol) ) * idel + rhs(:,:,2+ol) = rhs(:,:,2+ol) + & + ( a(1,2) * var(:,:,1+ol) + a(2,2) * var(:,:,2+ol) + & + a(3,2) * var(:,:,3+ol) + a(4,2) * var(:,:,4+ol) + & + a(5,2) * var(:,:,5+ol) ) * idel + rhs(:,:,3+ol) = rhs(:,:,3+ol) + & + ( a(1,3) * var(:,:,1+ol) + a(2,3) * var(:,:,2+ol) + & + a(3,3) * var(:,:,3+ol) + a(4,3) * var(:,:,4+ol) + & + a(5,3) * var(:,:,5+ol) + a(6,3) * var(:,:,6+ol) ) * idel + rhs(:,:,4+ol) = rhs(:,:,4+ol) + & + ( a(1,4) * var(:,:,1+ol) + a(2,4) * var(:,:,2+ol) + & + a(3,4) * var(:,:,3+ol) + a(4,4) * var(:,:,4+ol) + & + a(5,4) * var(:,:,5+ol) + a(6,4) * var(:,:,6+ol) + & + a(7,4) * var(:,:,7+ol) ) * idel - kl = 5 + kl = 5 + ol end if if ( bb(6) == 0 ) then kr = nk - gsize(3) else - rhs(:,:,nk-3) = rhs(:,:,nk-3) + & - ( a(1,4) * var(:,:,nk) + a(2,4) * var(:,:,nk-1) + & - a(3,4) * var(:,:,nk-2) + a(4,4) * var(:,:,nk-3) + & - a(5,4) * var(:,:,nk-4) + a(6,4) * var(:,:,nk-5) + & - a(7,4) * var(:,:,nk-6) ) * idel - rhs(:,:,nk-2) = rhs(:,:,nk-2) + & - ( a(1,3) * var(:,:,nk) + a(2,3) * var(:,:,nk-1) + & - a(3,3) * var(:,:,nk-2) + a(4,3) * var(:,:,nk-3) + & - a(5,3) * var(:,:,nk-4) + a(6,3) * var(:,:,nk-5) ) * idel - rhs(:,:,nk-1) = rhs(:,:,nk-1) + & - ( a(1,2) * var(:,:,nk) + a(2,2) * var(:,:,nk-1) + & - a(3,2) * var(:,:,nk-2) + a(4,2) * var(:,:,nk-3) + & - a(5,2) * var(:,:,nk-4) ) * idel - rhs(:,:,nk) = rhs(:,:,nk) + & - ( a(1,1) * var(:,:,nk) + a(2,1) * var(:,:,nk-1) + & - a(3,1) * var(:,:,nk-2) + a(4,1) * var(:,:,nk-3) ) * idel + or = nk - offset(6) + rhs(:,:,or-3) = rhs(:,:,or-3) + & + ( a(1,4) * var(:,:,or) + a(2,4) * var(:,:,or-1) + & + a(3,4) * var(:,:,or-2) + a(4,4) * var(:,:,or-3) + & + a(5,4) * var(:,:,or-4) + a(6,4) * var(:,:,or-5) + & + a(7,4) * var(:,:,or-6) ) * idel + rhs(:,:,or-2) = rhs(:,:,or-2) + & + ( a(1,3) * var(:,:,or) + a(2,3) * var(:,:,or-1) + & + a(3,3) * var(:,:,or-2) + a(4,3) * var(:,:,or-3) + & + a(5,3) * var(:,:,or-4) + a(6,3) * var(:,:,or-5) ) * idel + rhs(:,:,or-1) = rhs(:,:,or-1) + & + ( a(1,2) * var(:,:,or) + a(2,2) * var(:,:,or-1) + & + a(3,2) * var(:,:,or-2) + a(4,2) * var(:,:,or-3) + & + a(5,2) * var(:,:,or-4) ) * idel + rhs(:,:,or) = rhs(:,:,or) + & + ( a(1,1) * var(:,:,or) + a(2,1) * var(:,:,or-1) + & + a(3,1) * var(:,:,or-2) + a(4,1) * var(:,:,or-3) ) * idel - kr = nk - 4 + kr = or - 4 end if rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & ( -20.0_wp * var(:,:,kl:kr) + & |