opr_sx.f90

Go to the documentation of this file.

 
module opr_sx
  use gather_scatter, only : gs_t, gs_op_add
  use interpolation, only : interpolator_t
  use num_types, only : rp
  use space, only : space_t
  use coefs, only : coef_t
  use math, only : sub3, copy, rzero
  use mathops, only : opcolv
  implicit none
  private
 
  public :: opr_sx_dudxyz, opr_sx_opgrad, opr_sx_cdtp, opr_sx_conv1, &
       opr_sx_curl, opr_sx_cfl, opr_sx_lambda2, opr_sx_convect_scalar, &
       opr_sx_set_convect_rst
 
 
  interface
     module subroutine opr_sx_dudxyz(du, u, dr, ds, dt, coef)
       type(coef_t), intent(in), target :: coef
       real(kind=rp), intent(inout), &
            dimension(coef%Xh%lx, coef%Xh%ly, coef%Xh%lz, coef%msh%nelv) :: du
       real(kind=rp), intent(in), &
            dimension(coef%Xh%lx, coef%Xh%ly, coef%Xh%lz, coef%msh%nelv) :: &
            u, dr, ds, dt
     end subroutine opr_sx_dudxyz
 
     module subroutine opr_sx_opgrad(ux, uy, uz, u, coef)
       type(coef_t), intent(in) :: coef
       real(kind=rp), intent(inout) :: ux(coef%Xh%lxyz, coef%msh%nelv)
       real(kind=rp), intent(inout) :: uy(coef%Xh%lxyz, coef%msh%nelv)
       real(kind=rp), intent(inout) :: uz(coef%Xh%lxyz, coef%msh%nelv)
       real(kind=rp), intent(in) :: u(coef%Xh%lxyz, coef%msh%nelv)
     end subroutine opr_sx_opgrad
 
     module subroutine opr_sx_cdtp(dtx, x, dr, ds, dt, coef)
       type(coef_t), intent(in) :: coef
       real(kind=rp), intent(inout) :: dtx(coef%Xh%lxyz, coef%msh%nelv)
       real(kind=rp), intent(inout) :: x(coef%Xh%lxyz, coef%msh%nelv)
       real(kind=rp), intent(in) :: dr(coef%Xh%lxyz, coef%msh%nelv)
       real(kind=rp), intent(in) :: ds(coef%Xh%lxyz, coef%msh%nelv)
       real(kind=rp), intent(in) :: dt(coef%Xh%lxyz, coef%msh%nelv)
     end subroutine opr_sx_cdtp
 
     module subroutine opr_sx_conv1(du, u, vx, vy, vz, xh, coef, nelv)
       type(space_t), intent(in) :: Xh
       type(coef_t), intent(in) :: coef
       integer, intent(in) :: nelv
       real(kind=rp), intent(inout) :: du(xh%lxyz, nelv)
       real(kind=rp), intent(in) :: u(xh%lx, xh%ly, xh%lz, nelv)
       real(kind=rp), intent(in) :: vx(xh%lx, xh%ly, xh%lz, nelv)
       real(kind=rp), intent(in) :: vy(xh%lx, xh%ly, xh%lz, nelv)
       real(kind=rp), intent(in) :: vz(xh%lx, xh%ly, xh%lz, nelv)
     end subroutine opr_sx_conv1
 
     module subroutine opr_sx_convect_scalar(du, u, cr, cs, ct, xh_gll, xh_gl, &
          coef_gll, coef_gl, gll_to_gl)
       type(space_t), intent(in) :: Xh_GL
       type(space_t), intent(in) :: Xh_GLL
       type(coef_t), intent(in) :: coef_GLL
       type(coef_t), intent(in) :: coef_GL
       type(interpolator_t), intent(inout) :: GLL_to_GL
       real(kind=rp), intent(inout) :: &
            du(xh_gll%lx, xh_gll%ly, xh_gll%lz, coef_gl%msh%nelv)
       real(kind=rp), intent(inout) :: &
            u(xh_gl%lx, xh_gl%lx, xh_gl%lx, coef_gl%msh%nelv)
       real(kind=rp), intent(inout) :: cr(xh_gl%lxyz, coef_gl%msh%nelv)
       real(kind=rp), intent(inout) :: cs(xh_gl%lxyz, coef_gl%msh%nelv)
       real(kind=rp), intent(inout) :: ct(xh_gl%lxyz, coef_gl%msh%nelv)
 
     end subroutine opr_sx_convect_scalar
 
     module function opr_sx_cfl(dt, u, v, w, xh, coef, nelv) result(cfl)
       type(space_t), intent(in) :: Xh
       type(coef_t), intent(in) :: coef
       integer, intent(in) :: nelv
       real(kind=rp), intent(in) :: dt
       real(kind=rp), dimension(Xh%lx, Xh%ly, Xh%lz, nelv) :: u, v, w
       real(kind=rp) :: cfl
     end function opr_sx_cfl
 
     module subroutine opr_sx_lambda2(lambda2, u, v, w, coef)
       type(coef_t), intent(in) :: coef
       real(kind=rp), intent(inout), &
            dimension(coef%Xh%lx, coef%Xh%ly, coef%Xh%lz, coef%msh%nelv) :: lambda2
       real(kind=rp), intent(in), &
            dimension(coef%Xh%lx, coef%Xh%ly, coef%Xh%lz, coef%msh%nelv) :: u, v, w
     end subroutine opr_sx_lambda2
 
     module subroutine opr_sx_set_convect_rst(cr, cs, ct, cx, cy, cz, xh, coef)
       type(space_t), intent(inout) :: Xh
       type(coef_t), intent(inout) :: coef
       real(kind=rp), dimension(Xh%lxyz, coef%msh%nelv), &
            intent(inout) :: cr, cs, ct
       real(kind=rp), dimension(Xh%lxyz, coef%msh%nelv), &
            intent(in) :: cx, cy, cz
     end subroutine opr_sx_set_convect_rst
  end interface
 
contains
 
  subroutine opr_sx_curl(w1, w2, w3, u1, u2, u3, work1, work2, c_Xh)
    type(coef_t), intent(in) :: c_xh
    real(kind=rp), intent(inout), &
         dimension(c_Xh%Xh%lx, c_Xh%Xh%ly, c_Xh%Xh%lz, c_Xh%msh%nelv) :: w1
    real(kind=rp), intent(inout), &
         dimension(c_Xh%Xh%lx, c_Xh%Xh%ly, c_Xh%Xh%lz, c_Xh%msh%nelv) :: w2
    real(kind=rp), intent(inout), &
         dimension(c_Xh%Xh%lx, c_Xh%Xh%ly, c_Xh%Xh%lz, c_Xh%msh%nelv) :: w3
    real(kind=rp), intent(in), &
         dimension(c_Xh%Xh%lx, c_Xh%Xh%ly, c_Xh%Xh%lz, c_Xh%msh%nelv) :: u1
    real(kind=rp), intent(in), &
         dimension(c_Xh%Xh%lx, c_Xh%Xh%ly, c_Xh%Xh%lz, c_Xh%msh%nelv) :: u2
    real(kind=rp), intent(in), &
         dimension(c_Xh%Xh%lx, c_Xh%Xh%ly, c_Xh%Xh%lz, c_Xh%msh%nelv) :: u3
    real(kind=rp), intent(inout), &
         dimension(c_Xh%Xh%lx, c_Xh%Xh%ly, c_Xh%Xh%lz, c_Xh%msh%nelv) :: work1
    real(kind=rp), intent(inout), &
         dimension(c_Xh%Xh%lx, c_Xh%Xh%ly, c_Xh%Xh%lz, c_Xh%msh%nelv) :: work2
    integer :: gdim, n
 
    n = c_xh%dof%size()
    gdim = c_xh%msh%gdim
 
    !     this%work1=dw/dy ; this%work2=dv/dz
    call opr_sx_dudxyz(work1, u3, c_xh%drdy, c_xh%dsdy, c_xh%dtdy, c_xh)
    if (gdim .eq. 3) then
       call opr_sx_dudxyz(work2, u2, c_xh%drdz, c_xh%dsdz, c_xh%dtdz, c_xh)
       call sub3(w1, work1, work2, n)
    else
       call copy(w1, work1, n)
    end if
    !     this%work1=du/dz ; this%work2=dw/dx
    if (gdim .eq. 3) then
       call opr_sx_dudxyz(work1, u1, c_xh%drdz, c_xh%dsdz, c_xh%dtdz, c_xh)
       call opr_sx_dudxyz(work2, u3, c_xh%drdx, c_xh%dsdx, c_xh%dtdx, c_xh)
       call sub3(w2, work1, work2, n)
    else
       call rzero (work1, n)
       call opr_sx_dudxyz(work2, u3, c_xh%drdx, c_xh%dsdx, c_xh%dtdx, c_xh)
       call sub3(w2, work1, work2, n)
    end if
    !     this%work1=dv/dx ; this%work2=du/dy
    call opr_sx_dudxyz(work1, u2, c_xh%drdx, c_xh%dsdx, c_xh%dtdx, c_xh)
    call opr_sx_dudxyz(work2, u1, c_xh%drdy, c_xh%dsdy, c_xh%dtdy, c_xh)
    call sub3(w3, work1, work2, n)
    !!    BC dependent, Needs to change if cyclic
 
    call opcolv(w1, w2, w3, c_xh%B, gdim, n)
    call c_xh%gs_h%op(w1, n, gs_op_add)
    call c_xh%gs_h%op(w2, n, gs_op_add)
    call c_xh%gs_h%op(w3, n, gs_op_add)
    call opcolv(w1, w2, w3, c_xh%Binv, gdim, n)
 
  end subroutine opr_sx_curl
 
 
 
 
 
end module opr_sx