diff options
Diffstat (limited to 'src/Derivatives_2_1.F90')
| -rw-r--r-- | src/Derivatives_2_1.F90 | 70 |
1 files changed, 40 insertions, 30 deletions
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 |