diff options
-rw-r--r-- | param.ccl | 8 | ||||
-rw-r--r-- | src/Derivatives_2_1.F90 | 70 | ||||
-rw-r--r-- | src/Derivatives_4_2.F90 | 140 | ||||
-rw-r--r-- | src/Derivatives_4_3.F90 | 196 | ||||
-rw-r--r-- | src/Derivatives_4_3_min_err_coeff.F90 | 208 | ||||
-rw-r--r-- | src/Derivatives_6_3.F90 | 228 | ||||
-rw-r--r-- | src/Derivatives_6_3_min_err_coeff.F90 | 244 | ||||
-rw-r--r-- | src/Derivatives_6_5.F90 | 344 | ||||
-rw-r--r-- | src/Derivatives_6_5_min_err_coeff.F90 | 360 | ||||
-rw-r--r-- | src/Derivatives_8_4.F90 | 376 | ||||
-rw-r--r-- | src/Derivatives_8_4_min_err_coeff.F90 | 376 | ||||
-rw-r--r-- | src/Dissipation_2_1.F90 | 136 | ||||
-rw-r--r-- | src/Dissipation_2_1_alt.F90 | 136 | ||||
-rw-r--r-- | src/Dissipation_4_2.F90 | 216 | ||||
-rw-r--r-- | src/Dissipation_4_2_alt.F90 | 224 | ||||
-rw-r--r-- | src/Dissipation_4_3_min_err_coeff.F90 | 386 | ||||
-rw-r--r-- | src/Dissipation_4_3_min_err_coeff_alt.F90 | 592 | ||||
-rw-r--r-- | src/Dissipation_6_3.F90 | 320 | ||||
-rw-r--r-- | src/Dissipation_6_3_alt.F90 | 341 | ||||
-rw-r--r-- | src/Dissipation_6_5_min_err_coeff.F90 | 790 | ||||
-rw-r--r-- | src/Dissipation_6_5_min_err_coeff_alt.F90 | 698 | ||||
-rw-r--r-- | src/Dissipation_8_4.F90 | 453 | ||||
-rw-r--r-- | src/Dissipation_8_4_alt.F90 | 488 |
23 files changed, 3735 insertions, 3595 deletions
@@ -48,6 +48,14 @@ STRING vars "List of evolved grid functions that should have dissipation added" .* :: "Must be a valid list of grid functions" } "" +BOOLEAN zero_derivs_y "set all derivatives to 0 in the y-direction" +{ +} "no" + +BOOLEAN zero_derivs_z "set all derivatives to 0 in the z-direction" +{ +} "no" + # If you use this parameter be careful. You have to know what you are doing... BOOLEAN check_grid_sizes "Should we check grid sizes and ghost zones" { diff --git a/src/Derivatives_2_1.F90 b/src/Derivatives_2_1.F90 index 564e575..9720dd6 100644 --- a/src/Derivatives_2_1.F90 +++ b/src/Derivatives_2_1.F90 @@ -57,40 +57,50 @@ subroutine deriv_gf_2_1 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) end if dvar(il:ir,:,:) = a(1) * ( var(il+1:ir+1,:,:) - var(il-1:ir-1,:,:) ) * idel case (1) direction - if ( bb(1) == 0 ) then - jl = 1 + gsize - else - dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) ) * idel - dvar(:,2,:) = ( q(1,2) * var(:,1,:) + q(3,2) * var(:,3,:) ) * idel - jl = 3 - end if - if ( bb(2) == 0 ) then - jr = nj - gsize + if ( zero_derivs_y /= 0 ) then + dvar = zero else - dvar(:,nj-1,:) = - ( q(1,2) * var(:,nj,:) + & - q(3,2) * var(:,nj-2,:) ) * idel - dvar(:,nj,:) = - ( q(1,1) * var(:,nj,:) + & - q(2,1) * var(:,nj-1,:) ) * idel - jr = nj - 2 + if ( bb(1) == 0 ) then + jl = 1 + gsize + else + dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) ) * idel + dvar(:,2,:) = ( q(1,2) * var(:,1,:) + q(3,2) * var(:,3,:) ) * idel + jl = 3 + end if + if ( bb(2) == 0 ) then + jr = nj - gsize + else + dvar(:,nj-1,:) = - ( q(1,2) * var(:,nj,:) + & + q(3,2) * var(:,nj-2,:) ) * idel + dvar(:,nj,:) = - ( q(1,1) * var(:,nj,:) + & + q(2,1) * var(:,nj-1,:) ) * idel + jr = nj - 2 + end if + dvar(:,jl:jr,:) = a(1) * ( var(:,jl+1:jr+1,:) - & + var(:,jl-1:jr-1,:) ) * idel end if - dvar(:,jl:jr,:) = a(1) * ( var(:,jl+1:jr+1,:) - var(:,jl-1:jr-1,:) ) * idel case (2) direction - if ( bb(1) == 0 ) then - kl = 1 + gsize + if ( zero_derivs_z /= 0 ) then + dvar = zero else - dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) ) * idel - dvar(:,:,2) = ( q(1,2) * var(:,:,1) + q(3,2) * var(:,:,3) ) * idel - kl = 3 - end if - if ( bb(2) == 0 ) then - kr = nk - gsize - else - dvar(:,:,nk-1) = - ( q(1,2) * var(:,:,nk) + & - q(3,2) * var(:,:,nk-2) ) * idel - dvar(:,:,nk) = - ( q(1,1) * var(:,:,nk) + & - q(2,1) * var(:,:,nk-1) ) * idel - kr = nk - 2 + if ( bb(1) == 0 ) then + kl = 1 + gsize + else + dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) ) * idel + dvar(:,:,2) = ( q(1,2) * var(:,:,1) + q(3,2) * var(:,:,3) ) * idel + kl = 3 + end if + if ( bb(2) == 0 ) then + kr = nk - gsize + else + dvar(:,:,nk-1) = - ( q(1,2) * var(:,:,nk) + & + q(3,2) * var(:,:,nk-2) ) * idel + dvar(:,:,nk) = - ( q(1,1) * var(:,:,nk) + & + q(2,1) * var(:,:,nk-1) ) * idel + kr = nk - 2 + end if + dvar(:,:,kl:kr) = a(1) * ( var(:,:,kl+1:kr+1) - & + var(:,:,kl-1:kr-1) ) * idel end if - dvar(:,:,kl:kr) = a(1) * ( var(:,:,kl+1:kr+1) - var(:,:,kl-1:kr-1) ) * idel end select direction end subroutine deriv_gf_2_1 diff --git a/src/Derivatives_4_2.F90 b/src/Derivatives_4_2.F90 index ec56613..7fc40b1 100644 --- a/src/Derivatives_4_2.F90 +++ b/src/Derivatives_4_2.F90 @@ -85,76 +85,84 @@ subroutine deriv_gf_4_2 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) a(2) * ( var(il+2:ir+2,:,:) - & var(il-2:ir-2,:,:) ) ) * idel case (1) direction - if ( bb(1) == 0 ) then - jl = 1 + gsize - else - 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(3,2) * var(:,3,:) ) * idel - dvar(:,3,:) = ( q(1,3) * var(:,1,:) + q(2,3) * var(:,2,:) + & - q(4,3) * var(:,4,:) + q(5,3) * var(:,5,:) ) * idel - dvar(:,4,:) = ( q(1,4) * var(:,1,:) + q(3,4) * var(:,3,:) + & - q(5,4) * var(:,5,:) + q(6,4) * var(:,6,:) ) * idel - jl = 5 - end if - if ( bb(2) == 0 ) then - jr = nj - gsize + if ( zero_derivs_y /= 0 ) then + dvar = zero else - dvar(:,nj-3,:) = - ( q(1,4) * var(:,nj,:) + & - q(3,4) * var(:,nj-2,:) + & - q(5,4) * var(:,nj-4,:) + & - q(6,4) * var(:,nj-5,:) ) * idel - dvar(:,nj-2,:) = - ( q(1,3) * var(:,nj,:) + & - q(2,3) * var(:,nj-1,:) + & - q(4,3) * var(:,nj-3,:) + & - q(5,3) * var(:,nj-4,:) ) * idel - dvar(:,nj-1,:) = - ( q(1,2) * var(:,nj,:) + & - q(3,2) * var(:,nj-2,:) ) * 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 - jr = nj - 4 + if ( bb(1) == 0 ) then + jl = 1 + gsize + else + 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(3,2) * var(:,3,:) ) * idel + dvar(:,3,:) = ( q(1,3) * var(:,1,:) + q(2,3) * var(:,2,:) + & + q(4,3) * var(:,4,:) + q(5,3) * var(:,5,:) ) * idel + dvar(:,4,:) = ( q(1,4) * var(:,1,:) + q(3,4) * var(:,3,:) + & + q(5,4) * var(:,5,:) + q(6,4) * var(:,6,:) ) * idel + jl = 5 + end if + if ( bb(2) == 0 ) then + jr = nj - gsize + else + dvar(:,nj-3,:) = - ( q(1,4) * var(:,nj,:) + & + q(3,4) * var(:,nj-2,:) + & + q(5,4) * var(:,nj-4,:) + & + q(6,4) * var(:,nj-5,:) ) * idel + dvar(:,nj-2,:) = - ( q(1,3) * var(:,nj,:) + & + q(2,3) * var(:,nj-1,:) + & + q(4,3) * var(:,nj-3,:) + & + q(5,3) * var(:,nj-4,:) ) * idel + dvar(:,nj-1,:) = - ( q(1,2) * var(:,nj,:) + & + q(3,2) * var(:,nj-2,:) ) * 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 + jr = nj - 4 + end if + dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & + var(:,jl-1:jr-1,:) ) + & + a(2) * ( var(:,jl+2:jr+2,:) - & + var(:,jl-2:jr-2,:) ) ) * idel end if - dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & - var(:,jl-1:jr-1,:) ) + & - a(2) * ( var(:,jl+2:jr+2,:) - & - var(:,jl-2:jr-2,:) ) ) * idel case (2) direction - if ( bb(1) == 0 ) then - kl = 1 + gsize + if ( zero_derivs_z /= 0 ) then + dvar = zero else - 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(3,2) * var(:,:,3) ) * idel - dvar(:,:,3) = ( q(1,3) * var(:,:,1) + q(2,3) * var(:,:,2) + & - q(4,3) * var(:,:,4) + q(5,3) * var(:,:,5) ) * idel - dvar(:,:,4) = ( q(1,4) * var(:,:,1) + q(3,4) * var(:,:,3) + & - q(5,4) * var(:,:,5) + q(6,4) * var(:,:,6) ) * idel - kl = 5 - end if - if ( bb(2) == 0 ) then - kr = nk - gsize - else - dvar(:,:,nk-3) = - ( q(1,4) * var(:,:,nk) + & - q(3,4) * var(:,:,nk-2) + & - q(5,4) * var(:,:,nk-4) + & - q(6,4) * var(:,:,nk-5) ) * idel - dvar(:,:,nk-2) = - ( q(1,3) * var(:,:,nk) + & - q(2,3) * var(:,:,nk-1) + & - q(4,3) * var(:,:,nk-3) + & - q(5,3) * var(:,:,nk-4) ) * idel - dvar(:,:,nk-1) = - ( q(1,2) * var(:,:,nk) + & - q(3,2) * var(:,:,nk-2) ) * 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 - kr = nk - 4 + if ( bb(1) == 0 ) then + kl = 1 + gsize + else + 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(3,2) * var(:,:,3) ) * idel + dvar(:,:,3) = ( q(1,3) * var(:,:,1) + q(2,3) * var(:,:,2) + & + q(4,3) * var(:,:,4) + q(5,3) * var(:,:,5) ) * idel + dvar(:,:,4) = ( q(1,4) * var(:,:,1) + q(3,4) * var(:,:,3) + & + q(5,4) * var(:,:,5) + q(6,4) * var(:,:,6) ) * idel + kl = 5 + end if + if ( bb(2) == 0 ) then + kr = nk - gsize + else + dvar(:,:,nk-3) = - ( q(1,4) * var(:,:,nk) + & + q(3,4) * var(:,:,nk-2) + & + q(5,4) * var(:,:,nk-4) + & + q(6,4) * var(:,:,nk-5) ) * idel + dvar(:,:,nk-2) = - ( q(1,3) * var(:,:,nk) + & + q(2,3) * var(:,:,nk-1) + & + q(4,3) * var(:,:,nk-3) + & + q(5,3) * var(:,:,nk-4) ) * idel + dvar(:,:,nk-1) = - ( q(1,2) * var(:,:,nk) + & + q(3,2) * var(:,:,nk-2) ) * 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 + kr = nk - 4 + end if + dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & + var(:,:,kl-1:kr-1) ) + & + a(2) * ( var(:,:,kl+2:kr+2) - & + var(:,:,kl-2:kr-2) ) ) * idel end if - dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & - var(:,:,kl-1:kr-1) ) + & - a(2) * ( var(:,:,kl+2:kr+2) - & - var(:,:,kl-2:kr-2) ) ) * idel end select direction end subroutine deriv_gf_4_2 diff --git a/src/Derivatives_4_3.F90 b/src/Derivatives_4_3.F90 index 011a074..db7ebda 100644 --- a/src/Derivatives_4_3.F90 +++ b/src/Derivatives_4_3.F90 @@ -139,104 +139,112 @@ subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) a(2) * ( var(il+2:ir+2,:,:) - & var(il-2:ir-2,:,:) ) ) * idel case (1) direction - if ( bb(1) == 0 ) then - jl = 1 + gsize - else - 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 - jl = 6 - end if - if ( bb(2) == 0 ) then - jr = nj - gsize + if ( zero_derivs_y /= 0 ) then + dvar = zero else - 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 - jr = nj - 5 + if ( bb(1) == 0 ) then + jl = 1 + gsize + else + 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 + jl = 6 + end if + if ( bb(2) == 0 ) then + jr = nj - gsize + else + 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 + jr = nj - 5 + end if + dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & + var(:,jl-1:jr-1,:) ) + & + a(2) * ( var(:,jl+2:jr+2,:) - & + var(:,jl-2:jr-2,:) ) ) * idel end if - dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & - var(:,jl-1:jr-1,:) ) + & - a(2) * ( var(:,jl+2:jr+2,:) - & - var(:,jl-2:jr-2,:) ) ) * idel case (2) direction - if ( bb(1) == 0 ) then - kl = 1 + gsize + if ( zero_derivs_z /= 0 ) then + dvar = zero else - 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 - kl = 6 - end if - if ( bb(2) == 0 ) then - kr = nk - gsize - else - 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 - kr = nk - 5 + if ( bb(1) == 0 ) then + kl = 1 + gsize + else + 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 + kl = 6 + end if + if ( bb(2) == 0 ) then + kr = nk - gsize + else + 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 + kr = nk - 5 + end if + dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & + var(:,:,kl-1:kr-1) ) + & + a(2) * ( var(:,:,kl+2:kr+2) - & + var(:,:,kl-2:kr-2) ) ) * idel end if - dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & - var(:,:,kl-1:kr-1) ) + & - a(2) * ( var(:,:,kl+2:kr+2) - & - var(:,:,kl-2:kr-2) ) ) * idel end select direction end subroutine deriv_gf_4_3 diff --git a/src/Derivatives_4_3_min_err_coeff.F90 b/src/Derivatives_4_3_min_err_coeff.F90 index 938c4bc..659d6a4 100644 --- a/src/Derivatives_4_3_min_err_coeff.F90 +++ b/src/Derivatives_4_3_min_err_coeff.F90 @@ -126,110 +126,118 @@ subroutine deriv_gf_4_3_opt ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) a(2) * ( var(il+2:ir+2,:,:) - & var(il-2:ir-2,:,:) ) ) * idel case (1) direction - if ( bb(1) == 0 ) then - jl = 1 + gsize - else - dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & - q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) + & - q(5,1) * var(:,5,:) ) * 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,:) + & - q(7,2) * var(:,7,:) ) * 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,:) + & - q(7,3) * var(:,7,:) ) * 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 - jl = 6 - end if - if ( bb(2) == 0 ) then - jr = nj - gsize + if ( zero_derivs_y /= 0 ) then + dvar = zero else - 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,:) + & - q(7,3) * var(:,nj-6,:) ) * 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,:) + & - q(7,2) * var(:,nj-6,:) ) * 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,:) + & - q(5,1) * var(:,nj-4,:) ) * idel - jr = nj - 5 + if ( bb(1) == 0 ) then + jl = 1 + gsize + else + dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & + q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) + & + q(5,1) * var(:,5,:) ) * 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,:) + & + q(7,2) * var(:,7,:) ) * 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,:) + & + q(7,3) * var(:,7,:) ) * 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 + jl = 6 + end if + if ( bb(2) == 0 ) then + jr = nj - gsize + else + 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,:) + & + q(7,3) * var(:,nj-6,:) ) * 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,:) + & + q(7,2) * var(:,nj-6,:) ) * 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,:) + & + q(5,1) * var(:,nj-4,:) ) * idel + jr = nj - 5 + end if + dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & + var(:,jl-1:jr-1,:) ) + & + a(2) * ( var(:,jl+2:jr+2,:) - & + var(:,jl-2:jr-2,:) ) ) * idel end if - dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & - var(:,jl-1:jr-1,:) ) + & - a(2) * ( var(:,jl+2:jr+2,:) - & - var(:,jl-2:jr-2,:) ) ) * idel case (2) direction - if ( bb(1) == 0 ) then - kl = 1 + gsize + if ( zero_derivs_z /= 0 ) then + dvar = zero else - dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & - q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) + & - q(5,1) * var(:,:,5) ) * 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) + & - q(7,2) * var(:,:,7) ) * 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) + & - q(7,3) * var(:,:,7) ) * 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 - kl = 6 - end if - if ( bb(2) == 0 ) then - kr = nk - gsize - else - 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) + & - q(7,3) * var(:,:,nk-6) ) * 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) + & - q(7,2) * var(:,:,nk-6) ) * 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) + & - q(5,1) * var(:,:,nk-4) ) * idel - kr = nk - 5 + if ( bb(1) == 0 ) then + kl = 1 + gsize + else + dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & + q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) + & + q(5,1) * var(:,:,5) ) * 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) + & + q(7,2) * var(:,:,7) ) * 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) + & + q(7,3) * var(:,:,7) ) * 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 + kl = 6 + end if + if ( bb(2) == 0 ) then + kr = nk - gsize + else + 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) + & + q(7,3) * var(:,:,nk-6) ) * 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) + & + q(7,2) * var(:,:,nk-6) ) * 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) + & + q(5,1) * var(:,:,nk-4) ) * idel + kr = nk - 5 + end if + dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & + var(:,:,kl-1:kr-1) ) + & + a(2) * ( var(:,:,kl+2:kr+2) - & + var(:,:,kl-2:kr-2) ) ) * idel end if - dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & - var(:,:,kl-1:kr-1) ) + & - a(2) * ( var(:,:,kl+2:kr+2) - & - var(:,:,kl-2:kr-2) ) ) * idel end select direction end subroutine deriv_gf_4_3_opt diff --git a/src/Derivatives_6_3.F90 b/src/Derivatives_6_3.F90 index 3da882d..4c8643a 100644 --- a/src/Derivatives_6_3.F90 +++ b/src/Derivatives_6_3.F90 @@ -120,120 +120,128 @@ subroutine deriv_gf_6_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) a(3) * ( var(il+3:ir+3,:,:) - & var(il-3:ir-3,:,:) ) ) * idel case (1) direction - if ( bb(1) == 0 ) then - jl = 1 + gsize - else - dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & - q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) + & - q(5,1) * var(:,5,:) ) * idel - dvar(:,2,:) = ( q(1,2) * var(:,1,:) + 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(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(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(6,5) * var(:,6,:) + q(7,5) * var(:,7,:) + & - q(8,5) * var(:,8,:) ) * idel - dvar(:,6,:) = ( q(2,6) * var(:,2,:) + q(3,6) * var(:,3,:) + & - q(4,6) * var(:,4,:) + q(5,6) * var(:,5,:) + & - q(7,6) * var(:,7,:) + q(8,6) * var(:,8,:) + & - q(9,6) * var(:,9,:) ) * idel - jl = 7 - end if - if ( bb(2) == 0 ) then - jr = nj - gsize + if ( zero_derivs_y /= 0 ) then + dvar = zero else - 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,:) + & - q(5,1) * var(:,nj-4,:) ) * idel - dvar(:,nj-1,:) = - ( q(1,2) * var(:,nj,:) + 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-2,:) = - ( q(1,3) * var(:,nj,:) + q(2,3) * var(:,nj-1,:) + & - q(4,3) * var(:,nj-3,:) + q(5,3) * var(:,nj-4,:) + & - q(6,3) * var(:,nj-5,:) ) * idel - dvar(:,nj-3,:) = - ( q(1,4) * var(:,nj,:) + q(2,4) * var(:,nj-1,:) + & - q(3,4) * var(:,nj-2,:) + q(5,4) * var(:,nj-4,:) + & - q(6,4) * var(:,nj-5,:) + & - q(7,4) * var(:,nj-6,:) ) * idel - 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(6,5) * var(:,nj-5,:) + q(7,5) * var(:,nj-6,:) + & - q(8,5) * var(:,nj-7,:) ) * idel - dvar(:,nj-5,:) = - ( q(2,6) * var(:,nj-1,:) + q(3,6) * var(:,nj-2,:) + & - q(4,6) * var(:,nj-3,:) + q(5,6) * var(:,nj-4,:) + & - q(7,6) * var(:,nj-6,:) + q(8,6) * var(:,nj-7,:) + & - q(9,6) * var(:,nj-8,:) ) * idel - jr = nj - 6 + if ( bb(1) == 0 ) then + jl = 1 + gsize + else + dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & + q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) + & + q(5,1) * var(:,5,:) ) * idel + dvar(:,2,:) = ( q(1,2) * var(:,1,:) + 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(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(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(6,5) * var(:,6,:) + q(7,5) * var(:,7,:) + & + q(8,5) * var(:,8,:) ) * idel + dvar(:,6,:) = ( q(2,6) * var(:,2,:) + q(3,6) * var(:,3,:) + & + q(4,6) * var(:,4,:) + q(5,6) * var(:,5,:) + & + q(7,6) * var(:,7,:) + q(8,6) * var(:,8,:) + & + q(9,6) * var(:,9,:) ) * idel + jl = 7 + end if + if ( bb(2) == 0 ) then + jr = nj - gsize + else + 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,:) + & + q(5,1) * var(:,nj-4,:) ) * idel + dvar(:,nj-1,:) = - ( q(1,2) * var(:,nj,:) + 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-2,:) = - ( q(1,3) * var(:,nj,:) + q(2,3) * var(:,nj-1,:) + & + q(4,3) * var(:,nj-3,:) + q(5,3) * var(:,nj-4,:) + & + q(6,3) * var(:,nj-5,:) ) * idel + dvar(:,nj-3,:) = - ( q(1,4) * var(:,nj,:) + q(2,4) * var(:,nj-1,:) + & + q(3,4) * var(:,nj-2,:) + q(5,4) * var(:,nj-4,:) + & + q(6,4) * var(:,nj-5,:) + & + q(7,4) * var(:,nj-6,:) ) * idel + 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(6,5) * var(:,nj-5,:) + q(7,5) * var(:,nj-6,:) + & + q(8,5) * var(:,nj-7,:) ) * idel + dvar(:,nj-5,:) = - ( q(2,6) * var(:,nj-1,:) + q(3,6) * var(:,nj-2,:) + & + q(4,6) * var(:,nj-3,:) + q(5,6) * var(:,nj-4,:) + & + q(7,6) * var(:,nj-6,:) + q(8,6) * var(:,nj-7,:) + & + q(9,6) * var(:,nj-8,:) ) * idel + jr = nj - 6 + end if + dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & + var(:,jl-1:jr-1,:) ) + & + a(2) * ( var(:,jl+2:jr+2,:) - & + var(:,jl-2:jr-2,:) ) + & + a(3) * ( var(:,jl+3:jr+3,:) - & + var(:,jl-3:jr-3,:) ) ) * idel end if - dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & - var(:,jl-1:jr-1,:) ) + & - a(2) * ( var(:,jl+2:jr+2,:) - & - var(:,jl-2:jr-2,:) ) + & - a(3) * ( var(:,jl+3:jr+3,:) - & - var(:,jl-3:jr-3,:) ) ) * idel case (2) direction - if ( bb(1) == 0 ) then - kl = 1 + gsize + if ( zero_derivs_z /= 0 ) then + dvar = zero else - dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & - q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) + & - q(5,1) * var(:,:,5) ) * idel - dvar(:,:,2) = ( q(1,2) * var(:,:,1) + 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(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(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(6,5) * var(:,:,6) + q(7,5) * var(:,:,7) + & - q(8,5) * var(:,:,8) ) * idel - dvar(:,:,6) = ( q(2,6) * var(:,:,2) + q(3,6) * var(:,:,3) + & - q(4,6) * var(:,:,4) + q(5,6) * var(:,:,5) + & - q(7,6) * var(:,:,7) + q(8,6) * var(:,:,8) + & - q(9,6) * var(:,:,9) ) * idel - kl = 7 - end if - if ( bb(2) == 0 ) then - kr = nk - gsize - else - 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) + & - q(5,1) * var(:,:,nk-4) ) * idel - dvar(:,:,nk-1) = - ( q(1,2) * var(:,:,nk) + 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-2) = - ( q(1,3) * var(:,:,nk) + q(2,3) * var(:,:,nk-1) + & - q(4,3) * var(:,:,nk-3) + q(5,3) * var(:,:,nk-4) + & - q(6,3) * var(:,:,nk-5) ) * idel - dvar(:,:,nk-3) = - ( q(1,4) * var(:,:,nk) + q(2,4) * var(:,:,nk-1) + & - q(3,4) * var(:,:,nk-2) + q(5,4) * var(:,:,nk-4) + & - q(6,4) * var(:,:,nk-5) + & - q(7,4) * var(:,:,nk-6) ) * idel - 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(6,5) * var(:,:,nk-5) + q(7,5) * var(:,:,nk-6) + & - q(8,5) * var(:,:,nk-7) ) * idel - dvar(:,:,nk-5) = - ( q(2,6) * var(:,:,nk-1) + q(3,6) * var(:,:,nk-2) + & - q(4,6) * var(:,:,nk-3) + q(5,6) * var(:,:,nk-4) + & - q(7,6) * var(:,:,nk-6) + q(8,6) * var(:,:,nk-7) + & - q(9,6) * var(:,:,nk-8) ) * idel - kr = nk - 6 + if ( bb(1) == 0 ) then + kl = 1 + gsize + else + dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & + q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) + & + q(5,1) * var(:,:,5) ) * idel + dvar(:,:,2) = ( q(1,2) * var(:,:,1) + 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(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(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(6,5) * var(:,:,6) + q(7,5) * var(:,:,7) + & + q(8,5) * var(:,:,8) ) * idel + dvar(:,:,6) = ( q(2,6) * var(:,:,2) + q(3,6) * var(:,:,3) + & + q(4,6) * var(:,:,4) + q(5,6) * var(:,:,5) + & + q(7,6) * var(:,:,7) + q(8,6) * var(:,:,8) + & + q(9,6) * var(:,:,9) ) * idel + kl = 7 + end if + if ( bb(2) == 0 ) then + kr = nk - gsize + else + 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) + & + q(5,1) * var(:,:,nk-4) ) * idel + dvar(:,:,nk-1) = - ( q(1,2) * var(:,:,nk) + 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-2) = - ( q(1,3) * var(:,:,nk) + q(2,3) * var(:,:,nk-1) + & + q(4,3) * var(:,:,nk-3) + q(5,3) * var(:,:,nk-4) + & + q(6,3) * var(:,:,nk-5) ) * idel + dvar(:,:,nk-3) = - ( q(1,4) * var(:,:,nk) + q(2,4) * var(:,:,nk-1) + & + q(3,4) * var(:,:,nk-2) + q(5,4) * var(:,:,nk-4) + & + q(6,4) * var(:,:,nk-5) + & + q(7,4) * var(:,:,nk-6) ) * idel + 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(6,5) * var(:,:,nk-5) + q(7,5) * var(:,:,nk-6) + & + q(8,5) * var(:,:,nk-7) ) * idel + dvar(:,:,nk-5) = - ( q(2,6) * var(:,:,nk-1) + q(3,6) * var(:,:,nk-2) + & + q(4,6) * var(:,:,nk-3) + q(5,6) * var(:,:,nk-4) + & + q(7,6) * var(:,:,nk-6) + q(8,6) * var(:,:,nk-7) + & + q(9,6) * var(:,:,nk-8) ) * idel + kr = nk - 6 + end if + dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & + var(:,:,kl-1:kr-1) ) + & + a(2) * ( var(:,:,kl+2:kr+2) - & + var(:,:,kl-2:kr-2) ) + & + a(3) * ( var(:,:,kl+3:kr+3) - & + var(:,:,kl-3:kr-3) ) ) * idel end if - dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & - var(:,:,kl-1:kr-1) ) + & - a(2) * ( var(:,:,kl+2:kr+2) - & - var(:,:,kl-2:kr-2) ) + & - a(3) * ( var(:,:,kl+3:kr+3) - & - var(:,:,kl-3:kr-3) ) ) * idel end select direction end subroutine deriv_gf_6_3 diff --git a/src/Derivatives_6_3_min_err_coeff.F90 b/src/Derivatives_6_3_min_err_coeff.F90 index 603c66c..703eb4e 100644 --- a/src/Derivatives_6_3_min_err_coeff.F90 +++ b/src/Derivatives_6_3_min_err_coeff.F90 @@ -154,128 +154,136 @@ subroutine deriv_gf_6_3_opt ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) a(3) * ( var(il+3:ir+3,:,:) - & var(il-3:ir-3,:,:) ) ) * idel case (1) direction - if ( bb(1) == 0 ) then - jl = 1 + gsize - else - dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & - q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) + & - q(5,1) * var(:,5,:) + q(6,1) * var(:,6,:) ) * idel - dvar(:,2,:) = ( q(1,2) * var(:,1,:) + 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(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(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(6,5) * var(:,6,:) + q(7,5) * var(:,7,:) + & - q(8,5) * var(:,8,:) ) * idel - dvar(:,6,:) = ( q(1,6) * var(:,1,:) + q(2,6) * var(:,2,:) + & - q(3,6) * var(:,3,:) + q(4,6) * var(:,4,:) + & - q(5,6) * var(:,5,:) + q(7,6) * var(:,7,:) + & - q(8,6) * var(:,8,:) + q(9,6) * var(:,9,:) ) * idel - - jl = 7 - end if - if ( bb(2) == 0 ) then - jr = nj - gsize + if ( zero_derivs_y /= 0 ) then + dvar = zero else - dvar(:,nj-5,:) = - ( q(1,6) * var(:,nj,:) + q(2,6) * var(:,nj-1,:) + & - q(3,6) * var(:,nj-2,:) + q(4,6) * var(:,nj-3,:) + & - q(5,6) * var(:,nj-4,:) + q(7,6) * var(:,nj-6,:) + & - q(8,6) * var(:,nj-7,:) + & - q(9,6) * var(:,nj-8,:) ) * idel - 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(6,5) * var(:,nj-5,:) + q(7,5) * var(:,nj-6,:) + & - q(8,5) * var(:,nj-7,:) ) * idel - dvar(:,nj-3,:) = - ( q(1,4) * var(:,nj,:) + q(2,4) * var(:,nj-1,:) + & - q(3,4) * var(:,nj-2,:) + 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(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(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,:) + & - q(5,1) * var(:,nj-4,:) + & - q(6,1) * var(:,nj-5,:) ) * idel - - jr = nj - 6 + if ( bb(1) == 0 ) then + jl = 1 + gsize + else + dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & + q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) + & + q(5,1) * var(:,5,:) + q(6,1) * var(:,6,:) ) * idel + dvar(:,2,:) = ( q(1,2) * var(:,1,:) + 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(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(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(6,5) * var(:,6,:) + q(7,5) * var(:,7,:) + & + q(8,5) * var(:,8,:) ) * idel + dvar(:,6,:) = ( q(1,6) * var(:,1,:) + q(2,6) * var(:,2,:) + & + q(3,6) * var(:,3,:) + q(4,6) * var(:,4,:) + & + q(5,6) * var(:,5,:) + q(7,6) * var(:,7,:) + & + q(8,6) * var(:,8,:) + q(9,6) * var(:,9,:) ) * idel + + jl = 7 + end if + if ( bb(2) == 0 ) then + jr = nj - gsize + else + dvar(:,nj-5,:) = - ( q(1,6) * var(:,nj,:) + q(2,6) * var(:,nj-1,:) + & + q(3,6) * var(:,nj-2,:) + q(4,6) * var(:,nj-3,:) + & + q(5,6) * var(:,nj-4,:) + q(7,6) * var(:,nj-6,:) + & + q(8,6) * var(:,nj-7,:) + & + q(9,6) * var(:,nj-8,:) ) * idel + 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(6,5) * var(:,nj-5,:) + q(7,5) * var(:,nj-6,:) + & + q(8,5) * var(:,nj-7,:) ) * idel + dvar(:,nj-3,:) = - ( q(1,4) * var(:,nj,:) + q(2,4) * var(:,nj-1,:) + & + q(3,4) * var(:,nj-2,:) + 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(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(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,:) + & + q(5,1) * var(:,nj-4,:) + & + q(6,1) * var(:,nj-5,:) ) * idel + + jr = nj - 6 + end if + dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & + var(:,jl-1:jr-1,:) ) + & + a(2) * ( var(:,jl+2:jr+2,:) - & + var(:,jl-2:jr-2,:) ) + & + a(3) * ( var(:,jl+3:jr+3,:) - & + var(:,jl-3:jr-3,:) ) ) * idel end if - dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & - var(:,jl-1:jr-1,:) ) + & - a(2) * ( var(:,jl+2:jr+2,:) - & - var(:,jl-2:jr-2,:) ) + & - a(3) * ( var(:,jl+3:jr+3,:) - & - var(:,jl-3:jr-3,:) ) ) * idel case (2) direction - if ( bb(1) == 0 ) then - kl = 1 + gsize + if ( zero_derivs_z /= 0 ) then + dvar = zero else - dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & - q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) + & - q(5,1) * var(:,:,5) + q(6,1) * var(:,:,6) ) * idel - dvar(:,:,2) = ( q(1,2) * var(:,:,1) + 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(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(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(6,5) * var(:,:,6) + q(7,5) * var(:,:,7) + & - q(8,5) * var(:,:,8) ) * idel - dvar(:,:,6) = ( q(1,6) * var(:,:,1) + q(2,6) * var(:,:,2) + & - q(3,6) * var(:,:,3) + q(4,6) * var(:,:,4) + & - q(5,6) * var(:,:,5) + q(7,6) * var(:,:,7) + & - q(8,6) * var(:,:,8) + q(9,6) * var(:,:,9) ) * idel - - kl = 7 - end if - if ( bb(2) == 0 ) then - kr = nk - gsize - else - dvar(:,:,nk-5) = - ( q(1,6) * var(:,:,nk) + q(2,6) * var(:,:,nk-1) + & - q(3,6) * var(:,:,nk-2) + q(4,6) * var(:,:,nk-3) + & - q(5,6) * var(:,:,nk-4) + q(7,6) * var(:,:,nk-6) + & - q(8,6) * var(:,:,nk-7) + & - q(9,6) * var(:,:,nk-8) ) * idel - 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(6,5) * var(:,:,nk-5) + q(7,5) * var(:,:,nk-6) + & - q(8,5) * var(:,:,nk-7) ) * idel - dvar(:,:,nk-3) = - ( q(1,4) * var(:,:,nk) + q(2,4) * var(:,:,nk-1) + & - q(3,4) * var(:,:,nk-2) + 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(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(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) + & - q(5,1) * var(:,:,nk-4) + & - q(6,1) * var(:,:,nk-5) ) * idel - - kr = nk - 6 + if ( bb(1) == 0 ) then + kl = 1 + gsize + else + dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & + q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) + & + q(5,1) * var(:,:,5) + q(6,1) * var(:,:,6) ) * idel + dvar(:,:,2) = ( q(1,2) * var(:,:,1) + 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(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(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(6,5) * var(:,:,6) + q(7,5) * var(:,:,7) + & + q(8,5) * var(:,:,8) ) * idel + dvar(:,:,6) = ( q(1,6) * var(:,:,1) + q(2,6) * var(:,:,2) + & + q(3,6) * var(:,:,3) + q(4,6) * var(:,:,4) + & + q(5,6) * var(:,:,5) + q(7,6) * var(:,:,7) + & + q(8,6) * var(:,:,8) + q(9,6) * var(:,:,9) ) * idel + + kl = 7 + end if + if ( bb(2) == 0 ) then + kr = nk - gsize + else + dvar(:,:,nk-5) = - ( q(1,6) * var(:,:,nk) + q(2,6) * var(:,:,nk-1) + & + q(3,6) * var(:,:,nk-2) + q(4,6) * var(:,:,nk-3) + & + q(5,6) * var(:,:,nk-4) + q(7,6) * var(:,:,nk-6) + & + q(8,6) * var(:,:,nk-7) + & + q(9,6) * var(:,:,nk-8) ) * idel + 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(6,5) * var(:,:,nk-5) + q(7,5) * var(:,:,nk-6) + & + q(8,5) * var(:,:,nk-7) ) * idel + dvar(:,:,nk-3) = - ( q(1,4) * var(:,:,nk) + q(2,4) * var(:,:,nk-1) + & + q(3,4) * var(:,:,nk-2) + 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(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(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) + & + q(5,1) * var(:,:,nk-4) + & + q(6,1) * var(:,:,nk-5) ) * idel + + kr = nk - 6 + end if + dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & + var(:,:,kl-1:kr-1) ) + & + a(2) * ( var(:,:,kl+2:kr+2) - & + var(:,:,kl-2:kr-2) ) + & + a(3) * ( var(:,:,kl+3:kr+3) - & + var(:,:,kl-3:kr-3) ) ) * idel end if - dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & - var(:,:,kl-1:kr-1) ) + & - a(2) * ( var(:,:,kl+2:kr+2) - & - var(:,:,kl-2:kr-2) ) + & - a(3) * ( var(:,:,kl+3:kr+3) - & - var(:,:,kl-3:kr-3) ) ) * idel end select direction end subroutine deriv_gf_6_3_opt diff --git a/src/Derivatives_6_5.F90 b/src/Derivatives_6_5.F90 index 411bd59..09e2ac4 100644 --- a/src/Derivatives_6_5.F90 +++ b/src/Derivatives_6_5.F90 @@ -193,178 +193,186 @@ subroutine deriv_gf_6_5 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) a(3) * ( var(il+3:ir+3,:,:) - & var(il-3:ir-3,:,:) ) ) * idel case (1) direction - if ( bb(1) == 0 ) then - jl = 1 + gsize - else - dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & - q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) + & - q(5,1) * var(:,5,:) + q(6,1) * var(:,6,:) ) * 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,:) + & - q(7,2) * var(:,7,:) + q(8,2) * var(:,8,:) ) * 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,:) + & - q(7,3) * var(:,7,:) + q(8,3) * var(:,8,:) + & - q(9,3) * var(:,9,:) ) * 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,:) + q(8,4) * var(:,8,:) + & - q(9,4) * var(:,9,:) + q(10,4) * var(:,10,:) ) * 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,:) + q(8,5) * var(:,8,:) + & - q(9,5) * var(:,9,:) + q(10,5) * var(:,10,:) ) * idel - dvar(:,6,:) = ( q(1,6) * var(:,1,:) + q(2,6) * var(:,2,:) + & - q(3,6) * var(:,3,:) + q(4,6) * var(:,4,:) + & - q(5,6) * var(:,5,:) + q(6,6) * var(:,6,:) + & - q(7,6) * var(:,7,:) + q(8,6) * var(:,8,:) + & - q(9,6) * var(:,9,:) + q(10,6) * var(:,10,:) ) * idel - dvar(:,7,:) = ( q(1,7) * var(:,1,:) + q(2,7) * var(:,2,:) + & - q(3,7) * var(:,3,:) + q(4,7) * var(:,4,:) + & - q(5,7) * var(:,5,:) + q(6,7) * var(:,6,:) + & - q(7,7) * var(:,7,:) + q(8,7) * var(:,8,:) + & - q(9,7) * var(:,9,:) + q(10,7) * var(:,10,:) ) * idel - jl = 8 - end if - if ( bb(2) == 0 ) then - jr = nj - gsize + if ( zero_derivs_y /= 0 ) then + dvar = zero else - dvar(:,nj-6,:) = - ( q(1,7) * var(:,nj,:) + q(2,7) * var(:,nj-1,:) + & - q(3,7) * var(:,nj-2,:) + q(4,7) * var(:,nj-3,:) + & - q(5,7) * var(:,nj-4,:) + q(6,7) * var(:,nj-5,:) + & - q(7,7) * var(:,nj-6,:) + q(8,7) * var(:,nj-7,:) + & - q(9,7) * var(:,nj-8,:) + & - q(10,7) * var(:,nj-9,:) ) * idel - dvar(:,nj-5,:) = - ( q(1,6) * var(:,nj,:) + q(2,6) * var(:,nj-1,:) + & - q(3,6) * var(:,nj-2,:) + q(4,6) * var(:,nj-3,:) + & - q(5,6) * var(:,nj-4,:) + q(6,6) * var(:,nj-5,:) + & - q(7,6) * var(:,nj-6,:) + q(8,6) * var(:,nj-7,:) + & - q(9,6) * var(:,nj-8,:) + & - q(10,6) * var(:,nj-9,:) ) * idel - 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,:) + q(8,5) * var(:,nj-7,:) + & - q(9,5) * var(:,nj-8,:) + & - q(10,5) * var(:,nj-9,:) ) * 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,:) + q(8,4) * var(:,nj-7,:) + & - q(9,4) * var(:,nj-8,:) + & - q(10,4) * var(:,nj-9,:) ) * 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,:) + & - q(7,3) * var(:,nj-6,:) + q(8,3) * var(:,nj-7,:) + & - q(9,3) * var(:,nj-8,:)) * 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,:) + & - q(7,2) * var(:,nj-6,:) + & - q(8,2) * var(:,nj-7,:) ) * 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,:) + & - q(5,1) * var(:,nj-4,:) + & - q(6,1) * var(:,nj-5,:) ) * idel - jr = nj - 7 + if ( bb(1) == 0 ) then + jl = 1 + gsize + else + dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & + q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) + & + q(5,1) * var(:,5,:) + q(6,1) * var(:,6,:) ) * 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,:) + & + q(7,2) * var(:,7,:) + q(8,2) * var(:,8,:) ) * 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,:) + & + q(7,3) * var(:,7,:) + q(8,3) * var(:,8,:) + & + q(9,3) * var(:,9,:) ) * 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,:) + q(8,4) * var(:,8,:) + & + q(9,4) * var(:,9,:) + q(10,4) * var(:,10,:) ) * 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,:) + q(8,5) * var(:,8,:) + & + q(9,5) * var(:,9,:) + q(10,5) * var(:,10,:) ) * idel + dvar(:,6,:) = ( q(1,6) * var(:,1,:) + q(2,6) * var(:,2,:) + & + q(3,6) * var(:,3,:) + q(4,6) * var(:,4,:) + & + q(5,6) * var(:,5,:) + q(6,6) * var(:,6,:) + & + q(7,6) * var(:,7,:) + q(8,6) * var(:,8,:) + & + q(9,6) * var(:,9,:) + q(10,6) * var(:,10,:) ) * idel + dvar(:,7,:) = ( q(1,7) * var(:,1,:) + q(2,7) * var(:,2,:) + & + q(3,7) * var(:,3,:) + q(4,7) * var(:,4,:) + & + q(5,7) * var(:,5,:) + q(6,7) * var(:,6,:) + & + q(7,7) * var(:,7,:) + q(8,7) * var(:,8,:) + & + q(9,7) * var(:,9,:) + q(10,7) * var(:,10,:) ) * idel + jl = 8 + end if + if ( bb(2) == 0 ) then + jr = nj - gsize + else + dvar(:,nj-6,:) = - ( q(1,7) * var(:,nj,:) + q(2,7) * var(:,nj-1,:) + & + q(3,7) * var(:,nj-2,:) + q(4,7) * var(:,nj-3,:) + & + q(5,7) * var(:,nj-4,:) + q(6,7) * var(:,nj-5,:) + & + q(7,7) * var(:,nj-6,:) + q(8,7) * var(:,nj-7,:) + & + q(9,7) * var(:,nj-8,:) + & + q(10,7) * var(:,nj-9,:) ) * idel + dvar(:,nj-5,:) = - ( q(1,6) * var(:,nj,:) + q(2,6) * var(:,nj-1,:) + & + q(3,6) * var(:,nj-2,:) + q(4,6) * var(:,nj-3,:) + & + q(5,6) * var(:,nj-4,:) + q(6,6) * var(:,nj-5,:) + & + q(7,6) * var(:,nj-6,:) + q(8,6) * var(:,nj-7,:) + & + q(9,6) * var(:,nj-8,:) + & + q(10,6) * var(:,nj-9,:) ) * idel + 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,:) + q(8,5) * var(:,nj-7,:) + & + q(9,5) * var(:,nj-8,:) + & + q(10,5) * var(:,nj-9,:) ) * 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,:) + q(8,4) * var(:,nj-7,:) + & + q(9,4) * var(:,nj-8,:) + & + q(10,4) * var(:,nj-9,:) ) * 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,:) + & + q(7,3) * var(:,nj-6,:) + q(8,3) * var(:,nj-7,:) + & + q(9,3) * var(:,nj-8,:)) * 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,:) + & + q(7,2) * var(:,nj-6,:) + & + q(8,2) * var(:,nj-7,:) ) * 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,:) + & + q(5,1) * var(:,nj-4,:) + & + q(6,1) * var(:,nj-5,:) ) * idel + jr = nj - 7 + end if + dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & + var(:,jl-1:jr-1,:) ) + & + a(2) * ( var(:,jl+2:jr+2,:) - & + var(:,jl-2:jr-2,:) ) + & + a(3) * ( var(:,jl+3:jr+3,:) - & + var(:,jl-3:jr-3,:) ) ) * idel end if - dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & - var(:,jl-1:jr-1,:) ) + & - a(2) * ( var(:,jl+2:jr+2,:) - & - var(:,jl-2:jr-2,:) ) + & - a(3) * ( var(:,jl+3:jr+3,:) - & - var(:,jl-3:jr-3,:) ) ) * idel case (2) direction - if ( bb(1) == 0 ) then - kl = 1 + gsize + if ( zero_derivs_z /= 0 ) then + dvar = zero else - dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & - q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) + & - q(5,1) * var(:,:,5) + q(6,1) * var(:,:,6) ) * 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) + & - q(7,2) * var(:,:,7) + q(8,2) * var(:,:,8) ) * 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) + & - q(7,3) * var(:,:,7) + q(8,3) * var(:,:,8) + & - q(9,3) * var(:,:,9) ) * 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) + q(8,4) * var(:,:,8) + & - q(9,4) * var(:,:,9) + q(10,4) * var(:,:,10) ) * 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) + q(8,5) * var(:,:,8) + & - q(9,5) * var(:,:,9) + q(10,5) * var(:,:,10) ) * idel - dvar(:,:,6) = ( q(1,6) * var(:,:,1) + q(2,6) * var(:,:,2) + & - q(3,6) * var(:,:,3) + q(4,6) * var(:,:,4) + & - q(5,6) * var(:,:,5) + q(6,6) * var(:,:,6) + & - q(7,6) * var(:,:,7) + q(8,6) * var(:,:,8) + & - q(9,6) * var(:,:,9) + q(10,6) * var(:,:,10) ) * idel - dvar(:,:,7) = ( q(1,7) * var(:,:,1) + q(2,7) * var(:,:,2) + & - q(3,7) * var(:,:,3) + q(4,7) * var(:,:,4) + & - q(5,7) * var(:,:,5) + q(6,7) * var(:,:,6) + & - q(7,7) * var(:,:,7) + q(8,7) * var(:,:,8) + & - q(9,7) * var(:,:,9) + q(10,7) * var(:,:,10) ) * idel - kl = 8 - end if - if ( bb(2) == 0 ) then - kr = nk - gsize - else - dvar(:,:,nk-6) = - ( q(1,7) * var(:,:,nk) + q(2,7) * var(:,:,nk-1) + & - q(3,7) * var(:,:,nk-2) + q(4,7) * var(:,:,nk-3) + & - q(5,7) * var(:,:,nk-4) + q(6,7) * var(:,:,nk-5) + & - q(7,7) * var(:,:,nk-6) + q(8,7) * var(:,:,nk-7) + & - q(9,7) * var(:,:,nk-8) + & - q(10,7) * var(:,:,nk-9) ) * idel - dvar(:,:,nk-5) = - ( q(1,6) * var(:,:,nk) + q(2,6) * var(:,:,nk-1) + & - q(3,6) * var(:,:,nk-2) + q(4,6) * var(:,:,nk-3) + & - q(5,6) * var(:,:,nk-4) + q(6,6) * var(:,:,nk-5) + & - q(7,6) * var(:,:,nk-6) + q(8,6) * var(:,:,nk-7) + & - q(9,6) * var(:,:,nk-8) + & - q(10,6) * var(:,:,nk-9) ) * idel - 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) + q(8,5) * var(:,:,nk-7) + & - q(9,5) * var(:,:,nk-8) + & - q(10,5) * var(:,:,nk-9) ) * 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) + q(8,4) * var(:,:,nk-7) + & - q(9,4) * var(:,:,nk-8) + & - q(10,4) * var(:,:,nk-9) ) * 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) + & - q(7,3) * var(:,:,nk-6) + q(8,3) * var(:,:,nk-7) + & - q(9,3) * var(:,:,nk-8)) * 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) + & - q(7,2) * var(:,:,nk-6) + & - q(8,2) * var(:,:,nk-7) ) * 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) + & - q(5,1) * var(:,:,nk-4) + & - q(6,1) * var(:,:,nk-5) ) * idel - kr = nk - 7 + if ( bb(1) == 0 ) then + kl = 1 + gsize + else + dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & + q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) + & + q(5,1) * var(:,:,5) + q(6,1) * var(:,:,6) ) * 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) + & + q(7,2) * var(:,:,7) + q(8,2) * var(:,:,8) ) * 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) + & + q(7,3) * var(:,:,7) + q(8,3) * var(:,:,8) + & + q(9,3) * var(:,:,9) ) * 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) + q(8,4) * var(:,:,8) + & + q(9,4) * var(:,:,9) + q(10,4) * var(:,:,10) ) * 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) + q(8,5) * var(:,:,8) + & + q(9,5) * var(:,:,9) + q(10,5) * var(:,:,10) ) * idel + dvar(:,:,6) = ( q(1,6) * var(:,:,1) + q(2,6) * var(:,:,2) + & + q(3,6) * var(:,:,3) + q(4,6) * var(:,:,4) + & + q(5,6) * var(:,:,5) + q(6,6) * var(:,:,6) + & + q(7,6) * var(:,:,7) + q(8,6) * var(:,:,8) + & + q(9,6) * var(:,:,9) + q(10,6) * var(:,:,10) ) * idel + dvar(:,:,7) = ( q(1,7) * var(:,:,1) + q(2,7) * var(:,:,2) + & + q(3,7) * var(:,:,3) + q(4,7) * var(:,:,4) + & + q(5,7) * var(:,:,5) + q(6,7) * var(:,:,6) + & + q(7,7) * var(:,:,7) + q(8,7) * var(:,:,8) + & + q(9,7) * var(:,:,9) + q(10,7) * var(:,:,10) ) * idel + kl = 8 + end if + if ( bb(2) == 0 ) then + kr = nk - gsize + else + dvar(:,:,nk-6) = - ( q(1,7) * var(:,:,nk) + q(2,7) * var(:,:,nk-1) + & + q(3,7) * var(:,:,nk-2) + q(4,7) * var(:,:,nk-3) + & + q(5,7) * var(:,:,nk-4) + q(6,7) * var(:,:,nk-5) + & + q(7,7) * var(:,:,nk-6) + q(8,7) * var(:,:,nk-7) + & + q(9,7) * var(:,:,nk-8) + & + q(10,7) * var(:,:,nk-9) ) * idel + dvar(:,:,nk-5) = - ( q(1,6) * var(:,:,nk) + q(2,6) * var(:,:,nk-1) + & + q(3,6) * var(:,:,nk-2) + q(4,6) * var(:,:,nk-3) + & + q(5,6) * var(:,:,nk-4) + q(6,6) * var(:,:,nk-5) + & + q(7,6) * var(:,:,nk-6) + q(8,6) * var(:,:,nk-7) + & + q(9,6) * var(:,:,nk-8) + & + q(10,6) * var(:,:,nk-9) ) * idel + 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) + q(8,5) * var(:,:,nk-7) + & + q(9,5) * var(:,:,nk-8) + & + q(10,5) * var(:,:,nk-9) ) * 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) + q(8,4) * var(:,:,nk-7) + & + q(9,4) * var(:,:,nk-8) + & + q(10,4) * var(:,:,nk-9) ) * 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) + & + q(7,3) * var(:,:,nk-6) + q(8,3) * var(:,:,nk-7) + & + q(9,3) * var(:,:,nk-8)) * 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) + & + q(7,2) * var(:,:,nk-6) + & + q(8,2) * var(:,:,nk-7) ) * 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) + & + q(5,1) * var(:,:,nk-4) + & + q(6,1) * var(:,:,nk-5) ) * idel + kr = nk - 7 + end if + dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & + var(:,:,kl-1:kr-1) ) + & + a(2) * ( var(:,:,kl+2:kr+2) - & + var(:,:,kl-2:kr-2) ) + & + a(3) * ( var(:,:,kl+3:kr+3) - & + var(:,:,kl-3:kr-3) ) ) * idel end if - dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & - var(:,:,kl-1:kr-1) ) + & - a(2) * ( var(:,:,kl+2:kr+2) - & - var(:,:,kl-2:kr-2) ) + & - a(3) * ( var(:,:,kl+3:kr+3) - & - var(:,:,kl-3:kr-3) ) ) * idel end select direction end subroutine deriv_gf_6_5 diff --git a/src/Derivatives_6_5_min_err_coeff.F90 b/src/Derivatives_6_5_min_err_coeff.F90 index cad827a..dcab191 100644 --- a/src/Derivatives_6_5_min_err_coeff.F90 +++ b/src/Derivatives_6_5_min_err_coeff.F90 @@ -199,186 +199,194 @@ subroutine deriv_gf_6_5_opt ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) a(3) * ( var(il+3:ir+3,:,:) - & var(il-3:ir-3,:,:) ) ) * idel case (1) direction - if ( bb(1) == 0 ) then - jl = 1 + gsize - else - dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & - q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) + & - q(5,1) * var(:,5,:) + q(6,1) * var(:,6,:) + & - q(7,1) * var(:,7,:) ) * 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,:) + & - q(7,2) * var(:,7,:) + q(8,2) * var(:,8,:) + & - q(9,2) * var(:,9,:) + q(10,2) * var(:,10,:) ) * 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,:) + & - q(7,3) * var(:,7,:) + q(8,3) * var(:,8,:) + & - q(9,3) * var(:,9,:) + q(10,3) * var(:,10,:) ) * 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,:) + q(8,4) * var(:,8,:) + & - q(9,4) * var(:,9,:) + q(10,4) * var(:,10,:) ) * 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,:) + q(8,5) * var(:,8,:) + & - q(9,5) * var(:,9,:) + q(10,5) * var(:,10,:) ) * idel - dvar(:,6,:) = ( q(1,6) * var(:,1,:) + q(2,6) * var(:,2,:) + & - q(3,6) * var(:,3,:) + q(4,6) * var(:,4,:) + & - q(5,6) * var(:,5,:) + q(6,6) * var(:,6,:) + & - q(7,6) * var(:,7,:) + q(8,6) * var(:,8,:) + & - q(9,6) * var(:,9,:) + q(10,6) * var(:,10,:) ) * idel - dvar(:,7,:) = ( q(1,7) * var(:,1,:) + q(2,7) * var(:,2,:) + & - q(3,7) * var(:,3,:) + q(4,7) * var(:,4,:) + & - q(5,7) * var(:,5,:) + q(6,7) * var(:,6,:) + & - q(7,7) * var(:,7,:) + q(8,7) * var(:,8,:) + & - q(9,7) * var(:,9,:) + q(10,7) * var(:,10,:) ) * idel - jl = 8 - end if - if ( bb(2) == 0 ) then - jr = nj - gsize + if ( zero_derivs_y /= 0 ) then + dvar = zero else - dvar(:,nj-6,:) = - ( q(1,7) * var(:,nj,:) + q(2,7) * var(:,nj-1,:) + & - q(3,7) * var(:,nj-2,:) + q(4,7) * var(:,nj-3,:) + & - q(5,7) * var(:,nj-4,:) + q(6,7) * var(:,nj-5,:) + & - q(7,7) * var(:,nj-6,:) + q(8,7) * var(:,nj-7,:) + & - q(9,7) * var(:,nj-8,:) + & - q(10,7) * var(:,nj-9,:) ) * idel - dvar(:,nj-5,:) = - ( q(1,6) * var(:,nj,:) + q(2,6) * var(:,nj-1,:) + & - q(3,6) * var(:,nj-2,:) + q(4,6) * var(:,nj-3,:) + & - q(5,6) * var(:,nj-4,:) + q(6,6) * var(:,nj-5,:) + & - q(7,6) * var(:,nj-6,:) + q(8,6) * var(:,nj-7,:) + & - q(9,6) * var(:,nj-8,:) + & - q(10,6) * var(:,nj-9,:) ) * idel - 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,:) + q(8,5) * var(:,nj-7,:) + & - q(9,5) * var(:,nj-8,:) + & - q(10,5) * var(:,nj-9,:) ) * 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,:) + q(8,4) * var(:,nj-7,:) + & - q(9,4) * var(:,nj-8,:) + & - q(10,4) * var(:,nj-9,:) ) * 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,:) + & - q(7,3) * var(:,nj-6,:) + q(8,3) * var(:,nj-7,:) + & - q(9,3) * var(:,nj-8,:) + & - q(10,3) * var(:,nj-9,:) ) * 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,:) + & - q(7,2) * var(:,nj-6,:) + q(8,2) * var(:,nj-7,:) + & - q(9,2) * var(:,nj-8,:) + & - q(10,2) * var(:,nj-9,:) ) * 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,:) + & - q(5,1) * var(:,nj-4,:) + q(6,1) * var(:,nj-5,:) + & - q(7,1) * var(:,nj-6,:) ) * idel - jr = nj - 7 + if ( bb(1) == 0 ) then + jl = 1 + gsize + else + dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & + q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) + & + q(5,1) * var(:,5,:) + q(6,1) * var(:,6,:) + & + q(7,1) * var(:,7,:) ) * 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,:) + & + q(7,2) * var(:,7,:) + q(8,2) * var(:,8,:) + & + q(9,2) * var(:,9,:) + q(10,2) * var(:,10,:) ) * 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,:) + & + q(7,3) * var(:,7,:) + q(8,3) * var(:,8,:) + & + q(9,3) * var(:,9,:) + q(10,3) * var(:,10,:) ) * 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,:) + q(8,4) * var(:,8,:) + & + q(9,4) * var(:,9,:) + q(10,4) * var(:,10,:) ) * 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,:) + q(8,5) * var(:,8,:) + & + q(9,5) * var(:,9,:) + q(10,5) * var(:,10,:) ) * idel + dvar(:,6,:) = ( q(1,6) * var(:,1,:) + q(2,6) * var(:,2,:) + & + q(3,6) * var(:,3,:) + q(4,6) * var(:,4,:) + & + q(5,6) * var(:,5,:) + q(6,6) * var(:,6,:) + & + q(7,6) * var(:,7,:) + q(8,6) * var(:,8,:) + & + q(9,6) * var(:,9,:) + q(10,6) * var(:,10,:) ) * idel + dvar(:,7,:) = ( q(1,7) * var(:,1,:) + q(2,7) * var(:,2,:) + & + q(3,7) * var(:,3,:) + q(4,7) * var(:,4,:) + & + q(5,7) * var(:,5,:) + q(6,7) * var(:,6,:) + & + q(7,7) * var(:,7,:) + q(8,7) * var(:,8,:) + & + q(9,7) * var(:,9,:) + q(10,7) * var(:,10,:) ) * idel + jl = 8 + end if + if ( bb(2) == 0 ) then + jr = nj - gsize + else + dvar(:,nj-6,:) = - ( q(1,7) * var(:,nj,:) + q(2,7) * var(:,nj-1,:) + & + q(3,7) * var(:,nj-2,:) + q(4,7) * var(:,nj-3,:) + & + q(5,7) * var(:,nj-4,:) + q(6,7) * var(:,nj-5,:) + & + q(7,7) * var(:,nj-6,:) + q(8,7) * var(:,nj-7,:) + & + q(9,7) * var(:,nj-8,:) + & + q(10,7) * var(:,nj-9,:) ) * idel + dvar(:,nj-5,:) = - ( q(1,6) * var(:,nj,:) + q(2,6) * var(:,nj-1,:) + & + q(3,6) * var(:,nj-2,:) + q(4,6) * var(:,nj-3,:) + & + q(5,6) * var(:,nj-4,:) + q(6,6) * var(:,nj-5,:) + & + q(7,6) * var(:,nj-6,:) + q(8,6) * var(:,nj-7,:) + & + q(9,6) * var(:,nj-8,:) + & + q(10,6) * var(:,nj-9,:) ) * idel + 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,:) + q(8,5) * var(:,nj-7,:) + & + q(9,5) * var(:,nj-8,:) + & + q(10,5) * var(:,nj-9,:) ) * 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,:) + q(8,4) * var(:,nj-7,:) + & + q(9,4) * var(:,nj-8,:) + & + q(10,4) * var(:,nj-9,:) ) * 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,:) + & + q(7,3) * var(:,nj-6,:) + q(8,3) * var(:,nj-7,:) + & + q(9,3) * var(:,nj-8,:) + & + q(10,3) * var(:,nj-9,:) ) * 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,:) + & + q(7,2) * var(:,nj-6,:) + q(8,2) * var(:,nj-7,:) + & + q(9,2) * var(:,nj-8,:) + & + q(10,2) * var(:,nj-9,:) ) * 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,:) + & + q(5,1) * var(:,nj-4,:) + q(6,1) * var(:,nj-5,:) + & + q(7,1) * var(:,nj-6,:) ) * idel + jr = nj - 7 + end if + dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & + var(:,jl-1:jr-1,:) ) + & + a(2) * ( var(:,jl+2:jr+2,:) - & + var(:,jl-2:jr-2,:) ) + & + a(3) * ( var(:,jl+3:jr+3,:) - & + var(:,jl-3:jr-3,:) ) ) * idel end if - dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & - var(:,jl-1:jr-1,:) ) + & - a(2) * ( var(:,jl+2:jr+2,:) - & - var(:,jl-2:jr-2,:) ) + & - a(3) * ( var(:,jl+3:jr+3,:) - & - var(:,jl-3:jr-3,:) ) ) * idel case (2) direction - if ( bb(1) == 0 ) then - kl = 1 + gsize + if ( zero_derivs_z /= 0 ) then + dvar = zero else - dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & - q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) + & - q(5,1) * var(:,:,5) + q(6,1) * var(:,:,6) + & - q(7,1) * var(:,:,7) ) * 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) + & - q(7,2) * var(:,:,7) + q(8,2) * var(:,:,8) + & - q(9,2) * var(:,:,9) + q(10,2) * var(:,:,10) ) * 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) + & - q(7,3) * var(:,:,7) + q(8,3) * var(:,:,8) + & - q(9,3) * var(:,:,9) + q(10,3) * var(:,:,10) ) * 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) + q(8,4) * var(:,:,8) + & - q(9,4) * var(:,:,9) + q(10,4) * var(:,:,10) ) * 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) + q(8,5) * var(:,:,8) + & - q(9,5) * var(:,:,9) + q(10,5) * var(:,:,10) ) * idel - dvar(:,:,6) = ( q(1,6) * var(:,:,1) + q(2,6) * var(:,:,2) + & - q(3,6) * var(:,:,3) + q(4,6) * var(:,:,4) + & - q(5,6) * var(:,:,5) + q(6,6) * var(:,:,6) + & - q(7,6) * var(:,:,7) + q(8,6) * var(:,:,8) + & - q(9,6) * var(:,:,9) + q(10,6) * var(:,:,10) ) * idel - dvar(:,:,7) = ( q(1,7) * var(:,:,1) + q(2,7) * var(:,:,2) + & - q(3,7) * var(:,:,3) + q(4,7) * var(:,:,4) + & - q(5,7) * var(:,:,5) + q(6,7) * var(:,:,6) + & - q(7,7) * var(:,:,7) + q(8,7) * var(:,:,8) + & - q(9,7) * var(:,:,9) + q(10,7) * var(:,:,10) ) * idel - kl = 8 - end if - if ( bb(2) == 0 ) then - kr = nk - gsize - else - dvar(:,:,nk-6) = - ( q(1,7) * var(:,:,nk) + q(2,7) * var(:,:,nk-1) + & - q(3,7) * var(:,:,nk-2) + q(4,7) * var(:,:,nk-3) + & - q(5,7) * var(:,:,nk-4) + q(6,7) * var(:,:,nk-5) + & - q(7,7) * var(:,:,nk-6) + q(8,7) * var(:,:,nk-7) + & - q(9,7) * var(:,:,nk-8) + & - q(10,7) * var(:,:,nk-9) ) * idel - dvar(:,:,nk-5) = - ( q(1,6) * var(:,:,nk) + q(2,6) * var(:,:,nk-1) + & - q(3,6) * var(:,:,nk-2) + q(4,6) * var(:,:,nk-3) + & - q(5,6) * var(:,:,nk-4) + q(6,6) * var(:,:,nk-5) + & - q(7,6) * var(:,:,nk-6) + q(8,6) * var(:,:,nk-7) + & - q(9,6) * var(:,:,nk-8) + & - q(10,6) * var(:,:,nk-9) ) * idel - 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) + q(8,5) * var(:,:,nk-7) + & - q(9,5) * var(:,:,nk-8) + & - q(10,5) * var(:,:,nk-9) ) * 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) + q(8,4) * var(:,:,nk-7) + & - q(9,4) * var(:,:,nk-8) + & - q(10,4) * var(:,:,nk-9) ) * 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) + & - q(7,3) * var(:,:,nk-6) + q(8,3) * var(:,:,nk-7) + & - q(9,3) * var(:,:,nk-8) + & - q(10,3) * var(:,:,nk-9) ) * 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) + & - q(7,2) * var(:,:,nk-6) + q(8,2) * var(:,:,nk-7) + & - q(9,2) * var(:,:,nk-8) + & - q(10,2) * var(:,:,nk-9) ) * 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) + & - q(5,1) * var(:,:,nk-4) + q(6,1) * var(:,:,nk-5) + & - q(7,1) * var(:,:,nk-6) ) * idel - kr = nk - 7 + if ( bb(1) == 0 ) then + kl = 1 + gsize + else + dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & + q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) + & + q(5,1) * var(:,:,5) + q(6,1) * var(:,:,6) + & + q(7,1) * var(:,:,7) ) * 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) + & + q(7,2) * var(:,:,7) + q(8,2) * var(:,:,8) + & + q(9,2) * var(:,:,9) + q(10,2) * var(:,:,10) ) * 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) + & + q(7,3) * var(:,:,7) + q(8,3) * var(:,:,8) + & + q(9,3) * var(:,:,9) + q(10,3) * var(:,:,10) ) * 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) + q(8,4) * var(:,:,8) + & + q(9,4) * var(:,:,9) + q(10,4) * var(:,:,10) ) * 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) + q(8,5) * var(:,:,8) + & + q(9,5) * var(:,:,9) + q(10,5) * var(:,:,10) ) * idel + dvar(:,:,6) = ( q(1,6) * var(:,:,1) + q(2,6) * var(:,:,2) + & + q(3,6) * var(:,:,3) + q(4,6) * var(:,:,4) + & + q(5,6) * var(:,:,5) + q(6,6) * var(:,:,6) + & + q(7,6) * var(:,:,7) + q(8,6) * var(:,:,8) + & + q(9,6) * var(:,:,9) + q(10,6) * var(:,:,10) ) * idel + dvar(:,:,7) = ( q(1,7) * var(:,:,1) + q(2,7) * var(:,:,2) + & + q(3,7) * var(:,:,3) + q(4,7) * var(:,:,4) + & + q(5,7) * var(:,:,5) + q(6,7) * var(:,:,6) + & + q(7,7) * var(:,:,7) + q(8,7) * var(:,:,8) + & + q(9,7) * var(:,:,9) + q(10,7) * var(:,:,10) ) * idel + kl = 8 + end if + if ( bb(2) == 0 ) then + kr = nk - gsize + else + dvar(:,:,nk-6) = - ( q(1,7) * var(:,:,nk) + q(2,7) * var(:,:,nk-1) + & + q(3,7) * var(:,:,nk-2) + q(4,7) * var(:,:,nk-3) + & + q(5,7) * var(:,:,nk-4) + q(6,7) * var(:,:,nk-5) + & + q(7,7) * var(:,:,nk-6) + q(8,7) * var(:,:,nk-7) + & + q(9,7) * var(:,:,nk-8) + & + q(10,7) * var(:,:,nk-9) ) * idel + dvar(:,:,nk-5) = - ( q(1,6) * var(:,:,nk) + q(2,6) * var(:,:,nk-1) + & + q(3,6) * var(:,:,nk-2) + q(4,6) * var(:,:,nk-3) + & + q(5,6) * var(:,:,nk-4) + q(6,6) * var(:,:,nk-5) + & + q(7,6) * var(:,:,nk-6) + q(8,6) * var(:,:,nk-7) + & + q(9,6) * var(:,:,nk-8) + & + q(10,6) * var(:,:,nk-9) ) * idel + 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) + q(8,5) * var(:,:,nk-7) + & + q(9,5) * var(:,:,nk-8) + & + q(10,5) * var(:,:,nk-9) ) * 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) + q(8,4) * var(:,:,nk-7) + & + q(9,4) * var(:,:,nk-8) + & + q(10,4) * var(:,:,nk-9) ) * 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) + & + q(7,3) * var(:,:,nk-6) + q(8,3) * var(:,:,nk-7) + & + q(9,3) * var(:,:,nk-8) + & + q(10,3) * var(:,:,nk-9) ) * 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) + & + q(7,2) * var(:,:,nk-6) + q(8,2) * var(:,:,nk-7) + & + q(9,2) * var(:,:,nk-8) + & + q(10,2) * var(:,:,nk-9) ) * 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) + & + q(5,1) * var(:,:,nk-4) + q(6,1) * var(:,:,nk-5) + & + q(7,1) * var(:,:,nk-6) ) * idel + kr = nk - 7 + end if + dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & + var(:,:,kl-1:kr-1) ) + & + a(2) * ( var(:,:,kl+2:kr+2) - & + var(:,:,kl-2:kr-2) ) + & + a(3) * ( var(:,:,kl+3:kr+3) - & + var(:,:,kl-3:kr-3) ) ) * idel end if - dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & - var(:,:,kl-1:kr-1) ) + & - a(2) * ( var(:,:,kl+2:kr+2) - & - var(:,:,kl-2:kr-2) ) + & - a(3) * ( var(:,:,kl+3:kr+3) - & - var(:,:,kl-3:kr-3) ) ) * idel end select direction end subroutine deriv_gf_6_5_opt diff --git a/src/Derivatives_8_4.F90 b/src/Derivatives_8_4.F90 index fb312f1..3d7e477 100644 --- a/src/Derivatives_8_4.F90 +++ b/src/Derivatives_8_4.F90 @@ -264,194 +264,202 @@ subroutine deriv_gf_8_4 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) a(4) * ( var(il+4:ir+4,:,:) - & var(il-4:ir-4,:,:) ) ) * idel case (1) direction - if ( bb(1) == 0 ) then - jl = 1 + gsize - else - dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & - q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) + & - q(5,1) * var(:,5,:) + q(6,1) * var(:,6,:) + & - q(7,1) * var(:,7,:) + q(8,1) * var(:,8,:) ) * idel - dvar(:,2,:) = ( q(1,2) * var(:,1,:) + q(3,2) * var(:,3,:) + & - q(4,2) * var(:,4,:) + q(5,2) * var(:,5,:) + & - q(6,2) * var(:,6,:) + q(7,2) * var(:,7,:) + & - q(8,2) * var(:,8,:) ) * idel - dvar(:,3,:) = ( q(1,3) * var(:,1,:) + q(2,3) * var(:,2,:) + & - q(4,3) * var(:,4,:) + q(5,3) * var(:,5,:) + & - q(6,3) * var(:,6,:) + q(7,3) * var(:,7,:) + & - q(8,3) * var(:,8,:) ) * idel - dvar(:,4,:) = ( q(1,4) * var(:,1,:) + q(2,4) * var(:,2,:) + & - q(3,4) * var(:,3,:) + q(5,4) * var(:,5,:) + & - q(6,4) * var(:,6,:) + q(7,4) * var(:,7,:) + & - q(8,4) * var(:,8,:) ) * idel - dvar(:,5,:) = ( q(1,5) * var(:,1,:) + q(2,5) * var(:,2,:) + & - q(3,5) * var(:,3,:) + q(4,5) * var(:,4,:) + & - q(6,5) * var(:,6,:) + q(7,5) * var(:,7,:) + & - q(8,5) * var(:,8,:) + q(9,5) * var(:,9,:) ) * idel - dvar(:,6,:) = ( q(1,6) * var(:,1,:) + q(2,6) * var(:,2,:) + & - q(3,6) * var(:,3,:) + q(4,6) * var(:,4,:) + & - q(5,6) * var(:,5,:) + q(7,6) * var(:,7,:) + & - q(8,6) * var(:,8,:) + q(9,6) * var(:,9,:) + & - q(10,6) * var(:,10,:) ) * idel - dvar(:,7,:) = ( q(1,7) * var(:,1,:) + q(2,7) * var(:,2,:) + & - q(3,7) * var(:,3,:) + q(4,7) * var(:,4,:) + & - q(5,7) * var(:,5,:) + q(6,7) * var(:,6,:) + & - q(8,7) * var(:,8,:) + q(9,7) * var(:,9,:) + & - q(10,7) * var(:,10,:) + q(11,7) * var(:,11,:) ) * idel - dvar(:,8,:) = ( q(1,8) * var(:,1,:) + q(2,8) * var(:,2,:) + & - q(3,8) * var(:,3,:) + q(4,8) * var(:,4,:) + & - q(5,8) * var(:,5,:) + q(6,8) * var(:,6,:) + & - q(7,8) * var(:,7,:) + q(9,8) * var(:,9,:) + & - q(10,8) * var(:,10,:) + q(11,8) * var(:,11,:) + & - q(12,8) * var(:,12,:) ) * idel - jl = 9 - end if - if ( bb(2) == 0 ) then - jr = nj - gsize + if ( zero_derivs_y /= 0 ) then + dvar = zero else - 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,:) + & - q(5,1) * var(:,nj-4,:) + q(6,1) * var(:,nj-5,:) + & - q(7,1) * var(:,nj-6,:) + & - q(8,1) * var(:,nj-7,:) ) * idel - dvar(:,nj-1,:) = - ( q(1,2) * var(:,nj,:) + q(3,2) * var(:,nj-2,:) + & - q(4,2) * var(:,nj-3,:) + q(5,2) * var(:,nj-4,:) + & - q(6,2) * var(:,nj-5,:) + q(7,2) * var(:,nj-6,:) + & - q(8,2) * var(:,nj-7,:) ) * idel - dvar(:,nj-2,:) = - ( q(1,3) * var(:,nj,:) + q(2,3) * var(:,nj-1,:) + & - q(4,3) * var(:,nj-3,:) + q(5,3) * var(:,nj-4,:) + & - q(6,3) * var(:,nj-5,:) + q(7,3) * var(:,nj-6,:) + & - q(8,3) * var(:,nj-7,:) ) * idel - dvar(:,nj-3,:) = - ( q(1,4) * var(:,nj,:) + q(2,4) * var(:,nj-1,:) + & - q(3,4) * var(:,nj-2,:) + q(5,4) * var(:,nj-4,:) + & - q(6,4) * var(:,nj-5,:) + q(7,4) * var(:,nj-6,:) + & - q(8,4) * var(:,nj-7,:) ) * idel - 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(6,5) * var(:,nj-5,:) + q(7,5) * var(:,nj-6,:) + & - q(8,5) * var(:,nj-7,:) + & - q(9,5) * var(:,nj-8,:) ) * idel - dvar(:,nj-5,:) = - ( q(1,6) * var(:,nj,:) + q(2,6) * var(:,nj-1,:) + & - q(3,6) * var(:,nj-2,:) + q(4,6) * var(:,nj-3,:) + & - q(5,6) * var(:,nj-4,:) + q(7,6) * var(:,nj-6,:) + & - q(8,6) * var(:,nj-7,:) + q(9,6) * var(:,nj-8,:) + & - q(10,6) * var(:,nj-9,:) ) * idel - dvar(:,nj-6,:) = - ( q(1,7) * var(:,nj,:) + q(2,7) * var(:,nj-1,:) + & - q(3,7) * var(:,nj-2,:) + q(4,7) * var(:,nj-3,:) + & - q(5,7) * var(:,nj-4,:) + q(6,7) * var(:,nj-5,:) + & - q(8,7) * var(:,nj-7,:) + q(9,7) * var(:,nj-8,:) + & - q(10,7) * var(:,nj-9,:) + & - q(11,7) * var(:,nj-10,:) ) * idel - dvar(:,nj-7,:) = - ( q(1,8) * var(:,nj,:) + q(2,8) * var(:,nj-1,:) + & - q(3,8) * var(:,nj-2,:) + q(4,8) * var(:,nj-3,:) + & - q(5,8) * var(:,nj-4,:) + q(6,8) * var(:,nj-5,:) + & - q(7,8) * var(:,nj-6,:) + q(9,8) * var(:,nj-8,:) + & - q(10,8) * var(:,nj-9,:) + & - q(11,8) * var(:,nj-10,:) + & - q(12,8) * var(:,nj-11,:) ) * idel - jr = nj - 8 + if ( bb(1) == 0 ) then + jl = 1 + gsize + else + dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & + q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) + & + q(5,1) * var(:,5,:) + q(6,1) * var(:,6,:) + & + q(7,1) * var(:,7,:) + q(8,1) * var(:,8,:) ) * idel + dvar(:,2,:) = ( q(1,2) * var(:,1,:) + q(3,2) * var(:,3,:) + & + q(4,2) * var(:,4,:) + q(5,2) * var(:,5,:) + & + q(6,2) * var(:,6,:) + q(7,2) * var(:,7,:) + & + q(8,2) * var(:,8,:) ) * idel + dvar(:,3,:) = ( q(1,3) * var(:,1,:) + q(2,3) * var(:,2,:) + & + q(4,3) * var(:,4,:) + q(5,3) * var(:,5,:) + & + q(6,3) * var(:,6,:) + q(7,3) * var(:,7,:) + & + q(8,3) * var(:,8,:) ) * idel + dvar(:,4,:) = ( q(1,4) * var(:,1,:) + q(2,4) * var(:,2,:) + & + q(3,4) * var(:,3,:) + q(5,4) * var(:,5,:) + & + q(6,4) * var(:,6,:) + q(7,4) * var(:,7,:) + & + q(8,4) * var(:,8,:) ) * idel + dvar(:,5,:) = ( q(1,5) * var(:,1,:) + q(2,5) * var(:,2,:) + & + q(3,5) * var(:,3,:) + q(4,5) * var(:,4,:) + & + q(6,5) * var(:,6,:) + q(7,5) * var(:,7,:) + & + q(8,5) * var(:,8,:) + q(9,5) * var(:,9,:) ) * idel + dvar(:,6,:) = ( q(1,6) * var(:,1,:) + q(2,6) * var(:,2,:) + & + q(3,6) * var(:,3,:) + q(4,6) * var(:,4,:) + & + q(5,6) * var(:,5,:) + q(7,6) * var(:,7,:) + & + q(8,6) * var(:,8,:) + q(9,6) * var(:,9,:) + & + q(10,6) * var(:,10,:) ) * idel + dvar(:,7,:) = ( q(1,7) * var(:,1,:) + q(2,7) * var(:,2,:) + & + q(3,7) * var(:,3,:) + q(4,7) * var(:,4,:) + & + q(5,7) * var(:,5,:) + q(6,7) * var(:,6,:) + & + q(8,7) * var(:,8,:) + q(9,7) * var(:,9,:) + & + q(10,7) * var(:,10,:) + q(11,7) * var(:,11,:) ) * idel + dvar(:,8,:) = ( q(1,8) * var(:,1,:) + q(2,8) * var(:,2,:) + & + q(3,8) * var(:,3,:) + q(4,8) * var(:,4,:) + & + q(5,8) * var(:,5,:) + q(6,8) * var(:,6,:) + & + q(7,8) * var(:,7,:) + q(9,8) * var(:,9,:) + & + q(10,8) * var(:,10,:) + q(11,8) * var(:,11,:) + & + q(12,8) * var(:,12,:) ) * idel + jl = 9 + end if + if ( bb(2) == 0 ) then + jr = nj - gsize + else + 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,:) + & + q(5,1) * var(:,nj-4,:) + q(6,1) * var(:,nj-5,:) + & + q(7,1) * var(:,nj-6,:) + & + q(8,1) * var(:,nj-7,:) ) * idel + dvar(:,nj-1,:) = - ( q(1,2) * var(:,nj,:) + q(3,2) * var(:,nj-2,:) + & + q(4,2) * var(:,nj-3,:) + q(5,2) * var(:,nj-4,:) + & + q(6,2) * var(:,nj-5,:) + q(7,2) * var(:,nj-6,:) + & + q(8,2) * var(:,nj-7,:) ) * idel + dvar(:,nj-2,:) = - ( q(1,3) * var(:,nj,:) + q(2,3) * var(:,nj-1,:) + & + q(4,3) * var(:,nj-3,:) + q(5,3) * var(:,nj-4,:) + & + q(6,3) * var(:,nj-5,:) + q(7,3) * var(:,nj-6,:) + & + q(8,3) * var(:,nj-7,:) ) * idel + dvar(:,nj-3,:) = - ( q(1,4) * var(:,nj,:) + q(2,4) * var(:,nj-1,:) + & + q(3,4) * var(:,nj-2,:) + q(5,4) * var(:,nj-4,:) + & + q(6,4) * var(:,nj-5,:) + q(7,4) * var(:,nj-6,:) + & + q(8,4) * var(:,nj-7,:) ) * idel + 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(6,5) * var(:,nj-5,:) + q(7,5) * var(:,nj-6,:) + & + q(8,5) * var(:,nj-7,:) + & + q(9,5) * var(:,nj-8,:) ) * idel + dvar(:,nj-5,:) = - ( q(1,6) * var(:,nj,:) + q(2,6) * var(:,nj-1,:) + & + q(3,6) * var(:,nj-2,:) + q(4,6) * var(:,nj-3,:) + & + q(5,6) * var(:,nj-4,:) + q(7,6) * var(:,nj-6,:) + & + q(8,6) * var(:,nj-7,:) + q(9,6) * var(:,nj-8,:) + & + q(10,6) * var(:,nj-9,:) ) * idel + dvar(:,nj-6,:) = - ( q(1,7) * var(:,nj,:) + q(2,7) * var(:,nj-1,:) + & + q(3,7) * var(:,nj-2,:) + q(4,7) * var(:,nj-3,:) + & + q(5,7) * var(:,nj-4,:) + q(6,7) * var(:,nj-5,:) + & + q(8,7) * var(:,nj-7,:) + q(9,7) * var(:,nj-8,:) + & + q(10,7) * var(:,nj-9,:) + & + q(11,7) * var(:,nj-10,:) ) * idel + dvar(:,nj-7,:) = - ( q(1,8) * var(:,nj,:) + q(2,8) * var(:,nj-1,:) + & + q(3,8) * var(:,nj-2,:) + q(4,8) * var(:,nj-3,:) + & + q(5,8) * var(:,nj-4,:) + q(6,8) * var(:,nj-5,:) + & + q(7,8) * var(:,nj-6,:) + q(9,8) * var(:,nj-8,:) + & + q(10,8) * var(:,nj-9,:) + & + q(11,8) * var(:,nj-10,:) + & + q(12,8) * var(:,nj-11,:) ) * idel + jr = nj - 8 + end if + dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & + var(:,jl-1:jr-1,:) ) + & + a(2) * ( var(:,jl+2:jr+2,:) - & + var(:,jl-2:jr-2,:) ) + & + a(3) * ( var(:,jl+3:jr+3,:) - & + var(:,jl-3:jr-3,:) ) + & + a(4) * ( var(:,jl+4:jr+4,:) - & + var(:,jl-4:jr-4,:) ) ) * idel end if - dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & - var(:,jl-1:jr-1,:) ) + & - a(2) * ( var(:,jl+2:jr+2,:) - & - var(:,jl-2:jr-2,:) ) + & - a(3) * ( var(:,jl+3:jr+3,:) - & - var(:,jl-3:jr-3,:) ) + & - a(4) * ( var(:,jl+4:jr+4,:) - & - var(:,jl-4:jr-4,:) ) ) * idel case (2) direction - if ( bb(1) == 0 ) then - kl = 1 + gsize + if ( zero_derivs_z /= 0 ) then + dvar = zero else - dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & - q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) + & - q(5,1) * var(:,:,5) + q(6,1) * var(:,:,6) + & - q(7,1) * var(:,:,7) + q(8,1) * var(:,:,8) ) * idel - dvar(:,:,2) = ( q(1,2) * var(:,:,1) + q(3,2) * var(:,:,3) + & - q(4,2) * var(:,:,4) + q(5,2) * var(:,:,5) + & - q(6,2) * var(:,:,6) + q(7,2) * var(:,:,7) + & - q(8,2) * var(:,:,8) ) * idel - dvar(:,:,3) = ( q(1,3) * var(:,:,1) + q(2,3) * var(:,:,2) + & - q(4,3) * var(:,:,4) + q(5,3) * var(:,:,5) + & - q(6,3) * var(:,:,6) + q(7,3) * var(:,:,7) + & - q(8,3) * var(:,:,8) ) * idel - dvar(:,:,4) = ( q(1,4) * var(:,:,1) + q(2,4) * var(:,:,2) + & - q(3,4) * var(:,:,3) + q(5,4) * var(:,:,5) + & - q(6,4) * var(:,:,6) + q(7,4) * var(:,:,7) + & - q(8,4) * var(:,:,8) ) * idel - dvar(:,:,5) = ( q(1,5) * var(:,:,1) + q(2,5) * var(:,:,2) + & - q(3,5) * var(:,:,3) + q(4,5) * var(:,:,4) + & - q(6,5) * var(:,:,6) + q(7,5) * var(:,:,7) + & - q(8,5) * var(:,:,8) + q(9,5) * var(:,:,9) ) * idel - dvar(:,:,6) = ( q(1,6) * var(:,:,1) + q(2,6) * var(:,:,2) + & - q(3,6) * var(:,:,3) + q(4,6) * var(:,:,4) + & - q(5,6) * var(:,:,5) + q(7,6) * var(:,:,7) + & - q(8,6) * var(:,:,8) + q(9,6) * var(:,:,9) + & - q(10,6) * var(:,:,10) ) * idel - dvar(:,:,7) = ( q(1,7) * var(:,:,1) + q(2,7) * var(:,:,2) + & - q(3,7) * var(:,:,3) + q(4,7) * var(:,:,4) + & - q(5,7) * var(:,:,5) + q(6,7) * var(:,:,6) + & - q(8,7) * var(:,:,8) + q(9,7) * var(:,:,9) + & - q(10,7) * var(:,:,10) + q(11,7) * var(:,:,11) ) * idel - dvar(:,:,8) = ( q(1,8) * var(:,:,1) + q(2,8) * var(:,:,2) + & - q(3,8) * var(:,:,3) + q(4,8) * var(:,:,4) + & - q(5,8) * var(:,:,5) + q(6,8) * var(:,:,6) + & - q(7,8) * var(:,:,7) + q(9,8) * var(:,:,9) + & - q(10,8) * var(:,:,10) + q(11,8) * var(:,:,11) + & - q(12,8) * var(:,:,12) ) * idel - kl = 9 - end if - if ( bb(2) == 0 ) then - kr = nk - gsize - else - 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) + & - q(5,1) * var(:,:,nk-4) + q(6,1) * var(:,:,nk-5) + & - q(7,1) * var(:,:,nk-6) + & - q(8,1) * var(:,:,nk-7) ) * idel - dvar(:,:,nk-1) = - ( q(1,2) * var(:,:,nk) + q(3,2) * var(:,:,nk-2) + & - q(4,2) * var(:,:,nk-3) + q(5,2) * var(:,:,nk-4) + & - q(6,2) * var(:,:,nk-5) + q(7,2) * var(:,:,nk-6) + & - q(8,2) * var(:,:,nk-7) ) * idel - dvar(:,:,nk-2) = - ( q(1,3) * var(:,:,nk) + q(2,3) * var(:,:,nk-1) + & - q(4,3) * var(:,:,nk-3) + q(5,3) * var(:,:,nk-4) + & - q(6,3) * var(:,:,nk-5) + q(7,3) * var(:,:,nk-6) + & - q(8,3) * var(:,:,nk-7) ) * idel - dvar(:,:,nk-3) = - ( q(1,4) * var(:,:,nk) + q(2,4) * var(:,:,nk-1) + & - q(3,4) * var(:,:,nk-2) + q(5,4) * var(:,:,nk-4) + & - q(6,4) * var(:,:,nk-5) + q(7,4) * var(:,:,nk-6) + & - q(8,4) * var(:,:,nk-7) ) * idel - 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(6,5) * var(:,:,nk-5) + q(7,5) * var(:,:,nk-6) + & - q(8,5) * var(:,:,nk-7) + & - q(9,5) * var(:,:,nk-8) ) * idel - dvar(:,:,nk-5) = - ( q(1,6) * var(:,:,nk) + q(2,6) * var(:,:,nk-1) + & - q(3,6) * var(:,:,nk-2) + q(4,6) * var(:,:,nk-3) + & - q(5,6) * var(:,:,nk-4) + q(7,6) * var(:,:,nk-6) + & - q(8,6) * var(:,:,nk-7) + q(9,6) * var(:,:,nk-8) + & - q(10,6) * var(:,:,nk-9) ) * idel - dvar(:,:,nk-6) = - ( q(1,7) * var(:,:,nk) + q(2,7) * var(:,:,nk-1) + & - q(3,7) * var(:,:,nk-2) + q(4,7) * var(:,:,nk-3) + & - q(5,7) * var(:,:,nk-4) + q(6,7) * var(:,:,nk-5) + & - q(8,7) * var(:,:,nk-7) + q(9,7) * var(:,:,nk-8) + & - q(10,7) * var(:,:,nk-9) + & - q(11,7) * var(:,:,nk-10) ) * idel - dvar(:,:,nk-7) = - ( q(1,8) * var(:,:,nk) + q(2,8) * var(:,:,nk-1) + & - q(3,8) * var(:,:,nk-2) + q(4,8) * var(:,:,nk-3) + & - q(5,8) * var(:,:,nk-4) + q(6,8) * var(:,:,nk-5) + & - q(7,8) * var(:,:,nk-6) + q(9,8) * var(:,:,nk-8) + & - q(10,8) * var(:,:,nk-9) + & - q(11,8) * var(:,:,nk-10) + & - q(12,8) * var(:,:,nk-11) ) * idel - kr = nk - 8 + if ( bb(1) == 0 ) then + kl = 1 + gsize + else + dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & + q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) + & + q(5,1) * var(:,:,5) + q(6,1) * var(:,:,6) + & + q(7,1) * var(:,:,7) + q(8,1) * var(:,:,8) ) * idel + dvar(:,:,2) = ( q(1,2) * var(:,:,1) + q(3,2) * var(:,:,3) + & + q(4,2) * var(:,:,4) + q(5,2) * var(:,:,5) + & + q(6,2) * var(:,:,6) + q(7,2) * var(:,:,7) + & + q(8,2) * var(:,:,8) ) * idel + dvar(:,:,3) = ( q(1,3) * var(:,:,1) + q(2,3) * var(:,:,2) + & + q(4,3) * var(:,:,4) + q(5,3) * var(:,:,5) + & + q(6,3) * var(:,:,6) + q(7,3) * var(:,:,7) + & + q(8,3) * var(:,:,8) ) * idel + dvar(:,:,4) = ( q(1,4) * var(:,:,1) + q(2,4) * var(:,:,2) + & + q(3,4) * var(:,:,3) + q(5,4) * var(:,:,5) + & + q(6,4) * var(:,:,6) + q(7,4) * var(:,:,7) + & + q(8,4) * var(:,:,8) ) * idel + dvar(:,:,5) = ( q(1,5) * var(:,:,1) + q(2,5) * var(:,:,2) + & + q(3,5) * var(:,:,3) + q(4,5) * var(:,:,4) + & + q(6,5) * var(:,:,6) + q(7,5) * var(:,:,7) + & + q(8,5) * var(:,:,8) + q(9,5) * var(:,:,9) ) * idel + dvar(:,:,6) = ( q(1,6) * var(:,:,1) + q(2,6) * var(:,:,2) + & + q(3,6) * var(:,:,3) + q(4,6) * var(:,:,4) + & + q(5,6) * var(:,:,5) + q(7,6) * var(:,:,7) + & + q(8,6) * var(:,:,8) + q(9,6) * var(:,:,9) + & + q(10,6) * var(:,:,10) ) * idel + dvar(:,:,7) = ( q(1,7) * var(:,:,1) + q(2,7) * var(:,:,2) + & + q(3,7) * var(:,:,3) + q(4,7) * var(:,:,4) + & + q(5,7) * var(:,:,5) + q(6,7) * var(:,:,6) + & + q(8,7) * var(:,:,8) + q(9,7) * var(:,:,9) + & + q(10,7) * var(:,:,10) + q(11,7) * var(:,:,11) ) * idel + dvar(:,:,8) = ( q(1,8) * var(:,:,1) + q(2,8) * var(:,:,2) + & + q(3,8) * var(:,:,3) + q(4,8) * var(:,:,4) + & + q(5,8) * var(:,:,5) + q(6,8) * var(:,:,6) + & + q(7,8) * var(:,:,7) + q(9,8) * var(:,:,9) + & + q(10,8) * var(:,:,10) + q(11,8) * var(:,:,11) + & + q(12,8) * var(:,:,12) ) * idel + kl = 9 + end if + if ( bb(2) == 0 ) then + kr = nk - gsize + else + 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) + & + q(5,1) * var(:,:,nk-4) + q(6,1) * var(:,:,nk-5) + & + q(7,1) * var(:,:,nk-6) + & + q(8,1) * var(:,:,nk-7) ) * idel + dvar(:,:,nk-1) = - ( q(1,2) * var(:,:,nk) + q(3,2) * var(:,:,nk-2) + & + q(4,2) * var(:,:,nk-3) + q(5,2) * var(:,:,nk-4) + & + q(6,2) * var(:,:,nk-5) + q(7,2) * var(:,:,nk-6) + & + q(8,2) * var(:,:,nk-7) ) * idel + dvar(:,:,nk-2) = - ( q(1,3) * var(:,:,nk) + q(2,3) * var(:,:,nk-1) + & + q(4,3) * var(:,:,nk-3) + q(5,3) * var(:,:,nk-4) + & + q(6,3) * var(:,:,nk-5) + q(7,3) * var(:,:,nk-6) + & + q(8,3) * var(:,:,nk-7) ) * idel + dvar(:,:,nk-3) = - ( q(1,4) * var(:,:,nk) + q(2,4) * var(:,:,nk-1) + & + q(3,4) * var(:,:,nk-2) + q(5,4) * var(:,:,nk-4) + & + q(6,4) * var(:,:,nk-5) + q(7,4) * var(:,:,nk-6) + & + q(8,4) * var(:,:,nk-7) ) * idel + 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(6,5) * var(:,:,nk-5) + q(7,5) * var(:,:,nk-6) + & + q(8,5) * var(:,:,nk-7) + & + q(9,5) * var(:,:,nk-8) ) * idel + dvar(:,:,nk-5) = - ( q(1,6) * var(:,:,nk) + q(2,6) * var(:,:,nk-1) + & + q(3,6) * var(:,:,nk-2) + q(4,6) * var(:,:,nk-3) + & + q(5,6) * var(:,:,nk-4) + q(7,6) * var(:,:,nk-6) + & + q(8,6) * var(:,:,nk-7) + q(9,6) * var(:,:,nk-8) + & + q(10,6) * var(:,:,nk-9) ) * idel + dvar(:,:,nk-6) = - ( q(1,7) * var(:,:,nk) + q(2,7) * var(:,:,nk-1) + & + q(3,7) * var(:,:,nk-2) + q(4,7) * var(:,:,nk-3) + & + q(5,7) * var(:,:,nk-4) + q(6,7) * var(:,:,nk-5) + & + q(8,7) * var(:,:,nk-7) + q(9,7) * var(:,:,nk-8) + & + q(10,7) * var(:,:,nk-9) + & + q(11,7) * var(:,:,nk-10) ) * idel + dvar(:,:,nk-7) = - ( q(1,8) * var(:,:,nk) + q(2,8) * var(:,:,nk-1) + & + q(3,8) * var(:,:,nk-2) + q(4,8) * var(:,:,nk-3) + & + q(5,8) * var(:,:,nk-4) + q(6,8) * var(:,:,nk-5) + & + q(7,8) * var(:,:,nk-6) + q(9,8) * var(:,:,nk-8) + & + q(10,8) * var(:,:,nk-9) + & + q(11,8) * var(:,:,nk-10) + & + q(12,8) * var(:,:,nk-11) ) * idel + kr = nk - 8 + end if + dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & + var(:,:,kl-1:kr-1) ) + & + a(2) * ( var(:,:,kl+2:kr+2) - & + var(:,:,kl-2:kr-2) ) + & + a(3) * ( var(:,:,kl+3:kr+3) - & + var(:,:,kl-3:kr-3) ) + & + a(4) * ( var(:,:,kl+4:kr+4) - & + var(:,:,kl-4:kr-4) ) ) * idel end if - dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & - var(:,:,kl-1:kr-1) ) + & - a(2) * ( var(:,:,kl+2:kr+2) - & - var(:,:,kl-2:kr-2) ) + & - a(3) * ( var(:,:,kl+3:kr+3) - & - var(:,:,kl-3:kr-3) ) + & - a(4) * ( var(:,:,kl+4:kr+4) - & - var(:,:,kl-4:kr-4) ) ) * idel end select direction end subroutine deriv_gf_8_4 diff --git a/src/Derivatives_8_4_min_err_coeff.F90 b/src/Derivatives_8_4_min_err_coeff.F90 index a06ee18..42eae82 100644 --- a/src/Derivatives_8_4_min_err_coeff.F90 +++ b/src/Derivatives_8_4_min_err_coeff.F90 @@ -230,194 +230,202 @@ subroutine deriv_gf_8_4_opt ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) a(4) * ( var(il+4:ir+4,:,:) - & var(il-4:ir-4,:,:) ) ) * idel case (1) direction - if ( bb(1) == 0 ) then - jl = 1 + gsize - else - dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & - q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) + & - q(5,1) * var(:,5,:) + q(6,1) * var(:,6,:) + & - q(7,1) * var(:,7,:) + q(8,1) * var(:,8,:) ) * idel - dvar(:,2,:) = ( q(1,2) * var(:,1,:) + q(3,2) * var(:,3,:) + & - q(4,2) * var(:,4,:) + q(5,2) * var(:,5,:) + & - q(6,2) * var(:,6,:) + q(7,2) * var(:,7,:) + & - q(8,2) * var(:,8,:) ) * idel - dvar(:,3,:) = ( q(1,3) * var(:,1,:) + q(2,3) * var(:,2,:) + & - q(4,3) * var(:,4,:) + q(5,3) * var(:,5,:) + & - q(6,3) * var(:,6,:) + q(7,3) * var(:,7,:) + & - q(8,3) * var(:,8,:) ) * idel - dvar(:,4,:) = ( q(1,4) * var(:,1,:) + q(2,4) * var(:,2,:) + & - q(3,4) * var(:,3,:) + q(5,4) * var(:,5,:) + & - q(6,4) * var(:,6,:) + q(7,4) * var(:,7,:) + & - q(8,4) * var(:,8,:) ) * idel - dvar(:,5,:) = ( q(1,5) * var(:,1,:) + q(2,5) * var(:,2,:) + & - q(3,5) * var(:,3,:) + q(4,5) * var(:,4,:) + & - q(6,5) * var(:,6,:) + q(7,5) * var(:,7,:) + & - q(8,5) * var(:,8,:) + q(9,5) * var(:,9,:) ) * idel - dvar(:,6,:) = ( q(1,6) * var(:,1,:) + q(2,6) * var(:,2,:) + & - q(3,6) * var(:,3,:) + q(4,6) * var(:,4,:) + & - q(5,6) * var(:,5,:) + q(7,6) * var(:,7,:) + & - q(8,6) * var(:,8,:) + q(9,6) * var(:,9,:) + & - q(10,6) * var(:,10,:) ) * idel - dvar(:,7,:) = ( q(1,7) * var(:,1,:) + q(2,7) * var(:,2,:) + & - q(3,7) * var(:,3,:) + q(4,7) * var(:,4,:) + & - q(5,7) * var(:,5,:) + q(6,7) * var(:,6,:) + & - q(8,7) * var(:,8,:) + q(9,7) * var(:,9,:) + & - q(10,7) * var(:,10,:) + q(11,7) * var(:,11,:) ) * idel - dvar(:,8,:) = ( q(1,8) * var(:,1,:) + q(2,8) * var(:,2,:) + & - q(3,8) * var(:,3,:) + q(4,8) * var(:,4,:) + & - q(5,8) * var(:,5,:) + q(6,8) * var(:,6,:) + & - q(7,8) * var(:,7,:) + q(9,8) * var(:,9,:) + & - q(10,8) * var(:,10,:) + q(11,8) * var(:,11,:) + & - q(12,8) * var(:,12,:) ) * idel - jl = 9 - end if - if ( bb(2) == 0 ) then - jr = nj - gsize + if ( zero_derivs_y /= 0 ) then + dvar = zero else - 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,:) + & - q(5,1) * var(:,nj-4,:) + q(6,1) * var(:,nj-5,:) + & - q(7,1) * var(:,nj-6,:) + & - q(8,1) * var(:,nj-7,:) ) * idel - dvar(:,nj-1,:) = - ( q(1,2) * var(:,nj,:) + q(3,2) * var(:,nj-2,:) + & - q(4,2) * var(:,nj-3,:) + q(5,2) * var(:,nj-4,:) + & - q(6,2) * var(:,nj-5,:) + q(7,2) * var(:,nj-6,:) + & - q(8,2) * var(:,nj-7,:) ) * idel - dvar(:,nj-2,:) = - ( q(1,3) * var(:,nj,:) + q(2,3) * var(:,nj-1,:) + & - q(4,3) * var(:,nj-3,:) + q(5,3) * var(:,nj-4,:) + & - q(6,3) * var(:,nj-5,:) + q(7,3) * var(:,nj-6,:) + & - q(8,3) * var(:,nj-7,:) ) * idel - dvar(:,nj-3,:) = - ( q(1,4) * var(:,nj,:) + q(2,4) * var(:,nj-1,:) + & - q(3,4) * var(:,nj-2,:) + q(5,4) * var(:,nj-4,:) + & - q(6,4) * var(:,nj-5,:) + q(7,4) * var(:,nj-6,:) + & - q(8,4) * var(:,nj-7,:) ) * idel - 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(6,5) * var(:,nj-5,:) + q(7,5) * var(:,nj-6,:) + & - q(8,5) * var(:,nj-7,:) + & - q(9,5) * var(:,nj-8,:) ) * idel - dvar(:,nj-5,:) = - ( q(1,6) * var(:,nj,:) + q(2,6) * var(:,nj-1,:) + & - q(3,6) * var(:,nj-2,:) + q(4,6) * var(:,nj-3,:) + & - q(5,6) * var(:,nj-4,:) + q(7,6) * var(:,nj-6,:) + & - q(8,6) * var(:,nj-7,:) + q(9,6) * var(:,nj-8,:) + & - q(10,6) * var(:,nj-9,:) ) * idel - dvar(:,nj-6,:) = - ( q(1,7) * var(:,nj,:) + q(2,7) * var(:,nj-1,:) + & - q(3,7) * var(:,nj-2,:) + q(4,7) * var(:,nj-3,:) + & - q(5,7) * var(:,nj-4,:) + q(6,7) * var(:,nj-5,:) + & - q(8,7) * var(:,nj-7,:) + q(9,7) * var(:,nj-8,:) + & - q(10,7) * var(:,nj-9,:) + & - q(11,7) * var(:,nj-10,:) ) * idel - dvar(:,nj-7,:) = - ( q(1,8) * var(:,nj,:) + q(2,8) * var(:,nj-1,:) + & - q(3,8) * var(:,nj-2,:) + q(4,8) * var(:,nj-3,:) + & - q(5,8) * var(:,nj-4,:) + q(6,8) * var(:,nj-5,:) + & - q(7,8) * var(:,nj-6,:) + q(9,8) * var(:,nj-8,:) + & - q(10,8) * var(:,nj-9,:) + & - q(11,8) * var(:,nj-10,:) + & - q(12,8) * var(:,nj-11,:) ) * idel - jr = nj - 8 + if ( bb(1) == 0 ) then + jl = 1 + gsize + else + dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & + q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) + & + q(5,1) * var(:,5,:) + q(6,1) * var(:,6,:) + & + q(7,1) * var(:,7,:) + q(8,1) * var(:,8,:) ) * idel + dvar(:,2,:) = ( q(1,2) * var(:,1,:) + q(3,2) * var(:,3,:) + & + q(4,2) * var(:,4,:) + q(5,2) * var(:,5,:) + & + q(6,2) * var(:,6,:) + q(7,2) * var(:,7,:) + & + q(8,2) * var(:,8,:) ) * idel + dvar(:,3,:) = ( q(1,3) * var(:,1,:) + q(2,3) * var(:,2,:) + & + q(4,3) * var(:,4,:) + q(5,3) * var(:,5,:) + & + q(6,3) * var(:,6,:) + q(7,3) * var(:,7,:) + & + q(8,3) * var(:,8,:) ) * idel + dvar(:,4,:) = ( q(1,4) * var(:,1,:) + q(2,4) * var(:,2,:) + & + q(3,4) * var(:,3,:) + q(5,4) * var(:,5,:) + & + q(6,4) * var(:,6,:) + q(7,4) * var(:,7,:) + & + q(8,4) * var(:,8,:) ) * idel + dvar(:,5,:) = ( q(1,5) * var(:,1,:) + q(2,5) * var(:,2,:) + & + q(3,5) * var(:,3,:) + q(4,5) * var(:,4,:) + & + q(6,5) * var(:,6,:) + q(7,5) * var(:,7,:) + & + q(8,5) * var(:,8,:) + q(9,5) * var(:,9,:) ) * idel + dvar(:,6,:) = ( q(1,6) * var(:,1,:) + q(2,6) * var(:,2,:) + & + q(3,6) * var(:,3,:) + q(4,6) * var(:,4,:) + & + q(5,6) * var(:,5,:) + q(7,6) * var(:,7,:) + & + q(8,6) * var(:,8,:) + q(9,6) * var(:,9,:) + & + q(10,6) * var(:,10,:) ) * idel + dvar(:,7,:) = ( q(1,7) * var(:,1,:) + q(2,7) * var(:,2,:) + & + q(3,7) * var(:,3,:) + q(4,7) * var(:,4,:) + & + q(5,7) * var(:,5,:) + q(6,7) * var(:,6,:) + & + q(8,7) * var(:,8,:) + q(9,7) * var(:,9,:) + & + q(10,7) * var(:,10,:) + q(11,7) * var(:,11,:) ) * idel + dvar(:,8,:) = ( q(1,8) * var(:,1,:) + q(2,8) * var(:,2,:) + & + q(3,8) * var(:,3,:) + q(4,8) * var(:,4,:) + & + q(5,8) * var(:,5,:) + q(6,8) * var(:,6,:) + & + q(7,8) * var(:,7,:) + q(9,8) * var(:,9,:) + & + q(10,8) * var(:,10,:) + q(11,8) * var(:,11,:) + & + q(12,8) * var(:,12,:) ) * idel + jl = 9 + end if + if ( bb(2) == 0 ) then + jr = nj - gsize + else + 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,:) + & + q(5,1) * var(:,nj-4,:) + q(6,1) * var(:,nj-5,:) + & + q(7,1) * var(:,nj-6,:) + & + q(8,1) * var(:,nj-7,:) ) * idel + dvar(:,nj-1,:) = - ( q(1,2) * var(:,nj,:) + q(3,2) * var(:,nj-2,:) + & + q(4,2) * var(:,nj-3,:) + q(5,2) * var(:,nj-4,:) + & + q(6,2) * var(:,nj-5,:) + q(7,2) * var(:,nj-6,:) + & + q(8,2) * var(:,nj-7,:) ) * idel + dvar(:,nj-2,:) = - ( q(1,3) * var(:,nj,:) + q(2,3) * var(:,nj-1,:) + & + q(4,3) * var(:,nj-3,:) + q(5,3) * var(:,nj-4,:) + & + q(6,3) * var(:,nj-5,:) + q(7,3) * var(:,nj-6,:) + & + q(8,3) * var(:,nj-7,:) ) * idel + dvar(:,nj-3,:) = - ( q(1,4) * var(:,nj,:) + q(2,4) * var(:,nj-1,:) + & + q(3,4) * var(:,nj-2,:) + q(5,4) * var(:,nj-4,:) + & + q(6,4) * var(:,nj-5,:) + q(7,4) * var(:,nj-6,:) + & + q(8,4) * var(:,nj-7,:) ) * idel + 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(6,5) * var(:,nj-5,:) + q(7,5) * var(:,nj-6,:) + & + q(8,5) * var(:,nj-7,:) + & + q(9,5) * var(:,nj-8,:) ) * idel + dvar(:,nj-5,:) = - ( q(1,6) * var(:,nj,:) + q(2,6) * var(:,nj-1,:) + & + q(3,6) * var(:,nj-2,:) + q(4,6) * var(:,nj-3,:) + & + q(5,6) * var(:,nj-4,:) + q(7,6) * var(:,nj-6,:) + & + q(8,6) * var(:,nj-7,:) + q(9,6) * var(:,nj-8,:) + & + q(10,6) * var(:,nj-9,:) ) * idel + dvar(:,nj-6,:) = - ( q(1,7) * var(:,nj,:) + q(2,7) * var(:,nj-1,:) + & + q(3,7) * var(:,nj-2,:) + q(4,7) * var(:,nj-3,:) + & + q(5,7) * var(:,nj-4,:) + q(6,7) * var(:,nj-5,:) + & + q(8,7) * var(:,nj-7,:) + q(9,7) * var(:,nj-8,:) + & + q(10,7) * var(:,nj-9,:) + & + q(11,7) * var(:,nj-10,:) ) * idel + dvar(:,nj-7,:) = - ( q(1,8) * var(:,nj,:) + q(2,8) * var(:,nj-1,:) + & + q(3,8) * var(:,nj-2,:) + q(4,8) * var(:,nj-3,:) + & + q(5,8) * var(:,nj-4,:) + q(6,8) * var(:,nj-5,:) + & + q(7,8) * var(:,nj-6,:) + q(9,8) * var(:,nj-8,:) + & + q(10,8) * var(:,nj-9,:) + & + q(11,8) * var(:,nj-10,:) + & + q(12,8) * var(:,nj-11,:) ) * idel + jr = nj - 8 + end if + dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & + var(:,jl-1:jr-1,:) ) + & + a(2) * ( var(:,jl+2:jr+2,:) - & + var(:,jl-2:jr-2,:) ) + & + a(3) * ( var(:,jl+3:jr+3,:) - & + var(:,jl-3:jr-3,:) ) + & + a(4) * ( var(:,jl+4:jr+4,:) - & + var(:,jl-4:jr-4,:) ) ) * idel end if - dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & - var(:,jl-1:jr-1,:) ) + & - a(2) * ( var(:,jl+2:jr+2,:) - & - var(:,jl-2:jr-2,:) ) + & - a(3) * ( var(:,jl+3:jr+3,:) - & - var(:,jl-3:jr-3,:) ) + & - a(4) * ( var(:,jl+4:jr+4,:) - & - var(:,jl-4:jr-4,:) ) ) * idel case (2) direction - if ( bb(1) == 0 ) then - kl = 1 + gsize + if ( zero_derivs_z /= 0 ) then + dvar = zero else - dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & - q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) + & - q(5,1) * var(:,:,5) + q(6,1) * var(:,:,6) + & - q(7,1) * var(:,:,7) + q(8,1) * var(:,:,8) ) * idel - dvar(:,:,2) = ( q(1,2) * var(:,:,1) + q(3,2) * var(:,:,3) + & - q(4,2) * var(:,:,4) + q(5,2) * var(:,:,5) + & - q(6,2) * var(:,:,6) + q(7,2) * var(:,:,7) + & - q(8,2) * var(:,:,8) ) * idel - dvar(:,:,3) = ( q(1,3) * var(:,:,1) + q(2,3) * var(:,:,2) + & - q(4,3) * var(:,:,4) + q(5,3) * var(:,:,5) + & - q(6,3) * var(:,:,6) + q(7,3) * var(:,:,7) + & - q(8,3) * var(:,:,8) ) * idel - dvar(:,:,4) = ( q(1,4) * var(:,:,1) + q(2,4) * var(:,:,2) + & - q(3,4) * var(:,:,3) + q(5,4) * var(:,:,5) + & - q(6,4) * var(:,:,6) + q(7,4) * var(:,:,7) + & - q(8,4) * var(:,:,8) ) * idel - dvar(:,:,5) = ( q(1,5) * var(:,:,1) + q(2,5) * var(:,:,2) + & - q(3,5) * var(:,:,3) + q(4,5) * var(:,:,4) + & - q(6,5) * var(:,:,6) + q(7,5) * var(:,:,7) + & - q(8,5) * var(:,:,8) + q(9,5) * var(:,:,9) ) * idel - dvar(:,:,6) = ( q(1,6) * var(:,:,1) + q(2,6) * var(:,:,2) + & - q(3,6) * var(:,:,3) + q(4,6) * var(:,:,4) + & - q(5,6) * var(:,:,5) + q(7,6) * var(:,:,7) + & - q(8,6) * var(:,:,8) + q(9,6) * var(:,:,9) + & - q(10,6) * var(:,:,10) ) * idel - dvar(:,:,7) = ( q(1,7) * var(:,:,1) + q(2,7) * var(:,:,2) + & - q(3,7) * var(:,:,3) + q(4,7) * var(:,:,4) + & - q(5,7) * var(:,:,5) + q(6,7) * var(:,:,6) + & - q(8,7) * var(:,:,8) + q(9,7) * var(:,:,9) + & - q(10,7) * var(:,:,10) + q(11,7) * var(:,:,11) ) * idel - dvar(:,:,8) = ( q(1,8) * var(:,:,1) + q(2,8) * var(:,:,2) + & - q(3,8) * var(:,:,3) + q(4,8) * var(:,:,4) + & - q(5,8) * var(:,:,5) + q(6,8) * var(:,:,6) + & - q(7,8) * var(:,:,7) + q(9,8) * var(:,:,9) + & - q(10,8) * var(:,:,10) + q(11,8) * var(:,:,11) + & - q(12,8) * var(:,:,12) ) * idel - kl = 9 - end if - if ( bb(2) == 0 ) then - kr = nk - gsize - else - 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) + & - q(5,1) * var(:,:,nk-4) + q(6,1) * var(:,:,nk-5) + & - q(7,1) * var(:,:,nk-6) + & - q(8,1) * var(:,:,nk-7) ) * idel - dvar(:,:,nk-1) = - ( q(1,2) * var(:,:,nk) + q(3,2) * var(:,:,nk-2) + & - q(4,2) * var(:,:,nk-3) + q(5,2) * var(:,:,nk-4) + & - q(6,2) * var(:,:,nk-5) + q(7,2) * var(:,:,nk-6) + & - q(8,2) * var(:,:,nk-7) ) * idel - dvar(:,:,nk-2) = - ( q(1,3) * var(:,:,nk) + q(2,3) * var(:,:,nk-1) + & - q(4,3) * var(:,:,nk-3) + q(5,3) * var(:,:,nk-4) + & - q(6,3) * var(:,:,nk-5) + q(7,3) * var(:,:,nk-6) + & - q(8,3) * var(:,:,nk-7) ) * idel - dvar(:,:,nk-3) = - ( q(1,4) * var(:,:,nk) + q(2,4) * var(:,:,nk-1) + & - q(3,4) * var(:,:,nk-2) + q(5,4) * var(:,:,nk-4) + & - q(6,4) * var(:,:,nk-5) + q(7,4) * var(:,:,nk-6) + & - q(8,4) * var(:,:,nk-7) ) * idel - 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(6,5) * var(:,:,nk-5) + q(7,5) * var(:,:,nk-6) + & - q(8,5) * var(:,:,nk-7) + & - q(9,5) * var(:,:,nk-8) ) * idel - dvar(:,:,nk-5) = - ( q(1,6) * var(:,:,nk) + q(2,6) * var(:,:,nk-1) + & - q(3,6) * var(:,:,nk-2) + q(4,6) * var(:,:,nk-3) + & - q(5,6) * var(:,:,nk-4) + q(7,6) * var(:,:,nk-6) + & - q(8,6) * var(:,:,nk-7) + q(9,6) * var(:,:,nk-8) + & - q(10,6) * var(:,:,nk-9) ) * idel - dvar(:,:,nk-6) = - ( q(1,7) * var(:,:,nk) + q(2,7) * var(:,:,nk-1) + & - q(3,7) * var(:,:,nk-2) + q(4,7) * var(:,:,nk-3) + & - q(5,7) * var(:,:,nk-4) + q(6,7) * var(:,:,nk-5) + & - q(8,7) * var(:,:,nk-7) + q(9,7) * var(:,:,nk-8) + & - q(10,7) * var(:,:,nk-9) + & - q(11,7) * var(:,:,nk-10) ) * idel - dvar(:,:,nk-7) = - ( q(1,8) * var(:,:,nk) + q(2,8) * var(:,:,nk-1) + & - q(3,8) * var(:,:,nk-2) + q(4,8) * var(:,:,nk-3) + & - q(5,8) * var(:,:,nk-4) + q(6,8) * var(:,:,nk-5) + & - q(7,8) * var(:,:,nk-6) + q(9,8) * var(:,:,nk-8) + & - q(10,8) * var(:,:,nk-9) + & - q(11,8) * var(:,:,nk-10) + & - q(12,8) * var(:,:,nk-11) ) * idel - kr = nk - 8 + if ( bb(1) == 0 ) then + kl = 1 + gsize + else + dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & + q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) + & + q(5,1) * var(:,:,5) + q(6,1) * var(:,:,6) + & + q(7,1) * var(:,:,7) + q(8,1) * var(:,:,8) ) * idel + dvar(:,:,2) = ( q(1,2) * var(:,:,1) + q(3,2) * var(:,:,3) + & + q(4,2) * var(:,:,4) + q(5,2) * var(:,:,5) + & + q(6,2) * var(:,:,6) + q(7,2) * var(:,:,7) + & + q(8,2) * var(:,:,8) ) * idel + dvar(:,:,3) = ( q(1,3) * var(:,:,1) + q(2,3) * var(:,:,2) + & + q(4,3) * var(:,:,4) + q(5,3) * var(:,:,5) + & + q(6,3) * var(:,:,6) + q(7,3) * var(:,:,7) + & + q(8,3) * var(:,:,8) ) * idel + dvar(:,:,4) = ( q(1,4) * var(:,:,1) + q(2,4) * var(:,:,2) + & + q(3,4) * var(:,:,3) + q(5,4) * var(:,:,5) + & + q(6,4) * var(:,:,6) + q(7,4) * var(:,:,7) + & + q(8,4) * var(:,:,8) ) * idel + dvar(:,:,5) = ( q(1,5) * var(:,:,1) + q(2,5) * var(:,:,2) + & + q(3,5) * var(:,:,3) + q(4,5) * var(:,:,4) + & + q(6,5) * var(:,:,6) + q(7,5) * var(:,:,7) + & + q(8,5) * var(:,:,8) + q(9,5) * var(:,:,9) ) * idel + dvar(:,:,6) = ( q(1,6) * var(:,:,1) + q(2,6) * var(:,:,2) + & + q(3,6) * var(:,:,3) + q(4,6) * var(:,:,4) + & + q(5,6) * var(:,:,5) + q(7,6) * var(:,:,7) + & + q(8,6) * var(:,:,8) + q(9,6) * var(:,:,9) + & + q(10,6) * var(:,:,10) ) * idel + dvar(:,:,7) = ( q(1,7) * var(:,:,1) + q(2,7) * var(:,:,2) + & + q(3,7) * var(:,:,3) + q(4,7) * var(:,:,4) + & + q(5,7) * var(:,:,5) + q(6,7) * var(:,:,6) + & + q(8,7) * var(:,:,8) + q(9,7) * var(:,:,9) + & + q(10,7) * var(:,:,10) + q(11,7) * var(:,:,11) ) * idel + dvar(:,:,8) = ( q(1,8) * var(:,:,1) + q(2,8) * var(:,:,2) + & + q(3,8) * var(:,:,3) + q(4,8) * var(:,:,4) + & + q(5,8) * var(:,:,5) + q(6,8) * var(:,:,6) + & + q(7,8) * var(:,:,7) + q(9,8) * var(:,:,9) + & + q(10,8) * var(:,:,10) + q(11,8) * var(:,:,11) + & + q(12,8) * var(:,:,12) ) * idel + kl = 9 + end if + if ( bb(2) == 0 ) then + kr = nk - gsize + else + 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) + & + q(5,1) * var(:,:,nk-4) + q(6,1) * var(:,:,nk-5) + & + q(7,1) * var(:,:,nk-6) + & + q(8,1) * var(:,:,nk-7) ) * idel + dvar(:,:,nk-1) = - ( q(1,2) * var(:,:,nk) + q(3,2) * var(:,:,nk-2) + & + q(4,2) * var(:,:,nk-3) + q(5,2) * var(:,:,nk-4) + & + q(6,2) * var(:,:,nk-5) + q(7,2) * var(:,:,nk-6) + & + q(8,2) * var(:,:,nk-7) ) * idel + dvar(:,:,nk-2) = - ( q(1,3) * var(:,:,nk) + q(2,3) * var(:,:,nk-1) + & + q(4,3) * var(:,:,nk-3) + q(5,3) * var(:,:,nk-4) + & + q(6,3) * var(:,:,nk-5) + q(7,3) * var(:,:,nk-6) + & + q(8,3) * var(:,:,nk-7) ) * idel + dvar(:,:,nk-3) = - ( q(1,4) * var(:,:,nk) + q(2,4) * var(:,:,nk-1) + & + q(3,4) * var(:,:,nk-2) + q(5,4) * var(:,:,nk-4) + & + q(6,4) * var(:,:,nk-5) + q(7,4) * var(:,:,nk-6) + & + q(8,4) * var(:,:,nk-7) ) * idel + 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(6,5) * var(:,:,nk-5) + q(7,5) * var(:,:,nk-6) + & + q(8,5) * var(:,:,nk-7) + & + q(9,5) * var(:,:,nk-8) ) * idel + dvar(:,:,nk-5) = - ( q(1,6) * var(:,:,nk) + q(2,6) * var(:,:,nk-1) + & + q(3,6) * var(:,:,nk-2) + q(4,6) * var(:,:,nk-3) + & + q(5,6) * var(:,:,nk-4) + q(7,6) * var(:,:,nk-6) + & + q(8,6) * var(:,:,nk-7) + q(9,6) * var(:,:,nk-8) + & + q(10,6) * var(:,:,nk-9) ) * idel + dvar(:,:,nk-6) = - ( q(1,7) * var(:,:,nk) + q(2,7) * var(:,:,nk-1) + & + q(3,7) * var(:,:,nk-2) + q(4,7) * var(:,:,nk-3) + & + q(5,7) * var(:,:,nk-4) + q(6,7) * var(:,:,nk-5) + & + q(8,7) * var(:,:,nk-7) + q(9,7) * var(:,:,nk-8) + & + q(10,7) * var(:,:,nk-9) + & + q(11,7) * var(:,:,nk-10) ) * idel + dvar(:,:,nk-7) = - ( q(1,8) * var(:,:,nk) + q(2,8) * var(:,:,nk-1) + & + q(3,8) * var(:,:,nk-2) + q(4,8) * var(:,:,nk-3) + & + q(5,8) * var(:,:,nk-4) + q(6,8) * var(:,:,nk-5) + & + q(7,8) * var(:,:,nk-6) + q(9,8) * var(:,:,nk-8) + & + q(10,8) * var(:,:,nk-9) + & + q(11,8) * var(:,:,nk-10) + & + q(12,8) * var(:,:,nk-11) ) * idel + kr = nk - 8 + end if + dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & + var(:,:,kl-1:kr-1) ) + & + a(2) * ( var(:,:,kl+2:kr+2) - & + var(:,:,kl-2:kr-2) ) + & + a(3) * ( var(:,:,kl+3:kr+3) - & + var(:,:,kl-3:kr-3) ) + & + a(4) * ( var(:,:,kl+4:kr+4) - & + var(:,:,kl-4:kr-4) ) ) * idel end if - dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & - var(:,:,kl-1:kr-1) ) + & - a(2) * ( var(:,:,kl+2:kr+2) - & - var(:,:,kl-2:kr-2) ) + & - a(3) * ( var(:,:,kl+3:kr+3) - & - var(:,:,kl-3:kr-3) ) + & - a(4) * ( var(:,:,kl+4:kr+4) - & - var(:,:,kl-4:kr-4) ) ) * idel end select direction end subroutine deriv_gf_8_4_opt diff --git a/src/Dissipation_2_1.F90 b/src/Dissipation_2_1.F90 index 0904b33..0b625f4 100644 --- a/src/Dissipation_2_1.F90 +++ b/src/Dissipation_2_1.F90 @@ -60,75 +60,79 @@ subroutine dissipation_2_1 (var, ni, nj, nk, bb, gsize, delta, epsilon, rhs) ( var(il-1:ir-1,:,:) + & var(il+1:ir+1,:,:) ) ) * idel - call set_coeff ( a ) - - if ( scale_with_h > 0 ) then - idel = epsilon / ( 4 * delta(2) ) - else - idel = epsilon / 4 - 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,:) ) * idel - rhs(:,2,:) = rhs(:,2,:) + & - ( a(1,2) * var(:,1,:) + a(2,2) * var(:,2,:) + & - a(3,2) * var(:,3,:) ) * idel - - jl = 3 - end if - if ( bb(4) == 0 ) then - jr = nj - gsize(2) - else - rhs(:,nj-1,:) = rhs(:,nj-1,:) + & - ( a(1,2) * var(:,nj,:) + a(2,2) * var(:,nj-1,:) + & - a(3,2) * var(:,nj-2,:) ) * idel - rhs(:,nj,:) = rhs(:,nj,:) + & - ( a(1,1) * var(:,nj,:) + a(2,1) * var(:,nj-1,:) ) * idel - - jr = nj - 2 + if ( zero_derivs_y == 0 ) then + call set_coeff ( a ) + + if ( scale_with_h > 0 ) then + idel = epsilon / ( 4 * delta(2) ) + else + idel = epsilon / 4 + 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,:) ) * idel + rhs(:,2,:) = rhs(:,2,:) + & + ( a(1,2) * var(:,1,:) + a(2,2) * var(:,2,:) + & + a(3,2) * var(:,3,:) ) * idel + + jl = 3 + end if + if ( bb(4) == 0 ) then + jr = nj - gsize(2) + else + rhs(:,nj-1,:) = rhs(:,nj-1,:) + & + ( a(1,2) * var(:,nj,:) + a(2,2) * var(:,nj-1,:) + & + a(3,2) * var(:,nj-2,:) ) * idel + rhs(:,nj,:) = rhs(:,nj,:) + & + ( a(1,1) * var(:,nj,:) + a(2,1) * var(:,nj-1,:) ) * idel + + jr = nj - 2 + end if + rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & + ( -2.0_wp * var(:,jl:jr,:) + & + ( var(:,jl-1:jr-1,:) + & + var(:,jl+1:jr+1,:) ) ) * idel end if - rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & - ( -2.0_wp * var(:,jl:jr,:) + & - ( var(:,jl-1:jr-1,:) + & - var(:,jl+1:jr+1,:) ) ) * idel - - call set_coeff ( a ) - - if ( scale_with_h > 0 ) then - idel = epsilon / ( 4 * delta(3) ) - else - idel = epsilon / 4 - 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) ) * idel - rhs(:,:,2) = rhs(:,:,2) + & - ( a(1,2) * var(:,:,1) + a(2,2) * var(:,:,2) + & - a(3,2) * var(:,:,3) ) * idel - - kl = 3 - end if - if ( bb(6) == 0 ) then - kr = nk - gsize(3) - else - rhs(:,:,nk-1) = rhs(:,:,nk-1) + & - ( a(1,2) * var(:,:,nk) + a(2,2) * var(:,:,nk-1) + & - a(3,2) * var(:,:,nk-2) ) * idel - rhs(:,:,nk) = rhs(:,:,nk) + & - ( a(1,1) * var(:,:,nk) + a(2,1) * var(:,:,nk-1) ) * idel - kr = nk - 2 + if ( zero_derivs_z == 0 ) then + call set_coeff ( a ) + + if ( scale_with_h > 0 ) then + idel = epsilon / ( 4 * delta(3) ) + else + idel = epsilon / 4 + 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) ) * idel + rhs(:,:,2) = rhs(:,:,2) + & + ( a(1,2) * var(:,:,1) + a(2,2) * var(:,:,2) + & + a(3,2) * var(:,:,3) ) * idel + + kl = 3 + end if + if ( bb(6) == 0 ) then + kr = nk - gsize(3) + else + rhs(:,:,nk-1) = rhs(:,:,nk-1) + & + ( a(1,2) * var(:,:,nk) + a(2,2) * var(:,:,nk-1) + & + a(3,2) * var(:,:,nk-2) ) * idel + rhs(:,:,nk) = rhs(:,:,nk) + & + ( a(1,1) * var(:,:,nk) + a(2,1) * var(:,:,nk-1) ) * idel + + kr = nk - 2 + end if + rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & + ( -2.0_wp * var(:,:,kl:kr) + & + ( var(:,:,kl-1:kr-1) + & + var(:,:,kl+1:kr+1) ) ) * idel end if - rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & - ( -2.0_wp * var(:,:,kl:kr) + & - ( var(:,:,kl-1:kr-1) + & - var(:,:,kl+1:kr+1) ) ) * idel contains diff --git a/src/Dissipation_2_1_alt.F90 b/src/Dissipation_2_1_alt.F90 index 49ee0f0..3ee91b4 100644 --- a/src/Dissipation_2_1_alt.F90 +++ b/src/Dissipation_2_1_alt.F90 @@ -60,75 +60,79 @@ subroutine dissipation_2_1_alt (var, ni, nj, nk, bb, gsize, delta, epsilon, rhs) ( var(il-2:ir-2,:,:) + & var(il+2:ir+2,:,:) ) ) * idel - call set_coeff ( a ) - - idel = epsilon / ( 16 * delta(2) ) - - 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,:) ) * idel - - jl = 3 - end if - if ( bb(4) == 0 ) then - jr = nj - gsize(2) - else - 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,:) ) * idel - rhs(:,nj,:) = rhs(:,nj,:) + & - ( a(1,1) * var(:,nj,:) + a(2,1) * var(:,nj-1,:) + & - a(3,1) * var(:,nj-2,:) ) * idel - - jr = nj - 2 + if ( zero_derivs_y == 0 ) then + call set_coeff ( a ) + + idel = epsilon / ( 16 * delta(2) ) + + 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,:) ) * idel + + jl = 3 + end if + if ( bb(4) == 0 ) then + jr = nj - gsize(2) + else + 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,:) ) * idel + rhs(:,nj,:) = rhs(:,nj,:) + & + ( a(1,1) * var(:,nj,:) + a(2,1) * var(:,nj-1,:) + & + a(3,1) * var(:,nj-2,:) ) * idel + + jr = nj - 2 + end if + rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & + ( -6.0_wp * var(:,jl:jr,:) + & + 4.0_wp * ( var(:,jl-1:jr-1,:) + & + var(:,jl+1:jr+1,:) ) - & + ( var(:,jl-2:jr-2,:) + & + var(:,jl+2:jr+2,:) ) ) * idel end if - rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & - ( -6.0_wp * var(:,jl:jr,:) + & - 4.0_wp * ( var(:,jl-1:jr-1,:) + & - var(:,jl+1:jr+1,:) ) - & - ( var(:,jl-2:jr-2,:) + & - var(:,jl+2:jr+2,:) ) ) * idel - call set_coeff ( a ) - - idel = epsilon / ( 16 * delta(3) ) - - 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) ) * idel - - kl = 3 - end if - if ( bb(6) == 0 ) then - kr = nk - gsize(3) - else - 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) ) * idel - rhs(:,:,nk) = rhs(:,:,nk) + & - ( a(1,1) * var(:,:,nk) + a(2,1) * var(:,:,nk-1) + & - a(3,1) * var(:,:,nk-2) ) * idel - - kr = nk - 2 + if ( zero_derivs_z == 0 ) then + call set_coeff ( a ) + + idel = epsilon / ( 16 * delta(3) ) + + 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) ) * idel + + kl = 3 + end if + if ( bb(6) == 0 ) then + kr = nk - gsize(3) + else + 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) ) * idel + rhs(:,:,nk) = rhs(:,:,nk) + & + ( a(1,1) * var(:,:,nk) + a(2,1) * var(:,:,nk-1) + & + a(3,1) * var(:,:,nk-2) ) * idel + + kr = nk - 2 + end if + rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & + ( -6.0_wp * var(:,:,kl:kr) + & + 4.0_wp * ( var(:,:,kl-1:kr-1) + & + var(:,:,kl+1:kr+1) ) - & + ( var(:,:,kl-2:kr-2) + & + var(:,:,kl+2:kr+2) ) ) * idel end if - rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & - ( -6.0_wp * var(:,:,kl:kr) + & - 4.0_wp * ( var(:,:,kl-1:kr-1) + & - var(:,:,kl+1:kr+1) ) - & - ( var(:,:,kl-2:kr-2) + & - var(:,:,kl+2:kr+2) ) ) * idel contains diff --git a/src/Dissipation_4_2.F90 b/src/Dissipation_4_2.F90 index f43073e..95f7fde 100644 --- a/src/Dissipation_4_2.F90 +++ b/src/Dissipation_4_2.F90 @@ -80,115 +80,119 @@ subroutine dissipation_4_2 (var, ni, nj, nk, bb, gsize, delta, epsilon, rhs) ( var(il-2:ir-2,:,:) + & var(il+2:ir+2,:,:) ) ) * idel - call set_coeff ( a ) - - if ( scale_with_h > 0 ) then - idel = epsilon / ( 16 * delta(2) ) - else - idel = epsilon / 16 - 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,:) ) * 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,:) ) * 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,:) ) * idel - rhs(:,4,:) = rhs(:,4,:) + & - ( 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,:) ) * idel - - jl = 5 - end if - if ( bb(4) == 0 ) then - jr = nj - gsize(2) - else - rhs(:,nj-3,:) = rhs(:,nj-3,:) + & - ( 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,:) ) * 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,:) ) * 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,:) ) * idel - rhs(:,nj,:) = rhs(:,nj,:) + & - ( a(1,1) * var(:,nj,:) + a(2,1) * var(:,nj-1,:) + & - a(3,1) * var(:,nj-2,:) ) * idel - - jr = nj - 4 + if ( zero_derivs_y == 0 ) then + call set_coeff ( a ) + + if ( scale_with_h > 0 ) then + idel = epsilon / ( 16 * delta(2) ) + else + idel = epsilon / 16 + 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,:) ) * 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,:) ) * 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,:) ) * idel + rhs(:,4,:) = rhs(:,4,:) + & + ( 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,:) ) * idel + + jl = 5 + end if + if ( bb(4) == 0 ) then + jr = nj - gsize(2) + else + rhs(:,nj-3,:) = rhs(:,nj-3,:) + & + ( 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,:) ) * 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,:) ) * 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,:) ) * idel + rhs(:,nj,:) = rhs(:,nj,:) + & + ( a(1,1) * var(:,nj,:) + a(2,1) * var(:,nj-1,:) + & + a(3,1) * var(:,nj-2,:) ) * idel + + jr = nj - 4 + end if + rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & + ( -6.0_wp * var(:,jl:jr,:) + & + 4.0_wp * ( var(:,jl-1:jr-1,:) + & + var(:,jl+1:jr+1,:) ) - & + ( var(:,jl-2:jr-2,:) + & + var(:,jl+2:jr+2,:) ) ) * idel end if - rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & - ( -6.0_wp * var(:,jl:jr,:) + & - 4.0_wp * ( var(:,jl-1:jr-1,:) + & - var(:,jl+1:jr+1,:) ) - & - ( var(:,jl-2:jr-2,:) + & - var(:,jl+2:jr+2,:) ) ) * idel - call set_coeff ( a ) - - if ( scale_with_h > 0 ) then - idel = epsilon / ( 16 * delta(3) ) - else - idel = epsilon / 16 - 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) ) * 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) ) * 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) ) * idel - rhs(:,:,4) = rhs(:,:,4) + & - ( 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) ) * idel - - kl = 5 - end if - if ( bb(6) == 0 ) then - kr = nk - gsize(3) - else - rhs(:,:,nk-3) = rhs(:,:,nk-3) + & - ( 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) ) * 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) ) * 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) ) * idel - rhs(:,:,nk) = rhs(:,:,nk) + & - ( a(1,1) * var(:,:,nk) + a(2,1) * var(:,:,nk-1) + & - a(3,1) * var(:,:,nk-2) ) * idel - - kr = nk - 4 + if ( zero_derivs_z == 0 ) then + call set_coeff ( a ) + + if ( scale_with_h > 0 ) then + idel = epsilon / ( 16 * delta(3) ) + else + idel = epsilon / 16 + 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) ) * 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) ) * 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) ) * idel + rhs(:,:,4) = rhs(:,:,4) + & + ( 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) ) * idel + + kl = 5 + end if + if ( bb(6) == 0 ) then + kr = nk - gsize(3) + else + rhs(:,:,nk-3) = rhs(:,:,nk-3) + & + ( 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) ) * 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) ) * 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) ) * idel + rhs(:,:,nk) = rhs(:,:,nk) + & + ( a(1,1) * var(:,:,nk) + a(2,1) * var(:,:,nk-1) + & + a(3,1) * var(:,:,nk-2) ) * idel + + kr = nk - 4 + end if + rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & + ( -6.0_wp * var(:,:,kl:kr) + & + 4.0_wp * ( var(:,:,kl-1:kr-1) + & + var(:,:,kl+1:kr+1) ) - & + ( var(:,:,kl-2:kr-2) + & + var(:,:,kl+2:kr+2) ) ) * idel end if - rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & - ( -6.0_wp * var(:,:,kl:kr) + & - 4.0_wp * ( var(:,:,kl-1:kr-1) + & - var(:,:,kl+1:kr+1) ) - & - ( var(:,:,kl-2:kr-2) + & - var(:,:,kl+2:kr+2) ) ) * idel contains diff --git a/src/Dissipation_4_2_alt.F90 b/src/Dissipation_4_2_alt.F90 index c0a68ba..884f6b3 100644 --- a/src/Dissipation_4_2_alt.F90 +++ b/src/Dissipation_4_2_alt.F90 @@ -82,119 +82,123 @@ subroutine dissipation_4_2_alt (var, ni, nj, nk, bb, gsize, delta, epsilon, rhs) ( var(il-3:ir-3,:,:) + & var(il+3:ir+3,:,:) ) ) * idel - call set_coeff ( a ) - - idel = epsilon / ( 64 * delta(2) ) - - 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 - - jl = 5 - 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 - - jr = nj - 4 + if ( zero_derivs_y == 0 ) then + call set_coeff ( a ) + + idel = epsilon / ( 64 * delta(2) ) + + 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 + + jl = 5 + 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 + + jr = nj - 4 + end if + rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & + ( -20.0_wp * var(:,jl:jr,:) + & + 15.0_wp * ( var(:,jl-1:jr-1,:) + & + var(:,jl+1:jr+1,:) ) - & + 6.0_wp * ( var(:,jl-2:jr-2,:) + & + var(:,jl+2:jr+2,:) ) + & + ( var(:,jl-3:jr-3,:) + & + var(:,jl+3:jr+3,:) ) ) * idel end if - rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & - ( -20.0_wp * var(:,jl:jr,:) + & - 15.0_wp * ( var(:,jl-1:jr-1,:) + & - var(:,jl+1:jr+1,:) ) - & - 6.0_wp * ( var(:,jl-2:jr-2,:) + & - var(:,jl+2:jr+2,:) ) + & - ( var(:,jl-3:jr-3,:) + & - var(:,jl+3:jr+3,:) ) ) * idel - call set_coeff ( a ) - - idel = epsilon / ( 64 * delta(3) ) - - 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 - - kl = 5 - 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 - - kr = nk - 4 + if ( zero_derivs_z == 0 ) then + call set_coeff ( a ) + + idel = epsilon / ( 64 * delta(3) ) + + 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 + + kl = 5 + 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 + + kr = nk - 4 + end if + rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & + ( -20.0_wp * var(:,:,kl:kr) + & + 15.0_wp * ( var(:,:,kl-1:kr-1) + & + var(:,:,kl+1:kr+1) ) - & + 6.0_wp * ( var(:,:,kl-2:kr-2) + & + var(:,:,kl+2:kr+2) ) + & + ( var(:,:,kl-3:kr-3) + & + var(:,:,kl+3:kr+3) ) ) * idel end if - rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & - ( -20.0_wp * var(:,:,kl:kr) + & - 15.0_wp * ( var(:,:,kl-1:kr-1) + & - var(:,:,kl+1:kr+1) ) - & - 6.0_wp * ( var(:,:,kl-2:kr-2) + & - var(:,:,kl+2:kr+2) ) + & - ( var(:,:,kl-3:kr-3) + & - var(:,:,kl+3:kr+3) ) ) * idel contains diff --git a/src/Dissipation_4_3_min_err_coeff.F90 b/src/Dissipation_4_3_min_err_coeff.F90 index dc4178d..1a196ac 100644 --- a/src/Dissipation_4_3_min_err_coeff.F90 +++ b/src/Dissipation_4_3_min_err_coeff.F90 @@ -127,202 +127,206 @@ subroutine dissipation_4_3_opt (var, lsh, gsh, lbnd, bb, gsize, & deallocate ( a, d, b, h ) - allocate ( a(nj,nj), d(nj,nj), b(nj,nj), h(nj,nj) ) - a = 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 ) + if ( zero_derivs_y == 0 ) then + allocate ( a(nj,nj), d(nj,nj), b(nj,nj), h(nj,nj) ) + a = zero; d = zero; b = zero; h = zero; - call set_hmatrix ( h, bb(3:4) ) - - a = - transpose ( matmul ( h, matmul ( transpose(d), matmul ( b, d ) ) ) ) - - if ( scale_with_h > 0 ) then - idel = epsilon / ( 16 * delta(2) ) - else - idel = epsilon / 16 - 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,:) ) * 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 - 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 - 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 - - deallocate ( a, d, b, h ) - - allocate ( a(nk,nk), d(nk,nk), b(nk,nk), h(nk,nk) ) - a = 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 ) - - call set_hmatrix ( h, bb(5:6) ) - - a = - transpose ( matmul ( h, matmul ( transpose(d), matmul ( b, d ) ) ) ) - - if ( scale_with_h > 0 ) then - idel = epsilon / ( 16 * delta(3) ) - else - idel = epsilon / 16 + call set_dmatrix ( d, bb(3:4) ) + + call set_bmatrix ( b, bb(3:4), lsh(2), gsh(2), lbnd(2), delta(2), dfl ) + + call set_hmatrix ( h, bb(3:4) ) + + a = - transpose ( matmul ( h, matmul ( transpose(d), matmul ( b, d ) ) ) ) + + if ( scale_with_h > 0 ) then + idel = epsilon / ( 16 * delta(2) ) + else + idel = epsilon / 16 + 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,:) ) * 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 + 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 + 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 + + deallocate ( a, d, b, h ) 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) ) * 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 - 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 + if ( zero_derivs_z == 0 ) then + allocate ( a(nk,nk), d(nk,nk), b(nk,nk), h(nk,nk) ) + a = 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 ) + + call set_hmatrix ( h, bb(5:6) ) + + a = - transpose ( matmul ( h, matmul ( transpose(d), matmul ( b, d ) ) ) ) + + if ( scale_with_h > 0 ) then + idel = epsilon / ( 16 * delta(3) ) + else + idel = epsilon / 16 + 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) ) * 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 + 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 + 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 + + deallocate ( a, d, b, h ) 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 - - deallocate ( a, d, b, h ) - contains subroutine set_dmatrix ( d, bb ) diff --git a/src/Dissipation_4_3_min_err_coeff_alt.F90 b/src/Dissipation_4_3_min_err_coeff_alt.F90 index 4f103cd..ef619d0 100644 --- a/src/Dissipation_4_3_min_err_coeff_alt.F90 +++ b/src/Dissipation_4_3_min_err_coeff_alt.F90 @@ -182,307 +182,311 @@ subroutine dissipation_4_3_alt (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 + max(3,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 - max(3,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 + max(3,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 - - idel = epsilon / ( 64 * delta(2) ) - - - 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,:) ) * 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,:) ) * 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,:) ) * 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,:) ) * idel - - jl = 6 - - end if - - if ( bb(4) == 0 ) then - - jr = nj - gsize(2) - - else - - rhs(:,nj-4,:) = rhs(:,nj-4,:) + & - ( 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-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-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-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 - 5 - - 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 - max(3,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 + max(3,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 - max(3,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 - - idel = epsilon / ( 64 * delta(3) ) - - 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) ) * 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) ) * 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) ) * 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) ) * idel - - kl = 6 - + + idel = epsilon / ( 64 * delta(2) ) + + + 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,:) ) * 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,:) ) * 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,:) ) * 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,:) ) * idel + + jl = 6 + + end if + + if ( bb(4) == 0 ) then + + jr = nj - gsize(2) + + else + + rhs(:,nj-4,:) = rhs(:,nj-4,:) + & + ( 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-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-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-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 - 5 + + 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-4) = rhs(:,:,nk-4) + & - ( 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-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-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-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 - 5 - + 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 + max(3,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 - max(3,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 + + idel = epsilon / ( 64 * delta(3) ) + + 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) ) * 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) ) * 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) ) * 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) ) * idel + + kl = 6 + + end if + + if ( bb(6) == 0 ) then + + kr = nk - gsize(3) + + else + + rhs(:,:,nk-4) = rhs(:,:,nk-4) + & + ( 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-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-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-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 - 5 + + 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 ) diff --git a/src/Dissipation_6_3.F90 b/src/Dissipation_6_3.F90 index 1f0c74b..a93f84d 100644 --- a/src/Dissipation_6_3.F90 +++ b/src/Dissipation_6_3.F90 @@ -106,167 +106,171 @@ subroutine dissipation_6_3 (var, ni, nj, nk, bb, gsize, delta, epsilon, rhs) ( var(il-3:ir-3,:,:) + & var(il+3:ir+3,:,:) ) ) * idel - call set_coeff ( a ) - - 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,:) ) * 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 - rhs(:,5,:) = rhs(:,5,:) + & - ( 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,:) ) * idel - rhs(:,6,:) = rhs(:,6,:) + & - ( 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,:) ) * idel - - jl = 7 - end if - if ( bb(4) == 0 ) then - jr = nj - gsize(2) - else - rhs(:,nj-5,:) = rhs(:,nj-5,:) + & - ( a(3,6) * var(:,nj-2,:) + a(4,6) * var(:,nj-3,:) + & - a(5,6) * var(:,nj-4,:) + a(6,6) * var(:,nj-5,:) + & - a(7,6) * var(:,nj-6,:) + a(8,6) * var(:,nj-7,:) + & - a(9,6) * var(:,nj-8,:) ) * idel - rhs(:,nj-4,:) = rhs(:,nj-4,:) + & - ( a(2,5) * var(:,nj-1,:) + a(3,5) * var(:,nj-2,:) + & - a(4,5) * var(:,nj-3,:) + a(5,5) * var(:,nj-4,:) + & - a(6,5) * var(:,nj-5,:) + a(7,5) * var(:,nj-6,:) + & - a(8,5) * var(:,nj-7,:) ) * idel - 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 - - jr = nj - 6 + if ( zero_derivs_y == 0 ) then + call set_coeff ( a ) + + 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,:) ) * 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 + rhs(:,5,:) = rhs(:,5,:) + & + ( 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,:) ) * idel + rhs(:,6,:) = rhs(:,6,:) + & + ( 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,:) ) * idel + + jl = 7 + end if + if ( bb(4) == 0 ) then + jr = nj - gsize(2) + else + rhs(:,nj-5,:) = rhs(:,nj-5,:) + & + ( a(3,6) * var(:,nj-2,:) + a(4,6) * var(:,nj-3,:) + & + a(5,6) * var(:,nj-4,:) + a(6,6) * var(:,nj-5,:) + & + a(7,6) * var(:,nj-6,:) + a(8,6) * var(:,nj-7,:) + & + a(9,6) * var(:,nj-8,:) ) * idel + rhs(:,nj-4,:) = rhs(:,nj-4,:) + & + ( a(2,5) * var(:,nj-1,:) + a(3,5) * var(:,nj-2,:) + & + a(4,5) * var(:,nj-3,:) + a(5,5) * var(:,nj-4,:) + & + a(6,5) * var(:,nj-5,:) + a(7,5) * var(:,nj-6,:) + & + a(8,5) * var(:,nj-7,:) ) * idel + 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 + + jr = nj - 6 + end if + rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & + ( -20.0_wp * var(:,jl:jr,:) + & + 15.0_wp * ( var(:,jl-1:jr-1,:) + & + var(:,jl+1:jr+1,:) ) - & + 6.0_wp * ( var(:,jl-2:jr-2,:) + & + var(:,jl+2:jr+2,:) ) + & + ( var(:,jl-3:jr-3,:) + & + var(:,jl+3:jr+3,:) ) ) * idel end if - rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & - ( -20.0_wp * var(:,jl:jr,:) + & - 15.0_wp * ( var(:,jl-1:jr-1,:) + & - var(:,jl+1:jr+1,:) ) - & - 6.0_wp * ( var(:,jl-2:jr-2,:) + & - var(:,jl+2:jr+2,:) ) + & - ( var(:,jl-3:jr-3,:) + & - var(:,jl+3:jr+3,:) ) ) * idel - - call set_coeff ( a ) - - 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) ) * 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 - rhs(:,:,5) = rhs(:,:,5) + & - ( 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) ) * idel - rhs(:,:,6) = rhs(:,:,6) + & - ( 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) ) * idel - - kl = 7 - end if - if ( bb(6) == 0 ) then - kr = nk - gsize(3) - else - rhs(:,:,nk-5) = rhs(:,:,nk-5) + & - ( a(3,6) * var(:,:,nk-2) + a(4,6) * var(:,:,nk-3) + & - a(5,6) * var(:,:,nk-4) + a(6,6) * var(:,:,nk-5) + & - a(7,6) * var(:,:,nk-6) + a(8,6) * var(:,:,nk-7) + & - a(9,6) * var(:,:,nk-8) ) * idel - rhs(:,:,nk-4) = rhs(:,:,nk-4) + & - ( a(2,5) * var(:,:,nk-1) + a(3,5) * var(:,:,nk-2) + & - a(4,5) * var(:,:,nk-3) + a(5,5) * var(:,:,nk-4) + & - a(6,5) * var(:,:,nk-5) + a(7,5) * var(:,:,nk-6) + & - a(8,5) * var(:,:,nk-7) ) * idel - 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 - kr = nk - 6 + if ( zero_derivs_z == 0 ) then + call set_coeff ( a ) + + 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) ) * 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 + rhs(:,:,5) = rhs(:,:,5) + & + ( 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) ) * idel + rhs(:,:,6) = rhs(:,:,6) + & + ( 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) ) * idel + + kl = 7 + end if + if ( bb(6) == 0 ) then + kr = nk - gsize(3) + else + rhs(:,:,nk-5) = rhs(:,:,nk-5) + & + ( a(3,6) * var(:,:,nk-2) + a(4,6) * var(:,:,nk-3) + & + a(5,6) * var(:,:,nk-4) + a(6,6) * var(:,:,nk-5) + & + a(7,6) * var(:,:,nk-6) + a(8,6) * var(:,:,nk-7) + & + a(9,6) * var(:,:,nk-8) ) * idel + rhs(:,:,nk-4) = rhs(:,:,nk-4) + & + ( a(2,5) * var(:,:,nk-1) + a(3,5) * var(:,:,nk-2) + & + a(4,5) * var(:,:,nk-3) + a(5,5) * var(:,:,nk-4) + & + a(6,5) * var(:,:,nk-5) + a(7,5) * var(:,:,nk-6) + & + a(8,5) * var(:,:,nk-7) ) * idel + 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 + + kr = nk - 6 + end if + rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & + ( -20.0_wp * var(:,:,kl:kr) + & + 15.0_wp * ( var(:,:,kl-1:kr-1) + & + var(:,:,kl+1:kr+1) ) - & + 6.0_wp * ( var(:,:,kl-2:kr-2) + & + var(:,:,kl+2:kr+2) ) + & + ( var(:,:,kl-3:kr-3) + & + var(:,:,kl+3:kr+3) ) ) * idel end if - rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & - ( -20.0_wp * var(:,:,kl:kr) + & - 15.0_wp * ( var(:,:,kl-1:kr-1) + & - var(:,:,kl+1:kr+1) ) - & - 6.0_wp * ( var(:,:,kl-2:kr-2) + & - var(:,:,kl+2:kr+2) ) + & - ( var(:,:,kl-3:kr-3) + & - var(:,:,kl+3:kr+3) ) ) * idel contains diff --git a/src/Dissipation_6_3_alt.F90 b/src/Dissipation_6_3_alt.F90 index 4a19c11..5fd9bc9 100644 --- a/src/Dissipation_6_3_alt.F90 +++ b/src/Dissipation_6_3_alt.F90 @@ -111,178 +111,183 @@ subroutine dissipation_6_3_alt (var, ni, nj, nk, bb, gsize, delta, epsilon, rhs) ( var(il-4:ir-4,:,:) + & var(il+4:ir+4,:,:) ) ) * idel - call set_coeff ( a ) - - idel = epsilon / ( 256 * delta(2) ) - - 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,:) + & - a(5,1) * var(:,5,:) ) * 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,:) ) * 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,:) + a(8,4) * var(:,8,:) ) * 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,:) ) * idel - rhs(:,6,:) = rhs(:,6,:) + & - ( 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 - - jl = 7 + if ( zero_derivs_y == 0 ) then + call set_coeff ( a ) + + idel = epsilon / ( 256 * delta(2) ) + + 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,:) + & + a(5,1) * var(:,5,:) ) * 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,:) ) * 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,:) + a(8,4) * var(:,8,:) ) * 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,:) ) * idel + rhs(:,6,:) = rhs(:,6,:) + & + ( 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 + + jl = 7 + end if + if ( bb(4) == 0 ) then + jr = nj - gsize(2) + else + rhs(:,nj-5,:) = rhs(:,nj-5,:) + & + ( a(2,6) * var(:,nj-1,:) + a(3,6) * var(:,nj-2,:) + & + a(4,6) * var(:,nj-3,:) + a(5,6) * var(:,nj-4,:) + & + a(6,6) * var(:,nj-5,:) + a(7,6) * var(:,nj-6,:) + & + a(8,6) * var(:,nj-7,:) + a(9,6) * var(:,nj-8,:) + & + a(10,6) * var(:,nj-9,:) ) * idel + rhs(:,nj-4,:) = rhs(:,nj-4,:) + & + ( a(1,5) * var(:,nj,:) + a(2,5) * var(:,nj-1,:) + & + a(3,5) * var(:,nj-2,:) + a(4,5) * var(:,nj-3,:) + & + a(5,5) * var(:,nj-4,:) + a(6,5) * var(:,nj-5,:) + & + a(7,5) * var(:,nj-6,:) + a(8,5) * var(:,nj-7,:) + & + a(9,5) * var(:,nj-8,:) ) * idel + 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,:) + a(8,4) * var(:,nj-7,:) ) * 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,:) + & + a(7,3) * var(:,nj-6,:) ) * 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,:) + a(6,2) * var(:,nj-5,:) ) * 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,:) + & + a(5,1) * var(:,nj-4,:) ) * idel + + jr = nj - 6 + end if + rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & + ( -70.0_wp * var(:,jl:jr,:) + & + 56.0_wp * ( var(:,jl-1:jr-1,:) + & + var(:,jl+1:jr+1,:) ) - & + 28.0_wp * ( var(:,jl-2:jr-2,:) + & + var(:,jl+2:jr+2,:) ) + & + 8.0_wp * ( var(:,jl-3:jr-3,:) + & + var(:,jl+3:jr+3,:) ) - & + ( var(:,jl-4:jr-4,:) + & + var(:,jl+4:jr+4,:) ) ) * idel end if - if ( bb(4) == 0 ) then - jr = nj - gsize(2) - else - rhs(:,nj-5,:) = rhs(:,nj-5,:) + & - ( a(2,6) * var(:,nj-1,:) + a(3,6) * var(:,nj-2,:) + & - a(4,6) * var(:,nj-3,:) + a(5,6) * var(:,nj-4,:) + & - a(6,6) * var(:,nj-5,:) + a(7,6) * var(:,nj-6,:) + & - a(8,6) * var(:,nj-7,:) + a(9,6) * var(:,nj-8,:) + & - a(10,6) * var(:,nj-9,:) ) * idel - rhs(:,nj-4,:) = rhs(:,nj-4,:) + & - ( a(1,5) * var(:,nj,:) + a(2,5) * var(:,nj-1,:) + & - a(3,5) * var(:,nj-2,:) + a(4,5) * var(:,nj-3,:) + & - a(5,5) * var(:,nj-4,:) + a(6,5) * var(:,nj-5,:) + & - a(7,5) * var(:,nj-6,:) + a(8,5) * var(:,nj-7,:) + & - a(9,5) * var(:,nj-8,:) ) * idel - 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,:) + a(8,4) * var(:,nj-7,:) ) * 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,:) + & - a(7,3) * var(:,nj-6,:) ) * 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,:) + a(6,2) * var(:,nj-5,:) ) * 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,:) + & - a(5,1) * var(:,nj-4,:) ) * idel - jr = nj - 6 + if ( zero_derivs_z == 0 ) then + call set_coeff ( a ) + + idel = epsilon / ( 256 * delta(3) ) + + 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) + & + a(5,1) * var(:,:,5) ) * 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) ) * 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) + a(8,4) * var(:,:,8) ) * 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) ) * idel + rhs(:,:,6) = rhs(:,:,6) + & + ( 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 + + kl = 7 + end if + if ( bb(6) == 0 ) then + kr = nk - gsize(3) + else + rhs(:,:,nk-5) = rhs(:,:,nk-5) + & + ( a(2,6) * var(:,:,nk-1) + a(3,6) * var(:,:,nk-2) + & + a(4,6) * var(:,:,nk-3) + a(5,6) * var(:,:,nk-4) + & + a(6,6) * var(:,:,nk-5) + a(7,6) * var(:,:,nk-6) + & + a(8,6) * var(:,:,nk-7) + a(9,6) * var(:,:,nk-8) + & + a(10,6) * var(:,:,nk-9) ) * idel + rhs(:,:,nk-4) = rhs(:,:,nk-4) + & + ( a(1,5) * var(:,:,nk) + a(2,5) * var(:,:,nk-1) + & + a(3,5) * var(:,:,nk-2) + a(4,5) * var(:,:,nk-3) + & + a(5,5) * var(:,:,nk-4) + a(6,5) * var(:,:,nk-5) + & + a(7,5) * var(:,:,nk-6) + a(8,5) * var(:,:,nk-7) + & + a(9,5) * var(:,:,nk-8) ) * idel + 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) + a(8,4) * var(:,:,nk-7) ) * 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) + & + a(7,3) * var(:,:,nk-6) ) * 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) + a(6,2) * var(:,:,nk-5) ) * 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) + & + a(5,1) * var(:,:,nk-4) ) * idel + kr = nk - 6 + end if + rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & + ( -70.0_wp * var(:,:,kl:kr) + & + 56.0_wp * ( var(:,:,kl-1:kr-1) + & + var(:,:,kl+1:kr+1) ) - & + 28.0_wp * ( var(:,:,kl-2:kr-2) + & + var(:,:,kl+2:kr+2) ) + & + 8.0_wp * ( var(:,:,kl-3:kr-3) + & + var(:,:,kl+3:kr+3) ) - & + ( var(:,:,kl-4:kr-4) + & + var(:,:,kl+4:kr+4) ) ) * idel end if - rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & - ( -70.0_wp * var(:,jl:jr,:) + & - 56.0_wp * ( var(:,jl-1:jr-1,:) + & - var(:,jl+1:jr+1,:) ) - & - 28.0_wp * ( var(:,jl-2:jr-2,:) + & - var(:,jl+2:jr+2,:) ) + & - 8.0_wp * ( var(:,jl-3:jr-3,:) + & - var(:,jl+3:jr+3,:) ) - & - ( var(:,jl-4:jr-4,:) + & - var(:,jl+4:jr+4,:) ) ) * idel - - call set_coeff ( a ) - - idel = epsilon / ( 256 * delta(3) ) - - 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) + & - a(5,1) * var(:,:,5) ) * 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) ) * 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) + a(8,4) * var(:,:,8) ) * 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) ) * idel - rhs(:,:,6) = rhs(:,:,6) + & - ( 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 - kl = 7 - end if - if ( bb(6) == 0 ) then - kr = nk - gsize(3) - else - rhs(:,:,nk-5) = rhs(:,:,nk-5) + & - ( a(2,6) * var(:,:,nk-1) + a(3,6) * var(:,:,nk-2) + & - a(4,6) * var(:,:,nk-3) + a(5,6) * var(:,:,nk-4) + & - a(6,6) * var(:,:,nk-5) + a(7,6) * var(:,:,nk-6) + & - a(8,6) * var(:,:,nk-7) + a(9,6) * var(:,:,nk-8) + & - a(10,6) * var(:,:,nk-9) ) * idel - rhs(:,:,nk-4) = rhs(:,:,nk-4) + & - ( a(1,5) * var(:,:,nk) + a(2,5) * var(:,:,nk-1) + & - a(3,5) * var(:,:,nk-2) + a(4,5) * var(:,:,nk-3) + & - a(5,5) * var(:,:,nk-4) + a(6,5) * var(:,:,nk-5) + & - a(7,5) * var(:,:,nk-6) + a(8,5) * var(:,:,nk-7) + & - a(9,5) * var(:,:,nk-8) ) * idel - 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) + a(8,4) * var(:,:,nk-7) ) * 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) + & - a(7,3) * var(:,:,nk-6) ) * 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) + a(6,2) * var(:,:,nk-5) ) * 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) + & - a(5,1) * var(:,:,nk-4) ) * idel - kr = nk - 6 - end if - rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & - ( -70.0_wp * var(:,:,kl:kr) + & - 56.0_wp * ( var(:,:,kl-1:kr-1) + & - var(:,:,kl+1:kr+1) ) - & - 28.0_wp * ( var(:,:,kl-2:kr-2) + & - var(:,:,kl+2:kr+2) ) + & - 8.0_wp * ( var(:,:,kl-3:kr-3) + & - var(:,:,kl+3:kr+3) ) - & - ( var(:,:,kl-4:kr-4) + & - var(:,:,kl+4:kr+4) ) ) * idel contains subroutine set_coeff ( a ) 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 ) diff --git a/src/Dissipation_6_5_min_err_coeff_alt.F90 b/src/Dissipation_6_5_min_err_coeff_alt.F90 index 102ac5d..ac9b19a 100644 --- a/src/Dissipation_6_5_min_err_coeff_alt.F90 +++ b/src/Dissipation_6_5_min_err_coeff_alt.F90 @@ -208,358 +208,362 @@ subroutine dissipation_6_5_alt (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) ) - - call set_dmatrix ( d, bb(3:4) ) - - a = -transpose ( matmul ( h, matmul ( transpose(d), matmul ( b, d ) ) ) ) - - idel = epsdisl / ( 256.0_wp * delta(2) ) - - 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,:) + & - a(5,1) * var(:,5,:) ) * 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,:) + & - a(11,2) * var(:,11,:) ) * 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,:) + & - a(11,3) * var(:,11,:) ) * 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,:) + & - a(11,4) * var(:,11,:) ) * 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,:) + & - a(11,5) * var(:,11,:) ) * 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,:) + & - a(11,6) * var(:,11,:) ) * 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,:) + & - a(11,7) * var(:,11,:) ) * idel - - jl = 8 - - end if - - if ( bb(4) == 0 ) then - - jr = nj - gsize(2) - - else - - rhs(:,nj-6,:) = rhs(:,nj-6,:) + & - ( a(nj-10,nj-6) * var(:,nj-10,:) + & - 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-10,nj-5) * var(:,nj-10,:) + & - 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-10,nj-4) * var(:,nj-10,:) + & - 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-10,nj-3) * var(:,nj-10,:) + & - 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-10,nj-2) * var(:,nj-10,:) + & - 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-10,nj-1) * var(:,nj-10,:) + & - 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-4,nj) * var(:,nj-4,:) + & - 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-4,j) * var(:,j-4,:) + & - 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,:) + & - a(j+4,j) * var(:,j+4,:) ) * 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) ) - - call set_dmatrix ( d, bb(5:6) ) - - a = -transpose ( matmul ( h, matmul ( transpose(d), matmul ( b, d ) ) ) ) - - idel = epsdisl / ( 256.0_wp * delta(3) ) - - 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) + & - a(5,1) * var(:,:,5) ) * 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) + & - a(11,2) * var(:,:,11) ) * 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) + & - a(11,3) * var(:,:,11) ) * 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) + & - a(11,4) * var(:,:,11) ) * 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) + & - a(11,5) * var(:,:,11) ) * 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) + & - a(11,6) * var(:,:,11) ) * 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) + & - a(11,7) * var(:,:,11) ) * idel - - kl = 8 - + call set_bmatrix ( b, bb(3:4), lsh(2), gsh(2), lbnd(2), delta(2), dfl ) + + call set_hmatrix ( h, bb(3:4) ) + + call set_dmatrix ( d, bb(3:4) ) + + a = -transpose ( matmul ( h, matmul ( transpose(d), matmul ( b, d ) ) ) ) + + idel = epsdisl / ( 256.0_wp * delta(2) ) + + 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,:) + & + a(5,1) * var(:,5,:) ) * 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,:) + & + a(11,2) * var(:,11,:) ) * 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,:) + & + a(11,3) * var(:,11,:) ) * 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,:) + & + a(11,4) * var(:,11,:) ) * 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,:) + & + a(11,5) * var(:,11,:) ) * 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,:) + & + a(11,6) * var(:,11,:) ) * 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,:) + & + a(11,7) * var(:,11,:) ) * idel + + jl = 8 + + end if + + if ( bb(4) == 0 ) then + + jr = nj - gsize(2) + + else + + rhs(:,nj-6,:) = rhs(:,nj-6,:) + & + ( a(nj-10,nj-6) * var(:,nj-10,:) + & + 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-10,nj-5) * var(:,nj-10,:) + & + 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-10,nj-4) * var(:,nj-10,:) + & + 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-10,nj-3) * var(:,nj-10,:) + & + 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-10,nj-2) * var(:,nj-10,:) + & + 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-10,nj-1) * var(:,nj-10,:) + & + 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-4,nj) * var(:,nj-4,:) + & + 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-4,j) * var(:,j-4,:) + & + 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,:) + & + a(j+4,j) * var(:,j+4,:) ) * 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-10,nk-6) * var(:,:,nk-10) + & - 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-10,nk-5) * var(:,:,nk-10) + & - 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-10,nk-4) * var(:,:,nk-10) + & - 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-10,nk-3) * var(:,:,nk-10) + & - 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-10,nk-2) * var(:,:,nk-10) + & - 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-10,nk-1) * var(:,:,nk-10) + & - 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-4,nk) * var(:,:,nk-4) + & - 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) ) + + call set_dmatrix ( d, bb(5:6) ) + + a = -transpose ( matmul ( h, matmul ( transpose(d), matmul ( b, d ) ) ) ) + + idel = epsdisl / ( 256.0_wp * delta(3) ) + + 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) + & + a(5,1) * var(:,:,5) ) * 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) + & + a(11,2) * var(:,:,11) ) * 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) + & + a(11,3) * var(:,:,11) ) * 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) + & + a(11,4) * var(:,:,11) ) * 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) + & + a(11,5) * var(:,:,11) ) * 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) + & + a(11,6) * var(:,:,11) ) * 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) + & + a(11,7) * var(:,:,11) ) * idel + + kl = 8 + + end if + + if ( bb(6) == 0 ) then + + kr = nk - gsize(3) + + else + + rhs(:,:,nk-6) = rhs(:,:,nk-6) + & + ( a(nk-10,nk-6) * var(:,:,nk-10) + & + 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-10,nk-5) * var(:,:,nk-10) + & + 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-10,nk-4) * var(:,:,nk-10) + & + 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-10,nk-3) * var(:,:,nk-10) + & + 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-10,nk-2) * var(:,:,nk-10) + & + 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-10,nk-1) * var(:,:,nk-10) + & + 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-4,nk) * var(:,:,nk-4) + & + 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-4,k) * var(:,:,k-4) + & + 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) + & + a(k+4,k) * var(:,:,k+4) ) * idel + + end do + + deallocate ( a, d, b, h, tmp ) end if - do k = kl, kr - - rhs(:,:,k) = rhs(:,:,k) + & - ( a(k-4,k) * var(:,:,k-4) + & - 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) + & - a(k+4,k) * var(:,:,k+4) ) * idel - - end do - - deallocate ( a, d, b, h, tmp ) - contains subroutine set_dmatrix ( d, bb ) diff --git a/src/Dissipation_8_4.F90 b/src/Dissipation_8_4.F90 index ec98407..c81432b 100644 --- a/src/Dissipation_8_4.F90 +++ b/src/Dissipation_8_4.F90 @@ -139,234 +139,239 @@ subroutine dissipation_8_4 (var, ni, nj, nk, bb, gsize, delta, epsilon, rhs) ( var(il-4:ir-4,:,:) + & var(il+4:ir+4,:,:) ) ) * idel - call set_coeff ( a ) - - if ( scale_with_h > 0 ) then - idel = epsilon / ( 256 * delta(2) ) - else - idel = epsilon / 256 - 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,:) + & - a(5,1) * var(:,5,:) ) * 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,:) ) * 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,:) + a(8,4) * var(:,8,:) ) * 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,:) ) * idel - rhs(:,6,:) = rhs(:,6,:) + & - ( 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(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,:) + & - a(11,7) * var(:,11,:) ) * idel - rhs(:,8,:) = rhs(:,8,:) + & - ( a(4,8) * var(:,4,:) + a(5,8) * var(:,5,:) + & - a(6,8) * var(:,6,:) + a(7,8) * var(:,7,:) + & - a(8,8) * var(:,8,:) + a(9,8) * var(:,9,:) + & - a(10,8) * var(:,10,:) + a(11,8) * var(:,11,:) + & - a(12,8) * var(:,12,:) ) * idel - - jl = 9 - end if - if ( bb(4) == 0 ) then - jr = nj - gsize(2) - else - rhs(:,nj-7,:) = rhs(:,nj-7,:) + & - ( a(4,8) * var(:,nj-3,:) + a(5,8) * var(:,nj-4,:) + & - a(6,8) * var(:,nj-5,:) + a(7,8) * var(:,nj-6,:) + & - a(8,8) * var(:,nj-7,:) + a(9,8) * var(:,nj-8,:) + & - a(10,8) * var(:,nj-9,:) + a(11,8) * var(:,nj-10,:) + & - a(12,8) * var(:,nj-11,:) ) * idel - rhs(:,nj-6,:) = rhs(:,nj-6,:) + & - ( a(3,7) * var(:,nj-2,:) + a(4,7) * var(:,nj-3,:) + & - a(5,7) * var(:,nj-4,:) + a(6,7) * var(:,nj-5,:) + & - a(7,7) * var(:,nj-6,:) + a(8,7) * var(:,nj-7,:) + & - a(9,7) * var(:,nj-8,:) + a(10,7) * var(:,nj-9,:) + & - a(11,7) * var(:,nj-10,:) ) * idel - rhs(:,nj-5,:) = rhs(:,nj-5,:) + & - ( a(2,6) * var(:,nj-1,:) + a(3,6) * var(:,nj-2,:) + & - a(4,6) * var(:,nj-3,:) + a(5,6) * var(:,nj-4,:) + & - a(6,6) * var(:,nj-5,:) + a(7,6) * var(:,nj-6,:) + & - a(8,6) * var(:,nj-7,:) + a(9,6) * var(:,nj-8,:) + & - a(10,6) * var(:,nj-9,:) ) * idel - rhs(:,nj-4,:) = rhs(:,nj-4,:) + & - ( a(1,5) * var(:,nj,:) + a(2,5) * var(:,nj-1,:) + & - a(3,5) * var(:,nj-2,:) + a(4,5) * var(:,nj-3,:) + & - a(5,5) * var(:,nj-4,:) + a(6,5) * var(:,nj-5,:) + & - a(7,5) * var(:,nj-6,:) + a(8,5) * var(:,nj-7,:) + & - a(9,5) * var(:,nj-8,:) ) * idel - 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,:) + a(8,4) * var(:,nj-7,:) ) * 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,:) + & - a(7,3) * var(:,nj-6,:) ) * 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,:) + a(6,2) * var(:,nj-5,:) ) * 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,:) + & - a(5,1) * var(:,nj-4,:) ) * idel - - jr = nj - 8 + if ( zero_derivs_y == 0 ) then + call set_coeff ( a ) + + if ( scale_with_h > 0 ) then + idel = epsilon / ( 256 * delta(2) ) + else + idel = epsilon / 256 + 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,:) + & + a(5,1) * var(:,5,:) ) * 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,:) ) * 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,:) + a(8,4) * var(:,8,:) ) * 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,:) ) * idel + rhs(:,6,:) = rhs(:,6,:) + & + ( 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(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,:) + & + a(11,7) * var(:,11,:) ) * idel + rhs(:,8,:) = rhs(:,8,:) + & + ( a(4,8) * var(:,4,:) + a(5,8) * var(:,5,:) + & + a(6,8) * var(:,6,:) + a(7,8) * var(:,7,:) + & + a(8,8) * var(:,8,:) + a(9,8) * var(:,9,:) + & + a(10,8) * var(:,10,:) + a(11,8) * var(:,11,:) + & + a(12,8) * var(:,12,:) ) * idel + + jl = 9 + end if + if ( bb(4) == 0 ) then + jr = nj - gsize(2) + else + rhs(:,nj-7,:) = rhs(:,nj-7,:) + & + ( a(4,8) * var(:,nj-3,:) + a(5,8) * var(:,nj-4,:) + & + a(6,8) * var(:,nj-5,:) + a(7,8) * var(:,nj-6,:) + & + a(8,8) * var(:,nj-7,:) + a(9,8) * var(:,nj-8,:) + & + a(10,8) * var(:,nj-9,:) + a(11,8) * var(:,nj-10,:) + & + a(12,8) * var(:,nj-11,:) ) * idel + rhs(:,nj-6,:) = rhs(:,nj-6,:) + & + ( a(3,7) * var(:,nj-2,:) + a(4,7) * var(:,nj-3,:) + & + a(5,7) * var(:,nj-4,:) + a(6,7) * var(:,nj-5,:) + & + a(7,7) * var(:,nj-6,:) + a(8,7) * var(:,nj-7,:) + & + a(9,7) * var(:,nj-8,:) + a(10,7) * var(:,nj-9,:) + & + a(11,7) * var(:,nj-10,:) ) * idel + rhs(:,nj-5,:) = rhs(:,nj-5,:) + & + ( a(2,6) * var(:,nj-1,:) + a(3,6) * var(:,nj-2,:) + & + a(4,6) * var(:,nj-3,:) + a(5,6) * var(:,nj-4,:) + & + a(6,6) * var(:,nj-5,:) + a(7,6) * var(:,nj-6,:) + & + a(8,6) * var(:,nj-7,:) + a(9,6) * var(:,nj-8,:) + & + a(10,6) * var(:,nj-9,:) ) * idel + rhs(:,nj-4,:) = rhs(:,nj-4,:) + & + ( a(1,5) * var(:,nj,:) + a(2,5) * var(:,nj-1,:) + & + a(3,5) * var(:,nj-2,:) + a(4,5) * var(:,nj-3,:) + & + a(5,5) * var(:,nj-4,:) + a(6,5) * var(:,nj-5,:) + & + a(7,5) * var(:,nj-6,:) + a(8,5) * var(:,nj-7,:) + & + a(9,5) * var(:,nj-8,:) ) * idel + 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,:) + a(8,4) * var(:,nj-7,:) ) * 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,:) + & + a(7,3) * var(:,nj-6,:) ) * 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,:) + a(6,2) * var(:,nj-5,:) ) * 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,:) + & + a(5,1) * var(:,nj-4,:) ) * idel + + jr = nj - 8 + end if + rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & + ( -70.0_wp * var(:,jl:jr,:) + & + 56.0_wp * ( var(:,jl-1:jr-1,:) + & + var(:,jl+1:jr+1,:) ) - & + 28.0_wp * ( var(:,jl-2:jr-2,:) + & + var(:,jl+2:jr+2,:) ) + & + 8.0_wp * ( var(:,jl-3:jr-3,:) + & + var(:,jl+3:jr+3,:) ) - & + ( var(:,jl-4:jr-4,:) + & + var(:,jl+4:jr+4,:) ) ) * idel end if - rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & - ( -70.0_wp * var(:,jl:jr,:) + & - 56.0_wp * ( var(:,jl-1:jr-1,:) + & - var(:,jl+1:jr+1,:) ) - & - 28.0_wp * ( var(:,jl-2:jr-2,:) + & - var(:,jl+2:jr+2,:) ) + & - 8.0_wp * ( var(:,jl-3:jr-3,:) + & - var(:,jl+3:jr+3,:) ) - & - ( var(:,jl-4:jr-4,:) + & - var(:,jl+4:jr+4,:) ) ) * idel - - call set_coeff ( a ) - if ( scale_with_h > 0 ) then - idel = epsilon / ( 256 * delta(3) ) - else - idel = epsilon / 256 + if ( zero_derivs_z == 0 ) then + call set_coeff ( a ) + + if ( scale_with_h > 0 ) then + idel = epsilon / ( 256 * delta(3) ) + else + idel = epsilon / 256 + 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) + & + a(5,1) * var(:,:,5) ) * 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) ) * 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) + a(8,4) * var(:,:,8) ) * 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) ) * idel + rhs(:,:,6) = rhs(:,:,6) + & + ( 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(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) + & + a(11,7) * var(:,:,11) ) * idel + rhs(:,:,8) = rhs(:,:,8) + & + ( a(4,8) * var(:,:,4) + a(5,8) * var(:,:,5) + & + a(6,8) * var(:,:,6) + a(7,8) * var(:,:,7) + & + a(8,8) * var(:,:,8) + a(9,8) * var(:,:,9) + & + a(10,8) * var(:,:,10) + a(11,8) * var(:,:,11) + & + a(12,8) * var(:,:,12) ) * idel + + kl = 9 + end if + if ( bb(6) == 0 ) then + kr = nk - gsize(3) + else + rhs(:,:,nk-7) = rhs(:,:,nk-7) + & + ( a(4,8) * var(:,:,nk-3) + a(5,8) * var(:,:,nk-4) + & + a(6,8) * var(:,:,nk-5) + a(7,8) * var(:,:,nk-6) + & + a(8,8) * var(:,:,nk-7) + a(9,8) * var(:,:,nk-8) + & + a(10,8) * var(:,:,nk-9) + a(11,8) * var(:,:,nk-10) + & + a(12,8) * var(:,:,nk-11) ) * idel + rhs(:,:,nk-6) = rhs(:,:,nk-6) + & + ( a(3,7) * var(:,:,nk-2) + a(4,7) * var(:,:,nk-3) + & + a(5,7) * var(:,:,nk-4) + a(6,7) * var(:,:,nk-5) + & + a(7,7) * var(:,:,nk-6) + a(8,7) * var(:,:,nk-7) + & + a(9,7) * var(:,:,nk-8) + a(10,7) * var(:,:,nk-9) + & + a(11,7) * var(:,:,nk-10) ) * idel + rhs(:,:,nk-5) = rhs(:,:,nk-5) + & + ( a(2,6) * var(:,:,nk-1) + a(3,6) * var(:,:,nk-2) + & + a(4,6) * var(:,:,nk-3) + a(5,6) * var(:,:,nk-4) + & + a(6,6) * var(:,:,nk-5) + a(7,6) * var(:,:,nk-6) + & + a(8,6) * var(:,:,nk-7) + a(9,6) * var(:,:,nk-8) + & + a(10,6) * var(:,:,nk-9) ) * idel + rhs(:,:,nk-4) = rhs(:,:,nk-4) + & + ( a(1,5) * var(:,:,nk) + a(2,5) * var(:,:,nk-1) + & + a(3,5) * var(:,:,nk-2) + a(4,5) * var(:,:,nk-3) + & + a(5,5) * var(:,:,nk-4) + a(6,5) * var(:,:,nk-5) + & + a(7,5) * var(:,:,nk-6) + a(8,5) * var(:,:,nk-7) + & + a(9,5) * var(:,:,nk-8) ) * idel + 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) + a(8,4) * var(:,:,nk-7) ) * 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) + & + a(7,3) * var(:,:,nk-6) ) * 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) + a(6,2) * var(:,:,nk-5) ) * 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) + & + a(5,1) * var(:,:,nk-4) ) * idel + kr = nk - 8 + end if + rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & + ( -70.0_wp * var(:,:,kl:kr) + & + 56.0_wp * ( var(:,:,kl-1:kr-1) + & + var(:,:,kl+1:kr+1) ) - & + 28.0_wp * ( var(:,:,kl-2:kr-2) + & + var(:,:,kl+2:kr+2) ) + & + 8.0_wp * ( var(:,:,kl-3:kr-3) + & + var(:,:,kl+3:kr+3) ) - & + ( var(:,:,kl-4:kr-4) + & + var(:,:,kl+4:kr+4) ) ) * idel 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) + & - a(5,1) * var(:,:,5) ) * 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) ) * 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) + a(8,4) * var(:,:,8) ) * 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) ) * idel - rhs(:,:,6) = rhs(:,:,6) + & - ( 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(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) + & - a(11,7) * var(:,:,11) ) * idel - rhs(:,:,8) = rhs(:,:,8) + & - ( a(4,8) * var(:,:,4) + a(5,8) * var(:,:,5) + & - a(6,8) * var(:,:,6) + a(7,8) * var(:,:,7) + & - a(8,8) * var(:,:,8) + a(9,8) * var(:,:,9) + & - a(10,8) * var(:,:,10) + a(11,8) * var(:,:,11) + & - a(12,8) * var(:,:,12) ) * idel - - kl = 9 - end if - if ( bb(6) == 0 ) then - kr = nk - gsize(3) - else - rhs(:,:,nk-7) = rhs(:,:,nk-7) + & - ( a(4,8) * var(:,:,nk-3) + a(5,8) * var(:,:,nk-4) + & - a(6,8) * var(:,:,nk-5) + a(7,8) * var(:,:,nk-6) + & - a(8,8) * var(:,:,nk-7) + a(9,8) * var(:,:,nk-8) + & - a(10,8) * var(:,:,nk-9) + a(11,8) * var(:,:,nk-10) + & - a(12,8) * var(:,:,nk-11) ) * idel - rhs(:,:,nk-6) = rhs(:,:,nk-6) + & - ( a(3,7) * var(:,:,nk-2) + a(4,7) * var(:,:,nk-3) + & - a(5,7) * var(:,:,nk-4) + a(6,7) * var(:,:,nk-5) + & - a(7,7) * var(:,:,nk-6) + a(8,7) * var(:,:,nk-7) + & - a(9,7) * var(:,:,nk-8) + a(10,7) * var(:,:,nk-9) + & - a(11,7) * var(:,:,nk-10) ) * idel - rhs(:,:,nk-5) = rhs(:,:,nk-5) + & - ( a(2,6) * var(:,:,nk-1) + a(3,6) * var(:,:,nk-2) + & - a(4,6) * var(:,:,nk-3) + a(5,6) * var(:,:,nk-4) + & - a(6,6) * var(:,:,nk-5) + a(7,6) * var(:,:,nk-6) + & - a(8,6) * var(:,:,nk-7) + a(9,6) * var(:,:,nk-8) + & - a(10,6) * var(:,:,nk-9) ) * idel - rhs(:,:,nk-4) = rhs(:,:,nk-4) + & - ( a(1,5) * var(:,:,nk) + a(2,5) * var(:,:,nk-1) + & - a(3,5) * var(:,:,nk-2) + a(4,5) * var(:,:,nk-3) + & - a(5,5) * var(:,:,nk-4) + a(6,5) * var(:,:,nk-5) + & - a(7,5) * var(:,:,nk-6) + a(8,5) * var(:,:,nk-7) + & - a(9,5) * var(:,:,nk-8) ) * idel - 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) + a(8,4) * var(:,:,nk-7) ) * 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) + & - a(7,3) * var(:,:,nk-6) ) * 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) + a(6,2) * var(:,:,nk-5) ) * 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) + & - a(5,1) * var(:,:,nk-4) ) * idel - kr = nk - 8 - end if - rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & - ( -70.0_wp * var(:,:,kl:kr) + & - 56.0_wp * ( var(:,:,kl-1:kr-1) + & - var(:,:,kl+1:kr+1) ) - & - 28.0_wp * ( var(:,:,kl-2:kr-2) + & - var(:,:,kl+2:kr+2) ) + & - 8.0_wp * ( var(:,:,kl-3:kr-3) + & - var(:,:,kl+3:kr+3) ) - & - ( var(:,:,kl-4:kr-4) + & - var(:,:,kl+4:kr+4) ) ) * idel contains subroutine set_coeff ( a ) diff --git a/src/Dissipation_8_4_alt.F90 b/src/Dissipation_8_4_alt.F90 index 594b531..2cbd77d 100644 --- a/src/Dissipation_8_4_alt.F90 +++ b/src/Dissipation_8_4_alt.F90 @@ -147,251 +147,255 @@ subroutine dissipation_8_4_alt (var, ni, nj, nk, bb, gsize, delta, epsilon, rhs) ( var(il-5:ir-5,:,:) + & var(il+5:ir+5,:,:) ) ) * idel - call set_coeff ( a ) - - idel = epsilon / ( 1024 * delta(2) ) - - 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,:) + & - a(5,1) * var(:,5,:) + a(6,1) * var(:,6,:) ) * 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,:) + a(8,3) * var(:,8,:) ) * 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,:) ) * 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,:) + & - a(11,6) * var(:,11,:) ) * idel - rhs(:,7,:) = rhs(:,7,:) + & - ( 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,:) + a(11,7) * var(:,11,:) + & - a(12,7) * var(:,12,:) ) * idel - rhs(:,8,:) = rhs(:,8,:) + & - ( a(3,8) * var(:,3,:) + a(4,8) * var(:,4,:) + & - a(5,8) * var(:,5,:) + a(6,8) * var(:,6,:) + & - a(7,8) * var(:,7,:) + a(8,8) * var(:,8,:) + & - a(9,8) * var(:,9,:) + a(10,8) * var(:,10,:) + & - a(11,8) * var(:,11,:) + a(12,8) * var(:,12,:) + & - a(13,8) * var(:,13,:) ) * idel - - jl = 9 - end if - if ( bb(4) == 0 ) then - jr = nj - gsize(2) - else - rhs(:,nj-7,:) = rhs(:,nj-7,:) + & - ( a(3,8) * var(:,nj-2,:) + a(4,8) * var(:,nj-3,:) + & - a(5,8) * var(:,nj-4,:) + a(6,8) * var(:,nj-5,:) + & - a(7,8) * var(:,nj-6,:) + a(8,8) * var(:,nj-7,:) + & - a(9,8) * var(:,nj-8,:) + a(10,8) * var(:,nj-9,:) + & - a(11,8) * var(:,nj-10,:) + a(12,8) * var(:,nj-11,:) + & - a(13,8) * var(:,nj-12,:) ) * idel - rhs(:,nj-6,:) = rhs(:,nj-6,:) + & - ( a(2,7) * var(:,nj-1,:) + a(3,7) * var(:,nj-2,:) + & - a(4,7) * var(:,nj-3,:) + a(5,7) * var(:,nj-4,:) + & - a(6,7) * var(:,nj-5,:) + a(7,7) * var(:,nj-6,:) + & - a(8,7) * var(:,nj-7,:) + a(9,7) * var(:,nj-8,:) + & - a(10,7) * var(:,nj-9,:) + a(11,7) * var(:,nj-10,:) + & - a(12,7) * var(:,nj-11,:) ) * idel - rhs(:,nj-5,:) = rhs(:,nj-5,:) + & - ( a(1,6) * var(:,nj,:) + a(2,6) * var(:,nj-1,:) + & - a(3,6) * var(:,nj-2,:) + a(4,6) * var(:,nj-3,:) + & - a(5,6) * var(:,nj-4,:) + a(6,6) * var(:,nj-5,:) + & - a(7,6) * var(:,nj-6,:) + a(8,6) * var(:,nj-7,:) + & - a(9,6) * var(:,nj-8,:) + a(10,6) * var(:,nj-9,:) + & - a(11,6) * var(:,nj-10,:) ) * idel - rhs(:,nj-4,:) = rhs(:,nj-4,:) + & - ( a(1,5) * var(:,nj,:) + a(2,5) * var(:,nj-1,:) + & - a(3,5) * var(:,nj-2,:) + a(4,5) * var(:,nj-3,:) + & - a(5,5) * var(:,nj-4,:) + a(6,5) * var(:,nj-5,:) + & - a(7,5) * var(:,nj-6,:) + a(8,5) * var(:,nj-7,:) + & - a(9,5) * var(:,nj-8,:) + a(10,5) * var(:,nj-9,:) ) * idel - 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,:) + a(8,4) * var(:,nj-7,:) + & - a(9,4) * var(:,nj-8,:) ) * 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,:) + & - a(7,3) * var(:,nj-6,:) + a(8,3) * var(:,nj-7,:) ) * 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,:) + a(6,2) * var(:,nj-5,:) + & - a(7,2) * var(:,nj-6,:) ) * 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,:) + & - a(5,1) * var(:,nj-4,:) + a(6,1) * var(:,nj-5,:) ) * idel - - jr = nj - 8 + if ( zero_derivs_y == 0 ) then + call set_coeff ( a ) + + idel = epsilon / ( 1024 * delta(2) ) + + 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,:) + & + a(5,1) * var(:,5,:) + a(6,1) * var(:,6,:) ) * 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,:) + a(8,3) * var(:,8,:) ) * 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,:) ) * 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,:) + & + a(11,6) * var(:,11,:) ) * idel + rhs(:,7,:) = rhs(:,7,:) + & + ( 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,:) + a(11,7) * var(:,11,:) + & + a(12,7) * var(:,12,:) ) * idel + rhs(:,8,:) = rhs(:,8,:) + & + ( a(3,8) * var(:,3,:) + a(4,8) * var(:,4,:) + & + a(5,8) * var(:,5,:) + a(6,8) * var(:,6,:) + & + a(7,8) * var(:,7,:) + a(8,8) * var(:,8,:) + & + a(9,8) * var(:,9,:) + a(10,8) * var(:,10,:) + & + a(11,8) * var(:,11,:) + a(12,8) * var(:,12,:) + & + a(13,8) * var(:,13,:) ) * idel + + jl = 9 + end if + if ( bb(4) == 0 ) then + jr = nj - gsize(2) + else + rhs(:,nj-7,:) = rhs(:,nj-7,:) + & + ( a(3,8) * var(:,nj-2,:) + a(4,8) * var(:,nj-3,:) + & + a(5,8) * var(:,nj-4,:) + a(6,8) * var(:,nj-5,:) + & + a(7,8) * var(:,nj-6,:) + a(8,8) * var(:,nj-7,:) + & + a(9,8) * var(:,nj-8,:) + a(10,8) * var(:,nj-9,:) + & + a(11,8) * var(:,nj-10,:) + a(12,8) * var(:,nj-11,:) + & + a(13,8) * var(:,nj-12,:) ) * idel + rhs(:,nj-6,:) = rhs(:,nj-6,:) + & + ( a(2,7) * var(:,nj-1,:) + a(3,7) * var(:,nj-2,:) + & + a(4,7) * var(:,nj-3,:) + a(5,7) * var(:,nj-4,:) + & + a(6,7) * var(:,nj-5,:) + a(7,7) * var(:,nj-6,:) + & + a(8,7) * var(:,nj-7,:) + a(9,7) * var(:,nj-8,:) + & + a(10,7) * var(:,nj-9,:) + a(11,7) * var(:,nj-10,:) + & + a(12,7) * var(:,nj-11,:) ) * idel + rhs(:,nj-5,:) = rhs(:,nj-5,:) + & + ( a(1,6) * var(:,nj,:) + a(2,6) * var(:,nj-1,:) + & + a(3,6) * var(:,nj-2,:) + a(4,6) * var(:,nj-3,:) + & + a(5,6) * var(:,nj-4,:) + a(6,6) * var(:,nj-5,:) + & + a(7,6) * var(:,nj-6,:) + a(8,6) * var(:,nj-7,:) + & + a(9,6) * var(:,nj-8,:) + a(10,6) * var(:,nj-9,:) + & + a(11,6) * var(:,nj-10,:) ) * idel + rhs(:,nj-4,:) = rhs(:,nj-4,:) + & + ( a(1,5) * var(:,nj,:) + a(2,5) * var(:,nj-1,:) + & + a(3,5) * var(:,nj-2,:) + a(4,5) * var(:,nj-3,:) + & + a(5,5) * var(:,nj-4,:) + a(6,5) * var(:,nj-5,:) + & + a(7,5) * var(:,nj-6,:) + a(8,5) * var(:,nj-7,:) + & + a(9,5) * var(:,nj-8,:) + a(10,5) * var(:,nj-9,:) ) * idel + 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,:) + a(8,4) * var(:,nj-7,:) + & + a(9,4) * var(:,nj-8,:) ) * 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,:) + & + a(7,3) * var(:,nj-6,:) + a(8,3) * var(:,nj-7,:) ) * 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,:) + a(6,2) * var(:,nj-5,:) + & + a(7,2) * var(:,nj-6,:) ) * 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,:) + & + a(5,1) * var(:,nj-4,:) + a(6,1) * var(:,nj-5,:) ) * idel + + jr = nj - 8 + end if + rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & + ( -252.0_wp * var(:,jl:jr,:) + & + 210.0_wp * ( var(:,jl-1:jr-1,:) + & + var(:,jl+1:jr+1,:) ) - & + 120.0_wp * ( var(:,jl-2:jr-2,:) + & + var(:,jl+2:jr+2,:) ) + & + 45.0_wp * ( var(:,jl-3:jr-3,:) + & + var(:,jl+3:jr+3,:) ) - & + 10.0_wp * ( var(:,jl-4:jr-4,:) + & + var(:,jl+4:jr+4,:) ) + & + ( var(:,jl-5:jr-5,:) + & + var(:,jl+5:jr+5,:) ) ) * idel end if - rhs(:,jl:jr,:) = rhs(:,jl:jr,:) + & - ( -252.0_wp * var(:,jl:jr,:) + & - 210.0_wp * ( var(:,jl-1:jr-1,:) + & - var(:,jl+1:jr+1,:) ) - & - 120.0_wp * ( var(:,jl-2:jr-2,:) + & - var(:,jl+2:jr+2,:) ) + & - 45.0_wp * ( var(:,jl-3:jr-3,:) + & - var(:,jl+3:jr+3,:) ) - & - 10.0_wp * ( var(:,jl-4:jr-4,:) + & - var(:,jl+4:jr+4,:) ) + & - ( var(:,jl-5:jr-5,:) + & - var(:,jl+5:jr+5,:) ) ) * idel - - call set_coeff ( a ) - - idel = epsilon / ( 1024 * delta(3) ) - - 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) + & - a(5,1) * var(:,:,5) + a(6,1) * var(:,:,6) ) * 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) + a(8,3) * var(:,:,8) ) * 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) ) * 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) + & - a(11,6) * var(:,:,11) ) * idel - rhs(:,:,7) = rhs(:,:,7) + & - ( 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) + a(11,7) * var(:,:,11) + & - a(12,7) * var(:,:,12) ) * idel - rhs(:,:,8) = rhs(:,:,8) + & - ( a(3,8) * var(:,:,3) + a(4,8) * var(:,:,4) + & - a(5,8) * var(:,:,5) + a(6,8) * var(:,:,6) + & - a(7,8) * var(:,:,7) + a(8,8) * var(:,:,8) + & - a(9,8) * var(:,:,9) + a(10,8) * var(:,:,10) + & - a(11,8) * var(:,:,11) + a(12,8) * var(:,:,12) + & - a(13,8) * var(:,:,13) ) * idel - - kl = 9 - end if - if ( bb(6) == 0 ) then - kr = nk - gsize(3) - else - rhs(:,:,nk-7) = rhs(:,:,nk-7) + & - ( a(3,8) * var(:,:,nk-2) + a(4,8) * var(:,:,nk-3) + & - a(5,8) * var(:,:,nk-4) + a(6,8) * var(:,:,nk-5) + & - a(7,8) * var(:,:,nk-6) + a(8,8) * var(:,:,nk-7) + & - a(9,8) * var(:,:,nk-8) + a(10,8) * var(:,:,nk-9) + & - a(11,8) * var(:,:,nk-10) + a(12,8) * var(:,:,nk-11) + & - a(13,8) * var(:,:,nk-12) ) * idel - rhs(:,:,nk-6) = rhs(:,:,nk-6) + & - ( a(2,7) * var(:,:,nk-1) + a(3,7) * var(:,:,nk-2) + & - a(4,7) * var(:,:,nk-3) + a(5,7) * var(:,:,nk-4) + & - a(6,7) * var(:,:,nk-5) + a(7,7) * var(:,:,nk-6) + & - a(8,7) * var(:,:,nk-7) + a(9,7) * var(:,:,nk-8) + & - a(10,7) * var(:,:,nk-9) + a(11,7) * var(:,:,nk-10) + & - a(12,7) * var(:,:,nk-11) ) * idel - rhs(:,:,nk-5) = rhs(:,:,nk-5) + & - ( a(1,6) * var(:,:,nk) + a(2,6) * var(:,:,nk-1) + & - a(3,6) * var(:,:,nk-2) + a(4,6) * var(:,:,nk-3) + & - a(5,6) * var(:,:,nk-4) + a(6,6) * var(:,:,nk-5) + & - a(7,6) * var(:,:,nk-6) + a(8,6) * var(:,:,nk-7) + & - a(9,6) * var(:,:,nk-8) + a(10,6) * var(:,:,nk-9) + & - a(11,6) * var(:,:,nk-10) ) * idel - rhs(:,:,nk-4) = rhs(:,:,nk-4) + & - ( a(1,5) * var(:,:,nk) + a(2,5) * var(:,:,nk-1) + & - a(3,5) * var(:,:,nk-2) + a(4,5) * var(:,:,nk-3) + & - a(5,5) * var(:,:,nk-4) + a(6,5) * var(:,:,nk-5) + & - a(7,5) * var(:,:,nk-6) + a(8,5) * var(:,:,nk-7) + & - a(9,5) * var(:,:,nk-8) + a(10,5) * var(:,:,nk-9) ) * idel - 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) + a(8,4) * var(:,:,nk-7) + & - a(9,4) * var(:,:,nk-8) ) * 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) + & - a(7,3) * var(:,:,nk-6) + a(8,3) * var(:,:,nk-7) ) * 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) + a(6,2) * var(:,:,nk-5) + & - a(7,2) * var(:,:,nk-6) ) * 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) + & - a(5,1) * var(:,:,nk-4) + a(6,1) * var(:,:,nk-5) ) * idel - kr = nk - 8 + if ( zero_derivs_z == 0 ) then + call set_coeff ( a ) + + idel = epsilon / ( 1024 * delta(3) ) + + 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) + & + a(5,1) * var(:,:,5) + a(6,1) * var(:,:,6) ) * 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) + a(8,3) * var(:,:,8) ) * 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) ) * 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) + & + a(11,6) * var(:,:,11) ) * idel + rhs(:,:,7) = rhs(:,:,7) + & + ( 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) + a(11,7) * var(:,:,11) + & + a(12,7) * var(:,:,12) ) * idel + rhs(:,:,8) = rhs(:,:,8) + & + ( a(3,8) * var(:,:,3) + a(4,8) * var(:,:,4) + & + a(5,8) * var(:,:,5) + a(6,8) * var(:,:,6) + & + a(7,8) * var(:,:,7) + a(8,8) * var(:,:,8) + & + a(9,8) * var(:,:,9) + a(10,8) * var(:,:,10) + & + a(11,8) * var(:,:,11) + a(12,8) * var(:,:,12) + & + a(13,8) * var(:,:,13) ) * idel + + kl = 9 + end if + if ( bb(6) == 0 ) then + kr = nk - gsize(3) + else + rhs(:,:,nk-7) = rhs(:,:,nk-7) + & + ( a(3,8) * var(:,:,nk-2) + a(4,8) * var(:,:,nk-3) + & + a(5,8) * var(:,:,nk-4) + a(6,8) * var(:,:,nk-5) + & + a(7,8) * var(:,:,nk-6) + a(8,8) * var(:,:,nk-7) + & + a(9,8) * var(:,:,nk-8) + a(10,8) * var(:,:,nk-9) + & + a(11,8) * var(:,:,nk-10) + a(12,8) * var(:,:,nk-11) + & + a(13,8) * var(:,:,nk-12) ) * idel + rhs(:,:,nk-6) = rhs(:,:,nk-6) + & + ( a(2,7) * var(:,:,nk-1) + a(3,7) * var(:,:,nk-2) + & + a(4,7) * var(:,:,nk-3) + a(5,7) * var(:,:,nk-4) + & + a(6,7) * var(:,:,nk-5) + a(7,7) * var(:,:,nk-6) + & + a(8,7) * var(:,:,nk-7) + a(9,7) * var(:,:,nk-8) + & + a(10,7) * var(:,:,nk-9) + a(11,7) * var(:,:,nk-10) + & + a(12,7) * var(:,:,nk-11) ) * idel + rhs(:,:,nk-5) = rhs(:,:,nk-5) + & + ( a(1,6) * var(:,:,nk) + a(2,6) * var(:,:,nk-1) + & + a(3,6) * var(:,:,nk-2) + a(4,6) * var(:,:,nk-3) + & + a(5,6) * var(:,:,nk-4) + a(6,6) * var(:,:,nk-5) + & + a(7,6) * var(:,:,nk-6) + a(8,6) * var(:,:,nk-7) + & + a(9,6) * var(:,:,nk-8) + a(10,6) * var(:,:,nk-9) + & + a(11,6) * var(:,:,nk-10) ) * idel + rhs(:,:,nk-4) = rhs(:,:,nk-4) + & + ( a(1,5) * var(:,:,nk) + a(2,5) * var(:,:,nk-1) + & + a(3,5) * var(:,:,nk-2) + a(4,5) * var(:,:,nk-3) + & + a(5,5) * var(:,:,nk-4) + a(6,5) * var(:,:,nk-5) + & + a(7,5) * var(:,:,nk-6) + a(8,5) * var(:,:,nk-7) + & + a(9,5) * var(:,:,nk-8) + a(10,5) * var(:,:,nk-9) ) * idel + 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) + a(8,4) * var(:,:,nk-7) + & + a(9,4) * var(:,:,nk-8) ) * 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) + & + a(7,3) * var(:,:,nk-6) + a(8,3) * var(:,:,nk-7) ) * 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) + a(6,2) * var(:,:,nk-5) + & + a(7,2) * var(:,:,nk-6) ) * 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) + & + a(5,1) * var(:,:,nk-4) + a(6,1) * var(:,:,nk-5) ) * idel + + kr = nk - 8 + end if + rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & + ( -252.0_wp * var(:,:,kl:kr) + & + 210.0_wp * ( var(:,:,kl-1:kr-1) + & + var(:,:,kl+1:kr+1) ) - & + 120.0_wp * ( var(:,:,kl-2:kr-2) + & + var(:,:,kl+2:kr+2) ) + & + 45.0_wp * ( var(:,:,kl-3:kr-3) + & + var(:,:,kl+3:kr+3) ) - & + 10.0_wp * ( var(:,:,kl-4:kr-4) + & + var(:,:,kl+4:kr+4) ) + & + ( var(:,:,kl-5:kr-5) + & + var(:,:,kl+5:kr+5) ) ) * idel end if - rhs(:,:,kl:kr) = rhs(:,:,kl:kr) + & - ( -252.0_wp * var(:,:,kl:kr) + & - 210.0_wp * ( var(:,:,kl-1:kr-1) + & - var(:,:,kl+1:kr+1) ) - & - 120.0_wp * ( var(:,:,kl-2:kr-2) + & - var(:,:,kl+2:kr+2) ) + & - 45.0_wp * ( var(:,:,kl-3:kr-3) + & - var(:,:,kl+3:kr+3) ) - & - 10.0_wp * ( var(:,:,kl-4:kr-4) + & - var(:,:,kl+4:kr+4) ) + & - ( var(:,:,kl-5:kr-5) + & - var(:,:,kl+5:kr+5) ) ) * idel contains |