diff options
Diffstat (limited to 'src/Dissipation_6_5_min_err_coeff.F90')
| -rw-r--r-- | src/Dissipation_6_5_min_err_coeff.F90 | 790 |
1 files changed, 397 insertions, 393 deletions
diff --git a/src/Dissipation_6_5_min_err_coeff.F90 b/src/Dissipation_6_5_min_err_coeff.F90 index d4e15a4..63e0fbe 100644 --- a/src/Dissipation_6_5_min_err_coeff.F90 +++ b/src/Dissipation_6_5_min_err_coeff.F90 @@ -232,406 +232,410 @@ subroutine dissipation_6_5_opt (var, lsh, gsh, lbnd, bb, gsize, & deallocate ( a, d, b, h, tmp ) - allocate ( a(nj,nj), d(nj,nj), b(nj,nj), h(nj,nj), tmp(nj,nj) ) - a = zero; d = zero; b = zero; h = zero; - - call set_bmatrix ( b, bb(3:4), lsh(2), gsh(2), lbnd(2), delta(2), dfl ) + if ( zero_derivs_y == 0 ) then + allocate ( a(nj,nj), d(nj,nj), b(nj,nj), h(nj,nj), tmp(nj,nj) ) + a = zero; d = zero; b = zero; h = zero; - call set_hmatrix ( h, bb(3:4) ) - - if ( any ( bb(3:4) == 0 ) ) then - - call set_dmatrix ( d, bb(3:4) ) - - a = -transpose ( matmul ( h, matmul ( transpose(d), matmul ( b, d ) ) ) ) - - else - - if ( lsh(2) /= gsh(2) ) then - call CCTK_WARN(0, 'Internal error. Bounding box inconsistent with grid' ) - end if - - center = gsh(2) / 2 - jr = center + gsize(2) - - call set_dmatrix_half ( d(1:jr,1:jr), 1 ) - - tmp(1:jr,1:jr) = -transpose ( & - matmul ( h(1:jr,1:jr), & - matmul ( & - transpose ( d(1:jr,1:jr) ), & - matmul ( b(1:jr,1:jr), & - d(1:jr,1:jr) ) ) ) ) - - a(1:jr,1:center) = tmp(1:jr,1:center) - - jl = center + 1 - gsize(2) - d = zero - call set_dmatrix_half ( d(jl:nj,jl:nj), 2 ) - tmp(jl:nj,jl:nj) = -transpose ( & - matmul ( h(jl:nj,jl:nj), & + call set_bmatrix ( b, bb(3:4), lsh(2), gsh(2), lbnd(2), delta(2), dfl ) + + call set_hmatrix ( h, bb(3:4) ) + + if ( any ( bb(3:4) == 0 ) ) then + + call set_dmatrix ( d, bb(3:4) ) + + a = -transpose ( matmul ( h, matmul ( transpose(d), matmul ( b, d ) ) ) ) + + else + + if ( lsh(2) /= gsh(2) ) then + call CCTK_WARN(0, 'Internal error. Bounding box inconsistent with grid' ) + end if + + center = gsh(2) / 2 + jr = center + gsize(2) + + call set_dmatrix_half ( d(1:jr,1:jr), 1 ) + + tmp(1:jr,1:jr) = -transpose ( & + matmul ( h(1:jr,1:jr), & matmul ( & - transpose ( d(jl:nj,jl:nj) ), & - matmul ( b(jl:nj,jl:nj), & - d(jl:nj,jl:nj) ) ) ) ) - - a(jl:nj,center+1:nj) = tmp(jl:nj,center+1:nj) - - end if - - if ( scale_with_h > 0 ) then - idel = epsilon / ( 64 * delta(2) ) - else - idel = epsilon / 64 - end if - - 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,:) + a(6,2) * var(:,6,:) + & - a(7,2) * var(:,7,:) + a(8,2) * var(:,8,:) + & - a(9,2) * var(:,9,:) + a(10,2) * var(:,10,:) ) * 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,:) + a(8,3) * var(:,8,:) + & - a(9,3) * var(:,9,:) + a(10,3) * var(:,10,:) ) * 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,:) + a(8,4) * var(:,8,:) + & - a(9,4) * var(:,9,:) + a(10,4) * var(:,10,:) ) * 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,:) + a(8,5) * var(:,8,:) + & - a(9,5) * var(:,9,:) + a(10,5) * var(:,10,:) ) * idel - rhs(:,6,:) = rhs(:,6,:) + & - ( a(1,6) * var(:,1,:) + a(2,6) * var(:,2,:) + & - a(3,6) * var(:,3,:) + a(4,6) * var(:,4,:) + & - a(5,6) * var(:,5,:) + a(6,6) * var(:,6,:) + & - a(7,6) * var(:,7,:) + a(8,6) * var(:,8,:) + & - a(9,6) * var(:,9,:) + a(10,6) * var(:,10,:) ) * idel - rhs(:,7,:) = rhs(:,7,:) + & - ( a(1,7) * var(:,1,:) + a(2,7) * var(:,2,:) + & - a(3,7) * var(:,3,:) + a(4,7) * var(:,4,:) + & - a(5,7) * var(:,5,:) + a(6,7) * var(:,6,:) + & - a(7,7) * var(:,7,:) + a(8,7) * var(:,8,:) + & - a(9,7) * var(:,9,:) + a(10,7) * var(:,10,:) ) * idel - - jl = 8 - - end if - - if ( bb(4) == 0 ) then - - jr = nj - gsize(2) - - else - - rhs(:,nj-6,:) = rhs(:,nj-6,:) + & - ( a(nj-9,nj-6) * var(:,nj-9,:) + & - a(nj-8,nj-6) * var(:,nj-8,:) + & - a(nj-7,nj-6) * var(:,nj-7,:) + & - a(nj-6,nj-6) * var(:,nj-6,:) + & - a(nj-5,nj-6) * var(:,nj-5,:) + & - a(nj-4,nj-6) * var(:,nj-4,:) + & - a(nj-3,nj-6) * var(:,nj-3,:) + & - a(nj-2,nj-6) * var(:,nj-2,:) + & - a(nj-1,nj-6) * var(:,nj-1,:) + & - a(nj,nj-6) * var(:,nj,:) ) * idel - rhs(:,nj-5,:) = rhs(:,nj-5,:) + & - ( a(nj-9,nj-5) * var(:,nj-9,:) + & - a(nj-8,nj-5) * var(:,nj-8,:) + & - a(nj-7,nj-5) * var(:,nj-7,:) + & - a(nj-6,nj-5) * var(:,nj-6,:) + & - a(nj-5,nj-5) * var(:,nj-5,:) + & - a(nj-4,nj-5) * var(:,nj-4,:) + & - a(nj-3,nj-5) * var(:,nj-3,:) + & - a(nj-2,nj-5) * var(:,nj-2,:) + & - a(nj-1,nj-5) * var(:,nj-1,:) + & - a(nj,nj-5) * var(:,nj,:) ) * idel - rhs(:,nj-4,:) = rhs(:,nj-4,:) + & - ( a(nj-9,nj-4) * var(:,nj-9,:) + & - a(nj-8,nj-4) * var(:,nj-8,:) + & - a(nj-7,nj-4) * var(:,nj-7,:) + & - 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-9,nj-3) * var(:,nj-9,:) + & - a(nj-8,nj-3) * var(:,nj-8,:) + & - a(nj-7,nj-3) * var(:,nj-7,:) + & - 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-9,nj-2) * var(:,nj-9,:) + & - a(nj-8,nj-2) * var(:,nj-8,:) + & - a(nj-7,nj-2) * var(:,nj-7,:) + & - 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-9,nj-1) * var(:,nj-9,:) + & - a(nj-8,nj-1) * var(:,nj-8,:) + & - a(nj-7,nj-1) * var(:,nj-7,:) + & - 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-3,nj) * var(:,nj-3,:) + & - a(nj-2,nj) * var(:,nj-2,:) + & - a(nj-1,nj) * var(:,nj-1,:) + & - a(nj,nj) * var(:,nj,:) ) * idel - - jr = nj - 7 - - end if - - do j = jl, jr - - rhs(:,j,:) = rhs(:,j,:) + & - ( a(j-3,j) * var(:,j-3,:) + & - 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,:) + & - a(j+3,j) * var(:,j+3,:) ) * idel - - end do - - deallocate ( a, d, b, h, tmp ) - - allocate ( a(nk,nk), d(nk,nk), b(nk,nk), h(nk,nk), tmp(nk,nk) ) - a = zero; d = zero; b = zero; h = zero; tmp = zero - - call set_bmatrix ( b, bb(5:6), lsh(3), gsh(3), lbnd(3), delta(3), dfl ) - - call set_hmatrix ( h, bb(5:6) ) - - if ( any ( bb(5:6) == 0 ) ) then - - call set_dmatrix ( d, bb(5:6) ) - - a = -transpose ( matmul ( h, matmul ( transpose(d), matmul ( b, d ) ) ) ) - - else - - if ( lsh(3) /= gsh(3) ) then - call CCTK_WARN(0, 'Internal error. Bounding box inconsistent with grid' ) + transpose ( d(1:jr,1:jr) ), & + matmul ( b(1:jr,1:jr), & + d(1:jr,1:jr) ) ) ) ) + + a(1:jr,1:center) = tmp(1:jr,1:center) + + jl = center + 1 - gsize(2) + d = zero + call set_dmatrix_half ( d(jl:nj,jl:nj), 2 ) + tmp(jl:nj,jl:nj) = -transpose ( & + matmul ( h(jl:nj,jl:nj), & + matmul ( & + transpose ( d(jl:nj,jl:nj) ), & + matmul ( b(jl:nj,jl:nj), & + d(jl:nj,jl:nj) ) ) ) ) + + a(jl:nj,center+1:nj) = tmp(jl:nj,center+1:nj) + end if - - center = gsh(3) / 2 - kr = center + gsize(3) - - call set_dmatrix_half ( d(1:kr,1:kr), 1 ) - - tmp(1:kr,1:kr) = -transpose ( & - matmul ( h(1:kr,1:kr), & - matmul ( & - transpose ( d(1:kr,1:kr) ), & - matmul ( b(1:kr,1:kr), & - d(1:kr,1:kr) ) ) ) ) - - a(1:kr,1:center) = tmp(1:kr,1:center) - - kl = center + 1 - gsize(3) - d = zero - call set_dmatrix_half ( d(kl:nk,kl:nk), 2 ) - tmp(kl:nk,kl:nk) = -transpose ( & - matmul ( h(kl:nk,kl:nk), & - matmul ( & - transpose ( d(kl:nk,kl:nk) ), & - matmul ( b(kl:nk,kl:nk), & - d(kl:nk,kl:nk) ) ) ) ) - - a(kl:nk,center+1:nk) = tmp(kl:nk,center+1:nk) - - end if - - if ( scale_with_h > 0 ) then - idel = epsilon / ( 64 * delta(3) ) - else - idel = epsilon / 64 - end if - - 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) + a(6,2) * var(:,:,6) + & - a(7,2) * var(:,:,7) + a(8,2) * var(:,:,8) + & - a(9,2) * var(:,:,9) + a(10,2) * var(:,:,10) ) * 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) + a(8,3) * var(:,:,8) + & - a(9,3) * var(:,:,9) + a(10,3) * var(:,:,10) ) * 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) + a(8,4) * var(:,:,8) + & - a(9,4) * var(:,:,9) + a(10,4) * var(:,:,10) ) * 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) + a(8,5) * var(:,:,8) + & - a(9,5) * var(:,:,9) + a(10,5) * var(:,:,10) ) * idel - rhs(:,:,6) = rhs(:,:,6) + & - ( a(1,6) * var(:,:,1) + a(2,6) * var(:,:,2) + & - a(3,6) * var(:,:,3) + a(4,6) * var(:,:,4) + & - a(5,6) * var(:,:,5) + a(6,6) * var(:,:,6) + & - a(7,6) * var(:,:,7) + a(8,6) * var(:,:,8) + & - a(9,6) * var(:,:,9) + a(10,6) * var(:,:,10) ) * idel - rhs(:,:,7) = rhs(:,:,7) + & - ( a(1,7) * var(:,:,1) + a(2,7) * var(:,:,2) + & - a(3,7) * var(:,:,3) + a(4,7) * var(:,:,4) + & - a(5,7) * var(:,:,5) + a(6,7) * var(:,:,6) + & - a(7,7) * var(:,:,7) + a(8,7) * var(:,:,8) + & - a(9,7) * var(:,:,9) + a(10,7) * var(:,:,10) ) * idel - - kl = 8 - + + if ( scale_with_h > 0 ) then + idel = epsilon / ( 64 * delta(2) ) + else + idel = epsilon / 64 + end if + + 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,:) + a(6,2) * var(:,6,:) + & + a(7,2) * var(:,7,:) + a(8,2) * var(:,8,:) + & + a(9,2) * var(:,9,:) + a(10,2) * var(:,10,:) ) * 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,:) + a(8,3) * var(:,8,:) + & + a(9,3) * var(:,9,:) + a(10,3) * var(:,10,:) ) * 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,:) + a(8,4) * var(:,8,:) + & + a(9,4) * var(:,9,:) + a(10,4) * var(:,10,:) ) * 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,:) + a(8,5) * var(:,8,:) + & + a(9,5) * var(:,9,:) + a(10,5) * var(:,10,:) ) * idel + rhs(:,6,:) = rhs(:,6,:) + & + ( a(1,6) * var(:,1,:) + a(2,6) * var(:,2,:) + & + a(3,6) * var(:,3,:) + a(4,6) * var(:,4,:) + & + a(5,6) * var(:,5,:) + a(6,6) * var(:,6,:) + & + a(7,6) * var(:,7,:) + a(8,6) * var(:,8,:) + & + a(9,6) * var(:,9,:) + a(10,6) * var(:,10,:) ) * idel + rhs(:,7,:) = rhs(:,7,:) + & + ( a(1,7) * var(:,1,:) + a(2,7) * var(:,2,:) + & + a(3,7) * var(:,3,:) + a(4,7) * var(:,4,:) + & + a(5,7) * var(:,5,:) + a(6,7) * var(:,6,:) + & + a(7,7) * var(:,7,:) + a(8,7) * var(:,8,:) + & + a(9,7) * var(:,9,:) + a(10,7) * var(:,10,:) ) * idel + + jl = 8 + + end if + + if ( bb(4) == 0 ) then + + jr = nj - gsize(2) + + else + + rhs(:,nj-6,:) = rhs(:,nj-6,:) + & + ( a(nj-9,nj-6) * var(:,nj-9,:) + & + a(nj-8,nj-6) * var(:,nj-8,:) + & + a(nj-7,nj-6) * var(:,nj-7,:) + & + a(nj-6,nj-6) * var(:,nj-6,:) + & + a(nj-5,nj-6) * var(:,nj-5,:) + & + a(nj-4,nj-6) * var(:,nj-4,:) + & + a(nj-3,nj-6) * var(:,nj-3,:) + & + a(nj-2,nj-6) * var(:,nj-2,:) + & + a(nj-1,nj-6) * var(:,nj-1,:) + & + a(nj,nj-6) * var(:,nj,:) ) * idel + rhs(:,nj-5,:) = rhs(:,nj-5,:) + & + ( a(nj-9,nj-5) * var(:,nj-9,:) + & + a(nj-8,nj-5) * var(:,nj-8,:) + & + a(nj-7,nj-5) * var(:,nj-7,:) + & + a(nj-6,nj-5) * var(:,nj-6,:) + & + a(nj-5,nj-5) * var(:,nj-5,:) + & + a(nj-4,nj-5) * var(:,nj-4,:) + & + a(nj-3,nj-5) * var(:,nj-3,:) + & + a(nj-2,nj-5) * var(:,nj-2,:) + & + a(nj-1,nj-5) * var(:,nj-1,:) + & + a(nj,nj-5) * var(:,nj,:) ) * idel + rhs(:,nj-4,:) = rhs(:,nj-4,:) + & + ( a(nj-9,nj-4) * var(:,nj-9,:) + & + a(nj-8,nj-4) * var(:,nj-8,:) + & + a(nj-7,nj-4) * var(:,nj-7,:) + & + 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-9,nj-3) * var(:,nj-9,:) + & + a(nj-8,nj-3) * var(:,nj-8,:) + & + a(nj-7,nj-3) * var(:,nj-7,:) + & + 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-9,nj-2) * var(:,nj-9,:) + & + a(nj-8,nj-2) * var(:,nj-8,:) + & + a(nj-7,nj-2) * var(:,nj-7,:) + & + 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-9,nj-1) * var(:,nj-9,:) + & + a(nj-8,nj-1) * var(:,nj-8,:) + & + a(nj-7,nj-1) * var(:,nj-7,:) + & + 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-3,nj) * var(:,nj-3,:) + & + a(nj-2,nj) * var(:,nj-2,:) + & + a(nj-1,nj) * var(:,nj-1,:) + & + a(nj,nj) * var(:,nj,:) ) * idel + + jr = nj - 7 + + end if + + do j = jl, jr + + rhs(:,j,:) = rhs(:,j,:) + & + ( a(j-3,j) * var(:,j-3,:) + & + 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,:) + & + a(j+3,j) * var(:,j+3,:) ) * idel + + end do + + deallocate ( a, d, b, h, tmp ) end if - if ( bb(6) == 0 ) then - - kr = nk - gsize(3) - - else - - rhs(:,:,nk-6) = rhs(:,:,nk-6) + & - ( a(nk-9,nk-6) * var(:,:,nk-9) + & - a(nk-8,nk-6) * var(:,:,nk-8) + & - a(nk-7,nk-6) * var(:,:,nk-7) + & - a(nk-6,nk-6) * var(:,:,nk-6) + & - a(nk-5,nk-6) * var(:,:,nk-5) + & - a(nk-4,nk-6) * var(:,:,nk-4) + & - a(nk-3,nk-6) * var(:,:,nk-3) + & - a(nk-2,nk-6) * var(:,:,nk-2) + & - a(nk-1,nk-6) * var(:,:,nk-1) + & - a(nk,nk-6) * var(:,:,nk) ) * idel - rhs(:,:,nk-5) = rhs(:,:,nk-5) + & - ( a(nk-9,nk-5) * var(:,:,nk-9) + & - a(nk-8,nk-5) * var(:,:,nk-8) + & - a(nk-7,nk-5) * var(:,:,nk-7) + & - a(nk-6,nk-5) * var(:,:,nk-6) + & - a(nk-5,nk-5) * var(:,:,nk-5) + & - a(nk-4,nk-5) * var(:,:,nk-4) + & - a(nk-3,nk-5) * var(:,:,nk-3) + & - a(nk-2,nk-5) * var(:,:,nk-2) + & - a(nk-1,nk-5) * var(:,:,nk-1) + & - a(nk,nk-5) * var(:,:,nk) ) * idel - rhs(:,:,nk-4) = rhs(:,:,nk-4) + & - ( a(nk-9,nk-4) * var(:,:,nk-9) + & - a(nk-8,nk-4) * var(:,:,nk-8) + & - a(nk-7,nk-4) * var(:,:,nk-7) + & - 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-9,nk-3) * var(:,:,nk-9) + & - a(nk-8,nk-3) * var(:,:,nk-8) + & - a(nk-7,nk-3) * var(:,:,nk-7) + & - 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-9,nk-2) * var(:,:,nk-9) + & - a(nk-8,nk-2) * var(:,:,nk-8) + & - a(nk-7,nk-2) * var(:,:,nk-7) + & - 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-9,nk-1) * var(:,:,nk-9) + & - a(nk-8,nk-1) * var(:,:,nk-8) + & - a(nk-7,nk-1) * var(:,:,nk-7) + & - 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-3,nk) * var(:,:,nk-3) + & - a(nk-2,nk) * var(:,:,nk-2) + & - a(nk-1,nk) * var(:,:,nk-1) + & - a(nk,nk) * var(:,:,nk) ) * idel - - kr = nk - 7 - + if ( zero_derivs_z == 0 ) then + allocate ( a(nk,nk), d(nk,nk), b(nk,nk), h(nk,nk), tmp(nk,nk) ) + a = zero; d = zero; b = zero; h = zero; tmp = zero + + call set_bmatrix ( b, bb(5:6), lsh(3), gsh(3), lbnd(3), delta(3), dfl ) + + call set_hmatrix ( h, bb(5:6) ) + + if ( any ( bb(5:6) == 0 ) ) then + + call set_dmatrix ( d, bb(5:6) ) + + a = -transpose ( matmul ( h, matmul ( transpose(d), matmul ( b, d ) ) ) ) + + else + + if ( lsh(3) /= gsh(3) ) then + call CCTK_WARN(0, 'Internal error. Bounding box inconsistent with grid' ) + end if + + center = gsh(3) / 2 + kr = center + gsize(3) + + call set_dmatrix_half ( d(1:kr,1:kr), 1 ) + + tmp(1:kr,1:kr) = -transpose ( & + matmul ( h(1:kr,1:kr), & + matmul ( & + transpose ( d(1:kr,1:kr) ), & + matmul ( b(1:kr,1:kr), & + d(1:kr,1:kr) ) ) ) ) + + a(1:kr,1:center) = tmp(1:kr,1:center) + + kl = center + 1 - gsize(3) + d = zero + call set_dmatrix_half ( d(kl:nk,kl:nk), 2 ) + tmp(kl:nk,kl:nk) = -transpose ( & + matmul ( h(kl:nk,kl:nk), & + matmul ( & + transpose ( d(kl:nk,kl:nk) ), & + matmul ( b(kl:nk,kl:nk), & + d(kl:nk,kl:nk) ) ) ) ) + + a(kl:nk,center+1:nk) = tmp(kl:nk,center+1:nk) + + end if + + if ( scale_with_h > 0 ) then + idel = epsilon / ( 64 * delta(3) ) + else + idel = epsilon / 64 + end if + + 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) + a(6,2) * var(:,:,6) + & + a(7,2) * var(:,:,7) + a(8,2) * var(:,:,8) + & + a(9,2) * var(:,:,9) + a(10,2) * var(:,:,10) ) * 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) + a(8,3) * var(:,:,8) + & + a(9,3) * var(:,:,9) + a(10,3) * var(:,:,10) ) * 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) + a(8,4) * var(:,:,8) + & + a(9,4) * var(:,:,9) + a(10,4) * var(:,:,10) ) * 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) + a(8,5) * var(:,:,8) + & + a(9,5) * var(:,:,9) + a(10,5) * var(:,:,10) ) * idel + rhs(:,:,6) = rhs(:,:,6) + & + ( a(1,6) * var(:,:,1) + a(2,6) * var(:,:,2) + & + a(3,6) * var(:,:,3) + a(4,6) * var(:,:,4) + & + a(5,6) * var(:,:,5) + a(6,6) * var(:,:,6) + & + a(7,6) * var(:,:,7) + a(8,6) * var(:,:,8) + & + a(9,6) * var(:,:,9) + a(10,6) * var(:,:,10) ) * idel + rhs(:,:,7) = rhs(:,:,7) + & + ( a(1,7) * var(:,:,1) + a(2,7) * var(:,:,2) + & + a(3,7) * var(:,:,3) + a(4,7) * var(:,:,4) + & + a(5,7) * var(:,:,5) + a(6,7) * var(:,:,6) + & + a(7,7) * var(:,:,7) + a(8,7) * var(:,:,8) + & + a(9,7) * var(:,:,9) + a(10,7) * var(:,:,10) ) * idel + + kl = 8 + + end if + + if ( bb(6) == 0 ) then + + kr = nk - gsize(3) + + else + + rhs(:,:,nk-6) = rhs(:,:,nk-6) + & + ( a(nk-9,nk-6) * var(:,:,nk-9) + & + a(nk-8,nk-6) * var(:,:,nk-8) + & + a(nk-7,nk-6) * var(:,:,nk-7) + & + a(nk-6,nk-6) * var(:,:,nk-6) + & + a(nk-5,nk-6) * var(:,:,nk-5) + & + a(nk-4,nk-6) * var(:,:,nk-4) + & + a(nk-3,nk-6) * var(:,:,nk-3) + & + a(nk-2,nk-6) * var(:,:,nk-2) + & + a(nk-1,nk-6) * var(:,:,nk-1) + & + a(nk,nk-6) * var(:,:,nk) ) * idel + rhs(:,:,nk-5) = rhs(:,:,nk-5) + & + ( a(nk-9,nk-5) * var(:,:,nk-9) + & + a(nk-8,nk-5) * var(:,:,nk-8) + & + a(nk-7,nk-5) * var(:,:,nk-7) + & + a(nk-6,nk-5) * var(:,:,nk-6) + & + a(nk-5,nk-5) * var(:,:,nk-5) + & + a(nk-4,nk-5) * var(:,:,nk-4) + & + a(nk-3,nk-5) * var(:,:,nk-3) + & + a(nk-2,nk-5) * var(:,:,nk-2) + & + a(nk-1,nk-5) * var(:,:,nk-1) + & + a(nk,nk-5) * var(:,:,nk) ) * idel + rhs(:,:,nk-4) = rhs(:,:,nk-4) + & + ( a(nk-9,nk-4) * var(:,:,nk-9) + & + a(nk-8,nk-4) * var(:,:,nk-8) + & + a(nk-7,nk-4) * var(:,:,nk-7) + & + 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-9,nk-3) * var(:,:,nk-9) + & + a(nk-8,nk-3) * var(:,:,nk-8) + & + a(nk-7,nk-3) * var(:,:,nk-7) + & + 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-9,nk-2) * var(:,:,nk-9) + & + a(nk-8,nk-2) * var(:,:,nk-8) + & + a(nk-7,nk-2) * var(:,:,nk-7) + & + 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-9,nk-1) * var(:,:,nk-9) + & + a(nk-8,nk-1) * var(:,:,nk-8) + & + a(nk-7,nk-1) * var(:,:,nk-7) + & + 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-3,nk) * var(:,:,nk-3) + & + a(nk-2,nk) * var(:,:,nk-2) + & + a(nk-1,nk) * var(:,:,nk-1) + & + a(nk,nk) * var(:,:,nk) ) * idel + + kr = nk - 7 + + end if + + do k = kl, kr + + rhs(:,:,k) = rhs(:,:,k) + & + ( a(k-3,k) * var(:,:,k-3) + & + 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) + & + a(k+3,k) * var(:,:,k+3) ) * idel + + end do + + deallocate ( a, d, b, h, tmp ) end if - do k = kl, kr - - rhs(:,:,k) = rhs(:,:,k) + & - ( a(k-3,k) * var(:,:,k-3) + & - 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) + & - a(k+3,k) * var(:,:,k+3) ) * idel - - end do - - deallocate ( a, d, b, h, tmp ) - contains subroutine set_dmatrix ( d, bb ) |