diff options
Diffstat (limited to 'src/Dissipation_4_3_min_err_coeff.F90')
-rw-r--r-- | src/Dissipation_4_3_min_err_coeff.F90 | 500 |
1 files changed, 281 insertions, 219 deletions
diff --git a/src/Dissipation_4_3_min_err_coeff.F90 b/src/Dissipation_4_3_min_err_coeff.F90 index 7aa79ef..bd6a11d 100644 --- a/src/Dissipation_4_3_min_err_coeff.F90 +++ b/src/Dissipation_4_3_min_err_coeff.F90 @@ -5,7 +5,7 @@ #include "cctk_Functions.h" -subroutine dissipation_4_3_opt (var, lsh, gsh, lbnd, bb, gsize, & +subroutine dissipation_4_3_opt (var, lsh, gsh, lbnd, bb, gsize, offset, & delta, epsilon, dfl, npatches, patch, rhs) use dissipation_coeff @@ -21,12 +21,12 @@ subroutine dissipation_4_3_opt (var, lsh, gsh, lbnd, bb, gsize, & CCTK_REAL, dimension(lsh(1),lsh(2),lsh(3)), 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 CCTK_REAL, dimension(3), intent(in) :: dfl CCTK_INT, intent(in) :: npatches, patch - CCTK_REAL :: zero = 0.0 integer, parameter :: wp = kind(zero) integer :: i, j, k @@ -34,7 +34,8 @@ subroutine dissipation_4_3_opt (var, lsh, gsh, lbnd, bb, gsize, & CCTK_REAL, dimension(:,:), pointer :: a CCTK_REAL :: idel - CCTK_INT :: il, ir, jl, jr, kl, kr + CCTK_INT :: il, ir, jl, jr, kl, kr, ol, or + CCTK_INT :: nii, njj, nkk ni = lsh(1); nj = lsh(2); nk = lsh(3) @@ -46,18 +47,26 @@ subroutine dissipation_4_3_opt (var, lsh, gsh, lbnd, bb, gsize, & end if if ( .not. savedx(patch) ) then - allocate ( atmp(ni,ni), d(ni,ni), b(ni,ni), h(ni,ni) ) + nii = ni + if ( bb(1) /= 0 ) then + nii = nii - offset(1) + endif + if ( bb(2) /= 0 ) then + nii = nii - offset(2) + endif + allocate ( atmp(nii,nii), d(nii,nii), b(nii,nii), h(nii,nii) ) atmp = zero; d = zero; b = zero; h = zero call set_dmatrix ( d, bb(1:2) ) - call set_bmatrix ( b, bb(1:2), lsh(1), gsh(1), lbnd(1), delta(1), dfl(1) ) + call set_bmatrix ( b, bb(1:2), lsh(1), gsh(1)-offset(1)-offset(2), lbnd(1), delta(1), dfl(1) ) call set_hmatrix ( h, bb(1:2) ) + atmp = -transpose ( matmul ( h, matmul ( transpose(d), matmul ( b, d ) ) ) ) - allocate ( xcoeff(patch)%coeff(ni,ni) ) + allocate ( xcoeff(patch)%coeff(nii,nii) ) xcoeff(patch)%coeff = atmp @@ -78,95 +87,115 @@ subroutine dissipation_4_3_opt (var, lsh, gsh, lbnd, bb, gsize, & 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,:,:) ) * 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,:,:) + a(6,2) * var(6,:,:) + & - a(7,2) * var(7,:,:) ) * 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,:,:) + & - a(7,3) * var(7,:,:) ) * 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 - rhs(5,:,:) = rhs(5,:,:) + & - ( a(1,5) * var(1,:,:) + a(2,5) * var(2,:,:) + & - a(3,5) * var(3,:,:) + a(4,5) * var(4,:,:) + & - a(5,5) * var(5,:,:) + a(6,5) * var(6,:,:) + & - a(7,5) * var(7,:,:) ) * idel - il = 6 + 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,:,:) ) * 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,:,:) + a(6,2) * var(6+ol,:,:) + & + a(7,2) * var(7+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,:,:) + & + a(7,3) * var(7+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 + rhs(5+ol,:,:) = rhs(5+ol,:,:) + & + ( a(1,5) * var(1+ol,:,:) + a(2,5) * var(2+ol,:,:) + & + a(3,5) * var(3+ol,:,:) + a(4,5) * var(4+ol,:,:) + & + a(5,5) * var(5+ol,:,:) + a(6,5) * var(6+ol,:,:) + & + a(7,5) * var(7+ol,:,:) ) * idel + il = 6 + ol end if if ( bb(2) == 0 ) then ir = ni - gsize(1) else - rhs(ni-4,:,:) = rhs(ni-4,:,:) + & - ( a(ni-6,ni-4) * var(ni-6,:,:) + & - a(ni-5,ni-4) * var(ni-5,:,:) + & - a(ni-4,ni-4) * var(ni-4,:,:) + & - a(ni-3,ni-4) * var(ni-3,:,:) + & - a(ni-2,ni-4) * var(ni-2,:,:) + & - a(ni-1,ni-4) * var(ni-1,:,:) + & - a(ni,ni-4) * var(ni,:,:) ) * idel - rhs(ni-3,:,:) = rhs(ni-3,:,:) + & - ( a(ni-6,ni-3) * var(ni-6,:,:) + & - a(ni-5,ni-3) * var(ni-5,:,:) + & - a(ni-4,ni-3) * var(ni-4,:,:) + & - a(ni-3,ni-3) * var(ni-3,:,:) + & - a(ni-2,ni-3) * var(ni-2,:,:) + & - a(ni-1,ni-3) * var(ni-1,:,:) + & - a(ni,ni-3) * var(ni,:,:) ) * idel - rhs(ni-2,:,:) = rhs(ni-2,:,:) + & - ( a(ni-6,ni-2) * var(ni-6,:,:) + & - a(ni-5,ni-2) * var(ni-5,:,:) + & - a(ni-4,ni-2) * var(ni-4,:,:) + & - a(ni-3,ni-2) * var(ni-3,:,:) + & - a(ni-2,ni-2) * var(ni-2,:,:) + & - a(ni-1,ni-2) * var(ni-1,:,:) + & - a(ni,ni-2) * var(ni,:,:) ) * idel - rhs(ni-1,:,:) = rhs(ni-1,:,:) + & - ( a(ni-6,ni-1) * var(ni-6,:,:) + & - a(ni-5,ni-1) * var(ni-5,:,:) + & - a(ni-4,ni-1) * var(ni-4,:,:) + & - a(ni-3,ni-1) * var(ni-3,:,:) + & - a(ni-2,ni-1) * var(ni-2,:,:) + & - a(ni-1,ni-1) * var(ni-1,:,:) + & - a(ni,ni-1) * var(ni,:,:) ) * idel - rhs(ni,:,:) = rhs(ni,:,:) + & - ( a(ni-2,ni) * var(ni-2,:,:) + & - a(ni-1,ni) * var(ni-1,:,:) + & - a(ni,ni) * var(ni,:,:) ) * idel - ir = ni - 5 + or = ni - offset(2) + rhs(or-4,:,:) = rhs(or-4,:,:) + & + ( a(nii-6,nii-4) * var(or-6,:,:) + & + a(nii-5,nii-4) * var(or-5,:,:) + & + a(nii-4,nii-4) * var(or-4,:,:) + & + a(nii-3,nii-4) * var(or-3,:,:) + & + a(nii-2,nii-4) * var(or-2,:,:) + & + a(nii-1,nii-4) * var(or-1,:,:) + & + a(nii,nii-4) * var(or,:,:) ) * idel + rhs(or-3,:,:) = rhs(or-3,:,:) + & + ( a(nii-6,nii-3) * var(or-6,:,:) + & + a(nii-5,nii-3) * var(or-5,:,:) + & + a(nii-4,nii-3) * var(or-4,:,:) + & + a(nii-3,nii-3) * var(or-3,:,:) + & + a(nii-2,nii-3) * var(or-2,:,:) + & + a(nii-1,nii-3) * var(or-1,:,:) + & + a(nii,nii-3) * var(or,:,:) ) * idel + rhs(or-2,:,:) = rhs(or-2,:,:) + & + ( a(nii-6,nii-2) * var(or-6,:,:) + & + a(nii-5,nii-2) * var(or-5,:,:) + & + a(nii-4,nii-2) * var(or-4,:,:) + & + a(nii-3,nii-2) * var(or-3,:,:) + & + a(nii-2,nii-2) * var(or-2,:,:) + & + a(nii-1,nii-2) * var(or-1,:,:) + & + a(nii,nii-2) * var(or,:,:) ) * idel + rhs(or-1,:,:) = rhs(or-1,:,:) + & + ( a(nii-6,nii-1) * var(or-6,:,:) + & + a(nii-5,nii-1) * var(or-5,:,:) + & + a(nii-4,nii-1) * var(or-4,:,:) + & + a(nii-3,nii-1) * var(or-3,:,:) + & + a(nii-2,nii-1) * var(or-2,:,:) + & + a(nii-1,nii-1) * var(or-1,:,:) + & + a(nii,nii-1) * var(or,:,:) ) * idel + rhs(or,:,:) = rhs(or,:,:) + & + ( a(nii-2,nii) * var(or-2,:,:) + & + a(nii-1,nii) * var(or-1,:,:) + & + a(nii,nii) * var(or,:,:) ) * idel + ir = or - 5 end if - do i = il, ir - rhs(i,:,:) = rhs(i,:,:) + & - ( a(i-2,i) * var(i-2,:,:) + & - a(i-1,i) * var(i-1,:,:) + & - a(i,i) * var(i,:,:) + & - a(i+1,i) * var(i+1,:,:) + & - a(i+2,i) * var(i+2,:,:) ) * idel - end do + if (bb(1) /= 0) then + do i = il, ir + rhs(i,:,:) = rhs(i,:,:) + & + ( a(i-ol-2,i-ol) * var(i-2,:,:) + & + a(i-ol-1,i-ol) * var(i-1,:,:) + & + a(i-ol,i-ol) * var(i,:,:) + & + a(i-ol+1,i-ol) * var(i+1,:,:) + & + a(i-ol+2,i-ol) * var(i+2,:,:) ) * idel + end do + else + do i = il, ir + rhs(i,:,:) = rhs(i,:,:) + & + ( a(i-2,i) * var(i-2,:,:) + & + a(i-1,i) * var(i-1,:,:) + & + a(i,i) * var(i,:,:) + & + a(i+1,i) * var(i+1,:,:) + & + a(i+2,i) * var(i+2,:,:) ) * idel + end do + endif if ( zero_derivs_y == 0 ) then if ( .not. savedy(patch) ) then - allocate ( atmp(nj,nj), d(nj,nj), b(nj,nj), h(nj,nj) ) + njj = nj + if ( bb(3) /= 0 ) then + njj = njj - offset(3) + endif + if ( bb(4) /= 0 ) then + njj = njj - offset(4) + endif + allocate ( atmp(njj,njj), d(njj,njj), b(njj,njj), h(njj,njj) ) atmp = zero; d = zero; b = zero; h = zero call set_dmatrix ( d, bb(3:4) ) - call set_bmatrix ( b, bb(3:4), lsh(2), gsh(2), lbnd(2), delta(2), dfl(2) ) + call set_bmatrix ( b, bb(3:4), lsh(2), gsh(2)-offset(3)-offset(4), lbnd(2), delta(2), dfl(2) ) call set_hmatrix ( h, bb(3:4) ) atmp = -transpose ( matmul ( h, matmul ( transpose(d), matmul ( b, d ) ) ) ) - allocate ( ycoeff(patch)%coeff(nj,nj) ) + allocate ( ycoeff(patch)%coeff(njj,njj) ) ycoeff(patch)%coeff = atmp @@ -187,98 +216,118 @@ subroutine dissipation_4_3_opt (var, lsh, gsh, lbnd, bb, gsize, & 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,:) ) * 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,:) + a(6,2) * var(:,6,:) + & - a(7,2) * var(:,7,:) ) * 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,:) + & - a(7,3) * var(:,7,:) ) * 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 - rhs(:,5,:) = rhs(:,5,:) + & - ( a(1,5) * var(:,1,:) + a(2,5) * var(:,2,:) + & - a(3,5) * var(:,3,:) + a(4,5) * var(:,4,:) + & - a(5,5) * var(:,5,:) + a(6,5) * var(:,6,:) + & - a(7,5) * var(:,7,:) ) * idel - jl = 6 + 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,:) ) * 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,:) + a(6,2) * var(:,6+ol,:) + & + a(7,2) * var(:,7+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,:) + & + a(7,3) * var(:,7+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 + rhs(:,5+ol,:) = rhs(:,5+ol,:) + & + ( a(1,5) * var(:,1+ol,:) + a(2,5) * var(:,2+ol,:) + & + a(3,5) * var(:,3+ol,:) + a(4,5) * var(:,4+ol,:) + & + a(5,5) * var(:,5+ol,:) + a(6,5) * var(:,6+ol,:) + & + a(7,5) * var(:,7+ol,:) ) * idel + jl = 6 + ol end if if ( bb(4) == 0 ) then jr = nj - gsize(2) else - rhs(:,nj-4,:) = rhs(:,nj-4,:) + & - ( a(nj-6,nj-4) * var(:,nj-6,:) + & - a(nj-5,nj-4) * var(:,nj-5,:) + & - a(nj-4,nj-4) * var(:,nj-4,:) + & - a(nj-3,nj-4) * var(:,nj-3,:) + & - a(nj-2,nj-4) * var(:,nj-2,:) + & - a(nj-1,nj-4) * var(:,nj-1,:) + & - a(nj,nj-4) * var(:,nj,:) ) * idel - rhs(:,nj-3,:) = rhs(:,nj-3,:) + & - ( a(nj-6,nj-3) * var(:,nj-6,:) + & - a(nj-5,nj-3) * var(:,nj-5,:) + & - a(nj-4,nj-3) * var(:,nj-4,:) + & - a(nj-3,nj-3) * var(:,nj-3,:) + & - a(nj-2,nj-3) * var(:,nj-2,:) + & - a(nj-1,nj-3) * var(:,nj-1,:) + & - a(nj,nj-3) * var(:,nj,:) ) * idel - rhs(:,nj-2,:) = rhs(:,nj-2,:) + & - ( a(nj-6,nj-2) * var(:,nj-6,:) + & - a(nj-5,nj-2) * var(:,nj-5,:) + & - a(nj-4,nj-2) * var(:,nj-4,:) + & - a(nj-3,nj-2) * var(:,nj-3,:) + & - a(nj-2,nj-2) * var(:,nj-2,:) + & - a(nj-1,nj-2) * var(:,nj-1,:) + & - a(nj,nj-2) * var(:,nj,:) ) * idel - rhs(:,nj-1,:) = rhs(:,nj-1,:) + & - ( a(nj-6,nj-1) * var(:,nj-6,:) + & - a(nj-5,nj-1) * var(:,nj-5,:) + & - a(nj-4,nj-1) * var(:,nj-4,:) + & - a(nj-3,nj-1) * var(:,nj-3,:) + & - a(nj-2,nj-1) * var(:,nj-2,:) + & - a(nj-1,nj-1) * var(:,nj-1,:) + & - a(nj,nj-1) * var(:,nj,:) ) * idel - rhs(:,nj,:) = rhs(:,nj,:) + & - ( a(nj-2,nj) * var(:,nj-2,:) + & - a(nj-1,nj) * var(:,nj-1,:) + & - a(nj,nj) * var(:,nj,:) ) * idel - jr = nj - 5 + or = nj - offset(4) + rhs(:,or-4,:) = rhs(:,or-4,:) + & + ( a(njj-6,njj-4) * var(:,or-6,:) + & + a(njj-5,njj-4) * var(:,or-5,:) + & + a(njj-4,njj-4) * var(:,or-4,:) + & + a(njj-3,njj-4) * var(:,or-3,:) + & + a(njj-2,njj-4) * var(:,or-2,:) + & + a(njj-1,njj-4) * var(:,or-1,:) + & + a(njj,njj-4) * var(:,or,:) ) * idel + rhs(:,or-3,:) = rhs(:,or-3,:) + & + ( a(njj-6,njj-3) * var(:,or-6,:) + & + a(njj-5,njj-3) * var(:,or-5,:) + & + a(njj-4,njj-3) * var(:,or-4,:) + & + a(njj-3,njj-3) * var(:,or-3,:) + & + a(njj-2,njj-3) * var(:,or-2,:) + & + a(njj-1,njj-3) * var(:,or-1,:) + & + a(njj,njj-3) * var(:,or,:) ) * idel + rhs(:,or-2,:) = rhs(:,or-2,:) + & + ( a(njj-6,njj-2) * var(:,or-6,:) + & + a(njj-5,njj-2) * var(:,or-5,:) + & + a(njj-4,njj-2) * var(:,or-4,:) + & + a(njj-3,njj-2) * var(:,or-3,:) + & + a(njj-2,njj-2) * var(:,or-2,:) + & + a(njj-1,njj-2) * var(:,or-1,:) + & + a(njj,njj-2) * var(:,or,:) ) * idel + rhs(:,or-1,:) = rhs(:,or-1,:) + & + ( a(njj-6,njj-1) * var(:,or-6,:) + & + a(njj-5,njj-1) * var(:,or-5,:) + & + a(njj-4,njj-1) * var(:,or-4,:) + & + a(njj-3,njj-1) * var(:,or-3,:) + & + a(njj-2,njj-1) * var(:,or-2,:) + & + a(njj-1,njj-1) * var(:,or-1,:) + & + a(njj,njj-1) * var(:,or,:) ) * idel + rhs(:,or,:) = rhs(:,or,:) + & + ( a(njj-2,njj) * var(:,or-2,:) + & + a(njj-1,njj) * var(:,or-1,:) + & + a(njj,njj) * var(:,or,:) ) * idel + jr = or - 5 end if - do j = jl, jr - rhs(:,j,:) = rhs(:,j,:) + & - ( a(j-2,j) * var(:,j-2,:) + & - a(j-1,j) * var(:,j-1,:) + & - a(j,j) * var(:,j,:) + & - a(j+1,j) * var(:,j+1,:) + & - a(j+2,j) * var(:,j+2,:) ) * idel - end do + if ( bb(3) /= 0 ) then + do j = jl, jr + rhs(:,j,:) = rhs(:,j,:) + & + ( a(j-ol-2,j-ol) * var(:,j-2,:) + & + a(j-ol-1,j-ol) * var(:,j-1,:) + & + a(j-ol,j-ol) * var(:,j,:) + & + a(j-ol+1,j-ol) * var(:,j+1,:) + & + a(j-ol+2,j-ol) * var(:,j+2,:) ) * idel + end do + else + do j = jl, jr + rhs(:,j,:) = rhs(:,j,:) + & + ( a(j-2,j) * var(:,j-2,:) + & + a(j-1,j) * var(:,j-1,:) + & + a(j,j) * var(:,j,:) + & + a(j+1,j) * var(:,j+1,:) + & + a(j+2,j) * var(:,j+2,:) ) * idel + end do + endif end if if ( zero_derivs_z == 0 ) then if ( .not. savedz(patch) ) then - allocate ( atmp(nk,nk), d(nk,nk), b(nk,nk), h(nk,nk) ) + nkk = nk + if ( bb(5) /= 0 ) then + nkk = nkk - offset(5) + endif + if ( bb(6) /= 0 ) then + nkk = nkk - offset(6) + endif + allocate ( atmp(nkk,nkk), d(nkk,nkk), b(nkk,nkk), h(nkk,nkk) ) atmp = zero; d = zero; b = zero; h = zero call set_dmatrix ( d, bb(5:6) ) - call set_bmatrix ( b, bb(5:6), lsh(3), gsh(3), lbnd(3), delta(3), dfl(3) ) + call set_bmatrix ( b, bb(5:6), lsh(3), gsh(3)-offset(5)-offset(6), lbnd(3), delta(3), dfl(3) ) call set_hmatrix ( h, bb(5:6) ) atmp = -transpose ( matmul ( h, matmul ( transpose(d), matmul ( b, d ) ) ) ) - allocate ( zcoeff(patch)%coeff(nk,nk) ) + allocate ( zcoeff(patch)%coeff(nkk,nkk) ) zcoeff(patch)%coeff = atmp @@ -299,81 +348,94 @@ subroutine dissipation_4_3_opt (var, lsh, gsh, lbnd, bb, gsize, & 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) ) * 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) + a(6,2) * var(:,:,6) + & - a(7,2) * var(:,:,7) ) * 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) + & - a(7,3) * var(:,:,7) ) * 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 - rhs(:,:,5) = rhs(:,:,5) + & - ( a(1,5) * var(:,:,1) + a(2,5) * var(:,:,2) + & - a(3,5) * var(:,:,3) + a(4,5) * var(:,:,4) + & - a(5,5) * var(:,:,5) + a(6,5) * var(:,:,6) + & - a(7,5) * var(:,:,7) ) * idel - kl = 6 + 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) ) * 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) + a(6,2) * var(:,:,6+ol) + & + a(7,2) * var(:,:,7+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) + & + a(7,3) * var(:,:,7+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 + rhs(:,:,5+ol) = rhs(:,:,5+ol) + & + ( a(1,5) * var(:,:,1+ol) + a(2,5) * var(:,:,2+ol) + & + a(3,5) * var(:,:,3+ol) + a(4,5) * var(:,:,4+ol) + & + a(5,5) * var(:,:,5+ol) + a(6,5) * var(:,:,6+ol) + & + a(7,5) * var(:,:,7+ol) ) * idel + kl = 6 + ol end if if ( bb(6) == 0 ) then kr = nk - gsize(3) else - rhs(:,:,nk-4) = rhs(:,:,nk-4) + & - ( a(nk-6,nk-4) * var(:,:,nk-6) + & - a(nk-5,nk-4) * var(:,:,nk-5) + & - a(nk-4,nk-4) * var(:,:,nk-4) + & - a(nk-3,nk-4) * var(:,:,nk-3) + & - a(nk-2,nk-4) * var(:,:,nk-2) + & - a(nk-1,nk-4) * var(:,:,nk-1) + & - a(nk,nk-4) * var(:,:,nk) ) * idel - rhs(:,:,nk-3) = rhs(:,:,nk-3) + & - ( a(nk-6,nk-3) * var(:,:,nk-6) + & - a(nk-5,nk-3) * var(:,:,nk-5) + & - a(nk-4,nk-3) * var(:,:,nk-4) + & - a(nk-3,nk-3) * var(:,:,nk-3) + & - a(nk-2,nk-3) * var(:,:,nk-2) + & - a(nk-1,nk-3) * var(:,:,nk-1) + & - a(nk,nk-3) * var(:,:,nk) ) * idel - rhs(:,:,nk-2) = rhs(:,:,nk-2) + & - ( a(nk-6,nk-2) * var(:,:,nk-6) + & - a(nk-5,nk-2) * var(:,:,nk-5) + & - a(nk-4,nk-2) * var(:,:,nk-4) + & - a(nk-3,nk-2) * var(:,:,nk-3) + & - a(nk-2,nk-2) * var(:,:,nk-2) + & - a(nk-1,nk-2) * var(:,:,nk-1) + & - a(nk,nk-2) * var(:,:,nk) ) * idel - rhs(:,:,nk-1) = rhs(:,:,nk-1) + & - ( a(nk-6,nk-1) * var(:,:,nk-6) + & - a(nk-5,nk-1) * var(:,:,nk-5) + & - a(nk-4,nk-1) * var(:,:,nk-4) + & - a(nk-3,nk-1) * var(:,:,nk-3) + & - a(nk-2,nk-1) * var(:,:,nk-2) + & - a(nk-1,nk-1) * var(:,:,nk-1) + & - a(nk,nk-1) * var(:,:,nk) ) * idel - rhs(:,:,nk) = rhs(:,:,nk) + & - ( a(nk-2,nk) * var(:,:,nk-2) + & - a(nk-1,nk) * var(:,:,nk-1) + & - a(nk,nk) * var(:,:,nk) ) * idel - kr = nk - 5 + or = nk - offset(6) + rhs(:,:,or-4) = rhs(:,:,or-4) + & + ( a(nkk-6,nkk-4) * var(:,:,or-6) + & + a(nkk-5,nkk-4) * var(:,:,or-5) + & + a(nkk-4,nkk-4) * var(:,:,or-4) + & + a(nkk-3,nkk-4) * var(:,:,or-3) + & + a(nkk-2,nkk-4) * var(:,:,or-2) + & + a(nkk-1,nkk-4) * var(:,:,or-1) + & + a(nkk,nkk-4) * var(:,:,or) ) * idel + rhs(:,:,or-3) = rhs(:,:,or-3) + & + ( a(nkk-6,nkk-3) * var(:,:,or-6) + & + a(nkk-5,nkk-3) * var(:,:,or-5) + & + a(nkk-4,nkk-3) * var(:,:,or-4) + & + a(nkk-3,nkk-3) * var(:,:,or-3) + & + a(nkk-2,nkk-3) * var(:,:,or-2) + & + a(nkk-1,nkk-3) * var(:,:,or-1) + & + a(nkk,nkk-3) * var(:,:,or) ) * idel + rhs(:,:,or-2) = rhs(:,:,or-2) + & + ( a(nkk-6,nkk-2) * var(:,:,or-6) + & + a(nkk-5,nkk-2) * var(:,:,or-5) + & + a(nkk-4,nkk-2) * var(:,:,or-4) + & + a(nkk-3,nkk-2) * var(:,:,or-3) + & + a(nkk-2,nkk-2) * var(:,:,or-2) + & + a(nkk-1,nkk-2) * var(:,:,or-1) + & + a(nkk,nkk-2) * var(:,:,or) ) * idel + rhs(:,:,or-1) = rhs(:,:,or-1) + & + ( a(nkk-6,nkk-1) * var(:,:,or-6) + & + a(nkk-5,nkk-1) * var(:,:,or-5) + & + a(nkk-4,nkk-1) * var(:,:,or-4) + & + a(nkk-3,nkk-1) * var(:,:,or-3) + & + a(nkk-2,nkk-1) * var(:,:,or-2) + & + a(nkk-1,nkk-1) * var(:,:,or-1) + & + a(nkk,nkk-1) * var(:,:,or) ) * idel + rhs(:,:,or) = rhs(:,:,or) + & + ( a(nkk-2,nkk) * var(:,:,or-2) + & + a(nkk-1,nkk) * var(:,:,or-1) + & + a(nkk,nkk) * var(:,:,or) ) * idel + kr = or - 5 end if - do k = kl, kr - rhs(:,:,k) = rhs(:,:,k) + & - ( a(k-2,k) * var(:,:,k-2) + & - a(k-1,k) * var(:,:,k-1) + & - a(k,k) * var(:,:,k) + & - a(k+1,k) * var(:,:,k+1) + & - a(k+2,k) * var(:,:,k+2) ) * idel - end do + if ( bb(5) /= 0) then + do k = kl, kr + rhs(:,:,k) = rhs(:,:,k) + & + ( a(k-ol-2,k-ol) * var(:,:,k-2) + & + a(k-ol-1,k-ol) * var(:,:,k-1) + & + a(k-ol,k-ol) * var(:,:,k) + & + a(k-ol+1,k-ol) * var(:,:,k+1) + & + a(k-ol+2,k-ol) * var(:,:,k+2) ) * idel + end do + else + do k = kl, kr + rhs(:,:,k) = rhs(:,:,k) + & + ( a(k-2,k) * var(:,:,k-2) + & + a(k-1,k) * var(:,:,k-1) + & + a(k,k) * var(:,:,k) + & + a(k+1,k) * var(:,:,k+1) + & + a(k+2,k) * var(:,:,k+2) ) * idel + end do + endif end if |